def get_vapp_network_firewall(ctx, vapp_name, network_name):
"""Get the VappFirewall object.
It will restore sessions if expired. It will reads the client and
creates the VappFirewall object.
"""
restore_session(ctx, vdc_required=True)
client = ctx.obj['client']
vapp_dhcp = VappFirewall(client, vapp_name, network_name)
return vapp_dhcp
def test_0040_list_firewall_rule(self):
vapp_firewall = VappFirewall(
TestVappFirewall._client,
vapp_name=TestVappFirewall._vapp_name,
network_name=TestVappFirewall._network_name)
result = vapp_firewall.list_firewall_rule()
self.assertNotEqual(len(result), 0)
self.assertTrue(
any(d['Name'] == TestVappFirewall._test_firewall_rule_name
for d in result))
def test_0060_delete_firewall_rule(self):
vapp_firewall = VappFirewall(
TestVappFirewall._client,
vapp_name=TestVappFirewall._vapp_name,
network_name=TestVappFirewall._network_name)
task = vapp_firewall.delete_firewall_rule(
TestVappFirewall._test_firewall_rule_name)
result = TestVappFirewall._client.get_task_monitor().wait_for_success(
task=task)
self.assertEqual(result.get('status'), TaskStatus.SUCCESS.value)
vapp_firewall._reload()
result = vapp_firewall.list_firewall_rule()
self.assertFalse(
any(d['Name'] == TestVappFirewall._test_firewall_rule_name
for d in result))
def test_0030_add_firewall_rule(self):
vapp_firewall = VappFirewall(
TestVappFirewall._client,
vapp_name=TestVappFirewall._vapp_name,
network_name=TestVappFirewall._network_name)
task = vapp_firewall.add_firewall_rule(
name=TestVappFirewall._test_firewall_rule_name)
result = TestVappFirewall._client.get_task_monitor().wait_for_success(
task=task)
self.assertEqual(result.get('status'), TaskStatus.SUCCESS.value)
vapp_firewall._reload()
firewall_service = \
vapp_firewall.resource.Configuration.Features.FirewallService
self.assertTrue(hasattr(firewall_service, 'FirewallRule'))
self.assertEqual(TestVappFirewall._test_firewall_rule_name,
firewall_service.FirewallRule[1].Description)
def test_0020_set_default_action(self):
vapp_firewall = VappFirewall(
TestVappFirewall._client,
vapp_name=TestVappFirewall._vapp_name,
network_name=TestVappFirewall._network_name)
# set allow as default action and True log of firewall service
task = vapp_firewall.set_default_action(TestVappFirewall._allow,
TestVappFirewall._enable)
result = TestVappFirewall._client.get_task_monitor().wait_for_success(
task=task)
self.assertEqual(result.get('status'), TaskStatus.SUCCESS.value)
vapp_firewall._reload()
firewall_service = \
vapp_firewall.resource.Configuration.Features.FirewallService
self.assertEqual(TestVappFirewall._allow,
firewall_service.DefaultAction)
self.assertTrue(firewall_service.LogDefaultAction)
def test_0012_enable_firewall_service(self):
vapp_firewall = VappFirewall(
TestVappFirewall._client,
vapp_name=TestVappFirewall._vapp_name,
network_name=TestVappFirewall._network_name)
# disable firewall service
task = vapp_firewall.enable_firewall_service(TestVappFirewall._disable)
TestVappFirewall._client.get_task_monitor().wait_for_success(task=task)
vapp_firewall._reload()
firewall_service = \
vapp_firewall.resource.Configuration.Features.FirewallService
self.assertFalse(firewall_service.IsEnabled)
# enable firewall service
task = vapp_firewall.enable_firewall_service(TestVappFirewall._enable)
TestVappFirewall._client.get_task_monitor().wait_for_success(task=task)
vapp_firewall._reload()
firewall_service = \
vapp_firewall.resource.Configuration.Features.FirewallService
self.assertTrue(firewall_service.IsEnabled)