def _update_or_delete_entity_container(entity_version, learning_container_year,
type_entity):
if not entity_version:
_delete_entity(learning_container_year, type_entity)
else:
update_or_create_entity_container_year_with_components(
entity_version.entity, learning_container_year, type_entity)
def _reinitialize_entities_before_proposal(learning_container_year,
initial_entities_by_type):
for type_entity, id_entity in initial_entities_by_type.items():
initial_entity = entity.get_by_internal_id(id_entity)
if initial_entity:
update_or_create_entity_container_year_with_components(
initial_entity, learning_container_year, type_entity)
else:
current_entity_container_year = entity_container_year.find_by_learning_container_year_and_linktype(
learning_container_year, type_entity)
if current_entity_container_year is not None:
current_entity_container_year.delete()
def test_update_or_create_entity_container_year_with_components_type_requirement(self):
"""In this test, we ensure that when we create an entity_container type requirement,
we have an entity_component created"""
an_entity = EntityFactory()
a_learning_container_year = LearningContainerYearFactory(academic_year=self.academic_year)
LearningComponentYearFactory(acronym="CM", learning_container_year=a_learning_container_year)
LearningComponentYearFactory(acronym="TP", learning_container_year=a_learning_container_year)
link_type = random.choice(ENTITY_TYPES_VOLUME)
update_or_create_entity_container_year_with_components(an_entity, a_learning_container_year, link_type)
self.assertEqual(EntityContainerYear.objects.all().count(), 1)
self.assertEqual(EntityComponentYear.objects.all().count(), 2)
def test_update_or_create_entity_container_year_with_components_type_allocation(self):
"""In this test, we ensure that when we create an entity_container type allocation,
we have NO entity_component created"""
an_entity = EntityFactory()
a_learning_container_year = LearningContainerYearFactory(academic_year=self.academic_year)
LearningComponentYearFactory(acronym="CM", learning_container_year=a_learning_container_year)
LearningComponentYearFactory(acronym="TP", learning_container_year=a_learning_container_year)
link_type = entity_container_year_link_type.ALLOCATION_ENTITY
update_or_create_entity_container_year_with_components(an_entity, a_learning_container_year, link_type)
self.assertEqual(EntityContainerYear.objects.all().count(), 1)
self.assertEqual(EntityComponentYear.objects.all().count(), 0)
def test_update_or_create_entity_container_year_with_components_type_requirement(self):
"""In this test, we ensure that when we create an entity_container type requirement,
we have an entity_component created"""
an_entity = EntityFactory()
a_learning_container_year = LearningContainerYearFactory(academic_year=self.academic_year)
LearningComponentYearFactory(acronym="PM", learning_container_year=a_learning_container_year)
LearningComponentYearFactory(acronym="PP", learning_container_year=a_learning_container_year)
link_type = random.choice(REQUIREMENT_ENTITIES)
business_edition.update_or_create_entity_container_year_with_components(
an_entity, a_learning_container_year, link_type
)
self.assertEqual(EntityContainerYear.objects.filter(
learning_container_year=a_learning_container_year).count(), 1)
self.assertEqual(EntityComponentYear.objects.filter(
entity_container_year__learning_container_year=a_learning_container_year
).count(), 2)
