Add tests for CLI

#3

Author: dbanty

openapi_python_client/cli.py

@@ -1,3 +1,4 @@
+from importlib.metadata import version
 from typing import Optional
 
 import click
@@ -6,6 +7,7 @@
 
 
 @click.group()
+@click.version_option(version(__package__), prog_name="OpenAPI Python Client")
 def cli() -> None:
     """ Entrypoint into CLI """
     pass
@@ -17,6 +19,9 @@ def cli() -> None:
 def generate(url: Optional[str], path: Optional[str]) -> None:
     """ Generate a new OpenAPI Client library """
     if not url and not path:
-        print("You must either provide --url or --path")
+        click.secho("You must either provide --url or --path", fg="red")
+        exit(1)
+    elif url and path:
+        click.secho("Provide either --url or --path, not both", fg="red")
         exit(1)
     main(url=url, path=path)

tests/test_cli.py

@@ -0,0 +1,56 @@
+from click.testing import CliRunner
+
+
+def test_generate_no_params():
+    from openapi_python_client.cli import generate
+
+    runner = CliRunner()
+    result = runner.invoke(generate)
+
+    assert result.exit_code == 1
+    assert result.output == "You must either provide --url or --path\n"
+
+
+def test_generate_url_and_path():
+    from openapi_python_client.cli import generate
+
+    runner = CliRunner()
+    result = runner.invoke(generate, ["--url=blah", "--path=blahblah"])
+
+    assert result.exit_code == 1
+    assert result.output == "Provide either --url or --path, not both\n"
+
+
+def test_generate_url(mocker):
+    main = mocker.patch("openapi_python_client.main")
+    url = mocker.MagicMock()
+    from openapi_python_client.cli import generate
+
+    runner = CliRunner()
+    result = runner.invoke(generate, [f"--url={url}"])
+
+    assert result.exit_code == 0
+    main.assert_called_once_with(url=str(url), path=None)
+
+
+def test_generate_path(mocker):
+    main = mocker.patch("openapi_python_client.main")
+    path = mocker.MagicMock()
+    from openapi_python_client.cli import generate
+
+    runner = CliRunner()
+    result = runner.invoke(generate, [f"--path={path}"])
+
+    assert result.exit_code == 0
+    main.assert_called_once_with(path=str(path), url=None)
+
+
+def test_version():
+    from importlib.metadata import version
+    from openapi_python_client.cli import cli
+
+    runner = CliRunner()
+    result = runner.invoke(cli, ["--version"])
+
+    assert result.exit_code == 0
+    assert result.output == f"OpenAPI Python Client, version {version('openapi-python-client')}\n"