diff --git a/remote/account_service.go b/remote/account_service.go index 1ea0af7..5bf65a8 100644 --- a/remote/account_service.go +++ b/remote/account_service.go @@ -27,4 +27,5 @@ type AccountService interface { GetPoolSwapQuote(ctx context.Context, amount, from, fromTokenAddress, poolAddress, toTokenAddress string) (*models.PoolSwapQuoteResult, error) PoolSwap(ctx context.Context, amount, from, fromTokenAddress, poolAddress, toTokenAddress string) (*models.PoolSwapResult, error) GetSwapFromTokenMaxLimit(ctx context.Context, poolAddress, fromTokenAddress, toTokenAddress, publicKey string) (*models.MaxLimitResult, error) + CheckTokenInPool(ctx context.Context, poolAddress, tokenAddress string) (*models.TokenInPoolResult, error) } diff --git a/testutil/mocks/service_mock.go b/testutil/mocks/service_mock.go index aaa78bb..de19d54 100644 --- a/testutil/mocks/service_mock.go +++ b/testutil/mocks/service_mock.go @@ -105,3 +105,8 @@ func (m MockAccountService) GetSwapFromTokenMaxLimit(ctx context.Context, poolAd args := m.Called(poolAddress, fromTokenAddress, toTokenAddress, publicKey) return args.Get(0).(*models.MaxLimitResult), args.Error(1) } + +func (m MockAccountService) CheckTokenInPool(ctx context.Context, poolAddress, tokenAddress string) (*models.TokenInPoolResult, error) { + args := m.Called(poolAddress, tokenAddress) + return args.Get(0).(*models.TokenInPoolResult), args.Error(1) +} diff --git a/testutil/testservice/account_service.go b/testutil/testservice/account_service.go index 00906c0..46abb1a 100644 --- a/testutil/testservice/account_service.go +++ b/testutil/testservice/account_service.go @@ -104,3 +104,7 @@ func (m TestAccountService) PoolSwap(ctx context.Context, amount, from, fromToke func (m TestAccountService) GetSwapFromTokenMaxLimit(ctx context.Context, poolAddress, fromTokenAddress, toTokenAddress, publicKey string) (*models.MaxLimitResult, error) { return &models.MaxLimitResult{}, nil } + +func (m TestAccountService) CheckTokenInPool(ctx context.Context, poolAddress, tokenAddress string) (*models.TokenInPoolResult, error) { + return &models.TokenInPoolResult{}, nil +}