diff options
| -rw-r--r-- | libmproxy/cmdline.py | 2 | ||||
| -rw-r--r-- | test/test_cmdline.py | 29 |
2 files changed, 30 insertions, 1 deletions
diff --git a/libmproxy/cmdline.py b/libmproxy/cmdline.py index 789cfaec..58dbadad 100644 --- a/libmproxy/cmdline.py +++ b/libmproxy/cmdline.py @@ -7,7 +7,7 @@ def get_common_options(options): if options.stickycookie_all: stickycookie = ".*" elif options.stickycookie_filt: - stickycookie = stickycookie_filt + stickycookie = options.stickycookie_filt return dict( verbosity = options.verbose, wfile = options.wfile, 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() +] + |