def test_delete(self):
"""
Test the DeleteGroup function.
"""
group = Database.create_group()
data = APIClient.delete_group(self.client, group.name)
self.assertEqual(data['error'], False)
with self.assertRaises(DoesNotExist):
Database.get_group(group.name)
def test_target_rename_association(self):
"""
Tests the RenameTarget API function, check to make sure Sessions, Targets, and Groups.
"""
target = Database.create_target()
target_name = target.name
session_id = Database.create_session(target_name).session_id
orig_group = Database.create_group('some_group')
orig_group.whitelist_member(target.name)
action_id = Database.create_action(target_name).action_id
data = APIClient.rename_target(self.client, target_name, 'TEST')
self.assertEqual(data['error'], False)
target = Database.get_target('TEST')
self.assertIsNotNone(target)
with self.assertRaises(DoesNotExist):
Database.get_target(target_name)
self.assertEqual(target.name, 'TEST')
session = Database.get_session(session_id)
self.assertEqual(session.target_name, 'TEST')
action = Database.get_action(action_id)
self.assertEqual(action.target_name, 'TEST')
group = Database.get_group(orig_group.name)
self.assertIn(target.name, group.member_names)
def test_create(self):
"""
Test the CreateGroup API function.
"""
data = APIClient.create_group(self.client, 'SOME GROUP')
self.assertEqual(data['error'], False)
self.assertIsNotNone(Database.get_group('SOME GROUP'))
def test_whitelist_member(self):
"""
Test the whitelist_member function.
"""
group = Database.create_group('SOME GROUP')
target = Database.create_target()
group.whitelist_member(target.name)
group = Database.get_group('SOME GROUP')
self.assertListEqual([target.name], group.whitelist_members)
def test_add_member(self):
"""
Test the GetGroup API function.
"""
Database.create_group('TEST GROUP')
target = Database.create_target()
data = APIClient.add_group_member(self.client, 'TEST GROUP',
target.name)
self.assertEqual(data['error'], False)
group = Database.get_group('TEST GROUP')
self.assertListEqual(group.whitelist_members, [target.name])
def test_remove_member(self):
"""
Test the RemoveGroupMember function.
"""
targets = [
Database.create_target().name,
Database.create_target().name,
Database.create_target().name,
]
Database.create_group('REMOVE GROUP', targets)
data = APIClient.remove_group_member(self.client, 'REMOVE GROUP',
targets[0])
self.assertEqual(data['error'], False)
group = Database.get_group('REMOVE GROUP')
self.assertListEqual(sorted(targets[1:]), sorted(group.member_names))
def test_blacklist_member(self):
"""
Test the BlacklistGroupMember function.
"""
targets = [
Database.create_target().name,
Database.create_target().name,
Database.create_target().name,
]
Database.create_group('BLACKLIST GROUP', targets)
data = APIClient.blacklist_group_member(self.client, 'BLACKLIST GROUP',
targets[0])
self.assertEqual(data['error'], False)
group = Database.get_group('BLACKLIST GROUP')
self.assertListEqual(sorted(targets[1:]), sorted(group.member_names))
self.assertListEqual([targets[0]], group.blacklist_members)
