code cleanup

Author: 0xh3rman

Features/Swap/Sources/ViewModels/SwapSceneViewModel.swift

@@ -346,13 +346,7 @@ extension SwapSceneViewModel {
             var accumulated: [SwapperQuote] = []
             var errors: [Error] = []
 
-            for await result in swapQuotesProvider.fetchQuotes(
-                wallet: wallet,
-                fromAsset: input.fromAsset,
-                toAsset: input.toAsset,
-                amount: input.value,
-                useMaxAmount: input.useMaxAmount
-            ) {
+            for await result in swapQuotesProvider.fetchQuotes(wallet: wallet, input: input) {
                 try Task.checkCancellation()
 
                 switch result {

Features/Swap/Tests/SwapTests/SwapSceneViewModelTests.swift

@@ -138,7 +138,6 @@ struct SwapSceneViewModelTests {
         toValueMock: String = "250000000000"
     ) async -> SwapSceneViewModel {
         let swapper = GemSwapperMock(
-            quotes: [.mock(toValue: toValueMock)],
             quoteByProvider: .mock(toValue: toValueMock)
         )
         let model = SwapSceneViewModel.mock(swapper: swapper)

Packages/FeatureServices/SwapService/SwapQuotesProvider.swift

@@ -1,7 +1,5 @@
 // Copyright (c). Gem Wallet. All rights reserved.
 
-import Foundation
-import BigInt
 import Foundation
 import Primitives
 
@@ -10,7 +8,7 @@ import struct Gemstone.SwapperProviderType
 
 public protocol SwapQuotesProvidable: Sendable {
     func supportedAssets(for assetId: AssetId) -> ([Primitives.Chain], [Primitives.AssetId])
-    func fetchQuotes(wallet: Wallet, fromAsset: Asset, toAsset: Asset, amount: BigInt, useMaxAmount: Bool) -> AsyncStream<Result<SwapperQuote, Error>>
+    func fetchQuotes(wallet: Wallet, input: SwapQuoteInput) -> AsyncStream<Result<SwapperQuote, Error>>
 }
 
 public struct SwapQuotesProvider: SwapQuotesProvidable {
@@ -24,13 +22,12 @@ public struct SwapQuotesProvider: SwapQuotesProvidable {
         swapService.supportedAssets(for: assetId)
     }
 
-    public func fetchQuotes(wallet: Wallet, fromAsset: Asset, toAsset: Asset, amount: BigInt, useMaxAmount: Bool) -> AsyncStream<Result<SwapperQuote, Error>> {
+    public func fetchQuotes(wallet: Wallet, input: SwapQuoteInput) -> AsyncStream<Result<SwapperQuote, Error>> {
         AsyncStream { continuation in
             Task {
                 do {
-                    let input = SwapQuoteInput(fromAsset: fromAsset, toAsset: toAsset, value: amount, useMaxAmount: useMaxAmount)
-                    let walletAddress = try wallet.account(for: fromAsset.chain).address
-                    let destinationAddress = try wallet.account(for: toAsset.chain).address
+                    let walletAddress = try wallet.account(for: input.fromAsset.chain).address
+                    let destinationAddress = try wallet.account(for: input.toAsset.chain).address
                     let providers = try swapService.getProvidersForQuote(input: input, walletAddress: walletAddress, destinationAddress: destinationAddress)
                     await fetchFromProviders(providers, input: input, walletAddress: walletAddress, destinationAddress: destinationAddress, continuation: continuation)
                 } catch {

Packages/FeatureServices/SwapService/TestKit/GemSwapperMock.swift

@@ -15,7 +15,6 @@ import struct Gemstone.SwapperSwapResult
 
 public final class GemSwapperMock: GemSwapperProtocol {
     private let permit2ForQuote: Permit2ApprovalData
-    private let quotes: [SwapperQuote]
     private let quoteByProvider: SwapperQuote
     private let quoteData: GemSwapQuoteData
     private let providers: [SwapperProviderType]
@@ -28,7 +27,6 @@ public final class GemSwapperMock: GemSwapperProtocol {
 
     public init(
         permit2ForQuote: Permit2ApprovalData = .mock(),
-        quotes: [SwapperQuote] = [.mock()],
         quoteByProvider: SwapperQuote = .mock(),
         quoteData: GemSwapQuoteData = .mock(),
         providers: [SwapperProviderType] = [.mock()],
@@ -40,7 +38,6 @@ public final class GemSwapperMock: GemSwapperProtocol {
         fetchQuoteError: Error? = nil
     ) {
         self.permit2ForQuote = permit2ForQuote
-        self.quotes = quotes
         self.quoteByProvider = quoteByProvider
         self.quoteData = quoteData
         self.providers = providers
@@ -63,7 +60,7 @@ public final class GemSwapperMock: GemSwapperProtocol {
         if let error = fetchQuoteError {
             throw error
         }
-        return quotes
+        return [quoteByProvider]
     }
 
     public func fetchQuoteByProvider(provider: SwapperProvider, request: SwapperQuoteRequest) async throws -> SwapperQuote {