aboutsummaryrefslogtreecommitdiffstats
path: root/mitmproxy/test/test_platform_pf.py
blob: 8994ee0dc7a3ef3ad9ef7729d72de0dc7059eb59 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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)