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])