Unit tests for cmdline.py

1 files changed, 29 insertions, 0 deletions

diff --git a/test/test_cmdline.py b/test/test_cmdline.py
new file mode 100644
index 00000000..799e0e24
--- /dev/null
+++ b/test/test_cmdline.py
@@ -0,0 +1,29 @@
+import optparse
+import libpry
+from libmproxy import cmdline
+
+
+class uAll(libpry.AutoTree):
+    def test_common(self):
+        parser = optparse.OptionParser()
+        cmdline.common_options(parser)
+        opts, args = parser.parse_args(args=[])
+
+        assert cmdline.get_common_options(opts)
+
+        opts.stickycookie_all = True
+        v = cmdline.get_common_options(opts)
+        assert v["stickycookie"] == ".*"
+
+        opts.stickycookie_all = False
+        opts.stickycookie_filt = "foo"
+        v = cmdline.get_common_options(opts)
+        assert v["stickycookie"] == "foo"
+
+
+
+
+tests = [
+    uAll()
+]
+