Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-23 08:11:00 +00:00
Code Issues Projects Releases Wiki Activity

remove quotes around filename

Browse Source
This commit is contained in:
Maximilian Hils 2018-02-01 09:56:56 +01:00
parent b57f1a86af
commit 4decb5c2e3
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mitmproxy/addons/script.py
View File

@ -44,7 +44,9 @@ class Script:
def __init__(self, path):
self.name = "scriptmanager:" + path
self.path = path
self.fullpath = os.path.expanduser(path)
self.fullpath = os.path.expanduser(
path.strip("'\" ")
)
self.ns = None
self.last_load = 0

12
test/mitmproxy/addons/test_script.py
View File

@ -68,6 +68,18 @@ class TestScript:
with pytest.raises(exceptions.OptionsError):
script.Script("nonexistent")
def test_quotes_around_filename(self):
"""
Test that a script specified as '"foo.py"' works to support the calling convention of
mitmproxy 2.0, as e.g. used by Cuckoo Sandbox.
"""
path = tutils.test_data.path("mitmproxy/data/addonscripts/recorder/recorder.py")
s = script.Script(
'"{}"'.format(path)
)
assert '"' not in s.fullpath
def test_simple(self):
with taddons.context() as tctx:
sc = script.Script(