diff options
| author | Jenkins <jenkins@review.openstack.org> | 2013-03-11 21:13:31 +0000 |
|---|---|---|
| committer | Gerrit Code Review <review@openstack.org> | 2013-03-11 21:13:31 +0000 |
| commit | 4c5fb5657a38c9206bd91c1253b1a2456c1cd37f (patch) | |
| tree | ebd1ffc641962769114ae5d4e1ce2e6c46cbdf04 /nova/tests | |
| parent | 554ecca217916949753f693891c344f4c6f640c9 (diff) | |
| parent | 4ed651f918f1e3bd4a130b5ba560a4da6b1db436 (diff) | |
Merge "clear up method parameters for _modify_rules"
Diffstat (limited to 'nova/tests')
| -rw-r--r-- | nova/tests/test_iptables_network.py | 27 |
1 files changed, 16 insertions, 11 deletions
diff --git a/nova/tests/test_iptables_network.py b/nova/tests/test_iptables_network.py index e64dcf66b..0aaae7df0 100644 --- a/nova/tests/test_iptables_network.py +++ b/nova/tests/test_iptables_network.py @@ -93,12 +93,12 @@ class IptablesManagerTestCase(test.TestCase): table = self.manager.ipv4['filter'] table.add_rule('FORWARD', '-s 1.2.3.4/5 -j DROP') - new_lines = self.manager._modify_rules(current_lines, table) + new_lines = self.manager._modify_rules(current_lines, table, 'filter') self.assertTrue('[0:0] -A %s-FORWARD ' '-s 1.2.3.4/5 -j DROP' % self.binary_name in new_lines) table.remove_rule('FORWARD', '-s 1.2.3.4/5 -j DROP') - new_lines = self.manager._modify_rules(current_lines, table) + new_lines = self.manager._modify_rules(current_lines, table, 'filter') self.assertTrue('[0:0] -A %s-FORWARD ' '-s 1.2.3.4/5 -j DROP' % self.binary_name not in new_lines) @@ -118,22 +118,23 @@ class IptablesManagerTestCase(test.TestCase): ' -o eth0') table.add_rule('PREROUTING', '-d 10.10.10.11 -j DNAT --to 10.0.0.10') table.add_rule('OUTPUT', '-d 10.10.10.11 -j DNAT --to 10.0.0.10') - new_lines = self.manager._modify_rules(current_lines, table) + new_lines = self.manager._modify_rules(current_lines, table, 'nat') self.assertEqual(len(new_lines) - len(current_lines), 8) regex = '.*\s+%s(/32|\s+|$)' num_removed = table.remove_rules_regex(regex % '10.10.10.10') self.assertEqual(num_removed, 4) - new_lines = self.manager._modify_rules(current_lines, table) + new_lines = self.manager._modify_rules(current_lines, table, 'nat') self.assertEqual(len(new_lines) - len(current_lines), 4) num_removed = table.remove_rules_regex(regex % '10.10.10.11') self.assertEqual(num_removed, 4) - new_lines = self.manager._modify_rules(current_lines, table) + new_lines = self.manager._modify_rules(current_lines, table, 'nat') self.assertEqual(new_lines, current_lines) def test_nat_rules(self): current_lines = self.sample_nat new_lines = self.manager._modify_rules(current_lines, - self.manager.ipv4['nat']) + self.manager.ipv4['nat'], + 'nat') for line in [':%s-OUTPUT - [0:0]' % (self.binary_name), ':%s-float-snat - [0:0]' % (self.binary_name), @@ -168,7 +169,8 @@ class IptablesManagerTestCase(test.TestCase): def test_filter_rules(self): current_lines = self.sample_filter new_lines = self.manager._modify_rules(current_lines, - self.manager.ipv4['filter']) + self.manager.ipv4['filter'], + 'nat') for line in [':%s-FORWARD - [0:0]' % (self.binary_name), ':%s-INPUT - [0:0]' % (self.binary_name), @@ -205,7 +207,7 @@ class IptablesManagerTestCase(test.TestCase): current_lines = [] new_lines = self.manager._modify_rules(current_lines, self.manager.ipv4['filter'], - table_name='filter') + 'filter') for line in ['*filter', 'COMMIT']: @@ -226,7 +228,8 @@ class IptablesManagerTestCase(test.TestCase): current_lines[12:12] = ['[0:0] -A FORWARD -j iptables-top-rule'] self.flags(iptables_top_regex='-j iptables-top-rule') new_lines = self.manager._modify_rules(current_lines, - self.manager.ipv4['filter']) + self.manager.ipv4['filter'], + 'filter') self.assertEqual(current_lines, new_lines) def test_iptables_bottom_order(self): @@ -235,7 +238,8 @@ class IptablesManagerTestCase(test.TestCase): current_lines[26:26] = ['[0:0] -A FORWARD -j iptables-bottom-rule'] self.flags(iptables_bottom_regex='-j iptables-bottom-rule') new_lines = self.manager._modify_rules(current_lines, - self.manager.ipv4['filter']) + self.manager.ipv4['filter'], + 'filter') self.assertEqual(current_lines, new_lines) def test_iptables_preserve_order(self): @@ -246,5 +250,6 @@ class IptablesManagerTestCase(test.TestCase): self.flags(iptables_top_regex='-j iptables-top-rule') self.flags(iptables_bottom_regex='-j iptables-bottom-rule') new_lines = self.manager._modify_rules(current_lines, - self.manager.ipv4['filter']) + self.manager.ipv4['filter'], + 'filter') self.assertEqual(current_lines, new_lines) |