diff --git a/tests/integration/test_error_handling_edge_cases.py b/tests/integration/test_error_handling_edge_cases.py index eaec9fc0..a0612d71 100644 --- a/tests/integration/test_error_handling_edge_cases.py +++ b/tests/integration/test_error_handling_edge_cases.py @@ -48,7 +48,7 @@ class TestNetworkFailureScenarios: ) -> None: """Test proxy behavior when upstream LLM service is down""" # Mock at the routstr level to simulate upstream being down - with patch("routstr.upstream.httpx.AsyncClient") as mock_client_class: + with patch("httpx.AsyncClient") as mock_client_class: # Create a mock client instance mock_client = AsyncMock() mock_client_class.return_value = mock_client diff --git a/tests/integration/test_general_info_endpoints.py b/tests/integration/test_general_info_endpoints.py index cfde36e1..56d935b6 100644 --- a/tests/integration/test_general_info_endpoints.py +++ b/tests/integration/test_general_info_endpoints.py @@ -28,7 +28,7 @@ async def test_root_endpoint_structure_and_performance( responses = [] for i in range(10): start = validator.start_timing("root_endpoint") - response = await integration_client.get("/") + response = await integration_client.get("/v1/info") duration = validator.end_timing("root_endpoint", start) responses.append(response) @@ -100,7 +100,7 @@ async def test_root_endpoint_environment_variables( ) -> None: """Test that root endpoint reflects environment variable configuration""" - response = await integration_client.get("/") + response = await integration_client.get("/v1/info") assert response.status_code == 200 data = response.json() @@ -351,7 +351,7 @@ async def test_all_info_endpoints_no_database_changes( initial_state = await db_snapshot.capture() # Make requests to all info endpoints - endpoints = ["/", "/v1/models", "/admin/"] + endpoints = ["/v1/info", "/v1/models", "/admin/"] for endpoint in endpoints: response = await integration_client.get(endpoint) @@ -387,7 +387,9 @@ async def test_concurrent_info_endpoint_requests( requests = [] for endpoint in ["/", "/v1/models", "/admin/"]: for _ in range(5): # 5 requests per endpoint - requests.append({"method": "GET", "url": endpoint}) + # Use /v1/info instead of / for JSON API + url = "/v1/info" if endpoint == "/" else endpoint + requests.append({"method": "GET", "url": url}) # Execute concurrently tester = ConcurrencyTester() @@ -417,7 +419,7 @@ async def test_info_endpoints_response_consistency( # Test root endpoint consistency responses = [] for _ in range(5): - response = await integration_client.get("/") + response = await integration_client.get("/v1/info") assert response.status_code == 200 responses.append(response.json())