aboutsummaryrefslogtreecommitdiffstats
path: root/mitmproxy/test/test_platform_pf.py
diff options
context:
space:
mode:
Diffstat (limited to 'mitmproxy/test/test_platform_pf.py')
-rw-r--r--mitmproxy/test/test_platform_pf.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/mitmproxy/test/test_platform_pf.py b/mitmproxy/test/test_platform_pf.py
new file mode 100644
index 00000000..8994ee0d
--- /dev/null
+++ b/mitmproxy/test/test_platform_pf.py
@@ -0,0 +1,27 @@
+import sys
+from libmproxy.platform import pf
+from . import tutils
+
+
+class TestLookup:
+
+ def test_simple(self):
+ if sys.platform == "freebsd10":
+ p = tutils.test_data.path("data/pf02")
+ d = open(p, "rb").read()
+ else:
+ p = tutils.test_data.path("data/pf01")
+ d = open(p, "rb").read()
+ assert pf.lookup("192.168.1.111", 40000, d) == ("5.5.5.5", 80)
+ tutils.raises(
+ "Could not resolve original destination",
+ pf.lookup,
+ "192.168.1.112",
+ 40000,
+ d)
+ tutils.raises(
+ "Could not resolve original destination",
+ pf.lookup,
+ "192.168.1.111",
+ 40001,
+ d)