Add DNAT test case.

Author: ldx

iptc/test/test_targets.py

@@ -241,6 +241,54 @@ def test_insert(self):
                 self.fail("inserted rule does not match original")
 
 
+class TestDnatTarget(unittest.TestCase):
+    def setUp(self):
+        self.rule = iptc.Rule()
+        self.rule.dst = "127.0.0.2"
+        self.rule.protocol = "tcp"
+        self.rule.in_interface = "eth0"
+
+        self.match = iptc.Match(self.rule, "tcp")
+        self.rule.add_match(self.match)
+
+        self.target = iptc.Target(self.rule, "DNAT")
+        self.rule.target = self.target
+
+        self.chain = iptc.Chain(iptc.Table(iptc.Table.MANGLE),
+                                "iptc_test_dnat")
+        iptc.Table(iptc.Table.MANGLE).create_chain(self.chain)
+
+    def tearDown(self):
+        self.chain.flush()
+        self.chain.delete()
+
+    def test_mode(self):
+        for dst in ["1.2.3.4", "199.199.199.199-199.199.199.255",
+                    "1.2.3.4:5678", "1.2.3.4:5678-5688"]:
+            self.target.to_destination = dst
+            self.assertEquals(self.target.to_destination, dst)
+            self.target.reset()
+            self.target.to_destination = dst
+            self.target.random = "1"
+            self.assertEquals(self.target.to_destination, dst)
+            self.target.reset()
+            self.target.to_destination = dst
+            self.target.persistent = "1"
+            self.assertEquals(self.target.to_destination, dst)
+            self.target.reset()
+
+    def test_insert(self):
+        self.target.reset()
+        self.target.to_destination = "1.2.3.4"
+        self.rule.target = self.target
+
+        self.chain.insert_rule(self.rule)
+
+        for r in self.chain.rules:
+            if r != self.rule:
+                self.fail("inserted rule does not match original")
+
+
 class TestIPTMasqueradeTarget(unittest.TestCase):
     def setUp(self):
         self.rule = iptc.Rule()
@@ -302,8 +350,10 @@ def suite():
             TestIPTRedirectTarget)
         suite_masq = unittest.TestLoader().loadTestsFromTestCase(
             TestIPTMasqueradeTarget)
+        suite_dnat = unittest.TestLoader().loadTestsFromTestCase(
+            TestDnatTarget)
         return unittest.TestSuite([suite_target, suite_cluster, suite_redir,
-                                   suite_tos, suite_masq])
+                                   suite_tos, suite_masq, suite_dnat])
     else:
         return unittest.TestSuite([suite_target, suite_cluster, suite_tos])