first commit
This commit is contained in:
@@ -0,0 +1,78 @@
|
||||
import subprocess
|
||||
import sys
|
||||
import unittest
|
||||
from unittest import mock
|
||||
|
||||
from tests.python.support.paths import add_repo_path, repo_path
|
||||
|
||||
add_repo_path("skills/dxf/scripts")
|
||||
|
||||
from dxf import cli as dxf
|
||||
|
||||
|
||||
class DxfCliTests(unittest.TestCase):
|
||||
def test_requires_explicit_target(self) -> None:
|
||||
with self.assertRaises(SystemExit) as cm:
|
||||
dxf.main([])
|
||||
self.assertEqual(2, cm.exception.code)
|
||||
|
||||
def test_passes_targets_in_order(self) -> None:
|
||||
with mock.patch.object(dxf, "generate_dxf_targets", return_value=0) as generate:
|
||||
self.assertEqual(0, dxf.main(["drawings/second.py", "drawings/first.py"]))
|
||||
|
||||
generate.assert_called_once_with(["drawings/second.py", "drawings/first.py"], output=None, verbose=False)
|
||||
|
||||
def test_passes_verbose_flag(self) -> None:
|
||||
with mock.patch.object(dxf, "generate_dxf_targets", return_value=0) as generate:
|
||||
self.assertEqual(0, dxf.main(["drawings/part.py", "--verbose"]))
|
||||
|
||||
generate.assert_called_once_with(["drawings/part.py"], output=None, verbose=True)
|
||||
|
||||
def test_passes_output_flag(self) -> None:
|
||||
with mock.patch.object(dxf, "generate_dxf_targets", return_value=0) as generate:
|
||||
self.assertEqual(0, dxf.main(["drawings/part.py", "-o", "DXF/part.dxf"]))
|
||||
|
||||
generate.assert_called_once_with(["drawings/part.py"], output="DXF/part.dxf", verbose=False)
|
||||
|
||||
def test_output_flag_rejects_multiple_targets(self) -> None:
|
||||
with self.assertRaises(SystemExit) as cm:
|
||||
dxf.main(["drawings/first.py", "drawings/second.py", "-o", "DXF/first.dxf"])
|
||||
self.assertEqual(2, cm.exception.code)
|
||||
|
||||
def test_scripts_dxf_directory_invokes_cli(self) -> None:
|
||||
skill_root = repo_path("skills/dxf")
|
||||
result = subprocess.run(
|
||||
[sys.executable, "scripts/dxf", "--help"],
|
||||
cwd=skill_root,
|
||||
check=False,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
text=True,
|
||||
)
|
||||
self.assertEqual("", result.stderr)
|
||||
self.assertEqual(0, result.returncode)
|
||||
self.assertIn("usage: dxf", result.stdout)
|
||||
self.assertIn("--output", result.stdout)
|
||||
|
||||
def test_cli_import_does_not_import_heavy_cad_modules(self) -> None:
|
||||
skill_root = repo_path("skills/dxf")
|
||||
code = (
|
||||
"import sys; sys.path.insert(0, 'scripts'); import dxf.cli; "
|
||||
"print('OCP.OCP' in sys.modules); "
|
||||
"print('cadpy.step_scene' in sys.modules)"
|
||||
)
|
||||
result = subprocess.run(
|
||||
[sys.executable, "-c", code],
|
||||
cwd=skill_root,
|
||||
check=False,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
text=True,
|
||||
)
|
||||
self.assertEqual("", result.stderr)
|
||||
self.assertEqual(0, result.returncode)
|
||||
self.assertEqual(["False", "False"], result.stdout.strip().splitlines())
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user