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

maplocal addon: add some tests

Browse Source
This commit is contained in:
Martin Plattner 2020-07-08 01:34:48 +02:00
parent b1609697cd
commit 53644de820
1 changed files with 79 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

80
test/mitmproxy/addons/test_maplocal.py
View File

@ -1,4 +1,3 @@
import pytest
from pathlib import Path from pathlib import Path
from mitmproxy.addons.maplocal import MapLocal from mitmproxy.addons.maplocal import MapLocal
@ -32,3 +31,82 @@ class TestMapLocal:
spec = parse_modify_spec(":org:" + str(tmpdir), True, True) spec = parse_modify_spec(":org:" + str(tmpdir), True, True)
file_candidates = ml.file_candidates(url, spec) file_candidates = ml.file_candidates(url, spec)
assert file_candidates[0] == str(tmpdir) + "/index.html" assert file_candidates[0] == str(tmpdir) + "/index.html"
def test_sanitize_candidate_path(self, tmpdir):
base_dir = Path(str(tmpdir))
tmpdir.join("testdir1", "testdir2", "testdir3", "testdir4", "testfile").write("bar", ensure=True)
ml = MapLocal()
assert ml.sanitize_candidate_path(
base_dir.joinpath("..", "bar"), base_dir
) is None
assert ml.sanitize_candidate_path(
base_dir.joinpath(".."), base_dir
) is None
assert ml.sanitize_candidate_path(
base_dir.joinpath("..", ".."), base_dir
) is None
assert ml.sanitize_candidate_path(
base_dir.joinpath("..", "..", "..", "..", "..", "..", "etc", "passwd"), base_dir
) is None
assert ml.sanitize_candidate_path(
base_dir.joinpath("testdir1"), base_dir
) is not None
assert ml.sanitize_candidate_path(
base_dir.joinpath("testdir1", "testdir2"), base_dir
) is not None
assert ml.sanitize_candidate_path(
base_dir.joinpath("testdir1", "testdir2", "testdir3", "testdir4", "testfile"), base_dir
) is not None
assert ml.sanitize_candidate_path(
base_dir.joinpath("testdir1", "testdir2", "testdir3", "testdir4", "testfile"),
base_dir.joinpath("testdir1", "testdir2", "testdir3", "testdir4", "testfile")
) is not None
def test_modify_headers(self, tmpdir):
ml = MapLocal()
with taddons.context(ml) as tctx:
tmpfile = tmpdir.join("test1.jpg")
tmpfile.write("local content 1")
tctx.configure(
ml,
map_local=[
":jpg:" + str(tmpdir)
]
)
f = tflow.tflow()
f.request.url = b"https://example.org/images/test1.jpg"
ml.request(f)
assert f.response.content == b"local content 1"
tmpfile = tmpdir.join("images", "test2.jpg")
tmpfile.write("local content 2", ensure=True)
tctx.configure(
ml,
map_local=[
":jpg:" + str(tmpdir)
]
)
f = tflow.tflow()
f.request.url = b"https://example.org/images/test2.jpg"
ml.request(f)
assert f.response.content == b"local content 2"
tmpfile = tmpdir.join("images", "test3.jpg")
tmpfile.write("local content 3", ensure=True)
tctx.configure(
ml,
map_local=[
":jpg:" + str(tmpfile)
]
)
f = tflow.tflow()
f.request.url = b"https://example.org/images/test3.jpg"
ml.request(f)
assert f.response.content == b"local content 3"