✅ Add extra test for coverage

tiangolo · tiangolo · commit 5c0b4a63c81d · 2025-05-26T16:07:49.000+02:00
diff --git a/tests/test_types.py b/tests/test_types.py
@@ -0,0 +1,34 @@
+from enum import Enum
+
+import typer
+from typer.testing import CliRunner
+
+app = typer.Typer(context_settings={"token_normalize_func": str.lower})
+
+
+class User(str, Enum):
+    rick = "Rick"
+    morty = "Morty"
+
+
+@app.command()
+def hello(name: User = User.rick) -> None:
+    print(f"Hello {name.value}!")
+
+
+runner = CliRunner()
+
+
+def test_enum_choice() -> None:
+    # This test is only for coverage of the new custom TyperChoice class
+    result = runner.invoke(app, ["--name", "morty"], catch_exceptions=False)
+    # assert result.exit_code == 0
+    assert "Hello Morty!" in result.output
+
+    result = runner.invoke(app, ["--name", "Rick"])
+    assert result.exit_code == 0
+    assert "Hello Rick!" in result.output
+
+    result = runner.invoke(app, ["--name", "RICK"])
+    assert result.exit_code == 0
+    assert "Hello Rick!" in result.output
