TMSS-153: added task_draft as optional parameter

SAS/TMSS/test/tmss_test_data_django_models.py

@@ -145,13 +145,16 @@ def SchedulingUnitBlueprint_test_data():
             "draft": models.SchedulingUnitDraft.objects.create(**SchedulingUnitDraft_test_data()),
             "requirements_template": models.SchedulingUnitTemplate.objects.create(**SchedulingUnitTemplate_test_data())}
 
-def TaskBlueprint_test_data():
+def TaskBlueprint_test_data(task_draft: models.TaskDraft = None):
+    if task_draft is None:
+        task_draft = models.TaskDraft.objects.create(**TaskDraft_test_data())
+
     return {"name": 'my_task_blueprint',
                         "description": "",
                         "tags": [],
                         "specifications_doc": "{}",
                         "do_cancel": False,
-                        "draft": models.TaskDraft.objects.create(**TaskDraft_test_data()),
+                        "draft": task_draft,
                         "specifications_template": models.TaskTemplate.objects.create(**TaskTemplate_test_data()),
                         "scheduling_unit_blueprint": models.SchedulingUnitBlueprint.objects.create(**SchedulingUnitBlueprint_test_data())}