Fix testFlowTime test

rainlanguage · Aug 12, 2024 · dc2e925 · dc2e925
1 parent 7446ad2
commit dc2e925
Showing 1 changed file with 13 additions and 75 deletions.
diff --git a/test/concrete/flowErc1155/FlowTimeTest.sol b/test/concrete/flowErc1155/FlowTimeTest.sol
@@ -2,7 +2,6 @@
 pragma solidity ^0.8.19;
 
 import {Test, Vm} from "forge-std/Test.sol";
-import {StdUtils} from "forge-std/StdUtils.sol";
 import {FlowERC1155} from "../../../src/concrete/erc1155/FlowERC1155.sol";
 import {
     FlowUtilsAbstractTest,
@@ -14,96 +13,35 @@ import {
 import {IFlowERC1155V5} from "../../../src/interface/unstable/IFlowERC1155V5.sol";
 import {EvaluableV2, SignedContextV1} from "rain.interpreter.interface/interface/IInterpreterCallerV2.sol";
 import {FlowERC1155Test} from "../../abstract/FlowERC1155Test.sol";
-import {MockERC20} from "../../../lib/rain.factory/lib/rain.interpreter.interface/lib/forge-std/src/mocks/MockERC20.sol";
-import {FlowTransferV1} from "../../../src/interface/deprecated/v3/IFlowV3.sol";
-import {ERC20Transfer, ERC721Transfer, ERC1155Transfer} from "src/interface/unstable/IFlowV5.sol";
 import {SignContextLib} from "test/lib/SignContextLib.sol";
+import {DEFAULT_STATE_NAMESPACE} from "rain.interpreter.interface/interface/IInterpreterV2.sol";
+import {IInterpreterStoreV2} from "rain.interpreter.interface/interface/IInterpreterStoreV2.sol";
 
 contract FlowTimeTest is FlowUtilsAbstractTest, FlowERC1155Test {
     using SignContextLib for Vm;
 
-    /// Should validate multiple signed contexts
-    function testFlowTime(
-        string memory uri,
-        uint256[] memory context0,
-        uint256[] memory context1,
-        uint256 fuzzedKeyAlice
-    ) public {
-        (IFlowERC1155V5 erc1155Flow, EvaluableV2 memory evaluable) = deployIFlowERC1155V5(uri);
-
-        // Ensure the fuzzed key is within the valid range for secp256k1
-        uint256 aliceKey = (fuzzedKeyAlice % (SECP256K1_ORDER - 1)) + 1;
-        address alice = vm.addr(aliceKey);
-
-        MockERC20 erc20In = new MockERC20();
-        erc20In.initialize("InToken", "TKNI", 18);
-        vm.label(address(erc20In), "asdf");
-
-        MockERC20 erc20Out = new MockERC20();
-        erc20Out.initialize("OutToken", "TKNO", 18);
-        vm.label(address(erc20In), "asdf1");
-
-        ERC20Transfer[] memory erc20Transfers = new ERC20Transfer[](2);
-        // First ERC20 transfer
-        erc20Transfers[0] = ERC20Transfer({
-            from: alice,
-            to: address(erc1155Flow), // Contract address
-            token: address(erc20In),
-            amount: 10
-        });
-
-        // Second ERC20 transfer
-        erc20Transfers[1] = ERC20Transfer({
-            from: address(erc1155Flow), // Contract address
-            to: alice,
-            token: address(erc20Out),
-            amount: 20
-        });
+    function testFlowTime(string memory uri, uint256[] memory writeToStore) public {
+        vm.assume(writeToStore.length != 0);
 
-        deal(address(erc20In), alice, 1e18);
-        deal(address(erc20Out), alice, 1e18);
-
-        vm.startPrank(address(alice));
-
-        // Fund Alice and the contract with necessary tokens
-        erc20In.transfer(alice, erc20Transfers[0].amount);
-        //emit log_named_uint("Alice ERC20In Balance1", erc20In.balanceOf(alice));
-        erc20Out.transfer(address(erc1155Flow), erc20Transfers[1].amount);
-
-        // Approve ERC20 transfers
-        SignedContextV1[] memory signedContexts = new SignedContextV1[](2);
-        signedContexts[0] = vm.signContext(aliceKey, aliceKey, context0);
-        signedContexts[1] = vm.signContext(aliceKey, aliceKey, context1);
+        (IFlowERC1155V5 erc1155Flow, EvaluableV2 memory evaluable) = deployIFlowERC1155V5(uri);
 
-        uint256[] memory stack1 = generateFlowERC1155Stack(
+        uint256[] memory stack = generateFlowERC1155Stack(
             new ERC1155Transfer[](0),
             new ERC721Transfer[](0),
             new ERC20Transfer[](0),
             new ERC1155SupplyChange[](0),
             new ERC1155SupplyChange[](0)
         );
-        interpreterEval2MockCall(stack1, new uint256[](0));
-
-        // Perform the first flow with id 1234
-        uint256[] memory flowId1234 = new uint256[](1);
-        flowId1234[0] = 1234;
-        erc1155Flow.flow(evaluable, flowId1234, signedContexts);
 
-        // Fund Alice and the contract with necessary tokens
-        erc20In.transfer(alice, erc20Transfers[0].amount);
-        erc20Out.transfer(address(erc1155Flow), erc20Transfers[1].amount);
+        interpreterEval2MockCall(stack, writeToStore);
 
-        // Perform another flow with a different id 5678
-        uint256[] memory flowId5678 = new uint256[](1);
-        flowId5678[0] = 5678;
-        erc1155Flow.flow(evaluable, flowId5678, signedContexts);
+        vm.mockCall(address(iStore), abi.encodeWithSelector(IInterpreterStoreV2.set.selector), abi.encode());
 
-        // Fund Alice and the contract with necessary tokens
-        erc20In.transfer(alice, erc20Transfers[0].amount);
-        erc20Out.transfer(address(erc1155Flow), erc20Transfers[1].amount);
+        vm.expectCall(
+            address(iStore),
+            abi.encodeWithSelector(IInterpreterStoreV2.set.selector, DEFAULT_STATE_NAMESPACE, writeToStore)
+        );
 
-        // Attempt to perform the flow again with id 1234, should revert
-        vm.expectRevert();
-        erc1155Flow.flow(evaluable, flowId1234, signedContexts);
+        erc1155Flow.flow(evaluable, writeToStore, new SignedContextV1[](0));
     }
 }