Coverage for tests / test_tutorial / test_progressbar / test_tutorial006.py: 100%

1import subprocess 7 ctx1abcdefg

2import sys 7 ctx1abcdefg

3from unittest.mock import patch 7 ctx1abcdefg

4 

5from typer.testing import CliRunner 7 ctx1abcdefg

6 

7import docs_src.progressbar.tutorial006_py310 as mod 7 ctx1abcdefg

8from tests.utils import normalize_rich_output 7 ctx1abcdefg

9 

10app = mod.app 7 ctx1abcdefg

11 

12runner = CliRunner() 7 ctx1abcdefg

13 

14 

15def test_cli(): # Checking only final state of progress bar 7 ctx1abcdefg

16 with patch("time.sleep") as mock_sleep: 7 ctx1abcdefg

17 result = runner.invoke(app) 7 ctx1abcdefg

18 

19 # Replace all Rich formatting with `*` characters 

20 normalized_output = normalize_rich_output(result.output) 7 ctx1abcdefg

21 

22 assert result.exit_code == 0 7 ctx1abcdefg

23 assert mock_sleep.call_count == 4 7 ctx1abcdefg

24 assert "Processed 1000 things in batches." in normalized_output 7 ctx1abcdefg

25 

26 

27def test_script(): 7 ctx1abcdefg

28 result = subprocess.run( 7 ctx1abcdefg

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

30 capture_output=True, 

31 encoding="utf-8", 

32 ) 

33 assert "Usage" in result.stdout 7 ctx1abcdefg