✅ test: fix mock modules

TimeleapLabs · Jun 3, 2024 · 1fa5ac6 · 1fa5ac6
1 parent 1755610
commit 1fa5ac6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/service/pos/pos_mock.go b/internal/service/pos/pos_mock.go
@@ -27,8 +27,8 @@ func (m *MockService) GetVotingPower(address [20]byte, block *big.Int) (*big.Int
 	return big.NewInt(int64(args.Int(0))), args.Error(1)
 }
 
-func (m *MockService) GetVotingPowerOfEvm(address [20]byte, block *big.Int) (*big.Int, error) {
-	args := m.Called(address, block)
+func (m *MockService) GetVotingPowerOfEvm(ctx context.Context, evmAddress string) (*big.Int, error) {
+	args := m.Called(ctx, evmAddress)
 	return big.NewInt(int64(args.Int(0))), args.Error(1)
 }