diff --git a/internal/service/pos/pos_mock.go b/internal/service/pos/pos_mock.go index 2021089b..2cf30251 100644 --- 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) }