chg: fix test errors after merging the dev branch

unioncredit · Jul 1, 2024 · ea625f8 · ea625f8
1 parent 9fe5a80
commit ea625f8
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 126 deletions.
diff --git a/test/foundry/PureTokeAdapter.t.sol b/test/foundry/PureTokeAdapter.t.sol
diff --git a/test/foundry/UnionLens.t.sol b/test/foundry/UnionLens.t.sol
@@ -10,28 +10,28 @@ contract TestUnionLens is TestWrapper {
     function setUp() public virtual {
         deployMocks();
         vm.startPrank(ADMIN);
-        marketRegistryMock.setUToken(address(daiMock), address(uTokenMock));
-        marketRegistryMock.setUserManager(address(daiMock), address(userManagerMock));
+        marketRegistryMock.setUToken(address(erc20Mock), address(uTokenMock));
+        marketRegistryMock.setUserManager(address(erc20Mock), address(userManagerMock));
         vm.stopPrank();
         unionLens = new UnionLens(marketRegistryMock);
     }
 
     function testGetStakerAddresses() public {
-        address[] memory addresses = unionLens.getStakerAddresses(address(daiMock), address(1));
+        address[] memory addresses = unionLens.getStakerAddresses(address(erc20Mock), address(1));
         assertEq(addresses.length, 0);
 
         userManagerMock.updateTrust(address(1), 1 ether);
-        addresses = unionLens.getStakerAddresses(address(daiMock), address(1));
+        addresses = unionLens.getStakerAddresses(address(erc20Mock), address(1));
         assertEq(addresses.length, 1);
     }
 
     function testGetBorrowerAddresses() public {
-        address[] memory addresses = unionLens.getBorrowerAddresses(address(daiMock), address(1));
+        address[] memory addresses = unionLens.getBorrowerAddresses(address(erc20Mock), address(1));
         assertEq(addresses.length, 0);
 
         vm.startPrank(address(1));
         userManagerMock.updateTrust(address(2), 1 ether);
-        addresses = unionLens.getBorrowerAddresses(address(daiMock), address(1));
+        addresses = unionLens.getBorrowerAddresses(address(erc20Mock), address(1));
         assertEq(addresses.length, 1);
         vm.stopPrank();
     }

diff --git a/test/integration/pureTokenAdapter.ts b/test/integration/pureTokenAdapter.ts
@@ -48,8 +48,8 @@ describe("Pure Token Adapter", () => {
         await getDai(contracts.dai, staker, mintAmount);
         await contracts.dai.connect(staker).approve(contracts.userManager.address, ethers.constants.MaxUint256);
 
-        adapter = contracts.adapters.pureToken;
-        await contracts.adapters.pureToken.setCeiling(contracts.dai.address, ethers.constants.MaxUint256);
+        adapter = contracts.adapters.pureTokenAdapter;
+        await adapter.setCeiling(contracts.dai.address, ethers.constants.MaxUint256);
         await contracts.userManager.connect(staker).stake(stakeAmount);
         await helpers.updateTrust(staker, borrower, stakeAmount);
     };
@@ -81,10 +81,10 @@ describe("Pure Token Adapter", () => {
         });
 
         it("cannot borrow if overdue", async () => {
-            await helpers.withOverdueblocks(1, async () => {
+            await helpers.withOverdue(1, async () => {
                 const minBorrow = await contracts.uToken.minBorrow();
                 const resp = helpers.borrow(borrower, minBorrow);
-                await expect(resp).to.be.revertedWith("MemberIsOverdue()");
+                await expect(resp).to.be.revertedWithCustomError(contracts.uToken, "MemberIsOverdue");
             });
         });