Coverage for tests / test_suggest_commands.py: 100%

1import typer 7 ctx1abcdefg

2from typer.testing import CliRunner 7 ctx1abcdefg

3 

4runner = CliRunner() 7 ctx1abcdefg

5 

6 

7def test_typo_suggestion_enabled(): 7 ctx1abcdefg

8 """Test that typo suggestions work when enabled""" 

9 app = typer.Typer() 7 ctx1abcdefg

10 

11 @app.command() 7 ctx1abcdefg

12 def create(): # pragma: no cover 7 ctx1abcdefg

13 typer.echo("Creating...") 

14 

15 @app.command() 7 ctx1abcdefg

16 def delete(): # pragma: no cover 7 ctx1abcdefg

17 typer.echo("Deleting...") 

18 

19 result = runner.invoke(app, ["crate"]) 7 ctx1abcdefg

20 assert result.exit_code != 0 7 ctx1abcdefg

21 assert "No such command" in result.output 7 ctx1abcdefg

22 assert "Did you mean 'create'?" in result.output 7 ctx1abcdefg

23 

24 

25def test_typo_suggestion_multiple_matches(): 7 ctx1abcdefg

26 """Test that multiple suggestions are shown when there are multiple close matches""" 

27 app = typer.Typer() 7 ctx1abcdefg

28 

29 @app.command() 7 ctx1abcdefg

30 def create(): # pragma: no cover 7 ctx1abcdefg

31 typer.echo("Creating...") 

32 

33 @app.command() 7 ctx1abcdefg

34 def createnew(): # pragma: no cover 7 ctx1abcdefg

35 typer.echo("Creating new...") 

36 

37 result = runner.invoke(app, ["crate"]) 7 ctx1abcdefg

38 assert result.exit_code != 0 7 ctx1abcdefg

39 assert "No such command" in result.output 7 ctx1abcdefg

40 assert "Did you mean" in result.output 7 ctx1abcdefg

41 assert "create" in result.output and "createnew" in result.output 7 ctx1abcdefg

42 

43 

44def test_typo_suggestion_no_matches(): 7 ctx1abcdefg

45 """Test that no suggestions are shown when there are no close matches""" 

46 app = typer.Typer() 7 ctx1abcdefg

47 

48 @app.command() 7 ctx1abcdefg

49 def create(): # pragma: no cover 7 ctx1abcdefg

50 typer.echo("Creating...") 

51 

52 @app.command() 7 ctx1abcdefg

53 def delete(): # pragma: no cover 7 ctx1abcdefg

54 typer.echo("Deleting...") 

55 

56 result = runner.invoke(app, ["xyz"]) 7 ctx1abcdefg

57 assert result.exit_code != 0 7 ctx1abcdefg

58 assert "No such command" in result.output 7 ctx1abcdefg

59 assert "Did you mean" not in result.output 7 ctx1abcdefg

60 

61 

62def test_typo_suggestion_exact_match_works(): 7 ctx1abcdefg

63 """Test that exact matches still work normally""" 

64 app = typer.Typer() 7 ctx1abcdefg

65 

66 @app.command() 7 ctx1abcdefg

67 def create(): 7 ctx1abcdefg

68 typer.echo("Creating...") 7 ctx1abcdefg

69 

70 @app.command() 7 ctx1abcdefg

71 def delete(): 7 ctx1abcdefg

72 typer.echo("Deleting...") 7 ctx1abcdefg

73 

74 result = runner.invoke(app, ["create"]) 7 ctx1abcdefg

75 assert result.exit_code == 0 7 ctx1abcdefg

76 assert "Creating..." in result.output 7 ctx1abcdefg

77 

78 result = runner.invoke(app, ["delete"]) 7 ctx1abcdefg

79 assert result.exit_code == 0 7 ctx1abcdefg

80 assert "Deleting..." in result.output 7 ctx1abcdefg

81 

82 

83def test_typo_suggestion_disabled(): 7 ctx1abcdefg

84 """Test that typo suggestions can be explicitly disabled""" 

85 app = typer.Typer(suggest_commands=False) 7 ctx1abcdefg

86 

87 @app.command() 7 ctx1abcdefg

88 def create(): # pragma: no cover 7 ctx1abcdefg

89 typer.echo("Creating...") 

90 

91 @app.command() 7 ctx1abcdefg

92 def delete(): # pragma: no cover 7 ctx1abcdefg

93 typer.echo("Deleting...") 

94 

95 result = runner.invoke(app, ["crate"]) 7 ctx1abcdefg

96 assert result.exit_code != 0 7 ctx1abcdefg

97 assert "No such command" in result.output 7 ctx1abcdefg

98 assert "Did you mean" not in result.output 7 ctx1abcdefg