web tests

2026-06-21 17:57:01 -05:00
parent 4958eeb3ef
commit b9bb11ac02
11 changed files with 242 additions and 20 deletions
@@ -0,0 +1,125 @@
+from __future__ import annotations
+
+import pytest
+
+from personal_mcp.registry.models.prompt import PromptArgumentEntry
+from personal_mcp.registry.models.registry import PromptRecord
+from personal_mcp.registry.models.registry import PromptSummaryPayload
+from personal_mcp.registry.models.registry import ReferenceRecord
+from personal_mcp.registry.models.registry import SkillPatternPayload
+from personal_mcp.registry.models.registry import SkillRecord
+from personal_mcp.registry.models.registry import SkillSummaryPayload
+
+pytestmark = pytest.mark.unit
+
+
+def _make_skill_record() -> SkillRecord:
+    return SkillRecord(
+        skill_id="demo-skill",
+        name="demo-skill",
+        description="demo skill",
+        version="1.2.3",
+        tags=("testing", "catalog"),
+        capabilities=("resource://skills/demo-skill/document", "resource://catalog/skills_index"),
+        depends_on=("base-skill",),
+        document_uri="resource://skills/demo-skill/document",
+        document_relpath="skills/demo-skill/SKILL.md",
+        document_content="# demo",
+        references={
+            "zeta": ReferenceRecord(
+                ref_id="zeta",
+                uri="resource://skills/demo-skill/references/zeta",
+                relpath="skills/demo-skill/references/zeta.md",
+                mime_type="text/markdown",
+                title="Zeta",
+                content="# zeta",
+            ),
+            "alpha": ReferenceRecord(
+                ref_id="alpha",
+                uri="resource://skills/demo-skill/references/alpha",
+                relpath="skills/demo-skill/references/alpha.md",
+                mime_type="text/markdown",
+                title="Alpha",
+                content="# alpha",
+            ),
+        },
+    )
+
+
+def _make_prompt_record() -> PromptRecord:
+    return PromptRecord(
+        prompt_id="demo-prompt",
+        name="demo-prompt",
+        description="demo prompt",
+        version="0.9.0",
+        tags=("testing",),
+        capabilities=("resource://prompts/demo-prompt/document",),
+        arguments={
+            "topic": PromptArgumentEntry(
+                type="string",
+                required=True,
+                description="topic to discuss",
+            )
+        },
+        document_uri="resource://prompts/demo-prompt/document",
+        document_relpath="prompts/demo-prompt/PROMPT.md",
+        document_content="# demo",
+    )
+
+
+def test_skill_pattern_payload_from_record_shape() -> None:
+    record = _make_skill_record()
+
+    payload = SkillPatternPayload.from_record(record).model_dump()
+
+    assert payload == {
+        "id": "demo-skill",
+        "name": "demo-skill",
+        "version": "1.2.3",
+        "description": "demo skill",
+        "tags": ["testing", "catalog"],
+        "depends_on": ["base-skill"],
+        "capabilities": ["resource://skills/demo-skill/document", "resource://catalog/skills_index"],
+        "resources": ["resource://skills/demo-skill/document", "resource://catalog/skills_index"],
+    }
+
+
+def test_skill_summary_payload_from_record_shape() -> None:
+    record = _make_skill_record()
+
+    payload = SkillSummaryPayload.from_record(record).model_dump()
+
+    assert payload == {
+        "id": "demo-skill",
+        "name": "demo-skill",
+        "description": "demo skill",
+        "tags": ["testing", "catalog"],
+        "capabilities": ["resource://skills/demo-skill/document", "resource://catalog/skills_index"],
+        "version": "1.2.3",
+        "document_uri": "resource://skills/demo-skill/document",
+        "detail_uri": "resource://catalog/skills/demo-skill",
+        "resources": {
+            "document": "resource://skills/demo-skill/document",
+            "references": [
+                "resource://skills/demo-skill/references/alpha",
+                "resource://skills/demo-skill/references/zeta",
+            ],
+        },
+    }
+
+
+def test_prompt_summary_payload_from_record_shape() -> None:
+    record = _make_prompt_record()
+
+    payload = PromptSummaryPayload.from_record(record).model_dump()
+
+    assert payload == {
+        "id": "demo-prompt",
+        "name": "demo-prompt",
+        "description": "demo prompt",
+        "tags": ["testing"],
+        "capabilities": ["resource://prompts/demo-prompt/document"],
+        "version": "0.9.0",
+        "document_uri": "resource://prompts/demo-prompt/document",
+        "detail_uri": "resource://catalog/prompts/demo-prompt",
+    }