diff --git a/tests/test_cli.py b/tests/test_cli.py index 4079c582..07ba93d5 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -25,7 +25,10 @@ class MyTests(unittest.TestCase): 'jc -h': (False, None, None, []), 'jc -h --arp': (False, None, None, []), 'jc -h arp': (False, None, None, []), - 'jc -h arp -a': (False, None, None, []) + 'jc -h arp -a': (False, None, None, []), + 'jc --pretty dig': (True, ['dig'], '--dig', ['p']), + 'jc --pretty --monochrome --quiet --raw dig': (True, ['dig'], '--dig', ['p', 'm', 'q', 'r']), + 'jc --about --yaml-out': (False, None, None, []) } for command, expected_command in commands.items():