mirror of
https://github.com/retspen/webvirtcloud
synced 2026-10-18 09:00:07 +00:00
57 lines
2.9 KiB
Python
57 lines
2.9 KiB
Python
import argparse
|
|
import unittest
|
|
|
|
|
|
class TestCLIParsers(unittest.TestCase):
|
|
def test_novncd_parser_defaults(self):
|
|
# We test the parser specification without importing django
|
|
parser = argparse.ArgumentParser(description="WebVirtCloud noVNC daemon")
|
|
parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", default=False)
|
|
parser.add_argument("-d", "--debug", dest="debug", action="store_true", default=False)
|
|
parser.add_argument("-H", "--host", dest="host", action="store", default="0.0.0.0")
|
|
parser.add_argument("-p", "--port", dest="port", action="store", type=int, default=6080)
|
|
parser.add_argument("-c", "--cert", dest="cert", action="store", default="cert.pem")
|
|
|
|
opts = parser.parse_args([])
|
|
self.assertFalse(opts.verbose)
|
|
self.assertFalse(opts.debug)
|
|
self.assertEqual(opts.host, "0.0.0.0")
|
|
self.assertEqual(opts.port, 6080)
|
|
self.assertEqual(opts.cert, "cert.pem")
|
|
|
|
def test_novncd_parser_custom_args(self):
|
|
parser = argparse.ArgumentParser(description="WebVirtCloud noVNC daemon")
|
|
parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", default=False)
|
|
parser.add_argument("-d", "--debug", dest="debug", action="store_true", default=False)
|
|
parser.add_argument("-H", "--host", dest="host", action="store", default="0.0.0.0")
|
|
parser.add_argument("-p", "--port", dest="port", action="store", type=int, default=6080)
|
|
parser.add_argument("-c", "--cert", dest="cert", action="store", default="cert.pem")
|
|
|
|
opts = parser.parse_args(["-v", "-d", "-H", "192.168.1.50", "-p", "7000", "-c", "/etc/ssl/cert.pem"])
|
|
self.assertTrue(opts.verbose)
|
|
self.assertTrue(opts.debug)
|
|
self.assertEqual(opts.host, "192.168.1.50")
|
|
self.assertEqual(opts.port, 7000)
|
|
self.assertEqual(opts.cert, "/etc/ssl/cert.pem")
|
|
|
|
def test_socketiod_parser_defaults_and_custom(self):
|
|
parser = argparse.ArgumentParser(description="WebVirtCloud Socket.IO daemon")
|
|
parser.add_argument("-v", "--verbose", dest="verbose", action="store_true", default=False)
|
|
parser.add_argument("-d", "--debug", dest="debug", action="store_true", default=False)
|
|
parser.add_argument("-H", "--host", dest="host", action="store", default="localhost")
|
|
parser.add_argument("-p", "--port", dest="port", action="store", type=int, default=6081)
|
|
|
|
opts_def = parser.parse_args([])
|
|
self.assertFalse(opts_def.verbose)
|
|
self.assertFalse(opts_def.debug)
|
|
self.assertEqual(opts_def.host, "localhost")
|
|
self.assertEqual(opts_def.port, 6081)
|
|
|
|
opts_custom = parser.parse_args(["--verbose", "--host", "10.0.0.1", "--port", "9000"])
|
|
self.assertTrue(opts_custom.verbose)
|
|
self.assertEqual(opts_custom.host, "10.0.0.1")
|
|
self.assertEqual(opts_custom.port, 9000)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|