Coverage for tests/test_annotated.py: 100%

1import typer 9 ctx1iabgcdhef

2from typer.testing import CliRunner 9 ctx1iabgcdhef

3from typing_extensions import Annotated 9 ctx1iabgcdhef

4 

5from .utils import needs_py310 9 ctx1iabgcdhef

6 

7runner = CliRunner() 9 ctx1iabgcdhef

8 

9 

10def test_annotated_argument_with_default(): 9 ctx1iabgcdhef

11 app = typer.Typer() 9 ctx1iabgcdhef

12 

13 @app.command() 9 ctx1iabgcdhef

14 def cmd(val: Annotated[int, typer.Argument()] = 0): 9 ctx1iabgcdhef

15 print(f"hello {val}") 9 ctx1iabgcdhef

16 

17 result = runner.invoke(app) 9 ctx1iabgcdhef

18 assert result.exit_code == 0, result.output 9 ctx1iabgcdhef

19 assert "hello 0" in result.output 9 ctx1iabgcdhef

20 

21 result = runner.invoke(app, ["42"]) 9 ctx1iabgcdhef

22 assert result.exit_code == 0, result.output 9 ctx1iabgcdhef

23 assert "hello 42" in result.output 9 ctx1iabgcdhef

24 

25 

26@needs_py310 9 ctx1iabgcdhef

27def test_annotated_argument_in_string_type_with_default(): 8 ctx1abgcdhef

28 app = typer.Typer() 6 ctx1abcdef

29 

30 @app.command() 6 ctx1abcdef

31 def cmd(val: "Annotated[int, typer.Argument()]" = 0): 6 ctx1abcdef

32 print(f"hello {val}") 6 ctx1abcdef

33 

34 result = runner.invoke(app) 6 ctx1abcdef

35 assert result.exit_code == 0, result.output 6 ctx1abcdef

36 assert "hello 0" in result.output 6 ctx1abcdef

37 

38 result = runner.invoke(app, ["42"]) 6 ctx1abcdef

39 assert result.exit_code == 0, result.output 6 ctx1abcdef

40 assert "hello 42" in result.output 6 ctx1abcdef

41 

42 

43def test_annotated_argument_with_default_factory(): 9 ctx1iabgcdhef

44 app = typer.Typer() 9 ctx1iabgcdhef

45 

46 def make_string(): 9 ctx1iabgcdhef

47 return "I made it" 9 ctx1iabgcdhef

48 

49 @app.command() 9 ctx1iabgcdhef

50 def cmd(val: Annotated[str, typer.Argument(default_factory=make_string)]): 9 ctx1iabgcdhef

51 print(val) 9 ctx1iabgcdhef

52 

53 result = runner.invoke(app) 9 ctx1iabgcdhef

54 assert result.exit_code == 0, result.output 9 ctx1iabgcdhef

55 assert "I made it" in result.output 9 ctx1iabgcdhef

56 

57 result = runner.invoke(app, ["overridden"]) 9 ctx1iabgcdhef

58 assert result.exit_code == 0, result.output 9 ctx1iabgcdhef

59 assert "overridden" in result.output 9 ctx1iabgcdhef

60 

61 

62def test_annotated_option_with_argname_doesnt_mutate_multiple_calls(): 9 ctx1iabgcdhef

63 app = typer.Typer() 9 ctx1iabgcdhef

64 

65 @app.command() 9 ctx1iabgcdhef

66 def cmd(force: Annotated[bool, typer.Option("--force")] = False): 9 ctx1iabgcdhef

67 if force: 9 ctx1iabgcdhef

68 print("Forcing operation") 9 ctx1iabgcdhef

69 else: 

70 print("Not forcing") 9 ctx1iabgcdhef

71 

72 result = runner.invoke(app) 9 ctx1iabgcdhef

73 assert result.exit_code == 0, result.output 9 ctx1iabgcdhef

74 assert "Not forcing" in result.output 9 ctx1iabgcdhef

75 

76 result = runner.invoke(app, ["--force"]) 9 ctx1iabgcdhef

77 assert result.exit_code == 0, result.output 9 ctx1iabgcdhef

78 assert "Forcing operation" in result.output 9 ctx1iabgcdhef