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

1import subprocess 8 ctx1adebfgch

2import sys 8 ctx1adebfgch

3from unittest.mock import patch 8 ctx1adebfgch

4 

5from typer.testing import CliRunner 8 ctx1adebfgch

6 

7import docs_src.progressbar.tutorial004_py39 as mod 8 ctx1adebfgch

8from tests.utils import normalize_rich_output 8 ctx1adebfgch

9 

10app = mod.app 8 ctx1adebfgch

11 

12runner = CliRunner() 8 ctx1adebfgch

13 

14 

15def test_cli(): # Checking only final state of progress bar 8 ctx1adebfgch

16 consumed = [] 8 ctx1adebfgch

17 

18 def fake_iterate_user_ids(): 8 ctx1adebfgch

19 for i in range(100): 8 ctx1adebfgch

20 consumed.append(i) 8 ctx1adebfgch

21 yield i 8 ctx1adebfgch

22 

23 with ( 3 ctx1abc

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

25 patch( 

26 "docs_src.progressbar.tutorial004_py39.iterate_user_ids", 

27 side_effect=fake_iterate_user_ids, 

28 ), 

29 ): 

30 result = runner.invoke(app) 8 ctx1adebfgch

31 

32 # Replace all Rich formatting with `*` characters 

33 normalized_output = normalize_rich_output(result.output) 8 ctx1adebfgch

34 

35 assert result.exit_code == 0 8 ctx1adebfgch

36 assert len(consumed) == 100 8 ctx1adebfgch

37 assert mock_sleep.call_count == 100 8 ctx1adebfgch

38 assert "Processed 100 user IDs." in normalized_output 8 ctx1adebfgch

39 

40 

41def test_cli_no_mock_generator(): 8 ctx1adebfgch

42 with ( 3 ctx1abc

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

44 ): 

45 result = runner.invoke(app) 8 ctx1adebfgch

46 

47 # Replace all Rich formatting with `*` characters 

48 normalized_output = normalize_rich_output(result.output) 8 ctx1adebfgch

49 

50 assert result.exit_code == 0 8 ctx1adebfgch

51 assert mock_sleep.call_count == 100 8 ctx1adebfgch

52 assert "Processed 100 user IDs." in normalized_output 8 ctx1adebfgch

53 

54 

55def test_script(): 8 ctx1adebfgch

56 result = subprocess.run( 8 ctx1adebfgch

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

58 capture_output=True, 

59 encoding="utf-8", 

60 ) 

61 assert "Usage" in result.stdout 8 ctx1adebfgch