feat: add function get_role_manager

Signed-off-by: Andreas Bichinger <[email protected]>
casbin · Dec 19, 2020 · 7fb1879 · 7fb1879
1 parent 43e4329
commit 7fb1879
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/casbin/core_enforcer.py b/casbin/core_enforcer.py
@@ -121,6 +121,10 @@ def set_watcher(self, watcher):
         self.watcher = watcher
         pass
 
+    def get_role_manager(self):
+        """gets the current role manager."""
+        return self.rm
+
     def set_role_manager(self, rm):
         """sets the current role manager."""
 

diff --git a/tests/test_enforcer.py b/tests/test_enforcer.py
@@ -165,7 +165,7 @@ def test_enforce_rbac_with_pattern(self):
                          get_examples("rbac_with_pattern_policy.csv"))
 
         #set matching function to key_match2
-        e.rm.add_matching_func(casbin.util.key_match2)
+        e.get_role_manager().add_matching_func(casbin.util.key_match2)
 
         self.assertTrue(e.enforce("alice", "/book/1", "GET"))
         self.assertTrue(e.enforce("alice", "/book/2", "GET"))
@@ -177,7 +177,7 @@ def test_enforce_rbac_with_pattern(self):
         self.assertTrue(e.enforce("bob", "/pen/2", "GET"))
 
         #replace key_match2 with key_match3
-        e.rm.add_matching_func(casbin.util.key_match3)
+        e.get_role_manager().add_matching_func(casbin.util.key_match3)
         self.assertTrue(e.enforce("alice", "/book2/1", "GET"))
         self.assertTrue(e.enforce("alice", "/book2/2", "GET"))
         self.assertTrue(e.enforce("alice", "/pen2/1", "GET"))