Skip to content

Commit 1c33c35

Browse files
committed
Fixes in test_responses.py
1 parent fa3b46c commit 1c33c35

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

tests/unit/utils/test_responses.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -379,6 +379,7 @@ async def test_get_mcp_tools_with_kubernetes_auth(
379379
name="k8s-server",
380380
url="http://localhost:3000",
381381
authorization_headers={"Authorization": "kubernetes"},
382+
provider_id="x",
382383
),
383384
]
384385
mock_config = mocker.Mock()
@@ -396,6 +397,7 @@ async def test_get_mcp_tools_with_mcp_headers(self, mocker: MockerFixture) -> No
396397
name="fs",
397398
url="http://localhost:3000",
398399
authorization_headers={"Authorization": "client", "X-Custom": "client"},
400+
provider_id="x",
399401
),
400402
]
401403
mock_config = mocker.Mock()
@@ -429,6 +431,7 @@ async def test_get_mcp_tools_client_auth_no_mcp_headers(
429431
name="client-auth-server",
430432
url="http://localhost:3000",
431433
authorization_headers={"X-Custom": "client"},
434+
provider_id="x",
432435
),
433436
]
434437
mock_config = mocker.Mock()
@@ -452,6 +455,7 @@ async def test_get_mcp_tools_client_auth_missing_server_in_headers(
452455
name="client-auth-server",
453456
url="http://localhost:3000",
454457
authorization_headers={"X-Custom": "client"},
458+
provider_id="x",
455459
),
456460
]
457461
mock_config = mocker.Mock()
@@ -478,6 +482,7 @@ async def test_get_mcp_tools_with_static_headers(
478482
name="server1",
479483
url="http://localhost:3000",
480484
authorization_headers={"Authorization": str(secret_file)},
485+
provider_id="x",
481486
),
482487
]
483488
mock_config = mocker.Mock()
@@ -505,6 +510,7 @@ async def test_get_mcp_tools_with_mixed_headers(
505510
"X-API-Key": str(secret_file),
506511
"X-Custom": "client",
507512
},
513+
provider_id="x",
508514
),
509515
]
510516
mock_config = mocker.Mock()
@@ -535,11 +541,13 @@ async def test_get_mcp_tools_skips_server_with_missing_auth(
535541
name="missing-k8s-auth",
536542
url="http://localhost:3001",
537543
authorization_headers={"Authorization": "kubernetes"},
544+
provider_id="x",
538545
),
539546
ModelContextProtocolServer(
540547
name="missing-client-auth",
541548
url="http://localhost:3002",
542549
authorization_headers={"X-Token": "client"},
550+
provider_id="x",
543551
),
544552
]
545553
mock_config = mocker.Mock()
@@ -559,6 +567,7 @@ async def test_get_mcp_tools_includes_server_without_auth(
559567
name="public-server",
560568
url="http://localhost:3000",
561569
authorization_headers={},
570+
provider_id="x",
562571
),
563572
]
564573
mock_config = mocker.Mock()
@@ -580,6 +589,7 @@ async def test_get_mcp_tools_oauth_no_headers_raises_401_with_www_authenticate(
580589
name="oauth-server",
581590
url="http://localhost:3000",
582591
authorization_headers={"Authorization": "oauth"},
592+
provider_id="x",
583593
),
584594
]
585595
mock_config = mocker.Mock()

0 commit comments

Comments
 (0)