test: auctioneer

yearn · Aug 20, 2024 · ad395f7 · ad395f7
1 parent 52c7912
commit ad395f7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/test/BaseAuctioneer.t.sol b/src/test/BaseAuctioneer.t.sol
@@ -276,6 +276,7 @@ contract BaseAuctioneerTest is Setup {
         uint256 toTake = (_amount * _percent) / MAX_BPS;
         uint256 left = _amount - toTake;
         uint256 needed = auctioneer.getAmountNeeded(id, toTake);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -291,7 +292,7 @@ contract BaseAuctioneerTest is Setup {
 
         (, , , _available) = auctioneer.auctionInfo(id);
         assertEq(_available, left);
-        assertEq(ERC20(asset).balanceOf(address(this)), 0);
+        assertEq(ERC20(asset).balanceOf(address(this)), beforeAsset);
         assertEq(ERC20(from).balanceOf(address(this)), before + toTake);
         assertEq(ERC20(from).balanceOf(address(auctioneer)), left);
         assertEq(ERC20(asset).balanceOf(address(auctioneer)), needed);
@@ -430,6 +431,7 @@ contract BaseAuctioneerTest is Setup {
         uint256 toTake = (kickable * _percent) / MAX_BPS;
         uint256 left = _amount - toTake;
         uint256 needed = auctioneer.getAmountNeeded(id, toTake);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -447,7 +449,7 @@ contract BaseAuctioneerTest is Setup {
 
         (, , , _available) = auctioneer.auctionInfo(id);
         assertEq(_available, kickable - toTake);
-        assertEq(ERC20(asset).balanceOf(address(this)), 0);
+        assertEq(ERC20(asset).balanceOf(address(this)), beforeAsset);
         assertEq(ERC20(from).balanceOf(address(this)), before + toTake);
         assertEq(ERC20(from).balanceOf(address(auctioneer)), left);
         assertEq(ERC20(asset).balanceOf(address(auctioneer)), needed);