mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 16:17:49 +00:00
96 lines
2.5 KiB
Python
96 lines
2.5 KiB
Python
import optparse
|
|
import libpry
|
|
from libmproxy import cmdline
|
|
|
|
|
|
class uAll(libpry.AutoTree):
|
|
def test_parse_replace_hook(self):
|
|
x = cmdline.parse_replace_hook("/foo/bar/voing")
|
|
assert x == ("foo", "bar", "voing")
|
|
|
|
x = cmdline.parse_replace_hook("/foo/bar/vo/ing/")
|
|
assert x == ("foo", "bar", "vo/ing/")
|
|
|
|
x = cmdline.parse_replace_hook("/bar/voing")
|
|
assert x == (".*", "bar", "voing")
|
|
|
|
libpry.raises(
|
|
cmdline.ParseReplaceException,
|
|
cmdline.parse_replace_hook,
|
|
"/foo"
|
|
)
|
|
libpry.raises(
|
|
"replacement regex",
|
|
cmdline.parse_replace_hook,
|
|
"patt/[/rep"
|
|
)
|
|
libpry.raises(
|
|
"filter pattern",
|
|
cmdline.parse_replace_hook,
|
|
"/~/foo/rep"
|
|
)
|
|
libpry.raises(
|
|
"empty replacement regex",
|
|
cmdline.parse_replace_hook,
|
|
"//"
|
|
)
|
|
|
|
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
|
|
opts.stickyauth_all = True
|
|
v = cmdline.get_common_options(opts)
|
|
assert v["stickycookie"] == ".*"
|
|
assert v["stickyauth"] == ".*"
|
|
|
|
opts.stickycookie_all = False
|
|
opts.stickyauth_all = False
|
|
opts.stickycookie_filt = "foo"
|
|
opts.stickyauth_filt = "foo"
|
|
v = cmdline.get_common_options(opts)
|
|
assert v["stickycookie"] == "foo"
|
|
assert v["stickyauth"] == "foo"
|
|
|
|
opts.replace = ["/foo/bar/voing"]
|
|
v = cmdline.get_common_options(opts)
|
|
assert v["replacements"] == [("foo", "bar", "voing")]
|
|
|
|
opts.replace = ["//"]
|
|
libpry.raises(
|
|
"empty replacement regex",
|
|
cmdline.get_common_options,
|
|
opts
|
|
)
|
|
|
|
opts.replace = []
|
|
opts.replace_file = [("/foo/bar/nonexistent")]
|
|
libpry.raises(
|
|
"could not read replace file",
|
|
cmdline.get_common_options,
|
|
opts
|
|
)
|
|
|
|
opts.replace_file = [("/~/bar/nonexistent")]
|
|
libpry.raises(
|
|
"filter pattern",
|
|
cmdline.get_common_options,
|
|
opts
|
|
)
|
|
|
|
opts.replace_file = [("/foo/bar/./data/replace")]
|
|
v = cmdline.get_common_options(opts)["replacements"]
|
|
assert len(v) == 1
|
|
assert v[0][2].strip() == "replacecontents"
|
|
|
|
|
|
|
|
tests = [
|
|
uAll()
|
|
]
|
|
|