diff --git a/tests/registry/models/test_document_validation.py b/tests/registry/models/test_document_validation.py index 77bd485..12df168 100644 --- a/tests/registry/models/test_document_validation.py +++ b/tests/registry/models/test_document_validation.py @@ -42,19 +42,21 @@ def make_skill_frontmatter_payload( ) -> str: """Builds frontmatter payload YAML for skill conversion tests.""" canonical_name = name or skill_id + x_personal_mcp: dict[str, Any] = { + "id": skill_id, + "version": version, + "tags": list(tags), + "capabilities": list(capabilities or (f"resource://skills/{canonical_name}/document",)), + "depends_on": list(depends_on), + } + if references is not None: + x_personal_mcp["references"] = references + payload: dict[str, Any] = { "name": canonical_name, "description": description, - "x-personal-mcp": { - "id": skill_id, - "version": version, - "tags": list(tags), - "capabilities": list(capabilities or (f"resource://skills/{canonical_name}/document",)), - "depends_on": list(depends_on), - }, + "x-personal-mcp": x_personal_mcp, } - if references is not None: - payload["x-personal-mcp"]["references"] = references return yaml.safe_dump(payload, sort_keys=False) diff --git a/tests/web/test_endpoint_connections.py b/tests/web/test_endpoint_connections.py index ec3e284..ca25327 100644 --- a/tests/web/test_endpoint_connections.py +++ b/tests/web/test_endpoint_connections.py @@ -27,7 +27,7 @@ class TestMcpHttpEndpoints: @pytest.mark.asyncio async def test_serves_docs_entrypoint(self, client: AsyncClient) -> None: """Ensures GET /docs returns the docs site entrypoint response.""" - response = await client.get("/docs") + response = await client.get("/docs", follow_redirects=True) assert response.status_code == 200 assert "text/html" in response.headers["content-type"]