diff --git a/SAS/TMSS/backend/test/t_tmssapp_specification_django_API.py b/SAS/TMSS/backend/test/t_tmssapp_specification_django_API.py index 577932cd868df45bc7335df4a3c67f91ecbb56b3..b8d82ead9e47fbab49e00befc8742bedc634eee3 100755 --- a/SAS/TMSS/backend/test/t_tmssapp_specification_django_API.py +++ b/SAS/TMSS/backend/test/t_tmssapp_specification_django_API.py @@ -479,6 +479,19 @@ class SchedulingUnitDraftTest(unittest.TestCase): #When auto_ingest=False (in project), the scheduling units should be created with ingest_permission_required = True self.assertEqual(True, entry.ingest_permission_required) + def test_SchedulingUnitDraft_gets_created_with_correct_default_piggyback_allowed_flags(self): + + # setup + project = models.Project.objects.create(**Project_test_data()) + scheduling_set = models.SchedulingSet.objects.create(**SchedulingSet_test_data(project=project)) + scheduling_unit_draft = models.SchedulingUnitDraft.objects.create(**SchedulingUnitDraft_test_data(scheduling_set=scheduling_set)) + + scheduling_unit_draft.refresh_from_db() + + # assert + self.assertEqual(scheduling_unit_draft.piggyback_allowed_tbb, project.piggyback_allowed_tbb) + self.assertEqual(scheduling_unit_draft.piggyback_allowed_aartfaac, project.piggyback_allowed_aartfaac) + class TaskDraftTest(unittest.TestCase): @@ -722,6 +735,20 @@ class SchedulingUnitBlueprintTest(unittest.TestCase): #When auto_ingest=False (in project), the scheduling units should be created with ingest_permission_required = True self.assertEqual(True, entry.ingest_permission_required) + def test_SchedulingUnitBlueprint_gets_created_with_correct_default_piggyback_allowed_flags(self): + + # setup + project = models.Project.objects.create(**Project_test_data()) + scheduling_set = models.SchedulingSet.objects.create(**SchedulingSet_test_data(project=project)) + scheduling_unit_draft = models.SchedulingUnitDraft.objects.create(**SchedulingUnitDraft_test_data(scheduling_set=scheduling_set)) + scheduling_unit_blueprint = models.SchedulingUnitBlueprint.objects.create(**SchedulingUnitBlueprint_test_data(draft=scheduling_unit_draft)) + + scheduling_unit_blueprint.refresh_from_db() + + # assert + self.assertEqual(scheduling_unit_blueprint.piggyback_allowed_tbb, scheduling_unit_draft.piggyback_allowed_tbb) + self.assertEqual(scheduling_unit_blueprint.piggyback_allowed_aartfaac, scheduling_unit_draft.piggyback_allowed_aartfaac) + class TaskBlueprintTest(unittest.TestCase): @classmethod