list tests

Author: nabinchha

src/data_designer/cli/commands/list.py

@@ -17,6 +17,7 @@ def list_command(
     config_dir: str | None = typer.Option(None, "--config-dir", help="Custom configuration directory"),
 ) -> None:
     """List current Data Designer configurations.
+
     Args:
         config_dir: Optional custom configuration directory
 

tests/cli/commands/test_list.py

@@ -0,0 +1,40 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+from unittest.mock import patch
+
+from rich.table import Table
+
+from data_designer.cli.commands.list import display_models, display_providers, list_command
+from data_designer.cli.constants import DEFAULT_CONFIG_DIR
+from data_designer.cli.repositories.model_repository import ModelRepository
+from data_designer.cli.repositories.provider_repository import ProviderRepository
+
+
+@patch("data_designer.cli.commands.list.display_providers")
+@patch("data_designer.cli.commands.list.display_models")
+def test_list_command(mock_display_models, mock_display_providers):
+    """Test list command."""
+    list_command(config_dir=None)
+    mock_display_providers.assert_called_once()
+    mock_display_providers.call_args[0][0].config_dir == DEFAULT_CONFIG_DIR
+    mock_display_models.assert_called_once()
+    mock_display_models.call_args[0][0].config_dir == DEFAULT_CONFIG_DIR
+
+
+@patch("data_designer.cli.commands.list.console.print")
+def test_display_providers(mock_console_print):
+    """Test display providers."""
+    display_providers(ProviderRepository(DEFAULT_CONFIG_DIR))
+    mock_console_print.call_count > 1
+    assert isinstance(mock_console_print.call_args_list[0][0][0], Table)
+    mock_console_print.call_args_list[0][0][0].title == "Model Providers"
+
+
+@patch("data_designer.cli.commands.list.console.print")
+def test_display_models(mock_console_print):
+    """Test display models."""
+    display_models(ModelRepository(DEFAULT_CONFIG_DIR))
+    mock_console_print.call_count > 1
+    assert isinstance(mock_console_print.call_args_list[0][0][0], Table)
+    mock_console_print.call_args_list[0][0][0].title == "Model Configurations"

tests/cli/conftest.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
 from pathlib import Path
 
 import pytest

tests/cli/repositories/test_model_repository.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
 from pathlib import Path
 
 from data_designer.cli.constants import MODEL_CONFIGS_FILE_NAME

tests/cli/repositories/test_provider_repository.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
 from pathlib import Path
 
 from data_designer.cli.constants import MODEL_PROVIDERS_FILE_NAME

tests/cli/services/test_provider_service.py

@@ -1,3 +1,6 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
 from data_designer.cli.services.provider_service import ProviderService
 from data_designer.config.models import ModelProvider