From 53644de82093a2b3fc810f422ed6455420dde80d Mon Sep 17 00:00:00 2001
From: Martin Plattner <git@mplattner.at>
Date: Wed, 8 Jul 2020 01:34:48 +0200
Subject: [PATCH] maplocal addon: add some tests

---
 test/mitmproxy/addons/test_maplocal.py | 80 +++++++++++++++++++++++++-
 1 file changed, 79 insertions(+), 1 deletion(-)

diff --git a/test/mitmproxy/addons/test_maplocal.py b/test/mitmproxy/addons/test_maplocal.py
index f0e530524..5ecdc0031 100644
--- a/test/mitmproxy/addons/test_maplocal.py
+++ b/test/mitmproxy/addons/test_maplocal.py
@@ -1,4 +1,3 @@
-import pytest
 from pathlib import Path
 
 from mitmproxy.addons.maplocal import MapLocal
@@ -32,3 +31,82 @@ class TestMapLocal:
         spec = parse_modify_spec(":org:" + str(tmpdir), True, True)
         file_candidates = ml.file_candidates(url, spec)
         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"