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