aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_platform_pf.py
blob: 3250b035be1e3e3233f370f3a31d71adb1efbfdb (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
import tutils
import sys
from libmproxy.platform import pf


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)