test(migration): add compoundV3 migration tests

morpho-org · Jan 15, 2025 · 371b613 · 371b613
1 parent 4212dd0
commit 371b613
Show file tree

Hide file tree

Showing 5 changed files with 465 additions and 19 deletions.
diff --git a/packages/blue-migration-sdk-viem/src/fetchers/compoundV2/compoundV2.fetchers.ts b/packages/blue-migration-sdk-viem/src/fetchers/compoundV2/compoundV2.fetchers.ts
@@ -41,11 +41,6 @@ async function fetchCompoundV2InstancePosition(
 
   const chainId = parameters.chainId;
 
-  const migrationContracts =
-    MIGRATION_ADDRESSES[chainId][MigratableProtocol.compoundV2];
-
-  if (!migrationContracts) return null;
-
   if (deployless) {
     //TODO
   }

diff --git a/packages/blue-migration-sdk-viem/src/fetchers/compoundV3/compoundV3.fetchers.ts b/packages/blue-migration-sdk-viem/src/fetchers/compoundV3/compoundV3.fetchers.ts
@@ -20,17 +20,6 @@ async function fetchCompoundV3InstancePosition(
   client: Client,
   { deployless = true, ...parameters }: DeploylessFetchParameters = {},
 ) {
-  parameters.chainId = ChainUtils.parseSupportedChainId(
-    parameters.chainId ?? (await getChainId(client)),
-  );
-
-  const chainId = parameters.chainId;
-
-  const migrationContracts =
-    MIGRATION_ADDRESSES[chainId][MigratableProtocol.aaveV2];
-
-  if (!migrationContracts) return null;
-
   if (deployless) {
     //TODO
   }
@@ -143,7 +132,7 @@ export async function fetchCompoundV3Positions(
   const chainId = parameters.chainId;
 
   const migrationContracts =
-    MIGRATION_ADDRESSES[chainId][MigratableProtocol.aaveV2];
+    MIGRATION_ADDRESSES[chainId][MigratableProtocol.compoundV3];
 
   if (!migrationContracts) return [];
 

diff --git a/packages/blue-migration-sdk-viem/test/e2e/compoundV2/supply.test.ts b/packages/blue-migration-sdk-viem/test/e2e/compoundV2/supply.test.ts
@@ -198,7 +198,7 @@ describe("Supply position on COMPOUND V2", () => {
 
           testFn("should fetch user position", async ({ client }) => {
             const amount = parseUnits("10", underlyingDecimals);
-            await writeSupply(client, cToken, underlying, amount, false);
+            await writeSupply(client, cToken, underlying, amount);
 
             const allPositions = await fetchMigratablePositions(
               client.account.address,