Update test and cli args

Author: mxr

openapi_generator_cli/__init__.py

@@ -56,13 +56,11 @@ def run(args: list[str] | None = None) -> subprocess.CompletedProcess[bytes]:
     return subprocess.run(arguments, check=False)  # noqa: S603
 
 
-def cli() -> None:
+def cli(argv: list[str] | None = None) -> None:
     """Run the OpenAPI Generator CLI with the arguments provided on the command line."""
-    args = []
-    if len(sys.argv) > 1:
-        args = sys.argv[1:]
+    args = argv[1:] if argv is not None and len(argv) > 1 else []
     sys.exit(run(args).returncode)
 
 
 if __name__ == "__main__":
-    cli()
+    cli(sys.argv)

tests/test_cli.py

@@ -2,15 +2,11 @@
 
 import os
 import re
-import subprocess
-from unittest.mock import MagicMock, patch
 
 import pytest
 
 from openapi_generator_cli import cli, run
 
-RETURN_CODE = 23
-
 
 def test_cli_version(capfd: pytest.CaptureFixture[str]) -> None:
     result = run(args=["version"])
@@ -49,14 +45,14 @@ def test_invalid_arg(capfd: pytest.CaptureFixture[str]) -> None:
     )
 
 
-@patch("openapi_generator_cli.run", autospec=True)
-@patch("sys.argv", ["openapi-generator-cli", "version"])
-def test_cli_exits_with_returncode(run_mock: MagicMock) -> None:
-    run_mock.return_value = subprocess.CompletedProcess(args=[], returncode=RETURN_CODE)
-
+def test_cli_invalid_arg(capfd: pytest.CaptureFixture[str]) -> None:
     with pytest.raises(SystemExit) as exc_info:
-        cli()
-
-    assert exc_info.value.code == RETURN_CODE
+        cli(["openapi-generator-cli", "--invalid-arg-404"])
+    assert exc_info.value.code == 1
 
-    run_mock.assert_called_once_with(["version"])
+    captured = capfd.readouterr()
+    assert not captured.out
+    assert (
+        "Found unexpected parameters: [--invalid-arg-404]"
+        in captured.err.split(os.linesep)[0]
+    )