LCORE-739: added return type to unit tests

Author: tisnik

tests/unit/metrics/test_utis.py

@@ -3,7 +3,7 @@
 from metrics.utils import setup_model_metrics, update_llm_token_count_from_turn
 
 
-async def test_setup_model_metrics(mocker):
+async def test_setup_model_metrics(mocker) -> None:
     """Test the setup_model_metrics function."""
 
     # Mock the LlamaStackAsLibraryClient
@@ -76,7 +76,7 @@ async def test_setup_model_metrics(mocker):
     )
 
 
-def test_update_llm_token_count_from_turn(mocker):
+def test_update_llm_token_count_from_turn(mocker) -> None:
     """Test the update_llm_token_count_from_turn function."""
     mocker.patch("metrics.utils.Tokenizer.get_instance")
     mock_formatter_class = mocker.patch("metrics.utils.ChatFormat")

tests/unit/utils/test_common.py

@@ -18,7 +18,7 @@
 
 
 @pytest.mark.asyncio
-async def test_register_mcp_servers_empty_list(mocker):
+async def test_register_mcp_servers_empty_list(mocker) -> None:
     """Test register_mcp_servers with empty MCP servers list."""
     # Mock the logger
     mock_logger = Mock(spec=Logger)
@@ -49,7 +49,7 @@ async def test_register_mcp_servers_empty_list(mocker):
 
 
 @pytest.mark.asyncio
-async def test_register_mcp_servers_single_server_not_registered(mocker):
+async def test_register_mcp_servers_single_server_not_registered(mocker) -> None:
     """Test register_mcp_servers with single MCP server that is not yet registered."""
     # Mock the logger
     mock_logger = Mock(spec=Logger)
@@ -94,7 +94,7 @@ async def test_register_mcp_servers_single_server_not_registered(mocker):
 
 
 @pytest.mark.asyncio
-async def test_register_mcp_servers_single_server_already_registered(mocker):
+async def test_register_mcp_servers_single_server_already_registered(mocker) -> None:
     """Test register_mcp_servers with single MCP server that is already registered."""
     # Mock the logger
     mock_logger = Mock(spec=Logger)
@@ -132,7 +132,7 @@ async def test_register_mcp_servers_single_server_already_registered(mocker):
 
 
 @pytest.mark.asyncio
-async def test_register_mcp_servers_multiple_servers_mixed_registration(mocker):
+async def test_register_mcp_servers_multiple_servers_mixed_registration(mocker) -> None:
     """Test register_mcp_servers with multiple MCP servers - some registered, some not."""
     # Mock the logger
     mock_logger = Mock(spec=Logger)
@@ -194,7 +194,7 @@ async def test_register_mcp_servers_multiple_servers_mixed_registration(mocker):
 
 
 @pytest.mark.asyncio
-async def test_register_mcp_servers_with_custom_provider(mocker):
+async def test_register_mcp_servers_with_custom_provider(mocker) -> None:
     """Test register_mcp_servers with MCP server using custom provider."""
     # Mock the logger
     mock_logger = Mock(spec=Logger)
@@ -235,7 +235,7 @@ async def test_register_mcp_servers_with_custom_provider(mocker):
 
 
 @pytest.mark.asyncio
-async def test_register_mcp_servers_async_with_library_client(mocker):
+async def test_register_mcp_servers_async_with_library_client(mocker) -> None:
     """
     Test that `register_mcp_servers_async` correctly registers MCP
     servers when using the library client configuration.