Add tests for alias and real_name.

Author: ldx

iptc/test/test_matches.py

@@ -264,6 +264,40 @@ def test_iprange_tcpdport(self):
                 self.fail("inserted rule does not match original")
 
 
+class TestXTStateMatch(unittest.TestCase):
+    def setUp(self):
+        self.rule = iptc.Rule()
+        self.rule.src = "127.0.0.1"
+        self.rule.protocol = "tcp"
+        self.rule.target = iptc.Target(self.rule, "ACCEPT")
+
+        self.match = iptc.Match(self.rule, "state")
+
+        self.chain = iptc.Chain(iptc.Table(iptc.Table.FILTER),
+                                "iptc_test_state")
+        self.table = iptc.Table(iptc.Table.FILTER)
+        try:
+            self.chain.flush()
+            self.chain.delete()
+        except:
+            pass
+        self.table.create_chain(self.chain)
+
+    def tearDown(self):
+        self.chain.flush()
+        self.chain.delete()
+        pass
+
+    def test_state(self):
+        self.match.state = "RELATED,ESTABLISHED"
+        self.rule.add_match(self.match)
+        self.chain.insert_rule(self.rule)
+        rule = self.chain.rules[0]
+        m = rule.matches[0]
+        self.assertIn(m.name, ["state", "conntrack"])
+        self.assertEquals(m.state, "RELATED,ESTABLISHED")
+
+
 def suite():
     suite_match = unittest.TestLoader().loadTestsFromTestCase(TestMatch)
     suite_udp = unittest.TestLoader().loadTestsFromTestCase(TestXTUdpMatch)
@@ -273,8 +307,10 @@ def suite():
         TestCommentMatch)
     suite_iprange = unittest.TestLoader().loadTestsFromTestCase(
         TestIprangeMatch)
+    suite_state = unittest.TestLoader().loadTestsFromTestCase(TestXTStateMatch)
     return unittest.TestSuite([suite_match, suite_udp, suite_mark,
-                               suite_limit, suite_comment, suite_iprange])
+                               suite_limit, suite_comment, suite_iprange,
+                               suite_state])
 
 
 def run_tests():

iptc/test/test_targets.py

@@ -342,22 +342,56 @@ def test_insert(self):
             self.fail("inserted rule does not match original")
 
 
+class TestXTNotrackTarget(unittest.TestCase):
+    def setUp(self):
+        self.rule = iptc.Rule()
+        self.rule.dst = "127.0.0.2"
+        self.rule.protocol = "tcp"
+        self.rule.out_interface = "eth0"
+
+        self.target = iptc.Target(self.rule, "NOTRACK")
+        self.rule.target = self.target
+
+        self.chain = iptc.Chain(iptc.Table(iptc.Table.RAW),
+                                "iptc_test_notrack")
+        try:
+            self.chain.flush()
+            self.chain.delete()
+        except:
+            pass
+        iptc.Table(iptc.Table.RAW).create_chain(self.chain)
+
+    def tearDown(self):
+        self.chain.flush()
+        self.chain.delete()
+
+    def test_notrack(self):
+        self.chain.insert_rule(self.rule)
+        t = self.chain.rules[0].target
+        self.assertIn(t.name, ["NOTRACK", "CT"])
+
+
 def suite():
+    suites = []
     suite_target = unittest.TestLoader().loadTestsFromTestCase(TestTarget)
     suite_tos = unittest.TestLoader().loadTestsFromTestCase(TestXTTosTarget)
     suite_cluster = unittest.TestLoader().loadTestsFromTestCase(
         TestXTClusteripTarget)
+    suite_redir = unittest.TestLoader().loadTestsFromTestCase(
+        TestIPTRedirectTarget)
+    suite_masq = unittest.TestLoader().loadTestsFromTestCase(
+        TestIPTMasqueradeTarget)
+    suite_dnat = unittest.TestLoader().loadTestsFromTestCase(
+        TestDnatTarget)
+    suite_conntrack = unittest.TestLoader().loadTestsFromTestCase(
+        TestXTNotrackTarget)
+    suites.extend([suite_target, suite_cluster, suite_tos])
     if is_table_available(iptc.Table.NAT):
-        suite_redir = unittest.TestLoader().loadTestsFromTestCase(
-            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_dnat])
-    else:
-        return unittest.TestSuite([suite_target, suite_cluster, suite_tos])
+        suites.extend([suite_target, suite_cluster, suite_redir, suite_tos,
+                       suite_masq, suite_dnat])
+    if is_table_available(iptc.Table.RAW):
+        suites.extend([suite_conntrack])
+    return unittest.TestSuite(suites)
 
 
 def run_tests():