From cfcf41652fcb3b02b8635c55c61909a1781ef685 Mon Sep 17 00:00:00 2001
From: JIAXIN <murrayee@163.com>
Date: Thu, 30 Mar 2023 09:39:58 +0800
Subject: [PATCH] test(bridge-ui): add  unit test for
 checkIfTokenIsDeployedCrossChain util (#13483)

Co-authored-by: jeff <113397187+cyberhorsey@users.noreply.github.com>
---
 .../checkIfTokenIsDeployedCrossChain.spec.ts  | 122 ++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 packages/bridge-ui/src/utils/checkIfTokenIsDeployedCrossChain.spec.ts

diff --git a/packages/bridge-ui/src/utils/checkIfTokenIsDeployedCrossChain.spec.ts b/packages/bridge-ui/src/utils/checkIfTokenIsDeployedCrossChain.spec.ts
new file mode 100644
index 00000000000..53e19d423aa
--- /dev/null
+++ b/packages/bridge-ui/src/utils/checkIfTokenIsDeployedCrossChain.spec.ts
@@ -0,0 +1,122 @@
+import { ethers } from 'ethers';
+import { ETHToken } from '../token/tokens';
+import TokenVault from '../constants/abi/TokenVault';
+import type { Chain } from '../domain/chain';
+import type { Token } from '../domain/token';
+import { checkIfTokenIsDeployedCrossChain } from './checkIfTokenIsDeployedCrossChain';
+
+// mock the `ethers.providers.JsonRpcProvider` object for testing purposes
+const provider = new ethers.providers.JsonRpcProvider();
+jest.mock('../constants/envVars');
+
+describe('checkIfTokenIsDeployedCrossChain', () => {
+  const token: Token = {
+    name: 'Test Token',
+    symbol: 'TEST',
+    decimals: 18,
+    logoComponent: null,
+    addresses: [
+      { chainId: 1, address: '0x0000000000000000000000000000000000000001' },
+      { chainId: 4, address: '0x0000000000000000000000000000000000000002' },
+      { chainId: 5, address: '0x00' },
+    ],
+  };
+  const destTokenVaultAddress = '0x0000000000000000000000000000000000000004';
+
+  const toChain: Chain = {
+    id: 5,
+    name: 'MyChain',
+    rpc: 'http://mychain.rpc.com',
+    explorerUrl: '',
+    signalServiceAddress: '',
+    bridgeAddress: '0x1234567890',
+    headerSyncAddress: '0x0987654321',
+  };
+
+  const fromChain: Chain = {
+    id: 1,
+    name: 'SomeOtherChain',
+    rpc: 'http://otherchain.rpc.com',
+    explorerUrl: '',
+    signalServiceAddress: '',
+    bridgeAddress: '0x9876543210',
+    headerSyncAddress: '0x0123456789',
+  };
+
+  it('should return false when the token is ETH', async () => {
+    const ethToken: Token = {
+      ...token,
+      symbol: ETHToken.symbol,
+      addresses: [],
+    };
+    const result = await checkIfTokenIsDeployedCrossChain(
+      ethToken,
+      provider,
+      destTokenVaultAddress,
+      toChain,
+      fromChain,
+    );
+    expect(result).toBeFalsy();
+  });
+
+  it('should return false when the token is not deployed on the destination chain', async () => {
+    // mock the `ethers.Contract` object for testing purposes
+    const destTokenVaultContract = {
+      canonicalToBridged: jest
+        .fn()
+        .mockResolvedValue(ethers.constants.AddressZero),
+    };
+    jest
+      .spyOn(ethers, 'Contract')
+      .mockReturnValue(destTokenVaultContract as any);
+
+    const result = await checkIfTokenIsDeployedCrossChain(
+      token,
+      provider,
+      destTokenVaultAddress,
+      toChain,
+      fromChain,
+    );
+    expect(result).toBeFalsy();
+
+    expect(ethers.Contract).toHaveBeenCalledWith(
+      destTokenVaultAddress,
+      TokenVault,
+      provider,
+    );
+    expect(destTokenVaultContract.canonicalToBridged).toHaveBeenCalledWith(
+      fromChain.id,
+      token.addresses.find((a) => a.chainId === fromChain.id)?.address,
+    );
+  });
+
+  it('should return true when the token is deployed as BridgedERC20 on the destination chain', async () => {
+    const bridgedTokenAddress = '0x00';
+    // mock the `ethers.Contract` object for testing purposes
+    const destTokenVaultContract = {
+      canonicalToBridged: jest.fn().mockResolvedValue(bridgedTokenAddress),
+    };
+    jest
+      .spyOn(ethers, 'Contract')
+      .mockReturnValue(destTokenVaultContract as any);
+
+    const result = await checkIfTokenIsDeployedCrossChain(
+      token,
+      provider,
+      destTokenVaultAddress,
+      toChain,
+      fromChain,
+    );
+    expect(result).toBeTruthy();
+
+    expect(ethers.Contract).toHaveBeenCalledWith(
+      destTokenVaultAddress,
+      TokenVault,
+      provider,
+    );
+    expect(destTokenVaultContract.canonicalToBridged).toHaveBeenCalledWith(
+      fromChain.id,
+      token.addresses.find((a) => a.chainId === fromChain.id)?.address,
+    );
+  });
+});