Add more tests to confirm non-owner cannot remove hat from whitelist

decentdao · Nov 6, 2024 · f0d0acd · f0d0acd
1 parent e6e0bd1
commit f0d0acd
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/test/Azorius-LinearERC20VotingWithHatsProposalCreation.test.ts b/test/Azorius-LinearERC20VotingWithHatsProposalCreation.test.ts
@@ -255,6 +255,14 @@ describe('LinearERC20VotingWithHatsProposalCreation', () => {
     );
   });
 
+  it('Non-owner cannot remove a hat from the whitelist', async () => {
+    const hatId = 1; // Example hat ID
+
+    await expect(
+      linearERC20VotingWithHats.connect(rando).removeHatFromWhitelist(hatId),
+    ).to.be.revertedWith('Ownable: caller is not the owner');
+  });
+
   it('Cannot call setUp function again', async () => {
     const setupParams = ethers.AbiCoder.defaultAbiCoder().encode(
       ['address', 'address', 'address', 'uint32', 'uint256', 'uint256', 'address', 'uint256[]'],

diff --git a/test/Azorius-LinearERC721VotingWithHatsProposalCreation.test.ts b/test/Azorius-LinearERC721VotingWithHatsProposalCreation.test.ts
@@ -237,6 +237,14 @@ describe('LinearERC721VotingWithHatsProposalCreation', () => {
     );
   });
 
+  it('Non-owner cannot remove a hat from the whitelist', async () => {
+    const hatId = 1; // Example hat ID
+
+    await expect(
+      linearERC721VotingWithHats.connect(rando).removeHatFromWhitelist(hatId),
+    ).to.be.revertedWith('Ownable: caller is not the owner');
+  });
+
   it('Cannot call setUp function again', async () => {
     const setupParams = ethers.AbiCoder.defaultAbiCoder().encode(
       [