test: fix before amount

yearn · Aug 20, 2024 · 52c7912 · 52c7912
1 parent 8c6c122
commit 52c7912
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/src/test/Auction.t.sol b/src/test/Auction.t.sol
@@ -295,6 +295,7 @@ contract AuctionTest is Setup, ITaker {
         skip(auction.auctionLength() / 2);
 
         uint256 needed = auction.getAmountNeeded(id, _amount);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -311,7 +312,7 @@ contract AuctionTest is Setup, ITaker {
         (, , , _available) = auction.auctionInfo(id);
         assertEq(_available, 0);
 
-        assertEq(ERC20(asset).balanceOf(address(this)), 0);
+        assertEq(ERC20(asset).balanceOf(address(this)), beforeAsset);
         assertEq(ERC20(from).balanceOf(address(this)), before + _amount);
         assertEq(ERC20(from).balanceOf(address(auction)), 0);
         assertEq(ERC20(asset).balanceOf(address(mockStrategy)), needed);
@@ -351,6 +352,7 @@ contract AuctionTest is Setup, ITaker {
         uint256 toTake = (_amount * _percent) / MAX_BPS;
         uint256 left = _amount - toTake;
         uint256 needed = auction.getAmountNeeded(id, toTake);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -366,7 +368,7 @@ contract AuctionTest is Setup, ITaker {
 
         (, , , _available) = auction.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(auction)), left);
         assertEq(ERC20(asset).balanceOf(address(mockStrategy)), needed);
@@ -405,6 +407,7 @@ contract AuctionTest is Setup, ITaker {
         uint256 toTake = _amount / 2;
         uint256 left = _amount - toTake;
         uint256 needed = auction.getAmountNeeded(id, toTake);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -424,7 +427,7 @@ contract AuctionTest is Setup, ITaker {
 
         (, , , _available) = auction.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(auction)), left);
         assertEq(ERC20(asset).balanceOf(address(mockStrategy)), needed);

diff --git a/src/test/AuctionSwapper.t.sol b/src/test/AuctionSwapper.t.sol
@@ -309,6 +309,7 @@ contract AuctionSwapperTest is Setup {
         uint256 toTake = (_amount * _percent) / MAX_BPS;
         uint256 left = _amount - toTake;
         uint256 needed = auction.getAmountNeeded(id, toTake);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -324,7 +325,7 @@ contract AuctionSwapperTest is Setup {
 
         (, , , _available) = auction.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(auction)), left);
         assertEq(ERC20(asset).balanceOf(address(swapper)), needed);
@@ -470,6 +471,7 @@ contract AuctionSwapperTest is Setup {
         uint256 toTake = (kickable * _percent) / MAX_BPS;
         uint256 left = kickable - toTake;
         uint256 needed = auction.getAmountNeeded(id, toTake);
+        uint256 beforeAsset = ERC20(asset).balanceOf(address(this));
 
         airdrop(ERC20(asset), address(this), needed);
 
@@ -487,7 +489,7 @@ contract AuctionSwapperTest is Setup {
 
         (, , , _available) = auction.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(auction)), left);
         assertEq(ERC20(asset).balanceOf(address(swapper)), needed);