242
242
self.stubs.Set(db, 'instance_get', get_instance)
243
243
self.stubs.Set(db, 'network_get_associated_fixed_ips', get_associated)
245
def _test_add_snat_rule(self, expected):
246
def verify_add_rule(chain, rule):
247
self.assertEqual(chain, 'snat')
248
self.assertEqual(rule, expected)
250
self.stubs.Set(linux_net.iptables_manager.ipv4['nat'],
251
'add_rule', verify_add_rule)
252
linux_net.add_snat_rule('10.0.0.0/24')
254
def test_add_snat_rule(self):
255
self.flags(routing_source_ip='10.10.10.1')
256
expected = ('-s 10.0.0.0/24 -d 0.0.0.0/0 '
257
'-j SNAT --to-source 10.10.10.1 -o eth0')
258
self._test_add_snat_rule(expected)
260
def test_add_snat_rule_snat_range(self):
261
self.flags(routing_source_ip='10.10.10.1',
262
force_snat_range=['10.10.10.0/24'])
263
expected = ('-s 10.0.0.0/24 -d 10.10.10.0/24 '
264
'-j SNAT --to-source 10.10.10.1 -o eth0')
265
self._test_add_snat_rule(expected)
245
267
def test_update_dhcp_for_nw00(self):
246
268
self.flags(use_single_default_gateway=True)