From ba43610ff00bec4eeb50078833badff2d1032f8b Mon Sep 17 00:00:00 2001 From: alfred-mk Date: Thu, 13 Mar 2025 10:14:11 +0300 Subject: [PATCH] Added pool swap functions to the MockAccountService --- testutil/mocks/service_mock.go | 35 ++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) diff --git a/testutil/mocks/service_mock.go b/testutil/mocks/service_mock.go index bfa9222..42bcb28 100644 --- a/testutil/mocks/service_mock.go +++ b/testutil/mocks/service_mock.go @@ -57,3 +57,38 @@ func (m MockAccountService) RequestAlias(ctx context.Context, publicKey string, args := m.Called(publicKey, hint) return args.Get(0).(*models.RequestAliasResult), args.Error(1) } + +func (m MockAccountService) PoolDeposit(ctx context.Context, amount, from, poolAddress, tokenAddress string) (*models.PoolDepositResult, error) { + args := m.Called(amount, from, poolAddress, tokenAddress) + return args.Get(0).(*models.PoolDepositResult), args.Error(1) +} + +func (m MockAccountService) FetchTopPools(ctx context.Context) ([]dataserviceapi.PoolDetails, error) { + args := m.Called() + return args.Get(0).([]dataserviceapi.PoolDetails), args.Error(1) +} + +func (m MockAccountService) GetPoolSwappableFromVouchers(ctx context.Context, poolAddress, publicKey string) ([]dataserviceapi.TokenHoldings, error) { + args := m.Called(poolAddress, publicKey) + return args.Get(0).([]dataserviceapi.TokenHoldings), args.Error(1) +} + +func (m MockAccountService) GetPoolSwappableVouchers(ctx context.Context, poolAddress, publicKey string) ([]dataserviceapi.TokenHoldings, error) { + args := m.Called(poolAddress, publicKey) + return args.Get(0).([]dataserviceapi.TokenHoldings), args.Error(1) +} + +func (m MockAccountService) GetPoolSwapQuote(ctx context.Context, amount, from, fromTokenAddress, poolAddress, toTokenAddress string) (*models.PoolSwapQuoteResult, error) { + args := m.Called(amount, from, fromTokenAddress, poolAddress, toTokenAddress) + return args.Get(0).(*models.PoolSwapQuoteResult), args.Error(1) +} + +func (m MockAccountService) PoolSwap(ctx context.Context, amount, from, fromTokenAddress, poolAddress, toTokenAddress string) (*models.PoolSwapResult, error) { + args := m.Called(amount, from, fromTokenAddress, poolAddress, toTokenAddress) + return args.Get(0).(*models.PoolSwapResult), args.Error(1) +} + +func (m MockAccountService) GetSwapFromTokenMaxLimit(ctx context.Context, poolAddress, fromTokenAddress, toTokenAddress, publicKey string) (*models.MaxLimitResult, error) { + args := m.Called(poolAddress, fromTokenAddress, toTokenAddress, publicKey) + return args.Get(0).(*models.MaxLimitResult), args.Error(1) +}