fix: test on divested strategies

yieldprotocol · Jan 8, 2024 · 42252f3 · 42252f3
1 parent 10bdd1f
commit 42252f3
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/test/Unwind.t.sol b/test/Unwind.t.sol
@@ -22,6 +22,15 @@ contract UnwindTest is PRBTest, StdCheats {
     function setUp() public virtual {
         // Instantiate the contract-under-test.
         unwind = new Unwind();
+
+        for (uint i = 0; i < unwind.knownContractsLength(); i++) {
+            address target = unwind.knownContracts(i);
+            if (unwind.contractTypes(target) == Unwind.Type.STRATEGYV2 && IStrategy(target).state() == IStrategy.State.INVESTED) {
+                console2.log("%s: Divesting", ERC20(target).name());
+                IStrategy(target).divest();
+                console2.log("%s: Divested", ERC20(target).name());
+            }
+        }
     }