upd: events; fxRoot tests

0xPolygon · May 25, 2023 · f9bf56d · f9bf56d
1 parent 2a2bd82
commit f9bf56d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/test/FxRoot.t.sol b/test/FxRoot.t.sol
@@ -2,6 +2,8 @@
 pragma solidity ^0.8.0;
 
 import {FxBase, MockMessageProcessor} from "@utils/FxBase.sol";
+import {FxRoot} from "contracts/FxRoot.sol";
+import {FxChild} from "contracts/FxChild.sol";
 
 contract FxRootTest is FxBase {
     address public alice = makeAddr("alice");
@@ -14,8 +16,14 @@ contract FxRootTest is FxBase {
     }
 
     function test_SetFxChildAndStateSender() public {
-        assertEq(address(fxRoot.fxChild()), address(fxChild));
-        assertEq(address(fxRoot.stateSender()), address(stateSender));
+        FxChild fxChild2 = new FxChild();
+        FxRoot fxRoot2 = new FxRoot(address(stateSender));
+
+        fxRoot2.setFxChild(address(fxChild2));
+        fxChild2.setFxRoot(address(fxRoot2));
+
+        assertEq(address(fxRoot2.fxChild()), address(fxChild2));
+        assertEq(address(fxRoot2.stateSender()), address(stateSender));
     }
 
     function test_CannotSetFxChild() public {

diff --git a/test/utils/Events.sol b/test/utils/Events.sol
@@ -3,6 +3,7 @@ pragma solidity ^0.8.0;
 
 abstract contract Events {
     event TokenMappedERC20(address indexed rootToken, address indexed childToken); // root chain
+    event TokenMappedERC721(address indexed rootToken, address indexed childToken);
     event TokenMapped(address indexed rootToken, address indexed childToken); // child chain
 
     event FxWithdrawERC20(