fix: replace test-helpers mainnetTokens with common mainnetTokens (#75)

Co-authored-by: Bob Lu <[email protected]>
dinngo · Aug 9, 2024 · 35d465d · 35d465d
1 parent 71d51bc
commit 35d465d
Show file tree

Hide file tree

Showing 34 changed files with 339 additions and 260 deletions.
diff --git a/.changeset/shaggy-years-invite.md b/.changeset/shaggy-years-invite.md
@@ -0,0 +1,5 @@
+---
+'@protocolink/logics': patch
+---
+
+replace test-helpers mainnetTokens with common mainnetTokens
diff --git a/test/logics/aave-v2/borrow.test.ts b/test/logics/aave-v2/borrow.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev2 from 'src/logics/aave-v2';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,8 +15,8 @@ describe('mainnet-pb: Test AaveV2 Borrow Logic', function () {
     chainId = await getChainId();
     const [, user1, user2] = await hre.ethers.getSigners();
     users = [user1, user2];
-    await claimToken(chainId, user1.address, mainnetTokens.USDC, '20000');
-    await claimToken(chainId, user2.address, mainnetTokens.WETH, '100');
+    await claimToken(chainId, user1.address, common.mainnetTokens.USDC, '20000');
+    await claimToken(chainId, user2.address, common.mainnetTokens.WETH, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v2/deposit.test.ts b/test/logics/aave-v2/deposit.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev2 from 'src/logics/aave-v2';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -14,8 +14,8 @@ describe('mainnet-pb: Test AaveV2 Deposit Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '100');
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v2/flash-loan.test.ts b/test/logics/aave-v2/flash-loan.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev2 from 'src/logics/aave-v2';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,10 +15,16 @@ describe('mainnet-pb: Test AaveV2 FlashLoan Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '2');
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '2');
-    await claimToken(chainId, user.address, mainnetTokens.USDT, '2');
-    await claimToken(chainId, user.address, mainnetTokens.DAI, '2', '0x8A610c1C93da88c59F51A6264A4c70927814B320');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDT, '2');
+    await claimToken(
+      chainId,
+      user.address,
+      common.mainnetTokens.DAI,
+      '2',
+      '0x8A610c1C93da88c59F51A6264A4c70927814B320'
+    );
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v2/repay.test.ts b/test/logics/aave-v2/repay.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev2 from 'src/logics/aave-v2';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -16,10 +16,10 @@ describe('mainnet-pb: Test AaveV2 Repay Logic', function () {
     chainId = await getChainId();
     const [, user1, user2] = await hre.ethers.getSigners();
     users = [user1, user2];
-    await claimToken(chainId, user1.address, mainnetTokens.USDC, '40000');
-    await claimToken(chainId, user1.address, mainnetTokens.WETH, '100');
-    await claimToken(chainId, user2.address, mainnetTokens.USDC, '100');
-    await claimToken(chainId, user2.address, mainnetTokens.WETH, '100');
+    await claimToken(chainId, user1.address, common.mainnetTokens.USDC, '40000');
+    await claimToken(chainId, user1.address, common.mainnetTokens.WETH, '100');
+    await claimToken(chainId, user2.address, common.mainnetTokens.USDC, '100');
+    await claimToken(chainId, user2.address, common.mainnetTokens.WETH, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v2/withdraw.test.ts b/test/logics/aave-v2/withdraw.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev2 from 'src/logics/aave-v2';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,8 +15,8 @@ describe('mainnet-pb: Test AaveV2 Withdraw Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '100');
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v3/borrow.test.ts b/test/logics/aave-v3/borrow.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev3 from 'src/logics/aave-v3';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,8 +15,8 @@ describe('mainnet-pb: Test AaveV3 Borrow Logic', function () {
     chainId = await getChainId();
     const [, user1, user2] = await hre.ethers.getSigners();
     users = [user1, user2];
-    await claimToken(chainId, user1.address, mainnetTokens.USDC, '10000');
-    await claimToken(chainId, user2.address, mainnetTokens.WETH, '100');
+    await claimToken(chainId, user1.address, common.mainnetTokens.USDC, '10000');
+    await claimToken(chainId, user2.address, common.mainnetTokens.WETH, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v3/flash-loan.test.ts b/test/logics/aave-v3/flash-loan.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev3 from 'src/logics/aave-v3';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,11 +15,17 @@ describe('mainnet-pb: Test AaveV3 FlashLoan Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '2');
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '2');
-    await claimToken(chainId, user.address, mainnetTokens.USDT, '2');
-    await claimToken(chainId, user.address, mainnetTokens.DAI, '2', '0x8A610c1C93da88c59F51A6264A4c70927814B320');
-    await claimToken(chainId, user.address, mainnetTokens.WBTC, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDT, '2');
+    await claimToken(
+      chainId,
+      user.address,
+      common.mainnetTokens.DAI,
+      '2',
+      '0x8A610c1C93da88c59F51A6264A4c70927814B320'
+    );
+    await claimToken(chainId, user.address, common.mainnetTokens.WBTC, '2');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v3/repay.test.ts b/test/logics/aave-v3/repay.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev3 from 'src/logics/aave-v3';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -16,10 +16,10 @@ describe('mainnet-pb: Test AaveV3 Repay Logic', function () {
     chainId = await getChainId();
     const [, user1, user2] = await hre.ethers.getSigners();
     users = [user1, user2];
-    await claimToken(chainId, user1.address, mainnetTokens.USDC, '20000');
-    await claimToken(chainId, user1.address, mainnetTokens.WETH, '100');
-    await claimToken(chainId, user2.address, mainnetTokens.USDC, '100');
-    await claimToken(chainId, user2.address, mainnetTokens.WETH, '100');
+    await claimToken(chainId, user1.address, common.mainnetTokens.USDC, '20000');
+    await claimToken(chainId, user1.address, common.mainnetTokens.WETH, '100');
+    await claimToken(chainId, user2.address, common.mainnetTokens.USDC, '100');
+    await claimToken(chainId, user2.address, common.mainnetTokens.WETH, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v3/supply.test.ts b/test/logics/aave-v3/supply.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev3 from 'src/logics/aave-v3';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -14,8 +14,8 @@ describe('mainnet-pb: Test AaveV3 Supply Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '100');
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/aave-v3/withdraw.test.ts b/test/logics/aave-v3/withdraw.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as aavev3 from 'src/logics/aave-v3';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,8 +15,8 @@ describe('mainnet-pb: Test AaveV3 Withdraw Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '100');
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '100');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '100');
   });
 
   snapshotAndRevertEach();

diff --git a/test/logics/balancer-v2/flash-loan.test.ts b/test/logics/balancer-v2/flash-loan.test.ts
@@ -1,6 +1,6 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 import * as balancerv2 from 'src/logics/balancer-v2';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as core from '@protocolink/core';
 import { expect } from 'chai';
@@ -15,19 +15,25 @@ describe('mainnet-pb: Test BalancerV2 FlashLoan Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.WETH, '2');
-    await claimToken(chainId, user.address, mainnetTokens.USDC, '2');
-    await claimToken(chainId, user.address, mainnetTokens.USDT, '2');
-    await claimToken(chainId, user.address, mainnetTokens.DAI, '2', '0x8A610c1C93da88c59F51A6264A4c70927814B320');
+    await claimToken(chainId, user.address, common.mainnetTokens.WETH, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDC, '2');
+    await claimToken(chainId, user.address, common.mainnetTokens.USDT, '2');
+    await claimToken(
+      chainId,
+      user.address,
+      common.mainnetTokens.DAI,
+      '2',
+      '0x8A610c1C93da88c59F51A6264A4c70927814B320'
+    );
   });
 
   snapshotAndRevertEach();
 
   const testCases = [
-    { loans: new common.TokenAmounts([mainnetTokens.WETH, '1'], [mainnetTokens.USDC, '1']) },
-    { repays: new common.TokenAmounts([mainnetTokens.WETH, '1'], [mainnetTokens.USDC, '1']) },
-    { loans: new common.TokenAmounts([mainnetTokens.USDT, '1'], [mainnetTokens.DAI, '1']) },
-    { repays: new common.TokenAmounts([mainnetTokens.USDT, '1'], [mainnetTokens.DAI, '1']) },
+    { loans: new common.TokenAmounts([common.mainnetTokens.WETH, '1'], [common.mainnetTokens.USDC, '1']) },
+    { repays: new common.TokenAmounts([common.mainnetTokens.WETH, '1'], [common.mainnetTokens.USDC, '1']) },
+    { loans: new common.TokenAmounts([common.mainnetTokens.USDT, '1'], [common.mainnetTokens.DAI, '1']) },
+    { repays: new common.TokenAmounts([common.mainnetTokens.USDT, '1'], [common.mainnetTokens.DAI, '1']) },
   ];
 
   testCases.forEach((params, i) => {

diff --git a/test/logics/compound-v2/claim.test.ts b/test/logics/compound-v2/claim.test.ts
@@ -1,5 +1,5 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as compoundv2 from 'src/logics/compound-v2';
 import * as core from '@protocolink/core';
@@ -16,8 +16,14 @@ describe('mainnet-pb: Test CompoundV2 Claim Logic', function () {
     chainId = await getChainId();
     const [, user1, user2] = await hre.ethers.getSigners();
     users = [user1, user2];
-    await claimToken(chainId, user1.address, mainnetTokens.USDC, '5000');
-    await claimToken(chainId, user1.address, mainnetTokens.DAI, '5000', '0x8A610c1C93da88c59F51A6264A4c70927814B320');
+    await claimToken(chainId, user1.address, common.mainnetTokens.USDC, '5000');
+    await claimToken(
+      chainId,
+      user1.address,
+      common.mainnetTokens.DAI,
+      '5000',
+      '0x8A610c1C93da88c59F51A6264A4c70927814B320'
+    );
   });
 
   snapshotAndRevertEach();
@@ -27,26 +33,26 @@ describe('mainnet-pb: Test CompoundV2 Claim Logic', function () {
     {
       ownerIndex: 0,
       claimerIndex: 0,
-      supply: new common.TokenAmount(mainnetTokens.DAI, '1000'),
-      borrow: new common.TokenAmount(mainnetTokens.USDC, '100'),
+      supply: new common.TokenAmount(common.mainnetTokens.DAI, '1000'),
+      borrow: new common.TokenAmount(common.mainnetTokens.USDC, '100'),
     },
     {
       ownerIndex: 0,
       claimerIndex: 0,
-      supply: new common.TokenAmount(mainnetTokens.USDC, '3000'),
-      borrow: new common.TokenAmount(mainnetTokens.DAI, '100'),
+      supply: new common.TokenAmount(common.mainnetTokens.USDC, '3000'),
+      borrow: new common.TokenAmount(common.mainnetTokens.DAI, '100'),
     },
     {
       ownerIndex: 0,
       claimerIndex: 1,
-      supply: new common.TokenAmount(mainnetTokens.DAI, '1000'),
-      borrow: new common.TokenAmount(mainnetTokens.USDC, '100'),
+      supply: new common.TokenAmount(common.mainnetTokens.DAI, '1000'),
+      borrow: new common.TokenAmount(common.mainnetTokens.USDC, '100'),
     },
     {
       ownerIndex: 0,
       claimerIndex: 1,
-      supply: new common.TokenAmount(mainnetTokens.USDC, '3000'),
-      borrow: new common.TokenAmount(mainnetTokens.DAI, '100'),
+      supply: new common.TokenAmount(common.mainnetTokens.USDC, '3000'),
+      borrow: new common.TokenAmount(common.mainnetTokens.DAI, '100'),
     },
   ];
 

diff --git a/test/logics/compound-v2/repay.test.ts b/test/logics/compound-v2/repay.test.ts
@@ -1,5 +1,5 @@
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
-import { claimToken, getChainId, mainnetTokens, snapshotAndRevertEach } from '@protocolink/test-helpers';
+import { claimToken, getChainId, snapshotAndRevertEach } from '@protocolink/test-helpers';
 import * as common from '@protocolink/common';
 import * as compoundv2 from 'src/logics/compound-v2';
 import * as core from '@protocolink/core';
@@ -16,28 +16,34 @@ describe('mainnet-pb: Test CompoundV2 Repay Logic', function () {
   before(async function () {
     chainId = await getChainId();
     [, user] = await hre.ethers.getSigners();
-    await claimToken(chainId, user.address, mainnetTokens.WBTC, '10', '0x9ff58f4ffb29fa2266ab25e75e2a8b3503311656');
+    await claimToken(
+      chainId,
+      user.address,
+      common.mainnetTokens.WBTC,
+      '10',
+      '0x9ff58f4ffb29fa2266ab25e75e2a8b3503311656'
+    );
   });
 
   snapshotAndRevertEach();
 
   const testCases = [
     {
-      supply: new common.TokenAmount(mainnetTokens.ETH, '100'),
-      borrow: new common.TokenAmount(mainnetTokens.WBTC, '1'),
+      supply: new common.TokenAmount(common.mainnetTokens.ETH, '100'),
+      borrow: new common.TokenAmount(common.mainnetTokens.WBTC, '1'),
     },
     {
-      supply: new common.TokenAmount(mainnetTokens.WBTC, '1'),
-      borrow: new common.TokenAmount(mainnetTokens.ETH, '1'),
+      supply: new common.TokenAmount(common.mainnetTokens.WBTC, '1'),
+      borrow: new common.TokenAmount(common.mainnetTokens.ETH, '1'),
     },
     {
-      supply: new common.TokenAmount(mainnetTokens.ETH, '100'),
-      borrow: new common.TokenAmount(mainnetTokens.WBTC, '1'),
+      supply: new common.TokenAmount(common.mainnetTokens.ETH, '100'),
+      borrow: new common.TokenAmount(common.mainnetTokens.WBTC, '1'),
       balanceBps: 5000,
     },
     {
-      supply: new common.TokenAmount(mainnetTokens.WBTC, '1'),
-      borrow: new common.TokenAmount(mainnetTokens.ETH, '1'),
+      supply: new common.TokenAmount(common.mainnetTokens.WBTC, '1'),
+      borrow: new common.TokenAmount(common.mainnetTokens.ETH, '1'),
       balanceBps: 5000,
     },
   ];