From 428d7db3cdc9525ba721162b65a95934f385f456 Mon Sep 17 00:00:00 2001
From: josojo <josojo@hotmail.de>
Date: Fri, 12 Jan 2024 06:50:17 +0100
Subject: [PATCH] Improve test of testSetAndCheckClaimed (#185)

* improve test of testSetAndCheckClaimed

* linting
---
 test/ForkableBridge.t.sol | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/test/ForkableBridge.t.sol b/test/ForkableBridge.t.sol
index 9777dafb..8989868f 100644
--- a/test/ForkableBridge.t.sol
+++ b/test/ForkableBridge.t.sol
@@ -844,5 +844,27 @@ contract ForkableBridgeWrapperTest is Test {
         ForkableBridgeWrapper(child1).setAndCheckClaimed(index);
         vm.expectRevert(selector);
         ForkableBridgeWrapper(child2).setAndCheckClaimed(index);
+
+        // check that the _verify function within PolygonZkEVMBridge calls really
+        // the _setAndCheckClaimed function of the ForkableBridge contract
+        // and not the _setAndCheckClaimed function of the PolygonZkEVMBridge contract
+        bytes32[32] memory smtProof;
+        for (uint256 i = 0; i < 32; i++) {
+            smtProof[i] = forkableBridge.branch(i);
+        }
+        vm.expectRevert(selector);
+        // forkableBridgeWrapper.claimAsset will call the PolygonZkEVMBridge.claimAsset function
+        ForkableBridgeWrapper(child1).claimAsset(
+            smtProof,
+            uint32(index),
+            bytes32(0),
+            bytes32(0),
+            uint32(0),
+            address(0x1),
+            uint32(1),
+            address(0),
+            uint256(1),
+            bytes("0x")
+        );
     }
 }