def test_get_zone_cmds(self): firewall_name = 'fake_firewall0' eth_iface = 'eth0' fake_api = mock.NonCallableMock() fake_api.get_ethernet_if_id.return_value = eth_iface mac_address = '00:00:00:00:00:00' fake_apply_rule = mock.NonCallableMock() fake_apply_rule.router = { 'gw_port': { 'mac_address': mac_address }, l3_constants.INTERFACE_KEY: [{ 'mac_address': mac_address }] } trusted_zone_name = vyatta_utils.get_trusted_zone_name(fake_apply_rule) untrusted_zone_name = vyatta_utils.get_untrusted_zone_name( fake_apply_rule) cmds_actual = vyatta_utils.get_zone_cmds(fake_api, fake_apply_rule, firewall_name) cmds_expect = [ vyatta_client.DeleteCmd('zone-policy'), vyatta_client.SetCmd( vyatta_utils.ZONE_INTERFACE_CMD.format(trusted_zone_name, eth_iface)), vyatta_client.SetCmd( vyatta_utils.ZONE_INTERFACE_CMD.format(untrusted_zone_name, eth_iface)), vyatta_client.SetCmd( vyatta_utils.ZONE_FIREWALL_CMD.format( trusted_zone_name, untrusted_zone_name, parse.quote_plus(firewall_name))), vyatta_client.SetCmd( vyatta_utils.ZONE_FIREWALL_CMD.format( untrusted_zone_name, trusted_zone_name, parse.quote_plus(firewall_name))), ] self.assertEqual(cmds_expect, cmds_actual) fake_api.get_ethernet_if_id.assert_has_calls([ mock.call(mac_address), mock.call(mac_address), ])
def test_get_zone_cmds(self): firewall_name = 'fake_firewall0' eth_iface = 'eth0' fake_api = mock.NonCallableMock() fake_api.get_ethernet_if_id.return_value = eth_iface mac_address = '00:00:00:00:00:00' fake_apply_rule = mock.NonCallableMock() fake_apply_rule.router = { 'gw_port': { 'mac_address': mac_address}, l3_constants.INTERFACE_KEY: [{ 'mac_address': mac_address}] } trusted_zone_name = vyatta_utils.get_trusted_zone_name( fake_apply_rule) untrusted_zone_name = vyatta_utils.get_untrusted_zone_name( fake_apply_rule) cmds_actual = vyatta_utils.get_zone_cmds( fake_api, fake_apply_rule, firewall_name) cmds_expect = [ vyatta_client.DeleteCmd('zone-policy'), vyatta_client.SetCmd( vyatta_utils.ZONE_INTERFACE_CMD.format( trusted_zone_name, eth_iface)), vyatta_client.SetCmd( vyatta_utils.ZONE_INTERFACE_CMD.format( untrusted_zone_name, eth_iface)), vyatta_client.SetCmd( vyatta_utils.ZONE_FIREWALL_CMD.format( trusted_zone_name, untrusted_zone_name, parse.quote_plus(firewall_name))), vyatta_client.SetCmd( vyatta_utils.ZONE_FIREWALL_CMD.format( untrusted_zone_name, trusted_zone_name, parse.quote_plus(firewall_name))), ] self.assertEqual(cmds_expect, cmds_actual) fake_api.get_ethernet_if_id.assert_has_calls([ mock.call(mac_address), mock.call(mac_address), ])
def test_get_untrusted_zone_name(self): fake_apply_list = object() self.assertEqual( 'External_Untrust', vyatta_utils.get_untrusted_zone_name( fake_apply_list))
def test_get_untrusted_zone_name(self): fake_apply_list = object() self.assertEqual('External_Untrust', vyatta_utils.get_untrusted_zone_name(fake_apply_list))