stuff

tests/artifacts/test_catalog.py

@@ -0,0 +1,70 @@
+from __future__ import annotations
+
+from wf_artifacts import (
+    DependencyDiagnostic,
+    DiagnosticSeverity,
+    RequiredCapability,
+    WorkflowArtifact,
+    artifact_catalog_entry,
+    artifact_node_name,
+)
+
+
+def artifact() -> WorkflowArtifact:
+    return WorkflowArtifact(
+        id="summarize_docs",
+        version=2,
+        title="Summarize Docs",
+        description="Summarize retrieved documentation.",
+        input_schema={"type": "object", "properties": {"query": {"type": "string"}}},
+        output_schema={
+            "type": "object",
+            "properties": {"summary": {"type": "string"}},
+        },
+        outcomes=("done", "failed"),
+        plan={"name": "summarize_docs", "nodes": [{"id": "hidden"}]},
+        required_capabilities={
+            "context7.query-docs": RequiredCapability(
+                logical_source="context7",
+                capability_name="query-docs",
+                kind="tool",
+            )
+        },
+    )
+
+
+def test_artifact_node_name_is_stable_and_versioned() -> None:
+    name = artifact_node_name(artifact())
+
+    assert name == "workflow.summarize_docs.v2"
+
+
+def test_artifact_catalog_entry_is_nodespec_shaped_without_internal_plan() -> None:
+    entry = artifact_catalog_entry(artifact())
+    dumped = entry.model_dump(mode="json")
+
+    assert dumped["name"] == "workflow.summarize_docs.v2"
+    assert dumped["display_name"] == "Summarize Docs"
+    assert dumped["description"] == "Summarize retrieved documentation."
+    assert dumped["outcomes"] == ["done", "failed"]
+    assert dumped["input_schema"]["properties"]["query"]["type"] == "string"
+    assert dumped["output_schema"]["properties"]["summary"]["type"] == "string"
+    assert dumped["required_sources"] == ["context7"]
+    assert "plan" not in dumped
+
+
+def test_artifact_catalog_entry_includes_dependency_diagnostics() -> None:
+    diagnostic = DependencyDiagnostic(
+        severity=DiagnosticSeverity.ERROR,
+        code="source_missing",
+        logical_ref="context7.query-docs",
+        bound_source="context7.personal",
+        message="Bound source is unavailable.",
+    )
+
+    entry = artifact_catalog_entry(artifact(), diagnostics=[diagnostic])
+    dumped = entry.model_dump(mode="json")
+
+    assert len(dumped["diagnostics"]) == 1
+    assert dumped["diagnostics"][0]["code"] == "source_missing"
+    assert dumped["diagnostics"][0]["logical_ref"] == "context7.query-docs"