diff --git a/tests/test_cli.py b/tests/test_cli.py new file mode 100644 index 00000000..df19a110 --- /dev/null +++ b/tests/test_cli.py @@ -0,0 +1,36 @@ +import io +import sys +import unittest + +from httprunner.cli import main + + +class TestCli(unittest.TestCase): + + def setUp(self): + self.captured_output = io.StringIO() + sys.stdout = self.captured_output + + def test_show_version(self): + sys.argv = ["hrun", "-V"] + + with self.assertRaises(SystemExit) as cm: + main() + sys.stdout = sys.__stdout__ # Reset redirect. + + self.assertEqual(cm.exception.code, 0) + + from httprunner import __version__ + self.assertIn(__version__, self.captured_output.getvalue().strip()) + + def test_show_help(self): + sys.argv = ["hrun", "-h"] + + with self.assertRaises(SystemExit) as cm: + main() + sys.stdout = sys.__stdout__ # Reset redirect. + + self.assertEqual(cm.exception.code, 0) + + from httprunner import __description__ + self.assertIn(__description__, self.captured_output.getvalue().strip())