Merge pull request #11 from dadadave80:chore/fork-test-setup

Chore/fork-test-setup

Author: dadadave80

test/ChronicleTest.t.sol

@@ -17,7 +17,7 @@ contract ChronicleTest is DeployedChronicleState {
     }
 
     function test_ChronicleFacetsDeployed() public view {
-        assertEq(facetAddresses.length, 5);
+        assertEq(facetAddresses.length, 6);
         for (uint256 i; i < facetAddresses.length; ++i) {
             assertNotEq(address(facetAddresses[i]), address(0));
         }

test/fork/previewnet/ChroniclePreviewnetStateTest.t.sol

@@ -0,0 +1,62 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.30;
+
+import {Test} from "forge-std/Test.sol";
+import {PartiesFacet} from "@chronicle/facets/PartiesFacet.sol";
+import {ProductsFacet} from "@chronicle/facets/ProductsFacet.sol";
+import {SupplyChainFacet} from "@chronicle/facets/SupplyChainFacet.sol";
+import {DiamondCutFacet} from "@diamond/facets/DiamondCutFacet.sol";
+import {DiamondLoupeFacet} from "@diamond/facets/DiamondLoupeFacet.sol";
+import {OwnableRolesFacet} from "@diamond/facets/OwnableRolesFacet.sol";
+import {Role, Rating} from "@chronicle-types/PartyStorage.sol";
+
+contract ChroniclePreviewnetStateTest is Test {
+    DiamondCutFacet public diamondCutFacet;
+    DiamondLoupeFacet public diamondLoupeFacet;
+    OwnableRolesFacet public ownableRolesFacet;
+    PartiesFacet public partiesFacet;
+    ProductsFacet public productsFacet;
+    SupplyChainFacet public supplyChainFacet;
+
+    uint256 previewnetFork;
+    string PRE_RPC_URL = vm.envString("PRE_RPC_URL");
+
+    address TESTER = makeAddr("tester");
+
+    modifier previewnet() {
+        _previewnetSetup();
+        _;
+    }
+
+    function setUp() public {
+        previewnetFork = vm.createFork(PRE_RPC_URL);
+    }
+
+    function test_ChronicleOwner() public previewnet {
+        assertEq(ownableRolesFacet.owner(), vm.envAddress("PRE_CHR_OWNER"));
+    }
+
+    function test_RegisterParty() public previewnet {
+        vm.startPrank(TESTER);
+        partiesFacet.registerParty("Party", Role.Retailer);
+
+        assertTrue(partiesFacet.hasActiveRole(TESTER, Role.Retailer));
+        assertEq(partiesFacet.getParty(TESTER).name, "Party");
+        assertEq(partiesFacet.getParty(TESTER).addr, TESTER);
+        assertEq(uint8(partiesFacet.getParty(TESTER).role), uint8(Role.Retailer));
+        assertEq(partiesFacet.getParty(TESTER).active, true);
+        assertEq(partiesFacet.getParty(TESTER).frozen, false);
+        assertEq(uint8(partiesFacet.getParty(TESTER).rating), uint8(Rating.Zero));
+    }
+
+    function _previewnetSetup() internal {
+        vm.selectFork(previewnetFork);
+
+        diamondCutFacet = DiamondCutFacet(payable(vm.envAddress("PRE_CHR_ADDR")));
+        diamondLoupeFacet = DiamondLoupeFacet(payable(vm.envAddress("PRE_CHR_ADDR")));
+        ownableRolesFacet = OwnableRolesFacet(payable(vm.envAddress("PRE_CHR_ADDR")));
+        partiesFacet = PartiesFacet(payable(vm.envAddress("PRE_CHR_ADDR")));
+        productsFacet = ProductsFacet(payable(vm.envAddress("PRE_CHR_ADDR")));
+        supplyChainFacet = SupplyChainFacet(payable(vm.envAddress("PRE_CHR_ADDR")));
+    }
+}

test/helpers/ChronicleDeployer.sol

@@ -11,6 +11,7 @@ import {HelperContract} from "@diamond-test/helpers/HelperContract.sol";
 import {Chronicle} from "@chronicle/Chronicle.sol";
 import {PartiesFacet} from "@chronicle/facets/PartiesFacet.sol";
 import {ProductsFacet} from "@chronicle/facets/ProductsFacet.sol";
+import {SupplyChainFacet} from "@chronicle/facets/SupplyChainFacet.sol";
 import {InitHTCKeyTypes} from "@chronicle/initializers/InitHTCKeyTypes.sol";
 
 abstract contract ChronicleDeployer is HelperContract {
@@ -21,13 +22,14 @@ abstract contract ChronicleDeployer is HelperContract {
         OwnableRolesFacet ownableRolesFacet = new OwnableRolesFacet();
         PartiesFacet partiesFacet = new PartiesFacet();
         ProductsFacet productsFacet = new ProductsFacet();
+        SupplyChainFacet supplyChainFacet = new SupplyChainFacet();
 
         // deploy initializers
         MultiInit multiInit = new MultiInit();
         ERC165Init erc165Init = new ERC165Init();
         InitHTCKeyTypes initHtcKeyTypes = new InitHTCKeyTypes();
 
-        FacetCut[] memory facetCuts = new FacetCut[](5);
+        FacetCut[] memory facetCuts = new FacetCut[](6);
 
         facetCuts[0] = FacetCut({
             facetAddress: address(diamondCutFacet),
@@ -59,6 +61,12 @@ abstract contract ChronicleDeployer is HelperContract {
             functionSelectors: _generateSelectors("ProductsFacet")
         });
 
+        facetCuts[5] = FacetCut({
+            facetAddress: address(supplyChainFacet),
+            action: FacetCutAction.Add,
+            functionSelectors: _generateSelectors("SupplyChainFacet")
+        });
+
         address[] memory initAddr = new address[](2);
         bytes[] memory initData = new bytes[](2);
 

test/helpers/TestStates.sol

@@ -6,6 +6,7 @@ import {DiamondLoupeFacet} from "@diamond/facets/DiamondLoupeFacet.sol";
 import {OwnableRolesFacet} from "@diamond/facets/OwnableRolesFacet.sol";
 import {PartiesFacet} from "@chronicle/facets/PartiesFacet.sol";
 import {ProductsFacet} from "@chronicle/facets/ProductsFacet.sol";
+import {SupplyChainFacet} from "@chronicle/facets/SupplyChainFacet.sol";
 import {ChronicleDeployer} from "@chronicle-test/helpers/ChronicleDeployer.sol";
 
 abstract contract DeployedChronicleState is ChronicleDeployer {
@@ -15,20 +16,28 @@ abstract contract DeployedChronicleState is ChronicleDeployer {
     OwnableRolesFacet public ownableRolesFacet;
     PartiesFacet public partiesFacet;
     ProductsFacet public productsFacet;
+    SupplyChainFacet public supplyChainFacet;
     address[] public facetAddresses;
-    string[5] public facetNames =
-        ["DiamondCutFacet", "DiamondLoupeFacet", "OwnableRolesFacet", "PartiesFacet", "ProductsFacet"];
+    string[6] public facetNames = [
+        "DiamondCutFacet",
+        "DiamondLoupeFacet",
+        "OwnableRolesFacet",
+        "PartiesFacet",
+        "ProductsFacet",
+        "SupplyChainFacet"
+    ];
     address public constant CHRONICLE_OWNER = address(1337);
 
-    function setUp() public {
-        vm.startPrank(CHRONICLE_OWNER);
+    function setUp() public virtual {
+        vm.prank(CHRONICLE_OWNER);
         chronicle = _deployChronicle(CHRONICLE_OWNER);
 
         diamondCutFacet = DiamondCutFacet(chronicle);
         diamondLoupeFacet = DiamondLoupeFacet(chronicle);
         ownableRolesFacet = OwnableRolesFacet(chronicle);
         partiesFacet = PartiesFacet(chronicle);
         productsFacet = ProductsFacet(chronicle);
+        supplyChainFacet = SupplyChainFacet(chronicle);
 
         facetAddresses = diamondLoupeFacet.facetAddresses();
     }