feat: upgrade assets controllers to 42 with multichain token rates

app/components/UI/Tokens/index.test.tsx

@@ -40,7 +40,7 @@ jest.mock('../../../core/Engine', () => ({
       updateExchangeRate: jest.fn(() => Promise.resolve()),
     },
     TokenRatesController: {
-      updateExchangeRates: jest.fn(() => Promise.resolve()),
+      updateExchangeRatesByChainId: jest.fn(() => Promise.resolve()),
     },
     NetworkController: {
       getNetworkClientById: () => ({
@@ -359,7 +359,7 @@ describe('Tokens', () => {
         Engine.context.CurrencyRateController.updateExchangeRate,
       ).toHaveBeenCalled();
       expect(
-        Engine.context.TokenRatesController.updateExchangeRates,
+        Engine.context.TokenRatesController.updateExchangeRatesByChainId,
       ).toHaveBeenCalled();
     });
   });

app/components/UI/Tokens/index.tsx

@@ -183,7 +183,11 @@ const Tokens: React.FC<TokensI> = ({ tokens }) => {
         TokenDetectionController.detectTokens(),
         AccountTrackerController.refresh(),
         CurrencyRateController.updateExchangeRate(nativeCurrencies),
-        TokenRatesController.updateExchangeRates(),
+        ...Object.values(networkConfigurationsByChainId).map((network) =>
+          TokenRatesController.updateExchangeRatesByChainId(
+            { chainId: network.chainId, nativeCurrency: network.nativeCurrency }
+          ),
+        ),
       ];
       await Promise.all(actions).catch((error) => {
         Logger.error(error, 'Error while refreshing tokens');

app/components/hooks/AssetPolling/AssetPollingProvider.tsx

@@ -1,11 +1,13 @@
 import React, { ReactNode } from 'react';
 import useCurrencyRatePolling from './useCurrencyRatePolling';
+import useTokenRatesPolling from './useTokenRatesPolling';
 
 // This provider is a step towards making controller polling fully UI based.
 // Eventually, individual UI components will call the use*Polling hooks to
 // poll and return particular data. This polls globally in the meantime.
 export const AssetPollingProvider = ({ children }: { children: ReactNode }) => {
   useCurrencyRatePolling();
+  useTokenRatesPolling();
 
   return <>{children}</>;
 };

app/components/hooks/AssetPolling/useTokenRatesPolling.test.ts

@@ -0,0 +1,63 @@
+import useTokenRatesPolling from './useTokenRatesPolling';
+import { renderHookWithProvider } from '../../../util/test/renderWithProvider';
+import Engine from '../../../core/Engine';
+
+jest.mock('../../../core/Engine', () => ({
+  context: {
+    TokenRatesController: {
+      startPolling: jest.fn(),
+      stopPollingByPollingToken: jest.fn(),
+    },
+  },
+}));
+
+describe('useTokenRatesPolling', () => {
+
+  beforeEach(() => {
+    jest.resetAllMocks();
+  });
+
+  const state = {
+    engine: {
+      backgroundState: {
+        TokenRatesController: {
+          marketData: {},
+        },
+        NetworkController: {
+          networkConfigurationsByChainId: {
+            '0x1': {},
+            '0x89': {},
+          },
+        },
+      },
+    },
+  };
+
+  it('Should poll by provided chain ids', async () => {
+
+    renderHookWithProvider(() => useTokenRatesPolling({chainIds: ['0x1']}), {state});
+
+    const mockedTokenRatesController = jest.mocked(Engine.context.TokenRatesController);
+
+    expect(mockedTokenRatesController.startPolling).toHaveBeenCalledTimes(1);
+    expect(
+      mockedTokenRatesController.startPolling
+    ).toHaveBeenCalledWith({chainId: '0x1'});
+
+  });
+
+  it('Should poll by all chain ids in network state, if no chain ids are explicitly provided', async () => {
+
+    renderHookWithProvider(() => useTokenRatesPolling(), {state});
+
+    const mockedTokenRatesController = jest.mocked(Engine.context.TokenRatesController);
+
+    expect(mockedTokenRatesController.startPolling).toHaveBeenCalledTimes(2);
+    expect(
+      mockedTokenRatesController.startPolling
+    ).toHaveBeenCalledWith({chainId: '0x1'});
+    expect(
+      mockedTokenRatesController.startPolling
+    ).toHaveBeenCalledWith({chainId: '0x89'});
+  });
+});

app/components/hooks/AssetPolling/useTokenRatesPolling.ts

@@ -0,0 +1,33 @@
+import { useSelector } from 'react-redux';
+import usePolling from '../usePolling';
+import Engine from '../../../core/Engine';
+import { selectNetworkConfigurations } from '../../../selectors/networkController';
+import { Hex } from '@metamask/utils';
+import { selectContractExchangeRates, selectTokenMarketData } from '../../../selectors/tokenRatesController';
+
+const useTokenRatesPolling = ({ chainIds }: { chainIds?: Hex[] } = {}) => {
+
+  // Selectors to determine polling input
+  const networkConfigurations = useSelector(selectNetworkConfigurations);
+
+  // Selectors returning state updated by the polling
+  const contractExchangeRates = useSelector(selectContractExchangeRates);
+  const tokenMarketData = useSelector(selectTokenMarketData);
+
+  const { TokenRatesController } = Engine.context;
+
+  usePolling({
+    startPolling:
+    TokenRatesController.startPolling.bind(TokenRatesController),
+    stopPollingByPollingToken:
+    TokenRatesController.stopPollingByPollingToken.bind(TokenRatesController),
+    input: (chainIds ?? Object.keys(networkConfigurations)).map((chainId) => ({chainId: chainId as Hex})),
+  });
+
+  return {
+    contractExchangeRates,
+    tokenMarketData
+  };
+};
+
+export default useTokenRatesPolling;

app/core/Engine.ts

@@ -1542,6 +1542,8 @@ export class Engine {
           assetsContractController.getBalancesInSingleCall.bind(
             assetsContractController,
           ),
+        platform: 'mobile',
+        useAccountsAPI: true,
       }),
 
       new NftDetectionController({
@@ -1908,14 +1910,12 @@ export class Engine {
       TokenDetectionController,
       TokenListController,
       TransactionController,
-      TokenRatesController,
     } = this.context;
 
     TokenListController.start();
     TokenDetectionController.start();
     // leaving the reference of TransactionController here, rather than importing it from utils to avoid circular dependency
     TransactionController.startIncomingTransactionPolling();
-    TokenRatesController.start();
   }
 
   configureControllersOnNetworkChange() {
@@ -2158,11 +2158,11 @@ export class Engine {
     // SelectedNetworkController.unsetAllDomains()
 
     //Clear assets info
-    TokensController.reset();
-    NftController.reset();
+    TokensController.resetState();
+    NftController.resetState();
 
-    TokenBalancesController.reset();
-    TokenRatesController.reset();
+    TokenBalancesController.resetState();
+    TokenRatesController.resetState();
 
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     (TransactionController as any).update(() => ({

app/selectors/tokenRatesController.ts

@@ -14,3 +14,9 @@ export const selectContractExchangeRates = createSelector(
   (chainId: Hex, tokenRatesControllerState: TokenRatesControllerState) =>
     tokenRatesControllerState.marketData[chainId],
 );
+
+export const selectTokenMarketData = createSelector(
+  selectTokenRatesControllerState,
+  (tokenRatesControllerState: TokenRatesControllerState) =>
+    tokenRatesControllerState.marketData,
+);

package.json

@@ -140,7 +140,7 @@
     "@metamask/accounts-controller": "^18.2.1",
     "@metamask/address-book-controller": "^6.0.1",
     "@metamask/approval-controller": "^7.1.0",
-    "@metamask/assets-controllers": "^41.0.0",
+    "@metamask/assets-controllers": "^42.0.0",
     "@metamask/base-controller": "^7.0.1",
     "@metamask/composable-controller": "^3.0.0",
     "@metamask/contract-metadata": "^2.1.0",