Coverage for tests / test_types.py: 100%

1from enum import Enum 7 ctx1abcdefg

2 

3import typer 7 ctx1abcdefg

4from typer.testing import CliRunner 7 ctx1abcdefg

5 

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

7 

8 

9class User(str, Enum): 7 ctx1abcdefg

10 rick = "Rick" 7 ctx1abcdefg

11 morty = "Morty" 7 ctx1abcdefg

12 

13 

14@app.command() 7 ctx1abcdefg

15def hello(name: User = User.rick) -> None: 7 ctx1abcdefg

16 print(f"Hello {name.value}!") 7 ctx1abcdefg

17 

18 

19runner = CliRunner() 7 ctx1abcdefg

20 

21 

22def test_enum_choice() -> None: 7 ctx1abcdefg

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

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

25 assert result.exit_code == 0 7 ctx1abcdefg

26 assert "Hello Morty!" in result.output 7 ctx1abcdefg

27 

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

29 assert result.exit_code == 0 7 ctx1abcdefg

30 assert "Hello Rick!" in result.output 7 ctx1abcdefg

31 

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

33 assert result.exit_code == 0 7 ctx1abcdefg

34 assert "Hello Rick!" in result.output 7 ctx1abcdefg