fix: improve test

bgd-labs · Jul 17, 2024 · 64b3be8 · 64b3be8
1 parent 18a99a3
commit 64b3be8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/test/UpgradableOwnableWithGuardian.t copy.sol b/test/UpgradableOwnableWithGuardian.t copy.sol
@@ -61,13 +61,14 @@ contract TestOfUpgradableOwnableWithGuardian is Test {
     withGuardian.updateGuardian(newGuardian);
   }
 
-  function test_updateGuardian_eoa(address newGuardian) external {
-    vm.assume(newGuardian != owner && newGuardian != guardian);
+  function test_updateGuardian_eoa(address eoa, address newGuardian) external {
+    vm.assume(eoa != owner && eoa != guardian);
 
+    vm.prank(eoa);
     vm.expectRevert(
       abi.encodeWithSelector(
         UpgradableOwnableWithGuardian.OnlyGuardianOrOwnerInvalidCaller.selector,
-        address(this)
+        eoa
       )
     );
     withGuardian.updateGuardian(newGuardian);