def test_init(self):
authuser1 = GroupUser(group='group1', user='******')
self.assertIsNotNone(authuser1)
self.assertEqual("<GroupUser(id='n/a', group='group1', user='******')>", str(authuser1))
authuser2 = GroupUser(id=1, group='group1', user='******')
self.assertIsNotNone(authuser2)
self.assertEqual("<GroupUser(id='1', group='group1', user='******')>", str(authuser2))
def _upload_groups(self, yaml_data, verbose=False):
if 'groups' in yaml_data:
for group_name in yaml_data['groups'].keys():
auth_group = AuthoriseGroup(name=group_name, parent=None)
result = self.storage_engine.session.add(auth_group)
yaml_obj = yaml_data['groups'][group_name]
if 'roles' in yaml_obj:
roles_list = yaml_obj['roles']
splits = roles_list.split(",")
for role_name in splits:
role_name = role_name.strip()
group_role = GroupRole(group=group_name,
role=role_name)
if verbose is True:
outputLog(self, group_role)
self.storage_engine.session.add(group_role)
if 'groups' in yaml_obj:
groups_list = yaml_obj['groups']
splits = groups_list.split(",")
for element in splits:
inner_group_name = element.strip()
group_group = GroupGroup(group=group_name,
subgroup=inner_group_name)
if verbose is True:
outputLog(self, group_group)
self.storage_engine.session.add(group_group)
if 'users' in yaml_obj:
users_list = yaml_obj['users']
splits = users_list.split(",")
for user_name in splits:
user_name = user_name.strip()
group_user = GroupUser(group=group_name,
user=user_name)
if verbose is True:
outputLog(self, group_user)
self.storage_engine.session.add(group_user)