def check_rule_operation(self, rule_type, rules, vms_networks=None): if not vms_networks: vms_networks = [] fake_client = self.generate_client(vms_networks=vms_networks) fake_ctx = self.generate_context_for_security_group() fake_ctx._target.node.properties = {"rules": rules} # any calls for save configuration will be success gateway = fake_client._vdc_gateway self.set_services_conf_result(gateway, vcloud_plugin_common.TASK_STATUS_SUCCESS) # for check calls for add/delete rule gateway.add_fw_rule = mock.MagicMock(return_value=None) gateway.delete_fw_rule = mock.MagicMock(return_value=None) # any networks will be routed self.set_network_routed_in_client(fake_client) with mock.patch("vcloud_network_plugin.security_group.ctx", fake_ctx): with mock.patch("vcloud_plugin_common.ctx", fake_ctx): security_group._rule_operation(rule_type, fake_client) return gateway
def check_rule_operation_fail(self, rule_type, rules): fake_client = self.generate_client() fake_ctx = self.generate_context_for_security_group() fake_ctx._target.node.properties = {"rules": rules} # check busy gateway = fake_client._vdc_gateway self.set_gateway_busy(gateway) self.set_services_conf_result(fake_client._vdc_gateway, None) with mock.patch("vcloud_network_plugin.security_group.ctx", fake_ctx): with mock.patch("vcloud_plugin_common.ctx", fake_ctx): self.assertFalse(security_group._rule_operation(rule_type, fake_client))
def check_rule_operation(self, rule_type, rules, vms_networks=None): if not vms_networks: vms_networks = [] fake_client = self.generate_client(vms_networks=vms_networks) fake_ctx = self.generate_context_for_security_group() fake_ctx._target.node.properties = { 'rules': rules } # any calls for save configuration will be success gateway = fake_client._vdc_gateway self.set_services_conf_result( gateway, vcloud_plugin_common.TASK_STATUS_SUCCESS ) # for check calls for add/delete rule gateway.add_fw_rule = mock.MagicMock(return_value=None) gateway.delete_fw_rule = mock.MagicMock(return_value=None) # any networks will be routed self.set_network_routed_in_client(fake_client) with mock.patch('vcloud_network_plugin.security_group.ctx', fake_ctx): with mock.patch('vcloud_plugin_common.ctx', fake_ctx): security_group._rule_operation( rule_type, fake_client ) return gateway
def check_rule_operation_fail(self, rule_type, rules): fake_client = self.generate_client() fake_ctx = self.generate_context_for_security_group() fake_ctx._target.node.properties = { 'rules': rules } # check busy gateway = fake_client._vdc_gateway self.set_gateway_busy(gateway) self.set_services_conf_result( fake_client._vdc_gateway, None ) with mock.patch('vcloud_network_plugin.security_group.ctx', fake_ctx): with mock.patch('vcloud_plugin_common.ctx', fake_ctx): self.assertFalse(security_group._rule_operation( rule_type, fake_client ))