@@ -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