From 389ae1ac7472abd4e7a6d83096ea6bc1bc9b7835 Mon Sep 17 00:00:00 2001
From: JD <jann.driessen@gmx.de>
Date: Thu, 23 May 2024 18:04:04 +0200
Subject: [PATCH] test: fix zeroex provider tests

---
 src/quote/zeroEx/provider.test.ts | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/src/quote/zeroEx/provider.test.ts b/src/quote/zeroEx/provider.test.ts
index 9af80c4e..9bcec1ce 100644
--- a/src/quote/zeroEx/provider.test.ts
+++ b/src/quote/zeroEx/provider.test.ts
@@ -1,11 +1,15 @@
 import 'dotenv/config'
 
-import { LocalhostProvider, QuoteTokens, ZeroExApiSwapQuote } from 'tests/utils'
+import {
+  IndexZeroExSwapQuoteProvider,
+  LocalhostProvider,
+  QuoteTokens,
+} from 'tests/utils'
 import { wei } from 'utils'
 import { ZeroExQuoteProvider } from './provider'
 
 const provider = LocalhostProvider
-const zeroExApi = ZeroExApiSwapQuote
+const swapQuoteProvider = IndexZeroExSwapQuoteProvider
 
 const { dpi, dseth, eth, mvi } = QuoteTokens
 
@@ -18,7 +22,7 @@ describe('ZeroExQuoteProvider', () => {
     const indexTokenAmount = wei(1)
     const inputToken = eth
     const outputToken = dseth
-    const quoteProvider = new ZeroExQuoteProvider(provider, zeroExApi)
+    const quoteProvider = new ZeroExQuoteProvider(provider, swapQuoteProvider)
     const quote = await quoteProvider.getQuote({
       isMinting: true,
       inputToken,
@@ -37,7 +41,7 @@ describe('ZeroExQuoteProvider', () => {
     const indexTokenAmount = wei(1)
     const inputToken = dseth
     const outputToken = eth
-    const quoteProvider = new ZeroExQuoteProvider(provider, zeroExApi)
+    const quoteProvider = new ZeroExQuoteProvider(provider, swapQuoteProvider)
     const quote = await quoteProvider.getQuote({
       isMinting: false,
       inputToken,
@@ -67,7 +71,7 @@ describe('ZeroExQuoteProvider', () => {
       indexTokenAmount: wei(1),
       slippage: 0.5,
     }
-    const quoteProvider = new ZeroExQuoteProvider(provider, zeroExApi)
+    const quoteProvider = new ZeroExQuoteProvider(provider, swapQuoteProvider)
     const quote = await quoteProvider.getQuote(request)
     if (!quote) fail()
     expect(quote.componentQuotes.length).toBeGreaterThan(0)
@@ -86,7 +90,7 @@ describe('ZeroExQuoteProvider', () => {
       indexTokenAmount: wei(1),
       slippage: 0.5,
     }
-    const quoteProvider = new ZeroExQuoteProvider(provider, zeroExApi)
+    const quoteProvider = new ZeroExQuoteProvider(provider, swapQuoteProvider)
     const quote = await quoteProvider.getQuote(request)
     if (!quote) fail()
     expect(quote.componentQuotes.length).toBeGreaterThan(0)
@@ -105,7 +109,7 @@ describe('ZeroExQuoteProvider', () => {
       indexTokenAmount: wei(1),
       slippage: 0.5,
     }
-    const quoteProvider = new ZeroExQuoteProvider(provider, zeroExApi)
+    const quoteProvider = new ZeroExQuoteProvider(provider, swapQuoteProvider)
     const quote = await quoteProvider.getQuote(request)
     if (!quote) fail()
     expect(quote.componentQuotes.length).toBeGreaterThan(0)
@@ -124,7 +128,7 @@ describe('ZeroExQuoteProvider', () => {
       indexTokenAmount: wei(1),
       slippage: 0.5,
     }
-    const quoteProvider = new ZeroExQuoteProvider(provider, zeroExApi)
+    const quoteProvider = new ZeroExQuoteProvider(provider, swapQuoteProvider)
     const quote = await quoteProvider.getQuote(request)
     if (!quote) fail()
     expect(quote.componentQuotes.length).toBeGreaterThan(0)