Skip to content

Commit 8f2a6ec

Browse files
committed
Update tests for MCP Server API
Assisted-by: Claude Opus 4.6 Generated-by: Cursor Signed-off-by: Maysun J Faisal <maysunaneek@gmail.com>
1 parent 620ce41 commit 8f2a6ec

2 files changed

Lines changed: 7 additions & 3 deletions

File tree

tests/unit/app/test_routers.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
metrics,
2626
tools,
2727
mcp_auth,
28+
mcp_servers,
2829
rlsapi_v1,
2930
a2a,
3031
query,
@@ -109,12 +110,13 @@ def test_include_routers() -> None:
109110
include_routers(app)
110111

111112
# are all routers added?
112-
assert len(app.routers) == 21
113+
assert len(app.routers) == 22
113114
assert root.router in app.get_routers()
114115
assert info.router in app.get_routers()
115116
assert models.router in app.get_routers()
116117
assert tools.router in app.get_routers()
117118
assert mcp_auth.router in app.get_routers()
119+
assert mcp_servers.router in app.get_routers()
118120
assert shields.router in app.get_routers()
119121
assert providers.router in app.get_routers()
120122
assert query.router in app.get_routers()
@@ -147,12 +149,13 @@ def test_check_prefixes() -> None:
147149
include_routers(app)
148150

149151
# are all routers added?
150-
assert len(app.routers) == 21
152+
assert len(app.routers) == 22
151153
assert app.get_router_prefix(root.router) == ""
152154
assert app.get_router_prefix(info.router) == "/v1"
153155
assert app.get_router_prefix(models.router) == "/v1"
154156
assert app.get_router_prefix(tools.router) == "/v1"
155157
assert app.get_router_prefix(mcp_auth.router) == "/v1"
158+
assert app.get_router_prefix(mcp_servers.router) == "/v1"
156159
assert app.get_router_prefix(shields.router) == "/v1"
157160
assert app.get_router_prefix(providers.router) == "/v1"
158161
assert app.get_router_prefix(rags.router) == "/v1"

tests/unit/models/responses/test_error_responses.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -472,14 +472,15 @@ def test_openapi_response(self) -> None:
472472

473473
# Verify example count matches schema examples count
474474
assert len(examples) == expected_count
475-
assert expected_count == 5
475+
assert expected_count == 6
476476

477477
# Verify all labeled examples are present
478478
assert "conversation" in examples
479479
assert "provider" in examples
480480
assert "model" in examples
481481
assert "rag" in examples
482482
assert "streaming request" in examples
483+
assert "mcp server" in examples
483484

484485
# Verify example structure for one example
485486
conversation_example = examples["conversation"]

0 commit comments

Comments
 (0)