Capability sources, add default source under wf.
This commit is contained in:
@@ -97,6 +97,20 @@ def test_create_broker_server_exposes_tools_resources_and_prompts() -> None:
|
||||
assert "wf.std" in source_ids
|
||||
|
||||
|
||||
def test_broker_admin_tools_are_backed_by_wf_admin_source() -> None:
|
||||
service = WfMcpService(store=FileStore(local_temp_root() / "broker_admin_source"))
|
||||
server = create_broker_server(service)
|
||||
|
||||
tools = asyncio.run(server.list_tools())
|
||||
tool_names = {tool.name for tool in tools}
|
||||
|
||||
assert "list_spec_sources" in tool_names
|
||||
assert "get_planner_catalog" in tool_names
|
||||
assert "wf.admin.list_sources" in service.capability_sources[
|
||||
"wf.admin"
|
||||
].capabilities.tools
|
||||
|
||||
|
||||
def test_build_service_from_config_registers_connections() -> None:
|
||||
config = BrokerConfig(
|
||||
store_root=local_temp_root() / "broker_config_store",
|
||||
|
||||
Reference in New Issue
Block a user