feat: 完成工作流调度与上传下载 API

2026-08-08 21:04:21 +08:00
commit a8d11eafc6
32 changed files with 3675 additions and 0 deletions
@@ -0,0 +1,107 @@
+from fastapi.testclient import TestClient
+
+from app.main import app
+
+
+def definition() -> dict:
+    return {
+        "name": "echo-flow",
+        "version": 1,
+        "nodes": [
+            {
+                "id": "step",
+                "node_type": "echo",
+                "inputs": {"file_uri": "input.video_uri"},
+            }
+        ],
+        "edges": [],
+        "entry_inputs": {"video_uri": "file"},
+        "final_outputs": {"result": "step.file_uri"},
+    }
+
+
+def test_workflow_crud_and_publish() -> None:
+    with TestClient(app) as client:
+        created = client.post(
+            "/api/admin/workflows",
+            json={
+                "id": "echo-flow",
+                "name": "Echo Flow",
+                "description": "demo",
+                "definition": definition(),
+            },
+        )
+        assert created.status_code == 200
+        assert created.json()["id"] == "echo-flow"
+
+        assert client.get("/api/admin/workflows").status_code == 200
+        assert client.get("/api/admin/workflows/echo-flow").status_code == 200
+        assert client.get("/api/admin/workflows/missing").status_code == 404
+
+        validated = client.post(
+            "/api/admin/workflows/echo-flow/validate",
+            json=definition(),
+        )
+        assert validated.status_code == 200
+        assert validated.json()["valid"] is True
+        assert client.post(
+            "/api/admin/workflows/missing/validate",
+            json=definition(),
+        ).status_code == 404
+
+        published = client.post("/api/admin/workflows/echo-flow/publish")
+        assert published.status_code == 200
+        assert published.json()["published"] == "echo-flow"
+        assert client.post("/api/admin/workflows/missing/publish").status_code == 404
+
+        versions = client.get("/api/admin/workflows/echo-flow/versions")
+        assert versions.status_code == 200
+        assert len(versions.json()) == 1
+        assert client.get("/api/admin/workflows/missing/versions").status_code == 404
+
+        assert client.delete("/api/admin/workflows/echo-flow").status_code == 200
+        assert client.delete("/api/admin/workflows/echo-flow").status_code == 404
+
+
+def test_workflow_slug_without_id() -> None:
+    with TestClient(app) as client:
+        created = client.post(
+            "/api/admin/workflows",
+            json={
+                "name": "Echo Flow",
+                "definition": definition(),
+            },
+        )
+        assert created.status_code == 200
+        assert created.json()["id"] == "echo-flow"
+
+
+def test_workflow_validation_error() -> None:
+    with TestClient(app) as client:
+        response = client.post(
+            "/api/admin/workflows",
+            json={
+                "id": "bad",
+                "name": "Bad",
+                "definition": {
+                    "name": "Bad",
+                    "version": 1,
+                    "nodes": [
+                        {"id": "a", "node_type": "x"},
+                        {"id": "a", "node_type": "y"},
+                    ],
+                    "edges": [],
+                },
+            },
+        )
+        assert response.status_code == 422
+
+
+def test_publish_workflow_without_version() -> None:
+    with TestClient(app) as client:
+        db = app.state.db
+        db.upsert_workflow(
+            {"id": "empty", "name": "Empty", "published": 0, "latest_version": 0}
+        )
+        response = client.post("/api/admin/workflows/empty/publish")
+        assert response.status_code == 422