Add unit test for ` transport of por-address-list (#3890)

* Add unit test for openEdenUSDOAddress transport of por-address-list

* lint+compile

Author: dskloetc

packages/sources/por-address-list/test/unit/openEdenUSDOAddress.test.ts

@@ -0,0 +1,189 @@
+import { TransportDependencies } from '@chainlink/external-adapter-framework/transports'
+import { LoggerFactoryProvider } from '@chainlink/external-adapter-framework/util'
+import { makeStub } from '@chainlink/external-adapter-framework/util/testing-utils'
+import { BaseEndpointTypes, inputParameters } from '../../src/endpoint/openEdenUSDOAddress'
+import { AddressTransport } from '../../src/transport/openEdenUSDOAddress'
+
+type RequestParams = typeof inputParameters.validated
+
+const originalEnv = { ...process.env }
+
+const restoreEnv = () => {
+  for (const key of Object.keys(process.env)) {
+    if (key in originalEnv) {
+      process.env[key] = originalEnv[key]
+    } else {
+      delete process.env[key]
+    }
+  }
+}
+
+const ADDRESS_LIST_CONTRACT_ADDRESS = '0x440139321A15d14ce0729E004e91D66BaF1A08B0'
+
+const addressListContract = {
+  getPoRAddressListLength: jest.fn().mockResolvedValue(0),
+  getPoRAddressList: jest.fn().mockResolvedValue([]),
+}
+
+const contracts: Record<string, unknown> = makeStub('contracts', {
+  [ADDRESS_LIST_CONTRACT_ADDRESS]: addressListContract,
+})
+
+const makeEthers = () => {
+  return {
+    providers: {
+      JsonRpcProvider: jest.fn(),
+    },
+    Contract: function (address: string) {
+      return contracts[address]
+    },
+  }
+}
+
+jest.mock('ethers', () => ({
+  ethers: makeEthers(),
+}))
+
+const walletAddress = '0x5EaFF7af80488033Bc845709806D5Fae5291eB88'
+const tbillContractAddress = '0xdd50C053C096CB04A3e3362E2b622529EC5f2e8a'
+const tbillPriceOracleAddress = '0xCe9a6626Eb99eaeA829D7fA613d5D0A2eaE45F40'
+const usdcContractAddress = '0xA0b86991c6218b36c1d19D4a2e9Eb0cE3606eB48'
+const tbillAddress = makeStub('tbillAddress', {
+  chain: 'Ethereem Mainnet',
+  chainId: 1,
+  tokenSymbol: 'TBILL',
+  tokenAddress: tbillContractAddress,
+  tokenPriceOracle: tbillPriceOracleAddress,
+  yourVaultAddress: walletAddress,
+})
+const usdcAddress = makeStub('otherAddress', {
+  chain: 'Ethereem Mainnet',
+  chainId: 1,
+  tokenSymbol: 'USDC',
+  tokenAddress: usdcContractAddress,
+  yourVaultAddress: walletAddress,
+})
+
+LoggerFactoryProvider.set()
+
+describe('AddressTransport', () => {
+  const transportName = 'default_single_transport'
+  const endpointName = 'openedenAddress'
+
+  const adapterSettings = makeStub('adapterSettings', {
+    WARMUP_SUBSCRIPTION_TTL: 10_000,
+  } as unknown as BaseEndpointTypes['Settings'])
+
+  const responseCache = {
+    write: jest.fn(),
+  }
+
+  const dependencies = makeStub('dependencies', {
+    responseCache,
+    subscriptionSetFactory: {
+      buildSet: jest.fn(),
+    },
+  } as unknown as TransportDependencies<BaseEndpointTypes>)
+
+  let transport: AddressTransport
+
+  beforeEach(async () => {
+    restoreEnv()
+    jest.restoreAllMocks()
+    jest.useFakeTimers()
+
+    process.env.BASE_RPC_URL = 'https://base-rpc.example.com'
+    process.env.BASE_RPC_CHAIN_ID = '8453'
+
+    transport = new AddressTransport()
+
+    await transport.initialize(dependencies, adapterSettings, endpointName, transportName)
+  })
+
+  describe('_handleRequest', () => {
+    it('should return an empty list', async () => {
+      const params = makeStub('params', {
+        contractAddress: ADDRESS_LIST_CONTRACT_ADDRESS,
+        contractAddressNetwork: 'BASE',
+        type: 'tbill',
+      } as RequestParams)
+      const response = await transport._handleRequest(params)
+      expect(response).toEqual({
+        statusCode: 200,
+        data: {
+          result: [],
+        },
+        result: null,
+        timestamps: {
+          providerDataRequestedUnixMs: Date.now(),
+          providerDataReceivedUnixMs: Date.now(),
+        },
+      })
+    })
+
+    it('should return tbill address', async () => {
+      const addresses = [tbillAddress, usdcAddress]
+      addressListContract.getPoRAddressList.mockResolvedValue(addresses)
+      addressListContract.getPoRAddressListLength.mockResolvedValue(addresses.length)
+
+      const params = makeStub('params', {
+        contractAddress: ADDRESS_LIST_CONTRACT_ADDRESS,
+        contractAddressNetwork: 'BASE',
+        type: 'tbill',
+      } as RequestParams)
+      const response = await transport._handleRequest(params)
+      expect(response).toEqual({
+        statusCode: 200,
+        data: {
+          result: [
+            {
+              contractAddress: tbillContractAddress,
+              network: 'Ethereem Mainnet',
+              chainId: '1',
+              token: 'TBILL',
+              wallets: [walletAddress],
+              priceOracleAddress: tbillPriceOracleAddress,
+            },
+          ],
+        },
+        result: null,
+        timestamps: {
+          providerDataRequestedUnixMs: Date.now(),
+          providerDataReceivedUnixMs: Date.now(),
+        },
+      })
+    })
+
+    it('should return other address', async () => {
+      const addresses = [tbillAddress, usdcAddress]
+      addressListContract.getPoRAddressList.mockResolvedValue(addresses)
+      addressListContract.getPoRAddressListLength.mockResolvedValue(addresses.length)
+
+      const params = makeStub('params', {
+        contractAddress: ADDRESS_LIST_CONTRACT_ADDRESS,
+        contractAddressNetwork: 'BASE',
+        type: 'other',
+      } as RequestParams)
+      const response = await transport._handleRequest(params)
+      expect(response).toEqual({
+        statusCode: 200,
+        data: {
+          result: [
+            {
+              contractAddress: usdcContractAddress,
+              network: 'Ethereem Mainnet',
+              chainId: '1',
+              token: 'USDC',
+              wallets: [walletAddress],
+            },
+          ],
+        },
+        result: null,
+        timestamps: {
+          providerDataRequestedUnixMs: Date.now(),
+          providerDataReceivedUnixMs: Date.now(),
+        },
+      })
+    })
+  })
+})