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

1import subprocess 8 ctx1abcdefgh

2import sys 8 ctx1abcdefgh

3from unittest.mock import patch 8 ctx1abcdefgh

4 

5from rich.console import Console # noqa: TID251 8 ctx1abcdefgh

6from typer.testing import CliRunner 8 ctx1abcdefgh

7 

8import docs_src.printing.tutorial001_py39 as mod 8 ctx1abcdefgh

9from tests.utils import normalize_rich_output 8 ctx1abcdefgh

10 

11app = mod.app 8 ctx1abcdefgh

12 

13runner = CliRunner() 8 ctx1abcdefgh

14 

15 

16def test_cli(): 8 ctx1abcdefgh

17 console = Console(force_terminal=True, width=100) 8 ctx1abcdefgh

18 with patch("rich.get_console", return_value=console): 8 ctx1abcdefgh

19 result = runner.invoke(app) 8 ctx1abcdefgh

20 

21 assert result.exit_code == 0 8 ctx1abcdefgh

22 

23 # Replace all Rich formatting with `*` characters 

24 normalized_output = normalize_rich_output(result.output, squash_whitespaces=False) 8 ctx1abcdefgh

25 

26 assert ( 8 ctx1abcdefgh

27 "Here's the data\n" 

28 "*{*\n" 

29 " *'name'*: *'Rick'*,\n" 

30 " *'age'*: *42*,\n" 

31 " *'items'*: *[*{*'name'*: *'Portal Gun'*}*, *{*'name'*: *'Plumbus'*}*]*,\n" 

32 " *'active'*: *True*,\n" 

33 " *'affiliation'*: *None*\n" 

34 "*}*\n" 

35 ) in normalized_output 

36 

37 

38def test_script(): 8 ctx1abcdefgh

39 result = subprocess.run( 8 ctx1abcdefgh

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

41 capture_output=True, 

42 encoding="utf-8", 

43 ) 

44 assert "Usage" in result.stdout 8 ctx1abcdefgh