fix: handle cancelled jobs (#2)

* fix: handle cancelled jobs
voi-oss · Aug 30, 2023 · 0a1cebe · 0a1cebe
1 parent 5721ee5
commit 0a1cebe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/dbttoolkit/dbt_cloud/clients/dbt_cloud_client.py b/src/dbttoolkit/dbt_cloud/clients/dbt_cloud_client.py
@@ -144,6 +144,7 @@ def retrieve_completed_runs(
             run
             for run in runs
             if run["is_complete"]
+            and not run["is_cancelled"]
             and run["project_id"] == self.project_id
             and run["environment_id"] == self.environment_id
         ]

diff --git a/tests/dbt_cloud/actions/test_retrieve_most_recent_artifact.py b/tests/dbt_cloud/actions/test_retrieve_most_recent_artifact.py
@@ -59,6 +59,16 @@ def rest_api_run_result(job_id, dbt_cloud_ids):
         "id": 2,
         "is_complete": True,
         "is_success": False,
+        "is_cancelled": False,
+        "created_at": "2022-06-10 11:30:00.321339+00:00",
+        **common,
+    }
+
+    cancelled = {
+        "id": 3,
+        "is_complete": True,
+        "is_success": False,
+        "is_cancelled": True,
         "created_at": "2022-06-10 11:30:00.321339+00:00",
         **common,
     }
@@ -67,11 +77,12 @@ def rest_api_run_result(job_id, dbt_cloud_ids):
         "id": 1,
         "is_complete": True,
         "is_success": True,
+        "is_cancelled": False,
         "created_at": "2022-06-09 11:30:00.321339+00:00",
         **common,
     }
 
-    return {"data": [failure, success]}
+    return {"data": [failure, cancelled, success]}
 
 
 @fixture