fix: status when task is removed

DevoInc · Oct 4, 2024 · 6adac2d · 6adac2d
1 parent 84777b4
commit 6adac2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/test_api_tasks.py b/tests/integration/test_api_tasks.py
@@ -66,7 +66,7 @@ def test_jobs_cycle(setup_client, setup_query, job_name):
     jobs_ids = [job["id"] for job in result["object"] if job["friendlyName"].startswith("devo-sdk-api-test")]
     for job_id in jobs_ids:
         result = setup_client.remove_job(job_id)
-        assert result["object"]["status"] == "REMOVED"
+        assert result["object"]["status"] in ["REMOVED", "COMPLETED"]
 
 
 if __name__ == "__main__":