Fix the Mock A token references

mstable · Nov 24, 2020 · 50e7934 · 50e7934
1 parent fa38060
commit 50e7934
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/test/masset/TestMassetRedeemMulti.spec.ts b/test/masset/TestMassetRedeemMulti.spec.ts
@@ -14,7 +14,7 @@ import * as t from "types/generated";
 const { expect } = envSetup.configure();
 
 const MockERC20 = artifacts.require("MockERC20");
-const MockAToken = artifacts.require("MockAToken");
+const MockAToken = artifacts.require("MockATokenV2");
 const MockAave = artifacts.require("MockAaveV2");
 
 contract("Masset - RedeemMasset", async (accounts) => {
@@ -273,9 +273,7 @@ contract("Masset - RedeemMasset", async (accounts) => {
                     );
                     const expectedBassetsExact = await Promise.all(
                         basketComp.bAssets.map((b) =>
-                            simpleToExactAmount(10, 18)
-                                .mul(ratioScale)
-                                .div(new BN(b.ratio)),
+                            simpleToExactAmount(10, 18).mul(ratioScale).div(new BN(b.ratio)),
                         ),
                     );
                     const bAssetFees = expectedBassetsExact.map((b, i) =>

diff --git a/test/masset/TestMassetRedemption.spec.ts b/test/masset/TestMassetRedemption.spec.ts
@@ -15,7 +15,7 @@ const { expect } = envSetup.configure();
 
 const MockBasketManager1 = artifacts.require("MockBasketManager1");
 const MockERC20 = artifacts.require("MockERC20");
-const MockAToken = artifacts.require("MockAToken");
+const MockAToken = artifacts.require("MockATokenV2");
 const MockAave = artifacts.require("MockAaveV2");
 const AaveIntegration = artifacts.require("AaveIntegration");