fix enumerability of role given using _setupRole

OpenZeppelin · Feb 19, 2021 · f2dcdb2 · f2dcdb2
1 parent db0b96b
commit f2dcdb2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/contracts/access/AccessControlEnumerable.sol b/contracts/access/AccessControlEnumerable.sol
@@ -52,4 +52,12 @@ abstract contract AccessControlEnumerable is AccessControl {
         super.revokeRole(role, account);
         _roleMembers[role].remove(account);
     }
+
+    /**
+     * @dev Overload {_setupRole} to track enumerable memberships
+     */
+    function _setupRole(bytes32 role, address account) internal virtual override {
+        super._setupRole(role, account);
+        _roleMembers[role].add(account);
+    }
 }