class TestAccessmanager():
am = None
jsonpath = None
def get_jsonpath(self):
am = AccessManager()
# noinspection PyTypeChecker
am.init(DummyBot()) # Get variables like jsonpath
path = am.jsonpath
del am
return path
# noinspection PyPep8Naming
def delete_JSON_if_exists(self):
if os.path.exists(self.jsonpath):
os.remove(self.jsonpath)
def setup(self):
self.jsonpath = self.get_jsonpath()
self.delete_JSON_if_exists()
self.am = AccessManager()
self.am.jsonfile = self.am.jsonpath
# noinspection PyTypeChecker
self.am.init(DummyBot())
def teardown(self):
if os.path.exists(self.am.jsonpath):
os.remove(self.am.jsonpath)
if os.path.exists(self.am.jsonpath + ".bak"):
os.remove(self.am.jsonpath + ".bak")
def test_accessmanager_init(self):
assert self.am.jsonpath is not None
#TODO: something else to test?
def test_accessmanager_json(self):
acls = {"a": "b", "c": {"d": [1, 2 ,"3"]}}
groups = {"A": "B", "C": {"D": [10, 20 ,"30"]}}
self.am.acls = acls
self.am.groups = groups
self.am.write_JSON()
self.am.acls = None
self.am.groups = None
self.am.read_JSON()
assert self.am.acls == acls
assert self.am.groups == groups
def test_accessmanager_addgroup(self):
self.am.groups = {}
self.am.add_group("grp1")
assert self.am.groups == {'grp1': {'members': []}}
self.am.add_group("grp2", "member")
assert self.am.groups == {'grp2': {'members': ['member']}, 'grp1': {'members': []}}
self.am.add_group("grp3", ["member1", "member2"])
assert self.am.groups == {'grp3': {'members': ['member1', 'member2']},
'grp2': {'members': ['member']}, 'grp1': {'members': []}}
def test_accessmanager_removegroup(self):
self.am.groups = {}
self.am.remove_group("grp1")
assert self.am.groups == {}
self.am.groups = {'grp2': {'members': ['member']}, 'grp1': {'members': []}}
self.am.remove_group("grp2")
assert self.am.groups == {'grp1': {'members': []}}
def test_accessmanager_existsgroup(self):
self.am.groups = {}
self.am.add_group("grp1")
assert self.am.exists_group("grp1")
assert not self.am.exists_group("grp2")
def test_accessmanager_getgroup(self):
self.am.add_group("getgroup")
grp = self.am.get_group("getgroup")
assert grp.name == "getgroup"
def test_accessmanager_addtogroup(self):
self.am.add_group("test")
self.am.add_to_group("test", "user")
assert self.am.get_group("test").get_members() == ["user"]
def test_accessmanager_removefromgroup(self):
self.am.add_group("g1", ["a", "b"])
self.am.add_group("g2", ["a", "b"])
self.am.remove_from_group("g1", "a")
self.am.remove_from_group("g2", "b")
assert self.am.get_group("g1").get_members() == ["b"]
assert self.am.get_group("g2").get_members() == ["a"]
def test_accessmanager_createacl(self):
self.am.create_acl("a.b.c.*")
assert self.am.acls["a.b.c.*"] == {"groups": [], "members": []}
assert not "a.b.c.d" in self.am.acls.keys()
def test_accessmanager_removeacl(self):
self.am.create_acl("a")
self.am.create_acl("b")
self.am.remove_acl("a")
self.am.remove_acl("c")
assert list(self.am.acls.keys()) == ["b"]
def test_accessmanager_existsacl(self):
self.am.create_acl("a")
assert "a" in self.am.acls.keys()
assert "b" not in self.am.acls.keys()
def test_accessmanager_registeracl(self):
self.am.register_acl("acl1")
self.am.register_acl("acl1")
self.am.register_acl("acl1")
assert len(self.am.acls) == 1
assert ["%moderators"] == self.am.acls["acl1"]["groups"]
assert len(self.am.acls["acl1"]["members"]) == 0
self.am.add_group("a") # Make sure groups to test with exist
self.am.add_group("c")
self.am.register_acl("acl2", "a", "b")
self.am.register_acl("acl3", ["c"], ["d"])
assert self.am.acls["acl2"]["groups"] == ["a"]
assert self.am.acls["acl2"]["members"] == ["b"]
assert self.am.acls["acl3"]["groups"] == ["c"]
assert self.am.acls["acl3"]["members"] == ["d"]
def test_accessmanager_addgrouptoacl(self):
self.am.add_group("a") # Make sure groups to test with exist
self.am.add_group("b")
self.am.register_acl("acl", ["a"], [])
self.am.add_group_to_acl("acl", "b")
assert self.am.acls["acl"]["groups"] == ["a", "b"]
def test_accessmanager_removegroupfromacl(self):
self.am.add_group("a") # Make sure groups to test with exist
self.am.add_group("b")
self.am.register_acl("acl", ["a", "b"], [])
self.am.remove_group_from_acl("acl", "a")
assert self.am.acls["acl"]["groups"] == ["b"]
def test_accessmanager_addusertoacl(self):
self.am.register_acl("acl", [], ["a"])
self.am.add_user_to_acl("acl", "b")
assert self.am.acls["acl"]["members"] == ["a", "b"]
def test_accessmanager_removeuserfromacl(self):
self.am.register_acl("acl1", [], ["a", "b", "c"])
self.am.remove_user_from_acl("acl1", "b")
assert self.am.acls["acl1"]["members"] == ["a", "c"]
def test_accessmanager_expandgroups(self):
assert self.am.expand_groups("%moderators").sort() == ["%moderators", "%operators", "%owner"].sort()
assert self.am.expand_groups("%operators").sort() == ["%operators", "%owner"].sort()
assert self.am.expand_groups("%owner") == ["%owner"]
assert self.am.expand_groups(["test", "%operators"]).sort() == ["test", "%operators", "%owner"].sort()
def test_accessmanager_isinacl(self):
pass #TODO: Write access tests
