Spaces:
Runtime error
Runtime error
File size: 1,320 Bytes
0cac9cf | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | import unittest
from rich.console import Console
from queryquest.cli import is_quit_command, normalize_prompt_input, parse_args
class CliTests(unittest.TestCase):
def setUp(self) -> None:
self.console = Console(record=True)
def test_is_quit_command_variants(self) -> None:
self.assertTrue(is_quit_command("-q"))
self.assertTrue(is_quit_command("qq --quit"))
self.assertFalse(is_quit_command("show listings"))
def test_normalize_prompt_input(self) -> None:
self.assertEqual(normalize_prompt_input("-p hello"), ("hello", False))
self.assertEqual(normalize_prompt_input("--prompt world"), ("world", False))
self.assertEqual(normalize_prompt_input("-p"), ("", True))
def test_parse_args_prompt(self) -> None:
options = parse_args(["-p", "top 5 cities"], self.console)
self.assertFalse(options.setup)
self.assertEqual(options.prompt, "top 5 cities")
def test_parse_args_setup(self) -> None:
options = parse_args(["--setup"], self.console)
self.assertTrue(options.setup)
self.assertIsNone(options.prompt)
def test_parse_args_help_exits(self) -> None:
with self.assertRaises(SystemExit):
parse_args(["-h"], self.console)
if __name__ == "__main__":
unittest.main()
|