chore: simpler setup

morpho-org · Apr 4, 2024 · 7a71b62 · 7a71b62
1 parent 725c9ee
commit 7a71b62
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/forge/HalmosTest.sol b/test/forge/HalmosTest.sol
@@ -83,6 +83,7 @@ contract HalmosTest is SymTest, Test {
 
         bytes memory args;
 
+        // Todo: make it possible to call any market
         if (selector == morpho.supply.selector) {
             args = abi.encode(marketParams, assets, shares, onBehalf, emptyData);
         } else if (selector == morpho.repay.selector) {
@@ -93,7 +94,8 @@ contract HalmosTest is SymTest, Test {
             address borrower = svm.createAddress("borrower");
             args = abi.encode(marketParams, borrower, assets, shares, emptyData);
         } else if (selector == morpho.flashLoan.selector) {
-            address token = svm.createAddress("token");
+            // Todo: make it more general
+            address token = address(loanToken);
             args = abi.encode(marketParams, token, assets, emptyData);
         } else {
             args = svm.createBytes(1024, "data");