aboutsummaryrefslogtreecommitdiffstats
path: root/test/test_platform_pf.py
blob: d8718c8fd6d55fbfe93fdd3c66c5a67eccd7ecba (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 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)