Merge pull request #6 from Kolektivo/feat-5/update-badger-address

feat: enables updating badger address
Kolektivo · Jun 8, 2022 · 8ff6f4c · 8ff6f4c
2 parents 6e8009c + 67aa66f
commit 8ff6f4c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/contracts/Roles.sol b/contracts/Roles.sol
@@ -335,4 +335,8 @@ contract Roles is Module {
         );
         return execAndReturnData(to, value, data, operation);
     }
+
+    function updateBadger(address newBadger) external onlyOwner {
+        badger = IBadger(newBadger);
+    }
 }
diff --git a/test/Roles.spec.ts b/test/Roles.spec.ts
@@ -1680,4 +1680,20 @@ describe("RolesModifier", async () => {
       ).to.be.revertedWith("FunctionNotAllowed");
     });
   });
+
+  describe("updateBadger()", () => {
+    it("reverts if not authorized", async () => {
+      const { modifier } = await txSetup();
+      await expect(modifier.updateBadger(AddressOne)).to.be.revertedWith(
+        "Ownable: caller is not the owner"
+      );
+    });
+
+    it("sets new badger address", async () => {
+      const { modifier, owner } = await setupRolesWithOwnerAndInvoker();
+      await modifier.connect(owner).updateBadger(AddressOne);
+
+      expect(await modifier.badger()).to.equal(AddressOne);
+    });
+  });
 });