TMSS-152: added test parameter specifications_doc

SAS/TMSS/test/tmss_test_data_django_models.py

@@ -165,17 +165,20 @@ def SchedulingUnitDraft_test_data(name="my_scheduling_unit_draft", scheduling_se
             "scheduling_set": scheduling_set,
             "requirements_template": template }
 
-def TaskDraft_test_data(name: str="my_task_draft", specifications_template: models.TaskTemplate=None, scheduling_unit_draft: models.SchedulingUnitDraft=None) -> dict:
+def TaskDraft_test_data(name: str="my_task_draft", specifications_template: models.TaskTemplate=None, specifications_doc: dict=None, scheduling_unit_draft: models.SchedulingUnitDraft=None) -> dict:
     if specifications_template is None:
         specifications_template = models.TaskTemplate.objects.create(**TaskTemplate_test_data())
 
+    if specifications_doc is None:
+        specifications_doc = get_default_json_object_for_schema(specifications_template.schema)
+
     if scheduling_unit_draft is None:
         scheduling_unit_draft = models.SchedulingUnitDraft.objects.create(**SchedulingUnitDraft_test_data())
 
     return {"name": name,
             "description": "",
             "tags": [],
-            "specifications_doc": get_default_json_object_for_schema(specifications_template.schema),
+            "specifications_doc": specifications_doc,
             "copy_reason": models.CopyReason.objects.get(value='template'),
             "copies": None,
             "scheduling_unit_draft": scheduling_unit_draft,