From a98b9a7988a77b79dca4fd9582a26eceef86a77f Mon Sep 17 00:00:00 2001
From: Justin Buchanan <justbuchanan@gmail.com>
Date: Wed, 22 Nov 2023 19:51:06 -0800
Subject: [PATCH] add test for --expression argument

---
 tests/test_cli.py                     | 50 +++++++++++++++++++++++++++
 tests/testdata/no_toplevel_objects.py |  7 ++++
 2 files changed, 57 insertions(+)
 create mode 100644 tests/testdata/no_toplevel_objects.py

diff --git a/tests/test_cli.py b/tests/test_cli.py
index 75c1a6d..b8ce14f 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -407,3 +407,53 @@ def test_exit_codes():
 
     # Make sure that we got exit code 100 for a failed model build
     assert exitcode == 100
+
+
+def test_expression_argument():
+    """
+    Tests the CLI with the the expression argument.
+    """
+    test_file = helpers.get_test_file_location("no_toplevel_objects.py")
+
+    # Get a temporary output file location
+    temp_dir = tempfile.gettempdir()
+    temp_file = os.path.join(temp_dir, "temp_test_10.step")
+
+    command = [
+        "python",
+        "src/cq_cli/main.py",
+        "--codec",
+        "step",
+        "--infile",
+        test_file,
+        "--outfile",
+        temp_file,
+        "--expression",
+        "cube()",
+    ]
+    out, err, exitcode = helpers.cli_call(command)
+
+    # Read the STEP output back from the outfile
+    with open(temp_file, "r") as file:
+        step_str = file.read()
+
+    assert step_str.startswith("ISO-10303-21;")
+
+    # Run cq-cli on the same model file, but don't specify an --expression. This
+    # should fail because the file contains no top-level show_object() calls.
+    command = [
+        "python",
+        "src/cq_cli/main.py",
+        "--codec",
+        "step",
+        "--infile",
+        test_file,
+        "--outfile",
+        temp_file,
+    ]
+    out, err, exitcode = helpers.cli_call(command)
+
+    print("err: %s" % err.decode())
+
+    # cq-cli invocation should fail
+    assert exitcode == 200
diff --git a/tests/testdata/no_toplevel_objects.py b/tests/testdata/no_toplevel_objects.py
new file mode 100644
index 0000000..b45f5c8
--- /dev/null
+++ b/tests/testdata/no_toplevel_objects.py
@@ -0,0 +1,7 @@
+import cadquery as cq
+
+# This test file contains a method for creating a shape, but does not contain
+# any top-level `show_object()` calls.
+
+def cube():
+    return cq.Workplane().box(1, 1, 1)