def test_get_included_files_uids(self):
test_parent = FileObject(binary=b'parent_file')
test_child = FileObject(binary=b'1st child')
test_child2 = FileObject(binary=b'2nd child')
test_parent.add_included_file(test_child)
test_parent.add_included_file(test_child2)
self.assertEqual(len(test_parent.get_included_files_uids()), 2, 'number of uids not correct')
self.assertIn(test_child.get_uid(), test_parent.get_included_files_uids(), 'uid of first file not found')
self.assertIn(test_child2.get_uid(), test_parent.get_included_files_uids(), 'uid of second file not found')
def test_get_included_files_uids(self):
test_parent = FileObject(binary=b'parent_file')
test_child = FileObject(binary=b'1st child')
test_child2 = FileObject(binary=b'2nd child')
test_parent.add_included_file(test_child)
test_parent.add_included_file(test_child2)
assert len(test_parent.get_included_files_uids()
) == 2, 'number of uids not correct'
assert test_child.uid in test_parent.get_included_files_uids(
), 'uid of first file not found'
assert test_child2.uid in test_parent.get_included_files_uids(
), 'uid of second file not found'
def test_object_processing_one_child(self):
root_object = FileObject(binary=b'root_file')
child_object = FileObject(binary=b'first_child_object')
root_object.add_included_file(child_object)
self.base_plugin.in_queue.put(root_object)
processed_object = self.base_plugin.out_queue.get()
self.assertEqual(processed_object.uid, root_object.uid, 'uid changed')
self.assertTrue(
child_object.uid in root_object.get_included_files_uids(),
'child object not in processed file')
