restore withdraw-test; clean-up

compound-finance · Apr 19, 2022 · 94f1fa1 · 94f1fa1
1 parent c3d5693
commit 94f1fa1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/test/comet-ext-test.ts b/test/comet-ext-test.ts
@@ -1,5 +1,5 @@
 import { CometHarnessInterface, FaucetToken } from '../build/types';
-import { baseBalanceOf, expect, exp, makeProtocol, setTotalsBasic } from './helpers';
+import { expect, exp, makeProtocol, setTotalsBasic } from './helpers';
 import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
 
 describe('CometExt', function () {

diff --git a/test/withdraw-test.ts b/test/withdraw-test.ts
@@ -331,9 +331,9 @@ describe('withdraw', function () {
 
       // no USDC transferred
       expect(await USDC.balanceOf(comet.address)).to.eq(100e6);
-      // expect(await comet.baseBalanceOf(alice.address)).to.eq(0);
+      expect(await baseBalanceOf(comet, alice.address)).to.eq(0n);
       expect(await USDC.balanceOf(alice.address)).to.eq(0);
-      // expect(await comet.baseBalanceOf(bob.address)).to.eq(0);
+      expect(await baseBalanceOf(comet, bob.address)).to.eq(0n);
       expect(await USDC.balanceOf(bob.address)).to.eq(0);
     });
 
@@ -378,9 +378,9 @@ describe('withdraw', function () {
 
       // no USDC transferred
       expect(await USDC.balanceOf(comet.address)).to.eq(100e6);
-      // expect(await comet.baseBalanceOf(alice.address)).to.eq(0);
+      expect(await baseBalanceOf(comet, alice.address)).to.eq(0n);
       expect(await USDC.balanceOf(alice.address)).to.eq(0);
-      // expect(await comet.baseBalanceOf(bob.address)).to.eq(0);
+      expect(await baseBalanceOf(comet, bob.address)).to.eq(0n);
       expect(await USDC.balanceOf(bob.address)).to.eq(0);
     });
   });