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)