def test_update_non_shareable(self):
"""
Tests updating groups that are non-shareable.
"""
self.create_common_users_and_groups()
sync = SyncUserAndGroups(
tsurl=TS_URL,
username=TS_USER,
password=TS_PASSWORD,
disable_ssl=True,
)
auag = sync.get_all_users_and_groups()
# Change Group 1 and Group 2 and verify change took.
group1 = auag.get_group("Group 1")
group1.visibility = Visibility.NON_SHAREABLE
group2 = auag.get_group("Group 2")
group2.visibility = Visibility.DEFAULT
# sync updates
sync.sync_users_and_groups(users_and_groups=auag)
# verify changes
auag = sync.get_all_users_and_groups()
self.assertEqual(
auag.get_group("Group 1").visibility, Visibility.NON_SHAREABLE)
self.assertEqual(
auag.get_group("Group 2").visibility, Visibility.DEFAULT)
self.assertEqual(
auag.get_group('Group "3"').visibility, Visibility.NON_SHAREABLE)
def test_delete_one_group(self):
"""Tests deleting a single groups."""
self.create_common_users_and_groups()
sync = SyncUserAndGroups(
tsurl=TS_URL,
username=TS_USER,
password=TS_PASSWORD,
disable_ssl=True,
)
sync.delete_group(groupname="Group1")
auag = sync.get_all_users_and_groups()
self.assertFalse(auag.has_group("Group1"))
self.assertTrue(auag.has_group("Group2"))
self.assertTrue(auag.has_group('Group"3"'))
def test_delete_user_list(self):
"""Tests deleting of a list of users."""
self.create_common_users_and_groups()
sync = SyncUserAndGroups(
tsurl=TS_URL,
username=TS_USER,
password=TS_PASSWORD,
disable_ssl=True,
)
sync.delete_users(usernames=["User1", "User2", "UserX"])
auag = sync.get_all_users_and_groups()
self.assertFalse(auag.has_user("User1"))
self.assertFalse(auag.has_user("User2"))
self.assertTrue(auag.has_user('User"3"'))
def test_getting_all(self):
"""
Tests getting all users from the server.
"""
self.create_common_users_and_groups()
sync = SyncUserAndGroups(
tsurl=TS_URL,
username=TS_USER,
password=TS_PASSWORD,
disable_ssl=True,
)
auag = sync.get_all_users_and_groups()
# There are four constant users, tsadmin, guest, su, system
self.assertEqual(auag.number_users(), 7)
# There are two constant groups, Administrator and System
self.assertEqual(auag.number_groups(), 6)
def test_syncing_user_and_groups(self):
"""
Tests adding users and groups to ThoughtSpot.
"""
sync = SyncUserAndGroups(
tsurl=TS_URL,
username=TS_USER,
password=TS_PASSWORD,
disable_ssl=True,
)
auag = sync.get_all_users_and_groups()
group1 = auag.get_group("Group 1")
self.assertEqual("Group 1", group1.name)
self.assertEqual("This is Group 1", group1.displayName)
self.assertEqual("A group for testing.", group1.description)
self.assertEqual([], group1.groupNames)
self.assertEqual(Visibility.DEFAULT, group1.visibility)
group2 = auag.get_group("Group 2")
self.assertEqual("Group 2", group2.name)
self.assertEqual("This is Group 2", group2.displayName)
self.assertEqual("Another group for testing.", group2.description)
self.assertEqual(["Group 1"], group2.groupNames)
self.assertEqual(Visibility.NON_SHAREABLE, group2.visibility)
user1 = auag.get_user("User1")
self.assertEqual("User1", user1.name)
self.assertEqual("User 1", user1.displayName)
self.assertEqual(["All", "Group 1"], user1.groupNames)
self.assertEqual(Visibility.DEFAULT, user1.visibility)
user2 = auag.get_user("User2")
self.assertEqual("User2", user2.name)
self.assertEqual("User 2", user2.displayName)
self.assertEqual(sorted(["All", "Group 1", "Group 2"]),
sorted(user2.groupNames))
self.assertEqual(Visibility.NON_SHAREABLE, user2.visibility)
def dump_users_and_groups(args):
"""
Gets users and groups from the server and dumps them in the correct format.
:param args: The command line arguments.
"""
sync = SyncUserAndGroups(
tsurl=args.ts_url,
username=args.username,
password=args.password,
disable_ssl=args.disable_ssl,
)
all_users_and_groups = sync.get_all_users_and_groups()
print("writing to %s" % args.filename)
if args.output_type == "json":
with open(args.filename, "w") as outfile:
outfile.write(all_users_and_groups.to_json())
else:
writer = UGXLSWriter()
writer.write(users_and_groups=all_users_and_groups,
filename=args.filename)