164
164
self.assertTrue('-A %s -j run_tests.py-%s' \
165
165
% (chain, chain) in new_lines,
166
166
"Built-in chain %s not wrapped" % (chain,))
168
def test_will_empty_chain(self):
169
self.manager.ipv4['filter'].add_chain('test-chain')
170
self.manager.ipv4['filter'].add_rule('test-chain', '-j DROP')
171
old_count = len(self.manager.ipv4['filter'].rules)
172
self.manager.ipv4['filter'].empty_chain('test-chain')
173
self.assertEqual(old_count - 1, len(self.manager.ipv4['filter'].rules))
175
def test_will_empty_unwrapped_chain(self):
176
self.manager.ipv4['filter'].add_chain('test-chain', wrap=False)
177
self.manager.ipv4['filter'].add_rule('test-chain', '-j DROP',
179
old_count = len(self.manager.ipv4['filter'].rules)
180
self.manager.ipv4['filter'].empty_chain('test-chain', wrap=False)
181
self.assertEqual(old_count - 1, len(self.manager.ipv4['filter'].rules))
183
def test_will_not_empty_wrapped_when_unwrapped(self):
184
self.manager.ipv4['filter'].add_chain('test-chain')
185
self.manager.ipv4['filter'].add_rule('test-chain', '-j DROP')
186
old_count = len(self.manager.ipv4['filter'].rules)
187
self.manager.ipv4['filter'].empty_chain('test-chain', wrap=False)
188
self.assertEqual(old_count, len(self.manager.ipv4['filter'].rules))
190
def test_will_not_empty_unwrapped_when_wrapped(self):
191
self.manager.ipv4['filter'].add_chain('test-chain', wrap=False)
192
self.manager.ipv4['filter'].add_rule('test-chain', '-j DROP',
194
old_count = len(self.manager.ipv4['filter'].rules)
195
self.manager.ipv4['filter'].empty_chain('test-chain')
196
self.assertEqual(old_count, len(self.manager.ipv4['filter'].rules))