test: check balances

yearn · Sep 9, 2024 · fedb233 · fedb233
1 parent 3453fb9
commit fedb233
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/test/Accounting.t.sol b/src/test/Accounting.t.sol
@@ -606,5 +606,11 @@ contract AccountingTest is Setup {
 
         // Should just deposit the available amount.
         checkStrategyTotals(strategy, _amount, _amount, 0, _amount);
+
+        assertEq(asset.balanceOf(_address), 0);
+        assertEq(strategy.balanceOf(_address), _amount);
+        assertEq(asset.balanceOf(address(strategy)), 0);
+
+        assertEq(asset.balanceOf(address(yieldSource)), _amount);
     }
 }