test: lock down mcp model ownership shims

This commit is contained in:
lda
2026-06-06 15:07:02 +07:00 Verified
parent a28ca5434a
commit fa37f57c00
2 changed files with 19 additions and 0 deletions
+15
View File
@@ -36,3 +36,18 @@ def test_concern_package_imports_resolve() -> None:
assert WfMcpService.__name__ == "WfMcpService"
assert ProxyRuntime.__name__ == "ProxyRuntime"
assert callable(load_broker_config)
def test_models_module_reexports_canonical_model_owners() -> None:
from wf_mcp.auth import AuthRecord
from wf_mcp.broker.models import BrokerConfig, ConnectionConfig
from wf_mcp.catalog.models import CatalogSnapshot
from wf_mcp.models import AuthRecord as CompatAuthRecord
from wf_mcp.models import BrokerConfig as CompatBrokerConfig
from wf_mcp.models import CatalogSnapshot as CompatCatalogSnapshot
from wf_mcp.models import ConnectionConfig as CompatConnectionConfig
assert CompatAuthRecord is AuthRecord
assert CompatBrokerConfig is BrokerConfig
assert CompatCatalogSnapshot is CatalogSnapshot
assert CompatConnectionConfig is ConnectionConfig