test: add tests to cover new changes

MetaMask · Nov 22, 2024 · 6f3061d · 6f3061d
1 parent 3487bcd
commit 6f3061d
Show file tree

Hide file tree

Showing 2 changed files with 109 additions and 0 deletions.
diff --git a/packages/assets-controllers/src/TokenRatesController.test.ts b/packages/assets-controllers/src/TokenRatesController.test.ts
@@ -2348,6 +2348,55 @@ describe('TokenRatesController', () => {
         );
       });
 
+      it('correctly calls the Price API with unqiue native token addresses (e.g. MATIC)', async () => {
+        const tokenPricesService = buildMockTokenPricesService({
+          fetchTokenPrices: jest.fn().mockResolvedValue({
+            '0x0000000000000000000000000000000000001010': {
+              currency: 'MATIC',
+              tokenAddress: '0x0000000000000000000000000000000000001010',
+              value: 0.001,
+            },
+          }),
+        });
+
+        await withController(
+          {
+            options: { tokenPricesService },
+            mockNetworkClientConfigurationsByNetworkClientId: {
+              'AAAA-BBBB-CCCC-DDDD': buildCustomNetworkClientConfiguration({
+                chainId: '0x89',
+              }),
+            },
+          },
+          async ({
+            controller,
+            triggerTokensStateChange,
+            triggerNetworkStateChange,
+          }) => {
+            await callUpdateExchangeRatesMethod({
+              allTokens: {
+                '0x89': {
+                  [defaultSelectedAddress]: [],
+                },
+              },
+              chainId: '0x89',
+              controller,
+              triggerTokensStateChange,
+              triggerNetworkStateChange,
+              method,
+              nativeCurrency: 'MATIC',
+              selectedNetworkClientId: 'AAAA-BBBB-CCCC-DDDD',
+            });
+
+            expect(
+              controller.state.marketData['0x89'][
+                '0x0000000000000000000000000000000000001010'
+              ],
+            ).toBeDefined();
+          },
+        );
+      });
+
       it('only updates rates once when called twice', async () => {
         const tokenAddresses = [
           '0x0000000000000000000000000000000000000001',

diff --git a/packages/assets-controllers/src/token-prices-service/codefi-v2.test.ts b/packages/assets-controllers/src/token-prices-service/codefi-v2.test.ts
@@ -5,6 +5,8 @@ import {
   CodefiTokenPricesServiceV2,
   SUPPORTED_CHAIN_IDS,
   SUPPORTED_CURRENCIES,
+  ZERO_ADDRESS,
+  getNativeTokenAddress,
 } from './codefi-v2';
 
 // We're not customizing the default max delay
@@ -208,6 +210,51 @@ describe('CodefiTokenPricesServiceV2', () => {
       });
     });
 
+    it('calls the /spot-prices endpoint using the correct native token address', async () => {
+      const mockPriceAPI = nock('https://price.api.cx.metamask.io')
+        .get('/v2/chains/137/spot-prices')
+        .query({
+          tokenAddresses: '0x0000000000000000000000000000000000001010',
+          vsCurrency: 'ETH',
+          includeMarketData: 'true',
+        })
+        .reply(200, {
+          '0x0000000000000000000000000000000000001010': {
+            price: 14,
+            currency: 'ETH',
+            pricePercentChange1d: 1,
+            priceChange1d: 1,
+            marketCap: 117219.99428314982,
+            allTimeHigh: 0.00060467892389492,
+            allTimeLow: 0.00002303954000865728,
+            totalVolume: 5155.094053542448,
+            high1d: 0.00008020715848194385,
+            low1d: 0.00007792083564549064,
+            circulatingSupply: 1494269733.9526057,
+            dilutedMarketCap: 117669.5125951733,
+            marketCapPercentChange1d: 0.76671,
+            pricePercentChange1h: -1.0736342953259423,
+            pricePercentChange7d: -7.351582573655089,
+            pricePercentChange14d: -1.0799098946709822,
+            pricePercentChange30d: -25.776321124365992,
+            pricePercentChange200d: 46.091571238599165,
+            pricePercentChange1y: -2.2992517267242754,
+          },
+        });
+
+      const marketData =
+        await new CodefiTokenPricesServiceV2().fetchTokenPrices({
+          chainId: '0x89',
+          tokenAddresses: [],
+          currency: 'ETH',
+        });
+
+      expect(mockPriceAPI.isDone()).toBe(true);
+      expect(
+        marketData['0x0000000000000000000000000000000000001010'],
+      ).toBeDefined();
+    });
+
     it('should not include token price object for token address when token price in not included the response data', async () => {
       nock('https://price.api.cx.metamask.io')
         .get('/v2/chains/1/spot-prices')
@@ -1960,6 +2007,19 @@ describe('CodefiTokenPricesServiceV2', () => {
       ).toBe(false);
     });
   });
+
+  describe('getNativeTokenAddress', () => {
+    it('should return unique native token address for MATIC', () => {
+      expect(getNativeTokenAddress('0x89')).toBe(
+        '0x0000000000000000000000000000000000001010',
+      );
+    });
+    it('should return zero address for other chains', () => {
+      (['0x1', '0x2', '0x1337'] as const).forEach((chainId) => {
+        expect(getNativeTokenAddress(chainId)).toBe(ZERO_ADDRESS);
+      });
+    });
+  });
 });
 
 /**