test orgs 2

tests/rewrite/test_this_shit_works.py

@@ -557,9 +557,9 @@ def test():
     )
     assert d.status == RunStatus.COMPLETED, "oops"
     state = State.model_validate(d.state)
-    assert any(
+    assert any(i["name"] in context["pool"]["n_240"] for i in state.storage), (
-        i["name"] in context["pool"]["n_240"] for i in state.storage
+        "pity logic failed"
-    ), "pity logic failed"
+    )
     pprint(state.storage)
 
 

tests/wf_mcp/__init__.py

@@ -0,0 +1 @@
+"""MCP integration and proxy tests."""

tests/wf_mcp/test_broker_server.py

@@ -13,7 +13,7 @@ from wf_mcp import (
     load_broker_config,
 )
 
-from tests.test_wf_mcp_support import (
+from .test_support import (
     FailingDiscoveryAdapter,
     FakeAdapter,
     local_temp_root,

tests/wf_mcp/test_cli.py

@@ -9,7 +9,7 @@ from pydantic import ValidationError
 from wf_mcp.cli import build_parser, main
 from wf_mcp.broker_server import load_broker_config
 
-from tests.test_wf_mcp_support import local_temp_root
+from .test_support import local_temp_root
 
 
 def _write_config(path: Path) -> None:

tests/wf_mcp/test_sdk_adapter.py

@@ -6,7 +6,7 @@ import pytest
 
 from wf_mcp import ConnectionConfig, FileStore, McpSdkAdapter, WfMcpService
 
-from tests.test_wf_mcp_support import (
+from .test_support import (
     everything_server_connection,
     fixture_server_path,
     local_temp_root,

tests/wf_mcp/test_service.py

@@ -13,7 +13,7 @@ from wf_mcp import (
 )
 from wf_mcp.error_info import error_payload
 
-from tests.test_wf_mcp_support import (
+from .test_support import (
     FailingDiscoveryAdapter,
     FakeAdapter,
     echo_tool,

tests/wf_mcp/test_store.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from wf_mcp import AuthRecord, FileStore
 
-from tests.test_wf_mcp_support import local_temp_root
+from .test_support import local_temp_root
 
 
 def test_file_store_round_trips_auth() -> None:

tests/wf_mcp/test_support.py

@@ -57,7 +57,7 @@ def local_temp_root() -> Path:
 
 
 def fixture_server_path() -> str:
-    return str((Path(__file__).resolve().parent / "fixtures" / "mcp_echo_server.py"))
+    return str(Path(__file__).resolve().parents[1] / "fixtures" / "mcp_echo_server.py")
 
 
 def everything_server_connection() -> ConnectionConfig | None:

tests/wf_mcp/test_transparent_proxy.py

@@ -16,7 +16,7 @@ from wf_mcp import (
 )
 from wf_mcp.broker_server import load_broker_config
 
-from tests.test_wf_mcp_support import fixture_server_path, local_temp_root
+from .test_support import fixture_server_path, local_temp_root
 
 
 def _structured(result: Any) -> dict[str, Any]: