|
7 | 7 | from fastapi import HTTPException, Request, status |
8 | 8 | from llama_stack_client import APIConnectionError |
9 | 9 | from llama_stack_client.types import VersionInfo |
| 10 | +from authentication.interface import AuthTuple |
10 | 11 |
|
11 | 12 | from configuration import AppConfig |
12 | 13 | from app.endpoints.info import info_endpoint_handler |
@@ -40,7 +41,7 @@ async def test_info_endpoint_returns_service_information( |
40 | 41 | test_config: AppConfig, |
41 | 42 | mock_llama_stack_client: AsyncMockType, |
42 | 43 | test_request: Request, |
43 | | - test_auth: tuple[str, str, bool, str], |
| 44 | + test_auth: AuthTuple, |
44 | 45 | ) -> None: |
45 | 46 | """Test that info endpoint returns correct service information. |
46 | 47 |
|
@@ -76,7 +77,7 @@ async def test_info_endpoint_handles_connection_error( |
76 | 77 | test_config: AppConfig, |
77 | 78 | mock_llama_stack_client: AsyncMockType, |
78 | 79 | test_request: Request, |
79 | | - test_auth: tuple[str, str, bool, str], |
| 80 | + test_auth: AuthTuple, |
80 | 81 | mocker: MockerFixture, |
81 | 82 | ) -> None: |
82 | 83 | """Test that info endpoint properly handles Llama Stack connection errors. |
@@ -116,7 +117,7 @@ async def test_info_endpoint_uses_configuration_values( |
116 | 117 | test_config: AppConfig, |
117 | 118 | mock_llama_stack_client: AsyncMockType, |
118 | 119 | test_request: Request, |
119 | | - test_auth: tuple[str, str, bool, str], |
| 120 | + test_auth: AuthTuple, |
120 | 121 | ) -> None: |
121 | 122 | """Test that info endpoint correctly uses configuration values. |
122 | 123 |
|
|
0 commit comments