def _test_add_snat_rule(self, expected): def verify_add_rule(chain, rule): self.assertEqual(chain, 'snat') self.assertEqual(rule, expected) self.stubs.Set(linux_net.iptables_manager.ipv4['nat'], 'add_rule', verify_add_rule) linux_net.add_snat_rule('10.0.0.0/24')
def _test_add_snat_rule(self, expected, is_external): def verify_add_rule(chain, rule): self.assertEqual(chain, "snat") self.assertEqual(rule, expected) self.called = True self.stubs.Set(linux_net.iptables_manager.ipv4["nat"], "add_rule", verify_add_rule) self.called = False linux_net.add_snat_rule("10.0.0.0/24", is_external) if expected: self.assertTrue(self.called)
def _test_add_snat_rule(self, expected, is_external): def verify_add_rule(chain, rule): self.assertEqual(chain, 'snat') self.assertEqual(rule, expected) self.called = True self.stubs.Set(linux_net.iptables_manager.ipv4['nat'], 'add_rule', verify_add_rule) self.called = False linux_net.add_snat_rule('10.0.0.0/24', is_external) if expected: self.assertTrue(self.called)
def initialize_network(self, cidr): linux_net.add_snat_rule(cidr)