@portal-hq/web 3.6.2-alpha → 3.7.0

package/src/integrations/trading/zero-x/index.test.ts

@@ -7,10 +7,14 @@ import Mpc from '../../../mpc'
 import portalMock from '../../../__mocks/portal/portal'
 import {
   mockHost,
-  mockQuoteArgs,
   mockSourcesRes,
-  mockZeroXQuoteResponse,
+  mockZeroExQuoteV2Request,
+  mockZeroExQuoteV2Response,
+  mockZeroExOptions,
+  mockZeroExPriceRequest,
+  mockZeroExPriceResponse,
 } from '../../../__mocks/constants'
+import type { ZeroExPriceResponse } from '../../../../zero-x'
 
 describe('ZeroX', () => {
   let zeroX: ZeroX
@@ -28,46 +32,121 @@ describe('ZeroX', () => {
   })
 
   describe('getQuote', () => {
-    it('should correctly call mpc.getQuote', async () => {
+    it('should correctly call mpc.getSwapsQuoteV2 with args and options', async () => {
       const spy = jest
-        .spyOn(mpc, 'getQuote')
-        .mockResolvedValue(mockZeroXQuoteResponse)
+        .spyOn(mpc, 'getSwapsQuoteV2')
+        .mockResolvedValue(mockZeroExQuoteV2Response)
 
-      const result = await zeroX.getQuote('test', mockQuoteArgs, 'eip155:1')
+      const result = await zeroX.getQuote(
+        mockZeroExQuoteV2Request,
+        mockZeroExOptions,
+      )
+
+      expect(spy).toHaveBeenCalledTimes(1)
+      expect(spy).toHaveBeenCalledWith(
+        mockZeroExQuoteV2Request,
+        mockZeroExOptions,
+      )
+      expect(result).toEqual(mockZeroExQuoteV2Response)
+    })
+
+    it('should correctly call mpc.getSwapsQuoteV2 without options', async () => {
+      const spy = jest
+        .spyOn(mpc, 'getSwapsQuoteV2')
+        .mockResolvedValue(mockZeroExQuoteV2Response)
+
+      const result = await zeroX.getQuote(mockZeroExQuoteV2Request)
 
       expect(spy).toHaveBeenCalledTimes(1)
-      expect(spy).toHaveBeenCalledWith('test', mockQuoteArgs, 'eip155:1')
-      expect(result).toEqual(mockZeroXQuoteResponse)
+      expect(spy).toHaveBeenCalledWith(mockZeroExQuoteV2Request, undefined)
+      expect(result).toEqual(mockZeroExQuoteV2Response)
     })
 
-    it('should propagate errors from mpc.getQuote', async () => {
+    it('should propagate errors from mpc.getSwapsQuoteV2', async () => {
       const error = new Error('Test error')
-      jest.spyOn(mpc, 'getQuote').mockRejectedValue(error)
+      jest.spyOn(mpc, 'getSwapsQuoteV2').mockRejectedValue(error)
 
-      await expect(
-        zeroX.getQuote('test', mockQuoteArgs, 'eip155:1'),
-      ).rejects.toThrow('Test error')
+      await expect(zeroX.getQuote(mockZeroExQuoteV2Request)).rejects.toThrow(
+        'Test error',
+      )
     })
   })
 
   describe('getSources', () => {
-    it('should correctly call mpc.getSources', async () => {
+    it('should correctly call mpc.getSources with chainId and apiKey', async () => {
+      const spy = jest
+        .spyOn(mpc, 'getSwapsSourcesV2')
+        .mockResolvedValue(mockSourcesRes)
+
+      const result = await zeroX.getSources('eip155:1', {
+        zeroXApiKey: 'test-api-key',
+      })
+
+      console.log(`Result:`, result)
+      expect(spy).toHaveBeenCalledTimes(1)
+      expect(spy).toHaveBeenCalledWith(
+        { chainId: 'eip155:1' },
+        { zeroXApiKey: 'test-api-key' },
+      )
+      expect(result).toEqual(mockSourcesRes)
+    })
+
+    it('should correctly call mpc.getSources without apiKey', async () => {
       const spy = jest
-        .spyOn(mpc, 'getSources')
+        .spyOn(mpc, 'getSwapsSourcesV2')
         .mockResolvedValue(mockSourcesRes)
 
-      const result = await zeroX.getSources('test', 'eip155:1')
+      const result = await zeroX.getSources('eip155:1')
 
       expect(spy).toHaveBeenCalledTimes(1)
-      expect(spy).toHaveBeenCalledWith('test', 'eip155:1')
+      expect(spy).toHaveBeenCalledWith({ chainId: 'eip155:1' }, undefined)
       expect(result).toEqual(mockSourcesRes)
     })
 
     it('should propagate errors from mpc.getSources', async () => {
       const error = new Error('Test error')
-      jest.spyOn(mpc, 'getSources').mockRejectedValue(error)
+      jest.spyOn(mpc, 'getSwapsSourcesV2').mockRejectedValue(error)
+
+      await expect(zeroX.getSources('eip155:1')).rejects.toThrow('Test error')
+    })
+  })
+
+  describe('getPrice', () => {
+    it('should correctly call mpc.getSwapsPrice with args and options', async () => {
+      const spy = jest
+        .spyOn(mpc, 'getSwapsPrice')
+        .mockResolvedValue(mockZeroExPriceResponse as ZeroExPriceResponse)
+
+      const result = await zeroX.getPrice(
+        mockZeroExPriceRequest,
+        mockZeroExOptions,
+      )
+
+      expect(spy).toHaveBeenCalledTimes(1)
+      expect(spy).toHaveBeenCalledWith(
+        mockZeroExPriceRequest,
+        mockZeroExOptions,
+      )
+      expect(result).toEqual(mockZeroExPriceResponse)
+    })
+
+    it('should correctly call mpc.getSwapsPrice without options', async () => {
+      const spy = jest
+        .spyOn(mpc, 'getSwapsPrice')
+        .mockResolvedValue(mockZeroExPriceResponse as ZeroExPriceResponse)
+
+      const result = await zeroX.getPrice(mockZeroExPriceRequest)
+
+      expect(spy).toHaveBeenCalledTimes(1)
+      expect(spy).toHaveBeenCalledWith(mockZeroExPriceRequest, undefined)
+      expect(result).toEqual(mockZeroExPriceResponse)
+    })
+
+    it('should propagate errors from mpc.getSwapsPrice', async () => {
+      const error = new Error('Test error')
+      jest.spyOn(mpc, 'getSwapsPrice').mockRejectedValue(error)
 
-      await expect(zeroX.getSources('test', 'eip155:1')).rejects.toThrow(
+      await expect(zeroX.getPrice(mockZeroExPriceRequest)).rejects.toThrow(
         'Test error',
       )
     })

package/src/integrations/trading/zero-x/index.ts

@@ -1,5 +1,12 @@
 import Mpc from '../../../mpc'
-import { QuoteArgs, QuoteResponse } from '../../../../types'
+import {
+  ZeroExOptions,
+  ZeroExPriceRequest,
+  ZeroExPriceResponse,
+  ZeroExQuoteRequest,
+  ZeroExQuoteResponse,
+  ZeroExSourcesResponse,
+} from '../../../../zero-x'
 
 export default class ZeroX {
   private mpc: Mpc
@@ -17,11 +24,10 @@ export default class ZeroX {
    * @returns The quote response.
    */
   public async getQuote(
-    apiKey: string,
-    args: QuoteArgs,
-    chainId: string,
-  ): Promise<QuoteResponse> {
-    return this.mpc?.getQuote(apiKey, args, chainId)
+    args: ZeroExQuoteRequest,
+    options?: ZeroExOptions,
+  ): Promise<ZeroExQuoteResponse> {
+    return this.mpc?.getSwapsQuoteV2(args, options)
   }
 
   /**
@@ -32,9 +38,23 @@ export default class ZeroX {
    * @returns The sources response.
    */
   public async getSources(
-    apiKey: string,
     chainId: string,
-  ): Promise<Record<string, string>> {
-    return this.mpc?.getSources(apiKey, chainId)
+    options?: ZeroExOptions,
+  ): Promise<ZeroExSourcesResponse> {
+    return this.mpc?.getSwapsSourcesV2({ chainId }, options)
+  }
+
+  /**
+   * Get the price of a token from the Swaps API.
+   *
+   * @param args - The arguments for the price.
+   * @param options - The options for the price.
+   * @returns The price response.
+   */
+  public async getPrice(
+    args: ZeroExPriceRequest,
+    options?: ZeroExOptions,
+  ): Promise<ZeroExPriceResponse> {
+    return this.mpc?.getSwapsPrice(args, options)
   }
 }