Coverage for tests / test_tutorial / test_parameter_types / test_number / test_tutorial001.py: 100%

1import importlib 7 ctx1abcdefg

2import subprocess 7 ctx1abcdefg

3import sys 7 ctx1abcdefg

4from types import ModuleType 7 ctx1abcdefg

5 

6import pytest 7 ctx1abcdefg

7import typer 7 ctx1abcdefg

8import typer.core 7 ctx1abcdefg

9from typer.testing import CliRunner 7 ctx1abcdefg

10 

11runner = CliRunner() 7 ctx1abcdefg

12 

13 

14@pytest.fixture( 7 ctx1abcdefg

15 name="mod", 

16 params=[ 

17 pytest.param("tutorial001_py310"), 

18 pytest.param("tutorial001_an_py310"), 

19 ], 

20) 

21def get_mod(request: pytest.FixtureRequest) -> ModuleType: 7 ctx1abcdefg

22 module_name = f"docs_src.parameter_types.number.{request.param}" 7 ctx1abcdefg

23 mod = importlib.import_module(module_name) 7 ctx1abcdefg

24 return mod 7 ctx1abcdefg

25 

26 

27def test_type_repr(mod: ModuleType): 7 ctx1abcdefg

28 command = typer.main.get_command(mod.app) 7 ctx1abcdefg

29 

30 id_param = next(param for param in command.params if param.name == "ID") 7 ctx1abcdefg

31 assert repr(id_param.type) == "<IntRange 0<=x<=1000>" 7 ctx1abcdefg

32 

33 age_param = next(param for param in command.params if param.name == "age") 7 ctx1abcdefg

34 assert repr(age_param.type) == "<IntRange x>=18>" 7 ctx1abcdefg

35 

36 score_param = next(param for param in command.params if param.name == "score") 7 ctx1abcdefg

37 assert repr(score_param.type) == "<FloatRange x<=100>" 7 ctx1abcdefg

38 

39 

40def test_help(mod: ModuleType): 7 ctx1abcdefg

41 result = runner.invoke(mod.app, ["--help"]) 7 ctx1abcdefg

42 assert result.exit_code == 0 7 ctx1abcdefg

43 assert "--age" in result.output 7 ctx1abcdefg

44 assert "int range" in result.output 7 ctx1abcdefg

45 assert "--score" in result.output 7 ctx1abcdefg

46 assert "float range" in result.output 7 ctx1abcdefg

47 

48 

49def test_help_no_rich(monkeypatch: pytest.MonkeyPatch, mod: ModuleType): 7 ctx1abcdefg

50 monkeypatch.setattr(typer.core, "HAS_RICH", False) 7 ctx1abcdefg

51 result = runner.invoke(mod.app, ["--help"]) 7 ctx1abcdefg

52 assert result.exit_code == 0 7 ctx1abcdefg

53 assert "--age" in result.output 7 ctx1abcdefg

54 assert "int range" in result.output 7 ctx1abcdefg

55 assert "--score" in result.output 7 ctx1abcdefg

56 assert "float range" in result.output 7 ctx1abcdefg

57 

58 

59def test_params(mod: ModuleType): 7 ctx1abcdefg

60 result = runner.invoke(mod.app, ["5", "--age", "20", "--score", "90"]) 7 ctx1abcdefg

61 assert result.exit_code == 0 7 ctx1abcdefg

62 assert "ID is 5" in result.output 7 ctx1abcdefg

63 assert "--age is 20" in result.output 7 ctx1abcdefg

64 assert "--score is 90.0" in result.output 7 ctx1abcdefg

65 

66 

67def test_invalid_id(mod: ModuleType): 7 ctx1abcdefg

68 result = runner.invoke(mod.app, ["1002"]) 7 ctx1abcdefg

69 assert result.exit_code != 0 7 ctx1abcdefg

70 assert ( 7 ctx1abcdefg

71 "Invalid value for 'ID': 1002 is not in the range 0<=x<=1000." in result.output 

72 ) 

73 

74 

75def test_invalid_age(mod: ModuleType): 7 ctx1abcdefg

76 result = runner.invoke(mod.app, ["5", "--age", "15"]) 7 ctx1abcdefg

77 assert result.exit_code != 0 7 ctx1abcdefg

78 assert "Invalid value for '--age'" in result.output 7 ctx1abcdefg

79 assert "15 is not in the range x>=18" in result.output 7 ctx1abcdefg

80 

81 

82def test_invalid_score(monkeypatch: pytest.MonkeyPatch, mod: ModuleType): 7 ctx1abcdefg

83 monkeypatch.setattr(typer.core, "HAS_RICH", False) 7 ctx1abcdefg

84 result = runner.invoke(mod.app, ["5", "--age", "20", "--score", "100.5"]) 7 ctx1abcdefg

85 assert result.exit_code != 0 7 ctx1abcdefg

86 assert "Invalid value for '--score'" in result.output 7 ctx1abcdefg

87 assert "100.5 is not in the range x<=100." in result.output 7 ctx1abcdefg

88 

89 

90def test_negative_score(mod: ModuleType): 7 ctx1abcdefg

91 result = runner.invoke(mod.app, ["5", "--age", "20", "--score", "-5"]) 7 ctx1abcdefg

92 assert result.exit_code == 0 7 ctx1abcdefg

93 assert "ID is 5" in result.output 7 ctx1abcdefg

94 assert "--age is 20" in result.output 7 ctx1abcdefg

95 assert "--score is -5.0" in result.output 7 ctx1abcdefg

96 

97 

98def test_script(mod: ModuleType): 7 ctx1abcdefg

99 result = subprocess.run( 7 ctx1abcdefg

100 [sys.executable, "-m", "coverage", "run", mod.__file__, "--help"], 

101 capture_output=True, 

102 encoding="utf-8", 

103 ) 

104 assert "Usage" in result.stdout 7 ctx1abcdefg