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

1import subprocess 8 ctx1abcdefgh

2import sys 8 ctx1abcdefgh

3from unittest.mock import patch 8 ctx1abcdefgh

4 

5import typer 8 ctx1abcdefgh

6from typer.testing import CliRunner 8 ctx1abcdefgh

7 

8import docs_src.progressbar.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_one_step(): 8 ctx1abcdefgh

17 with patch("time.sleep") as sleep_mock: 8 ctx1abcdefgh

18 sleep_mock.side_effect = typer.Exit() # Exit on first `time.sleep()` call 8 ctx1abcdefgh

19 result = runner.invoke(app) 8 ctx1abcdefgh

20 

21 # Replace all Rich formatting with `*` characters 

22 normalized_output = normalize_rich_output(result.output) 8 ctx1abcdefgh

23 

24 assert result.exit_code == 0 8 ctx1abcdefgh

25 assert "Processing... 0%" in normalized_output 8 ctx1abcdefgh

26 

27 

28def test_cli(): 8 ctx1abcdefgh

29 with patch("time.sleep") as mock_sleep: 8 ctx1abcdefgh

30 result = runner.invoke(app) 8 ctx1abcdefgh

31 

32 # Replace all Rich formatting with `*` characters 

33 normalized_output = normalize_rich_output(result.output) 8 ctx1abcdefgh

34 

35 assert result.exit_code == 0 8 ctx1abcdefgh

36 assert mock_sleep.call_count == 100 8 ctx1abcdefgh

37 assert "Processing..." in normalized_output 8 ctx1abcdefgh

38 assert "100%" in normalized_output 8 ctx1abcdefgh

39 assert "Processed 100 things." in normalized_output 8 ctx1abcdefgh

40 

41 

42def test_script(): 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