Coverage for tests / test_tutorial / test_printing / test_tutorial002.py: 100%

1import subprocess 7 ctx1abcdefg

2import sys 7 ctx1abcdefg

3from unittest.mock import patch 7 ctx1abcdefg

4 

5from rich.console import Console # noqa: TID251 7 ctx1abcdefg

6from typer.testing import CliRunner 7 ctx1abcdefg

7 

8import docs_src.printing.tutorial002_py310 as mod 7 ctx1abcdefg

9from tests.utils import normalize_rich_output 7 ctx1abcdefg

10 

11app = mod.app 7 ctx1abcdefg

12 

13runner = CliRunner() 7 ctx1abcdefg

14 

15 

16def test_cli(): 7 ctx1abcdefg

17 console = Console(force_terminal=True, width=100) 7 ctx1abcdefg

18 with patch("rich.get_console", return_value=console): 7 ctx1abcdefg

19 result = runner.invoke(app) 7 ctx1abcdefg

20 

21 # Replace all Rich formatting with `*` characters 

22 normalized_output = normalize_rich_output(result.output) 7 ctx1abcdefg

23 

24 assert result.exit_code == 0 7 ctx1abcdefg

25 assert "*Alert!* *Portal gun* shooting! *" in normalized_output 7 ctx1abcdefg

26 

27 

28def test_cli_without_formatting(): 7 ctx1abcdefg

29 result = runner.invoke(app) 7 ctx1abcdefg

30 

31 assert result.exit_code == 0 7 ctx1abcdefg

32 assert "Alert! Portal gun shooting! 💥" in result.output 7 ctx1abcdefg

33 

34 

35def test_script(): 7 ctx1abcdefg

36 result = subprocess.run( 7 ctx1abcdefg

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

38 capture_output=True, 

39 encoding="utf-8", 

40 ) 

41 assert "Usage" in result.stdout 7 ctx1abcdefg