Add a test for CLI, but not fully done so disabled

llama_stack/cli/tests/test_stack_build.py

@@ -0,0 +1,105 @@
+from argparse import Namespace
+from unittest.mock import MagicMock, patch
+
+import pytest
+from llama_stack.distribution.datatypes import BuildConfig
+from llama_stack.cli.stack.build import StackBuild
+
+
+# temporary while we make the tests work
+pytest.skip(allow_module_level=True)
+
+
+@pytest.fixture
+def stack_build():
+    parser = MagicMock()
+    subparsers = MagicMock()
+    return StackBuild(subparsers)
+
+
+def test_stack_build_initialization(stack_build):
+    assert stack_build.parser is not None
+    assert stack_build.parser.set_defaults.called_once_with(
+        func=stack_build._run_stack_build_command
+    )
+
+
+@patch("llama_stack.distribution.build.build_image")
+def test_run_stack_build_command_with_config(
+    mock_build_image, mock_build_config, stack_build
+):
+    args = Namespace(
+        config="test_config.yaml",
+        template=None,
+        list_templates=False,
+        name=None,
+        image_type="conda",
+    )
+
+    with patch("builtins.open", MagicMock()):
+        with patch("yaml.safe_load") as mock_yaml_load:
+            mock_yaml_load.return_value = {"name": "test_build", "image_type": "conda"}
+            mock_build_config.return_value = MagicMock()
+
+            stack_build._run_stack_build_command(args)
+
+            mock_build_config.assert_called_once()
+            mock_build_image.assert_called_once()
+
+
+@patch("llama_stack.cli.table.print_table")
+def test_run_stack_build_command_list_templates(mock_print_table, stack_build):
+    args = Namespace(list_templates=True)
+
+    stack_build._run_stack_build_command(args)
+
+    mock_print_table.assert_called_once()
+
+
+@patch("prompt_toolkit.prompt")
+@patch("llama_stack.distribution.datatypes.BuildConfig")
+@patch("llama_stack.distribution.build.build_image")
+def test_run_stack_build_command_interactive(
+    mock_build_image, mock_build_config, mock_prompt, stack_build
+):
+    args = Namespace(
+        config=None, template=None, list_templates=False, name=None, image_type=None
+    )
+
+    mock_prompt.side_effect = [
+        "test_name",
+        "conda",
+        "meta-reference",
+        "test description",
+    ]
+    mock_build_config.return_value = MagicMock()
+
+    stack_build._run_stack_build_command(args)
+
+    assert mock_prompt.call_count == 4
+    mock_build_config.assert_called_once()
+    mock_build_image.assert_called_once()
+
+
+@patch("llama_stack.distribution.datatypes.BuildConfig")
+@patch("llama_stack.distribution.build.build_image")
+def test_run_stack_build_command_with_template(
+    mock_build_image, mock_build_config, stack_build
+):
+    args = Namespace(
+        config=None,
+        template="test_template",
+        list_templates=False,
+        name="test_name",
+        image_type="docker",
+    )
+
+    with patch("builtins.open", MagicMock()):
+        with patch("yaml.safe_load") as mock_yaml_load:
+            mock_yaml_load.return_value = {"name": "test_build", "image_type": "conda"}
+            mock_build_config.return_value = MagicMock()
+
+            stack_build._run_stack_build_command(args)
+
+            mock_build_config.assert_called_once()
+            mock_build_image.assert_called_once()