Modify implementation

OffchainLabs · Apr 29, 2024 · 37c8538 · 37c8538
1 parent 1827189
commit 37c8538
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/test-e2e/tokenBridgeDeploymentTest.ts b/test-e2e/tokenBridgeDeploymentTest.ts
@@ -193,9 +193,9 @@ describe('tokenBridge', () => {
     await checkL2Ownership(l2Deployment, usingFeeToken)
     await checkLogicContracts(usingFeeToken, l2Deployment)
 
-    // This should always be the last check, because WETH gateway registration is a 
+    // This should always be the last check, because WETH gateway registration is a
     // separate step that should be done after token bridge is atomically deployed
-    if (!usingFeeToken) {
+    if (!usingFeeToken && !isLocalDeployment(l1Provider, l2Provider)) {
       await checkWethGatewayIsRegistered(
         L1WethGateway__factory.connect(l1Deployment.wethGateway, l1Provider),
         L1GatewayRouter__factory.connect(l1Deployment.router, l1Provider),
@@ -835,3 +835,13 @@ interface L2DeploymentAddresses {
   upgradeExecutor: string
   multicall: string
 }
+
+function isLocalDeployment(
+  l1Provider: JsonRpcProvider,
+  l2Provider: JsonRpcProvider
+) {
+  return (
+    l1Provider.network.chainId === 412346 ||
+    l2Provider.network.chainId === 333333
+  )
+}