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

1import subprocess 7 ctx1acdefbg

2import sys 7 ctx1acdefbg

3from unittest.mock import patch 7 ctx1acdefbg

4 

5from typer.testing import CliRunner 7 ctx1acdefbg

6 

7import docs_src.progressbar.tutorial004_py310 as mod 7 ctx1acdefbg

8from tests.utils import normalize_rich_output 7 ctx1acdefbg

9 

10app = mod.app 7 ctx1acdefbg

11 

12runner = CliRunner() 7 ctx1acdefbg

13 

14 

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

16 consumed = [] 7 ctx1acdefbg

17 

18 def fake_iterate_user_ids(): 7 ctx1acdefbg

19 for i in range(100): 7 ctx1acdefbg

20 consumed.append(i) 7 ctx1acdefbg

21 yield i 7 ctx1acdefbg

22 

23 with ( 2 ctx1ab

24 patch("time.sleep") as mock_sleep, 

25 patch( 

26 "docs_src.progressbar.tutorial004_py310.iterate_user_ids", 

27 side_effect=fake_iterate_user_ids, 

28 ), 

29 ): 

30 result = runner.invoke(app) 7 ctx1acdefbg

31 

32 # Replace all Rich formatting with `*` characters 

33 normalized_output = normalize_rich_output(result.output) 7 ctx1acdefbg

34 

35 assert result.exit_code == 0 7 ctx1acdefbg

36 assert len(consumed) == 100 7 ctx1acdefbg

37 assert mock_sleep.call_count == 100 7 ctx1acdefbg

38 assert "Processed 100 user IDs." in normalized_output 7 ctx1acdefbg

39 

40 

41def test_cli_no_mock_generator(): 7 ctx1acdefbg

42 with ( 2 ctx1ab

43 patch("time.sleep") as mock_sleep, 

44 ): 

45 result = runner.invoke(app) 7 ctx1acdefbg

46 

47 # Replace all Rich formatting with `*` characters 

48 normalized_output = normalize_rich_output(result.output) 7 ctx1acdefbg

49 

50 assert result.exit_code == 0 7 ctx1acdefbg

51 assert mock_sleep.call_count == 100 7 ctx1acdefbg

52 assert "Processed 100 user IDs." in normalized_output 7 ctx1acdefbg

53 

54 

55def test_script(): 7 ctx1acdefbg

56 result = subprocess.run( 7 ctx1acdefbg

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

58 capture_output=True, 

59 encoding="utf-8", 

60 ) 

61 assert "Usage" in result.stdout 7 ctx1acdefbg