Add basic test for cli.build

Author: pont-us

test/test_cli.py

@@ -1,4 +1,4 @@
-from unittest.mock import patch
+from unittest.mock import patch, ANY
 import pytest
 from click.testing import CliRunner
 
@@ -12,8 +12,18 @@
 @pytest.mark.parametrize("batch_arg", [[], ["--batch"]])
 @pytest.mark.parametrize("server_arg", [[], ["--server"]])
 @pytest.mark.parametrize("from_saved_arg", [[], ["--from-saved"]])
-def test_make_script(run_mock, convert_mock, init_mock, tmp_path, verbose_arg, batch_arg, server_arg, from_saved_arg):
+def test_make_script(
+    run_mock,
+    convert_mock,
+    init_mock,
+    tmp_path,
+    verbose_arg,
+    batch_arg,
+    server_arg,
+    from_saved_arg,
+):
     from xcengine.cli import logging
+
     logging.getLogger().setLevel(logging.WARN)
     nb_path = tmp_path / "foo.ipynb"
     nb_path.touch()
@@ -22,23 +32,41 @@ def test_make_script(run_mock, convert_mock, init_mock, tmp_path, verbose_arg, b
     runner = CliRunner()
     result = runner.invoke(
         cli,
-        verbose_arg +
-        [
-            "make-script"] +
-        batch_arg + server_arg + from_saved_arg +
-[
+        verbose_arg
+        + ["make-script"]
+        + batch_arg
+        + server_arg
+        + from_saved_arg
+        + [
             str(nb_path),
             str(output_dir),
         ],
     )
-    convert_mock.assert_called()
-    init_mock.assert_called()
+    convert_mock.assert_called_once_with(
+        output_dir=output_dir, clear_output=False
+    )
+    init_mock.assert_called_once_with(nb_path)
     if batch_arg or server_arg:
-        run_mock.assert_called_with(
-            [
-                "python3",
-                output_dir / "execute.py" ] +
-            batch_arg + server_arg + from_saved_arg
+        run_mock.assert_called_once_with(
+            ["python3", output_dir / "execute.py"]
+            + batch_arg
+            + server_arg
+            + from_saved_arg
         )
-    assert logging.getLogger().getEffectiveLevel() == (logging.DEBUG if "--verbose" in verbose_arg else logging.WARNING)
+    assert logging.getLogger().getEffectiveLevel() == (
+        logging.DEBUG if "--verbose" in verbose_arg else logging.WARNING
+    )
     assert result.exit_code == 0
+
+
+@patch("xcengine.core.ImageBuilder.__init__")
+@patch("xcengine.core.ImageBuilder.build")
+def test_image_build(build_mock, init_mock, tmp_path):
+    nb_path = tmp_path / "foo.ipynb"
+    nb_path.touch()
+    runner = CliRunner()
+    tag = "foo"
+    result = runner.invoke(cli, ["image", "build", "--tag", tag, str(nb_path)])
+    init_mock.assert_called_once_with(
+        notebook=nb_path, environment=None, tag=tag, build_dir=ANY
+    )