Increase coverage

pyproject.toml

@@ -36,6 +36,7 @@ dependencies = [
     "shellingham >=1.3.0",
     "rich >=12.3.0",
     "annotated-doc >=0.0.2",
+    "colorama; platform_system == 'Windows'",
 ]
 readme = "README.md"
 

tests/test_completion/choice_case_insensitive_example.py

@@ -0,0 +1,19 @@
+from enum import Enum
+
+import typer
+
+app = typer.Typer()
+
+
+class User(str, Enum):
+    rick = "rick"
+    morty = "morty"
+
+
+@app.command()
+def main(name: User = typer.Option(User.rick, "--name", case_sensitive=False)):
+    print(name.value)
+
+
+if __name__ == "__main__":
+    app()

tests/test_completion/choice_example.py

@@ -0,0 +1,19 @@
+from enum import Enum
+
+import typer
+
+app = typer.Typer()
+
+
+class User(str, Enum):
+    rick = "rick"
+    morty = "morty"
+
+
+@app.command()
+def main(name: User = typer.Option(User.rick, "--name")):
+    print(name.value)
+
+
+if __name__ == "__main__":
+    app()

tests/test_completion/file_example.py

@@ -0,0 +1,12 @@
+import typer
+
+app = typer.Typer()
+
+
+@app.command()
+def main(config: typer.FileText = typer.Option(...)):
+    print(config.read())
+
+
+if __name__ == "__main__":
+    app()

tests/test_completion/test_completion_choice.py

@@ -0,0 +1,50 @@
+import os
+import subprocess
+import sys
+
+from . import choice_case_insensitive_example as mod_case_insensitive
+from . import choice_example as mod
+
+
+def test_script() -> None:
+    result = subprocess.run(
+        [sys.executable, "-m", "coverage", "run", mod.__file__, "--name", "rick"],
+        capture_output=True,
+        encoding="utf-8",
+    )
+    assert result.returncode == 0
+    assert "rick" in result.stdout
+
+
+def test_completion_choice_bash() -> None:
+    result = subprocess.run(
+        [sys.executable, "-m", "coverage", "run", mod.__file__, " "],
+        capture_output=True,
+        encoding="utf-8",
+        env={
+            **os.environ,
+            "_CHOICE_EXAMPLE.PY_COMPLETE": "complete_bash",
+            "COMP_WORDS": "choice_example.py --name mo",
+            "COMP_CWORD": "2",
+        },
+    )
+    assert result.returncode == 0
+    assert "morty" in result.stdout
+    assert "rick" not in result.stdout
+
+
+def test_completion_choice_bash_case_insensitive() -> None:
+    result = subprocess.run(
+        [sys.executable, "-m", "coverage", "run", mod_case_insensitive.__file__, " "],
+        capture_output=True,
+        encoding="utf-8",
+        env={
+            **os.environ,
+            "_CHOICE_CASE_INSENSITIVE_EXAMPLE.PY_COMPLETE": "complete_bash",
+            "COMP_WORDS": "choice_case_insensitive_example.py --name MO",
+            "COMP_CWORD": "2",
+        },
+    )
+    assert result.returncode == 0
+    assert "morty" in result.stdout
+    assert "rick" not in result.stdout

tests/test_completion/test_completion_file.py

@@ -0,0 +1,39 @@
+import os
+import subprocess
+import sys
+
+from . import file_example as mod
+
+
+def test_script() -> None:
+    result = subprocess.run(
+        [
+            sys.executable,
+            "-m",
+            "coverage",
+            "run",
+            mod.__file__,
+            "--config",
+            mod.__file__,
+        ],
+        capture_output=True,
+        encoding="utf-8",
+    )
+    assert result.returncode == 0
+    assert "def main(config: typer.FileText = typer.Option(...)):" in result.stdout
+
+
+def test_completion_file_bash() -> None:
+    result = subprocess.run(
+        [sys.executable, "-m", "coverage", "run", mod.__file__, " "],
+        capture_output=True,
+        encoding="utf-8",
+        env={
+            **os.environ,
+            "_FILE_EXAMPLE.PY_COMPLETE": "complete_bash",
+            "COMP_WORDS": "file_example.py --config file_ex",
+            "COMP_CWORD": "2",
+        },
+    )
+    assert result.returncode == 0
+    assert "file_ex" in result.stdout

tests/test_others.py

@@ -37,7 +37,7 @@ def test_too_many_parsers():
     def custom_parser(value: str) -> int:
         return int(value)  # pragma: no cover
 
-    class CustomClickParser(_click.ParamType):
+    class CustomClickParser(_click.types.ParamType):
         name = "custom_parser"
 
         def convert(
@@ -61,7 +61,7 @@ def test_valid_parser_permutations():
     def custom_parser(value: str) -> int:
         return int(value)  # pragma: no cover
 
-    class CustomClickParser(_click.ParamType):
+    class CustomClickParser(_click.types.ParamType):
         name = "custom_parser"
 
         def convert(

tests/test_tutorial/test_parameter_types/test_bool/test_tutorial001.py

@@ -4,6 +4,7 @@
 from types import ModuleType
 
 import pytest
+import typer
 from typer.testing import CliRunner
 
 runner = CliRunner()
@@ -22,6 +23,12 @@ def get_mod(request: pytest.FixtureRequest) -> ModuleType:
     return mod
 
 
+def test_type_repr(mod: ModuleType):
+    command = typer.main.get_command(mod.app)
+    force_param = next(param for param in command.params if param.name == "force")
+    assert repr(force_param.type) == "BOOL"
+
+
 def test_help(mod: ModuleType):
     result = runner.invoke(mod.app, ["--help"])
     assert result.exit_code == 0

tests/test_tutorial/test_parameter_types/test_datetime/test_tutorial001.py

@@ -1,6 +1,8 @@
 import subprocess
 import sys
+from datetime import datetime
 
+import typer
 from typer.testing import CliRunner
 
 from docs_src.parameter_types.datetime import tutorial001_py310 as mod
@@ -9,6 +11,12 @@
 app = mod.app
 
 
+def test_type_repr():
+    command = typer.main.get_command(app)
+    birth_param = next(param for param in command.params if param.name == "birth")
+    assert repr(birth_param.type) == "DateTime"
+
+
 def test_help():
     result = runner.invoke(app, ["--help"])
     assert result.exit_code == 0
@@ -22,6 +30,15 @@ def test_main():
     assert "Birth hour: 10" in result.output
 
 
+def test_main_datetime_object():
+    result = runner.invoke(
+        app, [], default_map={"birth": datetime(1956, 1, 31, 10, 0, 0)}
+    )
+    assert result.exit_code == 0
+    assert "Interesting day to be born: 1956-01-31 10:00:00" in result.output
+    assert "Birth hour: 10" in result.output
+
+
 def test_invalid():
     result = runner.invoke(app, ["july-19-1989"])
     assert result.exit_code != 0

tests/test_tutorial/test_parameter_types/test_index/test_tutorial001.py

@@ -1,6 +1,7 @@
 import subprocess
 import sys
 
+import typer
 from typer.testing import CliRunner
 
 from docs_src.parameter_types.index import tutorial001_py310 as mod
@@ -9,6 +10,16 @@
 app = mod.app
 
 
+def test_type_repr():
+    command = typer.main.get_command(app)
+    age_param = next(param for param in command.params if param.name == "age")
+    height_meters_param = next(
+        param for param in command.params if param.name == "height_meters"
+    )
+    assert repr(age_param.type) == "INT"
+    assert repr(height_meters_param.type) == "FLOAT"
+
+
 def test_help():
     result = runner.invoke(app, ["--help"])
     assert result.exit_code == 0

tests/test_tutorial/test_parameter_types/test_number/test_tutorial001.py

@@ -24,6 +24,19 @@ def get_mod(request: pytest.FixtureRequest) -> ModuleType:
     return mod
 
 
+def test_type_repr(mod: ModuleType):
+    command = typer.main.get_command(mod.app)
+
+    id_param = next(param for param in command.params if param.name == "id")
+    assert repr(id_param.type) == "<IntRange 0<=x<=1000>"
+
+    age_param = next(param for param in command.params if param.name == "age")
+    assert repr(age_param.type) == "<IntRange x>=18>"
+
+    score_param = next(param for param in command.params if param.name == "score")
+    assert repr(score_param.type) == "<FloatRange x<=100>"
+
+
 def test_help(mod: ModuleType):
     result = runner.invoke(mod.app, ["--help"])
     assert result.exit_code == 0

tests/test_tutorial/test_parameter_types/test_uuid/test_tutorial001.py

@@ -1,6 +1,8 @@
 import subprocess
 import sys
+import uuid
 
+import typer
 from typer.testing import CliRunner
 
 from docs_src.parameter_types.uuid import tutorial001_py310 as mod
@@ -9,13 +11,27 @@
 app = mod.app
 
 
+def test_type_repr():
+    command = typer.main.get_command(app)
+    user_id_param = next(param for param in command.params if param.name == "user_id")
+    assert repr(user_id_param.type) == "UUID"
+
+
 def test_main():
     result = runner.invoke(app, ["d48edaa6-871a-4082-a196-4daab372d4a1"])
     assert result.exit_code == 0
     assert "USER_ID is d48edaa6-871a-4082-a196-4daab372d4a1" in result.output
     assert "UUID version is: 4" in result.output
 
 
+def test_main_with_uuid_object():
+    user_id = uuid.UUID("d48edaa6-871a-4082-a196-4daab372d4a1")
+    result = runner.invoke(app, [], default_map={"user_id": user_id})
+    assert result.exit_code == 0
+    assert "USER_ID is d48edaa6-871a-4082-a196-4daab372d4a1" in result.output
+    assert "UUID version is: 4" in result.output
+
+
 def test_invalid_uuid():
     result = runner.invoke(app, ["7479706572-72756c6573"])
     assert result.exit_code != 0