def test_build_entity_version_by_entity_id_parent(self):
all_current_entities_version = entity_version.find_all_current_entities_version(
)
result = entity_version._build_entity_version_by_entity_id(
all_current_entities_version)
expected_keys = [v.entity_id for v in all_current_entities_version]
self.assertListEqual(list(sorted(result.keys())),
sorted(expected_keys))
def test_build_direct_children_by_entity_version_id(self):
entity_version_by_entity_id = entity_version._build_entity_version_by_entity_id(
entity_version.find_all_current_entities_version())
result = entity_version._build_direct_children_by_entity_version_id(entity_version_by_entity_id)
count_entities_version_with_children = 4
self.assertEqual(len(result.keys()), count_entities_version_with_children)
root_direct_children = [self.SC, self.LOCI]
self.assertEqual(set(result[self.root.id]), set(root_direct_children))
sc_direct_children = [self.MATH, self.PHYS]
self.assertEqual(set(result[self.SC.id]), set(sc_direct_children))
self.assertNotIn(self.MATH.id, result) # No children for MATH
def test_build_all_children_by_entity_version_id(self):
all_current_entites_versions = entity_version.find_all_current_entities_version(
)
entity_version_by_entity_id = entity_version._build_entity_version_by_entity_id(
all_current_entites_versions)
direct_children_by_entity_version_id = entity_version \
._build_direct_children_by_entity_version_id(entity_version_by_entity_id)
result = entity_version._build_all_children_by_entity_version_id(
direct_children_by_entity_version_id)
count_entities_version_with_children = 4
self.assertEqual(len(result.keys()),
count_entities_version_with_children)
root_all_children = [
self.SC, self.LOCI, self.MATH, self.PHYS, self.URBA, self.BARC
]
self.assertEqual(set(result[self.root.id]), set(root_all_children))
sc_all_children = [self.MATH, self.PHYS]
self.assertEqual(set(result[self.SC.id]), set(sc_all_children))
self.assertNotIn(self.MATH.id, result.keys())
