Coverage for tests / test_tutorial / test_parameter_types / test_custom_types / test_tutorial002.py: 100%

1import importlib 8 ctx1abcdefgh

2import subprocess 8 ctx1abcdefgh

3import sys 8 ctx1abcdefgh

4from types import ModuleType 8 ctx1abcdefgh

5 

6import pytest 8 ctx1abcdefgh

7from typer.testing import CliRunner 8 ctx1abcdefgh

8 

9runner = CliRunner() 8 ctx1abcdefgh

10 

11 

12@pytest.fixture( 8 ctx1abcdefgh

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial002_py39"), 

16 pytest.param("tutorial002_an_py39"), 

17 ], 

18) 

19def get_mod(request: pytest.FixtureRequest) -> ModuleType: 8 ctx1abcdefgh

20 module_name = f"docs_src.parameter_types.custom_types.{request.param}" 8 ctx1abcdefgh

21 mod = importlib.import_module(module_name) 8 ctx1abcdefgh

22 return mod 8 ctx1abcdefgh

23 

24 

25def test_help(mod: ModuleType): 8 ctx1abcdefgh

26 result = runner.invoke(mod.app, ["--help"]) 8 ctx1abcdefgh

27 assert result.exit_code == 0 8 ctx1abcdefgh

28 

29 

30def test_parse_custom_type(mod: ModuleType): 8 ctx1abcdefgh

31 result = runner.invoke(mod.app, ["0", "--custom-opt", "1"]) 8 ctx1abcdefgh

32 assert "custom_arg is <CustomClass: value=000>" in result.output 8 ctx1abcdefgh

33 assert "custom-opt is <CustomClass: value=111>" in result.output 8 ctx1abcdefgh

34 

35 

36def test_parse_custom_type_with_default(mod: ModuleType): 8 ctx1abcdefgh

37 result = runner.invoke(mod.app, ["0"]) 8 ctx1abcdefgh

38 assert "custom_arg is <CustomClass: value=000>" in result.output 8 ctx1abcdefgh

39 assert "custom-opt is <CustomClass: value=FooFooFoo>" in result.output 8 ctx1abcdefgh

40 

41 

42def test_script(mod: ModuleType): 8 ctx1abcdefgh

43 result = subprocess.run( 8 ctx1abcdefgh

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

45 capture_output=True, 

46 encoding="utf-8", 

47 ) 

48 assert "Usage" in result.stdout 8 ctx1abcdefgh