diff options
| -rw-r--r-- | tests/test_cmd2.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tests/test_cmd2.py b/tests/test_cmd2.py index 45051f3a..5e9e4ce6 100644 --- a/tests/test_cmd2.py +++ b/tests/test_cmd2.py @@ -2217,6 +2217,10 @@ def test_disable_and_enable_category(disable_commands_app): out = run_cmd(disable_commands_app, 'help has_no_help_func') assert out == [message_to_print] + visible_commands = disable_commands_app.get_visible_commands() + assert 'has_help_func' not in visible_commands + assert 'has_no_help_func' not in visible_commands + # Enable the category disable_commands_app.enable_category(disable_commands_app.category_name) @@ -2233,6 +2237,10 @@ def test_disable_and_enable_category(disable_commands_app): out = run_cmd(disable_commands_app, 'help has_no_help_func') assert out == ["Help for has_no_help_func"] + visible_commands = disable_commands_app.get_visible_commands() + assert 'has_help_func' in visible_commands + assert 'has_no_help_func' in visible_commands + def test_enable_enabled_command(disable_commands_app): # Test enabling a command that is not disabled saved_len = len(disable_commands_app.disabled_commands) |