Add test_calculateL2TokenAddress_Registered

jwbda · Feb 9, 2024 · 6da2210 · 6da2210
1 parent 3362b78
commit 6da2210
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/test-foundry/L2CustomGateway.t.sol b/test-foundry/L2CustomGateway.t.sol
@@ -33,7 +33,7 @@ contract L2CustomGatewayTest is L2ArbitrumGatewayTest {
         );
     }
 
-    function test_calculateL2TokenAddress_Registered() public {
+    function test_calculateL2TokenAddress_Registered() public virtual {
         address l2CustomToken = _registerToken();
         assertEq(
             l2CustomGateway.calculateL2TokenAddress(l1CustomToken),

diff --git a/test-foundry/L2ReverseCustomGateway.t.sol b/test-foundry/L2ReverseCustomGateway.t.sol
@@ -25,6 +25,15 @@ contract L2ReverseCustomGatewayTest is L2CustomGatewayTest {
     }
 
     /* solhint-disable func-name-mixedcase */
+    function test_calculateL2TokenAddress_Registered() public override {
+        address l1CustomToken = _registerToken();
+        assertEq(
+            l2CustomGateway.calculateL2TokenAddress(l1CustomToken),
+            address(l2MintedToken),
+            "Invalid L2 token"
+        );
+    }
+
     function test_finalizeInboundTransfer() public override {
         // fund gateway with tokens being withdrawn
         vm.prank(address(l2ReverseCustomGateway));