tests: status

Author: ric-evans

skydriver/rest_handlers.py

@@ -39,6 +39,7 @@
     is_testing,
 )
 from .database import schema
+from .database.schema import PENDING_EWMS_WORKFLOW
 from .ewms import request_stop_on_ewms
 from .k8s.scan_backlog import put_on_backlog
 from .k8s.scanner_instance import SkyScanK8sJobFactory
@@ -1046,14 +1047,21 @@ async def get(self, scan_id: str) -> None:
                     pods_411["pod_message"] = "pod(s) not found"
                     LOGGER.exception(e)
 
-        # respond
+        # scan state
         scan_state = await get_scan_state(manifest, self.ewms_rc)
-        if manifest.ewms_workflow_id:
+
+        # ewms
+        if (
+            manifest.ewms_workflow_id
+            and manifest.ewms_workflow_id != PENDING_EWMS_WORKFLOW
+        ):
             clusters = await ewms.get_taskforce_phases(
                 self.ewms_rc, manifest.ewms_workflow_id
             )
         else:
             clusters = []
+
+        # respond
         resp = {
             "scan_state": scan_state,
             "is_deleted": manifest.is_deleted,

tests/integration/dummy_ewms.py

@@ -9,7 +9,7 @@
 
 app = Flask(__name__)
 
-DONT_CALL_IT_A_DB: dict[str, Any] = {}
+DONT_CALL_IT_A_DB__WORKFLOWS: dict[str, Any] = {}
 
 
 @app.route("/v0/workflows", methods=["POST"])
@@ -28,7 +28,7 @@ def dummy_workflows_post():
         # add more fields only if needed in tests--keep things simple
     }
 
-    DONT_CALL_IT_A_DB[workflow_id] = minimal_wf_doc
+    DONT_CALL_IT_A_DB__WORKFLOWS[workflow_id] = minimal_wf_doc
 
     return jsonify(
         {
@@ -39,21 +39,37 @@ def dummy_workflows_post():
 
 @app.route("/v0/workflows/<workflow_id>", methods=["GET"])
 def dummy_workflows_get(workflow_id: str):
-    return jsonify(DONT_CALL_IT_A_DB[workflow_id])
+    return jsonify(DONT_CALL_IT_A_DB__WORKFLOWS[workflow_id])
 
 
 @app.route("/v0/workflows/<workflow_id>/actions/abort", methods=["POST"])
 def dummy_workflows_abort(workflow_id: str):
-    DONT_CALL_IT_A_DB[workflow_id].update({"deactivated": "abort"})
+    DONT_CALL_IT_A_DB__WORKFLOWS[workflow_id].update({"deactivated": "abort"})
     return jsonify({})
 
 
 @app.route("/v0/workflows/<workflow_id>/actions/finished", methods=["POST"])
 def dummy_workflows_finished(workflow_id: str):
-    DONT_CALL_IT_A_DB[workflow_id].update({"deactivated": "finished"})
+    DONT_CALL_IT_A_DB__WORKFLOWS[workflow_id].update({"deactivated": "finished"})
     return jsonify({})
 
 
+@app.route("/v0/query/taskforces", methods=["POST"])
+def dummy_query_taskforces():
+    workflow_id = request.get_json("workflow_id")
+
+    # respond with correctly-syntaxed gibberish
+    resp = {
+        "taskforces": [
+            {
+                "taskforce": f"TF-{workflow_id}",
+                "phase": "the-best-phase-ever",
+            }
+        ]
+    }
+    return jsonify(resp)
+
+
 if __name__ == "__main__":
     app.run(
         debug=True,