fix: tests for hats gatekeeper and ownable

privacy-scaling-explorations · Apr 17, 2024 · 9653154 · 9653154
1 parent 188b259
commit 9653154
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/contracts/tests/HatsGatekeeper.test.ts b/contracts/tests/HatsGatekeeper.test.ts
@@ -150,8 +150,9 @@ describe("HatsProtocol Gatekeeper", () => {
       });
 
       it("should fail to set MACI instance when the caller is not the owner", async () => {
-        await expect(hatsGatekeeperSingle.connect(voter).setMaciInstance(signerAddress)).to.be.revertedWith(
-          "Ownable: caller is not the owner",
+        await expect(hatsGatekeeperSingle.connect(voter).setMaciInstance(signerAddress)).to.be.revertedWithCustomError(
+          hatsGatekeeperSingle,
+          "OwnableUnauthorizedAccount",
         );
       });
     });
@@ -252,9 +253,9 @@ describe("HatsProtocol Gatekeeper", () => {
       });
 
       it("should fail to set MACI instance when the caller is not the owner", async () => {
-        await expect(hatsGatekeeperMultiple.connect(voter).setMaciInstance(signerAddress)).to.be.revertedWith(
-          "Ownable: caller is not the owner",
-        );
+        await expect(
+          hatsGatekeeperMultiple.connect(voter).setMaciInstance(signerAddress),
+        ).to.be.revertedWithCustomError(hatsGatekeeperMultiple, "OwnableUnauthorizedAccount");
       });
     });