Skip to content

Commit 4b23ea9

Browse files
committed
fix the tests
1 parent 1f404af commit 4b23ea9

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/server/mcp_server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,8 +77,8 @@ def create_mcp_server() -> FastMCP:
7777
description='List the Amazon Advertising profiles for a given remote identity ID. Returns a list of profiles.',
7878
)(service_tools.get_amazon_advertising_profiles)
7979
mcp.tool(
80-
name='get_table_schemas',
81-
description='Get the schemas for a given table name from the rules API. Returns the schema if found.',
80+
name='get_table_schema',
81+
description='Get the schema for a given table name from the rules API. Returns the schema if found.',
8282
)(service_tools.get_table_schema)
8383
mcp.tool(
8484
name='get_suggested_table_names',

tests/server/test_mcp_server.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def fake_create_auth_middleware(config, *, jwt_middleware, auth_manager):
7272
"get_suggested_table_names",
7373
"get_healthchecks",
7474
"get_jobs",
75-
"create_oneoff_jobs",
75+
"create_job",
7676
"get_subscriptions",
7777
"get_storage_subscriptions",
7878
"get_product_stage_ids",
@@ -114,7 +114,7 @@ def fake_create_auth_middleware(config, *, jwt_middleware, auth_manager):
114114
"get_suggested_table_names",
115115
"get_healthchecks",
116116
"get_jobs",
117-
"create_oneoff_jobs",
117+
"create_job",
118118
"get_subscriptions",
119119
"get_storage_subscriptions",
120120
"get_product_stage_ids",

0 commit comments

Comments
 (0)