fix foundry test

matter-labs · Sep 6, 2024 · acd428b · acd428b
1 parent 068c5c0
commit acd428b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/l1-contracts/test/foundry/unit/concrete/Bridgehub/experimental_bridge.t.sol b/l1-contracts/test/foundry/unit/concrete/Bridgehub/experimental_bridge.t.sol
@@ -249,11 +249,12 @@ contract ExperimentalBridgeTest is Test {
     }
 
     function test_addToken_cannotBeCalledByRandomAddress(address randomAddress, address randomCaller) public {
-        if (randomCaller != bridgeOwner) {
-            vm.prank(randomCaller);
-            vm.expectRevert(bytes("Ownable: caller is not the owner"));
-            bridgeHub.addToken(randomAddress);
-        }
+        vm.assume(randomAddress != bridgeOwner);
+        vm.assume(randomAddress != bridgeHub.admin());
+
+        vm.prank(randomCaller);
+        vm.expectRevert(bytes("Bridgehub: not owner or admin"));
+        bridgeHub.addToken(randomAddress);
 
         assertTrue(!bridgeHub.tokenIsRegistered(randomAddress), "This random address is not registered as a token");