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

test ruff check fix fmt

Browse Source
This commit is contained in:
lda
2026-05-29 23:57:59 +07:00 Verified
parent 68c12e80b5
commit c3b6854e46
16 changed files with 59 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/wf_mcp/proxy/conftest.py
-1
View File
@@ -4,7 +4,6 @@ import sys
from typing import Any
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.proxy import create_proxy_client
from ..test_support import fixture_server_path, local_temp_root
tests/wf_mcp/proxy/test_admin.py
-2
View File
@@ -3,12 +3,10 @@ from __future__ import annotations
import asyncio
import json
import sys
from pathlib import Path
import mcp.types as mcp_types
from wf_mcp.broker import load_broker_config
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.proxy import create_proxy_client
from ..test_support import fixture_server_path, local_temp_root
tests/wf_mcp/proxy/test_errors.py
+1 -7
View File
@@ -1,15 +1,10 @@
from __future__ import annotations
import asyncio
import json
import sys
from pathlib import Path
from typing import Any
from wf_mcp.broker import load_broker_config
from wf_mcp.events import EventBus, InMemoryEventSink
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.proxy import ProxyRuntime, create_proxy_client
from wf_mcp.proxy import ProxyRuntime
from wf_mcp.proxy.reload_events import (
ProxyReloadResult,
reload_change_events,
@@ -17,7 +12,6 @@ from wf_mcp.proxy.reload_events import (
from wf_mcp.proxy.tools import ProxyToolPayload, ProxyToolsPage
from ..test_support import fixture_server_path, local_temp_root
from .conftest import structured
def test_proxy_runtime_reload_publishes_local_change_events() -> None:
tests/wf_mcp/server/conftest.py
-1
View File
@@ -5,7 +5,6 @@ import sys
from typing import Any
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.server import create_server_client
from ..test_support import fixture_server_path, local_temp_root
tests/wf_mcp/server/test_config.py
+1 -5
View File
@@ -1,14 +1,10 @@
from __future__ import annotations
import asyncio
import json
import sys
from wf_mcp.broker.config import build_service_from_config, load_broker_config
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.broker.config import build_service_from_config
from wf_mcp.server import create_server_client
from ..test_support import fixture_server_path, local_temp_root
from .conftest import server_config, structured
tests/wf_mcp/server/test_docs.py
+2 -2
View File
@@ -7,11 +7,11 @@ import sys
from mcp import types as mcp_types
from wf_mcp.broker.config import load_broker_config
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.models import BrokerConfig
from wf_mcp.server import create_server_client
from ..test_support import fixture_server_path, local_temp_root
from .conftest import server_config, structured
from .conftest import structured
def test_server_exposes_platform_documentation_resources() -> None:
tests/wf_mcp/server/test_tools.py
+3 -6
View File
@@ -1,15 +1,12 @@
from __future__ import annotations
import asyncio
import re
import sys
from typing import Any
from wf_mcp.models import BrokerConfig, ConnectionConfig
from wf_mcp.models import BrokerConfig
from wf_mcp.server import create_server_client
from ..test_support import fixture_server_path, local_temp_root
from .conftest import assert_safe_tool_maps, server_config, structured
from ..test_support import local_temp_root
from .conftest import assert_safe_tool_maps, server_config
def test_server_search_mode_pins_stable_control_and_workflow_tools() -> None:
tests/wf_mcp/service/conftest.py
+3 -19
View File
@@ -1,34 +1,18 @@
from __future__ import annotations
import asyncio
import shutil
from typing import Any, cast
from typing import Any
from wf_artifacts import FileDraftWorkspaceStore, WorkflowDeployment
from wf_authoring import NodeSpec, build_async_registry, node
from wf_core import END, NodeUse, RunStatus, RuntimeContext
from wf_mcp.broker import WfMcpService
from wf_authoring import node
from wf_core import END
from wf_mcp.capabilities import DiscoveredTool
from wf_mcp.models import AuthRecord, ConnectionConfig, RawWorkflowPlan
from wf_mcp.runtime import ToolExecutor
from wf_mcp.sdk import ToolCallResult
from wf_mcp.shared.errors import error_payload
from wf_mcp.storage import FileStore
from wf_platform import (
CapabilityBuckets,
CapabilitySource,
SourceVisibility,
)
from ..test_support import (
EchoInput,
EchoOutput,
FailingDiscoveryAdapter,
FakeAdapter,
echo_tool,
finalize_tool,
input_binding,
local_temp_root,
output_binding,
)
tests/wf_mcp/service/test_catalog.py
+1 -1
View File
@@ -4,7 +4,7 @@ import asyncio
import shutil
from wf_artifacts import FileDraftWorkspaceStore
from wf_authoring import NodeSpec, node
from wf_authoring import NodeSpec
from wf_core import RunStatus
from wf_mcp.broker import WfMcpService
from wf_mcp.models import ConnectionConfig
tests/wf_mcp/service/test_events.py
+1 -3
View File
@@ -20,10 +20,9 @@ from ..test_support import (
def test_service_records_tool_call_events() -> None:
from wf_authoring import node
from wf_core import END, RunStatus
from ..test_support import echo_tool, input_binding, output_binding
from ..test_support import input_binding, output_binding
from .conftest import raw_plan
service = WfMcpService(store=FileStore(local_temp_root() / "tool_event_store"))
@@ -75,7 +74,6 @@ def test_service_records_tool_call_events() -> None:
def test_service_rejects_text_binding_for_raw_mcp_content_contract() -> None:
from wf_authoring import node
from wf_core import END
from ..test_support import input_binding, output_binding
tests/wf_mcp/service/test_sources.py
+35 -7
View File
@@ -2,7 +2,7 @@ from __future__ import annotations
import asyncio
from wf_authoring import NodeSpec, node
from wf_authoring import NodeSpec
from wf_mcp.broker import WfMcpService
from wf_mcp.models import ConnectionConfig
from wf_mcp.storage import FileStore
@@ -47,8 +47,18 @@ def test_service_compiles_and_runs_raw_plan() -> None:
"id": "echo",
"type": "node",
"node": "demo.personal.echo_tool",
"input": [{"target": {"root": "local", "parts": ["text"]}, "path": {"root": "input", "parts": ["text"]}}],
"output": [{"source": {"root": "local", "parts": ["echoed"]}, "target": {"root": "state", "parts": ["echoed"]}}],
"input": [
{
"target": {"root": "local", "parts": ["text"]},
"path": {"root": "input", "parts": ["text"]},
}
],
"output": [
{
"source": {"root": "local", "parts": ["echoed"]},
"target": {"root": "state", "parts": ["echoed"]},
}
],
}
],
edges=[{"from": "echo", "outcome": "ok", "to": "__end__"}],
@@ -69,17 +79,35 @@ def test_service_preserves_raw_plan_root_output_bindings() -> None:
plan = raw_plan(
name="root_output_plan",
input_schema={"type": "object", "properties": {"text": {"type": "string"}}, "required": ["text"]},
input_schema={
"type": "object",
"properties": {"text": {"type": "string"}},
"required": ["text"],
},
state_schema={"fields": {"echoed": {"type": "string"}}},
output_schema={"type": "object", "properties": {"echoed": {"type": "string"}}, "required": ["echoed"]},
output_schema={
"type": "object",
"properties": {"echoed": {"type": "string"}},
"required": ["echoed"],
},
start="echo",
nodes=[
{
"id": "echo",
"type": "node",
"node": "demo.personal.echo_tool",
"input": [{"target": {"root": "local", "parts": ["text"]}, "path": {"root": "input", "parts": ["text"]}}],
"output": [{"source": {"root": "local", "parts": ["echoed"]}, "target": {"root": "state", "parts": ["echoed"]}}],
"input": [
{
"target": {"root": "local", "parts": ["text"]},
"path": {"root": "input", "parts": ["text"]},
}
],
"output": [
{
"source": {"root": "local", "parts": ["echoed"]},
"target": {"root": "state", "parts": ["echoed"]},
}
],
}
],
edges=[{"from": "echo", "outcome": "ok", "to": "__end__"}],
tests/wf_mcp/workflow_surface/conftest.py
+3 -14
View File
@@ -1,6 +1,5 @@
from __future__ import annotations
import asyncio
from typing import Any
from pydantic import BaseModel
@@ -9,26 +8,16 @@ from wf_artifacts import (
FileWorkflowArtifactStore,
RequiredCapability,
WorkflowArtifact,
WorkflowDeployment,
)
from wf_authoring import node, reducer
from wf_mcp.broker import WfMcpService
from wf_mcp.models import AuthRecord, ConnectionConfig, RawWorkflowPlan
from wf_mcp.models import AuthRecord, ConnectionConfig
from wf_mcp.sdk import ToolCallResult
from wf_mcp.storage import FileStore
from wf_mcp.workflow_surface import TraceRange, WorkflowSurfaceHandlers
from wf_mcp.workflow_surface.models import CreateMinimalDraftWorkspaceRequest
from wf_core.models.steps import InputPathBinding, OutputBinding
from wf_core.paths import GraphSourcePath, LocalPath, StatePath
from wf_mcp.workflow_surface import WorkflowSurfaceHandlers
from wf_mcp.capabilities import DiscoveredTool
from wf_platform import (
CapabilityBuckets,
CapabilitySource,
SourcePermissions,
SourceVisibility,
)
from ..test_support import echo_tool, input_binding, local_temp_root, output_binding
from ..test_support import input_binding, output_binding
class AmountInput(BaseModel):
tests/wf_mcp/workflow_surface/test_artifacts.py
+1 -1
View File
@@ -5,7 +5,7 @@ import asyncio
from wf_artifacts import FileWorkflowArtifactStore
from ..test_support import local_temp_root
from .conftest import artifact, echo_artifact, handlers
from .conftest import artifact, handlers
def test_workflow_surface_lists_artifact_catalog_entries() -> None:
tests/wf_mcp/workflow_surface/test_capabilities.py
+6 -16
View File
@@ -2,7 +2,7 @@ from __future__ import annotations
import asyncio
from wf_artifacts import FileWorkflowArtifactStore, WorkflowArtifact
from wf_artifacts import FileWorkflowArtifactStore
from wf_mcp.broker import WfMcpService
from wf_mcp.models import ConnectionConfig
from wf_mcp.storage import FileStore
@@ -37,13 +37,9 @@ def test_workflow_surface_lists_planner_visible_capabilities() -> None:
def test_workflow_surface_filters_stdlib_capabilities_by_source() -> None:
h = handlers(
FileWorkflowArtifactStore(local_temp_root() / "surface_filtered_caps")
)
h = handlers(FileWorkflowArtifactStore(local_temp_root() / "surface_filtered_caps"))
payload = asyncio.run(
h.list_capabilities(source_id="wf.std", query="truthy")
)
payload = asyncio.run(h.list_capabilities(source_id="wf.std", query="truthy"))
assert [capability["name"] for capability in payload["capabilities"]] == [
"wf.std.truthy"
@@ -98,9 +94,7 @@ def test_workflow_surface_lists_saved_wrapper_capabilities() -> None:
artifact_store.save_artifact(echo_artifact())
h = handlers(artifact_store)
payload = asyncio.run(
h.list_capabilities(source_id="workflow", query="echo")
)
payload = asyncio.run(h.list_capabilities(source_id="workflow", query="echo"))
names = [capability["name"] for capability in payload["capabilities"]]
assert names == ["workflow.echo_wrapper.v1"]
@@ -115,13 +109,9 @@ def test_workflow_surface_lists_saved_wrapper_capabilities() -> None:
def test_workflow_surface_inspects_one_capability() -> None:
h = handlers(
FileWorkflowArtifactStore(local_temp_root() / "surface_inspect_cap")
)
h = handlers(FileWorkflowArtifactStore(local_temp_root() / "surface_inspect_cap"))
payload = asyncio.run(
h.inspect_capability(qualified_name="wf.std.runtime_error")
)
payload = asyncio.run(h.inspect_capability(qualified_name="wf.std.runtime_error"))
assert payload["name"] == "wf.std.runtime_error"
assert payload["outcomes"] == ["ok"]
tests/wf_mcp/workflow_surface/test_drafts.py
+1 -4
View File
@@ -13,7 +13,6 @@ from wf_core.paths import GraphSourcePath, LocalPath, StatePath
from ..test_support import echo_tool, local_temp_root
from .conftest import (
echo_artifact,
echo_draft,
handlers,
mcp_echo_tool,
@@ -222,9 +221,7 @@ def test_workflow_surface_deletes_draft_workspace() -> None:
)
deleted = asyncio.run(h.delete_draft_workspace(workspace_id="echo_draft"))
deleted_again = asyncio.run(
h.delete_draft_workspace(workspace_id="echo_draft")
)
deleted_again = asyncio.run(h.delete_draft_workspace(workspace_id="echo_draft"))
listed = asyncio.run(h.list_draft_workspaces())
assert deleted["deleted"] is True
tests/wf_mcp/workflow_surface/test_runs.py
+1 -6
View File
@@ -3,7 +3,6 @@ from __future__ import annotations
import asyncio
from wf_artifacts import FileWorkflowArtifactStore, WorkflowDeployment
from wf_authoring import node, reducer
from wf_mcp.broker import WfMcpService
from wf_mcp.models import ConnectionConfig
from wf_mcp.storage import FileStore
@@ -15,18 +14,14 @@ from wf_platform import (
SourceVisibility,
)
from ..test_support import echo_tool, input_binding, local_temp_root, output_binding
from ..test_support import echo_tool, local_temp_root
from .conftest import (
AmountInput,
AmountOutput,
amount_tool,
changed_echo_tool,
custom_reducer_artifact,
echo_artifact,
echo_draft,
failing_artifact,
failing_tool,
handlers,
logical_echo_artifact,
multiply,
)