Skip to content

Commit 7fb1879

Browse files
committed
feat: add function get_role_manager
Signed-off-by: Andreas Bichinger <andreas.bichinger@gmail.com>
1 parent 43e4329 commit 7fb1879

2 files changed

Lines changed: 6 additions & 2 deletions

File tree

casbin/core_enforcer.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,10 @@ def set_watcher(self, watcher):
121121
self.watcher = watcher
122122
pass
123123

124+
def get_role_manager(self):
125+
"""gets the current role manager."""
126+
return self.rm
127+
124128
def set_role_manager(self, rm):
125129
"""sets the current role manager."""
126130

tests/test_enforcer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def test_enforce_rbac_with_pattern(self):
165165
get_examples("rbac_with_pattern_policy.csv"))
166166

167167
#set matching function to key_match2
168-
e.rm.add_matching_func(casbin.util.key_match2)
168+
e.get_role_manager().add_matching_func(casbin.util.key_match2)
169169

170170
self.assertTrue(e.enforce("alice", "/book/1", "GET"))
171171
self.assertTrue(e.enforce("alice", "/book/2", "GET"))
@@ -177,7 +177,7 @@ def test_enforce_rbac_with_pattern(self):
177177
self.assertTrue(e.enforce("bob", "/pen/2", "GET"))
178178

179179
#replace key_match2 with key_match3
180-
e.rm.add_matching_func(casbin.util.key_match3)
180+
e.get_role_manager().add_matching_func(casbin.util.key_match3)
181181
self.assertTrue(e.enforce("alice", "/book2/1", "GET"))
182182
self.assertTrue(e.enforce("alice", "/book2/2", "GET"))
183183
self.assertTrue(e.enforce("alice", "/pen2/1", "GET"))

0 commit comments

Comments
 (0)