Coverage for tests / test_tutorial / test_parameter_types / test_custom_types / 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

7from typer.testing import CliRunner 7 ctx1abcdefg

8 

9runner = CliRunner() 7 ctx1abcdefg

10 

11 

12@pytest.fixture( 7 ctx1abcdefg

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial001_py310"), 

16 pytest.param("tutorial001_an_py310"), 

17 ], 

18) 

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

20 module_name = f"docs_src.parameter_types.custom_types.{request.param}" 7 ctx1abcdefg

21 mod = importlib.import_module(module_name) 7 ctx1abcdefg

22 return mod 7 ctx1abcdefg

23 

24 

25def test_help(mod: ModuleType): 7 ctx1abcdefg

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

27 assert result.exit_code == 0 7 ctx1abcdefg

28 

29 

30def test_parse_custom_type(mod: ModuleType): 7 ctx1abcdefg

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

32 assert "custom_arg is <CustomClass: value=00>" in result.output 7 ctx1abcdefg

33 assert "custom-opt is <CustomClass: value=11>" in result.output 7 ctx1abcdefg

34 

35 

36def test_parse_custom_type_with_default(mod: ModuleType): 7 ctx1abcdefg

37 result = runner.invoke(mod.app, ["0"]) 7 ctx1abcdefg

38 assert "custom_arg is <CustomClass: value=00>" in result.output 7 ctx1abcdefg

39 assert "custom-opt is <CustomClass: value=FooFoo>" in result.output 7 ctx1abcdefg

40 

41 

42def test_script(mod: ModuleType): 7 ctx1abcdefg

43 result = subprocess.run( 7 ctx1abcdefg

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

45 capture_output=True, 

46 encoding="utf-8", 

47 ) 

48 assert "Usage" in result.stdout 7 ctx1abcdefg