Coverage for tests/test_types.py: 100%

1from enum import Enum 9 ctx1abcdefghi

2 

3import typer 9 ctx1abcdefghi

4from typer.testing import CliRunner 9 ctx1abcdefghi

5 

6app = typer.Typer(context_settings={"token_normalize_func": str.lower}) 9 ctx1abcdefghi

7 

8 

9class User(str, Enum): 9 ctx1abcdefghi

10 rick = "Rick" 9 ctx1abcdefghi

11 morty = "Morty" 9 ctx1abcdefghi

12 

13 

14@app.command() 9 ctx1abcdefghi

15def hello(name: User = User.rick) -> None: 9 ctx1abcdefghi

16 print(f"Hello {name.value}!") 9 ctx1abcdefghi

17 

18 

19runner = CliRunner() 9 ctx1abcdefghi

20 

21 

22def test_enum_choice() -> None: 9 ctx1abcdefghi

23 # This test is only for coverage of the new custom TyperChoice class 

24 result = runner.invoke(app, ["--name", "morty"], catch_exceptions=False) 9 ctx1abcdefghi

25 assert result.exit_code == 0 9 ctx1abcdefghi

26 assert "Hello Morty!" in result.output 9 ctx1abcdefghi

27 

28 result = runner.invoke(app, ["--name", "Rick"]) 9 ctx1abcdefghi

29 assert result.exit_code == 0 9 ctx1abcdefghi

30 assert "Hello Rick!" in result.output 9 ctx1abcdefghi

31 

32 result = runner.invoke(app, ["--name", "RICK"]) 9 ctx1abcdefghi

33 assert result.exit_code == 0 9 ctx1abcdefghi

34 assert "Hello Rick!" in result.output 9 ctx1abcdefghi