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)