diff --git a/src/test/BaseAuctioneer.t.sol b/src/test/BaseAuctioneer.t.sol index c689418..ea9df23 100644 --- 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);