test: ensure verify asks for ERC20 balances

defi-wonderland · Jul 17, 2024 · 56dcaa8 · 56dcaa8
1 parent 068e06f
commit 56dcaa8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/test/unit/BCoWPool/BCoWPool_Verify.t.sol b/test/unit/BCoWPool/BCoWPool_Verify.t.sol
@@ -123,7 +123,10 @@ contract BCoWPoolVerify is BCoWPoolBase {
   function test_WhenPreconditionsAreMet(uint256 _sellAmount) external {
     _sellAmount = bound(_sellAmount, 0, validOrder.sellAmount);
     validOrder.sellAmount = _sellAmount;
-    // it should return
+    // it should ask the balance of the buy token
+    vm.expectCall(tokenIn, abi.encodeCall(IERC20.balanceOf, (address(bCoWPool))));
+    // it should ask the balance of the sell token
+    vm.expectCall(tokenOut, abi.encodeCall(IERC20.balanceOf, (address(bCoWPool))));
     bCoWPool.verify(validOrder);
   }
 }
diff --git a/test/unit/BCoWPool/BCoWPool_Verify.tree b/test/unit/BCoWPool/BCoWPool_Verify.tree
@@ -20,4 +20,5 @@ BCoWPool::Verify
 ├── when calculated token amount out is less than order sell amount
 │   └── it should revert
 └── when preconditions are met
-    └── it should return
+    ├── it should ask the balance of the buy token
+    └── it should ask the balance of the sell token