def test_add_groups_to_group(self):
"""Tests creation of following structure.
Group: parent_group {
Group: member_group1,
Group: member_group2
}
"""
parent_group = "parent_group"
member_group1 = "member_group1"
member_group2 = "member_group2"
ts = TSApiWrapper(DISABLE_SSL)
ts.login(HOSTPORT, USERNAME, PASSWORD)
# Create Parent Group
ts.create_group(parent_group, parent_group)
# Create members
ts.create_group(member_group1, member_group1)
ts.create_group(member_group2, member_group2)
# Add member 1 to parent
parent_gid = ts.get_groupid_with_name(parent_group).data
member_gid1 = ts.get_groupid_with_name(member_group1).data
member_gid2 = ts.get_groupid_with_name(member_group2).data
# Update parent group with members.
self.assertEqual(len(ts.list_groups_in_group(parent_gid).data), 0)
ts.add_groups_to_group([member_gid1], parent_gid)
self.assertEqual(len(ts.list_groups_in_group(parent_gid).data), 1)
ts.add_groups_to_group([member_gid2], parent_gid)
self.assertEqual(len(ts.list_groups_in_group(parent_gid).data), 2)
# Cleanup
ts.delete_groups([member_gid1, member_gid2, parent_gid])
def test_empty_group_relationship(self):
"""Tests creation of following structure.
Group: parent_group {
Group: member_group {}
User: member_user
}
Updates parent_group to {}
Checks for parent_group members to be empty.
"""
parent_group = "parent_group"
member_group = "member_group"
member_user = "******"
ts = TSApiWrapper(DISABLE_SSL)
ts.login(HOSTPORT, USERNAME, PASSWORD)
# Create Parent Group
ts.create_group(parent_group, parent_group)
# Create members
ts.create_group(member_group, member_group)
ts.create_user(member_user, member_user)
# Update parent group with members.
parent_gid = ts.get_groupid_with_name(parent_group).data
member_gid = ts.get_groupid_with_name(member_group).data
member_uid = ts.get_userid_with_name(member_user).data
ts.update_groups_to_group([member_gid], parent_gid)
ts.update_users_to_group([member_uid], parent_gid)
# Test for membership.
groups = ts.list_groups_in_group(parent_gid).data
self.assertEqual(groups[0].id, member_gid)
users = ts.list_users_in_group(parent_gid).data
self.assertEqual(users[0].id, member_uid)
# Update parent group with no members.
ts.update_users_to_group([], parent_gid)
ts.update_groups_to_group([], parent_gid)
# Test for membership.
groups = ts.list_groups_in_group(parent_gid).data
self.assertEqual(groups, [])
users = ts.list_users_in_group(parent_gid).data
self.assertEqual(users, [])
# Cleanup
ts.delete_groups([member_gid, parent_gid])
ts.delete_users([member_uid])