def test_rules():
redir = Redirector()
redir.localaddr = "1.2.3.4"
assert "-j DNAT -p tcp -m multiport --destination-ports 123 --to-destination 1.2.3.4:555" == \
redir._create_nat_rule("tcp", [123], 555)
assert "-j DNAT -p tcp -m multiport --destination-ports 123,456 --to-destination 1.2.3.4:555" == \
redir._create_nat_rule("tcp", [123, 456], 555)
assert "-j DNAT -p tcp -m multiport --destination-ports 123,456 --to-destination 1.2.3.4" == \
redir._create_nat_rule("tcp", [123, 456], None)
assert "-j DNAT -p tcp --to-destination 1.2.3.4:555" == redir._create_nat_rule("tcp", [], outport=555)
assert "-j DNAT --to-destination 1.2.3.4" == redir._create_nat_rule(None, [], None)
def test_rules():
redir = Redirector()
redir.localaddr = "1.2.3.4"
assert "-j DNAT -p tcp -m multiport --destination-ports 123 --to-destination 1.2.3.4:555" == \
redir._create_nat_rule("tcp", [123], 555)
assert "-j DNAT -p tcp -m multiport --destination-ports 123,456 --to-destination 1.2.3.4:555" == \
redir._create_nat_rule("tcp", [123, 456], 555)
assert "-j DNAT -p tcp -m multiport --destination-ports 123,456 --to-destination 1.2.3.4" == \
redir._create_nat_rule("tcp", [123, 456], None)
assert "-j DNAT -p tcp --to-destination 1.2.3.4:555" == redir._create_nat_rule(
"tcp", [], outport=555)
assert "-j DNAT --to-destination 1.2.3.4" == redir._create_nat_rule(
None, [], None)
