Passing the instance type in the test

unity-sds · Dec 30, 2024 · 50355a6 · 50355a6
1 parent dc816e5
commit 50355a6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/unity-test/system/integration/step_defs/test_cwl_workflows_with_airflow_api.py b/unity-test/system/integration/step_defs/test_cwl_workflows_with_airflow_api.py
@@ -89,6 +89,7 @@ def trigger_dag(airflow_api_url, airflow_api_auth, venue, test_case):
         request_memory = DAG_PARAMETERS[test_case]["request_memory"]
         request_cpu = DAG_PARAMETERS[test_case]["request_cpu"]
         request_storage = DAG_PARAMETERS[test_case]["request_storage"]
+        request_instance_type = DAG_PARAMETERS[test_case]["request_instance_type"]
         use_ecr = DAG_PARAMETERS[test_case]["use_ecr"]
         response = requests.post(
             f"{airflow_api_url}/api/v1/dags/{DAG_ID}/dagRuns",
@@ -100,6 +101,7 @@ def trigger_dag(airflow_api_url, airflow_api_auth, venue, test_case):
                     "request_memory": f"{request_memory}",
                     "request_cpu": f"{request_cpu}",
                     "request_storage": f"{request_storage}",
+                    "request_instance_type": f"{request_instance_type}",
                     "use_ecr": use_ecr,
                 }
             },