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_find_pass(self):
"""
This test creates a session and attempts to find it,
it's target, and it's history document.
"""
session1 = Database.create_session()
session2 = Database.get_session(session1.session_id)
self.assertEqual(session1, session2)
self.assertIsNotNone(Database.get_target(session2.target_name))
self.assertIsNotNone(session1.history)
self.assertIsNotNone(session2.history)
