Coverage for tests / test_types.py: 100%

1from enum import Enum 8 ctx1abcdefgh

2 

3import typer 8 ctx1abcdefgh

4from typer.testing import CliRunner 8 ctx1abcdefgh

5 

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

7 

8 

9class User(str, Enum): 8 ctx1abcdefgh

10 rick = "Rick" 8 ctx1abcdefgh

11 morty = "Morty" 8 ctx1abcdefgh

12 

13 

14@app.command() 8 ctx1abcdefgh

15def hello(name: User = User.rick) -> None: 8 ctx1abcdefgh

16 print(f"Hello {name.value}!") 8 ctx1abcdefgh

17 

18 

19runner = CliRunner() 8 ctx1abcdefgh

20 

21 

22def test_enum_choice() -> None: 8 ctx1abcdefgh

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

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

25 assert result.exit_code == 0 8 ctx1abcdefgh

26 assert "Hello Morty!" in result.output 8 ctx1abcdefgh

27 

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

29 assert result.exit_code == 0 8 ctx1abcdefgh

30 assert "Hello Rick!" in result.output 8 ctx1abcdefgh

31 

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

33 assert result.exit_code == 0 8 ctx1abcdefgh

34 assert "Hello Rick!" in result.output 8 ctx1abcdefgh