lda.chat/lda-wf
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: migrate service async tests

Browse Source
This commit is contained in:
lda
2026-06-04 19:12:12 +07:00 Unverified
parent f9673a4ad0
commit e001b524d6
8 changed files with 163 additions and 213 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/wf_mcp/service/test_adapters.py
+6 -8
View File
@@ -1,7 +1,5 @@
from __future__ import annotations
import asyncio
from wf_mcp.broker import WfMcpService
from wf_mcp.models import ConnectionConfig
from wf_mcp.storage import FileStore
@@ -13,21 +11,21 @@ from ..test_support import (
from .conftest import ContentOnlyOutputAdapter
def test_service_catalog_preserves_json_schema_description_metadata() -> None:
async def test_service_catalog_preserves_json_schema_description_metadata() -> None:
service = WfMcpService(store=FileStore(local_temp_root() / "schema_doc_store"))
service.register_connection(
ConnectionConfig(id="demo.personal", server="demo", account="personal")
)
service.register_adapter("demo", FakeAdapter())
asyncio.run(service.refresh_connection_catalog("demo.personal"))
await service.refresh_connection_catalog("demo.personal")
payload = service.get_catalog().as_payload()
node = payload["nodes"][0]
assert node["input_schema"]["properties"]["text"]["description"] == "Text to echo"
def test_service_preserves_content_only_tool_output_schema_for_workflows() -> None:
async def test_service_preserves_content_only_tool_output_schema_for_workflows() -> None:
service = WfMcpService(store=FileStore(local_temp_root() / "content_only_store"))
service.register_connection(
ConnectionConfig(
@@ -36,7 +34,7 @@ def test_service_preserves_content_only_tool_output_schema_for_workflows() -> No
)
service.register_adapter("everything", ContentOnlyOutputAdapter())
asyncio.run(service.refresh_connection_catalog("everything.default"))
await service.refresh_connection_catalog("everything.default")
payload = service.get_catalog().as_payload()
node = payload["nodes"][0]
@@ -44,14 +42,14 @@ def test_service_preserves_content_only_tool_output_schema_for_workflows() -> No
assert node["output_schema"]["required"] == ["content"]
def test_service_wrapped_tool_adapter_model_validates_simple_schema_types() -> None:
async def test_service_wrapped_tool_adapter_model_validates_simple_schema_types() -> None:
service = WfMcpService(store=FileStore(local_temp_root() / "adapter_model_store"))
service.register_connection(
ConnectionConfig(id="demo.personal", server="demo", account="personal")
)
service.register_adapter("demo", FakeAdapter())
asyncio.run(service.refresh_connection_catalog("demo.personal"))
await service.refresh_connection_catalog("demo.personal")
source = service.capability_sources["demo.personal"]
spec = source.capabilities.node_specs["demo.personal.echo_tool"]