# Licensed under the Apache License, Version 2.0 (the "License"); you may # not use this file except in compliance with the License. You may obtain # a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the # License for the specific language governing permissions and limitations # under the License. """Unit tests for ComputeManager().""" import contextlib import copy import datetime import fixtures as std_fixtures import time from unittest import mock from cinderclient import exceptions as cinder_exception from cursive import exception as cursive_exception import ddt from eventlet import event as eventlet_event from eventlet import timeout as eventlet_timeout from keystoneauth1 import exceptions as keystone_exception import netaddr from oslo_log import log as logging import oslo_messaging as messaging from oslo_serialization import jsonutils from oslo_service import fixture as service_fixture from oslo_utils.fixture import uuidsentinel as uuids from oslo_utils import timeutils from oslo_utils import uuidutils import testtools import nova from nova.compute import build_results from nova.compute import manager from nova.compute import power_state from nova.compute import resource_tracker from nova.compute import task_states from nova.compute import utils as compute_utils from nova.compute import vm_states from nova.conductor import api as conductor_api import nova.conf from nova import context from nova.db.main import api as db from nova import exception from nova.network import model as network_model from nova.network import neutron as neutronv2_api from nova import objects from nova.objects import base as base_obj from nova.objects import block_device as block_device_obj from nova.objects import fields from nova.objects import instance as instance_obj from nova.objects import migrate_data as migrate_data_obj from nova.objects import network_request as net_req_obj from nova.objects import service as service_obj from nova.pci import request as pci_request from nova.scheduler.client import report from nova import test from nova.tests import fixtures from nova.tests.unit.api.openstack import fakes from nova.tests.unit.compute import fake_resource_tracker from nova.tests.unit import fake_block_device from nova.tests.unit import fake_flavor from nova.tests.unit import fake_instance from nova.tests.unit import fake_network from nova.tests.unit import fake_network_cache_model from nova.tests.unit.objects import test_instance_fault from nova.tests.unit.objects import test_instance_info_cache from nova.tests.unit.objects import test_instance_numa from nova.virt.block_device import DriverVolumeBlockDevice as driver_bdm_volume from nova.virt import driver as virt_driver from nova.virt import event as virtevent from nova.virt import fake as fake_driver from nova.virt import hardware from nova.virt import node as virt_node from nova.volume import cinder CONF = nova.conf.CONF fake_host_list = [mock.sentinel.host1] @ddt.ddt class ComputeManagerUnitTestCase(test.NoDBTestCase, fake_resource_tracker.RTMockMixin): # os-brick>=5.1 now uses external file system locks instead of internal # locks so we need to set up locking REQUIRES_LOCKING = True STUB_COMPUTE_ID = False def setUp(self): super(ComputeManagerUnitTestCase, self).setUp() self.compute = manager.ComputeManager() self.context = context.RequestContext(fakes.FAKE_USER_ID, fakes.FAKE_PROJECT_ID) self.useFixture(fixtures.SpawnIsSynchronousFixture()) self.useFixture(fixtures.EventReporterStub()) self.allocations = { uuids.provider1: { "generation": 0, "resources": { "VCPU": 1, "MEMORY_MB": 512 } } } @mock.patch.object(manager.ComputeManager, '_get_power_state') @mock.patch.object(manager.ComputeManager, '_sync_instance_power_state') @mock.patch.object(objects.Instance, 'get_by_uuid') @mock.patch.object(objects.Migration, 'get_by_instance_and_status') @mock.patch.object(neutronv2_api.API, 'migrate_instance_start') def _test_handle_lifecycle_event(self, migrate_instance_start, mock_get_migration, mock_get, mock_sync, mock_get_power_state, transition, event_pwr_state, current_pwr_state): event = mock.Mock() mock_get.return_value = fake_instance.fake_instance_obj(self.context, task_state=task_states.MIGRATING) event.get_transition.return_value = transition mock_get_power_state.return_value = current_pwr_state self.compute.handle_lifecycle_event(event) expected_attrs = [] if transition in [virtevent.EVENT_LIFECYCLE_POSTCOPY_STARTED, virtevent.EVENT_LIFECYCLE_MIGRATION_COMPLETED]: expected_attrs.append('info_cache') mock_get.assert_called_once_with( test.MatchType(context.RequestContext), event.get_instance_uuid.return_value, expected_attrs=expected_attrs) if event_pwr_state == current_pwr_state: mock_sync.assert_called_with(mock.ANY, mock_get.return_value, event_pwr_state) else: self.assertFalse(mock_sync.called) migrate_finish_statuses = { virtevent.EVENT_LIFECYCLE_POSTCOPY_STARTED: 'running (post-copy)', virtevent.EVENT_LIFECYCLE_MIGRATION_COMPLETED: 'running' } if transition in migrate_finish_statuses: mock_get_migration.assert_called_with( test.MatchType(context.RequestContext), mock_get.return_value.uuid, migrate_finish_statuses[transition]) migrate_instance_start.assert_called_once_with( test.MatchType(context.RequestContext), mock_get.return_value, mock_get_migration.return_value) else: mock_get_migration.assert_not_called() migrate_instance_start.assert_not_called() def test_handle_lifecycle_event(self): event_map = {virtevent.EVENT_LIFECYCLE_STOPPED: power_state.SHUTDOWN, virtevent.EVENT_LIFECYCLE_STARTED: power_state.RUNNING, virtevent.EVENT_LIFECYCLE_PAUSED: power_state.PAUSED, virtevent.EVENT_LIFECYCLE_RESUMED: power_state.RUNNING, virtevent.EVENT_LIFECYCLE_SUSPENDED: power_state.SUSPENDED, virtevent.EVENT_LIFECYCLE_POSTCOPY_STARTED: power_state.PAUSED, virtevent.EVENT_LIFECYCLE_MIGRATION_COMPLETED: power_state.PAUSED, } for transition, pwr_state in event_map.items(): self._test_handle_lifecycle_event(transition=transition, event_pwr_state=pwr_state, current_pwr_state=pwr_state) def test_handle_lifecycle_event_state_mismatch(self): self._test_handle_lifecycle_event( transition=virtevent.EVENT_LIFECYCLE_STOPPED, event_pwr_state=power_state.SHUTDOWN, current_pwr_state=power_state.RUNNING) @mock.patch('nova.objects.Instance.get_by_uuid') @mock.patch('nova.compute.manager.ComputeManager.' '_sync_instance_power_state') @mock.patch('nova.objects.Migration.get_by_instance_and_status', side_effect=exception.MigrationNotFoundByStatus( instance_id=uuids.instance, status='running (post-copy)')) def test_handle_lifecycle_event_postcopy_migration_not_found( self, mock_get_migration, mock_sync, mock_get_instance): """Tests a EVENT_LIFECYCLE_POSTCOPY_STARTED scenario where the migration record is not found by the expected status. """ inst = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, task_state=task_states.MIGRATING) mock_get_instance.return_value = inst event = virtevent.LifecycleEvent( uuids.instance, virtevent.EVENT_LIFECYCLE_POSTCOPY_STARTED) with mock.patch.object(self.compute, '_get_power_state', return_value=power_state.PAUSED): with mock.patch.object(self.compute.network_api, 'migrate_instance_start') as mig_start: self.compute.handle_lifecycle_event(event) # Since we failed to find the migration record, we shouldn't call # migrate_instance_start. mig_start.assert_not_called() mock_get_migration.assert_called_once_with( test.MatchType(context.RequestContext), uuids.instance, 'running (post-copy)') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_delete_instance_info_cache_delete_ordering(self, mock_notify): call_tracker = mock.Mock() call_tracker.clear_events_for_instance.return_value = None mgr_class = self.compute.__class__ orig_delete = mgr_class._delete_instance specd_compute = mock.create_autospec(mgr_class) # spec out everything except for the method we really want # to test, then use call_tracker to verify call sequence specd_compute._delete_instance = orig_delete specd_compute.host = 'compute' mock_inst = mock.Mock() mock_inst.uuid = uuids.instance mock_inst.save = mock.Mock() mock_inst.destroy = mock.Mock() mock_inst.system_metadata = mock.Mock() def _mark_notify(*args, **kwargs): call_tracker._notify_about_instance_usage(*args, **kwargs) def _mark_shutdown(*args, **kwargs): call_tracker._shutdown_instance(*args, **kwargs) specd_compute.instance_events = call_tracker specd_compute._notify_about_instance_usage = _mark_notify specd_compute._shutdown_instance = _mark_shutdown mock_bdms = mock.Mock() specd_compute._delete_instance(specd_compute, self.context, mock_inst, mock_bdms) methods_called = [n for n, a, k in call_tracker.mock_calls] self.assertEqual(['clear_events_for_instance', '_notify_about_instance_usage', '_shutdown_instance', '_notify_about_instance_usage'], methods_called) mock_notify.assert_has_calls([ mock.call(self.context, mock_inst, specd_compute.host, action='delete', phase='start', bdms=mock_bdms), mock.call(self.context, mock_inst, specd_compute.host, action='delete', phase='end', bdms=mock_bdms)]) @mock.patch.object(objects.Instance, 'destroy') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_complete_deletion') @mock.patch.object(manager.ComputeManager, '_cleanup_volumes') @mock.patch.object(manager.ComputeManager, '_shutdown_instance') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') def _test_delete_instance_with_accels(self, instance, mock_inst_usage, mock_inst_action, mock_shutdown, mock_cleanup_vols, mock_complete_del, mock_inst_save, mock_inst_destroy): self.compute._delete_instance(self.context, instance, bdms=None) @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'delete_arqs_for_instance') def test_delete_instance_with_accels_ok(self, mock_del_arqs): # _delete_instance() calls Cyborg to delete ARQs, if # the extra specs has a device profile name. instance = fake_instance.fake_instance_obj(self.context) instance.flavor.extra_specs = {'accel:device_profile': 'mydp'} self._test_delete_instance_with_accels(instance) mock_del_arqs.assert_called_once_with(instance.uuid) @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'delete_arqs_for_instance') def test_delete_instance_with_accels_no_dp(self, mock_del_arqs): # _delete_instance() does not call Cyborg to delete ARQs, if # the extra specs has no device profile name. instance = fake_instance.fake_instance_obj(self.context) self._test_delete_instance_with_accels(instance) mock_del_arqs.assert_not_called() def _make_compute_node(self, hyp_hostname, cn_id): cn = mock.Mock(spec_set=['hypervisor_hostname', 'id', 'uuid', 'destroy']) cn.id = cn_id cn.hypervisor_hostname = hyp_hostname return cn def test_update_available_resource_for_node(self): rt = self._mock_rt(spec_set=['update_available_resource']) self.compute._update_available_resource_for_node( self.context, mock.sentinel.node, ) rt.update_available_resource.assert_called_once_with( self.context, mock.sentinel.node, startup=False, ) @mock.patch('nova.compute.manager.LOG') def test_update_available_resource_for_node_reshape_failed(self, log_mock): """ReshapeFailed logs and reraises.""" rt = self._mock_rt(spec_set=['update_available_resource']) rt.update_available_resource.side_effect = exception.ReshapeFailed( error='error') self.assertRaises(exception.ReshapeFailed, self.compute._update_available_resource_for_node, self.context, mock.sentinel.node, # While we're here, unit test the startup kwarg startup=True) rt.update_available_resource.assert_called_once_with( self.context, mock.sentinel.node, startup=True) log_mock.critical.assert_called_once() @mock.patch('nova.compute.manager.LOG') def test_update_available_resource_for_node_reshape_needed(self, log_mock): """ReshapeNeeded logs and reraises.""" rt = self._mock_rt(spec_set=['update_available_resource']) rt.update_available_resource.side_effect = exception.ReshapeNeeded() self.assertRaises(exception.ReshapeNeeded, self.compute._update_available_resource_for_node, self.context, mock.sentinel.node, # While we're here, unit test the startup kwarg startup=True) rt.update_available_resource.assert_called_once_with( self.context, mock.sentinel.node, startup=True) log_mock.exception.assert_called_once() def test_update_available_resource_for_node_pci_placement_failed_startup( self ): """If the PCI placement translation failed during startup then the exception is raised up to kill the service """ rt = self._mock_rt(spec_set=['update_available_resource']) rt.update_available_resource.side_effect = ( exception.PlacementPciException(error='error')) self.assertRaises( exception.PlacementPciException, self.compute._update_available_resource_for_node, self.context, mock.sentinel.node, startup=True, ) rt.update_available_resource.assert_called_once_with( self.context, mock.sentinel.node, startup=True) @mock.patch('nova.compute.manager.LOG') def test_update_available_resource_for_node_pci_placement_failed_later( self, mock_log ): """If the PCI placement translation failed later (not at startup) during a periodic then the exception is just logged """ rt = self._mock_rt(spec_set=['update_available_resource']) rt.update_available_resource.side_effect = ( exception.PlacementPciException(error='error')) self.compute._update_available_resource_for_node( self.context, mock.sentinel.node, startup=False) rt.update_available_resource.assert_called_once_with( self.context, mock.sentinel.node, startup=False) mock_log.exception.assert_called_once_with( 'Error updating PCI resources for node %(node)s.', {'node': mock.sentinel.node} ) @mock.patch.object(manager, 'LOG') @mock.patch.object(manager.ComputeManager, '_update_available_resource_for_node') @mock.patch.object(fake_driver.FakeDriver, 'get_available_nodes') @mock.patch.object(manager.ComputeManager, '_get_compute_nodes_in_db') def test_update_available_resource(self, get_db_nodes, get_avail_nodes, update_mock, mock_log): mock_rt = self._mock_rt() rc_mock = self.useFixture(fixtures.fixtures.MockPatchObject( self.compute, 'reportclient')).mock rc_mock.delete_resource_provider.side_effect = ( keystone_exception.EndpointNotFound) db_nodes = [self._make_compute_node('node%s' % i, i) for i in range(1, 5)] avail_nodes = set(['node2', 'node3', 'node4', 'node5']) avail_nodes_l = list(avail_nodes) get_db_nodes.return_value = db_nodes get_avail_nodes.return_value = avail_nodes self.compute.update_available_resource(self.context, startup=True) get_db_nodes.assert_called_once_with(self.context, avail_nodes, use_slave=True, startup=True) self.assertEqual(len(avail_nodes_l), update_mock.call_count) update_mock.assert_has_calls( [mock.call(self.context, node, startup=True) for node in avail_nodes_l] ) # First node in set should have been removed from DB # Last node in set should have been added to DB. for db_node in db_nodes: if db_node.hypervisor_hostname == 'node1': db_node.destroy.assert_called_once_with() rc_mock.delete_resource_provider.assert_called_once_with( self.context, db_node, cascade=True) mock_rt.remove_node.assert_called_once_with('node1') mock_log.error.assert_called_once_with( "Failed to delete compute node resource provider for " "compute node %s: %s", db_node.uuid, mock.ANY) else: self.assertFalse(db_node.destroy.called) self.assertEqual(1, mock_rt.remove_node.call_count) mock_rt.clean_compute_node_cache.assert_called_once_with(db_nodes) @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'delete_resource_provider') @mock.patch.object(manager.ComputeManager, '_update_available_resource_for_node') @mock.patch.object(fake_driver.FakeDriver, 'get_available_nodes') @mock.patch.object(manager.ComputeManager, '_get_compute_nodes_in_db') def test_update_available_resource_not_ready(self, get_db_nodes, get_avail_nodes, update_mock, del_rp_mock): db_nodes = [self._make_compute_node('node1', 1)] get_db_nodes.return_value = db_nodes get_avail_nodes.side_effect = exception.VirtDriverNotReady self.compute.update_available_resource(self.context) # these shouldn't get processed on VirtDriverNotReady update_mock.assert_not_called() del_rp_mock.assert_not_called() @mock.patch.object(manager.ComputeManager, '_update_available_resource_for_node') @mock.patch.object(fake_driver.FakeDriver, 'get_available_nodes') @mock.patch.object(manager.ComputeManager, '_get_compute_nodes_in_db') def test_update_available_resource_destroy_rebalance( self, get_db_nodes, get_avail_nodes, update_mock): mock_rt = self._mock_rt() rc_mock = self.useFixture(fixtures.fixtures.MockPatchObject( self.compute, 'reportclient')).mock db_nodes = [self._make_compute_node('node1', 1)] get_db_nodes.return_value = db_nodes # Destroy can fail if nodes were rebalanced between getting the node # list and calling destroy. db_nodes[0].destroy.side_effect = exception.ObjectActionError( action='destroy', reason='host changed') get_avail_nodes.return_value = set() self.compute.update_available_resource(self.context) get_db_nodes.assert_called_once_with(self.context, set(), use_slave=True, startup=False) self.assertEqual(0, update_mock.call_count) db_nodes[0].destroy.assert_called_once_with() self.assertEqual(0, rc_mock.delete_resource_provider.call_count) mock_rt.remove_node.assert_called_once_with('node1') rc_mock.invalidate_resource_provider.assert_called_once_with( db_nodes[0].uuid) @mock.patch('nova.context.get_admin_context') def test_pre_start_hook(self, get_admin_context): """Very simple test just to make sure update_available_resource is called as expected. """ with mock.patch.object( self.compute, 'update_available_resource') as update_res: self.compute.pre_start_hook() update_res.assert_called_once_with( get_admin_context.return_value, startup=True) @mock.patch.object(objects.ComputeNodeList, 'get_all_by_host', side_effect=exception.NotFound) @mock.patch('nova.compute.manager.LOG') def test_get_compute_nodes_in_db_on_startup(self, mock_log, get_all_by_host): """Tests to make sure we only log a warning when we do not find a compute node on startup since this may be expected. """ self.assertEqual([], self.compute._get_compute_nodes_in_db( self.context, {'fake-node'}, startup=True)) get_all_by_host.assert_called_once_with( self.context, self.compute.host, use_slave=False) self.assertTrue(mock_log.warning.called) self.assertFalse(mock_log.error.called) @mock.patch.object(objects.ComputeNodeList, 'get_all_by_host', side_effect=exception.NotFound) @mock.patch('nova.compute.manager.LOG') def test_get_compute_nodes_in_db_not_found_no_nodenames( self, mock_log, get_all_by_host): """Tests to make sure that _get_compute_nodes_in_db does not log anything when ComputeNodeList.get_all_by_host raises NotFound and the driver did not report any nodenames. """ self.assertEqual([], self.compute._get_compute_nodes_in_db( self.context, set())) get_all_by_host.assert_called_once_with( self.context, self.compute.host, use_slave=False) mock_log.assert_not_called() def _trusted_certs_setup_instance(self, include_trusted_certs=True): instance = fake_instance.fake_instance_obj(self.context) if include_trusted_certs: instance.trusted_certs = objects.trusted_certs.TrustedCerts( ids=['fake-trusted-cert-1', 'fake-trusted-cert-2']) else: instance.trusted_certs = None return instance def test_check_trusted_certs_provided_no_support(self): instance = self._trusted_certs_setup_instance() with mock.patch.dict(self.compute.driver.capabilities, supports_trusted_certs=False): self.assertRaises(exception.BuildAbortException, self.compute._check_trusted_certs, instance) def test_check_trusted_certs_not_provided_no_support(self): instance = self._trusted_certs_setup_instance( include_trusted_certs=False) with mock.patch.dict(self.compute.driver.capabilities, supports_trusted_certs=False): self.compute._check_trusted_certs(instance) def test_check_trusted_certs_provided_support(self): instance = self._trusted_certs_setup_instance() with mock.patch.dict(self.compute.driver.capabilities, supports_trusted_certs=True): self.compute._check_trusted_certs(instance) def test_check_device_tagging_no_tagging(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', instance_uuid=uuids.instance)]) net_req = net_req_obj.NetworkRequest(tag=None) net_req_list = net_req_obj.NetworkRequestList(objects=[net_req]) with mock.patch.dict(self.compute.driver.capabilities, supports_device_tagging=False): self.compute._check_device_tagging(net_req_list, bdms) def test_check_device_tagging_no_networks(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', instance_uuid=uuids.instance)]) with mock.patch.dict(self.compute.driver.capabilities, supports_device_tagging=False): self.compute._check_device_tagging(None, bdms) def test_check_device_tagging_tagged_net_req_no_virt_support(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', instance_uuid=uuids.instance)]) net_req = net_req_obj.NetworkRequest(port_id=uuids.bar, tag='foo') net_req_list = net_req_obj.NetworkRequestList(objects=[net_req]) with mock.patch.dict(self.compute.driver.capabilities, supports_device_tagging=False): self.assertRaises(exception.BuildAbortException, self.compute._check_device_tagging, net_req_list, bdms) def test_check_device_tagging_tagged_bdm_no_driver_support(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', tag='foo', instance_uuid=uuids.instance)]) with mock.patch.dict(self.compute.driver.capabilities, supports_device_tagging=False): self.assertRaises(exception.BuildAbortException, self.compute._check_device_tagging, None, bdms) def test_check_device_tagging_tagged_bdm_no_driver_support_declared(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', tag='foo', instance_uuid=uuids.instance)]) with mock.patch.dict(self.compute.driver.capabilities): self.compute.driver.capabilities.pop('supports_device_tagging', None) self.assertRaises(exception.BuildAbortException, self.compute._check_device_tagging, None, bdms) def test_check_device_tagging_tagged_bdm_with_driver_support(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', tag='foo', instance_uuid=uuids.instance)]) net_req = net_req_obj.NetworkRequest(network_id=uuids.bar) net_req_list = net_req_obj.NetworkRequestList(objects=[net_req]) with mock.patch.dict(self.compute.driver.capabilities, supports_device_tagging=True): self.compute._check_device_tagging(net_req_list, bdms) def test_check_device_tagging_tagged_net_req_with_driver_support(self): bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(source_type='volume', destination_type='volume', instance_uuid=uuids.instance)]) net_req = net_req_obj.NetworkRequest(network_id=uuids.bar, tag='foo') net_req_list = net_req_obj.NetworkRequestList(objects=[net_req]) with mock.patch.dict(self.compute.driver.capabilities, supports_device_tagging=True): self.compute._check_device_tagging(net_req_list, bdms) @mock.patch.object(objects.BlockDeviceMapping, 'create') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid', return_value=objects.BlockDeviceMappingList()) def test_reserve_block_device_name_with_tag(self, mock_get, mock_create): instance = fake_instance.fake_instance_obj(self.context) with test.nested( mock.patch.object(self.compute, '_get_device_name_for_instance', return_value='/dev/vda'), mock.patch.dict(self.compute.driver.capabilities, supports_tagged_attach_volume=True)): bdm = self.compute.reserve_block_device_name( self.context, instance, None, None, None, None, 'foo', False) self.assertEqual('foo', bdm.tag) @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') def test_reserve_block_device_name_raises(self, _): with mock.patch.dict(self.compute.driver.capabilities, supports_tagged_attach_volume=False): self.assertRaises(exception.VolumeTaggedAttachNotSupported, self.compute.reserve_block_device_name, self.context, fake_instance.fake_instance_obj(self.context), 'fake_device', 'fake_volume_id', 'fake_disk_bus', 'fake_device_type', 'foo', False) @mock.patch.object(objects.BlockDeviceMapping, 'create') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid', return_value=objects.BlockDeviceMappingList()) def test_reserve_block_device_name_multiattach(self, mock_get, mock_create): """Tests the case that multiattach=True and the driver supports it.""" instance = fake_instance.fake_instance_obj(self.context) with test.nested( mock.patch.object(self.compute, '_get_device_name_for_instance', return_value='/dev/vda'), mock.patch.dict(self.compute.driver.capabilities, supports_multiattach=True)): self.compute.reserve_block_device_name( self.context, instance, device=None, volume_id=uuids.volume_id, disk_bus=None, device_type=None, tag=None, multiattach=True) @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') def test_reserve_block_device_name_multiattach_raises(self, _): with mock.patch.dict(self.compute.driver.capabilities, supports_multiattach=False): self.assertRaises(exception.MultiattachNotSupportedByVirtDriver, self.compute.reserve_block_device_name, self.context, fake_instance.fake_instance_obj(self.context), 'fake_device', 'fake_volume_id', 'fake_disk_bus', 'fake_device_type', tag=None, multiattach=True) @mock.patch.object(compute_utils, 'add_instance_fault_from_exc', new=mock.Mock()) @mock.patch.object(objects.BlockDeviceMapping, 'create', new=mock.Mock()) @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def test_reserve_block_device_name_raises_on_duplicate(self, mock_get): instance = fake_instance.fake_instance_obj(self.context) vol_bdm = objects.BlockDeviceMapping( self.context, id=1, instance_uuid=instance.uuid, volume_id="myinstanceuuid", source_type='volume', destination_type='volume', delete_on_termination=True, connection_info=None, tag='fake-tag', device_name='/dev/fake0', attachment_id=uuids.attachment_id) mock_get.return_value = objects.BlockDeviceMappingList( objects=[vol_bdm]) self.assertRaises(exception.InvalidVolume, self.compute.reserve_block_device_name, self.context, instance, None, "myinstanceuuid", None, None, 'foo', False) @mock.patch.object(objects.Instance, 'save') @mock.patch.object(time, 'sleep') def test_allocate_network_succeeds_after_retries( self, mock_sleep, mock_save): self.flags(network_allocate_retries=8) instance = fake_instance.fake_instance_obj( self.context, expected_attrs=['system_metadata']) req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='fake')]) sec_groups = 'fake-sec-groups' final_result = 'meow' rp_mapping = {} network_arqs = {} expected_sleep_times = [mock.call(t) for t in (1, 2, 4, 8, 16, 30, 30)] with mock.patch.object( self.compute.network_api, 'allocate_for_instance', side_effect=[test.TestingException()] * 7 + [final_result]): res = self.compute._allocate_network_async(self.context, instance, req_networks, sec_groups, rp_mapping, network_arqs) self.assertEqual(7, mock_sleep.call_count) mock_sleep.assert_has_calls(expected_sleep_times) self.assertEqual(final_result, res) # Ensure save is not called in while allocating networks, the instance # is saved after the allocation. self.assertFalse(mock_save.called) self.assertEqual('True', instance.system_metadata['network_allocated']) def test_allocate_network_fails(self): self.flags(network_allocate_retries=0) instance = {} req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='fake')]) sec_groups = 'fake-sec-groups' rp_mapping = {} network_arqs = None with mock.patch.object( self.compute.network_api, 'allocate_for_instance', side_effect=test.TestingException) as mock_allocate: self.assertRaises(test.TestingException, self.compute._allocate_network_async, self.context, instance, req_networks, sec_groups, rp_mapping, network_arqs) mock_allocate.assert_called_once_with( self.context, instance, requested_networks=req_networks, security_groups=sec_groups, bind_host_id=instance.get('host'), resource_provider_mapping=rp_mapping, network_arqs=network_arqs) @mock.patch.object(manager.ComputeManager, '_instance_update') @mock.patch.object(time, 'sleep') def test_allocate_network_with_conf_value_is_one( self, sleep, _instance_update): self.flags(network_allocate_retries=1) instance = fake_instance.fake_instance_obj( self.context, expected_attrs=['system_metadata']) req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='fake')]) sec_groups = 'fake-sec-groups' final_result = 'zhangtralon' rp_mapping = {} with mock.patch.object(self.compute.network_api, 'allocate_for_instance', side_effect = [test.TestingException(), final_result]): res = self.compute._allocate_network_async(self.context, instance, req_networks, sec_groups, rp_mapping, None) self.assertEqual(final_result, res) self.assertEqual(1, sleep.call_count) def test_allocate_network_skip_for_no_allocate(self): # Ensures that we don't do anything if requested_networks has 'none' # for the network_id. req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='none')]) nwinfo = self.compute._allocate_network_async( self.context, mock.sentinel.instance, req_networks, security_groups=['default'], resource_provider_mapping={}, network_arqs=None) self.assertEqual(0, len(nwinfo)) @mock.patch('nova.compute.manager.ComputeManager.' '_do_build_and_run_instance') def _test_max_concurrent_builds(self, mock_dbari): with mock.patch.object(self.compute, '_build_semaphore') as mock_sem: instance = objects.Instance(uuid=uuidutils.generate_uuid()) for i in (1, 2, 3): self.compute.build_and_run_instance(self.context, instance, mock.sentinel.image, mock.sentinel.request_spec, {}, []) self.assertEqual(3, mock_sem.__enter__.call_count) def test_max_concurrent_builds_limited(self): self.flags(max_concurrent_builds=2) self._test_max_concurrent_builds() def test_max_concurrent_builds_unlimited(self): self.flags(max_concurrent_builds=0) self._test_max_concurrent_builds() def test_max_concurrent_builds_semaphore_limited(self): self.flags(max_concurrent_builds=123) self.assertEqual(123, manager.ComputeManager()._build_semaphore.balance) def test_max_concurrent_builds_semaphore_unlimited(self): self.flags(max_concurrent_builds=0) compute = manager.ComputeManager() self.assertEqual(0, compute._build_semaphore.balance) self.assertIsInstance(compute._build_semaphore, compute_utils.UnlimitedSemaphore) @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager.' '_snapshot_instance') def _test_max_concurrent_snapshots(self, mock_si, mock_inst_save): with mock.patch.object(self.compute, '_snapshot_semaphore') as mock_sem: instance = objects.Instance(uuid=uuidutils.generate_uuid()) for i in (1, 2, 3): self.compute.snapshot_instance(self.context, mock.sentinel.image, instance) self.assertEqual(3, mock_sem.__enter__.call_count) def test_max_concurrent_snapshots_limited(self): self.flags(max_concurrent_snapshots=2) self._test_max_concurrent_snapshots() def test_max_concurrent_snapshots_unlimited(self): self.flags(max_concurrent_snapshots=0) self._test_max_concurrent_snapshots() def test_max_concurrent_snapshots_semaphore_limited(self): self.flags(max_concurrent_snapshots=123) self.assertEqual(123, manager.ComputeManager()._snapshot_semaphore.balance) def test_max_concurrent_snapshots_semaphore_unlimited(self): self.flags(max_concurrent_snapshots=0) compute = manager.ComputeManager() self.assertEqual(0, compute._snapshot_semaphore.balance) self.assertIsInstance(compute._snapshot_semaphore, compute_utils.UnlimitedSemaphore) def test_nil_out_inst_obj_host_and_node_sets_nil(self): instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance, host='foo-host', node='foo-node', launched_on='foo-host') self.assertIsNotNone(instance.host) self.assertIsNotNone(instance.node) self.assertIsNotNone(instance.launched_on) self.compute._nil_out_instance_obj_host_and_node(instance) self.assertIsNone(instance.host) self.assertIsNone(instance.node) self.assertIsNone(instance.launched_on) def test_init_host(self): our_host = self.compute.host inst = fake_instance.fake_db_instance( vm_state=vm_states.ACTIVE, info_cache=dict(test_instance_info_cache.fake_info_cache, network_info=None), security_groups=None) startup_instances = [inst, inst, inst] def _make_instance_list(db_list): return instance_obj._make_instance_list( self.context, objects.InstanceList(), db_list, None) @mock.patch.object(manager.ComputeManager, '_ensure_existing_node_identity') @mock.patch.object(manager.ComputeManager, '_get_nodes') @mock.patch.object(manager.ComputeManager, '_error_out_instances_whose_build_was_interrupted') @mock.patch.object(fake_driver.FakeDriver, 'init_host') @mock.patch.object(objects.InstanceList, 'get_by_host') @mock.patch.object(context, 'get_admin_context') @mock.patch.object(manager.ComputeManager, '_destroy_evacuated_instances') @mock.patch.object(manager.ComputeManager, '_validate_pinning_configuration') @mock.patch.object(manager.ComputeManager, '_validate_vtpm_configuration') @mock.patch.object(manager.ComputeManager, '_init_instance') @mock.patch.object(self.compute, '_update_scheduler_instance_info') def _do_mock_calls(mock_update_scheduler, mock_inst_init, mock_validate_vtpm, mock_validate_pinning, mock_destroy, mock_admin_ctxt, mock_host_get, mock_init_host, mock_error_interrupted, mock_get_nodes, mock_existing_node): mock_admin_ctxt.return_value = self.context inst_list = _make_instance_list(startup_instances) mock_host_get.return_value = inst_list our_node = objects.ComputeNode( host=self.compute.host, uuid=uuids.our_node_uuid, hypervisor_hostname='fake-node') mock_get_nodes.return_value = {uuids.our_node_uuid: our_node} self.compute.init_host(None) mock_existing_node.assert_not_called() mock_validate_pinning.assert_called_once_with(inst_list) mock_validate_vtpm.assert_called_once_with(inst_list) mock_destroy.assert_called_once_with( self.context, {uuids.our_node_uuid: our_node}) mock_inst_init.assert_has_calls( [mock.call(self.context, inst_list[0]), mock.call(self.context, inst_list[1]), mock.call(self.context, inst_list[2])]) mock_init_host.assert_called_once_with(host=our_host) mock_host_get.assert_called_once_with(self.context, our_host, expected_attrs=['info_cache', 'metadata', 'numa_topology']) mock_update_scheduler.assert_called_once_with( self.context, inst_list) mock_error_interrupted.assert_called_once_with( self.context, {inst.uuid for inst in inst_list}, mock_get_nodes.return_value.keys()) _do_mock_calls() @mock.patch('nova.compute.manager.ComputeManager._get_nodes') @mock.patch('nova.compute.manager.ComputeManager.' '_error_out_instances_whose_build_was_interrupted') @mock.patch('nova.objects.InstanceList.get_by_host', return_value=objects.InstanceList()) @mock.patch('nova.compute.manager.ComputeManager.' '_destroy_evacuated_instances') @mock.patch('nova.compute.manager.ComputeManager._init_instance', mock.NonCallableMock()) @mock.patch('nova.compute.manager.ComputeManager.' '_update_scheduler_instance_info', mock.NonCallableMock()) def test_init_host_no_instances( self, mock_destroy_evac_instances, mock_get_by_host, mock_error_interrupted, mock_get_nodes): """Tests the case that init_host runs and there are no instances on this host yet (it's brand new). Uses NonCallableMock for the methods we assert should not be called. """ mock_get_nodes.return_value = { uuids.cn_uuid1: objects.ComputeNode( uuid=uuids.cn_uuid1, hypervisor_hostname='node1', host=self.compute.host)} self.compute.init_host(None) mock_error_interrupted.assert_called_once_with( test.MatchType(nova.context.RequestContext), set(), mock_get_nodes.return_value.keys()) mock_get_nodes.assert_called_once_with( test.MatchType(nova.context.RequestContext)) @mock.patch('nova.objects.InstanceList') @mock.patch('nova.objects.MigrationList.get_by_filters') @mock.patch('nova.objects.ComputeNodeList.get_all_by_uuids') def test_cleanup_host(self, mock_cnlist_get, mock_miglist_get, mock_instance_list): # just testing whether the cleanup_host method # when fired will invoke the underlying driver's # equivalent method. mock_miglist_get.return_value = [] mock_instance_list.get_by_host.return_value = [] mock_cnlist_get.return_value = [] with mock.patch.object(self.compute, 'driver') as mock_driver: self.compute.init_host(None) mock_driver.init_host.assert_called_once_with(host='fake-mini') self.compute.cleanup_host() # register_event_listener is called on startup (init_host) and # in cleanup_host mock_driver.register_event_listener.assert_has_calls([ mock.call(self.compute.handle_events), mock.call(None)]) mock_driver.cleanup_host.assert_called_once_with(host='fake-mini') def test_cleanup_live_migrations_in_pool_with_record(self): fake_future = mock.MagicMock() fake_instance_uuid = uuids.instance fake_migration = objects.Migration( uuid=uuids.migration, instance_uuid=fake_instance_uuid) fake_migration.save = mock.MagicMock() self.compute._waiting_live_migrations[fake_instance_uuid] = ( fake_migration, fake_future) with mock.patch.object(self.compute, '_live_migration_executor' ) as mock_migration_pool: self.compute._cleanup_live_migrations_in_pool() mock_migration_pool.shutdown.assert_called_once_with(wait=False) self.assertEqual('cancelled', fake_migration.status) fake_future.cancel.assert_called_once_with() self.assertEqual({}, self.compute._waiting_live_migrations) # test again with Future is None self.compute._waiting_live_migrations[fake_instance_uuid] = ( None, None) self.compute._cleanup_live_migrations_in_pool() mock_migration_pool.shutdown.assert_called_with(wait=False) self.assertEqual(2, mock_migration_pool.shutdown.call_count) self.assertEqual({}, self.compute._waiting_live_migrations) def test_init_virt_events_disabled(self): self.flags(handle_virt_lifecycle_events=False, group='workarounds') with mock.patch.object(self.compute.driver, 'register_event_listener') as mock_register: self.compute.init_virt_events() self.assertFalse(mock_register.called) @mock.patch('nova.compute.manager.ComputeManager._get_nodes') @mock.patch.object(manager.ComputeManager, '_error_out_instances_whose_build_was_interrupted') @mock.patch('nova.scheduler.utils.resources_from_flavor') @mock.patch.object(manager.ComputeManager, '_get_instances_on_driver') @mock.patch.object(manager.ComputeManager, 'init_virt_events') @mock.patch.object(context, 'get_admin_context') @mock.patch.object(objects.InstanceList, 'get_by_host') @mock.patch.object(fake_driver.FakeDriver, 'destroy') @mock.patch.object(fake_driver.FakeDriver, 'init_host') @mock.patch('nova.utils.temporary_mutation') @mock.patch('nova.objects.MigrationList.get_by_filters') @mock.patch('nova.objects.Migration.save') def test_init_host_with_evacuated_instance(self, mock_save, mock_mig_get, mock_temp_mut, mock_init_host, mock_destroy, mock_host_get, mock_admin_ctxt, mock_init_virt, mock_get_inst, mock_resources, mock_error_interrupted, mock_get_nodes): our_host = self.compute.host not_our_host = 'not-' + our_host deleted_instance = fake_instance.fake_instance_obj( self.context, host=not_our_host, uuid=uuids.deleted_instance) migration = objects.Migration(instance_uuid=deleted_instance.uuid) migration.source_node = 'fake-node' mock_mig_get.return_value = [migration] mock_admin_ctxt.return_value = self.context mock_host_get.return_value = objects.InstanceList() our_node = objects.ComputeNode( host=our_host, uuid=uuids.our_node_uuid, hypervisor_hostname='fake-node') mock_get_nodes.return_value = {uuids.our_node_uuid: our_node} mock_resources.return_value = mock.sentinel.my_resources # simulate failed instance mock_get_inst.return_value = [deleted_instance] with test.nested( mock.patch.object( self.compute.network_api, 'get_instance_nw_info', side_effect = exception.InstanceNotFound( instance_id=deleted_instance['uuid'])), mock.patch.object( self.compute.reportclient, 'remove_provider_tree_from_instance_allocation') ) as (mock_get_net, mock_remove_allocation): self.compute.init_host(None) mock_remove_allocation.assert_called_once_with( self.context, deleted_instance.uuid, uuids.our_node_uuid) mock_init_host.assert_called_once_with(host=our_host) mock_host_get.assert_called_once_with(self.context, our_host, expected_attrs=['info_cache', 'metadata', 'numa_topology']) mock_init_virt.assert_called_once_with() mock_temp_mut.assert_called_once_with(self.context, read_deleted='yes') mock_get_inst.assert_called_once_with(self.context) mock_get_net.assert_called_once_with(self.context, deleted_instance) # ensure driver.destroy is called so that driver may # clean up any dangling files mock_destroy.assert_called_once_with(self.context, deleted_instance, mock.ANY, mock.ANY, mock.ANY) mock_save.assert_called_once_with() mock_error_interrupted.assert_called_once_with( self.context, {deleted_instance.uuid}, mock_get_nodes.return_value.keys()) @mock.patch('nova.compute.manager.ComputeManager._get_nodes') @mock.patch.object(manager.ComputeManager, '_error_out_instances_whose_build_was_interrupted') @mock.patch.object(context, 'get_admin_context') @mock.patch.object(objects.InstanceList, 'get_by_host') @mock.patch.object(fake_driver.FakeDriver, 'init_host') @mock.patch('nova.compute.manager.ComputeManager._init_instance') @mock.patch('nova.compute.manager.ComputeManager.' '_destroy_evacuated_instances') def test_init_host_with_in_progress_evacuations(self, mock_destroy_evac, mock_init_instance, mock_init_host, mock_host_get, mock_admin_ctxt, mock_error_interrupted, mock_get_nodes): """Assert that init_instance is not called for instances that are evacuating from the host during init_host. """ active_instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, uuid=uuids.active_instance) active_instance.system_metadata = {} evacuating_instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, uuid=uuids.evac_instance) evacuating_instance.system_metadata = {} instance_list = objects.InstanceList(self.context, objects=[active_instance, evacuating_instance]) mock_host_get.return_value = instance_list mock_admin_ctxt.return_value = self.context mock_destroy_evac.return_value = { uuids.evac_instance: evacuating_instance } our_node = objects.ComputeNode( host=self.compute.host, uuid=uuids.our_node_uuid, hypervisor_hostname='fake-node') mock_get_nodes.return_value = {uuids.our_node_uuid: our_node} self.compute.init_host(None) mock_init_instance.assert_called_once_with( self.context, active_instance) mock_error_interrupted.assert_called_once_with( self.context, {active_instance.uuid, evacuating_instance.uuid}, mock_get_nodes.return_value.keys()) @mock.patch.object(objects.ComputeNodeList, 'get_all_by_uuids') @mock.patch.object(fake_driver.FakeDriver, 'get_nodenames_by_uuid') def test_get_nodes(self, mock_driver_get_nodes, mock_get_by_uuid): mock_driver_get_nodes.return_value = {uuids.node_fake_node1: 'host', uuids.node_fake_node2: 'host'} # NOTE(danms): The fake driver, by default, uses # uuidsentinel.node_$node_name, so we can predict the uuids it will # return here. cn1 = objects.ComputeNode(uuid=uuids.node_fake_node1, hypervisor_hostname='host') cn2 = objects.ComputeNode(uuid=uuids.node_fake_node2, hypervisor_hostname='host') mock_get_by_uuid.return_value = [cn1, cn2] nodes = self.compute._get_nodes(self.context) self.assertEqual({uuids.node_fake_node1: cn1, uuids.node_fake_node2: cn2}, nodes) mock_driver_get_nodes.assert_called_once_with() mock_get_by_uuid.assert_called_once_with(self.context, [uuids.node_fake_node1, uuids.node_fake_node2]) @mock.patch.object(objects.ComputeNodeList, 'get_all_by_uuids') @mock.patch.object(fake_driver.FakeDriver, 'get_nodenames_by_uuid') def test_get_nodes_mismatch(self, mock_driver_get_nodes, mock_get_by_uuid): # Virt driver reports a (hypervisor_) hostname of 'host1' mock_driver_get_nodes.return_value = {uuids.node_fake_node1: 'host1', uuids.node_fake_node2: 'host1'} # The database records for our compute nodes (by UUID) show a # hypervisor_hostname of 'host2' cn1 = objects.ComputeNode(uuid=uuids.node_fake_node1, hypervisor_hostname='host2') cn2 = objects.ComputeNode(uuid=uuids.node_fake_node2, hypervisor_hostname='host2') mock_get_by_uuid.return_value = [cn1, cn2] # Possible hostname (as reported by the virt driver) rename, # which should abort our startup self.assertRaises(exception.InvalidConfiguration, self.compute._get_nodes, self.context) @mock.patch.object(manager.LOG, 'warning') @mock.patch.object( objects.ComputeNode, 'get_by_host_and_nodename', new_callable=mock.NonCallableMock) @mock.patch.object( fake_driver.FakeDriver, 'get_available_nodes', side_effect=exception.VirtDriverNotReady) def test_get_nodes_driver_not_ready( self, mock_driver_get_nodes, mock_get_by_host_and_node, mock_log_warning): mock_driver_get_nodes.return_value = ['fake-node1', 'fake-node2'] nodes = self.compute._get_nodes(self.context) self.assertEqual({}, nodes) mock_log_warning.assert_called_once_with( "Virt driver is not ready. If this is the first time this service " "is starting on this host, then you can ignore this warning.") @mock.patch.object(manager.LOG, 'warning') @mock.patch.object(objects.ComputeNodeList, 'get_all_by_uuids') @mock.patch.object(fake_driver.FakeDriver, 'get_nodenames_by_uuid') def test_get_nodes_node_not_found( self, mock_driver_get_nodes, mock_get_all_by_uuids, mock_log_warning): mock_driver_get_nodes.return_value = {uuids.node_1: 'fake-node1'} mock_get_all_by_uuids.return_value = [] nodes = self.compute._get_nodes(self.context) self.assertEqual({}, nodes) mock_driver_get_nodes.assert_called_once_with() mock_get_all_by_uuids.assert_called_once_with(self.context, [uuids.node_1]) mock_log_warning.assert_called_once_with( "Compute nodes %s for host %s were not found in the database. " "If this is the first time this service is starting on this host, " "then you can ignore this warning.", [uuids.node_1], 'fake-mini') def test_init_host_disk_devices_configuration_failure(self): self.flags(max_disk_devices_to_attach=0, group='compute') self.assertRaises(exception.InvalidConfiguration, self.compute.init_host, None) @mock.patch.object(objects.InstanceList, 'get_by_host', new=mock.Mock()) @mock.patch('nova.objects.ComputeNodeList.get_all_by_uuids', new=mock.Mock(return_value=[mock.MagicMock()])) @mock.patch('nova.compute.manager.ComputeManager.' '_validate_pinning_configuration') def test_init_host_pinning_configuration_validation_failure(self, mock_validate_pinning): """Test that we fail init_host if the pinning configuration check fails. """ mock_validate_pinning.side_effect = exception.InvalidConfiguration self.assertRaises(exception.InvalidConfiguration, self.compute.init_host, None) @mock.patch.object(objects.InstanceList, 'get_by_host', new=mock.Mock()) @mock.patch('nova.compute.manager.ComputeManager.' '_validate_pinning_configuration', new=mock.Mock()) @mock.patch('nova.objects.ComputeNodeList.get_all_by_uuids', new=mock.Mock(return_value=[mock.MagicMock()])) @mock.patch('nova.compute.manager.ComputeManager.' '_validate_vtpm_configuration') def test_init_host_vtpm_configuration_validation_failure(self, mock_validate_vtpm): """Test that we fail init_host if the vTPM configuration check fails. """ mock_validate_vtpm.side_effect = exception.InvalidConfiguration self.assertRaises(exception.InvalidConfiguration, self.compute.init_host, None) @mock.patch.object(objects.Instance, 'save') @mock.patch.object(objects.InstanceList, 'get_by_filters') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_resource_provider') def test_init_host_with_interrupted_instance_build( self, mock_get_allocations, mock_get_instances, mock_instance_save): active_instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, uuid=uuids.active_instance) evacuating_instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, uuid=uuids.evac_instance) interrupted_instance = fake_instance.fake_instance_obj( self.context, host=None, uuid=uuids.interrupted_instance, vm_state=vm_states.BUILDING) # we have 3 different instances. We need consumers for each instance # in placement and an extra consumer that is not an instance allocations = { uuids.active_instance: "fake-resources-active", uuids.evac_instance: "fake-resources-evacuating", uuids.interrupted_instance: "fake-resources-interrupted", uuids.not_an_instance: "fake-resources-not-an-instance", } mock_get_allocations.return_value = report.ProviderAllocInfo( allocations=allocations) # get is called with a uuid filter containing interrupted_instance, # error_instance, and not_an_instance but it will only return the # interrupted_instance as the error_instance is not in building state # and not_an_instance does not match with any instance in the db. mock_get_instances.return_value = objects.InstanceList( self.context, objects=[interrupted_instance]) # interrupted_instance and error_instance is not in the list passed in # because it is not assigned to the compute and therefore not processed # by init_host and init_instance self.compute._error_out_instances_whose_build_was_interrupted( self.context, {inst.uuid for inst in [active_instance, evacuating_instance]}, [uuids.cn_uuid]) mock_get_instances.assert_called_once_with( self.context, {'vm_state': 'building', 'uuid': {uuids.interrupted_instance, uuids.not_an_instance} }, expected_attrs=[]) # this is expected to be called only once for interrupted_instance mock_instance_save.assert_called_once_with() self.assertEqual(vm_states.ERROR, interrupted_instance.vm_state) @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_resource_provider') def test_init_host_with_interrupted_instance_build_compute_rp_not_found( self, mock_get_allocations): mock_get_allocations.side_effect = [ exception.ResourceProviderAllocationRetrievalFailed( rp_uuid=uuids.cn1_uuid, error='error'), report.ProviderAllocInfo(allocations={}) ] self.compute._error_out_instances_whose_build_was_interrupted( self.context, set(), [uuids.cn1_uuid, uuids.cn2_uuid]) # check that nova skip the node that is not found in placement and # continue with the next mock_get_allocations.assert_has_calls( [ mock.call(self.context, uuids.cn1_uuid), mock.call(self.context, uuids.cn2_uuid), ] ) def test_init_instance_with_binding_failed_vif_type(self): # this instance will plug a 'binding_failed' vif instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, info_cache=None, power_state=power_state.RUNNING, vm_state=vm_states.ACTIVE, task_state=None, host=self.compute.host, expected_attrs=['info_cache']) with test.nested( mock.patch.object(context, 'get_admin_context', return_value=self.context), mock.patch.object(objects.Instance, 'get_network_info', return_value=network_model.NetworkInfo()), mock.patch.object(self.compute.driver, 'plug_vifs', side_effect=exception.VirtualInterfacePlugException( "Unexpected vif_type=binding_failed")), mock.patch.object(self.compute, '_set_instance_obj_error_state') ) as (get_admin_context, get_nw_info, plug_vifs, set_error_state): self.compute._init_instance(self.context, instance) set_error_state.assert_called_once_with(instance) def test_init_instance_vif_plug_fails_missing_pci(self): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, info_cache=None, power_state=power_state.RUNNING, vm_state=vm_states.ACTIVE, task_state=None, host=self.compute.host, expected_attrs=['info_cache']) with test.nested( mock.patch.object(context, 'get_admin_context', return_value=self.context), mock.patch.object(objects.Instance, 'get_network_info', return_value=network_model.NetworkInfo()), mock.patch.object(self.compute.driver, 'plug_vifs', side_effect=exception.PciDeviceNotFoundById("pci-addr")), mock.patch("nova.compute.manager.LOG.exception"), ) as (get_admin_context, get_nw_info, plug_vifs, log_exception): # as this does not raise, we are sure that the compute service # continues initializing the rest of the instances self.compute._init_instance(self.context, instance) log_exception.assert_called_once_with( "Virtual interface plugging failed for instance. Probably the " "vnic_type of the bound port has been changed. Nova does not " "support such change.", instance=instance ) def _test__validate_pinning_configuration(self, supports_pcpus=True): instance_1 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_1) instance_2 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_2) instance_3 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_3) instance_4 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_4) instance_5 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_5) instance_1.numa_topology = None numa_wo_pinning = test_instance_numa.get_fake_obj_numa_topology( self.context) numa_wo_pinning.cells[0].pcpuset = set() numa_wo_pinning.cells[1].pcpuset = set() instance_2.numa_topology = numa_wo_pinning numa_w_pinning = test_instance_numa.get_fake_obj_numa_topology( self.context) numa_w_pinning.cells[0].pin_vcpus((1, 10), (2, 11)) numa_w_pinning.cells[0].cpuset = set() numa_w_pinning.cells[0].cpu_policy = ( fields.CPUAllocationPolicy.DEDICATED) numa_w_pinning.cells[1].pin_vcpus((3, 0), (4, 1)) numa_w_pinning.cells[1].cpuset = set() numa_w_pinning.cells[1].cpu_policy = ( fields.CPUAllocationPolicy.DEDICATED) instance_3.numa_topology = numa_w_pinning instance_4.deleted = True numa_mixed_pinning = test_instance_numa.get_fake_obj_numa_topology( self.context) numa_mixed_pinning.cells[0].cpuset = set([5, 6]) numa_mixed_pinning.cells[0].pin_vcpus((1, 8), (2, 9)) numa_mixed_pinning.cells[0].cpu_policy = ( fields.CPUAllocationPolicy.MIXED) numa_mixed_pinning.cells[1].cpuset = set([7, 8]) numa_mixed_pinning.cells[1].pin_vcpus((3, 10), (4, 11)) numa_mixed_pinning.cells[1].cpu_policy = ( fields.CPUAllocationPolicy.MIXED) instance_5.numa_topology = numa_mixed_pinning instance_5.vcpus = 8 instances = objects.InstanceList(objects=[ instance_1, instance_2, instance_3, instance_4, instance_5]) with mock.patch.dict(self.compute.driver.capabilities, supports_pcpus=supports_pcpus): self.compute._validate_pinning_configuration(instances) def test__validate_pinning_configuration_valid_config(self): """Test that configuring proper 'cpu_dedicated_set' and 'cpu_shared_set', all tests passed. """ self.flags(cpu_shared_set='2-7', cpu_dedicated_set='0-1,8-15', group='compute') self._test__validate_pinning_configuration() def test__validate_pinning_configuration_invalid_unpinned_config(self): """Test that configuring only 'cpu_dedicated_set' when there are unpinned instances on the host results in an error. """ self.flags(cpu_dedicated_set='0-7', group='compute') ex = self.assertRaises( exception.InvalidConfiguration, self._test__validate_pinning_configuration) self.assertIn('This host has unpinned instances but has no CPUs ' 'set aside for this purpose;', str(ex)) def test__validate_pinning_configuration_invalid_pinned_config(self): """Test that configuring only 'cpu_shared_set' when there are pinned instances on the host results in an error """ self.flags(cpu_shared_set='0-7', group='compute') ex = self.assertRaises( exception.InvalidConfiguration, self._test__validate_pinning_configuration) self.assertIn('This host has pinned instances but has no CPUs ' 'set aside for this purpose;', str(ex)) @mock.patch.object(manager.LOG, 'warning') def test__validate_pinning_configuration_warning(self, mock_log): """Test that configuring 'cpu_dedicated_set' such that some pinned cores of the instance are outside the range it specifies results in a warning. """ self.flags(cpu_shared_set='0-7', cpu_dedicated_set='8-15', group='compute') self._test__validate_pinning_configuration() self.assertEqual(1, mock_log.call_count) self.assertIn('Instance is pinned to host CPUs %(cpus)s ' 'but one or more of these CPUs are not included in ', str(mock_log.call_args[0])) def test__validate_pinning_configuration_no_config(self): """Test that not configuring 'cpu_dedicated_set' or 'cpu_shared_set' when there are mixed instances on the host results in an error. """ ex = self.assertRaises( exception.InvalidConfiguration, self._test__validate_pinning_configuration) self.assertIn("This host has mixed instance requesting both pinned " "and unpinned CPUs but hasn't set aside unpinned CPUs " "for this purpose;", str(ex)) def test__validate_pinning_configuration_not_supported(self): """Test that the entire check is skipped if the driver doesn't even support PCPUs. """ self._test__validate_pinning_configuration(supports_pcpus=False) def _test__validate_vtpm_configuration(self, supports_vtpm): instance_1 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_1) instance_2 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_2) instance_3 = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance_3) image_meta = objects.ImageMeta.from_dict({}) instance_2.flavor.extra_specs = {'hw:tpm_version': '2.0'} instance_3.deleted = True instances = objects.InstanceList(objects=[ instance_1, instance_2, instance_3]) with test.nested( mock.patch.dict( self.compute.driver.capabilities, supports_vtpm=supports_vtpm, ), mock.patch.object( objects.ImageMeta, 'from_instance', return_value=image_meta, ), ): self.compute._validate_vtpm_configuration(instances) def test__validate_vtpm_configuration_unsupported(self): """Test that the check fails if the driver does not support vTPM and instances request it. """ ex = self.assertRaises( exception.InvalidConfiguration, self._test__validate_vtpm_configuration, supports_vtpm=False) self.assertIn( 'This host has instances with the vTPM feature enabled, but the ' 'host is not correctly configured; ', str(ex)) def test__validate_vtpm_configuration_supported(self): """Test that the entire check is skipped if the driver supports vTPM. """ self._test__validate_vtpm_configuration(supports_vtpm=True) def test__get_power_state_InstanceNotFound(self): instance = fake_instance.fake_instance_obj( self.context, power_state=power_state.RUNNING) with mock.patch.object(self.compute.driver, 'get_info', side_effect=exception.InstanceNotFound(instance_id=1)): self.assertEqual( power_state.NOSTATE, self.compute._get_power_state(instance), ) def test__get_power_state_NotFound(self): instance = fake_instance.fake_instance_obj( self.context, power_state=power_state.RUNNING) with mock.patch.object(self.compute.driver, 'get_info', side_effect=exception.NotFound()): self.assertRaises(exception.NotFound, self.compute._get_power_state, instance) @mock.patch.object(manager.ComputeManager, '_get_power_state') @mock.patch.object(fake_driver.FakeDriver, 'plug_vifs') @mock.patch.object(fake_driver.FakeDriver, 'resume_state_on_host_boot') @mock.patch.object(manager.ComputeManager, '_get_instance_block_device_info') @mock.patch.object(manager.ComputeManager, '_set_instance_obj_error_state') def test_init_instance_failed_resume_sets_error(self, mock_set_inst, mock_get_inst, mock_resume, mock_plug, mock_get_power): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, info_cache=None, power_state=power_state.RUNNING, vm_state=vm_states.ACTIVE, task_state=None, host=self.compute.host, expected_attrs=['info_cache']) self.flags(resume_guests_state_on_host_boot=True) mock_get_power.side_effect = (power_state.SHUTDOWN, power_state.SHUTDOWN) mock_get_inst.return_value = 'fake-bdm' mock_resume.side_effect = test.TestingException self.compute._init_instance('fake-context', instance) mock_get_power.assert_has_calls([mock.call(instance), mock.call(instance)]) mock_plug.assert_called_once_with(instance, mock.ANY) mock_get_inst.assert_called_once_with(mock.ANY, instance) mock_resume.assert_called_once_with(mock.ANY, instance, mock.ANY, 'fake-bdm') mock_set_inst.assert_called_once_with(instance) @mock.patch.object(objects.BlockDeviceMapping, 'destroy') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(objects.Instance, 'destroy') @mock.patch.object(objects.Instance, 'obj_load_attr') @mock.patch.object(objects.quotas, 'ids_from_instance') def test_init_instance_complete_partial_deletion( self, mock_ids_from_instance, mock_inst_destroy, mock_obj_load_attr, mock_get_by_instance_uuid, mock_bdm_destroy): """Test to complete deletion for instances in DELETED status but not marked as deleted in the DB """ instance = fake_instance.fake_instance_obj( self.context, project_id=fakes.FAKE_PROJECT_ID, uuid=uuids.instance, vcpus=1, memory_mb=64, power_state=power_state.SHUTDOWN, vm_state=vm_states.DELETED, host=self.compute.host, task_state=None, deleted=False, deleted_at=None, metadata={}, system_metadata={}, expected_attrs=['metadata', 'system_metadata']) # Make sure instance vm_state is marked as 'DELETED' but instance is # not destroyed from db. self.assertEqual(vm_states.DELETED, instance.vm_state) self.assertFalse(instance.deleted) def fake_inst_destroy(): instance.deleted = True instance.deleted_at = timeutils.utcnow() mock_ids_from_instance.return_value = (instance.project_id, instance.user_id) mock_inst_destroy.side_effect = fake_inst_destroy() self.compute._init_instance(self.context, instance) # Make sure that instance.destroy method was called and # instance was deleted from db. self.assertNotEqual(0, instance.deleted) @mock.patch('nova.compute.manager.LOG') def test_init_instance_complete_partial_deletion_raises_exception( self, mock_log): instance = fake_instance.fake_instance_obj( self.context, project_id=fakes.FAKE_PROJECT_ID, uuid=uuids.instance, vcpus=1, memory_mb=64, power_state=power_state.SHUTDOWN, vm_state=vm_states.DELETED, host=self.compute.host, task_state=None, deleted=False, deleted_at=None, metadata={}, system_metadata={}, expected_attrs=['metadata', 'system_metadata']) with mock.patch.object(self.compute, '_complete_partial_deletion') as mock_deletion: mock_deletion.side_effect = test.TestingException() self.compute._init_instance(self, instance) msg = u'Failed to complete a deletion' mock_log.exception.assert_called_once_with(msg, instance=instance) def test_init_instance_stuck_in_deleting(self): instance = fake_instance.fake_instance_obj( self.context, project_id=fakes.FAKE_PROJECT_ID, uuid=uuids.instance, vcpus=1, memory_mb=64, power_state=power_state.RUNNING, vm_state=vm_states.ACTIVE, host=self.compute.host, task_state=task_states.DELETING) bdms = [] with test.nested( mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid', return_value=bdms), mock.patch.object(self.compute, '_delete_instance'), mock.patch.object(instance, 'obj_load_attr') ) as (mock_get, mock_delete, mock_load): self.compute._init_instance(self.context, instance) mock_get.assert_called_once_with(self.context, instance.uuid) mock_delete.assert_called_once_with(self.context, instance, bdms) @mock.patch.object(objects.Instance, 'get_by_uuid') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def test_init_instance_stuck_in_deleting_raises_exception( self, mock_get_by_instance_uuid, mock_get_by_uuid): instance = fake_instance.fake_instance_obj( self.context, project_id=fakes.FAKE_PROJECT_ID, uuid=uuids.instance, vcpus=1, memory_mb=64, metadata={}, system_metadata={}, host=self.compute.host, vm_state=vm_states.ACTIVE, task_state=task_states.DELETING, expected_attrs=['metadata', 'system_metadata']) bdms = [] def _create_patch(name, attr): patcher = mock.patch.object(name, attr) mocked_obj = patcher.start() self.addCleanup(patcher.stop) return mocked_obj mock_delete_instance = _create_patch(self.compute, '_delete_instance') mock_set_instance_error_state = _create_patch( self.compute, '_set_instance_obj_error_state') mock_get_by_instance_uuid.return_value = bdms mock_get_by_uuid.return_value = instance mock_delete_instance.side_effect = test.TestingException('test') self.compute._init_instance(self.context, instance) mock_set_instance_error_state.assert_called_once_with(instance) def _test_init_instance_reverts_crashed_migrations(self, old_vm_state=None): power_on = True if (not old_vm_state or old_vm_state == vm_states.ACTIVE) else False sys_meta = { 'old_vm_state': old_vm_state } instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_states.ERROR, task_state=task_states.RESIZE_MIGRATING, power_state=power_state.SHUTDOWN, system_metadata=sys_meta, host=self.compute.host, expected_attrs=['system_metadata']) instance.migration_context = objects.MigrationContext(migration_id=42) migration = objects.Migration(source_compute='fake-host1', dest_compute='fake-host2') with test.nested( mock.patch.object(objects.Instance, 'get_network_info', return_value=network_model.NetworkInfo()), mock.patch.object(self.compute.driver, 'plug_vifs'), mock.patch.object(self.compute.driver, 'finish_revert_migration'), mock.patch.object(self.compute, '_get_instance_block_device_info', return_value=[]), mock.patch.object(self.compute.driver, 'get_info'), mock.patch.object(instance, 'save'), mock.patch.object(self.compute, '_retry_reboot', return_value=(False, None)), mock.patch.object(objects.Migration, 'get_by_id_and_instance', return_value=migration) ) as (mock_get_nw, mock_plug, mock_finish, mock_get_inst, mock_get_info, mock_save, mock_retry, mock_get_mig): mock_get_info.side_effect = ( hardware.InstanceInfo(state=power_state.SHUTDOWN), hardware.InstanceInfo(state=power_state.SHUTDOWN)) self.compute._init_instance(self.context, instance) mock_get_mig.assert_called_with(self.context, 42, instance.uuid) mock_retry.assert_called_once_with(instance, power_state.SHUTDOWN) mock_get_nw.assert_called_once_with() mock_plug.assert_called_once_with(instance, []) mock_get_inst.assert_called_once_with(self.context, instance) mock_finish.assert_called_once_with(self.context, instance, [], migration, [], power_on) mock_save.assert_called_once_with() mock_get_info.assert_has_calls( [mock.call(instance, use_cache=False), mock.call(instance, use_cache=False)]) self.assertIsNone(instance.task_state) def test_init_instance_reverts_crashed_migration_from_active(self): self._test_init_instance_reverts_crashed_migrations( old_vm_state=vm_states.ACTIVE) def test_init_instance_reverts_crashed_migration_from_stopped(self): self._test_init_instance_reverts_crashed_migrations( old_vm_state=vm_states.STOPPED) def test_init_instance_reverts_crashed_migration_no_old_state(self): self._test_init_instance_reverts_crashed_migrations(old_vm_state=None) def test_init_instance_resets_crashed_live_migration(self): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_states.ACTIVE, host=self.compute.host, task_state=task_states.MIGRATING) migration = objects.Migration(source_compute='fake-host1', id=39, dest_compute='fake-host2') with test.nested( mock.patch.object(instance, 'save'), mock.patch('nova.objects.Instance.get_network_info', return_value=network_model.NetworkInfo()), mock.patch.object(objects.Migration, 'get_by_instance_and_status', return_value=migration), mock.patch.object(self.compute, 'live_migration_abort'), mock.patch.object(self.compute, '_set_migration_status') ) as (save, get_nw_info, mock_get_status, mock_abort, mock_set_migr): self.compute._init_instance(self.context, instance) save.assert_called_once_with(expected_task_state=['migrating']) get_nw_info.assert_called_once_with() mock_get_status.assert_called_with(self.context, instance.uuid, 'running') mock_abort.assert_called_with(self.context, instance, migration.id) mock_set_migr.assert_called_with(migration, 'error') self.assertIsNone(instance.task_state) self.assertEqual(vm_states.ACTIVE, instance.vm_state) def _test_init_instance_sets_building_error(self, vm_state, task_state=None): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_state, host=self.compute.host, task_state=task_state) with mock.patch.object(instance, 'save') as save: self.compute._init_instance(self.context, instance) save.assert_called_once_with() self.assertIsNone(instance.task_state) self.assertEqual(vm_states.ERROR, instance.vm_state) def test_init_instance_sets_building_error(self): self._test_init_instance_sets_building_error(vm_states.BUILDING) def test_init_instance_sets_rebuilding_errors(self): tasks = [task_states.REBUILDING, task_states.REBUILD_BLOCK_DEVICE_MAPPING, task_states.REBUILD_SPAWNING] vms = [vm_states.ACTIVE, vm_states.STOPPED] for vm_state in vms: for task_state in tasks: self._test_init_instance_sets_building_error( vm_state, task_state) def _test_init_instance_sets_building_tasks_error(self, instance): instance.host = self.compute.host with mock.patch.object(instance, 'save') as save: self.compute._init_instance(self.context, instance) save.assert_called_once_with() self.assertIsNone(instance.task_state) self.assertEqual(vm_states.ERROR, instance.vm_state) def test_init_instance_sets_building_tasks_error_scheduling(self): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=None, task_state=task_states.SCHEDULING) self._test_init_instance_sets_building_tasks_error(instance) def test_init_instance_sets_building_tasks_error_block_device(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = None instance.task_state = task_states.BLOCK_DEVICE_MAPPING self._test_init_instance_sets_building_tasks_error(instance) def test_init_instance_sets_building_tasks_error_networking(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = None instance.task_state = task_states.NETWORKING self._test_init_instance_sets_building_tasks_error(instance) def test_init_instance_sets_building_tasks_error_spawning(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = None instance.task_state = task_states.SPAWNING self._test_init_instance_sets_building_tasks_error(instance) def _test_init_instance_cleans_image_states(self, instance): with mock.patch.object(instance, 'save') as save: self.compute._get_power_state = mock.Mock() instance.info_cache = None instance.power_state = power_state.RUNNING instance.host = self.compute.host self.compute._init_instance(self.context, instance) save.assert_called_once_with() self.assertIsNone(instance.task_state) @mock.patch('nova.compute.manager.ComputeManager._get_power_state', return_value=power_state.RUNNING) @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def _test_init_instance_cleans_task_states(self, powerstate, state, mock_get_uuid, mock_get_power_state): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.info_cache = None instance.power_state = power_state.RUNNING instance.vm_state = vm_states.ACTIVE instance.task_state = state instance.host = self.compute.host mock_get_power_state.return_value = powerstate self.compute._init_instance(self.context, instance) return instance def test_init_instance_cleans_image_state_pending_upload(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.IMAGE_PENDING_UPLOAD self._test_init_instance_cleans_image_states(instance) def test_init_instance_cleans_image_state_uploading(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.IMAGE_UPLOADING self._test_init_instance_cleans_image_states(instance) def test_init_instance_cleans_image_state_snapshot(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.IMAGE_SNAPSHOT self._test_init_instance_cleans_image_states(instance) def test_init_instance_cleans_image_state_snapshot_pending(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.IMAGE_SNAPSHOT_PENDING self._test_init_instance_cleans_image_states(instance) @mock.patch.object(objects.Instance, 'save') def test_init_instance_cleans_running_pausing(self, mock_save): instance = self._test_init_instance_cleans_task_states( power_state.RUNNING, task_states.PAUSING) mock_save.assert_called_once_with() self.assertEqual(vm_states.ACTIVE, instance.vm_state) self.assertIsNone(instance.task_state) @mock.patch.object(objects.Instance, 'save') def test_init_instance_cleans_running_unpausing(self, mock_save): instance = self._test_init_instance_cleans_task_states( power_state.RUNNING, task_states.UNPAUSING) mock_save.assert_called_once_with() self.assertEqual(vm_states.ACTIVE, instance.vm_state) self.assertIsNone(instance.task_state) @mock.patch('nova.compute.manager.ComputeManager.unpause_instance') def test_init_instance_cleans_paused_unpausing(self, mock_unpause): def fake_unpause(context, instance): instance.task_state = None mock_unpause.side_effect = fake_unpause instance = self._test_init_instance_cleans_task_states( power_state.PAUSED, task_states.UNPAUSING) mock_unpause.assert_called_once_with(self.context, instance) self.assertEqual(vm_states.ACTIVE, instance.vm_state) self.assertIsNone(instance.task_state) def test_init_instance_deletes_error_deleting_instance(self): instance = fake_instance.fake_instance_obj( self.context, project_id=fakes.FAKE_PROJECT_ID, uuid=uuids.instance, vcpus=1, memory_mb=64, vm_state=vm_states.ERROR, host=self.compute.host, task_state=task_states.DELETING) bdms = [] with test.nested( mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid', return_value=bdms), mock.patch.object(self.compute, '_delete_instance'), mock.patch.object(instance, 'obj_load_attr') ) as (mock_get, mock_delete, mock_load): self.compute._init_instance(self.context, instance) mock_get.assert_called_once_with(self.context, instance.uuid) mock_delete.assert_called_once_with(self.context, instance, bdms) def test_init_instance_resize_prep(self): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_states.ACTIVE, host=self.compute.host, task_state=task_states.RESIZE_PREP, power_state=power_state.RUNNING) with test.nested( mock.patch.object(self.compute, '_get_power_state', return_value=power_state.RUNNING), mock.patch.object(objects.Instance, 'get_network_info'), mock.patch.object(instance, 'save', autospec=True) ) as (mock_get_power_state, mock_nw_info, mock_instance_save): self.compute._init_instance(self.context, instance) mock_instance_save.assert_called_once_with() self.assertIsNone(instance.task_state) @mock.patch('nova.virt.fake.FakeDriver.power_off') @mock.patch.object(compute_utils, 'get_value_from_system_metadata', return_value=CONF.shutdown_timeout) def test_power_off_values(self, mock_get_metadata, mock_power_off): self.flags(shutdown_retry_interval=20, group='compute') instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_states.ACTIVE, task_state=task_states.POWERING_OFF) self.compute._power_off_instance(instance, clean_shutdown=True) mock_power_off.assert_called_once_with( instance, CONF.shutdown_timeout, 20) @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.objects.Instance.get_network_info') @mock.patch( 'nova.compute.manager.ComputeManager._get_instance_block_device_info') @mock.patch('nova.virt.driver.ComputeDriver.destroy') @mock.patch('nova.virt.fake.FakeDriver.get_volume_connector') @mock.patch('nova.compute.utils.notify_about_instance_action') @mock.patch( 'nova.compute.manager.ComputeManager._notify_about_instance_usage') def test_shutdown_instance_versioned_notifications(self, mock_notify_unversioned, mock_notify, mock_connector, mock_destroy, mock_blk_device_info, mock_nw_info, mock_elevated): mock_elevated.return_value = self.context instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_states.ERROR, task_state=task_states.DELETING) bdms = [mock.Mock(id=1, is_volume=True)] self.compute._shutdown_instance(self.context, instance, bdms, notify=True, try_deallocate_networks=False) mock_notify.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', action='shutdown', phase='start', bdms=bdms), mock.call(self.context, instance, 'fake-mini', action='shutdown', phase='end', bdms=bdms)]) @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.objects.Instance.get_network_info') @mock.patch( 'nova.compute.manager.ComputeManager._get_instance_block_device_info') @mock.patch('nova.virt.driver.ComputeDriver.destroy') @mock.patch('nova.virt.fake.FakeDriver.get_volume_connector') def _test_shutdown_instance_exception(self, exc, mock_connector, mock_destroy, mock_blk_device_info, mock_nw_info, mock_elevated): mock_connector.side_effect = exc mock_elevated.return_value = self.context instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance, vm_state=vm_states.ERROR, task_state=task_states.DELETING) bdms = [mock.Mock(id=1, is_volume=True, attachment_id=None)] self.compute._shutdown_instance(self.context, instance, bdms, notify=False, try_deallocate_networks=False) mock_connector.assert_called_once_with(instance) def test_shutdown_instance_endpoint_not_found(self): exc = cinder_exception.EndpointNotFound self._test_shutdown_instance_exception(exc) def test_shutdown_instance_client_exception(self): exc = cinder_exception.ClientException(code=9001) self._test_shutdown_instance_exception(exc) def test_shutdown_instance_volume_not_found(self): exc = exception.VolumeNotFound(volume_id=42) self._test_shutdown_instance_exception(exc) def test_shutdown_instance_disk_not_found(self): exc = exception.DiskNotFound(location="not\\here") self._test_shutdown_instance_exception(exc) def test_shutdown_instance_other_exception(self): exc = Exception('some other exception') self._test_shutdown_instance_exception(exc) def _test_init_instance_retries_reboot(self, instance, reboot_type, return_power_state): instance.host = self.compute.host with test.nested( mock.patch.object(self.compute, '_get_power_state', return_value=return_power_state), mock.patch.object(self.compute, 'reboot_instance'), mock.patch.object(objects.Instance, 'get_network_info') ) as ( _get_power_state, reboot_instance, get_network_info ): self.compute._init_instance(self.context, instance) call = mock.call(self.context, instance, block_device_info=None, reboot_type=reboot_type) reboot_instance.assert_has_calls([call]) def test_init_instance_retries_reboot_pending(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_PENDING for state in vm_states.ALLOW_SOFT_REBOOT: instance.vm_state = state self._test_init_instance_retries_reboot(instance, 'SOFT', power_state.RUNNING) def test_init_instance_retries_reboot_pending_hard(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_PENDING_HARD for state in vm_states.ALLOW_HARD_REBOOT: # NOTE(dave-mcnally) while a reboot of a vm in error state is # possible we don't attempt to recover an error during init if state == vm_states.ERROR: continue instance.vm_state = state self._test_init_instance_retries_reboot(instance, 'HARD', power_state.RUNNING) def test_init_instance_retries_reboot_pending_soft_became_hard(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_PENDING for state in vm_states.ALLOW_HARD_REBOOT: # NOTE(dave-mcnally) while a reboot of a vm in error state is # possible we don't attempt to recover an error during init if state == vm_states.ERROR: continue instance.vm_state = state with mock.patch.object(instance, 'save'): self._test_init_instance_retries_reboot(instance, 'HARD', power_state.SHUTDOWN) self.assertEqual(task_states.REBOOT_PENDING_HARD, instance.task_state) def test_init_instance_retries_reboot_started(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.REBOOT_STARTED with mock.patch.object(instance, 'save'): self._test_init_instance_retries_reboot(instance, 'HARD', power_state.NOSTATE) def test_init_instance_retries_reboot_started_hard(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.REBOOT_STARTED_HARD self._test_init_instance_retries_reboot(instance, 'HARD', power_state.NOSTATE) def _test_init_instance_cleans_reboot_state(self, instance): instance.host = self.compute.host with test.nested( mock.patch.object(self.compute, '_get_power_state', return_value=power_state.RUNNING), mock.patch.object(instance, 'save', autospec=True), mock.patch.object(objects.Instance, 'get_network_info') ) as ( _get_power_state, instance_save, get_network_info ): self.compute._init_instance(self.context, instance) instance_save.assert_called_once_with() self.assertIsNone(instance.task_state) self.assertEqual(vm_states.ACTIVE, instance.vm_state) def test_init_instance_cleans_image_state_reboot_started(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.REBOOT_STARTED instance.power_state = power_state.RUNNING self._test_init_instance_cleans_reboot_state(instance) def test_init_instance_cleans_image_state_reboot_started_hard(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.REBOOT_STARTED_HARD instance.power_state = power_state.RUNNING self._test_init_instance_cleans_reboot_state(instance) def test_init_instance_retries_power_off(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.POWERING_OFF instance.host = self.compute.host with mock.patch.object(self.compute, 'stop_instance'): self.compute._init_instance(self.context, instance) call = mock.call(self.context, instance, True) self.compute.stop_instance.assert_has_calls([call]) def test_init_instance_retries_power_on(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.POWERING_ON instance.host = self.compute.host with mock.patch.object(self.compute, 'start_instance'): self.compute._init_instance(self.context, instance) call = mock.call(self.context, instance) self.compute.start_instance.assert_has_calls([call]) def test_init_instance_retries_power_on_silent_exception(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.POWERING_ON instance.host = self.compute.host with mock.patch.object(self.compute, 'start_instance', return_value=Exception): init_return = self.compute._init_instance(self.context, instance) call = mock.call(self.context, instance) self.compute.start_instance.assert_has_calls([call]) self.assertIsNone(init_return) def test_init_instance_retries_power_off_silent_exception(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.POWERING_OFF instance.host = self.compute.host with mock.patch.object(self.compute, 'stop_instance', return_value=Exception): init_return = self.compute._init_instance(self.context, instance) call = mock.call(self.context, instance, True) self.compute.stop_instance.assert_has_calls([call]) self.assertIsNone(init_return) def test_get_power_state(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.STOPPED instance.task_state = None instance.host = self.compute.host with mock.patch.object(self.compute.driver, 'get_info') as mock_info: mock_info.return_value = hardware.InstanceInfo( state=power_state.SHUTDOWN) res = self.compute._get_power_state(instance) mock_info.assert_called_once_with(instance, use_cache=False) self.assertEqual(res, power_state.SHUTDOWN) @mock.patch('nova.objects.InstanceList.get_by_filters') def test_get_instances_on_driver(self, mock_instance_list): driver_instances = [] for x in range(10): driver_instances.append(fake_instance.fake_db_instance()) def _make_instance_list(db_list): return instance_obj._make_instance_list( self.context, objects.InstanceList(), db_list, None) driver_uuids = [inst['uuid'] for inst in driver_instances] mock_instance_list.return_value = _make_instance_list(driver_instances) with mock.patch.object(self.compute.driver, 'list_instance_uuids') as mock_driver_uuids: mock_driver_uuids.return_value = driver_uuids result = self.compute._get_instances_on_driver(self.context) self.assertEqual([x['uuid'] for x in driver_instances], [x['uuid'] for x in result]) expected_filters = {'uuid': driver_uuids} mock_instance_list.assert_called_with(self.context, expected_filters, use_slave=True) @mock.patch('nova.objects.InstanceList.get_by_filters') def test_get_instances_on_driver_empty(self, mock_instance_list): with mock.patch.object(self.compute.driver, 'list_instance_uuids') as mock_driver_uuids: mock_driver_uuids.return_value = [] result = self.compute._get_instances_on_driver(self.context) # Short circuit DB call, get_by_filters should not be called self.assertEqual(0, mock_instance_list.call_count) self.assertEqual(1, mock_driver_uuids.call_count) self.assertEqual([], [x['uuid'] for x in result]) @mock.patch('nova.objects.InstanceList.get_by_filters') def test_get_instances_on_driver_fallback(self, mock_instance_list): # Test getting instances when driver doesn't support # 'list_instance_uuids' self.compute.host = 'host' filters = {} self.flags(instance_name_template='inst-%i') all_instances = [] driver_instances = [] for x in range(10): instance = fake_instance.fake_db_instance(name='inst-%i' % x, id=x) if x % 2: driver_instances.append(instance) all_instances.append(instance) def _make_instance_list(db_list): return instance_obj._make_instance_list( self.context, objects.InstanceList(), db_list, None) driver_instance_names = [inst['name'] for inst in driver_instances] mock_instance_list.return_value = _make_instance_list(all_instances) with test.nested( mock.patch.object(self.compute.driver, 'list_instance_uuids'), mock.patch.object(self.compute.driver, 'list_instances') ) as ( mock_driver_uuids, mock_driver_instances ): mock_driver_uuids.side_effect = NotImplementedError() mock_driver_instances.return_value = driver_instance_names result = self.compute._get_instances_on_driver(self.context, filters) self.assertEqual([x['uuid'] for x in driver_instances], [x['uuid'] for x in result]) expected_filters = {'host': self.compute.host} mock_instance_list.assert_called_with(self.context, expected_filters, use_slave=True) @mock.patch.object(compute_utils, 'notify_usage_exists') @mock.patch.object(objects.TaskLog, 'end_task') @mock.patch.object(objects.TaskLog, 'begin_task') @mock.patch.object(objects.InstanceList, 'get_active_by_window_joined') @mock.patch.object(objects.TaskLog, 'get') def test_instance_usage_audit(self, mock_get, mock_get_active, mock_begin, mock_end, mock_notify): instances = [objects.Instance(uuid=uuids.instance)] def fake_task_log(*a, **k): pass def fake_get(*a, **k): return instances mock_get.side_effect = fake_task_log mock_get_active.side_effect = fake_get mock_begin.side_effect = fake_task_log mock_end.side_effect = fake_task_log self.flags(instance_usage_audit=True) self.compute._instance_usage_audit(self.context) mock_notify.assert_called_once_with( self.compute.notifier, self.context, instances[0], 'fake-mini', ignore_missing_network_data=False) self.assertTrue(mock_get.called) self.assertTrue(mock_get_active.called) self.assertTrue(mock_begin.called) self.assertTrue(mock_end.called) @mock.patch.object(objects.InstanceList, 'get_by_host') def test_sync_power_states(self, mock_get): instance = mock.Mock() mock_get.return_value = [instance] with mock.patch.object(self.compute._sync_power_pool, 'spawn_n') as mock_spawn: self.compute._sync_power_states(mock.sentinel.context) mock_get.assert_called_with(mock.sentinel.context, self.compute.host, expected_attrs=[], use_slave=True) mock_spawn.assert_called_once_with(mock.ANY, instance) @mock.patch('nova.objects.InstanceList.get_by_host', new=mock.Mock()) @mock.patch('nova.compute.manager.ComputeManager.' '_query_driver_power_state_and_sync', new_callable=mock.NonCallableMock) def test_sync_power_states_virt_driver_not_ready(self, _mock_sync): """"Tests that the periodic task exits early if the driver raises VirtDriverNotReady. """ with mock.patch.object( self.compute.driver, 'get_num_instances', side_effect=exception.VirtDriverNotReady) as gni: self.compute._sync_power_states(mock.sentinel.context) gni.assert_called_once_with() def _get_sync_instance(self, power_state, vm_state, task_state=None, shutdown_terminate=False): instance = objects.Instance() instance.uuid = uuids.instance instance.power_state = power_state instance.vm_state = vm_state instance.host = self.compute.host instance.task_state = task_state instance.shutdown_terminate = shutdown_terminate return instance @mock.patch.object(objects.Instance, 'refresh') def test_sync_instance_power_state_match(self, mock_refresh): instance = self._get_sync_instance(power_state.RUNNING, vm_states.ACTIVE) self.compute._sync_instance_power_state(self.context, instance, power_state.RUNNING) mock_refresh.assert_called_once_with(use_slave=False) @mock.patch.object(fake_driver.FakeDriver, 'get_info') @mock.patch.object(objects.Instance, 'refresh') @mock.patch.object(objects.Instance, 'save') def test_sync_instance_power_state_running_stopped(self, mock_save, mock_refresh, mock_get_info): mock_get_info.return_value = hardware.InstanceInfo( state=power_state.SHUTDOWN) instance = self._get_sync_instance(power_state.RUNNING, vm_states.ACTIVE) self.compute._sync_instance_power_state(self.context, instance, power_state.SHUTDOWN) self.assertEqual(instance.power_state, power_state.SHUTDOWN) mock_refresh.assert_called_once_with(use_slave=False) self.assertTrue(mock_save.called) mock_get_info.assert_called_once_with(instance, use_cache=False) def _test_sync_to_stop(self, vm_power_state, vm_state, driver_power_state, stop=True, force=False, shutdown_terminate=False): instance = self._get_sync_instance( vm_power_state, vm_state, shutdown_terminate=shutdown_terminate) with test.nested( mock.patch.object(objects.Instance, 'refresh'), mock.patch.object(objects.Instance, 'save'), mock.patch.object(self.compute.compute_api, 'stop'), mock.patch.object(self.compute.compute_api, 'delete'), mock.patch.object(self.compute.compute_api, 'force_stop'), mock.patch.object(self.compute.driver, 'get_info') ) as (mock_refresh, mock_save, mock_stop, mock_delete, mock_force, mock_get_info): mock_get_info.return_value = hardware.InstanceInfo( state=driver_power_state) self.compute._sync_instance_power_state(self.context, instance, driver_power_state) if shutdown_terminate: mock_delete.assert_called_once_with(self.context, instance) elif stop: if force: mock_force.assert_called_once_with(self.context, instance) else: mock_stop.assert_called_once_with(self.context, instance) if (vm_state == vm_states.ACTIVE and vm_power_state in (power_state.SHUTDOWN, power_state.CRASHED)): mock_get_info.assert_called_once_with(instance, use_cache=False) mock_refresh.assert_called_once_with(use_slave=False) self.assertTrue(mock_save.called) def test_sync_instance_power_state_to_stop(self): for ps in (power_state.SHUTDOWN, power_state.CRASHED, power_state.SUSPENDED): self._test_sync_to_stop(power_state.RUNNING, vm_states.ACTIVE, ps) for ps in (power_state.SHUTDOWN, power_state.CRASHED): self._test_sync_to_stop(power_state.PAUSED, vm_states.PAUSED, ps, force=True) self._test_sync_to_stop(power_state.SHUTDOWN, vm_states.STOPPED, power_state.RUNNING, force=True) def test_sync_instance_power_state_to_terminate(self): self._test_sync_to_stop(power_state.RUNNING, vm_states.ACTIVE, power_state.SHUTDOWN, force=False, shutdown_terminate=True) def test_sync_instance_power_state_to_no_stop(self): for ps in (power_state.PAUSED, power_state.NOSTATE): self._test_sync_to_stop(power_state.RUNNING, vm_states.ACTIVE, ps, stop=False) for vs in (vm_states.SOFT_DELETED, vm_states.DELETED): for ps in (power_state.NOSTATE, power_state.SHUTDOWN): self._test_sync_to_stop(power_state.RUNNING, vs, ps, stop=False) @mock.patch('nova.compute.manager.ComputeManager.' '_sync_instance_power_state') def test_query_driver_power_state_and_sync_pending_task( self, mock_sync_power_state): with mock.patch.object(self.compute.driver, 'get_info') as mock_get_info: db_instance = objects.Instance(uuid=uuids.db_instance, task_state=task_states.POWERING_OFF) self.compute._query_driver_power_state_and_sync(self.context, db_instance) self.assertFalse(mock_get_info.called) self.assertFalse(mock_sync_power_state.called) @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch('nova.compute.manager.ComputeManager.' '_sync_instance_power_state') def test_query_driver_power_state_and_sync_not_found_driver( self, mock_sync_power_state, mock_claim): error = exception.InstanceNotFound(instance_id=1) with mock.patch.object(self.compute.driver, 'get_info', side_effect=error) as mock_get_info: db_instance = objects.Instance(uuid=uuids.db_instance, task_state=None) self.compute._query_driver_power_state_and_sync(self.context, db_instance) mock_get_info.assert_called_once_with(db_instance) mock_sync_power_state.assert_called_once_with(self.context, db_instance, power_state.NOSTATE, use_slave=True) def test_cleanup_running_deleted_instances_virt_driver_not_ready(self): """Tests the scenario that the driver raises VirtDriverNotReady when listing instances so the task returns early. """ self.flags(running_deleted_instance_action='shutdown') with mock.patch.object(self.compute, '_running_deleted_instances', side_effect=exception.VirtDriverNotReady) as ls: # Mock the virt driver to make sure nothing calls it. with mock.patch.object(self.compute, 'driver', new_callable=mock.NonCallableMock): self.compute._cleanup_running_deleted_instances(self.context) ls.assert_called_once_with(test.MatchType(context.RequestContext)) @mock.patch.object(virt_driver.ComputeDriver, 'delete_instance_files') @mock.patch.object(objects.InstanceList, 'get_by_filters') def test_run_pending_deletes(self, mock_get, mock_delete): self.flags(instance_delete_interval=10) class FakeInstance(object): def __init__(self, uuid, name, smd): self.uuid = uuid self.name = name self.system_metadata = smd self.cleaned = False def __getitem__(self, name): return getattr(self, name) def save(self): pass def _fake_get(ctx, filter, expected_attrs, use_slave): mock_get.assert_called_once_with( {'read_deleted': 'yes'}, {'deleted': True, 'soft_deleted': False, 'host': 'fake-mini', 'cleaned': False}, expected_attrs=['system_metadata'], use_slave=True) return [a, b, c] a = FakeInstance(uuids.instanceA, 'apple', {'clean_attempts': '100'}) b = FakeInstance(uuids.instanceB, 'orange', {'clean_attempts': '3'}) c = FakeInstance(uuids.instanceC, 'banana', {}) mock_get.side_effect = _fake_get mock_delete.side_effect = [True, False] self.compute._run_pending_deletes({}) self.assertFalse(a.cleaned) self.assertEqual('100', a.system_metadata['clean_attempts']) self.assertTrue(b.cleaned) self.assertEqual('4', b.system_metadata['clean_attempts']) self.assertFalse(c.cleaned) self.assertEqual('1', c.system_metadata['clean_attempts']) mock_delete.assert_has_calls([mock.call(mock.ANY), mock.call(mock.ANY)]) @mock.patch.object(objects.Instance, 'mutated_migration_context') @mock.patch.object(objects.Migration, 'save') @mock.patch.object(objects.MigrationList, 'get_by_filters') @mock.patch.object(objects.InstanceList, 'get_by_filters') def _test_cleanup_incomplete_migrations(self, inst_host, mock_inst_get_by_filters, mock_migration_get_by_filters, mock_save, mock_mutated_mgr_ctxt): def fake_inst(context, uuid, host): inst = objects.Instance(context) inst.uuid = uuid inst.host = host return inst def fake_migration(uuid, status, inst_uuid, src_host, dest_host): migration = objects.Migration() migration.uuid = uuid migration.status = status migration.instance_uuid = inst_uuid migration.source_compute = src_host migration.dest_compute = dest_host return migration fake_instances = [fake_inst(self.context, uuids.instance_1, inst_host), fake_inst(self.context, uuids.instance_2, inst_host)] fake_migrations = [fake_migration(uuids.mig1, 'error', uuids.instance_1, 'fake-host', 'fake-mini'), fake_migration(uuids.mig2, 'error', uuids.instance_2, 'fake-host', 'fake-mini')] mock_migration_get_by_filters.return_value = fake_migrations mock_inst_get_by_filters.return_value = fake_instances with test.nested( mock.patch.object(self.compute.driver, 'delete_instance_files'), mock.patch.object(self.compute.driver, 'cleanup_lingering_instance_resources') ) as (mock_delete_files, mock_cleanup_resources): self.compute._cleanup_incomplete_migrations(self.context) # Ensure that migration status is set to 'failed' after instance # files deletion for those instances whose instance.host is not # same as compute host where periodic task is running. for inst in fake_instances: if inst.host != CONF.host: for mig in fake_migrations: if inst.uuid == mig.instance_uuid: self.assertEqual('failed', mig.status) def test_cleanup_incomplete_migrations_dest_node(self): """Test to ensure instance files are deleted from destination node. If instance gets deleted during resizing/revert-resizing operation, in that case instance files gets deleted from instance.host (source host here), but there is possibility that instance files could be present on destination node. This test ensures that `_cleanup_incomplete_migration` periodic task deletes orphaned instance files from destination compute node. """ self.flags(host='fake-mini') self._test_cleanup_incomplete_migrations('fake-host') def test_cleanup_incomplete_migrations_source_node(self): """Test to ensure instance files are deleted from source node. If instance gets deleted during resizing/revert-resizing operation, in that case instance files gets deleted from instance.host (dest host here), but there is possibility that instance files could be present on source node. This test ensures that `_cleanup_incomplete_migration` periodic task deletes orphaned instance files from source compute node. """ self.flags(host='fake-host') self._test_cleanup_incomplete_migrations('fake-mini') def test_attach_interface_failure(self): # Test that the fault methods are invoked when an attach fails db_instance = fake_instance.fake_db_instance() f_instance = objects.Instance._from_db_object(self.context, objects.Instance(), db_instance) f_instance.flavor = objects.Flavor() f_instance.system_metadata = {} f_instance.pci_requests = objects.InstancePCIRequests(requests=[]) e = exception.InterfaceAttachFailed(instance_uuid=f_instance.uuid) @mock.patch('nova.network.neutron.API.create_resource_requests') @mock.patch.object(self.compute, '_claim_pci_device_for_interface_attach', return_value=None) @mock.patch.object(compute_utils, 'EventReporter') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(self.compute.network_api, 'allocate_for_instance', side_effect=e) @mock.patch.object(self.compute, '_instance_update', side_effect=lambda *a, **k: {}) def do_test( update, meth, add_fault, notify, event, mock_claim_pci, mock_create_resource_req): mock_create_resource_req.return_value = ( None, [], mock.sentinel.req_lvl_params) self.assertRaises(exception.InterfaceAttachFailed, self.compute.attach_interface, self.context, f_instance, uuids.network_id, uuids.port_id, None, None) add_fault.assert_has_calls([ mock.call(self.context, f_instance, e, mock.ANY)]) event.assert_called_once_with( self.context, 'compute_attach_interface', CONF.host, f_instance.uuid, graceful_exit=False) with mock.patch.dict(self.compute.driver.capabilities, supports_attach_interface=True): do_test() def test_detach_interface_failure(self): # Test that the fault methods are invoked when a detach fails # Build test data that will cause a PortNotFound exception nw_info = network_model.NetworkInfo([]) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=uuids.instance) f_instance = objects.Instance(id=3, uuid=uuids.instance, info_cache=info_cache) @mock.patch.object(compute_utils, 'EventReporter') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(compute_utils, 'refresh_info_cache_for_instance') @mock.patch.object(self.compute, '_set_instance_obj_error_state') def do_test(meth, refresh, add_fault, event): self.assertRaises(exception.PortNotFound, self.compute.detach_interface, self.context, f_instance, 'port_id') add_fault.assert_has_calls( [mock.call(self.context, f_instance, mock.ANY, mock.ANY)]) event.assert_called_once_with( self.context, 'compute_detach_interface', CONF.host, f_instance.uuid, graceful_exit=False) refresh.assert_called_once_with(self.context, f_instance) do_test() @mock.patch('nova.compute.manager.LOG.log') @mock.patch.object(compute_utils, 'EventReporter') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(compute_utils, 'refresh_info_cache_for_instance') def test_detach_interface_instance_not_found(self, mock_refresh, mock_notify, mock_fault, mock_event, mock_log): nw_info = network_model.NetworkInfo([ network_model.VIF(uuids.port_id)]) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=uuids.instance) instance = objects.Instance(id=1, uuid=uuids.instance, info_cache=info_cache) with mock.patch.object(self.compute.driver, 'detach_interface', side_effect=exception.InstanceNotFound( instance_id=uuids.instance)): self.assertRaises(exception.InterfaceDetachFailed, self.compute.detach_interface, self.context, instance, uuids.port_id) mock_refresh.assert_called_once_with(self.context, instance) self.assertEqual(1, mock_log.call_count) self.assertEqual(logging.DEBUG, mock_log.call_args[0][0]) @mock.patch.object(compute_utils, 'EventReporter') @mock.patch.object(virt_driver.ComputeDriver, 'get_volume_connector', return_value={}) @mock.patch.object(manager.ComputeManager, '_instance_update', return_value={}) @mock.patch.object(db, 'instance_fault_create') @mock.patch.object(db, 'block_device_mapping_update') @mock.patch.object(db, 'block_device_mapping_get_by_instance_and_volume_id') @mock.patch.object(cinder.API, 'migrate_volume_completion') @mock.patch.object(cinder.API, 'terminate_connection') @mock.patch.object(cinder.API, 'unreserve_volume') @mock.patch.object(cinder.API, 'get') @mock.patch.object(cinder.API, 'roll_detaching') @mock.patch.object(compute_utils, 'notify_about_volume_swap') def _test_swap_volume(self, mock_notify, mock_roll_detaching, mock_cinder_get, mock_unreserve_volume, mock_terminate_connection, mock_migrate_volume_completion, mock_bdm_get, mock_bdm_update, mock_instance_fault_create, mock_instance_update, mock_get_volume_connector, mock_event, expected_exception=None): # This test ensures that volume_id arguments are passed to volume_api # and that volume states are OK volumes = {} volumes[uuids.old_volume] = {'id': uuids.old_volume, 'display_name': 'old_volume', 'status': 'detaching', 'size': 1} volumes[uuids.new_volume] = {'id': uuids.new_volume, 'display_name': 'new_volume', 'status': 'available', 'size': 2} fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'device_name': '/dev/vdb', 'source_type': 'volume', 'destination_type': 'volume', 'instance_uuid': uuids.instance, 'connection_info': '{"foo": "bar"}', 'volume_id': uuids.old_volume, 'attachment_id': None}) def fake_vol_api_roll_detaching(context, volume_id): self.assertTrue(uuidutils.is_uuid_like(volume_id)) if volumes[volume_id]['status'] == 'detaching': volumes[volume_id]['status'] = 'in-use' def fake_vol_api_func(context, volume, *args): self.assertTrue(uuidutils.is_uuid_like(volume)) return {} def fake_vol_get(context, volume_id): self.assertTrue(uuidutils.is_uuid_like(volume_id)) return volumes[volume_id] def fake_vol_unreserve(context, volume_id): self.assertTrue(uuidutils.is_uuid_like(volume_id)) if volumes[volume_id]['status'] == 'attaching': volumes[volume_id]['status'] = 'available' def fake_vol_migrate_volume_completion(context, old_volume_id, new_volume_id, error=False): self.assertTrue(uuidutils.is_uuid_like(old_volume_id)) self.assertTrue(uuidutils.is_uuid_like(new_volume_id)) volumes[old_volume_id]['status'] = 'in-use' return {'save_volume_id': new_volume_id} def fake_block_device_mapping_update(ctxt, id, updates, legacy): self.assertEqual(2, updates['volume_size']) return fake_bdm mock_roll_detaching.side_effect = fake_vol_api_roll_detaching mock_terminate_connection.side_effect = fake_vol_api_func mock_cinder_get.side_effect = fake_vol_get mock_migrate_volume_completion.side_effect = ( fake_vol_migrate_volume_completion) mock_unreserve_volume.side_effect = fake_vol_unreserve mock_bdm_get.return_value = fake_bdm mock_bdm_update.side_effect = fake_block_device_mapping_update mock_instance_fault_create.return_value = ( test_instance_fault.fake_faults['fake-uuid'][0]) instance1 = fake_instance.fake_instance_obj( self.context, **{'uuid': uuids.instance}) if expected_exception: volumes[uuids.old_volume]['status'] = 'detaching' volumes[uuids.new_volume]['status'] = 'attaching' self.assertRaises(expected_exception, self.compute.swap_volume, self.context, uuids.old_volume, uuids.new_volume, instance1, None) self.assertEqual('in-use', volumes[uuids.old_volume]['status']) self.assertEqual('available', volumes[uuids.new_volume]['status']) self.assertEqual(2, mock_notify.call_count) mock_notify.assert_any_call( test.MatchType(context.RequestContext), instance1, self.compute.host, fields.NotificationPhase.START, uuids.old_volume, uuids.new_volume) mock_notify.assert_any_call( test.MatchType(context.RequestContext), instance1, self.compute.host, fields.NotificationPhase.ERROR, uuids.old_volume, uuids.new_volume, test.MatchType(expected_exception)) else: self.compute.swap_volume(self.context, uuids.old_volume, uuids.new_volume, instance1, None) self.assertEqual(volumes[uuids.old_volume]['status'], 'in-use') self.assertEqual(2, mock_notify.call_count) mock_notify.assert_any_call(test.MatchType(context.RequestContext), instance1, self.compute.host, fields.NotificationPhase.START, uuids.old_volume, uuids.new_volume) mock_notify.assert_any_call(test.MatchType(context.RequestContext), instance1, self.compute.host, fields.NotificationPhase.END, uuids.old_volume, uuids.new_volume) mock_event.assert_called_once_with(self.context, 'compute_swap_volume', CONF.host, instance1.uuid, graceful_exit=False) def _assert_volume_api(self, context, volume, *args): self.assertTrue(uuidutils.is_uuid_like(volume)) return {} def _assert_swap_volume(self, context, old_connection_info, new_connection_info, instance, mountpoint, resize_to): self.assertEqual(2, resize_to) @mock.patch.object(cinder.API, 'initialize_connection') @mock.patch.object(fake_driver.FakeDriver, 'swap_volume') def test_swap_volume_volume_api_usage(self, mock_swap_volume, mock_initialize_connection): mock_initialize_connection.side_effect = self._assert_volume_api mock_swap_volume.side_effect = self._assert_swap_volume self._test_swap_volume() @mock.patch.object(cinder.API, 'initialize_connection') @mock.patch.object(fake_driver.FakeDriver, 'swap_volume', side_effect=test.TestingException()) def test_swap_volume_with_compute_driver_exception( self, mock_swap_volume, mock_initialize_connection): mock_initialize_connection.side_effect = self._assert_volume_api self._test_swap_volume(expected_exception=test.TestingException) @mock.patch.object(cinder.API, 'initialize_connection', side_effect=test.TestingException()) @mock.patch.object(fake_driver.FakeDriver, 'swap_volume') def test_swap_volume_with_initialize_connection_exception( self, mock_swap_volume, mock_initialize_connection): self._test_swap_volume(expected_exception=test.TestingException) @mock.patch('nova.compute.utils.notify_about_volume_swap') @mock.patch( 'nova.db.main.api.block_device_mapping_get_by_instance_and_volume_id') @mock.patch('nova.db.main.api.block_device_mapping_update') @mock.patch('nova.volume.cinder.API.get') @mock.patch('nova.virt.libvirt.LibvirtDriver.get_volume_connector') @mock.patch('nova.compute.manager.ComputeManager._swap_volume') def test_swap_volume_delete_on_termination_flag(self, swap_volume_mock, volume_connector_mock, get_volume_mock, update_bdm_mock, get_bdm_mock, notify_mock): # This test ensures that delete_on_termination flag arguments # are reserved volumes = {} old_volume_id = uuids.fake volumes[old_volume_id] = {'id': old_volume_id, 'display_name': 'old_volume', 'status': 'detaching', 'size': 2} new_volume_id = uuids.fake_2 volumes[new_volume_id] = {'id': new_volume_id, 'display_name': 'new_volume', 'status': 'available', 'size': 2} fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'device_name': '/dev/vdb', 'source_type': 'volume', 'destination_type': 'volume', 'instance_uuid': uuids.instance, 'delete_on_termination': True, 'connection_info': '{"foo": "bar"}', 'attachment_id': None}) comp_ret = {'save_volume_id': old_volume_id} new_info = {"foo": "bar", "serial": old_volume_id} swap_volume_mock.return_value = (comp_ret, new_info) volume_connector_mock.return_value = {} update_bdm_mock.return_value = fake_bdm get_bdm_mock.return_value = fake_bdm get_volume_mock.return_value = volumes[old_volume_id] self.compute.swap_volume(self.context, old_volume_id, new_volume_id, fake_instance.fake_instance_obj(self.context, **{'uuid': uuids.instance}), None) update_values = {'no_device': False, 'connection_info': jsonutils.dumps(new_info), 'volume_id': old_volume_id, 'source_type': u'volume', 'snapshot_id': None, 'destination_type': u'volume'} update_bdm_mock.assert_called_once_with(mock.ANY, mock.ANY, update_values, legacy=False) @mock.patch.object(compute_utils, 'notify_about_volume_swap') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') @mock.patch('nova.volume.cinder.API.get') @mock.patch('nova.volume.cinder.API.attachment_update') @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.attachment_complete') @mock.patch('nova.volume.cinder.API.migrate_volume_completion', return_value={'save_volume_id': uuids.old_volume_id}) def test_swap_volume_with_new_attachment_id_cinder_migrate_true( self, migrate_volume_completion, attachment_complete, attachment_delete, attachment_update, get_volume, get_bdm, notify_about_volume_swap): """Tests a swap volume operation with a new style volume attachment passed in from the compute API, and the case that Cinder initiated the swap volume because of a volume retype situation. This is a happy path test. Since it is a retype there is no volume size change. """ bdm = objects.BlockDeviceMapping( volume_id=uuids.old_volume_id, device_name='/dev/vda', attachment_id=uuids.old_attachment_id, connection_info='{"data": {}}', volume_size=1) old_volume = { 'id': uuids.old_volume_id, 'size': 1, 'status': 'retyping', 'migration_status': 'migrating', 'multiattach': False } new_volume = { 'id': uuids.new_volume_id, 'size': 1, 'status': 'reserved', 'migration_status': 'migrating', 'multiattach': False } attachment_update.return_value = {"connection_info": {"data": {}}} get_bdm.return_value = bdm get_volume.side_effect = (old_volume, new_volume) instance = fake_instance.fake_instance_obj(self.context) with test.nested( mock.patch.object(self.context, 'elevated', return_value=self.context), mock.patch.object(self.compute.driver, 'get_volume_connector', return_value=mock.sentinel.connector), mock.patch.object(bdm, 'save') ) as ( mock_elevated, mock_get_volume_connector, mock_save ): self.compute.swap_volume( self.context, uuids.old_volume_id, uuids.new_volume_id, instance, uuids.new_attachment_id) # Assert the expected calls. get_bdm.assert_called_once_with( self.context, uuids.old_volume_id, instance.uuid) # We updated the new attachment with the host connector. attachment_update.assert_called_once_with( self.context, uuids.new_attachment_id, mock.sentinel.connector, bdm.device_name) # We tell Cinder that the new volume is connected attachment_complete.assert_called_once_with( self.context, uuids.new_attachment_id) # After a successful swap volume, we deleted the old attachment. attachment_delete.assert_called_once_with( self.context, uuids.old_attachment_id) # After a successful swap volume, we tell Cinder so it can complete # the retype operation. migrate_volume_completion.assert_called_once_with( self.context, uuids.old_volume_id, uuids.new_volume_id, error=False) # The BDM should have been updated. Since it's a retype, the old # volume ID is returned from Cinder so that's what goes into the # BDM but the new attachment ID is saved. mock_save.assert_called_once_with() self.assertEqual(uuids.old_volume_id, bdm.volume_id) self.assertEqual(uuids.new_attachment_id, bdm.attachment_id) self.assertEqual(1, bdm.volume_size) new_conn_info = jsonutils.loads(bdm.connection_info) self.assertEqual(uuids.old_volume_id, new_conn_info['serial']) self.assertEqual(uuids.old_volume_id, new_conn_info['volume_id']) self.assertEqual( uuids.old_volume_id, new_conn_info['data']['volume_id']) @mock.patch.object(compute_utils, 'notify_about_volume_swap') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') @mock.patch('nova.volume.cinder.API.get') @mock.patch('nova.volume.cinder.API.attachment_update') @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.attachment_complete') @mock.patch('nova.volume.cinder.API.migrate_volume_completion') def test_swap_volume_with_new_attachment_id_cinder_migrate_false( self, migrate_volume_completion, attachment_complete, attachment_delete, attachment_update, get_volume, get_bdm, notify_about_volume_swap): """Tests a swap volume operation with a new style volume attachment passed in from the compute API, and the case that Cinder did not initiate the swap volume. This is a happy path test. Since it is not a retype we also change the size. """ bdm = objects.BlockDeviceMapping( volume_id=uuids.old_volume_id, device_name='/dev/vda', attachment_id=uuids.old_attachment_id, connection_info='{"data": {}}') old_volume = { 'id': uuids.old_volume_id, 'size': 1, 'status': 'detaching', 'multiattach': False } new_volume = { 'id': uuids.new_volume_id, 'size': 2, 'status': 'reserved', 'multiattach': False } attachment_update.return_value = {"connection_info": {"data": {}}} get_bdm.return_value = bdm get_volume.side_effect = (old_volume, new_volume) instance = fake_instance.fake_instance_obj(self.context) with test.nested( mock.patch.object(self.context, 'elevated', return_value=self.context), mock.patch.object(self.compute.driver, 'get_volume_connector', return_value=mock.sentinel.connector), mock.patch.object(bdm, 'save') ) as ( mock_elevated, mock_get_volume_connector, mock_save ): self.compute.swap_volume( self.context, uuids.old_volume_id, uuids.new_volume_id, instance, uuids.new_attachment_id) # Assert the expected calls. get_bdm.assert_called_once_with( self.context, uuids.old_volume_id, instance.uuid) # We updated the new attachment with the host connector. attachment_update.assert_called_once_with( self.context, uuids.new_attachment_id, mock.sentinel.connector, bdm.device_name) # We tell Cinder that the new volume is connected attachment_complete.assert_called_once_with( self.context, uuids.new_attachment_id) # After a successful swap volume, we deleted the old attachment. attachment_delete.assert_called_once_with( self.context, uuids.old_attachment_id) # After a successful swap volume, since it was not a # Cinder-initiated call, we don't call migrate_volume_completion. migrate_volume_completion.assert_not_called() # The BDM should have been updated. Since it's a not a retype, the # volume_id is now the new volume ID. mock_save.assert_called_once_with() self.assertEqual(uuids.new_volume_id, bdm.volume_id) self.assertEqual(uuids.new_attachment_id, bdm.attachment_id) self.assertEqual(2, bdm.volume_size) new_conn_info = jsonutils.loads(bdm.connection_info) self.assertEqual(uuids.new_volume_id, new_conn_info['serial']) self.assertEqual(uuids.new_volume_id, new_conn_info['volume_id']) self.assertEqual( uuids.new_volume_id, new_conn_info['data']['volume_id']) self.assertNotIn('multiattach', new_conn_info) @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(compute_utils, 'notify_about_volume_swap') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') @mock.patch('nova.volume.cinder.API.get') @mock.patch('nova.volume.cinder.API.attachment_update', side_effect=exception.VolumeAttachmentNotFound( attachment_id=uuids.new_attachment_id)) @mock.patch('nova.volume.cinder.API.roll_detaching') @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.migrate_volume_completion') def test_swap_volume_with_new_attachment_id_attachment_update_fails( self, migrate_volume_completion, attachment_delete, roll_detaching, attachment_update, get_volume, get_bdm, notify_about_volume_swap, add_instance_fault_from_exc): """Tests a swap volume operation with a new style volume attachment passed in from the compute API, and the case that Cinder initiated the swap volume because of a volume migrate situation. This is a negative test where attachment_update fails. """ bdm = objects.BlockDeviceMapping( volume_id=uuids.old_volume_id, device_name='/dev/vda', attachment_id=uuids.old_attachment_id, connection_info='{"data": {}}') old_volume = { 'id': uuids.old_volume_id, 'size': 1, 'status': 'in-use', 'migration_status': 'migrating', 'multiattach': False } new_volume = { 'id': uuids.new_volume_id, 'size': 1, 'status': 'reserved', 'migration_status': 'migrating', 'multiattach': False } get_bdm.return_value = bdm get_volume.side_effect = (old_volume, new_volume) instance = fake_instance.fake_instance_obj(self.context) with test.nested( mock.patch.object(self.context, 'elevated', return_value=self.context), mock.patch.object(self.compute.driver, 'get_volume_connector', return_value=mock.sentinel.connector) ) as ( mock_elevated, mock_get_volume_connector ): self.assertRaises( exception.VolumeAttachmentNotFound, self.compute.swap_volume, self.context, uuids.old_volume_id, uuids.new_volume_id, instance, uuids.new_attachment_id) # Assert the expected calls. get_bdm.assert_called_once_with( self.context, uuids.old_volume_id, instance.uuid) # We tried to update the new attachment with the host connector. attachment_update.assert_called_once_with( self.context, uuids.new_attachment_id, mock.sentinel.connector, bdm.device_name) # After a failure, we rollback the detaching status of the old # volume. roll_detaching.assert_called_once_with( self.context, uuids.old_volume_id) # After a failure, we deleted the new attachment. attachment_delete.assert_called_once_with( self.context, uuids.new_attachment_id) # After a failure for a Cinder-initiated swap volume, we called # migrate_volume_completion to let Cinder know things blew up. migrate_volume_completion.assert_called_once_with( self.context, uuids.old_volume_id, uuids.new_volume_id, error=True) @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(compute_utils, 'notify_about_volume_swap') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') @mock.patch('nova.volume.cinder.API.get') @mock.patch('nova.volume.cinder.API.attachment_update') @mock.patch('nova.volume.cinder.API.roll_detaching') @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.migrate_volume_completion') def test_swap_volume_with_new_attachment_id_driver_swap_fails( self, migrate_volume_completion, attachment_delete, roll_detaching, attachment_update, get_volume, get_bdm, notify_about_volume_swap, add_instance_fault_from_exc): """Tests a swap volume operation with a new style volume attachment passed in from the compute API, and the case that Cinder did not initiate the swap volume. This is a negative test where the compute driver swap_volume method fails. """ bdm = objects.BlockDeviceMapping( volume_id=uuids.old_volume_id, device_name='/dev/vda', attachment_id=uuids.old_attachment_id, connection_info='{"data": {}}') old_volume = { 'id': uuids.old_volume_id, 'size': 1, 'status': 'detaching', 'multiattach': False } new_volume = { 'id': uuids.new_volume_id, 'size': 2, 'status': 'reserved', 'multiattach': False } attachment_update.return_value = {"connection_info": {"data": {}}} get_bdm.return_value = bdm get_volume.side_effect = (old_volume, new_volume) instance = fake_instance.fake_instance_obj(self.context) with test.nested( mock.patch.object(self.context, 'elevated', return_value=self.context), mock.patch.object(self.compute.driver, 'get_volume_connector', return_value=mock.sentinel.connector), mock.patch.object(self.compute.driver, 'swap_volume', side_effect=test.TestingException('yikes')) ) as ( mock_elevated, mock_get_volume_connector, mock_driver_swap ): self.assertRaises( test.TestingException, self.compute.swap_volume, self.context, uuids.old_volume_id, uuids.new_volume_id, instance, uuids.new_attachment_id) # Assert the expected calls. # The new connection_info has the new_volume_id as the serial. new_cinfo = mock_driver_swap.call_args[0][2] self.assertIn('serial', new_cinfo) self.assertEqual(uuids.new_volume_id, new_cinfo['serial']) get_bdm.assert_called_once_with( self.context, uuids.old_volume_id, instance.uuid) # We updated the new attachment with the host connector. attachment_update.assert_called_once_with( self.context, uuids.new_attachment_id, mock.sentinel.connector, bdm.device_name) # After a failure, we rollback the detaching status of the old # volume. roll_detaching.assert_called_once_with( self.context, uuids.old_volume_id) # After a failed swap volume, we deleted the new attachment. attachment_delete.assert_called_once_with( self.context, uuids.new_attachment_id) # After a failed swap volume, since it was not a # Cinder-initiated call, we don't call migrate_volume_completion. migrate_volume_completion.assert_not_called() @mock.patch('nova.volume.cinder.API.attachment_update') def test_swap_volume_with_multiattach(self, attachment_update): """Tests swap volume where the volume being swapped-to supports multiattach as well as the compute driver, so the attachment for the new volume (created in the API) is updated with the host connector and the new_connection_info is updated with the multiattach flag. """ bdm = objects.BlockDeviceMapping( volume_id=uuids.old_volume_id, device_name='/dev/vda', attachment_id=uuids.old_attachment_id, connection_info='{"data": {}}') new_volume = { 'id': uuids.new_volume_id, 'size': 2, 'status': 'reserved', 'multiattach': True } attachment_update.return_value = {"connection_info": {"data": {}}} connector = mock.sentinel.connector with mock.patch.dict(self.compute.driver.capabilities, {'supports_multiattach': True}): _, new_cinfo = self.compute._init_volume_connection( self.context, new_volume, uuids.old_volume_id, connector, bdm, uuids.new_attachment_id, bdm.device_name) self.assertEqual(uuids.new_volume_id, new_cinfo['serial']) self.assertIn('multiattach', new_cinfo) self.assertTrue(new_cinfo['multiattach']) attachment_update.assert_called_once_with( self.context, uuids.new_attachment_id, connector, bdm.device_name) def test_swap_volume_with_multiattach_no_driver_support(self): """Tests a swap volume scenario where the new volume being swapped-to supports multiattach but the virt driver does not, so swap volume fails. """ bdm = objects.BlockDeviceMapping( volume_id=uuids.old_volume_id, device_name='/dev/vda', attachment_id=uuids.old_attachment_id, connection_info='{"data": {}}') new_volume = { 'id': uuids.new_volume_id, 'size': 2, 'status': 'reserved', 'multiattach': True } connector = {'host': 'localhost'} with mock.patch.dict(self.compute.driver.capabilities, {'supports_multiattach': False}): self.assertRaises(exception.MultiattachNotSupportedByVirtDriver, self.compute._init_volume_connection, self.context, new_volume, uuids.old_volume_id, connector, bdm, uuids.new_attachment_id, bdm.device_name) def test_live_migration_claim(self): mock_claim = mock.Mock() mock_claim.claimed_numa_topology = mock.sentinel.claimed_topology stub_image_meta = objects.ImageMeta() post_claim_md = migrate_data_obj.LiveMigrateData() with test.nested( mock.patch.object(self.compute.rt, 'live_migration_claim', return_value=mock_claim), mock.patch.object(self.compute, '_get_nodename', return_value='fake-dest-node'), mock.patch.object(objects.ImageMeta, 'from_instance', return_value=stub_image_meta), mock.patch.object(fake_driver.FakeDriver, 'post_claim_migrate_data', return_value=post_claim_md) ) as (mock_lm_claim, mock_get_nodename, mock_from_instance, mock_post_claim_migrate_data): instance = objects.Instance(flavor=objects.Flavor()) md = objects.LibvirtLiveMigrateData() migration = objects.Migration() self.assertEqual( post_claim_md, self.compute._live_migration_claim( self.context, instance, md, migration, mock.sentinel.limits, None)) mock_lm_claim.assert_called_once_with( self.context, instance, 'fake-dest-node', migration, mock.sentinel.limits, None) mock_post_claim_migrate_data.assert_called_once_with( self.context, instance, md, mock_claim) def test_live_migration_claim_claim_raises(self): stub_image_meta = objects.ImageMeta() with test.nested( mock.patch.object( self.compute.rt, 'live_migration_claim', side_effect=exception.ComputeResourcesUnavailable( reason='bork')), mock.patch.object(self.compute, '_get_nodename', return_value='fake-dest-node'), mock.patch.object(objects.ImageMeta, 'from_instance', return_value=stub_image_meta), mock.patch.object(fake_driver.FakeDriver, 'post_claim_migrate_data') ) as (mock_lm_claim, mock_get_nodename, mock_from_instance, mock_post_claim_migrate_data): instance = objects.Instance(flavor=objects.Flavor()) migration = objects.Migration() self.assertRaises( exception.MigrationPreCheckError, self.compute._live_migration_claim, self.context, instance, objects.LibvirtLiveMigrateData(), migration, mock.sentinel.limits, None) mock_lm_claim.assert_called_once_with( self.context, instance, 'fake-dest-node', migration, mock.sentinel.limits, None) mock_get_nodename.assert_called_once_with(instance) mock_post_claim_migrate_data.assert_not_called() def test_drop_move_claim_at_destination(self): instance = objects.Instance(flavor=objects.Flavor()) with test.nested( mock.patch.object(self.compute.rt, 'drop_move_claim'), mock.patch.object(self.compute, '_get_nodename', return_value='fake-node') ) as (mock_drop_move_claim, mock_get_nodename): self.compute.drop_move_claim_at_destination(self.context, instance) mock_get_nodename.assert_called_once_with(instance) mock_drop_move_claim.assert_called_once_with( self.context, instance, 'fake-node', flavor=instance.flavor) @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(fake_driver.FakeDriver, 'check_can_live_migrate_source') @mock.patch.object(manager.ComputeManager, '_get_instance_block_device_info') @mock.patch.object(compute_utils, 'is_volume_backed_instance') @mock.patch.object(compute_utils, 'EventReporter') @mock.patch.object(manager.ComputeManager, '_source_can_numa_live_migrate') def test_check_can_live_migrate_source( self, mock_src_can_numa, mock_event, mock_volume, mock_get_inst, mock_check, mock_get_bdms): fake_bdms = objects.BlockDeviceMappingList() mock_get_bdms.return_value = fake_bdms drvr_check_result = migrate_data_obj.LiveMigrateData() mock_check.return_value = drvr_check_result can_numa_result = migrate_data_obj.LiveMigrateData() mock_src_can_numa.return_value = can_numa_result is_volume_backed = 'volume_backed' dest_check_data = migrate_data_obj.LiveMigrateData() db_instance = fake_instance.fake_db_instance() instance = objects.Instance._from_db_object( self.context, objects.Instance(), db_instance) mock_volume.return_value = is_volume_backed mock_get_inst.return_value = {'block_device_mapping': 'fake'} result = self.compute.check_can_live_migrate_source( self.context, instance=instance, dest_check_data=dest_check_data) self.assertEqual(can_numa_result, result) mock_src_can_numa.assert_called_once_with( self.context, dest_check_data, drvr_check_result) mock_event.assert_called_once_with( self.context, 'compute_check_can_live_migrate_source', CONF.host, instance.uuid, graceful_exit=False) mock_check.assert_called_once_with(self.context, instance, dest_check_data, {'block_device_mapping': 'fake'}) mock_volume.assert_called_once_with(self.context, instance, fake_bdms) mock_get_inst.assert_called_once_with(self.context, instance, refresh_conn_info=False, bdms=fake_bdms) self.assertTrue(dest_check_data.is_volume_backed) def _test_check_can_live_migrate_destination(self, do_raise=False, src_numa_lm=None): db_instance = fake_instance.fake_db_instance(host='fake-host') instance = objects.Instance._from_db_object( self.context, objects.Instance(), db_instance) instance.host = 'fake-host' block_migration = 'block_migration' disk_over_commit = 'disk_over_commit' src_info = 'src_info' dest_info = 'dest_info' dest_check_data = objects.LibvirtLiveMigrateData( dst_supports_numa_live_migration=True) mig_data = objects.LibvirtLiveMigrateData() if src_numa_lm is not None: mig_data.src_supports_numa_live_migration = src_numa_lm with test.nested( mock.patch.object(self.compute, '_live_migration_claim'), mock.patch.object(self.compute, '_get_compute_info'), mock.patch.object(self.compute.driver, 'check_can_live_migrate_destination'), mock.patch.object(self.compute.compute_rpcapi, 'check_can_live_migrate_source'), mock.patch.object(self.compute.driver, 'cleanup_live_migration_destination_check'), mock.patch.object(db, 'instance_fault_create'), mock.patch.object(compute_utils, 'EventReporter'), mock.patch.object(migrate_data_obj.VIFMigrateData, 'create_skeleton_migrate_vifs', return_value=[]), mock.patch.object(instance, 'get_network_info'), mock.patch.object(self.compute, '_claim_pci_for_instance_vifs'), mock.patch.object(self.compute, '_update_migrate_vifs_profile_with_pci'), mock.patch.object(self.compute, '_dest_can_numa_live_migrate'), mock.patch('nova.compute.manager.LOG'), ) as (mock_lm_claim, mock_get, mock_check_dest, mock_check_src, mock_check_clean, mock_fault_create, mock_event, mock_create_mig_vif, mock_nw_info, mock_claim_pci, mock_update_mig_vif, mock_dest_can_numa, mock_log): mock_get.side_effect = (src_info, dest_info) mock_check_dest.return_value = dest_check_data mock_dest_can_numa.return_value = dest_check_data post_claim_md = objects.LibvirtLiveMigrateData( dst_numa_info=objects.LibvirtLiveMigrateNUMAInfo()) mock_lm_claim.return_value = post_claim_md if do_raise: mock_check_src.side_effect = test.TestingException mock_fault_create.return_value = \ test_instance_fault.fake_faults['fake-uuid'][0] else: mock_check_src.return_value = mig_data migration = objects.Migration() limits = objects.SchedulerLimits() result = self.compute.check_can_live_migrate_destination( self.context, instance=instance, block_migration=block_migration, disk_over_commit=disk_over_commit, migration=migration, limits=limits) if do_raise: mock_fault_create.assert_called_once_with(self.context, mock.ANY) mock_check_src.assert_called_once_with(self.context, instance, dest_check_data) mock_dest_can_numa.assert_called_with(dest_check_data, migration) if not src_numa_lm: mock_lm_claim.assert_not_called() self.assertEqual(mig_data, result) mock_log.info.assert_called() self.assertThat( mock_log.info.call_args[0][0], testtools.matchers.MatchesRegex( 'Destination was ready for NUMA live migration')) else: mock_lm_claim.assert_called_once_with( self.context, instance, mig_data, migration, limits, None) self.assertEqual(post_claim_md, result) mock_check_clean.assert_called_once_with(self.context, dest_check_data) mock_get.assert_has_calls([mock.call(self.context, 'fake-host'), mock.call(self.context, CONF.host)]) mock_check_dest.assert_called_once_with(self.context, instance, src_info, dest_info, block_migration, disk_over_commit) mock_event.assert_called_once_with( self.context, 'compute_check_can_live_migrate_destination', CONF.host, instance.uuid, graceful_exit=False) return result @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_check_can_live_migrate_destination_success(self): self.useFixture(std_fixtures.MonkeyPatch( 'nova.network.neutron.API.has_port_binding_extension', lambda *args: True)) self._test_check_can_live_migrate_destination() @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_check_can_live_migrate_destination_fail(self): self.useFixture(std_fixtures.MonkeyPatch( 'nova.network.neutron.API.has_port_binding_extension', lambda *args: True)) self.assertRaises( test.TestingException, self._test_check_can_live_migrate_destination, do_raise=True) @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_check_can_live_migrate_destination_contains_vifs(self): self.useFixture(std_fixtures.MonkeyPatch( 'nova.network.neutron.API.has_port_binding_extension', lambda *args: True)) migrate_data = self._test_check_can_live_migrate_destination() self.assertIn('vifs', migrate_data) self.assertIsNotNone(migrate_data.vifs) @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_check_can_live_migrate_destination_no_binding_extended(self): self.useFixture(std_fixtures.MonkeyPatch( 'nova.network.neutron.API.has_port_binding_extension', lambda *args: False)) migrate_data = self._test_check_can_live_migrate_destination() self.assertNotIn('vifs', migrate_data) @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_check_can_live_migrate_destination_src_numa_lm_false(self): self.useFixture(std_fixtures.MonkeyPatch( 'nova.network.neutron.API.has_port_binding_extension', lambda *args: True)) self._test_check_can_live_migrate_destination(src_numa_lm=False) @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_check_can_live_migrate_destination_src_numa_lm_true(self): self.useFixture(std_fixtures.MonkeyPatch( 'nova.network.neutron.API.has_port_binding_extension', lambda *args: True)) self._test_check_can_live_migrate_destination(src_numa_lm=True) @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') def test_check_can_live_migrate_destination_fail_group_policy( self, mock_fail_db): instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, vm_state=vm_states.ACTIVE, node='fake-node') ex = exception.RescheduledException( instance_uuid=instance.uuid, reason="policy violated") with mock.patch.object(self.compute, '_validate_instance_group_policy', side_effect=ex): self.assertRaises( exception.MigrationPreCheckError, self.compute.check_can_live_migrate_destination, self.context, instance, None, None, None, None) def test_dest_can_numa_live_migrate(self): positive_dest_check_data = objects.LibvirtLiveMigrateData( dst_supports_numa_live_migration=True) negative_dest_check_data = objects.LibvirtLiveMigrateData() self.assertNotIn( 'dst_supports_numa_live_migration', self.compute._dest_can_numa_live_migrate( copy.deepcopy(positive_dest_check_data), None)) self.assertIn( 'dst_supports_numa_live_migration', self.compute._dest_can_numa_live_migrate( copy.deepcopy(positive_dest_check_data), 'fake-migration')) self.assertNotIn( 'dst_supports_numa_live_migration', self.compute._dest_can_numa_live_migrate( copy.deepcopy(negative_dest_check_data), 'fake-migration')) self.assertNotIn( 'dst_supports_numa_live_migration', self.compute._dest_can_numa_live_migrate( copy.deepcopy(negative_dest_check_data), None)) def test_source_can_numa_live_migrate(self): positive_dest_check_data = objects.LibvirtLiveMigrateData( dst_supports_numa_live_migration=True) negative_dest_check_data = objects.LibvirtLiveMigrateData() positive_source_check_data = objects.LibvirtLiveMigrateData( src_supports_numa_live_migration=True) negative_source_check_data = objects.LibvirtLiveMigrateData() with mock.patch.object(self.compute.compute_rpcapi, 'supports_numa_live_migration', return_value=True ) as mock_supports_numa_lm: self.assertIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, positive_dest_check_data, copy.deepcopy(positive_source_check_data))) mock_supports_numa_lm.assert_called_once_with(self.context) self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, positive_dest_check_data, copy.deepcopy(negative_source_check_data))) self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, negative_dest_check_data, copy.deepcopy(positive_source_check_data))) self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, negative_dest_check_data, copy.deepcopy(negative_source_check_data))) with mock.patch.object(self.compute.compute_rpcapi, 'supports_numa_live_migration', return_value=False ) as mock_supports_numa_lm: self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, positive_dest_check_data, copy.deepcopy(positive_source_check_data))) mock_supports_numa_lm.assert_called_once_with(self.context) self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, positive_dest_check_data, copy.deepcopy(negative_source_check_data))) self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, negative_dest_check_data, copy.deepcopy(positive_source_check_data))) self.assertNotIn( 'src_supports_numa_live_migration', self.compute._source_can_numa_live_migrate( self.context, negative_dest_check_data, copy.deepcopy(negative_source_check_data))) @mock.patch('nova.compute.manager.InstanceEvents._lock_name') def test_prepare_for_instance_event(self, lock_name_mock): inst_obj = objects.Instance(uuid=uuids.instance) result = self.compute.instance_events.prepare_for_instance_event( inst_obj, 'test-event', None) self.assertIn(uuids.instance, self.compute.instance_events._events) self.assertIn(('test-event', None), self.compute.instance_events._events[uuids.instance]) self.assertEqual( result, self.compute.instance_events._events[uuids.instance] [('test-event', None)]) self.assertTrue(hasattr(result, 'send')) lock_name_mock.assert_called_once_with(inst_obj) @mock.patch('nova.compute.manager.InstanceEvents._lock_name') def test_pop_instance_event(self, lock_name_mock): event = eventlet_event.Event() self.compute.instance_events._events = { uuids.instance: { ('network-vif-plugged', None): event, } } inst_obj = objects.Instance(uuid=uuids.instance) event_obj = objects.InstanceExternalEvent(name='network-vif-plugged', tag=None) result = self.compute.instance_events.pop_instance_event(inst_obj, event_obj) self.assertEqual(result, event) lock_name_mock.assert_called_once_with(inst_obj) @mock.patch('nova.compute.manager.InstanceEvents._lock_name') def test_clear_events_for_instance(self, lock_name_mock): event = eventlet_event.Event() self.compute.instance_events._events = { uuids.instance: { ('test-event', None): event, } } inst_obj = objects.Instance(uuid=uuids.instance) result = self.compute.instance_events.clear_events_for_instance( inst_obj) self.assertEqual(result, {'test-event-None': event}) lock_name_mock.assert_called_once_with(inst_obj) def test_instance_events_lock_name(self): inst_obj = objects.Instance(uuid=uuids.instance) result = self.compute.instance_events._lock_name(inst_obj) self.assertEqual(result, "%s-events" % uuids.instance) def test_prepare_for_instance_event_again(self): inst_obj = objects.Instance(uuid=uuids.instance) self.compute.instance_events.prepare_for_instance_event( inst_obj, 'test-event', None) # A second attempt will avoid creating a new list; make sure we # get the current list result = self.compute.instance_events.prepare_for_instance_event( inst_obj, 'test-event', None) self.assertIn(uuids.instance, self.compute.instance_events._events) self.assertIn(('test-event', None), self.compute.instance_events._events[uuids.instance]) self.assertEqual( result, self.compute.instance_events._events[uuids.instance] [('test-event', None)]) self.assertTrue(hasattr(result, 'send')) def test_process_instance_event(self): event = eventlet_event.Event() self.compute.instance_events._events = { uuids.instance: { ('network-vif-plugged', None): event, } } inst_obj = objects.Instance(uuid=uuids.instance) event_obj = objects.InstanceExternalEvent(name='network-vif-plugged', tag=None) self.compute._process_instance_event(inst_obj, event_obj) self.assertTrue(event.ready()) self.assertEqual(event_obj, event.wait()) self.assertEqual({}, self.compute.instance_events._events) @ddt.data(task_states.DELETING, task_states.MIGRATING) @mock.patch('nova.compute.manager.LOG') def test_process_instance_event_expected_task(self, task_state, mock_log): """Tests that we don't log a warning when we get a network-vif-unplugged event for an instance that's undergoing a task state transition that will generate the expected event. """ inst_obj = objects.Instance(uuid=uuids.instance, vm_state=vm_states.ACTIVE, task_state=task_state) event_obj = objects.InstanceExternalEvent(name='network-vif-unplugged', tag=uuids.port_id) with mock.patch.object(self.compute.instance_events, 'pop_instance_event', return_value=None): self.compute._process_instance_event(inst_obj, event_obj) # assert we logged at debug level mock_log.debug.assert_called() self.assertThat(mock_log.debug.call_args[0][0], testtools.matchers.MatchesRegex( 'Received event .* for instance with task_state ' '%s')) @mock.patch('nova.compute.manager.LOG') def test_process_instance_event_unexpected_warning(self, mock_log): """Tests that we log a warning when we get an unexpected event.""" inst_obj = objects.Instance(uuid=uuids.instance, vm_state=vm_states.ACTIVE, task_state=None) event_obj = objects.InstanceExternalEvent(name='network-vif-unplugged', tag=uuids.port_id) with mock.patch.object(self.compute.instance_events, 'pop_instance_event', return_value=None): self.compute._process_instance_event(inst_obj, event_obj) # assert we logged at warning level mock_log.warning.assert_called() self.assertThat(mock_log.warning.call_args[0][0], testtools.matchers.MatchesRegex( 'Received unexpected event .* for ' 'instance with vm_state .* and ' 'task_state .*.')) def test_process_instance_vif_deleted_event(self): vif1 = fake_network_cache_model.new_vif() vif1['id'] = '1' vif2 = fake_network_cache_model.new_vif() vif2['id'] = '2' nw_info = network_model.NetworkInfo([vif1, vif2]) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=uuids.instance) inst_obj = objects.Instance(id=3, uuid=uuids.instance, info_cache=info_cache) @mock.patch.object(manager.neutron, 'update_instance_cache_with_nw_info') @mock.patch.object(self.compute.driver, 'detach_interface') def do_test(detach_interface, update_instance_cache_with_nw_info): self.compute._process_instance_vif_deleted_event(self.context, inst_obj, vif2['id']) update_instance_cache_with_nw_info.assert_called_once_with( self.compute.network_api, self.context, inst_obj, nw_info=[vif1]) detach_interface.assert_called_once_with(self.context, inst_obj, vif2) do_test() def test_process_instance_vif_deleted_event_not_implemented_error(self): """Tests the case where driver.detach_interface raises NotImplementedError. """ vif = fake_network_cache_model.new_vif() nw_info = network_model.NetworkInfo([vif]) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=uuids.instance) inst_obj = objects.Instance(id=3, uuid=uuids.instance, info_cache=info_cache) @mock.patch.object(manager.neutron, 'update_instance_cache_with_nw_info') @mock.patch.object(self.compute.driver, 'detach_interface', side_effect=NotImplementedError) def do_test(detach_interface, update_instance_cache_with_nw_info): self.compute._process_instance_vif_deleted_event( self.context, inst_obj, vif['id']) update_instance_cache_with_nw_info.assert_called_once_with( self.compute.network_api, self.context, inst_obj, nw_info=[]) detach_interface.assert_called_once_with( self.context, inst_obj, vif) do_test() @mock.patch('nova.compute.manager.LOG.info') # This is needed for py35. @mock.patch('nova.compute.manager.LOG.log') def test_process_instance_vif_deleted_event_instance_not_found( self, mock_log, mock_log_info): """Tests the case where driver.detach_interface raises InstanceNotFound. """ vif = fake_network_cache_model.new_vif() nw_info = network_model.NetworkInfo([vif]) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=uuids.instance) inst_obj = objects.Instance(id=3, uuid=uuids.instance, info_cache=info_cache) @mock.patch.object(manager.neutron, 'update_instance_cache_with_nw_info') @mock.patch.object(self.compute.driver, 'detach_interface', side_effect=exception.InstanceNotFound( instance_id=uuids.instance)) def do_test(detach_interface, update_instance_cache_with_nw_info): self.compute._process_instance_vif_deleted_event( self.context, inst_obj, vif['id']) update_instance_cache_with_nw_info.assert_called_once_with( self.compute.network_api, self.context, inst_obj, nw_info=[]) detach_interface.assert_called_once_with( self.context, inst_obj, vif) # LOG.log should have been called with a DEBUG level message. self.assertEqual(1, mock_log.call_count, mock_log.mock_calls) self.assertEqual(logging.DEBUG, mock_log.call_args[0][0]) do_test() def test_power_update(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.STOPPED instance.task_state = None instance.power_state = power_state.SHUTDOWN instance.host = self.compute.host with test.nested( mock.patch.object(nova.compute.utils, 'notify_about_instance_action'), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute.driver, 'power_update_event'), mock.patch.object(objects.Instance, 'save'), mock.patch.object(manager, 'LOG') ) as ( mock_instance_notify, mock_instance_usage, mock_event, mock_save, mock_log ): self.compute.power_update(self.context, instance, "POWER_ON") calls = [mock.call(self.context, instance, self.compute.host, action=fields.NotificationAction.POWER_ON, phase=fields.NotificationPhase.START), mock.call(self.context, instance, self.compute.host, action=fields.NotificationAction.POWER_ON, phase=fields.NotificationPhase.END)] mock_instance_notify.assert_has_calls(calls) calls = [mock.call(self.context, instance, "power_on.start"), mock.call(self.context, instance, "power_on.end")] mock_instance_usage.assert_has_calls(calls) mock_event.assert_called_once_with(instance, 'POWER_ON') mock_save.assert_called_once_with( expected_task_state=[None]) self.assertEqual(2, mock_log.debug.call_count) self.assertIn('Trying to', mock_log.debug.call_args[0][0]) def test_power_update_not_implemented(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.id = 1 instance.vm_state = vm_states.STOPPED instance.task_state = None instance.power_state = power_state.SHUTDOWN instance.host = self.compute.host with test.nested( mock.patch.object(nova.compute.utils, 'notify_about_instance_action'), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute.driver, 'power_update_event', side_effect=NotImplementedError()), mock.patch.object(instance, 'save'), mock.patch.object(nova.compute.utils, 'add_instance_fault_from_exc'), ) as ( mock_instance_notify, mock_instance_usage, mock_event, mock_save, mock_fault ): self.assertRaises(NotImplementedError, self.compute.power_update, self.context, instance, "POWER_ON") self.assertIsNone(instance.task_state) self.assertEqual(2, mock_save.call_count) # second save is done by revert_task_state mock_save.assert_has_calls( [mock.call(expected_task_state=[None]), mock.call()]) mock_instance_notify.assert_called_once_with( self.context, instance, self.compute.host, action=fields.NotificationAction.POWER_ON, phase=fields.NotificationPhase.START) mock_instance_usage.assert_called_once_with( self.context, instance, "power_on.start") mock_fault.assert_called_once_with( self.context, instance, mock.ANY, mock.ANY) def test_external_instance_event_power_update_invalid_state(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance1 instance.id = 1 instance.vm_state = vm_states.ACTIVE instance.task_state = task_states.POWERING_OFF instance.power_state = power_state.RUNNING instance.host = 'host1' instance.migration_context = None with test.nested( mock.patch.object(nova.compute.utils, 'notify_about_instance_action'), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute.driver, 'power_update_event'), mock.patch.object(objects.Instance, 'save'), mock.patch.object(manager, 'LOG') ) as ( mock_instance_notify, mock_instance_usage, mock_event, mock_save, mock_log ): self.compute.power_update(self.context, instance, "POWER_ON") mock_instance_notify.assert_not_called() mock_instance_usage.assert_not_called() mock_event.assert_not_called() mock_save.assert_not_called() self.assertEqual(1, mock_log.info.call_count) self.assertIn('is a no-op', mock_log.info.call_args[0][0]) def test_external_instance_event_power_update_unexpected_task_state(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance1 instance.id = 1 instance.vm_state = vm_states.ACTIVE instance.task_state = None instance.power_state = power_state.RUNNING instance.host = 'host1' instance.migration_context = None with test.nested( mock.patch.object(nova.compute.utils, 'notify_about_instance_action'), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute.driver, 'power_update_event'), mock.patch.object(objects.Instance, 'save', side_effect=exception.UnexpectedTaskStateError("blah")), mock.patch.object(manager, 'LOG') ) as ( mock_instance_notify, mock_instance_usage, mock_event, mock_save, mock_log ): self.compute.power_update(self.context, instance, "POWER_OFF") mock_instance_notify.assert_not_called() mock_instance_usage.assert_not_called() mock_event.assert_not_called() self.assertEqual(1, mock_log.info.call_count) self.assertIn('possibly preempted', mock_log.info.call_args[0][0]) def test_extend_volume(self): inst_obj = objects.Instance(id=3, uuid=uuids.instance) connection_info = {'foo': 'bar'} new_size = 20 bdm = objects.BlockDeviceMapping( source_type='volume', destination_type='volume', volume_id=uuids.volume_id, volume_size=10, instance_uuid=uuids.instance, device_name='/dev/vda', connection_info=jsonutils.dumps(connection_info)) @mock.patch.object(self.compute, 'volume_api') @mock.patch.object(self.compute.driver, 'extend_volume') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') @mock.patch.object(objects.BlockDeviceMapping, 'save') def do_test(bdm_save, bdm_get_by_vol_and_inst, extend_volume, volume_api): bdm_get_by_vol_and_inst.return_value = bdm volume_api.get.return_value = {'size': new_size} self.compute.extend_volume( self.context, inst_obj, uuids.volume_id) bdm_save.assert_called_once_with() extend_volume.assert_called_once_with( self.context, connection_info, inst_obj, new_size * pow(1024, 3)) do_test() def test_extend_volume_not_implemented_error(self): """Tests the case where driver.extend_volume raises NotImplementedError. """ inst_obj = objects.Instance(id=3, uuid=uuids.instance) connection_info = {'foo': 'bar'} bdm = objects.BlockDeviceMapping( source_type='volume', destination_type='volume', volume_id=uuids.volume_id, volume_size=10, instance_uuid=uuids.instance, device_name='/dev/vda', connection_info=jsonutils.dumps(connection_info)) @mock.patch.object(self.compute, 'volume_api') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') @mock.patch.object(objects.BlockDeviceMapping, 'save') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') def do_test(add_fault_mock, bdm_save, bdm_get_by_vol_and_inst, volume_api): bdm_get_by_vol_and_inst.return_value = bdm volume_api.get.return_value = {'size': 20} self.assertRaises( exception.ExtendVolumeNotSupported, self.compute.extend_volume, self.context, inst_obj, uuids.volume_id) add_fault_mock.assert_called_once_with( self.context, inst_obj, mock.ANY, mock.ANY) with mock.patch.dict(self.compute.driver.capabilities, supports_extend_volume=False): do_test() def test_extend_volume_volume_not_found(self): """Tests the case where driver.extend_volume tries to extend a volume not attached to the specified instance. """ inst_obj = objects.Instance(id=3, uuid=uuids.instance) @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance', side_effect=exception.NotFound()) def do_test(bdm_get_by_vol_and_inst): self.compute.extend_volume( self.context, inst_obj, uuids.volume_id) do_test() def test_external_instance_event(self): instances = [ objects.Instance(id=1, uuid=uuids.instance_1), objects.Instance(id=2, uuid=uuids.instance_2), objects.Instance(id=3, uuid=uuids.instance_3), objects.Instance(id=4, uuid=uuids.instance_4), objects.Instance(id=4, uuid=uuids.instance_5)] events = [ objects.InstanceExternalEvent(name='network-changed', tag='tag1', instance_uuid=uuids.instance_1), objects.InstanceExternalEvent(name='network-vif-plugged', instance_uuid=uuids.instance_2, tag='tag2'), objects.InstanceExternalEvent(name='network-vif-deleted', instance_uuid=uuids.instance_3, tag='tag3'), objects.InstanceExternalEvent(name='volume-extended', instance_uuid=uuids.instance_4, tag='tag4'), objects.InstanceExternalEvent(name='power-update', instance_uuid=uuids.instance_5, tag='POWER_ON')] @mock.patch.object(self.compute, 'power_update') @mock.patch.object(self.compute, 'extend_volume') @mock.patch.object(self.compute, '_process_instance_vif_deleted_event') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info') @mock.patch.object(self.compute, '_process_instance_event') def do_test(_process_instance_event, get_instance_nw_info, _process_instance_vif_deleted_event, extend_volume, power_update): self.compute.external_instance_event(self.context, instances, events) get_instance_nw_info.assert_called_once_with(self.context, instances[0], refresh_vif_id='tag1') _process_instance_event.assert_called_once_with(instances[1], events[1]) _process_instance_vif_deleted_event.assert_called_once_with( self.context, instances[2], events[2].tag) extend_volume.assert_called_once_with( self.context, instances[3], events[3].tag) power_update.assert_called_once_with( self.context, instances[4], events[4].tag) do_test() def test_external_instance_event_with_exception(self): vif1 = fake_network_cache_model.new_vif() vif1['id'] = '1' vif2 = fake_network_cache_model.new_vif() vif2['id'] = '2' nw_info = network_model.NetworkInfo([vif1, vif2]) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=uuids.instance_2) instances = [ objects.Instance(id=1, uuid=uuids.instance_1), objects.Instance(id=2, uuid=uuids.instance_2, info_cache=info_cache), objects.Instance(id=3, uuid=uuids.instance_3), # instance_4 doesn't have info_cache set so it will be lazy-loaded # and blow up with an InstanceNotFound error. objects.Instance(id=4, uuid=uuids.instance_4), objects.Instance(id=5, uuid=uuids.instance_5), ] events = [ objects.InstanceExternalEvent(name='network-changed', tag='tag1', instance_uuid=uuids.instance_1), objects.InstanceExternalEvent(name='network-vif-deleted', instance_uuid=uuids.instance_2, tag='2'), objects.InstanceExternalEvent(name='network-vif-plugged', instance_uuid=uuids.instance_3, tag='tag3'), objects.InstanceExternalEvent(name='network-vif-deleted', instance_uuid=uuids.instance_4, tag='tag4'), objects.InstanceExternalEvent(name='volume-extended', instance_uuid=uuids.instance_5, tag='tag5'), ] # Make sure all the four events are handled despite the exceptions in # processing events 1, 2, 4 and 5. @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance', side_effect=exception.InstanceNotFound( instance_id=uuids.instance_5)) @mock.patch.object(instances[3], 'obj_load_attr', side_effect=exception.InstanceNotFound( instance_id=uuids.instance_4)) @mock.patch.object(manager.neutron, 'update_instance_cache_with_nw_info') @mock.patch.object(self.compute.driver, 'detach_interface', side_effect=exception.NovaException) @mock.patch.object(self.compute.network_api, 'get_instance_nw_info', side_effect=exception.InstanceInfoCacheNotFound( instance_uuid=uuids.instance_1)) @mock.patch.object(self.compute, '_process_instance_event') def do_test(_process_instance_event, get_instance_nw_info, detach_interface, update_instance_cache_with_nw_info, obj_load_attr, bdm_get_by_vol_and_inst): self.compute.external_instance_event(self.context, instances, events) get_instance_nw_info.assert_called_once_with(self.context, instances[0], refresh_vif_id='tag1') update_instance_cache_with_nw_info.assert_called_once_with( self.compute.network_api, self.context, instances[1], nw_info=[vif1]) detach_interface.assert_called_once_with(self.context, instances[1], vif2) _process_instance_event.assert_called_once_with(instances[2], events[2]) obj_load_attr.assert_called_once_with('info_cache') bdm_get_by_vol_and_inst.assert_called_once_with( self.context, 'tag5', instances[4].uuid) do_test() def test_cancel_all_events(self): inst = objects.Instance(uuid=uuids.instance) fake_eventlet_event = mock.MagicMock() self.compute.instance_events._events = { inst.uuid: { ('network-vif-plugged', uuids.portid): fake_eventlet_event, } } self.compute.instance_events.cancel_all_events() # call it again to make sure we handle that gracefully self.compute.instance_events.cancel_all_events() self.assertTrue(fake_eventlet_event.send.called) event = fake_eventlet_event.send.call_args_list[0][0][0] self.assertEqual('network-vif-plugged', event.name) self.assertEqual(uuids.portid, event.tag) self.assertEqual('failed', event.status) def test_cleanup_cancels_all_events(self): with mock.patch.object(self.compute, 'instance_events') as mock_ev: self.compute.cleanup_host() mock_ev.cancel_all_events.assert_called_once_with() def test_cleanup_blocks_new_events(self): instance = objects.Instance(uuid=uuids.instance) self.compute.instance_events.cancel_all_events() callback = mock.MagicMock() body = mock.MagicMock() with self.compute.virtapi.wait_for_instance_event( instance, [('network-vif-plugged', 'bar')], error_callback=callback): body() self.assertTrue(body.called) callback.assert_called_once_with('network-vif-plugged-bar', instance) def test_pop_events_fails_gracefully(self): inst = objects.Instance(uuid=uuids.instance) event = mock.MagicMock() self.compute.instance_events._events = None self.assertIsNone( self.compute.instance_events.pop_instance_event(inst, event)) def test_clear_events_fails_gracefully(self): inst = objects.Instance(uuid=uuids.instance) self.compute.instance_events._events = None self.assertEqual( self.compute.instance_events.clear_events_for_instance(inst), {}) def test_retry_reboot_pending_soft(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_PENDING instance.vm_state = vm_states.ACTIVE allow_reboot, reboot_type = self.compute._retry_reboot( instance, power_state.RUNNING) self.assertTrue(allow_reboot) self.assertEqual(reboot_type, 'SOFT') def test_retry_reboot_pending_hard(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_PENDING_HARD instance.vm_state = vm_states.ACTIVE allow_reboot, reboot_type = self.compute._retry_reboot( instance, power_state.RUNNING) self.assertTrue(allow_reboot) self.assertEqual(reboot_type, 'HARD') def test_retry_reboot_starting_soft_off(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_STARTED allow_reboot, reboot_type = self.compute._retry_reboot( instance, power_state.NOSTATE) self.assertTrue(allow_reboot) self.assertEqual(reboot_type, 'HARD') def test_retry_reboot_starting_hard_off(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_STARTED_HARD allow_reboot, reboot_type = self.compute._retry_reboot( instance, power_state.NOSTATE) self.assertTrue(allow_reboot) self.assertEqual(reboot_type, 'HARD') def test_retry_reboot_starting_hard_on(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = task_states.REBOOT_STARTED_HARD allow_reboot, reboot_type = self.compute._retry_reboot( instance, power_state.RUNNING) self.assertFalse(allow_reboot) self.assertEqual(reboot_type, 'HARD') def test_retry_reboot_no_reboot(self): instance = objects.Instance(self.context) instance.uuid = uuids.instance instance.task_state = 'bar' allow_reboot, reboot_type = self.compute._retry_reboot( instance, power_state.RUNNING) self.assertFalse(allow_reboot) self.assertEqual(reboot_type, 'HARD') @mock.patch('nova.objects.BlockDeviceMapping.get_by_volume_and_instance') def test_remove_volume_connection(self, bdm_get): inst = mock.Mock() inst.uuid = uuids.instance_uuid fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.volume_id, 'device_name': '/dev/vdb', 'connection_info': '{"test": "test"}'}) bdm = objects.BlockDeviceMapping(context=self.context, **fake_bdm) bdm_get.return_value = bdm with test.nested( mock.patch.object(self.compute, 'volume_api'), mock.patch.object(self.compute, 'driver'), mock.patch.object(driver_bdm_volume, 'driver_detach'), ) as (mock_volume_api, mock_virt_driver, mock_driver_detach): connector = mock.Mock() def fake_driver_detach(context, instance, volume_api, virt_driver): # This is just here to validate the function signature. pass # There should be an easier way to do this with autospec... mock_driver_detach.side_effect = fake_driver_detach mock_virt_driver.get_volume_connector.return_value = connector self.compute.remove_volume_connection(self.context, uuids.volume_id, inst) bdm_get.assert_called_once_with(self.context, uuids.volume_id, uuids.instance_uuid) mock_driver_detach.assert_called_once_with(self.context, inst, mock_volume_api, mock_virt_driver) mock_volume_api.terminate_connection.assert_called_once_with( self.context, uuids.volume_id, connector) def test_remove_volume_connection_cinder_v3_api(self): instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) volume_id = uuids.volume vol_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) vol_bdm.attachment_id = uuids.attachment @mock.patch.object(self.compute.volume_api, 'terminate_connection') @mock.patch.object(self.compute, 'driver') @mock.patch.object(driver_bdm_volume, 'driver_detach') @mock.patch.object(objects.BlockDeviceMapping, 'get_by_volume_and_instance') def _test(mock_get_bdms, mock_detach, mock_driver, mock_terminate): mock_get_bdms.return_value = vol_bdm self.compute.remove_volume_connection(self.context, volume_id, instance) mock_detach.assert_called_once_with(self.context, instance, self.compute.volume_api, mock_driver) mock_terminate.assert_not_called() _test() @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.virt.block_device.DriverVolumeBlockDevice.driver_detach') def test_remove_volume_connection_v3_delete_attachment_true( self, driver_detach, attachment_delete): """Tests _remove_volume_connection with a cinder v3 style volume attachment and delete_attachment=True. """ instance = objects.Instance(uuid=uuids.instance) volume_id = uuids.volume vol_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) vol_bdm.attachment_id = uuids.attachment_id self.compute._remove_volume_connection( self.context, vol_bdm, instance, delete_attachment=True) driver_detach.assert_called_once_with( self.context, instance, self.compute.volume_api, self.compute.driver) attachment_delete.assert_called_once_with( self.context, vol_bdm.attachment_id) def test_delete_disk_metadata(self): bdm = objects.BlockDeviceMapping(volume_id=uuids.volume_id, tag='foo') instance = fake_instance.fake_instance_obj(self.context) instance.device_metadata = objects.InstanceDeviceMetadata( devices=[objects.DiskMetadata(serial=uuids.volume_id, tag='foo')]) instance.save = mock.Mock() self.compute._delete_disk_metadata(instance, bdm) self.assertEqual(0, len(instance.device_metadata.devices)) instance.save.assert_called_once_with() def test_delete_disk_metadata_no_serial(self): bdm = objects.BlockDeviceMapping(tag='foo') instance = fake_instance.fake_instance_obj(self.context) instance.device_metadata = objects.InstanceDeviceMetadata( devices=[objects.DiskMetadata(tag='foo')]) self.compute._delete_disk_metadata(instance, bdm) # NOTE(artom) This looks weird because we haven't deleted anything, but # it's normal behaviour for when DiskMetadata doesn't have serial set # and we can't find it based on BlockDeviceMapping's volume_id. self.assertEqual(1, len(instance.device_metadata.devices)) def test_detach_volume(self): # TODO(lyarwood): Test DriverVolumeBlockDevice.detach in # ../virt/test_block_device.py self._test_detach_volume() def test_detach_volume_not_destroy_bdm(self): # TODO(lyarwood): Test DriverVolumeBlockDevice.detach in # ../virt/test_block_device.py self._test_detach_volume(destroy_bdm=False) @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch.object(driver_bdm_volume, 'detach') @mock.patch('nova.compute.utils.notify_about_volume_attach_detach') @mock.patch('nova.compute.manager.ComputeManager._delete_disk_metadata') def test_detach_untagged_volume_metadata_not_deleted( self, mock_delete_metadata, _, __, ___): inst_obj = mock.Mock() fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.volume, 'device_name': '/dev/vdb', 'connection_info': '{"test": "test"}'}) bdm = objects.BlockDeviceMapping(context=self.context, **fake_bdm) self.compute._detach_volume(self.context, bdm, inst_obj, destroy_bdm=False, attachment_id=uuids.attachment) self.assertFalse(mock_delete_metadata.called) @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch.object(driver_bdm_volume, 'detach') @mock.patch('nova.compute.utils.notify_about_volume_attach_detach') @mock.patch('nova.compute.manager.ComputeManager._delete_disk_metadata') def test_detach_tagged_volume(self, mock_delete_metadata, _, __, ___): inst_obj = mock.Mock() fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.volume, 'device_name': '/dev/vdb', 'connection_info': '{"test": "test"}', 'tag': 'foo'}) bdm = objects.BlockDeviceMapping(context=self.context, **fake_bdm) self.compute._detach_volume(self.context, bdm, inst_obj, destroy_bdm=False, attachment_id=uuids.attachment) mock_delete_metadata.assert_called_once_with(inst_obj, bdm) @mock.patch.object(driver_bdm_volume, 'detach') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_volume_attach_detach') def _test_detach_volume(self, mock_notify_attach_detach, notify_inst_usage, detach, destroy_bdm=True): # TODO(lyarwood): Test DriverVolumeBlockDevice.detach in # ../virt/test_block_device.py volume_id = uuids.volume inst_obj = mock.Mock() inst_obj.uuid = uuids.instance inst_obj.host = CONF.host attachment_id = uuids.attachment fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'connection_info': '{"test": "test"}'}) bdm = objects.BlockDeviceMapping(context=self.context, **fake_bdm) with test.nested( mock.patch.object(self.compute, 'volume_api'), mock.patch.object(self.compute, 'driver'), mock.patch.object(bdm, 'destroy'), ) as (volume_api, driver, bdm_destroy): self.compute._detach_volume(self.context, bdm, inst_obj, destroy_bdm=destroy_bdm, attachment_id=attachment_id) detach.assert_called_once_with(self.context, inst_obj, self.compute.volume_api, self.compute.driver, attachment_id=attachment_id, destroy_bdm=destroy_bdm) notify_inst_usage.assert_called_once_with( self.context, inst_obj, "volume.detach", extra_usage_info={'volume_id': volume_id}) if destroy_bdm: bdm_destroy.assert_called_once_with() else: self.assertFalse(bdm_destroy.called) mock_notify_attach_detach.assert_has_calls([ mock.call(self.context, inst_obj, 'fake-mini', action='volume_detach', phase='start', volume_id=volume_id), mock.call(self.context, inst_obj, 'fake-mini', action='volume_detach', phase='end', volume_id=volume_id), ]) def test_detach_volume_evacuate(self): """For evacuate, terminate_connection is called with original host.""" # TODO(lyarwood): Test DriverVolumeBlockDevice.driver_detach in # ../virt/test_block_device.py expected_connector = {'host': 'evacuated-host'} conn_info_str = '{"connector": {"host": "evacuated-host"}}' self._test_detach_volume_evacuate(conn_info_str, expected=expected_connector) def test_detach_volume_evacuate_legacy(self): """Test coverage for evacuate with legacy attachments. In this case, legacy means the volume was attached to the instance before nova stashed the connector in connection_info. The connector sent to terminate_connection will still be for the local host in this case because nova does not have the info to get the connector for the original (evacuated) host. """ # TODO(lyarwood): Test DriverVolumeBlockDevice.driver_detach in # ../virt/test_block_device.py conn_info_str = '{"foo": "bar"}' # Has no 'connector'. self._test_detach_volume_evacuate(conn_info_str) def test_detach_volume_evacuate_mismatch(self): """Test coverage for evacuate with connector mismatch. For evacuate, if the stashed connector also has the wrong host, then log it and stay with the local connector. """ # TODO(lyarwood): Test DriverVolumeBlockDevice.driver_detach in # ../virt/test_block_device.py conn_info_str = '{"connector": {"host": "other-host"}}' self._test_detach_volume_evacuate(conn_info_str) @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_volume_attach_detach') def _test_detach_volume_evacuate(self, conn_info_str, mock_notify_attach_detach, notify_inst_usage, expected=None): """Re-usable code for detach volume evacuate test cases. :param conn_info_str: String form of the stashed connector. :param expected: Dict of the connector that is expected in the terminate call (optional). Default is to expect the local connector to be used. """ # TODO(lyarwood): Test DriverVolumeBlockDevice.driver_detach in # ../virt/test_block_device.py volume_id = 'vol_id' instance = fake_instance.fake_instance_obj(self.context, host='evacuated-host') fake_bdm = fake_block_device.FakeDbBlockDeviceDict( {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'connection_info': '{"test": "test"}'}) bdm = objects.BlockDeviceMapping(context=self.context, **fake_bdm) bdm.connection_info = conn_info_str local_connector = {'host': 'local-connector-host'} expected_connector = local_connector if not expected else expected with test.nested( mock.patch.object(self.compute, 'volume_api'), mock.patch.object(self.compute, 'driver'), mock.patch.object(driver_bdm_volume, 'driver_detach'), ) as (volume_api, driver, driver_detach): driver.get_volume_connector.return_value = local_connector self.compute._detach_volume(self.context, bdm, instance, destroy_bdm=False) driver_detach.assert_not_called() driver.get_volume_connector.assert_called_once_with(instance) volume_api.terminate_connection.assert_called_once_with( self.context, volume_id, expected_connector) volume_api.detach.assert_called_once_with(mock.ANY, volume_id, instance.uuid, None) notify_inst_usage.assert_called_once_with( self.context, instance, "volume.detach", extra_usage_info={'volume_id': volume_id} ) mock_notify_attach_detach.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', action='volume_detach', phase='start', volume_id=volume_id), mock.call(self.context, instance, 'fake-mini', action='volume_detach', phase='end', volume_id=volume_id), ]) @mock.patch('nova.compute.utils.notify_about_instance_rescue_action') def _test_rescue(self, mock_notify, clean_shutdown=True): instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE) fake_nw_info = network_model.NetworkInfo() rescue_image_meta = objects.ImageMeta.from_dict( {'id': uuids.image_id, 'name': uuids.image_name}) with test.nested( mock.patch.object(self.context, 'elevated', return_value=self.context), mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value=fake_nw_info), mock.patch.object(self.compute, '_get_rescue_image', return_value=rescue_image_meta), mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid', return_value=mock.sentinel.bdms), mock.patch.object(self.compute, '_get_instance_block_device_info', return_value=mock.sentinel.block_device_info), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute, '_power_off_instance'), mock.patch.object(self.compute.driver, 'rescue'), mock.patch.object(compute_utils, 'notify_usage_exists'), mock.patch.object(self.compute, '_get_power_state', return_value=power_state.RUNNING), mock.patch.object(instance, 'save') ) as ( elevated_context, get_nw_info, get_rescue_image, get_bdm_list, get_block_info, notify_instance_usage, power_off_instance, driver_rescue, notify_usage_exists, get_power_state, instance_save ): self.compute.rescue_instance( self.context, instance, rescue_password='verybadpass', rescue_image_ref=None, clean_shutdown=clean_shutdown) # assert the field values on the instance object self.assertEqual(vm_states.RESCUED, instance.vm_state) self.assertIsNone(instance.task_state) self.assertEqual(power_state.RUNNING, instance.power_state) self.assertIsNotNone(instance.launched_at) # assert our mock calls get_nw_info.assert_called_once_with(self.context, instance) get_rescue_image.assert_called_once_with( self.context, instance, None) get_bdm_list.assert_called_once_with(self.context, instance.uuid) get_block_info.assert_called_once_with(self.context, instance, bdms=mock.sentinel.bdms) extra_usage_info = {'rescue_image_name': uuids.image_name} notify_calls = [ mock.call(self.context, instance, "rescue.start", extra_usage_info=extra_usage_info, network_info=fake_nw_info), mock.call(self.context, instance, "rescue.end", extra_usage_info=extra_usage_info, network_info=fake_nw_info) ] notify_instance_usage.assert_has_calls(notify_calls) power_off_instance.assert_called_once_with(instance, clean_shutdown) driver_rescue.assert_called_once_with( self.context, instance, fake_nw_info, rescue_image_meta, 'verybadpass', mock.sentinel.block_device_info) notify_usage_exists.assert_called_once_with(self.compute.notifier, self.context, instance, 'fake-mini', current_period=True) mock_notify.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', None, phase='start'), mock.call(self.context, instance, 'fake-mini', None, phase='end')]) instance_save.assert_called_once_with( expected_task_state=task_states.RESCUING) def test_rescue(self): self._test_rescue() def test_rescue_forced_shutdown(self): self._test_rescue(clean_shutdown=False) @mock.patch('nova.compute.utils.notify_about_instance_action') def test_unrescue(self, mock_notify): instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.RESCUED) fake_nw_info = network_model.NetworkInfo() with test.nested( mock.patch.object(self.context, 'elevated', return_value=self.context), mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value=fake_nw_info), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute.driver, 'unrescue'), mock.patch.object(self.compute, '_get_power_state', return_value=power_state.RUNNING), mock.patch.object(instance, 'save') ) as ( elevated_context, get_nw_info, notify_instance_usage, driver_unrescue, get_power_state, instance_save ): self.compute.unrescue_instance(self.context, instance) # assert the field values on the instance object self.assertEqual(vm_states.ACTIVE, instance.vm_state) self.assertIsNone(instance.task_state) self.assertEqual(power_state.RUNNING, instance.power_state) # assert our mock calls get_nw_info.assert_called_once_with(self.context, instance) notify_calls = [ mock.call(self.context, instance, "unrescue.start", network_info=fake_nw_info), mock.call(self.context, instance, "unrescue.end", network_info=fake_nw_info) ] notify_instance_usage.assert_has_calls(notify_calls) driver_unrescue.assert_called_once_with(self.context, instance) mock_notify.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', action='unrescue', phase='start'), mock.call(self.context, instance, 'fake-mini', action='unrescue', phase='end')]) instance_save.assert_called_once_with( expected_task_state=task_states.UNRESCUING) @mock.patch('nova.compute.manager.ComputeManager._get_power_state', return_value=power_state.RUNNING) @mock.patch.object(objects.Instance, 'save') def test_set_admin_password(self, instance_save_mock, power_state_mock): # Ensure instance can have its admin password set. instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE, task_state=task_states.UPDATING_PASSWORD) @mock.patch.object(self.context, 'elevated', return_value=self.context) @mock.patch.object(self.compute.driver, 'set_admin_password') def do_test(driver_mock, elevated_mock): # call the manager method self.compute.set_admin_password(self.context, instance, 'fake-pass') # make our assertions self.assertEqual(vm_states.ACTIVE, instance.vm_state) self.assertIsNone(instance.task_state) power_state_mock.assert_called_once_with(instance) driver_mock.assert_called_once_with(instance, 'fake-pass') instance_save_mock.assert_called_once_with( expected_task_state=task_states.UPDATING_PASSWORD) do_test() @mock.patch('nova.compute.manager.ComputeManager._get_power_state', return_value=power_state.NOSTATE) @mock.patch('nova.compute.manager.ComputeManager._instance_update') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') def test_set_admin_password_bad_state(self, add_fault_mock, instance_save_mock, update_mock, power_state_mock): # Test setting password while instance is rebuilding. instance = fake_instance.fake_instance_obj(self.context) with mock.patch.object(self.context, 'elevated', return_value=self.context): # call the manager method self.assertRaises(exception.InstancePasswordSetFailed, self.compute.set_admin_password, self.context, instance, None) # make our assertions power_state_mock.assert_called_once_with(instance) instance_save_mock.assert_called_once_with( expected_task_state=task_states.UPDATING_PASSWORD) add_fault_mock.assert_called_once_with( self.context, instance, mock.ANY, mock.ANY) @mock.patch('nova.utils.generate_password', return_value='fake-pass') @mock.patch('nova.compute.manager.ComputeManager._get_power_state', return_value=power_state.RUNNING) @mock.patch('nova.compute.manager.ComputeManager._instance_update') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') def _do_test_set_admin_password_driver_error(self, exc, expected_vm_state, expected_task_state, expected_exception, add_fault_mock, instance_save_mock, update_mock, power_state_mock, gen_password_mock): # Ensure expected exception is raised if set_admin_password fails. instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE, task_state=task_states.UPDATING_PASSWORD) @mock.patch.object(self.context, 'elevated', return_value=self.context) @mock.patch.object(self.compute.driver, 'set_admin_password', side_effect=exc) def do_test(driver_mock, elevated_mock): # error raised from the driver should not reveal internal # information so a new error is raised self.assertRaises(expected_exception, self.compute.set_admin_password, self.context, instance=instance, new_pass=None) if expected_exception != exception.InstancePasswordSetFailed: instance_save_mock.assert_called_once_with( expected_task_state=task_states.UPDATING_PASSWORD) self.assertEqual(expected_vm_state, instance.vm_state) # check revert_task_state decorator update_mock.assert_called_once_with( self.context, instance, task_state=expected_task_state) # check wrap_instance_fault decorator add_fault_mock.assert_called_once_with( self.context, instance, mock.ANY, mock.ANY) do_test() def test_set_admin_password_driver_not_authorized(self): # Ensure expected exception is raised if set_admin_password not # authorized. exc = exception.Forbidden('Internal error') expected_exception = exception.InstancePasswordSetFailed self._do_test_set_admin_password_driver_error( exc, vm_states.ACTIVE, None, expected_exception) def test_set_admin_password_driver_not_implemented(self): # Ensure expected exception is raised if set_admin_password not # implemented by driver. exc = NotImplementedError() expected_exception = NotImplementedError self._do_test_set_admin_password_driver_error( exc, vm_states.ACTIVE, None, expected_exception) def test_set_admin_password_driver_not_supported(self): exc = exception.SetAdminPasswdNotSupported() expected_exception = exception.SetAdminPasswdNotSupported self._do_test_set_admin_password_driver_error( exc, vm_states.ACTIVE, None, expected_exception) def test_set_admin_password_guest_agent_no_enabled(self): exc = exception.QemuGuestAgentNotEnabled() expected_exception = exception.InstanceAgentNotEnabled self._do_test_set_admin_password_driver_error( exc, vm_states.ACTIVE, None, expected_exception) def test_destroy_evacuated_instances_no_migrations(self): with mock.patch( 'nova.objects.MigrationList.get_by_filters') as migration_list: migration_list.return_value = [] result = self.compute._destroy_evacuated_instances( self.context, {}) self.assertEqual({}, result) def test_destroy_evacuated_instances(self): our_host = self.compute.host flavor = objects.Flavor() instance_1 = objects.Instance(self.context, flavor=flavor) instance_1.uuid = uuids.instance_1 instance_1.task_state = None instance_1.vm_state = vm_states.ACTIVE instance_1.host = 'not-' + our_host instance_1.user_id = uuids.user_id instance_1.project_id = uuids.project_id instance_2 = objects.Instance(self.context, flavor=flavor) instance_2.uuid = uuids.instance_2 instance_2.task_state = None instance_2.vm_state = vm_states.ACTIVE instance_2.host = 'not-' + our_host instance_2.user_id = uuids.user_id instance_2.project_id = uuids.project_id instance_2.deleted = False # Only instance 2 has a migration record migration = objects.Migration(instance_uuid=instance_2.uuid) # Consider the migration successful migration.status = 'done' migration.source_node = 'fake-node' node_cache = { uuids.our_node_uuid: objects.ComputeNode( uuid=uuids.our_node_uuid, hypervisor_hostname='fake-node') } with test.nested( mock.patch.object(self.compute, '_get_instances_on_driver', return_value=[instance_1, instance_2]), mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value=None), mock.patch.object(self.compute, '_get_instance_block_device_info', return_value={}), mock.patch.object(self.compute, '_is_instance_storage_shared', return_value=False), mock.patch.object(self.compute.driver, 'destroy'), mock.patch('nova.objects.MigrationList.get_by_filters'), mock.patch('nova.objects.Migration.save'), mock.patch('nova.scheduler.utils.resources_from_flavor'), mock.patch.object(self.compute.reportclient, 'remove_provider_tree_from_instance_allocation') ) as (_get_instances_on_driver, get_instance_nw_info, _get_instance_block_device_info, _is_instance_storage_shared, destroy, migration_list, migration_save, get_resources, remove_allocation): migration_list.return_value = [migration] get_resources.return_value = mock.sentinel.resources self.compute._destroy_evacuated_instances(self.context, node_cache) # Only instance 2 should be deleted. Instance 1 is still running # here, but no migration from our host exists, so ignore it destroy.assert_called_once_with(self.context, instance_2, None, {}, True) remove_allocation.assert_called_once_with( self.context, instance_2.uuid, uuids.our_node_uuid) def test_destroy_evacuated_instances_node_deleted(self): our_host = self.compute.host flavor = objects.Flavor() instance_1 = objects.Instance(self.context, flavor=flavor) instance_1.uuid = uuids.instance_1 instance_1.task_state = None instance_1.vm_state = vm_states.ACTIVE instance_1.host = 'not-' + our_host instance_1.user_id = uuids.user_id instance_1.project_id = uuids.project_id instance_1.deleted = False instance_2 = objects.Instance(self.context, flavor=flavor) instance_2.uuid = uuids.instance_2 instance_2.task_state = None instance_2.vm_state = vm_states.ACTIVE instance_2.host = 'not-' + our_host instance_2.user_id = uuids.user_id instance_2.project_id = uuids.project_id instance_2.deleted = False migration_1 = objects.Migration(instance_uuid=instance_1.uuid) # Consider the migration successful but the node was deleted while the # compute was down migration_1.status = 'done' migration_1.source_node = 'deleted-node' migration_2 = objects.Migration(instance_uuid=instance_2.uuid) # Consider the migration successful migration_2.status = 'done' migration_2.source_node = 'fake-node' node_cache = { uuids.our_node_uuid: objects.ComputeNode( uuid=uuids.our_node_uuid, hypervisor_hostname='fake-node') } with test.nested( mock.patch.object(self.compute, '_get_instances_on_driver', return_value=[instance_1, instance_2]), mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value=None), mock.patch.object(self.compute, '_get_instance_block_device_info', return_value={}), mock.patch.object(self.compute, '_is_instance_storage_shared', return_value=False), mock.patch.object(self.compute.driver, 'destroy'), mock.patch('nova.objects.MigrationList.get_by_filters'), mock.patch('nova.objects.Migration.save'), mock.patch('nova.scheduler.utils.resources_from_flavor'), mock.patch.object(self.compute.reportclient, 'remove_provider_tree_from_instance_allocation') ) as (_get_instances_on_driver, get_instance_nw_info, _get_instance_block_device_info, _is_instance_storage_shared, destroy, migration_list, migration_save, get_resources, remove_allocation): migration_list.return_value = [migration_1, migration_2] get_resources.return_value = mock.sentinel.resources self.compute._destroy_evacuated_instances(self.context, node_cache) # both instance_1 and instance_2 is destroyed in the driver destroy.assert_has_calls( [mock.call(self.context, instance_1, None, {}, True), mock.call(self.context, instance_2, None, {}, True)], any_order=True) # but only instance_2 is deallocated as the compute node for # instance_1 is already deleted remove_allocation.assert_called_once_with( self.context, instance_2.uuid, uuids.our_node_uuid) def test_destroy_evacuated_instances_not_on_hypervisor(self): our_host = self.compute.host flavor = objects.Flavor() instance_1 = objects.Instance(self.context, flavor=flavor) instance_1.uuid = uuids.instance_1 instance_1.task_state = None instance_1.vm_state = vm_states.ACTIVE instance_1.host = 'not-' + our_host instance_1.user_id = uuids.user_id instance_1.project_id = uuids.project_id instance_1.deleted = False migration_1 = objects.Migration(instance_uuid=instance_1.uuid) migration_1.status = 'done' migration_1.source_node = 'our-node' node_cache = { uuids.our_node_uuid: objects.ComputeNode( uuid=uuids.our_node_uuid, hypervisor_hostname='our-node') } with test.nested( mock.patch.object(self.compute, '_get_instances_on_driver', return_value=[]), mock.patch.object(self.compute.driver, 'destroy'), mock.patch('nova.objects.MigrationList.get_by_filters'), mock.patch('nova.objects.Migration.save'), mock.patch('nova.scheduler.utils.resources_from_flavor'), mock.patch.object(self.compute.reportclient, 'remove_provider_tree_from_instance_allocation'), mock.patch('nova.objects.Instance.get_by_uuid') ) as (_get_instances_on_driver, destroy, migration_list, migration_save, get_resources, remove_allocation, instance_get_by_uuid): migration_list.return_value = [migration_1] instance_get_by_uuid.return_value = instance_1 get_resources.return_value = mock.sentinel.resources self.compute._destroy_evacuated_instances(self.context, node_cache) # nothing to be destroyed as the driver returned no instances on # the hypervisor self.assertFalse(destroy.called) # but our only instance still cleaned up in placement remove_allocation.assert_called_once_with( self.context, instance_1.uuid, uuids.our_node_uuid) instance_get_by_uuid.assert_called_once_with( self.context, instance_1.uuid) def test_destroy_evacuated_instances_not_on_hyp_and_instance_deleted(self): migration_1 = objects.Migration(instance_uuid=uuids.instance_1) migration_1.status = 'done' migration_1.source_node = 'our-node' with test.nested( mock.patch.object(self.compute, '_get_instances_on_driver', return_value=[]), mock.patch.object(self.compute.driver, 'destroy'), mock.patch('nova.objects.MigrationList.get_by_filters'), mock.patch('nova.objects.Migration.save'), mock.patch('nova.scheduler.utils.resources_from_flavor'), mock.patch.object(self.compute.reportclient, 'remove_provider_tree_from_instance_allocation'), mock.patch('nova.objects.Instance.get_by_uuid') ) as (_get_instances_on_driver, destroy, migration_list, migration_save, get_resources, remove_allocation, instance_get_by_uuid): migration_list.return_value = [migration_1] instance_get_by_uuid.side_effect = exception.InstanceNotFound( instance_id=uuids.instance_1) self.compute._destroy_evacuated_instances(self.context, {}) # nothing to be destroyed as the driver returned no instances on # the hypervisor self.assertFalse(destroy.called) instance_get_by_uuid.assert_called_once_with( self.context, uuids.instance_1) # nothing to be cleaned as the instance was deleted already self.assertFalse(remove_allocation.called) @mock.patch('nova.compute.manager.ComputeManager.' '_destroy_evacuated_instances') @mock.patch('nova.compute.manager.LOG') def test_init_host_foreign_instance(self, mock_log, mock_destroy): inst = mock.MagicMock() inst.host = self.compute.host + '-alt' self.compute._init_instance(mock.sentinel.context, inst) self.assertFalse(inst.save.called) self.assertTrue(mock_log.warning.called) msg = mock_log.warning.call_args_list[0] self.assertIn('appears to not be owned by this host', msg[0][0]) def test_init_host_pci_device_spec_validation_failure(self): # Tests that we fail init_host if there is a pci.device_spec # configured incorrectly. self.flags( device_spec=[ # it's invalid to specify both in the same devspec jsonutils.dumps({'address': 'foo', 'devname': 'bar'}) ], group='pci' ) self.assertRaises(exception.PciDeviceInvalidDeviceName, self.compute.init_host, None) @mock.patch('nova.compute.manager.ComputeManager._instance_update') def test_error_out_instance_on_exception_not_implemented_err(self, inst_update_mock): instance = fake_instance.fake_instance_obj(self.context) def do_test(): with self.compute._error_out_instance_on_exception( self.context, instance, instance_state=vm_states.STOPPED): raise NotImplementedError('test') self.assertRaises(NotImplementedError, do_test) inst_update_mock.assert_called_once_with( self.context, instance, vm_state=vm_states.STOPPED, task_state=None) @mock.patch('nova.compute.manager.ComputeManager._instance_update') def test_error_out_instance_on_exception_inst_fault_rollback(self, inst_update_mock): instance = fake_instance.fake_instance_obj(self.context) def do_test(): with self.compute._error_out_instance_on_exception( self.context, instance, instance_state=vm_states.STOPPED): raise exception.InstanceFaultRollback( inner_exception=test.TestingException('test')) self.assertRaises(test.TestingException, do_test) # The vm_state should be set to the instance_state parameter. inst_update_mock.assert_called_once_with( self.context, instance, vm_state=vm_states.STOPPED, task_state=None) @mock.patch('nova.compute.manager.ComputeManager.' '_set_instance_obj_error_state') def test_error_out_instance_on_exception_unknown_with_quotas(self, set_error): instance = fake_instance.fake_instance_obj(self.context) def do_test(): with self.compute._error_out_instance_on_exception( self.context, instance): raise test.TestingException('test') self.assertRaises(test.TestingException, do_test) set_error.assert_called_once_with(instance) @mock.patch('nova.compute.manager.ComputeManager.' '_detach_volume') def test_cleanup_volumes(self, mock_detach): instance = fake_instance.fake_instance_obj(self.context) bdm_do_not_delete_dict = fake_block_device.FakeDbBlockDeviceDict( {'volume_id': 'fake-id1', 'source_type': 'image', 'delete_on_termination': False}) bdm_delete_dict = fake_block_device.FakeDbBlockDeviceDict( {'volume_id': 'fake-id2', 'source_type': 'image', 'delete_on_termination': True}) bdms = block_device_obj.block_device_make_list(self.context, [bdm_do_not_delete_dict, bdm_delete_dict]) with mock.patch.object(self.compute.volume_api, 'delete') as volume_delete: self.compute._cleanup_volumes(self.context, instance, bdms) calls = [mock.call(self.context, bdm, instance, destroy_bdm=bdm.delete_on_termination) for bdm in bdms] self.assertEqual(calls, mock_detach.call_args_list) volume_delete.assert_called_once_with(self.context, bdms[1].volume_id) @mock.patch('nova.compute.manager.ComputeManager.' '_detach_volume') def test_cleanup_volumes_exception_do_not_raise(self, mock_detach): instance = fake_instance.fake_instance_obj(self.context) bdm_dict1 = fake_block_device.FakeDbBlockDeviceDict( {'volume_id': 'fake-id1', 'source_type': 'image', 'delete_on_termination': True}) bdm_dict2 = fake_block_device.FakeDbBlockDeviceDict( {'volume_id': 'fake-id2', 'source_type': 'image', 'delete_on_termination': True}) bdms = block_device_obj.block_device_make_list(self.context, [bdm_dict1, bdm_dict2]) with mock.patch.object(self.compute.volume_api, 'delete', side_effect=[test.TestingException(), None]) as volume_delete: self.compute._cleanup_volumes(self.context, instance, bdms, raise_exc=False) calls = [mock.call(self.context, bdm.volume_id) for bdm in bdms] self.assertEqual(calls, volume_delete.call_args_list) calls = [mock.call(self.context, bdm, instance, destroy_bdm=True) for bdm in bdms] self.assertEqual(calls, mock_detach.call_args_list) @mock.patch('nova.compute.manager.ComputeManager.' '_detach_volume') def test_cleanup_volumes_exception_raise(self, mock_detach): instance = fake_instance.fake_instance_obj(self.context) bdm_dict1 = fake_block_device.FakeDbBlockDeviceDict( {'volume_id': 'fake-id1', 'source_type': 'image', 'delete_on_termination': True}) bdm_dict2 = fake_block_device.FakeDbBlockDeviceDict( {'volume_id': 'fake-id2', 'source_type': 'image', 'delete_on_termination': True}) bdms = block_device_obj.block_device_make_list(self.context, [bdm_dict1, bdm_dict2]) with mock.patch.object(self.compute.volume_api, 'delete', side_effect=[test.TestingException(), None]) as volume_delete: self.assertRaises(test.TestingException, self.compute._cleanup_volumes, self.context, instance, bdms) calls = [mock.call(self.context, bdm.volume_id) for bdm in bdms] self.assertEqual(calls, volume_delete.call_args_list) calls = [mock.call(self.context, bdm, instance, destroy_bdm=bdm.delete_on_termination) for bdm in bdms] self.assertEqual(calls, mock_detach.call_args_list) @mock.patch('nova.compute.manager.ComputeManager._detach_volume', side_effect=exception.CinderConnectionFailed(reason='idk')) def test_cleanup_volumes_detach_fails_raise_exc(self, mock_detach): instance = fake_instance.fake_instance_obj(self.context) bdms = block_device_obj.block_device_make_list( self.context, [fake_block_device.FakeDbBlockDeviceDict( {'volume_id': uuids.volume_id, 'source_type': 'volume', 'destination_type': 'volume', 'delete_on_termination': False})]) self.assertRaises(exception.CinderConnectionFailed, self.compute._cleanup_volumes, self.context, instance, bdms) mock_detach.assert_called_once_with( self.context, bdms[0], instance, destroy_bdm=False) def test_stop_instance_task_state_none_power_state_shutdown(self): # Tests that stop_instance doesn't puke when the instance power_state # is shutdown and the task_state is None. instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE, task_state=None, power_state=power_state.SHUTDOWN) @mock.patch.object(self.compute, '_get_power_state', return_value=power_state.SHUTDOWN) @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, '_power_off_instance') @mock.patch.object(instance, 'save') def do_test(save_mock, power_off_mock, notify_mock, notify_action_mock, get_state_mock): # run the code self.compute.stop_instance(self.context, instance, True) # assert the calls self.assertEqual(2, get_state_mock.call_count) notify_mock.assert_has_calls([ mock.call(self.context, instance, 'power_off.start'), mock.call(self.context, instance, 'power_off.end') ]) notify_action_mock.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', action='power_off', phase='start'), mock.call(self.context, instance, 'fake-mini', action='power_off', phase='end'), ]) power_off_mock.assert_called_once_with(instance, True) save_mock.assert_called_once_with( expected_task_state=[task_states.POWERING_OFF, None]) self.assertEqual(power_state.SHUTDOWN, instance.power_state) self.assertIsNone(instance.task_state) self.assertEqual(vm_states.STOPPED, instance.vm_state) do_test() @mock.patch.object(manager.ComputeManager, '_set_migration_status') @mock.patch.object(manager.ComputeManager, '_do_rebuild_instance_with_claim') @mock.patch('nova.compute.utils.notify_about_instance_rebuild') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') def _test_rebuild_ex(self, instance, exc, mock_notify_about_instance_usage, mock_notify, mock_rebuild, mock_set, recreate=False, scheduled_node=None): mock_rebuild.side_effect = exc self.compute.rebuild_instance( self.context, instance, None, None, None, None, None, None, recreate, False, False, None, scheduled_node, {}, None, [], False, None) mock_set.assert_called_once_with(None, 'failed') mock_notify_about_instance_usage.assert_called_once_with( mock.ANY, instance, 'rebuild.error', fault=mock_rebuild.side_effect ) mock_notify.assert_called_once_with( mock.ANY, instance, 'fake-mini', phase='error', exception=exc, bdms=None) def test_rebuild_deleting(self): instance = fake_instance.fake_instance_obj(self.context) ex = exception.UnexpectedDeletingTaskStateError( instance_uuid=instance.uuid, expected='expected', actual='actual') self._test_rebuild_ex(instance, ex) def test_rebuild_notfound(self): instance = fake_instance.fake_instance_obj(self.context) ex = exception.InstanceNotFound(instance_id=instance.uuid) self._test_rebuild_ex(instance, ex) @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_error_out_instance_on_exception') def test_rebuild_driver_error_same_host(self, mock_error, mock_aiffe): instance = fake_instance.fake_instance_obj(self.context) ex = test.TestingException('foo') rt = self._mock_rt() self.assertRaises(test.TestingException, self._test_rebuild_ex, instance, ex) self.assertFalse( rt.delete_allocation_for_evacuated_instance.called) @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_error_out_instance_on_exception') def test_rebuild_driver_error_evacuate(self, mock_error, mock_aiffe, mock_elevated): mock_elevated.return_value = self.context instance = fake_instance.fake_instance_obj(self.context) instance.system_metadata = {} ex = test.TestingException('foo') rt = self._mock_rt() self.assertRaises(test.TestingException, self._test_rebuild_ex, instance, ex, recreate=True, scheduled_node='foo') delete_alloc = rt.delete_allocation_for_evacuated_instance delete_alloc.assert_called_once_with(self.context, instance, 'foo', node_type='destination') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.compute.resource_tracker.ResourceTracker.' 'delete_allocation_for_evacuated_instance') def test_rebuild_compute_resources_unavailable(self, mock_delete_alloc, mock_add_fault, mock_save): """Tests that when the rebuild_claim fails with ComputeResourcesUnavailable the vm_state on the instance remains unchanged. """ instance = fake_instance.fake_instance_obj(self.context) instance.vm_state = vm_states.ACTIVE ex = exception.ComputeResourcesUnavailable(reason='out of foo') self.assertRaises(messaging.ExpectedException, self._test_rebuild_ex, instance, ex) # Make sure the instance vm_state did not change. self.assertEqual(vm_states.ACTIVE, instance.vm_state) mock_delete_alloc.assert_called_once() mock_save.assert_called() mock_add_fault.assert_called_once() @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.objects.instance.Instance.drop_migration_context') @mock.patch('nova.objects.instance.Instance.apply_migration_context') @mock.patch('nova.objects.instance.Instance.mutated_migration_context') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') @mock.patch('nova.network.neutron.API.setup_instance_network_on_host') @mock.patch('nova.network.neutron.API.setup_networks_on_host') @mock.patch('nova.objects.instance.Instance.save') @mock.patch('nova.compute.utils.notify_about_instance_rebuild') @mock.patch('nova.compute.utils.notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.objects.instance.Instance.image_meta', new_callable=mock.PropertyMock) @mock.patch('nova.compute.manager.ComputeManager.' '_validate_instance_group_policy') @mock.patch('nova.compute.manager.ComputeManager._set_migration_status') @mock.patch('nova.compute.resource_tracker.ResourceTracker.rebuild_claim') def test_evacuate_late_server_group_policy_check( self, mock_rebuild_claim, mock_set_migration_status, mock_validate_policy, mock_image_meta, mock_notify_exists, mock_notify_legacy, mock_notify, mock_instance_save, mock_setup_networks, mock_setup_instance_network, mock_get_bdms, mock_mutate_migration, mock_appy_migration, mock_drop_migration, mock_context_elevated): self.flags(api_servers=['http://localhost/image/v2'], group='glance') instance = fake_instance.fake_instance_obj(self.context) instance.trusted_certs = None instance.info_cache = None elevated_context = mock.Mock() mock_context_elevated.return_value = elevated_context request_spec = objects.RequestSpec() request_spec.scheduler_hints = {'group': [uuids.group]} with mock.patch.object(self.compute, 'network_api'): self.compute.rebuild_instance( self.context, instance, None, None, None, None, None, None, recreate=True, on_shared_storage=None, preserve_ephemeral=False, migration=None, scheduled_node='fake-node', limits={}, request_spec=request_spec, accel_uuids=[], reimage_boot_volume=False, target_state=None) mock_validate_policy.assert_called_once_with( elevated_context, instance, {'group': [uuids.group]}) @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.compute.resource_tracker.ResourceTracker.' 'delete_allocation_for_evacuated_instance') @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.objects.instance.Instance.save') @mock.patch('nova.compute.utils.notify_about_instance_rebuild') @mock.patch('nova.compute.utils.notify_about_instance_usage') @mock.patch('nova.compute.manager.ComputeManager.' '_validate_instance_group_policy') @mock.patch('nova.compute.manager.ComputeManager._set_migration_status') @mock.patch('nova.compute.resource_tracker.ResourceTracker.rebuild_claim') def test_evacuate_late_server_group_policy_check_fails( self, mock_rebuild_claim, mock_set_migration_status, mock_validate_policy, mock_notify_legacy, mock_notify, mock_instance_save, mock_context_elevated, mock_delete_allocation, mock_instance_fault): instance = fake_instance.fake_instance_obj(self.context) instance.info_cache = None instance.system_metadata = {} instance.vm_state = vm_states.ACTIVE elevated_context = mock.Mock() mock_context_elevated.return_value = elevated_context request_spec = objects.RequestSpec() request_spec.scheduler_hints = {'group': [uuids.group]} exc = exception.RescheduledException( instance_uuid=instance.uuid, reason='policy violation') mock_validate_policy.side_effect = exc self.assertRaises( messaging.ExpectedException, self.compute.rebuild_instance, self.context, instance, None, None, None, None, None, None, recreate=True, on_shared_storage=None, preserve_ephemeral=False, migration=None, scheduled_node='fake-node', limits={}, request_spec=request_spec, accel_uuids=[], reimage_boot_volume=False, target_state=None) mock_validate_policy.assert_called_once_with( elevated_context, instance, {'group': [uuids.group]}) mock_delete_allocation.assert_called_once_with( elevated_context, instance, 'fake-node', node_type='destination') mock_notify.assert_called_once_with( elevated_context, instance, 'fake-mini', bdms=None, exception=exc, phase='error') # Make sure the instance vm_state did not change. self.assertEqual(vm_states.ACTIVE, instance.vm_state) def test_rebuild_node_not_updated_if_not_recreate(self): node = uuidutils.generate_uuid() # ironic node uuid instance = fake_instance.fake_instance_obj(self.context, node=node) instance.migration_context = None migration = objects.Migration(status='accepted') with test.nested( mock.patch.object(self.compute, '_get_compute_info'), mock.patch.object(self.compute, '_do_rebuild_instance_with_claim'), mock.patch.object(objects.Instance, 'save'), mock.patch.object(migration, 'save'), ) as (mock_get, mock_rebuild, mock_save, mock_migration_save): self.compute.rebuild_instance( self.context, instance, None, None, None, None, None, None, False, False, False, migration, None, {}, None, [], False, None) self.assertFalse(mock_get.called) self.assertEqual(node, instance.node) self.assertEqual('done', migration.status) mock_migration_save.assert_called_once_with() def test_rebuild_node_updated_if_recreate(self): dead_node = uuidutils.generate_uuid() img_sys_meta = {'image_hw_numa_nodes': 1} instance = fake_instance.fake_instance_obj(self.context, node=dead_node) instance.system_metadata = img_sys_meta instance.migration_context = None mock_rt = self._mock_rt() with test.nested( mock.patch.object(self.compute, '_get_compute_info'), mock.patch.object(self.compute, '_do_rebuild_instance'), ) as (mock_get, mock_rebuild): mock_get.return_value.hypervisor_hostname = 'new-node' self.compute.rebuild_instance( self.context, instance, None, None, None, None, None, None, True, False, False, mock.sentinel.migration, None, {}, None, [], False, None) mock_get.assert_called_once_with(mock.ANY, self.compute.host) mock_rt.finish_evacuation.assert_called_once_with( instance, 'new-node', mock.sentinel.migration) # Make sure the rebuild_claim was called with the proper image_meta # from the instance. mock_rt.rebuild_claim.assert_called_once() self.assertIn('image_meta', mock_rt.rebuild_claim.call_args[1]) actual_image_meta = mock_rt.rebuild_claim.call_args[1][ 'image_meta'].properties self.assertIn('hw_numa_nodes', actual_image_meta) self.assertEqual(1, actual_image_meta.hw_numa_nodes) @mock.patch.object(compute_utils, 'notify_about_instance_rebuild') @mock.patch.object(compute_utils, 'notify_usage_exists') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(objects.ImageMeta, 'from_instance') @mock.patch.object(objects.Instance, 'save', return_value=None) def test_rebuild_nw_updated_if_recreate(self, mock_save, mock_image_ref, mock_notify, mock_notify_exists, mock_notify_rebuild): with test.nested( mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute.network_api, 'setup_networks_on_host'), mock.patch.object(self.compute.network_api, 'setup_instance_network_on_host'), mock.patch.object(self.compute.network_api, 'get_instance_nw_info'), mock.patch.object(self.compute, '_get_instance_block_device_info', return_value='fake-bdminfo'), mock.patch.object(self.compute, '_check_trusted_certs'), ) as( mock_notify_usage, mock_setup, mock_setup_inst, mock_get_nw_info, mock_get_blk, mock_check_trusted_certs ): self.flags(group="glance", api_servers="http://127.0.0.1:9292") instance = fake_instance.fake_instance_obj(self.context) orig_vif = fake_network_cache_model.new_vif( {'profile': {"pci_slot": "0000:01:00.1"}}) orig_nw_info = network_model.NetworkInfo([orig_vif]) new_vif = fake_network_cache_model.new_vif( {'profile': {"pci_slot": "0000:02:00.1"}}) new_nw_info = network_model.NetworkInfo([new_vif]) info_cache = objects.InstanceInfoCache(network_info=orig_nw_info, instance_uuid=instance.uuid) instance.info_cache = info_cache instance.task_state = task_states.REBUILDING instance.migration_context = None instance.numa_topology = None instance.pci_requests = None instance.pci_devices = None orig_image_ref = None image_meta = instance.image_meta injected_files = [] new_pass = None orig_sys_metadata = None bdms = [] recreate = True on_shared_storage = None preserve_ephemeral = None mock_get_nw_info.return_value = new_nw_info self.compute._do_rebuild_instance(self.context, instance, orig_image_ref, image_meta, injected_files, new_pass, orig_sys_metadata, bdms, recreate, on_shared_storage, preserve_ephemeral, {}, {}, self.allocations, mock.sentinel.mapping, [], False, None) mock_notify_usage.assert_has_calls( [mock.call(self.context, instance, "rebuild.start", extra_usage_info=mock.ANY), mock.call(self.context, instance, "rebuild.end", network_info=new_nw_info, extra_usage_info=mock.ANY)]) self.assertTrue(mock_image_ref.called) self.assertTrue(mock_save.called) self.assertTrue(mock_notify_exists.called) mock_setup.assert_called_once_with(self.context, instance, mock.ANY) mock_setup_inst.assert_called_once_with( self.context, instance, mock.ANY, mock.ANY, provider_mappings=mock.sentinel.mapping) mock_get_nw_info.assert_called_once_with(self.context, instance) @ddt.data((False, False), (False, True), (True, False), (True, True)) @ddt.unpack def test_rebuild_default_impl(self, is_vol_backed, reimage_boot_vol): fake_image_meta = mock.MagicMock(id='fake_id') def _detach(context, bdms, detach_root_bdm=True): # NOTE(rpodolyaka): check that instance has been powered off by # the time we detach block devices, exact calls arguments will be # checked below self.assertTrue(mock_power_off.called) self.assertFalse(mock_destroy.called) def _attach(context, instance, bdms): return {'block_device_mapping': 'shared_block_storage'} def _spawn(context, instance, image_meta, injected_files, admin_password, allocations, network_info=None, block_device_info=None, accel_info=None): self.assertEqual(block_device_info['block_device_mapping'], 'shared_block_storage') with test.nested( mock.patch.object(self.compute.driver, 'destroy', return_value=None), mock.patch.object(self.compute.driver, 'spawn', side_effect=_spawn), mock.patch.object(objects.Instance, 'save', return_value=None), mock.patch.object(self.compute, '_power_off_instance', return_value=None), mock.patch.object(self.compute, '_get_accel_info', return_value=[]), mock.patch.object(compute_utils, 'is_volume_backed_instance', return_value=is_vol_backed), mock.patch.object(self.compute, '_rebuild_volume_backed_instance'), mock.patch.object(compute_utils, 'get_root_bdm') ) as( mock_destroy, mock_spawn, mock_save, mock_power_off, mock_accel_info, mock_is_volume_backed, mock_rebuild_vol_backed_inst, mock_get_root, ): instance = fake_instance.fake_instance_obj(self.context) instance.migration_context = None instance.numa_topology = None instance.pci_requests = None instance.pci_devices = None instance.device_metadata = None instance.task_state = task_states.REBUILDING instance.save(expected_task_state=[task_states.REBUILDING]) fake_block_device_info = { 'block_device_mapping': [ {'attachment_id': '341a8917-f74d-4473-8ee7-4ca05e5e0ab3', 'volume_id': 'b7c93bb9-dfe4-41af-aa56-e6b28342fd8f', 'connection_info': {'driver_volume_type': 'iscsi', 'data': {'target_discovered': False, 'target_portal': '127.0.0.1:3260', 'target_iqn': 'iqn.2010-10.org.openstack:volume-' 'b7c93bb9-dfe4-41af-aa56-e6b28342fd8f', 'target_lun': 0}}}]} self.compute._rebuild_default_impl(self.context, instance, fake_image_meta, [], admin_password='new_pass', bdms=[], allocations={}, detach_block_devices=_detach, attach_block_devices=_attach, network_info=None, evacuate=False, block_device_info= fake_block_device_info, preserve_ephemeral=False, reimage_boot_volume= reimage_boot_vol) self.assertTrue(mock_save.called) self.assertTrue(mock_spawn.called) mock_destroy.assert_called_once_with( self.context, instance, network_info=None, block_device_info=fake_block_device_info) mock_power_off.assert_called_once_with( instance, clean_shutdown=True) if is_vol_backed and reimage_boot_vol: mock_rebuild_vol_backed_inst.assert_called_once_with( self.context, instance, [], fake_image_meta.id) else: mock_rebuild_vol_backed_inst.assert_not_called() @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.attachment_create', return_value={'id': uuids.new_attachment_id}) @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') def test__rebuild_volume_backed_instance( self, wait_inst_event, attach_create, attach_delete): fake_conn_info = '{}' fake_device = 'fake_vda' root_bdm = mock.MagicMock( volume_id=uuids.volume_id, connection_info=fake_conn_info, device_name=fake_device, attachment_id=uuids.old_attachment_id, save=mock.MagicMock()) bdms = [root_bdm] events = [('volume-reimaged', root_bdm.volume_id)] image_size_gb = 1 deadline = CONF.reimage_timeout_per_gb * image_size_gb with test.nested( mock.patch.object(objects.Instance, 'save', return_value=None), mock.patch.object(compute_utils, 'get_root_bdm', return_value=root_bdm), mock.patch.object(self.compute, 'volume_api'), mock.patch.object(self.compute.image_api, 'get'), ) as ( mock_save, mock_get_root_bdm, mock_vol_api, mock_get_img ): instance = fake_instance.fake_instance_obj(self.context) instance.task_state = task_states.REBUILDING # 1024 ** 3 = 1073741824 mock_get_img.return_value = {'size': 1073741824} self.compute._rebuild_volume_backed_instance( self.context, instance, bdms, uuids.image_id) mock_vol_api.attachment_create.assert_called_once_with( self.context, uuids.volume_id, instance.uuid) mock_vol_api.attachment_delete.assert_called_once_with( self.context, uuids.old_attachment_id) mock_vol_api.reimage_volume.assert_called_once_with( self.context, uuids.volume_id, uuids.image_id, reimage_reserved=True) mock_get_img.assert_called_once_with( self.context, uuids.image_id) mock_get_root_bdm.assert_called_once_with( self.context, instance, bdms) wait_inst_event.assert_called_once_with( instance, events, deadline=deadline, error_callback=self.compute._reimage_failed_callback) @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.attachment_create', return_value={'id': uuids.new_attachment_id}) @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') def test__rebuild_volume_backed_instance_image_not_found( self, wait_inst_event, attach_create, attach_delete): fake_conn_info = '{}' fake_device = 'fake_vda' root_bdm = mock.MagicMock( volume_id=uuids.volume_id, connection_info=fake_conn_info, device_name=fake_device, attachment_id=uuids.old_attachment_id, save=mock.MagicMock()) bdms = [root_bdm] with test.nested( mock.patch.object(objects.Instance, 'save', return_value=None), mock.patch.object(compute_utils, 'get_root_bdm', return_value=root_bdm), mock.patch.object(self.compute, 'volume_api'), mock.patch.object(self.compute.image_api, 'get'), ) as( mock_save, mock_get_root_bdm, mock_vol_api, mock_get_img ): mock_get_img.side_effect = exception.ImageNotFound( image_id=uuids.image_id) instance = fake_instance.fake_instance_obj(self.context) instance.task_state = task_states.REBUILDING instance.save(expected_task_state=[task_states.REBUILDING]) mock_get_img.return_value = {'size': 1} self.assertRaises( exception.BuildAbortException, self.compute._rebuild_volume_backed_instance, self.context, instance, bdms, uuids.image_id) mock_vol_api.attachment_create.assert_called_once_with( self.context, uuids.volume_id, instance.uuid) mock_vol_api.attachment_delete.assert_called_once_with( self.context, uuids.old_attachment_id) mock_get_img.assert_called_once_with( self.context, uuids.image_id) @mock.patch.object(objects.Instance, 'save', return_value=None) @mock.patch.object(fake_driver.SmallFakeDriver, 'detach_volume') @mock.patch.object(cinder.API, 'roll_detaching') def test__detach_root_volume(self, mock_roll_detach, mock_detach, mock_save): exception_list = [ '', exception.DiskNotFound(location="not\\here"), exception.DeviceDetachFailed(device="fake_dev", reason="unknown"), ] mock_detach.side_effect = exception_list fake_conn_info = '{}' fake_device = 'fake_vda' root_bdm = mock.MagicMock( volume_id=uuids.volume_id, connection_info=fake_conn_info, device_name=fake_device, attachment_id=uuids.old_attachment_id, save=mock.MagicMock()) instance = fake_instance.fake_instance_obj(self.context) instance.task_state = task_states.REBUILDING instance.save(expected_task_state=[task_states.REBUILDING]) self.compute._detach_root_volume(self.context, instance, root_bdm) self.compute._detach_root_volume(self.context, instance, root_bdm) self.assertRaises(exception.DeviceDetachFailed, self.compute._detach_root_volume, self.context, instance, root_bdm) mock_roll_detach.assert_called_with(self.context, uuids.volume_id) self.assertRaises(Exception, self.compute._detach_root_volume, # noqa self.context, instance, root_bdm) mock_roll_detach.assert_called_with(self.context, uuids.volume_id) def test_do_rebuild_instance_check_trusted_certs(self): """Tests the scenario that we're rebuilding an instance with trusted_certs on a host that does not support trusted certs so a BuildAbortException is raised. """ instance = self._trusted_certs_setup_instance() instance.system_metadata = {} with mock.patch.dict(self.compute.driver.capabilities, supports_trusted_certs=False): ex = self.assertRaises( exception.BuildAbortException, self.compute._do_rebuild_instance, self.context, instance, instance.image_ref, instance.image_meta, injected_files=[], new_pass=None, orig_sys_metadata={}, bdms=objects.BlockDeviceMapping(), evacuate=False, on_shared_storage=None, preserve_ephemeral=False, migration=objects.Migration(), request_spec=objects.RequestSpec(), allocations=self.allocations, request_group_resource_providers_mapping=mock.sentinel.mapping, accel_uuids=[], reimage_boot_volume=False, target_state=None) self.assertIn('Trusted image certificates provided on host', str(ex)) def test_reverts_task_state_instance_not_found(self): # Tests that the reverts_task_state decorator in the compute manager # will not trace when an InstanceNotFound is raised. instance = objects.Instance(uuid=uuids.instance, task_state="FAKE") instance_update_mock = mock.Mock( side_effect=exception.InstanceNotFound(instance_id=instance.uuid)) self.compute._instance_update = instance_update_mock log_mock = mock.Mock() manager.LOG = log_mock @manager.reverts_task_state def fake_function(self, context, instance): raise test.TestingException() self.assertRaises(test.TestingException, fake_function, self, self.context, instance) self.assertFalse(log_mock.called) @mock.patch.object(nova.scheduler.client.query.SchedulerQueryClient, 'update_instance_info') def test_update_scheduler_instance_info(self, mock_update): instance = objects.Instance(uuid=uuids.instance) self.compute._update_scheduler_instance_info(self.context, instance) self.assertEqual(mock_update.call_count, 1) args = mock_update.call_args[0] self.assertNotEqual(args[0], self.context) self.assertIsInstance(args[0], self.context.__class__) self.assertEqual(args[1], self.compute.host) # Send a single instance; check that the method converts to an # InstanceList self.assertIsInstance(args[2], objects.InstanceList) self.assertEqual(args[2].objects[0], instance) @mock.patch.object(nova.scheduler.client.query.SchedulerQueryClient, 'delete_instance_info') def test_delete_scheduler_instance_info(self, mock_delete): self.compute._delete_scheduler_instance_info(self.context, mock.sentinel.inst_uuid) self.assertEqual(mock_delete.call_count, 1) args = mock_delete.call_args[0] self.assertNotEqual(args[0], self.context) self.assertIsInstance(args[0], self.context.__class__) self.assertEqual(args[1], self.compute.host) self.assertEqual(args[2], mock.sentinel.inst_uuid) @ddt.data(('vnc', 'spice', 'rdp', 'serial_console', 'mks'), ('spice', 'vnc', 'rdp', 'serial_console', 'mks'), ('rdp', 'vnc', 'spice', 'serial_console', 'mks'), ('serial_console', 'vnc', 'spice', 'rdp', 'mks'), ('mks', 'vnc', 'spice', 'rdp', 'serial_console')) @ddt.unpack @mock.patch('nova.objects.ConsoleAuthToken.' 'clean_console_auths_for_instance') def test_clean_instance_console_tokens(self, g1, g2, g3, g4, g5, mock_clean): # Enable one of each of the console types and disable the rest self.flags(enabled=True, group=g1) for g in [g2, g3, g4, g5]: self.flags(enabled=False, group=g) instance = objects.Instance(uuid=uuids.instance) self.compute._clean_instance_console_tokens(self.context, instance) mock_clean.assert_called_once_with(self.context, instance.uuid) @mock.patch('nova.objects.ConsoleAuthToken.' 'clean_console_auths_for_instance') def test_clean_instance_console_tokens_no_consoles_enabled(self, mock_clean): for g in ['vnc', 'spice', 'rdp', 'serial_console', 'mks']: self.flags(enabled=False, group=g) instance = objects.Instance(uuid=uuids.instance) self.compute._clean_instance_console_tokens(self.context, instance) mock_clean.assert_not_called() @mock.patch('nova.objects.ConsoleAuthToken.clean_expired_console_auths') def test_cleanup_expired_console_auth_tokens(self, mock_clean): self.compute._cleanup_expired_console_auth_tokens(self.context) mock_clean.assert_called_once_with(self.context) @mock.patch.object(nova.context.RequestContext, 'elevated') @mock.patch.object(nova.objects.InstanceList, 'get_by_host') @mock.patch.object(nova.scheduler.client.query.SchedulerQueryClient, 'sync_instance_info') def test_sync_scheduler_instance_info(self, mock_sync, mock_get_by_host, mock_elevated): inst1 = objects.Instance(uuid=uuids.instance_1) inst2 = objects.Instance(uuid=uuids.instance_2) inst3 = objects.Instance(uuid=uuids.instance_3) exp_uuids = [inst.uuid for inst in [inst1, inst2, inst3]] mock_get_by_host.return_value = objects.InstanceList( objects=[inst1, inst2, inst3]) fake_elevated = context.get_admin_context() mock_elevated.return_value = fake_elevated self.compute._sync_scheduler_instance_info(self.context) mock_get_by_host.assert_called_once_with( fake_elevated, self.compute.host, expected_attrs=[], use_slave=True) mock_sync.assert_called_once_with(fake_elevated, self.compute.host, exp_uuids) @mock.patch.object(nova.scheduler.client.query.SchedulerQueryClient, 'sync_instance_info') @mock.patch.object(nova.scheduler.client.query.SchedulerQueryClient, 'delete_instance_info') @mock.patch.object(nova.scheduler.client.query.SchedulerQueryClient, 'update_instance_info') def test_scheduler_info_updates_off(self, mock_update, mock_delete, mock_sync): mgr = self.compute mgr.send_instance_updates = False mgr._update_scheduler_instance_info(self.context, mock.sentinel.instance) mgr._delete_scheduler_instance_info(self.context, mock.sentinel.instance_uuid) mgr._sync_scheduler_instance_info(self.context) # None of the calls should have been made self.assertFalse(mock_update.called) self.assertFalse(mock_delete.called) self.assertFalse(mock_sync.called) def test_set_instance_obj_error_state_with_clean_task_state(self): instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.BUILDING, task_state=task_states.SPAWNING) with mock.patch.object(instance, 'save'): self.compute._set_instance_obj_error_state(instance, clean_task_state=True) self.assertEqual(vm_states.ERROR, instance.vm_state) self.assertIsNone(instance.task_state) def test_set_instance_obj_error_state_by_default(self): instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.BUILDING, task_state=task_states.SPAWNING) with mock.patch.object(instance, 'save'): self.compute._set_instance_obj_error_state(instance) self.assertEqual(vm_states.ERROR, instance.vm_state) self.assertEqual(task_states.SPAWNING, instance.task_state) @mock.patch.object(objects.Instance, 'save') def test_instance_update(self, mock_save): instance = objects.Instance(task_state=task_states.SCHEDULING, vm_state=vm_states.BUILDING) updates = {'task_state': None, 'vm_state': vm_states.ERROR} with mock.patch.object(self.compute, '_update_resource_tracker') as mock_rt: self.compute._instance_update(self.context, instance, **updates) self.assertIsNone(instance.task_state) self.assertEqual(vm_states.ERROR, instance.vm_state) mock_save.assert_called_once_with() mock_rt.assert_called_once_with(self.context, instance) def test_reset_reloads_rpcapi(self): orig_rpc = self.compute.compute_rpcapi with mock.patch('nova.compute.rpcapi.ComputeAPI') as mock_rpc: self.compute.reset() mock_rpc.assert_called_once_with() self.assertIsNot(orig_rpc, self.compute.compute_rpcapi) def test_reset_clears_provider_cache(self): # Seed the cache so we can tell we've cleared it reportclient = self.compute.reportclient ptree = reportclient._provider_tree ptree.new_root('foo', uuids.foo) self.assertEqual([uuids.foo], ptree.get_provider_uuids()) times = reportclient._association_refresh_time times[uuids.foo] = time.time() self.compute.reset() ptree = reportclient._provider_tree self.assertEqual([], ptree.get_provider_uuids()) times = reportclient._association_refresh_time self.assertEqual({}, times) @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') @mock.patch('nova.compute.manager.ComputeManager._delete_instance') def test_terminate_instance_no_bdm_volume_id(self, mock_delete_instance, mock_bdm_get_by_inst): # Tests that we refresh the bdm list if a volume bdm does not have the # volume_id set. instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ERROR, task_state=task_states.DELETING) bdm = fake_block_device.FakeDbBlockDeviceDict( {'source_type': 'snapshot', 'destination_type': 'volume', 'instance_uuid': instance.uuid, 'device_name': '/dev/vda'}) bdms = block_device_obj.block_device_make_list(self.context, [bdm]) # since the bdms passed in don't have a volume_id, we'll go back to the # database looking for updated versions mock_bdm_get_by_inst.return_value = bdms self.compute.terminate_instance(self.context, instance, bdms) mock_bdm_get_by_inst.assert_called_once_with( self.context, instance.uuid) mock_delete_instance.assert_called_once_with( self.context, instance, bdms) @mock.patch('nova.context.RequestContext.elevated') def test_terminate_instance_no_network_info(self, mock_elevated): # Tests that we refresh the network info if it was empty instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE) empty_nw_info = network_model.NetworkInfo() instance.info_cache = objects.InstanceInfoCache( network_info=empty_nw_info) vif = fake_network_cache_model.new_vif() nw_info = network_model.NetworkInfo([vif]) bdms = objects.BlockDeviceMappingList() elevated = context.get_admin_context() mock_elevated.return_value = elevated # Call shutdown instance with test.nested( mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value=nw_info), mock.patch.object(self.compute, '_get_instance_block_device_info'), mock.patch.object(self.compute.driver, 'destroy') ) as ( mock_nw_api_info, mock_get_bdi, mock_destroy ): self.compute._shutdown_instance(self.context, instance, bdms, notify=False, try_deallocate_networks=False) # Verify mock_nw_api_info.assert_called_once_with(elevated, instance) mock_get_bdi.assert_called_once_with(elevated, instance, bdms=bdms) # destroy should have been called with the refresh network_info mock_destroy.assert_called_once_with( elevated, instance, nw_info, mock_get_bdi.return_value) @mock.patch('nova.compute.utils.notify_about_instance_action') @mock.patch.object(nova.compute.manager.ComputeManager, '_notify_about_instance_usage') @mock.patch.object(compute_utils, 'EventReporter') def test_trigger_crash_dump(self, event_mock, notify_mock, mock_instance_action_notify): instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE) self.compute.trigger_crash_dump(self.context, instance) mock_instance_action_notify.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', action='trigger_crash_dump', phase='start'), mock.call(self.context, instance, 'fake-mini', action='trigger_crash_dump', phase='end')]) notify_mock.assert_has_calls([ mock.call(self.context, instance, 'trigger_crash_dump.start'), mock.call(self.context, instance, 'trigger_crash_dump.end') ]) self.assertIsNone(instance.task_state) self.assertEqual(vm_states.ACTIVE, instance.vm_state) def test_instance_restore_notification(self): inst_obj = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.SOFT_DELETED) with test.nested( mock.patch.object(nova.compute.utils, 'notify_about_instance_action'), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(objects.Instance, 'save'), mock.patch.object(self.compute.driver, 'restore') ) as ( fake_notify, fake_usage, fake_save, fake_restore ): self.compute.restore_instance(self.context, inst_obj) fake_notify.assert_has_calls([ mock.call(self.context, inst_obj, 'fake-mini', action='restore', phase='start'), mock.call(self.context, inst_obj, 'fake-mini', action='restore', phase='end')]) def test_delete_image_on_error_image_not_found_ignored(self): """Tests that we don't log an exception trace if we get a 404 when trying to delete an image as part of the image cleanup decorator. """ @manager.delete_image_on_error def some_image_related_op(self, context, image_id, instance): raise test.TestingException('oops!') image_id = uuids.image_id instance = objects.Instance(uuid=uuids.instance_uuid) with mock.patch.object(manager.LOG, 'exception') as mock_log: with mock.patch.object( self, 'image_api', create=True) as mock_image_api: mock_image_api.delete.side_effect = ( exception.ImageNotFound(image_id=image_id)) self.assertRaises(test.TestingException, some_image_related_op, self, self.context, image_id, instance) mock_image_api.delete.assert_called_once_with( self.context, image_id) # make sure nothing was logged at exception level mock_log.assert_not_called() @mock.patch('nova.volume.cinder.API.attachment_delete') @mock.patch('nova.volume.cinder.API.attachment_create', return_value={'id': uuids.attachment_id}) @mock.patch('nova.objects.BlockDeviceMapping.save') @mock.patch('nova.volume.cinder.API.terminate_connection') def test_terminate_volume_connections(self, mock_term_conn, mock_bdm_save, mock_attach_create, mock_attach_delete): """Tests _terminate_volume_connections with cinder v2 style, cinder v3.44 style, and non-volume BDMs. """ bdms = objects.BlockDeviceMappingList( objects=[ # We use two old-style BDMs to make sure we only build the # connector object once. objects.BlockDeviceMapping(volume_id=uuids.v2_volume_id_1, destination_type='volume', attachment_id=None), objects.BlockDeviceMapping(volume_id=uuids.v2_volume_id_2, destination_type='volume', attachment_id=None), objects.BlockDeviceMapping(volume_id=uuids.v3_volume_id, destination_type='volume', attachment_id=uuids.attach_id), objects.BlockDeviceMapping(volume_id=None, destination_type='local') ]) instance = fake_instance.fake_instance_obj( self.context, vm_state=vm_states.ACTIVE) fake_connector = mock.sentinel.fake_connector with mock.patch.object(self.compute.driver, 'get_volume_connector', return_value=fake_connector) as connector_mock: self.compute._terminate_volume_connections( self.context, instance, bdms) # assert we called terminate_connection twice (once per old volume bdm) mock_term_conn.assert_has_calls([ mock.call(self.context, uuids.v2_volume_id_1, fake_connector), mock.call(self.context, uuids.v2_volume_id_2, fake_connector) ]) # assert we only build the connector once connector_mock.assert_called_once_with(instance) # assert we called delete_attachment once for the single new volume bdm mock_attach_delete.assert_called_once_with( self.context, uuids.attach_id) mock_attach_create.assert_called_once_with( self.context, uuids.v3_volume_id, instance.uuid) def test_instance_soft_delete_notification(self): inst_obj = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.ACTIVE) inst_obj.system_metadata = {} with test.nested( mock.patch.object(nova.compute.utils, 'notify_about_instance_action'), mock.patch.object(nova.compute.utils, 'notify_about_instance_usage'), mock.patch.object(objects.Instance, 'save'), mock.patch.object(self.compute.driver, 'soft_delete') ) as (fake_notify, fake_legacy_notify, fake_save, fake_soft_delete): self.compute.soft_delete_instance(self.context, inst_obj) fake_notify.assert_has_calls([ mock.call(self.context, inst_obj, action='soft_delete', source='nova-compute', host='fake-mini', phase='start'), mock.call(self.context, inst_obj, action='soft_delete', source='nova-compute', host='fake-mini', phase='end')]) def test_get_scheduler_hints(self): # 1. No hints and no request_spec. self.assertEqual({}, self.compute._get_scheduler_hints({})) # 2. Hints come from the filter_properties. hints = {'foo': 'bar'} filter_properties = {'scheduler_hints': hints} self.assertEqual( hints, self.compute._get_scheduler_hints(filter_properties)) # 3. Hints come from filter_properties because reqspec is empty. reqspec = objects.RequestSpec.from_primitives(self.context, {}, {}) self.assertEqual( hints, self.compute._get_scheduler_hints( filter_properties, reqspec)) # 4. Hints come from the request spec. reqspec_hints = {'boo': 'baz'} reqspec = objects.RequestSpec.from_primitives( self.context, {}, {'scheduler_hints': reqspec_hints}) # The RequestSpec unconditionally stores hints as a key=list # unlike filter_properties which just stores whatever came in from # the API request. expected_reqspec_hints = {'boo': ['baz']} self.assertDictEqual( expected_reqspec_hints, self.compute._get_scheduler_hints( filter_properties, reqspec)) def test_notify_volume_usage_detach_no_block_stats(self): """Tests the case that the virt driver returns None from the block_stats() method and no notification is sent, similar to the virt driver raising NotImplementedError. """ self.flags(volume_usage_poll_interval=60) fake_instance = objects.Instance() fake_bdm = objects.BlockDeviceMapping(device_name='/dev/vda') with mock.patch.object(self.compute.driver, 'block_stats', return_value=None) as block_stats: # Assert a notification isn't sent. with mock.patch.object(self.compute.notifier, 'info', new_callable=mock.NonCallableMock): self.compute._notify_volume_usage_detach( self.context, fake_instance, fake_bdm) block_stats.assert_called_once_with(fake_instance, 'vda') @mock.patch('nova.compute.manager.LOG') def test_cache_images_unsupported(self, mock_log): r = self.compute.cache_images(self.context, ['an-image']) self.assertEqual({'an-image': 'unsupported'}, r) mock_log.warning.assert_called_once_with( 'Virt driver does not support image pre-caching; ignoring request') def test_cache_image_existing(self): with mock.patch.object(self.compute.driver, 'cache_image') as c: c.return_value = False r = self.compute.cache_images(self.context, ['an-image']) self.assertEqual({'an-image': 'existing'}, r) def test_cache_image_downloaded(self): with mock.patch.object(self.compute.driver, 'cache_image') as c: c.return_value = True r = self.compute.cache_images(self.context, ['an-image']) self.assertEqual({'an-image': 'cached'}, r) def test_cache_image_failed(self): with mock.patch.object(self.compute.driver, 'cache_image') as c: c.side_effect = test.TestingException('foo') r = self.compute.cache_images(self.context, ['an-image']) self.assertEqual({'an-image': 'error'}, r) def test_cache_images_multi(self): with mock.patch.object(self.compute.driver, 'cache_image') as c: c.side_effect = [True, False] r = self.compute.cache_images(self.context, ['one-image', 'two-image']) self.assertEqual({'one-image': 'cached', 'two-image': 'existing'}, r) @mock.patch.object(virt_node, 'write_local_node_uuid') @mock.patch.object(virt_node, 'read_local_node_uuid') def test_ensure_node_uuid_not_needed_version(self, mock_read, mock_write): # Make sure an up-to-date service bypasses the persistence service_ref = service_obj.Service() self.assertEqual(service_obj.SERVICE_VERSION, service_ref.version) mock_read.return_value = 'not none' mock_write.assert_not_called() self.compute._ensure_existing_node_identity(service_ref) mock_write.assert_not_called() @mock.patch.object(virt_node, 'write_local_node_uuid') def test_ensure_node_uuid_not_needed_ironic(self, mock_node): # Make sure an old service for ironic does not write a local node uuid with mock.patch.object(service_obj, 'SERVICE_VERSION', new=60): service_ref = service_obj.Service() self.assertEqual(60, service_ref.version) self.flags(compute_driver='ironic') self.compute._ensure_existing_node_identity(service_ref) mock_node.assert_not_called() @mock.patch.object(virt_node, 'write_local_node_uuid') @mock.patch.object(virt_node, 'read_local_node_uuid') def test_ensure_node_uuid_not_needed_preprovisioned(self, mock_read_node, mock_write_node): # Make sure an old service does not write a uuid if one is present with mock.patch.object(service_obj, 'SERVICE_VERSION', new=60): service_ref = service_obj.Service() self.assertEqual(60, service_ref.version) mock_read_node.return_value = str(uuids.SOME_UUID) self.compute._ensure_existing_node_identity(service_ref) mock_write_node.assert_not_called() @mock.patch.object(virt_node, 'write_local_node_uuid') @mock.patch.object(virt_node, 'read_local_node_uuid') @mock.patch('nova.objects.ComputeNodeList.get_all_by_host') def test_ensure_node_uuid_upgrade_no_node(self, mock_get_cn, mock_read_node, mock_write_node): # If we are not a new service, we have no pre-provisioned node uuid # and we find no nodes in the database, we do not write a local # node uuid *and* we abort startup since something is likely wrong. with mock.patch.object(service_obj, 'SERVICE_VERSION', new=60): service_ref = service_obj.Service() self.assertEqual(60, service_ref.version) mock_read_node.return_value = None mock_get_cn.return_value = [] self.assertRaises(exception.InvalidConfiguration, self.compute._ensure_existing_node_identity, service_ref) mock_get_cn.assert_called_once_with(mock.ANY, self.compute.host) mock_write_node.assert_not_called() @mock.patch.object(virt_node, 'write_local_node_uuid') @mock.patch.object(virt_node, 'read_local_node_uuid') @mock.patch('nova.objects.ComputeNodeList.get_all_by_host') def test_ensure_node_uuid_upgrade_multi_node(self, mock_get_cn, mock_read_node, mock_write_node): # If we are not a new service, we have no pre-provisioned node uuid # and we find multiple nodes in the database, we do not write a local # node uuid *and* we abort startup since something is likely wrong. with mock.patch.object(service_obj, 'SERVICE_VERSION', new=60): service_ref = service_obj.Service() self.assertEqual(60, service_ref.version) mock_read_node.return_value = None mock_get_cn.return_value = [1, 2] self.assertRaises(exception.InvalidConfiguration, self.compute._ensure_existing_node_identity, service_ref) mock_get_cn.assert_called_once_with(mock.ANY, self.compute.host) mock_write_node.assert_not_called() @mock.patch.object(virt_node, 'write_local_node_uuid') @mock.patch.object(virt_node, 'read_local_node_uuid') @mock.patch('nova.objects.ComputeNodeList.get_all_by_host') def test_ensure_node_uuid_upgrade_writes_node_uuid(self, mock_get_cn, mock_read_node, mock_write_node): # If we are not a new service, there is no pre-provisioned local # compute node uuid, and we find exactly one compute node in the # database for our host, we persist that. with mock.patch.object(service_obj, 'SERVICE_VERSION', new=60): service_ref = service_obj.Service() self.assertEqual(60, service_ref.version) mock_read_node.return_value = None mock_get_cn.return_value = [ objects.ComputeNode(uuid=str(uuids.compute)), ] self.compute._ensure_existing_node_identity(service_ref) mock_get_cn.assert_called_once_with(mock.ANY, self.compute.host) mock_write_node.assert_called_once_with(str(uuids.compute)) @mock.patch.object(virt_node, 'read_local_node_uuid') def test_ensure_node_uuid_missing_file_ironic(self, mock_read): mock_service = mock.MagicMock( version=service_obj.NODE_IDENTITY_VERSION) mock_read.return_value = None self.assertRaises(exception.InvalidConfiguration, self.compute._ensure_existing_node_identity, mock_service) mock_read.assert_called_once_with() # Now make sure that ironic causes this exact configuration to pass self.flags(compute_driver='ironic') self.compute._ensure_existing_node_identity(mock_service) def test_ensure_node_uuid_called_by_init_host(self): # test_init_host() above ensures that we do not call # _ensure_existing_node_identity() in the service_ref=None case. # Since testing init_host() requires a billion mocks, this # tests that we do call it when expected, but make it raise # to avoid running the rest of init_host(). with mock.patch.object(self.compute, '_ensure_existing_node_identity') as m: m.side_effect = test.TestingException self.assertRaises(test.TestingException, self.compute.init_host, mock.sentinel.service_ref) m.assert_called_once_with(mock.sentinel.service_ref) def test_check_for_host_rename_ironic(self): self.flags(compute_driver='ironic') # Passing None here makes sure we take the early exit because of our # virt driver nodes = {uuids.node1: mock.MagicMock(uuid=uuids.node1, host='not-this-host')} self.compute._check_for_host_rename(nodes) def test_check_for_host_rename_renamed_only(self): nodes = {uuids.node1: mock.MagicMock(uuid=uuids.node1, host='not-this-host')} self.assertRaises(exception.InvalidConfiguration, self.compute._check_for_host_rename, nodes) def test_check_for_host_rename_renamed_one(self): nodes = {uuids.node1: mock.MagicMock(uuid=uuids.node1, host=self.compute.host), uuids.node2: mock.MagicMock(uuid=uuids.node2, host='not-this-host')} self.assertRaises(exception.InvalidConfiguration, self.compute._check_for_host_rename, nodes) def test_check_for_host_rename_not_renamed(self): nodes = {uuids.node1: mock.MagicMock(uuid=uuids.node1, host=self.compute.host)} with mock.patch.object(manager.LOG, 'debug') as mock_debug: self.compute._check_for_host_rename(nodes) mock_debug.assert_called_once_with( 'Verified node %s matches my host %s', uuids.node1, self.compute.host) @mock.patch('nova.compute.manager.ComputeManager._get_nodes') def test_check_for_host_rename_called_by_init_host(self, mock_nodes): # Since testing init_host() requires a billion mocks, this # tests that we do call it when expected, but make it raise # to avoid running the rest of init_host(). with mock.patch.object(self.compute, '_check_for_host_rename') as m: m.side_effect = test.TestingException self.assertRaises(test.TestingException, self.compute.init_host, None) m.assert_called_once_with(mock_nodes.return_value) class ComputeManagerBuildInstanceTestCase(test.NoDBTestCase): def setUp(self): super(ComputeManagerBuildInstanceTestCase, self).setUp() self.compute = manager.ComputeManager() self.context = context.RequestContext(fakes.FAKE_USER_ID, fakes.FAKE_PROJECT_ID) self.instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.ACTIVE, expected_attrs=['metadata', 'system_metadata', 'info_cache']) self.instance.trusted_certs = None # avoid lazy-load failures self.instance.pci_requests = None # avoid lazy-load failures self.admin_pass = 'pass' self.injected_files = [] self.image = {} self.node = 'fake-node' self.limits = {} self.requested_networks = [] self.security_groups = [] self.block_device_mapping = [] self.accel_uuids = None self.filter_properties = {'retry': {'num_attempts': 1, 'hosts': [[self.compute.host, 'fake-node']]}} self.resource_provider_mapping = None self.accel_uuids = [] self.network_arqs = {} self.useFixture(fixtures.SpawnIsSynchronousFixture()) def fake_network_info(): return network_model.NetworkInfo([{'address': '1.2.3.4'}]) self.network_info = network_model.NetworkInfoAsyncWrapper( fake_network_info) self.block_device_info = self.compute._prep_block_device(context, self.instance, self.block_device_mapping) # override tracker with a version that doesn't need the database: fake_rt = fake_resource_tracker.FakeResourceTracker(self.compute.host, self.compute.driver) self.compute.rt = fake_rt self.compute.driver._set_nodes([self.node]) self.compute.rt.compute_nodes = {self.node: objects.ComputeNode()} self.allocations = { uuids.provider1: { "generation": 0, "resources": { "VCPU": 1, "MEMORY_MB": 512 } } } self.mock_get_allocs = self.useFixture( fixtures.fixtures.MockPatchObject( self.compute.reportclient, 'get_allocations_for_consumer')).mock self.mock_get_allocs.return_value = self.allocations def _do_build_instance_update(self, mock_save, reschedule_update=False): mock_save.return_value = self.instance if reschedule_update: mock_save.side_effect = (self.instance, self.instance) @staticmethod def _assert_build_instance_update(mock_save, reschedule_update=False): if reschedule_update: mock_save.assert_has_calls([ mock.call(expected_task_state=(task_states.SCHEDULING, None)), mock.call()]) else: mock_save.assert_called_once_with(expected_task_state= (task_states.SCHEDULING, None)) def _instance_action_events(self, mock_start, mock_finish): mock_start.assert_called_once_with(self.context, self.instance.uuid, mock.ANY, host=CONF.host, want_result=False) mock_finish.assert_called_once_with(self.context, self.instance.uuid, mock.ANY, exc_val=mock.ANY, exc_tb=mock.ANY, want_result=False) @mock.patch.object(objects.Instance, 'save') @mock.patch.object(nova.compute.manager.ComputeManager, '_default_block_device_names') @mock.patch.object(nova.compute.manager.ComputeManager, '_prep_block_device') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(nova.compute.manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_networks_before_block_device_mapping') def _test_accel_build_resources(self, accel_uuids, network_arqs, requested_networks, mock_prep_net, mock_build_net, mock_prep_spawn, mock_prep_bd, mock_bdnames, mock_save): args = (self.context, self.instance, requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, accel_uuids) resources = [] with self.compute._build_resources(*args) as resources: pass mock_build_net.assert_called_once_with(self.context, self.instance, requested_networks, mock.ANY, mock.ANY, network_arqs) return resources @mock.patch.object(nova.compute.manager.ComputeManager, '_get_bound_arq_resources') def test_accel_build_resources_no_device_profile(self, mock_get_arqs): # If dp_name is None, accel path is a no-op. self.instance.flavor.extra_specs = {} self._test_accel_build_resources(None, {}, self.requested_networks) mock_get_arqs.assert_not_called() @mock.patch.object(nova.compute.manager.ComputeManager, '_get_bound_arq_resources') def test_accel_build_resources(self, mock_get_arqs): # Happy path for accels in build_resources dp_name = "mydp" self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_list = [fixtures.CyborgFixture.bound_arq_list[0]] mock_get_arqs.return_value = arq_list arq_uuids = [arq['uuid'] for arq in arq_list] resources = self._test_accel_build_resources(arq_uuids, {}, self.requested_networks) mock_get_arqs.assert_called_once_with( self.context, self.instance, arq_uuids) self.assertEqual(sorted(resources['accel_info']), sorted(arq_list)) @mock.patch.object(compute_utils, 'delete_arqs_if_needed') @mock.patch.object(nova.compute.manager.ComputeManager, '_get_bound_arq_resources') def test_accel_build_resources_exception(self, mock_get_arqs, mock_clean_arq): dp_name = "mydp" self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_list = fixtures.CyborgFixture.bound_arq_list mock_get_arqs.return_value = arq_list # ensure there are arqs arq_uuids = [arq['uuid'] for arq in arq_list] mock_get_arqs.side_effect = ( exception.AcceleratorRequestOpFailed(op='get', msg='')) self.assertRaises(exception.BuildAbortException, self._test_accel_build_resources, arq_uuids, None, self.requested_networks) mock_clean_arq.assert_called_once_with( self.context, self.instance, arq_uuids) @mock.patch.object(nova.compute.manager.ComputeManager, '_get_bound_arq_resources') def test_accel_build_resources_with_port_device_profile(self, mock_get_arqs): # If dp_name is None, accel path is a no-op. self.instance.flavor.extra_specs = {} arqs = [ {'uuid': uuids.arq_uuid1}, {'uuid': uuids.arq_uuid2}, {'uuid': uuids.arq_uuid3}] mock_get_arqs.return_value = arqs accel_uuids = [arqs[0]['uuid'], arqs[1]['uuid'], arqs[2]['uuid']] request_tuples = [('123', '1.2.3.4', uuids.fakeid, None, arqs[0]['uuid'], 'smart_nic')] requests = objects.NetworkRequestList.from_tuples(request_tuples) expect_spec_arqs = [arqs[1], arqs[2]] expect_port_arqs = {arqs[0]['uuid']: arqs[0]} resources = self._test_accel_build_resources(accel_uuids, expect_port_arqs, requests) mock_get_arqs.assert_called_with(self.context, self.instance, accel_uuids) self.assertEqual(expect_spec_arqs, resources['accel_info']) @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'exit_wait_early') @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'get_arqs_for_instance') def test_arq_bind_wait_exit_early(self, mock_get_arqs, mock_wait_inst_ev, mock_exit_wait_early): # Bound ARQs available on first query, quit early. dp_name = fixtures.CyborgFixture.dp_name arq_list = [fixtures.CyborgFixture.bound_arq_list[0]] self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_events = [('accelerator-request-bound', arq['uuid']) for arq in arq_list] arq_uuids = [arq['uuid'] for arq in arq_list] mock_get_arqs.return_value = arq_list ret_arqs = self.compute._get_bound_arq_resources( self.context, self.instance, arq_uuids) mock_wait_inst_ev.assert_called_once_with( self.instance, arq_events, deadline=mock.ANY) mock_exit_wait_early.assert_called_once_with(arq_events) mock_get_arqs.assert_has_calls([ mock.call(self.instance.uuid, only_resolved=True)]) self.assertEqual(sorted(ret_arqs), sorted(arq_list)) @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'exit_wait_early') @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'get_arqs_for_instance') def test_arq_bind_wait_exit_early_no_arq_uuids(self, mock_get_arqs, mock_wait_inst_ev, mock_exit_wait_early): # If no ARQ UUIDs are passed in, call Cyborg to get the ARQs. # Then, if bound ARQs available on first query, quit early. dp_name = fixtures.CyborgFixture.dp_name arq_list = [fixtures.CyborgFixture.bound_arq_list[0]] self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_events = [('accelerator-request-bound', arq['uuid']) for arq in arq_list] mock_get_arqs.side_effect = [arq_list, arq_list] ret_arqs = self.compute._get_bound_arq_resources( self.context, self.instance, arq_uuids=None) mock_wait_inst_ev.assert_called_once_with( self.instance, arq_events, deadline=mock.ANY) mock_exit_wait_early.assert_called_once_with(arq_events) mock_get_arqs.assert_has_calls([ mock.call(self.instance.uuid), mock.call(self.instance.uuid, only_resolved=True)]) self.assertEqual(sorted(ret_arqs), sorted(arq_list)) @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'exit_wait_early') @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'get_arqs_for_instance') def test_arq_bind_wait(self, mock_get_arqs, mock_wait_inst_ev, mock_exit_wait_early): # If binding is in progress, must wait. dp_name = fixtures.CyborgFixture.dp_name arq_list = [fixtures.CyborgFixture.bound_arq_list[0]] self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_events = [('accelerator-request-bound', arq['uuid']) for arq in arq_list] arq_uuids = [arq['uuid'] for arq in arq_list] # get_arqs_for_instance gets called 2 times, returning the # resolved ARQs first, and the full list finally mock_get_arqs.side_effect = [[], arq_list] ret_arqs = self.compute._get_bound_arq_resources( self.context, self.instance, arq_uuids) mock_wait_inst_ev.assert_called_once_with( self.instance, arq_events, deadline=mock.ANY) mock_exit_wait_early.assert_not_called() self.assertEqual(sorted(ret_arqs), sorted(arq_list)) mock_get_arqs.assert_has_calls([ mock.call(self.instance.uuid, only_resolved=True), mock.call(self.instance.uuid)]) @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'exit_wait_early') @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'get_arqs_for_instance') def test_arq_bind_timeout(self, mock_get_arqs, mock_wait_inst_ev, mock_exit_wait_early): # If binding fails even after wait, exception is thrown dp_name = fixtures.CyborgFixture.dp_name arq_list = fixtures.CyborgFixture.bound_arq_list self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_events = [('accelerator-request-bound', arq['uuid']) for arq in arq_list] arq_uuids = [arq['uuid'] for arq in arq_list] mock_get_arqs.return_value = arq_list mock_wait_inst_ev.side_effect = eventlet_timeout.Timeout self.assertRaises(eventlet_timeout.Timeout, self.compute._get_bound_arq_resources, self.context, self.instance, arq_uuids) mock_wait_inst_ev.assert_called_once_with( self.instance, arq_events, deadline=mock.ANY) mock_exit_wait_early.assert_not_called() mock_get_arqs.assert_not_called() @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'exit_wait_early') @mock.patch.object(nova.compute.manager.ComputeVirtAPI, 'wait_for_instance_event') @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'get_arqs_for_instance') def test_arq_bind_exception(self, mock_get_arqs, mock_wait_inst_ev, mock_exit_wait_early): # If the code inside the context manager of _get_bound_arq_resources # raises an exception, that exception must be handled. dp_name = fixtures.CyborgFixture.dp_name arq_list = fixtures.CyborgFixture.bound_arq_list self.instance.flavor.extra_specs = {"accel:device_profile": dp_name} arq_events = [('accelerator-request-bound', arq['uuid']) for arq in arq_list] arq_uuids = [arq['uuid'] for arq in arq_list] mock_get_arqs.side_effect = ( exception.AcceleratorRequestOpFailed(op='', msg='')) self.assertRaises(exception.AcceleratorRequestOpFailed, self.compute._get_bound_arq_resources, self.context, self.instance, arq_uuids) mock_wait_inst_ev.assert_called_once_with( self.instance, arq_events, deadline=mock.ANY) mock_exit_wait_early.assert_not_called() mock_get_arqs.assert_called_once_with( self.instance.uuid, only_resolved=True) @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch.object(fake_driver.FakeDriver, 'spawn') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_consumer') @mock.patch.object(manager.ComputeManager, '_get_request_group_mapping') @mock.patch.object(manager.ComputeManager, '_check_trusted_certs') @mock.patch.object(manager.ComputeManager, '_check_device_tagging') @mock.patch.object(compute_utils, 'notify_about_instance_create') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') def test_spawn_called_with_accel_info(self, mock_ins_usage, mock_ins_create, mock_dev_tag, mock_certs, mock_req_group_map, mock_get_allocations, mock_ins_save, mock_spawn, mock_claim): accel_info = [{'k1': 'v1', 'k2': 'v2'}] @contextlib.contextmanager def fake_build_resources(compute_mgr, *args, **kwargs): yield { 'block_device_info': None, 'network_info': None, 'accel_info': accel_info, } self.stub_out('nova.compute.manager.ComputeManager._build_resources', fake_build_resources) mock_req_group_map.return_value = None mock_get_allocations.return_value = mock.sentinel.allocation self.compute._build_and_run_instance(self.context, self.instance, self.image, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, filter_properties=self.filter_properties) mock_spawn.assert_called_once_with(self.context, self.instance, mock.ANY, self.injected_files, self.admin_pass, mock.ANY, network_info=None, block_device_info=None, accel_info=accel_info) @mock.patch.object(objects.Instance, 'save') @mock.patch.object(nova.compute.manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(nova.compute.manager.ComputeManager, '_default_block_device_names') @mock.patch.object(nova.compute.manager.ComputeManager, '_prep_block_device') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_networks_before_block_device_mapping') @mock.patch.object(virt_driver.ComputeDriver, 'clean_networks_preparation') @mock.patch.object(nova.compute.manager.ComputeManager, '_get_bound_arq_resources') def _test_delete_arqs_exception(self, mock_get_arqs, mock_clean_net, mock_prep_net, mock_prep_spawn, mock_prep_bd, mock_bdnames, mock_build_net, mock_save): # called _get_bound_arq_resources only if we has accel_uuids self.accel_uuids = {uuids.arq1} args = (self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids) mock_get_arqs.side_effect = ( exception.AcceleratorRequestOpFailed(op='get', msg='')) with self.compute._build_resources(*args): raise test.TestingException() @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'delete_arqs_for_instance') def test_delete_arqs_if_build_res_exception(self, mock_del_arqs): # Cyborg is called to delete ARQs if exception is thrown inside # the context of # _build_resources(). self.instance.flavor.extra_specs = {'accel:device_profile': 'mydp'} self.assertRaisesRegex(exception.BuildAbortException, 'Failure getting accelerator requests', self._test_delete_arqs_exception) mock_del_arqs.assert_called_once_with(self.instance.uuid) @mock.patch('nova.accelerator.cyborg._CyborgClient.' 'delete_arqs_for_instance') def test_delete_arqs_if_build_res_exception_no_dp(self, mock_del_arqs): # Cyborg is not called to delete ARQs, even if an exception is # thrown inside the context of _build_resources(), if there is no # device profile name in the extra specs. self.instance.flavor.extra_specs = {} self.assertRaises(exception.BuildAbortException, self._test_delete_arqs_exception) # force delete arqs by uuid while get instance arqs failed mock_del_arqs.assert_not_called() def test_build_and_run_instance_called_with_proper_args(self): self._test_build_and_run_instance() def test_build_and_run_instance_with_unlimited_max_concurrent_builds(self): self.flags(max_concurrent_builds=0) self.compute = manager.ComputeManager() self._test_build_and_run_instance() @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def _test_build_and_run_instance(self, mock_build, mock_save, mock_start, mock_finish): self._do_build_instance_update(mock_save) orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.ACTIVE, ret) return ret with mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids=self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save) mock_build.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) # This test when sending an icehouse compatible rpc call to juno compute # node, NetworkRequest object can load from three items tuple. @mock.patch.object(compute_utils, 'EventReporter') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager._build_and_run_instance') def test_build_and_run_instance_with_icehouse_requested_network( self, mock_build_and_run, mock_save, mock_event): mock_save.return_value = self.instance self.compute.build_and_run_instance(self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids = self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=[objects.NetworkRequest( network_id='fake_network_id', address='10.0.0.1', port_id=uuids.port_instance)], security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) requested_network = mock_build_and_run.call_args[0][5][0] self.assertEqual('fake_network_id', requested_network.network_id) self.assertEqual('10.0.0.1', str(requested_network.address)) self.assertEqual(uuids.port_instance, requested_network.port_id) @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(manager.ComputeManager, '_cleanup_volumes') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(manager.ComputeManager, '_set_instance_obj_error_state') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def test_build_abort_exception(self, mock_build_run, mock_build, mock_set, mock_nil, mock_add, mock_clean_vol, mock_clean_net, mock_save, mock_start, mock_finish): self._do_build_instance_update(mock_save) mock_build_run.side_effect = exception.BuildAbortException(reason='', instance_uuid=self.instance.uuid) orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.FAILED, ret) return ret with mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids = self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save) mock_build_run.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_clean_net.assert_called_once_with(self.context, self.instance, self.requested_networks) mock_clean_vol.assert_called_once_with(self.context, self.instance, self.block_device_mapping, raise_exc=False) mock_add.assert_called_once_with(self.context, self.instance, mock.ANY, mock.ANY) mock_nil.assert_called_once_with(self.instance) mock_set.assert_called_once_with(self.instance, clean_task_state=True) @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(manager.ComputeManager, '_set_instance_obj_error_state') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def test_rescheduled_exception(self, mock_build_run, mock_build, mock_set, mock_nil, mock_save, mock_start, mock_finish): self._do_build_instance_update(mock_save, reschedule_update=True) mock_build_run.side_effect = exception.RescheduledException(reason='', instance_uuid=self.instance.uuid) orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.RESCHEDULED, ret) return ret with test.nested( mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ), mock.patch.object( self.compute.network_api, 'get_instance_nw_info', ), ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids=self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save, reschedule_update=True) mock_build_run.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_nil.assert_called_once_with(self.instance) mock_build.assert_called_once_with(self.context, [self.instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch.object(manager.ComputeManager, '_shutdown_instance') @mock.patch.object(manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(fake_driver.FakeDriver, 'spawn') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') def test_rescheduled_exception_with_non_ascii_exception(self, mock_notify, mock_save, mock_spawn, mock_build, mock_shutdown, mock_claim): exc = exception.NovaException(u's\xe9quence') mock_build.return_value = self.network_info mock_spawn.side_effect = exc self.assertRaises(exception.RescheduledException, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, self.accel_uuids) mock_save.assert_has_calls([ mock.call(), mock.call(expected_task_state='block_device_mapping'), ]) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, 'create.start', extra_usage_info={'image_name': self.image.get('name')}), mock.call(self.context, self.instance, 'create.error', fault=exc)]) mock_build.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_shutdown.assert_called_once_with(self.context, self.instance, self.block_device_mapping, self.requested_networks, try_deallocate_networks=False) mock_spawn.assert_called_once_with(self.context, self.instance, test.MatchType(objects.ImageMeta), self.injected_files, self.admin_pass, self.allocations, network_info=self.network_info, block_device_info=self.block_device_info, accel_info=[]) @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') def test_rescheduled_exception_with_network_allocated(self, mock_event_finish, mock_event_start, mock_ins_save, mock_build_ins, mock_build_and_run): instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.ACTIVE, system_metadata={'network_allocated': 'True'}, expected_attrs=['metadata', 'system_metadata', 'info_cache']) mock_ins_save.return_value = instance mock_build_and_run.side_effect = exception.RescheduledException( reason='', instance_uuid=self.instance.uuid) with mock.patch.object( self.compute.network_api, 'get_instance_nw_info', ): self.compute._do_build_and_run_instance( self.context, instance, self.image, request_spec={}, filter_properties=self.filter_properties, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list, accel_uuids=self.accel_uuids) mock_build_and_run.assert_called_once_with(self.context, instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_build_ins.assert_called_once_with(self.context, [instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') def test_rescheduled_exception_with_network_allocated_with_neutron(self, mock_event_finish, mock_event_start, mock_ins_save, mock_build_ins, mock_cleanup_network, mock_build_and_run): """Tests that we always cleanup allocated networks for the instance when using neutron and before we reschedule off the failed host. """ instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.ACTIVE, system_metadata={'network_allocated': 'True'}, expected_attrs=['metadata', 'system_metadata', 'info_cache']) mock_ins_save.return_value = instance mock_build_and_run.side_effect = exception.RescheduledException( reason='', instance_uuid=self.instance.uuid) self.compute._do_build_and_run_instance(self.context, instance, self.image, request_spec={}, filter_properties=self.filter_properties, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list, accel_uuids=self.accel_uuids) mock_build_and_run.assert_called_once_with(self.context, instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_cleanup_network.assert_called_once_with( self.context, instance, self.requested_networks) mock_build_ins.assert_called_once_with(self.context, [instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') def test_rescheduled_exception_with_sriov_network_allocated(self, mock_event_finish, mock_event_start, mock_ins_save, mock_cleanup_network, mock_build_ins, mock_build_and_run): vif1 = fake_network_cache_model.new_vif() vif1['id'] = '1' vif1['vnic_type'] = network_model.VNIC_TYPE_NORMAL vif2 = fake_network_cache_model.new_vif() vif2['id'] = '2' vif1['vnic_type'] = network_model.VNIC_TYPE_DIRECT nw_info = network_model.NetworkInfo([vif1, vif2]) instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.ACTIVE, system_metadata={'network_allocated': 'True'}, expected_attrs=['metadata', 'system_metadata', 'info_cache']) info_cache = objects.InstanceInfoCache(network_info=nw_info, instance_uuid=instance.uuid) instance.info_cache = info_cache mock_ins_save.return_value = instance mock_build_and_run.side_effect = exception.RescheduledException( reason='', instance_uuid=self.instance.uuid) self.compute._do_build_and_run_instance(self.context, instance, self.image, request_spec={}, filter_properties=self.filter_properties, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list, accel_uuids=self.accel_uuids) mock_build_and_run.assert_called_once_with(self.context, instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_cleanup_network.assert_called_once_with( self.context, instance, self.requested_networks) mock_build_ins.assert_called_once_with(self.context, [instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(manager.ComputeManager, '_set_instance_obj_error_state') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def test_rescheduled_exception_without_retry(self, mock_build_run, mock_add, mock_set, mock_clean_net, mock_nil, mock_save, mock_start, mock_finish): self._do_build_instance_update(mock_save) mock_build_run.side_effect = exception.RescheduledException(reason='', instance_uuid=self.instance.uuid) orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.FAILED, ret) return ret with mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties={}, accel_uuids=[], injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save) mock_build_run.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, {}, {}, self.accel_uuids) mock_clean_net.assert_called_once_with(self.context, self.instance, self.requested_networks) mock_add.assert_called_once_with(self.context, self.instance, mock.ANY, mock.ANY, fault_message=mock.ANY) mock_nil.assert_called_once_with(self.instance) mock_set.assert_called_once_with(self.instance, clean_task_state=True) @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def test_rescheduled_exception_do_not_deallocate_network(self, mock_build_run, mock_build, mock_nil, mock_clean_net, mock_save, mock_start, mock_finish): self._do_build_instance_update(mock_save, reschedule_update=True) mock_build_run.side_effect = exception.RescheduledException(reason='', instance_uuid=self.instance.uuid) orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.RESCHEDULED, ret) return ret with mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids=self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save, reschedule_update=True) mock_build_run.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_nil.assert_called_once_with(self.instance) mock_build.assert_called_once_with(self.context, [self.instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def test_rescheduled_exception_deallocate_network(self, mock_build_run, mock_build, mock_nil, mock_clean, mock_save, mock_start, mock_finish): self._do_build_instance_update(mock_save, reschedule_update=True) mock_build_run.side_effect = exception.RescheduledException(reason='', instance_uuid=self.instance.uuid) orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.RESCHEDULED, ret) return ret with mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids=self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save, reschedule_update=True) mock_build_run.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_clean.assert_called_once_with(self.context, self.instance, self.requested_networks) mock_nil.assert_called_once_with(self.instance) mock_build.assert_called_once_with(self.context, [self.instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_cleanup_allocated_networks') @mock.patch.object(manager.ComputeManager, '_cleanup_volumes') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(manager.ComputeManager, '_set_instance_obj_error_state') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(manager.ComputeManager, '_build_and_run_instance') def _test_build_and_run_exceptions(self, exc, mock_build_run, mock_build, mock_set, mock_nil, mock_add, mock_clean_vol, mock_clean_net, mock_save, mock_start, mock_finish, set_error=False, cleanup_volumes=False, nil_out_host_and_node=False): self._do_build_instance_update(mock_save) mock_build_run.side_effect = exc orig_do_build_and_run = self.compute._do_build_and_run_instance def _wrapped_do_build_and_run_instance(*args, **kwargs): ret = orig_do_build_and_run(*args, **kwargs) self.assertEqual(build_results.FAILED, ret) return ret with mock.patch.object( self.compute, '_do_build_and_run_instance', side_effect=_wrapped_do_build_and_run_instance, ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids=self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save) if cleanup_volumes: mock_clean_vol.assert_called_once_with(self.context, self.instance, self.block_device_mapping, raise_exc=False) if nil_out_host_and_node: mock_nil.assert_called_once_with(self.instance) if set_error: mock_add.assert_called_once_with(self.context, self.instance, mock.ANY, mock.ANY) mock_set.assert_called_once_with( self.instance, clean_task_state=True) mock_build_run.assert_called_once_with(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, {}, self.accel_uuids) mock_clean_net.assert_called_once_with(self.context, self.instance, self.requested_networks) def test_build_and_run_notfound_exception(self): self._test_build_and_run_exceptions(exception.InstanceNotFound( instance_id='')) def test_build_and_run_unexpecteddeleting_exception(self): self._test_build_and_run_exceptions( exception.UnexpectedDeletingTaskStateError( instance_uuid=uuids.instance, expected={}, actual={})) @mock.patch('nova.compute.manager.LOG.error') def test_build_and_run_buildabort_exception(self, mock_le): self._test_build_and_run_exceptions( exception.BuildAbortException(instance_uuid='', reason=''), set_error=True, cleanup_volumes=True, nil_out_host_and_node=True) mock_le.assert_called_once_with('Build of instance aborted: ', instance=mock.ANY) def test_build_and_run_unhandled_exception(self): self._test_build_and_run_exceptions(test.TestingException(), set_error=True, cleanup_volumes=True, nil_out_host_and_node=True) @mock.patch.object(manager.ComputeManager, '_do_build_and_run_instance') @mock.patch('nova.compute.stats.Stats.build_failed') def test_build_failures_reported(self, mock_failed, mock_dbari): mock_dbari.return_value = build_results.FAILED instance = objects.Instance(uuid=uuids.instance) for i in range(0, 10): self.compute.build_and_run_instance(self.context, instance, None, None, None, []) self.assertEqual(10, mock_failed.call_count) @mock.patch.object(manager.ComputeManager, '_do_build_and_run_instance') @mock.patch('nova.compute.stats.Stats.build_failed') def test_build_failures_not_reported(self, mock_failed, mock_dbari): self.flags(consecutive_build_service_disable_threshold=0, group='compute') mock_dbari.return_value = build_results.FAILED instance = objects.Instance(uuid=uuids.instance) for i in range(0, 10): self.compute.build_and_run_instance(self.context, instance, None, None, None, []) mock_failed.assert_not_called() @mock.patch.object(manager.ComputeManager, '_do_build_and_run_instance') @mock.patch.object(manager.ComputeManager, '_build_failed') @mock.patch.object(manager.ComputeManager, '_build_succeeded') def test_transient_build_failures_no_report(self, mock_succeeded, mock_failed, mock_dbari): results = [build_results.FAILED, build_results.ACTIVE, build_results.RESCHEDULED] def _fake_build(*a, **k): if results: return results.pop(0) else: return build_results.ACTIVE mock_dbari.side_effect = _fake_build instance = objects.Instance(uuid=uuids.instance) for i in range(0, 10): self.compute.build_and_run_instance(self.context, instance, None, None, None, []) self.assertEqual(2, mock_failed.call_count) self.assertEqual(8, mock_succeeded.call_count) @mock.patch.object(manager.ComputeManager, '_do_build_and_run_instance') @mock.patch.object(manager.ComputeManager, '_build_failed') @mock.patch.object(manager.ComputeManager, '_build_succeeded') def test_build_reschedules_reported(self, mock_succeeded, mock_failed, mock_dbari): mock_dbari.return_value = build_results.RESCHEDULED instance = objects.Instance(uuid=uuids.instance) for i in range(0, 10): self.compute.build_and_run_instance(self.context, instance, None, None, None, []) self.assertEqual(10, mock_failed.call_count) mock_succeeded.assert_not_called() @mock.patch.object(manager.ComputeManager, '_do_build_and_run_instance') @mock.patch( 'nova.exception_wrapper._emit_versioned_exception_notification', new=mock.Mock()) @mock.patch( 'nova.exception_wrapper._emit_legacy_exception_notification', new=mock.Mock()) @mock.patch( 'nova.compute.utils.add_instance_fault_from_exc', new=mock.Mock()) @mock.patch.object(manager.ComputeManager, '_build_failed') @mock.patch.object(manager.ComputeManager, '_build_succeeded') def test_build_exceptions_reported( self, mock_succeeded, mock_failed, mock_dbari, ): mock_dbari.side_effect = test.TestingException() instance = objects.Instance(uuid=uuids.instance, task_state=None) for i in range(0, 10): self.assertRaises(test.TestingException, self.compute.build_and_run_instance, self.context, instance, None, None, None, []) self.assertEqual(10, mock_failed.call_count) mock_succeeded.assert_not_called() @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch.object(manager.ComputeManager, '_shutdown_instance') @mock.patch.object(manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(fake_driver.FakeDriver, 'spawn') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') def _test_instance_exception(self, exc, raised_exc, mock_notify, mock_save, mock_spawn, mock_build, mock_shutdown, mock_claim): """This method test the instance related InstanceNotFound and reschedule on exception errors. The test cases get from arguments. :param exc: Injected exception into the code under test :param exception: Raised exception in test case :param result: At end the excepted state """ mock_build.return_value = self.network_info mock_spawn.side_effect = exc self.assertRaises(raised_exc, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, self.accel_uuids) mock_save.assert_has_calls([ mock.call(), mock.call(expected_task_state='block_device_mapping')]) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, 'create.start', extra_usage_info={'image_name': self.image.get('name')}), mock.call(self.context, self.instance, 'create.error', fault=exc)]) mock_build.assert_called_once_with( self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_shutdown.assert_called_once_with( self.context, self.instance, self.block_device_mapping, self.requested_networks, try_deallocate_networks=False) mock_spawn.assert_called_once_with( self.context, self.instance, test.MatchType(objects.ImageMeta), self.injected_files, self.admin_pass, self.allocations, network_info=self.network_info, block_device_info=self.block_device_info, accel_info=[]) def test_instance_not_found(self): got_exc = exception.InstanceNotFound(instance_id=1) self._test_instance_exception(got_exc, exception.InstanceNotFound) def test_reschedule_on_exception(self): got_exc = test.TestingException() self._test_instance_exception(got_exc, exception.RescheduledException) def test_spawn_network_auto_alloc_failure(self): # This isn't really a driver.spawn failure, it's a failure from # network_api.allocate_for_instance, but testing it here is convenient. self._test_build_and_run_spawn_exceptions( exception.UnableToAutoAllocateNetwork( project_id=self.context.project_id)) def test_spawn_network_fixed_ip_not_valid_on_host_failure(self): self._test_build_and_run_spawn_exceptions( exception.FixedIpInvalidOnHost(port_id='fake-port-id')) def test_build_and_run_no_more_fixedips_exception(self): self._test_build_and_run_spawn_exceptions( exception.NoMoreFixedIps("error messge")) def test_build_and_run_flavor_disk_smaller_image_exception(self): self._test_build_and_run_spawn_exceptions( exception.FlavorDiskSmallerThanImage( flavor_size=0, image_size=1)) def test_build_and_run_flavor_disk_smaller_min_disk(self): self._test_build_and_run_spawn_exceptions( exception.FlavorDiskSmallerThanMinDisk( flavor_size=0, image_min_disk=1)) def test_build_and_run_flavor_memory_too_small_exception(self): self._test_build_and_run_spawn_exceptions( exception.FlavorMemoryTooSmall()) def test_build_and_run_image_not_active_exception(self): self._test_build_and_run_spawn_exceptions( exception.ImageNotActive(image_id=self.image.get('id'))) def test_build_and_run_image_unacceptable_exception(self): self._test_build_and_run_spawn_exceptions( exception.ImageUnacceptable(image_id=self.image.get('id'), reason="")) def test_build_and_run_invalid_disk_info_exception(self): self._test_build_and_run_spawn_exceptions( exception.InvalidDiskInfo(reason="")) def test_build_and_run_invalid_disk_format_exception(self): self._test_build_and_run_spawn_exceptions( exception.InvalidDiskFormat(disk_format="")) def test_build_and_run_signature_verification_error(self): self._test_build_and_run_spawn_exceptions( cursive_exception.SignatureVerificationError(reason="")) def test_build_and_run_certificate_validation_error(self): self._test_build_and_run_spawn_exceptions( exception.CertificateValidationFailed(cert_uuid='trusted-cert-id', reason="")) def test_build_and_run_volume_encryption_not_supported(self): self._test_build_and_run_spawn_exceptions( exception.VolumeEncryptionNotSupported(volume_type='something', volume_id='something')) def test_build_and_run_invalid_input(self): self._test_build_and_run_spawn_exceptions( exception.InvalidInput(reason="")) def test_build_and_run_requested_vram_too_high(self): self._test_build_and_run_spawn_exceptions( exception.RequestedVRamTooHigh(req_vram=200, max_vram=100)) def _test_build_and_run_spawn_exceptions(self, exc): with test.nested( mock.patch.object(self.compute.driver, 'spawn', side_effect=exc), mock.patch.object(self.instance, 'save', side_effect=[self.instance, self.instance, self.instance]), mock.patch.object(self.compute, '_build_networks_for_instance', return_value=self.network_info), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute, '_shutdown_instance'), mock.patch.object(self.compute, '_validate_instance_group_policy'), mock.patch.object(self.compute.rt, 'instance_claim'), mock.patch('nova.compute.utils.notify_about_instance_create') ) as (spawn, save, _build_networks_for_instance, _notify_about_instance_usage, _shutdown_instance, _validate_instance_group_policy, mock_claim, mock_notify): self.assertRaises(exception.BuildAbortException, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, self.accel_uuids) _validate_instance_group_policy.assert_called_once_with( self.context, self.instance, {}) _build_networks_for_instance.assert_has_calls( [mock.call(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs)]) _notify_about_instance_usage.assert_has_calls([ mock.call(self.context, self.instance, 'create.start', extra_usage_info={'image_name': self.image.get('name')}), mock.call(self.context, self.instance, 'create.error', fault=exc)]) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, 'fake-mini', phase='start', bdms=[]), mock.call(self.context, self.instance, 'fake-mini', phase='error', exception=exc, bdms=[])]) save.assert_has_calls([ mock.call(), mock.call( expected_task_state=task_states.BLOCK_DEVICE_MAPPING)]) spawn.assert_has_calls([mock.call(self.context, self.instance, test.MatchType(objects.ImageMeta), self.injected_files, self.admin_pass, self.allocations, network_info=self.network_info, block_device_info=self.block_device_info, accel_info=[])]) _shutdown_instance.assert_called_once_with(self.context, self.instance, self.block_device_mapping, self.requested_networks, try_deallocate_networks=False) @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') @mock.patch.object(objects.InstanceActionEvent, 'event_finish_with_failure') @mock.patch.object(objects.InstanceActionEvent, 'event_start') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_nil_out_instance_obj_host_and_node') @mock.patch.object(conductor_api.ComputeTaskAPI, 'build_instances') @mock.patch.object(resource_tracker.ResourceTracker, 'instance_claim') def test_reschedule_on_resources_unavailable(self, mock_claim, mock_build, mock_nil, mock_save, mock_start, mock_finish, mock_notify): reason = 'resource unavailable' exc = exception.ComputeResourcesUnavailable(reason=reason) mock_claim.side_effect = exc self._do_build_instance_update(mock_save, reschedule_update=True) with mock.patch.object( self.compute.network_api, 'get_instance_nw_info', ): self.compute.build_and_run_instance( self.context, self.instance, self.image, request_spec={}, filter_properties=self.filter_properties, accel_uuids=self.accel_uuids, injected_files=self.injected_files, admin_password=self.admin_pass, requested_networks=self.requested_networks, security_groups=self.security_groups, block_device_mapping=self.block_device_mapping, node=self.node, limits=self.limits, host_list=fake_host_list) self._instance_action_events(mock_start, mock_finish) self._assert_build_instance_update(mock_save, reschedule_update=True) mock_claim.assert_called_once_with(self.context, self.instance, self.node, self.allocations, self.limits) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, 'create.start', extra_usage_info= {'image_name': self.image.get('name')}), mock.call(self.context, self.instance, 'create.error', fault=exc)]) mock_build.assert_called_once_with(self.context, [self.instance], self.image, self.filter_properties, self.admin_pass, self.injected_files, self.requested_networks, self.security_groups, self.block_device_mapping, request_spec={}, host_lists=[fake_host_list]) mock_nil.assert_called_once_with(self.instance) @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch.object(manager.ComputeManager, '_build_resources') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') def test_build_resources_buildabort_reraise(self, mock_notify, mock_save, mock_build, mock_claim): exc = exception.BuildAbortException( instance_uuid=self.instance.uuid, reason='') mock_build.side_effect = exc self.assertRaises(exception.BuildAbortException, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, request_spec=[], accel_uuids=self.accel_uuids) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, 'create.start', extra_usage_info={'image_name': self.image.get('name')}), mock.call(self.context, self.instance, 'create.error', fault=exc)]) mock_build.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, test.MatchType(objects.ImageMeta), self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(manager.ComputeManager, '_prep_block_device') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_networks_before_block_device_mapping') @mock.patch.object(virt_driver.ComputeDriver, 'clean_networks_preparation') def test_build_resources_reraises_on_failed_bdm_prep( self, mock_clean, mock_prepnet, mock_prep, mock_build, mock_save, mock_prepspawn, mock_failedspawn): mock_save.return_value = self.instance mock_build.return_value = self.network_info mock_prep.side_effect = test.TestingException try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): pass except Exception as e: self.assertIsInstance(e, exception.BuildAbortException) mock_save.assert_called_once_with() mock_build.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_prep.assert_called_once_with(self.context, self.instance, self.block_device_mapping) mock_prepnet.assert_called_once_with(self.instance, self.network_info) mock_clean.assert_called_once_with(self.instance, self.network_info) mock_prepspawn.assert_called_once_with(self.instance) mock_failedspawn.assert_called_once_with(self.instance) @mock.patch('nova.virt.block_device.attach_block_devices', side_effect=exception.VolumeNotCreated('oops!')) def test_prep_block_device_maintain_original_error_message(self, mock_attach): """Tests that when attach_block_devices raises an Exception, the re-raised InvalidBDM has the original error message which contains the actual details of the failure. """ bdms = objects.BlockDeviceMappingList( objects=[fake_block_device.fake_bdm_object( self.context, dict(source_type='image', destination_type='volume', boot_index=0, image_id=uuids.image_id, device_name='/dev/vda', volume_size=1))]) ex = self.assertRaises(exception.InvalidBDM, self.compute._prep_block_device, self.context, self.instance, bdms) self.assertEqual('oops!', str(ex)) @mock.patch('nova.objects.InstanceGroup.get_by_hint') def test_validate_policy_honors_workaround_disabled(self, mock_get): instance = objects.Instance(uuid=uuids.instance) hints = {'group': 'foo'} mock_get.return_value = objects.InstanceGroup(policy=None, uuid=uuids.group) self.compute._validate_instance_group_policy(self.context, instance, hints) mock_get.assert_called_once_with(self.context, 'foo') @mock.patch('nova.objects.InstanceGroup.get_by_hint') def test_validate_policy_honors_workaround_enabled(self, mock_get): self.flags(disable_group_policy_check_upcall=True, group='workarounds') instance = objects.Instance(uuid=uuids.instance) hints = {'group': 'foo'} self.compute._validate_instance_group_policy(self.context, instance, hints) self.assertFalse(mock_get.called) @mock.patch('nova.objects.InstanceGroup.get_by_hint') def test_validate_instance_group_policy_handles_hint_list(self, mock_get): """Tests that _validate_instance_group_policy handles getting scheduler_hints from a RequestSpec which stores the hints as a key=list pair. """ instance = objects.Instance(uuid=uuids.instance) hints = {'group': [uuids.group_hint]} self.compute._validate_instance_group_policy(self.context, instance, hints) mock_get.assert_called_once_with(self.context, uuids.group_hint) @mock.patch('nova.objects.InstanceGroup.get_by_hint') def test_validate_instance_group_policy_deleted_group(self, mock_get): """Tests that _validate_instance_group_policy handles the case where the scheduler hint has a group but that group has been deleted. This tests is a reproducer for bug: #1890244 """ instance = objects.Instance(uuid=uuids.instance) hints = {'group': [uuids.group_hint]} mock_get.side_effect = exception.InstanceGroupNotFound( group_uuid=uuids.group_hint ) # This implicitly asserts that no exception is raised since # uncaught exceptions would be treated as a test failure. self.compute._validate_instance_group_policy( self.context, instance, hints ) # and this just assert that we did in fact invoke the method # that raises to ensure that if we refactor in the future this # this test will fail if the function we mock is no longer called. mock_get.assert_called_once_with(self.context, uuids.group_hint) @mock.patch('nova.objects.InstanceGroup.get_by_uuid') @mock.patch('nova.objects.InstanceList.get_uuids_by_host') @mock.patch('nova.objects.InstanceGroup.get_by_hint') @mock.patch.object(fake_driver.FakeDriver, 'get_available_nodes') @mock.patch('nova.objects.MigrationList.get_in_progress_by_host_and_node') def test_validate_instance_group_policy_with_rules( self, migration_list, nodes, mock_get_by_hint, mock_get_by_host, mock_get_by_uuid): # Create 2 instance in same host, inst2 created before inst1 instance = objects.Instance(uuid=uuids.inst1) hints = {'group': [uuids.group_hint]} existing_insts = [uuids.inst1, uuids.inst2] members_uuids = [uuids.inst1, uuids.inst2] mock_get_by_host.return_value = existing_insts # if group policy rules limit to 1, raise RescheduledException group = objects.InstanceGroup( policy='anti-affinity', rules={'max_server_per_host': '1'}, hosts=['host1'], members=members_uuids, uuid=uuids.group) mock_get_by_hint.return_value = group mock_get_by_uuid.return_value = group nodes.return_value = ['nodename'] migration_list.return_value = [objects.Migration( uuid=uuids.migration, instance_uuid=uuids.instance)] self.assertRaises(exception.RescheduledException, self.compute._validate_instance_group_policy, self.context, instance, hints) # if group policy rules limit change to 2, validate OK group2 = objects.InstanceGroup( policy='anti-affinity', rules={'max_server_per_host': 2}, hosts=['host1'], members=members_uuids, uuid=uuids.group) mock_get_by_hint.return_value = group2 mock_get_by_uuid.return_value = group2 self.compute._validate_instance_group_policy(self.context, instance, hints) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_networks_before_block_device_mapping') @mock.patch.object(virt_driver.ComputeDriver, 'clean_networks_preparation') def test_failed_bdm_prep_from_delete_raises_unexpected(self, mock_clean, mock_prepnet, mock_prepspawn, mock_failedspawn): with test.nested( mock.patch.object(self.compute, '_build_networks_for_instance', return_value=self.network_info), mock.patch.object(self.instance, 'save', side_effect=exception.UnexpectedDeletingTaskStateError( instance_uuid=uuids.instance, actual={'task_state': task_states.DELETING}, expected={'task_state': None})), ) as (_build_networks_for_instance, save): try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): pass except Exception as e: self.assertIsInstance(e, exception.UnexpectedDeletingTaskStateError) _build_networks_for_instance.assert_has_calls( [mock.call(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs)]) save.assert_has_calls([mock.call()]) mock_prepnet.assert_called_once_with(self.instance, self.network_info) mock_clean.assert_called_once_with(self.instance, self.network_info) mock_prepspawn.assert_called_once_with(self.instance) mock_failedspawn.assert_called_once_with(self.instance) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_networks_before_block_device_mapping') @mock.patch.object(virt_driver.ComputeDriver, 'clean_networks_preparation') def test_failed_prepare_for_spawn(self, mock_clean, mock_prepnet, mock_prepspawn, mock_failedspawn): mock_prepspawn.side_effect = exception.ComputeResourcesUnavailable( reason="asdf") with mock.patch.object(self.compute, '_build_networks_for_instance', return_value=self.network_info ) as _build_networks_for_instance: try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): pass except Exception as e: self.assertIsInstance(e, exception.ComputeResourcesUnavailable) _build_networks_for_instance.assert_has_calls( [mock.call(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs)]) mock_prepnet.assert_not_called() mock_clean.assert_called_once_with(self.instance, self.network_info) mock_prepspawn.assert_called_once_with(self.instance) mock_failedspawn.assert_called_once_with(self.instance) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(manager.ComputeManager, '_build_networks_for_instance') def test_build_resources_aborts_on_failed_network_alloc(self, mock_build, mock_prepspawn, mock_failedspawn): mock_build.side_effect = test.TestingException try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): pass except Exception as e: self.assertIsInstance(e, exception.BuildAbortException) mock_build.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) # This exception is raised prior to initial prep and cleanup # with the virt driver, and as such these should not record # any calls. mock_prepspawn.assert_not_called() mock_failedspawn.assert_not_called() @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') def test_failed_network_alloc_from_delete_raises_unexpected(self, mock_prepspawn, mock_failedspawn): with mock.patch.object(self.compute, '_build_networks_for_instance') as _build_networks: exc = exception.UnexpectedDeletingTaskStateError _build_networks.side_effect = exc( instance_uuid=uuids.instance, actual={'task_state': task_states.DELETING}, expected={'task_state': None}) try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): pass except Exception as e: self.assertIsInstance(e, exc) _build_networks.assert_has_calls( [mock.call(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs)]) mock_prepspawn.assert_not_called() mock_failedspawn.assert_not_called() @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(manager.ComputeManager, '_shutdown_instance') @mock.patch.object(objects.Instance, 'save') def test_build_resources_cleans_up_and_reraises_on_spawn_failure(self, mock_save, mock_shutdown, mock_build, mock_prepspawn, mock_failedspawn): mock_save.return_value = self.instance mock_build.return_value = self.network_info test_exception = test.TestingException() def fake_spawn(): raise test_exception try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): fake_spawn() except Exception as e: self.assertEqual(test_exception, e) mock_save.assert_called_once_with() mock_build.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_shutdown.assert_called_once_with(self.context, self.instance, self.block_device_mapping, self.requested_networks, try_deallocate_networks=False) mock_prepspawn.assert_called_once_with(self.instance) # Complete should have occured with _shutdown_instance # so calling after the fact is not necessary. mock_failedspawn.assert_not_called() @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch('nova.network.model.NetworkInfoAsyncWrapper.wait') @mock.patch( 'nova.compute.manager.ComputeManager._build_networks_for_instance') @mock.patch('nova.objects.Instance.save') def test_build_resources_instance_not_found_before_yield( self, mock_save, mock_build_network, mock_info_wait, mock_prepspawn, mock_failedspawn): mock_build_network.return_value = self.network_info expected_exc = exception.InstanceNotFound( instance_id=self.instance.uuid) mock_save.side_effect = expected_exc try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): raise test.TestingException() except Exception as e: self.assertEqual(expected_exc, e) mock_build_network.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_info_wait.assert_called_once_with(do_raise=False) mock_prepspawn.assert_called_once_with(self.instance) mock_failedspawn.assert_called_once_with(self.instance) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch('nova.network.model.NetworkInfoAsyncWrapper.wait') @mock.patch( 'nova.compute.manager.ComputeManager._build_networks_for_instance') @mock.patch('nova.objects.Instance.save') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_networks_before_block_device_mapping') @mock.patch.object(virt_driver.ComputeDriver, 'clean_networks_preparation') def test_build_resources_unexpected_task_error_before_yield( self, mock_clean, mock_prepnet, mock_save, mock_build_network, mock_info_wait, mock_prepspawn, mock_failedspawn): mock_build_network.return_value = self.network_info mock_save.side_effect = exception.UnexpectedTaskStateError( instance_uuid=uuids.instance, expected={}, actual={}) try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): raise test.TestingException() except exception.BuildAbortException: pass mock_build_network.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_info_wait.assert_called_once_with(do_raise=False) mock_prepnet.assert_called_once_with(self.instance, self.network_info) mock_clean.assert_called_once_with(self.instance, self.network_info) mock_prepspawn.assert_called_once_with(self.instance) mock_failedspawn.assert_called_once_with(self.instance) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch('nova.network.model.NetworkInfoAsyncWrapper.wait') @mock.patch( 'nova.compute.manager.ComputeManager._build_networks_for_instance') @mock.patch('nova.objects.Instance.save') def test_build_resources_exception_before_yield( self, mock_save, mock_build_network, mock_info_wait, mock_prepspawn, mock_failedspawn): mock_build_network.return_value = self.network_info mock_save.side_effect = Exception() try: with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): raise test.TestingException() except exception.BuildAbortException: pass mock_build_network.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_info_wait.assert_called_once_with(do_raise=False) mock_prepspawn.assert_called_once_with(self.instance) mock_failedspawn.assert_called_once_with(self.instance) @mock.patch.object(virt_driver.ComputeDriver, 'failed_spawn_cleanup') @mock.patch.object(virt_driver.ComputeDriver, 'prepare_for_spawn') @mock.patch.object(manager.ComputeManager, '_build_networks_for_instance') @mock.patch.object(manager.ComputeManager, '_shutdown_instance') @mock.patch.object(objects.Instance, 'save') @mock.patch('nova.compute.manager.LOG') def test_build_resources_aborts_on_cleanup_failure(self, mock_log, mock_save, mock_shutdown, mock_build, mock_prepspawn, mock_failedspawn): mock_save.return_value = self.instance mock_build.return_value = self.network_info mock_shutdown.side_effect = test.TestingException('Failed to shutdown') def fake_spawn(): raise test.TestingException('Failed to spawn') with self.assertRaisesRegex(exception.BuildAbortException, 'Failed to spawn'): with self.compute._build_resources(self.context, self.instance, self.requested_networks, self.security_groups, self.image, self.block_device_mapping, self.resource_provider_mapping, self.accel_uuids): fake_spawn() self.assertTrue(mock_log.warning.called) msg = mock_log.warning.call_args_list[0] self.assertIn('Failed to shutdown', msg[0][1]) mock_save.assert_called_once_with() mock_build.assert_called_once_with(self.context, self.instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_shutdown.assert_called_once_with(self.context, self.instance, self.block_device_mapping, self.requested_networks, try_deallocate_networks=False) mock_prepspawn.assert_called_once_with(self.instance) # Complete should have occured with _shutdown_instance # so calling after the fact is not necessary. mock_failedspawn.assert_not_called() @mock.patch.object(manager.ComputeManager, '_allocate_network') def test_build_networks_if_not_allocated(self, mock_allocate): instance = fake_instance.fake_instance_obj(self.context, system_metadata={}, expected_attrs=['system_metadata']) nw_info_obj = self.compute._build_networks_for_instance(self.context, instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_allocate.assert_called_once_with(self.context, instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) self.assertTrue(hasattr(nw_info_obj, 'wait'), "wait must be there") @mock.patch.object(manager.ComputeManager, '_allocate_network') def test_build_networks_if_allocated_false(self, mock_allocate): instance = fake_instance.fake_instance_obj(self.context, system_metadata=dict(network_allocated='False'), expected_attrs=['system_metadata']) nw_info_obj = self.compute._build_networks_for_instance(self.context, instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_allocate.assert_called_once_with(self.context, instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) self.assertTrue(hasattr(nw_info_obj, 'wait'), "wait must be there") @mock.patch.object(manager.ComputeManager, '_allocate_network') def test_return_networks_if_found(self, mock_allocate): instance = fake_instance.fake_instance_obj(self.context, system_metadata=dict(network_allocated='True'), expected_attrs=['system_metadata']) def fake_network_info(): return network_model.NetworkInfo([{'address': '123.123.123.123'}]) with test.nested( mock.patch.object( self.compute.network_api, 'setup_instance_network_on_host'), mock.patch.object( self.compute.network_api, 'get_instance_nw_info')) as ( mock_setup, mock_get ): # this should be a NetworkInfo, not NetworkInfoAsyncWrapper, to # match what get_instance_nw_info really returns mock_get.return_value = fake_network_info() self.compute._build_networks_for_instance(self.context, instance, self.requested_networks, self.security_groups, self.resource_provider_mapping, self.network_arqs) mock_get.assert_called_once_with(self.context, instance) mock_setup.assert_called_once_with(self.context, instance, instance.host) def test__cleanup_allocated_networks__instance_not_found(self): with test.nested( mock.patch.object(self.compute.network_api, 'get_instance_nw_info'), mock.patch.object(self.compute.driver, 'unplug_vifs'), mock.patch.object(self.compute, '_deallocate_network'), mock.patch.object(self.instance, 'save', side_effect=exception.InstanceNotFound(instance_id='')) ) as (mock_nwinfo, mock_unplug, mock_deallocate_network, mock_save): # Testing that this doesn't raise an exception self.compute._cleanup_allocated_networks( self.context, self.instance, self.requested_networks) mock_nwinfo.assert_called_once_with( self.context, self.instance) mock_unplug.assert_called_once_with( self.instance, mock_nwinfo.return_value) mock_deallocate_network.assert_called_once_with( self.context, self.instance, self.requested_networks) mock_save.assert_called_once_with() self.assertEqual( 'False', self.instance.system_metadata['network_allocated']) @mock.patch('nova.compute.manager.LOG') def test__cleanup_allocated_networks__error(self, mock_log): with test.nested( mock.patch.object( self.compute.network_api, 'get_instance_nw_info', side_effect=Exception('some neutron error') ), mock.patch.object(self.compute.driver, 'unplug_vifs'), ) as (mock_nwinfo, mock_unplug): self.compute._cleanup_allocated_networks( self.context, self.instance, self.requested_networks) mock_nwinfo.assert_called_once_with(self.context, self.instance) self.assertEqual(1, mock_log.warning.call_count) self.assertIn( 'Failed to update network info cache', mock_log.warning.call_args[0][0], ) mock_unplug.assert_not_called() def test_split_network_arqs(self): arqs = [ {'uuid': uuids.arq_uuid1}, {'uuid': uuids.arq_uuid2}, {'uuid': uuids.arq_uuid3}] request_tuples = [('123', '1.2.3.4', uuids.fakeid, None, arqs[0]['uuid'], 'smart_nic')] requests = objects.NetworkRequestList.from_tuples(request_tuples) spec_arqs, port_arqs = self.compute._split_network_arqs(arqs, requests) expect_spec_arqs = {arqs[1]['uuid']: arqs[1], arqs[2]['uuid']: arqs[2]} expect_port_arqs = {arqs[0]['uuid']: arqs[0]} self.assertEqual(expect_spec_arqs, spec_arqs) self.assertEqual(expect_port_arqs, port_arqs) def test_deallocate_network_none_requested(self): # Tests that we don't deallocate networks if 'none' were # specifically requested. req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='none')]) with mock.patch.object(self.compute.network_api, 'deallocate_for_instance') as deallocate: self.compute._deallocate_network( self.context, mock.sentinel.instance, req_networks) self.assertFalse(deallocate.called) def test_deallocate_network_auto_requested_or_none_provided(self): # Tests that we deallocate networks if we were requested to # auto-allocate networks or requested_networks=None. req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='auto')]) for requested_networks in (req_networks, None): with mock.patch.object(self.compute.network_api, 'deallocate_for_instance') as deallocate: self.compute._deallocate_network( self.context, mock.sentinel.instance, requested_networks) deallocate.assert_called_once_with( self.context, mock.sentinel.instance, requested_networks=requested_networks) @mock.patch('nova.compute.manager.ComputeManager._deallocate_network') @mock.patch('nova.compute.manager.LOG.warning') def test_try_deallocate_network_retry_direct(self, warning_mock, deallocate_network_mock): """Tests that _try_deallocate_network will retry calling _deallocate_network on keystone ConnectFailure errors up to a limit. """ self.useFixture(service_fixture.SleepFixture()) deallocate_network_mock.side_effect = \ keystone_exception.connection.ConnectFailure req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='auto')]) instance = mock.MagicMock() ctxt = mock.MagicMock() self.assertRaises(keystone_exception.connection.ConnectFailure, self.compute._try_deallocate_network, ctxt, instance, req_networks) # should come to 3 retries and 1 default call , total 4 self.assertEqual(4, deallocate_network_mock.call_count) # And we should have logged a warning. warning_mock.assert_called() self.assertIn('Failed to deallocate network for instance; retrying.', warning_mock.call_args[0][0]) @mock.patch('nova.compute.manager.ComputeManager._deallocate_network') @mock.patch('nova.compute.manager.LOG.warning') def test_try_deallocate_network_no_retry(self, warning_mock, deallocate_network_mock): """Tests that _try_deallocate_network will not retry _deallocate_network for non-ConnectFailure errors. """ deallocate_network_mock.side_effect = test.TestingException('oops') req_networks = objects.NetworkRequestList( objects=[objects.NetworkRequest(network_id='auto')]) instance = mock.MagicMock() ctxt = mock.MagicMock() self.assertRaises(test.TestingException, self.compute._try_deallocate_network, ctxt, instance, req_networks) deallocate_network_mock.assert_called_once_with( ctxt, instance, req_networks) warning_mock.assert_not_called() @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch('nova.compute.utils.notify_about_instance_create') @mock.patch.object(manager.ComputeManager, '_instance_update') def test_launched_at_in_create_end_notification(self, mock_instance_update, mock_notify_instance_create, mock_claim): def fake_notify(*args, **kwargs): if args[2] == 'create.end': # Check that launched_at is set on the instance self.assertIsNotNone(args[1].launched_at) with test.nested( mock.patch.object(self.compute, '_update_scheduler_instance_info'), mock.patch.object(self.compute.driver, 'spawn'), mock.patch.object(self.compute, '_build_networks_for_instance', return_value=[]), mock.patch.object(self.instance, 'save'), mock.patch.object(self.compute, '_notify_about_instance_usage', side_effect=fake_notify) ) as (mock_upd, mock_spawn, mock_networks, mock_save, mock_notify): self.compute._build_and_run_instance(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, self.accel_uuids) expected_call = mock.call(self.context, self.instance, 'create.end', extra_usage_info={'message': u'Success'}, network_info=[]) create_end_call = mock_notify.call_args_list[ mock_notify.call_count - 1] self.assertEqual(expected_call, create_end_call) mock_notify_instance_create.assert_has_calls([ mock.call(self.context, self.instance, 'fake-mini', phase='start', bdms=[]), mock.call(self.context, self.instance, 'fake-mini', phase='end', bdms=[])]) def test_access_ip_set_when_instance_set_to_active(self): self.flags(default_access_ip_network_name='test1') instance = fake_instance.fake_db_instance() @mock.patch.object(self.compute.rt, 'instance_claim') @mock.patch.object(db, 'instance_update_and_get_original', return_value=({}, instance)) @mock.patch.object(self.compute.driver, 'spawn') @mock.patch.object(self.compute, '_build_networks_for_instance', return_value=fake_network.fake_get_instance_nw_info(self)) @mock.patch.object(db, 'instance_extra_update_by_uuid') @mock.patch.object(self.compute, '_notify_about_instance_usage') def _check_access_ip(mock_notify, mock_extra, mock_networks, mock_spawn, mock_db_update, mock_claim): self.compute._build_and_run_instance(self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, self.accel_uuids) updates = {'vm_state': u'active', 'access_ip_v6': netaddr.IPAddress('2001:db8:0:1:dcad:beff:feef:1'), 'access_ip_v4': netaddr.IPAddress('192.168.1.100'), 'power_state': 0, 'task_state': None, 'launched_at': mock.ANY, 'expected_task_state': 'spawning'} expected_call = mock.call(self.context, self.instance.uuid, updates, columns_to_join=['metadata', 'system_metadata', 'info_cache', 'tags']) last_update_call = mock_db_update.call_args_list[ mock_db_update.call_count - 1] self.assertEqual(expected_call, last_update_call) _check_access_ip() @mock.patch('nova.compute.resource_tracker.ResourceTracker.instance_claim') @mock.patch.object(manager.ComputeManager, '_instance_update') def test_create_error_on_instance_delete(self, mock_instance_update, mock_claim): def fake_notify(*args, **kwargs): if args[2] == 'create.error': # Check that launched_at is set on the instance self.assertIsNotNone(args[1].launched_at) exc = exception.InstanceNotFound(instance_id='') with test.nested( mock.patch.object(self.compute.driver, 'spawn'), mock.patch.object(self.compute, '_build_networks_for_instance', return_value=[]), mock.patch.object(self.instance, 'save', side_effect=[None, None, exc]), mock.patch.object(self.compute, '_notify_about_instance_usage', side_effect=fake_notify) ) as (mock_spawn, mock_networks, mock_save, mock_notify): self.assertRaises(exception.InstanceNotFound, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, self.accel_uuids) expected_call = mock.call(self.context, self.instance, 'create.error', fault=exc) create_error_call = mock_notify.call_args_list[ mock_notify.call_count - 1] self.assertEqual(expected_call, create_error_call) def test_build_with_resource_request_in_the_request_spec(self): request_spec = objects.RequestSpec( requested_resources=[ objects.RequestGroup( requester_id=uuids.port1, provider_uuids=[uuids.rp1])]) self.instance.pci_requests = objects.InstancePCIRequests(requests=[]) with test.nested( mock.patch.object(self.compute.driver, 'spawn'), mock.patch.object( self.compute, '_build_networks_for_instance', return_value=[]), mock.patch.object(self.instance, 'save'), mock.patch.object(self.compute.rt, 'instance_claim'), ) as (mock_spawn, mock_networks, mock_save, mock_claim): self.compute._build_and_run_instance( self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, request_spec, self.accel_uuids) mock_networks.assert_called_once_with( self.context, self.instance, self.requested_networks, self.security_groups, {uuids.port1: [uuids.rp1]}, self.network_arqs) def test_build_with_resource_request_sriov_port(self): request_spec = objects.RequestSpec( requested_resources=[ objects.RequestGroup( requester_id=uuids.port1, provider_uuids=[uuids.rp1])]) # NOTE(gibi): the first request will not match to any request group # this is the case when the request is not coming from a Neutron port # but from flavor or when the instance is old enough that the # requester_id field is not filled. # The second request will match with the request group in the request # spec and will trigger an update on that pci request. # The third request is coming from a Neutron port which doesn't have # resource request and therefore no matching request group exists in # the request spec. self.instance.pci_requests = objects.InstancePCIRequests(requests=[ objects.InstancePCIRequest( request_id=uuids.req0, ), objects.InstancePCIRequest( request_id=uuids.req1, requester_id=uuids.port1, spec=[{'vendor_id': '1377', 'product_id': '0047'}]), objects.InstancePCIRequest( request_id=uuids.req2, requester_id=uuids.port2, ), ]) with test.nested( mock.patch.object(self.compute.driver, 'spawn'), mock.patch.object(self.compute, '_build_networks_for_instance', return_value=[]), mock.patch.object(self.instance, 'save'), mock.patch('nova.scheduler.client.report.' 'SchedulerReportClient._get_resource_provider'), mock.patch.object(self.compute.rt, 'instance_claim'), ) as (mock_spawn, mock_networks, mock_save, mock_get_rp, mock_claim): mock_get_rp.return_value = { 'uuid': uuids.rp1, 'name': 'compute1:sriov-agent:ens3' } self.compute._build_and_run_instance( self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, request_spec, self.accel_uuids) mock_networks.assert_called_once_with( self.context, self.instance, self.requested_networks, self.security_groups, {uuids.port1: [uuids.rp1]}, {}) mock_get_rp.assert_called_once_with(self.context, uuids.rp1) # As the second pci request matched with the request group from the # request spec. So that pci request is extended with the # parent_ifname calculated from the corresponding RP name. self.assertEqual( [{'parent_ifname': 'ens3', 'vendor_id': '1377', 'product_id': '0047'}], self.instance.pci_requests.requests[1].spec) # the rest of the pci requests are unchanged self.assertNotIn('spec', self.instance.pci_requests.requests[0]) self.assertNotIn('spec', self.instance.pci_requests.requests[2]) def test_build_with_resource_request_sriov_rp_not_found(self): request_spec = objects.RequestSpec( requested_resources=[ objects.RequestGroup( requester_id=uuids.port1, provider_uuids=[uuids.rp1])]) self.instance.pci_requests = objects.InstancePCIRequests( requests=[ objects.InstancePCIRequest( requester_id=uuids.port1, request_id=uuids.req1 ) ] ) with mock.patch('nova.scheduler.client.report.SchedulerReportClient.' '_get_resource_provider') as (mock_get_rp): mock_get_rp.return_value = None self.assertRaises( exception.ResourceProviderNotFound, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, request_spec, self.accel_uuids) def test_build_with_resource_request_sriov_rp_wrongly_formatted_name(self): request_spec = objects.RequestSpec( requested_resources=[ objects.RequestGroup( requester_id=uuids.port1, provider_uuids=[uuids.rp1])]) self.instance.pci_requests = objects.InstancePCIRequests( requests=[ objects.InstancePCIRequest( requester_id=uuids.port1, request_id=uuids.req1 ) ] ) with mock.patch('nova.scheduler.client.report.SchedulerReportClient.' '_get_resource_provider') as (mock_get_rp): mock_get_rp.return_value = { 'uuid': uuids.rp1, 'name': 'my-awesome-rp' } self.assertRaises( exception.BuildAbortException, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, request_spec, self.accel_uuids) def test_build_with_resource_request_more_than_one_providers(self): request_spec = objects.RequestSpec( requested_resources=[ objects.RequestGroup( requester_id=uuids.port1, provider_uuids=[uuids.rp1, uuids.rp2])]) self.instance.pci_requests = objects.InstancePCIRequests( requests=[ objects.InstancePCIRequest( requester_id=uuids.port1, request_id=uuids.req1 ) ] ) self.assertRaises( exception.BuildAbortException, self.compute._build_and_run_instance, self.context, self.instance, self.image, self.injected_files, self.admin_pass, self.requested_networks, self.security_groups, self.block_device_mapping, self.node, self.limits, self.filter_properties, request_spec, self.accel_uuids) class ComputeManagerErrorsOutMigrationTestCase(test.NoDBTestCase): def setUp(self): super(ComputeManagerErrorsOutMigrationTestCase, self).setUp() self.context = context.RequestContext(fakes.FAKE_USER_ID, fakes.FAKE_PROJECT_ID) self.instance = fake_instance.fake_instance_obj(self.context) self.migration = objects.Migration() self.migration.instance_uuid = self.instance.uuid self.migration.status = 'migrating' self.migration.id = 0 @mock.patch.object(objects.Migration, 'save') def test_decorator(self, mock_save): # Tests that errors_out_migration decorator in compute manager sets # migration status to 'error' when an exception is raised from # decorated method @manager.errors_out_migration def fake_function(self, context, instance, migration): raise test.TestingException() self.assertRaises(test.TestingException, fake_function, self, self.context, self.instance, self.migration) self.assertEqual('error', self.migration.status) mock_save.assert_called_once_with() @mock.patch.object(objects.Migration, 'save') def test_contextmanager(self, mock_save): # Tests that errors_out_migration_ctxt context manager in compute # manager sets migration status to 'error' when an exception is raised # from decorated method def test_function(): with manager.errors_out_migration_ctxt(self.migration): raise test.TestingException() self.assertRaises(test.TestingException, test_function) self.assertEqual('error', self.migration.status) mock_save.assert_called_once_with() @ddt.ddt class ComputeManagerMigrationTestCase(test.NoDBTestCase, fake_resource_tracker.RTMockMixin): class TestResizeError(Exception): pass def setUp(self): super(ComputeManagerMigrationTestCase, self).setUp() self.notifier = self.useFixture(fixtures.NotificationFixture(self)) self.flags(compute_driver='fake.SameHostColdMigrateDriver') self.compute = manager.ComputeManager() self.context = context.RequestContext(fakes.FAKE_USER_ID, fakes.FAKE_PROJECT_ID) self.image = {} self.instance = fake_instance.fake_instance_obj(self.context, vm_state=vm_states.ACTIVE, expected_attrs=['metadata', 'system_metadata', 'info_cache']) self.migration = objects.Migration( context=self.context.elevated(), id=1, uuid=uuids.migration_uuid, instance_uuid=self.instance.uuid, new_instance_type_id=7, dest_compute='dest_compute', dest_node='dest_node', dest_host=None, source_compute='source_compute', source_node='source_node', status='migrating') self.migration.save = mock.MagicMock() self.useFixture(fixtures.SpawnIsSynchronousFixture()) self.useFixture(fixtures.EventReporterStub()) @contextlib.contextmanager def _mock_finish_resize(self): with test.nested( mock.patch.object(self.compute, '_finish_resize'), mock.patch.object(db, 'instance_fault_create'), mock.patch.object(self.compute, '_update_resource_tracker'), mock.patch.object(self.instance, 'save'), mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') ) as (_finish_resize, fault_create, instance_update, instance_save, get_bdm): fault_create.return_value = ( test_instance_fault.fake_faults['fake-uuid'][0]) yield _finish_resize def test_finish_resize_failure(self): self.migration.status = 'post-migrating' with self._mock_finish_resize() as _finish_resize: _finish_resize.side_effect = self.TestResizeError self.assertRaises( self.TestResizeError, self.compute.finish_resize, context=self.context, disk_info=[], image=self.image, instance=self.instance, migration=self.migration, request_spec=objects.RequestSpec() ) # Assert that we set the migration to an error state self.assertEqual("error", self.migration.status) @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') def test_finish_resize_notify_failure(self, notify): self.migration.status = 'post-migrating' with self._mock_finish_resize(): notify.side_effect = self.TestResizeError self.assertRaises( self.TestResizeError, self.compute.finish_resize, context=self.context, disk_info=[], image=self.image, instance=self.instance, migration=self.migration, request_spec=objects.RequestSpec() ) # Assert that we set the migration to an error state self.assertEqual("error", self.migration.status) @contextlib.contextmanager def _mock_resize_instance(self): with test.nested( mock.patch.object(self.compute.driver, 'migrate_disk_and_power_off'), mock.patch.object(db, 'instance_fault_create'), mock.patch.object(self.compute, '_update_resource_tracker'), mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.instance, 'save'), mock.patch.object(self.compute, '_notify_about_instance_usage'), mock.patch.object(self.compute, '_get_instance_block_device_info'), mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid'), mock.patch.object(objects.Flavor, 'get_by_id'), mock.patch.object(self.compute, '_terminate_volume_connections'), mock.patch.object(self.compute, 'compute_rpcapi'), ) as ( migrate_disk_and_power_off, fault_create, instance_update, network_api, save_inst, notify, vol_block_info, bdm, flavor, terminate_volume_connections, compute_rpcapi ): fault_create.return_value = ( test_instance_fault.fake_faults['fake-uuid'][0]) yield (migrate_disk_and_power_off, notify) def test_resize_instance_failure(self): with self._mock_resize_instance() as ( migrate_disk_and_power_off, notify): migrate_disk_and_power_off.side_effect = self.TestResizeError self.assertRaises( self.TestResizeError, self.compute.resize_instance, context=self.context, instance=self.instance, image=self.image, migration=self.migration, flavor='type', clean_shutdown=True, request_spec=objects.RequestSpec()) # Assert that we set the migration to an error state self.assertEqual("error", self.migration.status) def test_resize_instance_fail_rollback_stays_stopped(self): """Tests that when the driver's migrate_disk_and_power_off method raises InstanceFaultRollback that the instance vm_state is preserved rather than reset to ACTIVE which would be wrong if resizing a STOPPED server. """ with self._mock_resize_instance() as ( migrate_disk_and_power_off, notify): migrate_disk_and_power_off.side_effect = \ exception.InstanceFaultRollback( exception.ResizeError(reason='unable to resize disk down')) self.instance.vm_state = vm_states.STOPPED self.assertRaises( exception.ResizeError, self.compute.resize_instance, context=self.context, instance=self.instance, image=self.image, migration=self.migration, flavor='type', clean_shutdown=True, request_spec=objects.RequestSpec()) # Assert the instance vm_state was unchanged. self.assertEqual(vm_states.STOPPED, self.instance.vm_state) def test_resize_instance_notify_failure(self): # Raise an exception sending the end notification, which is after we # cast the migration to the destination host def fake_notify(context, instance, event, network_info=None): if event == 'resize.end': raise self.TestResizeError() with self._mock_resize_instance() as ( migrate_disk_and_power_off, notify_about_instance_action): notify_about_instance_action.side_effect = fake_notify self.assertRaises( self.TestResizeError, self.compute.resize_instance, context=self.context, instance=self.instance, image=self.image, migration=self.migration, flavor='type', clean_shutdown=True, request_spec=objects.RequestSpec()) # Assert that we did not set the migration to an error state self.assertEqual('post-migrating', self.migration.status) def _test_revert_resize_instance_destroy_disks(self, is_shared=False): # This test asserts that _is_instance_storage_shared() is called from # revert_resize() and the return value is passed to driver.destroy(). # Otherwise we could regress this. @mock.patch('nova.compute.rpcapi.ComputeAPI.finish_revert_resize') @mock.patch.object(self.instance, 'revert_migration_context') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info') @mock.patch.object(self.compute, '_is_instance_storage_shared') @mock.patch.object(self.compute, 'finish_revert_resize') @mock.patch.object(self.compute, '_instance_update') @mock.patch.object(self.compute.driver, 'destroy') @mock.patch.object(self.compute.network_api, 'setup_networks_on_host') @mock.patch.object(self.compute.network_api, 'migrate_instance_start') @mock.patch.object(compute_utils, 'notify_usage_exists') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def do_test(get_by_instance_uuid, notify_usage_exists, migrate_instance_start, setup_networks_on_host, destroy, _instance_update, finish_revert_resize, _is_instance_storage_shared, get_instance_nw_info, revert_migration_context, mock_finish_revert): self._mock_rt() # NOTE(danms): Before a revert, the instance is "on" # the destination host/node self.migration.uuid = uuids.migration self.migration.source_compute = 'src' self.migration.source_node = 'srcnode' self.migration.dest_compute = self.instance.host self.migration.dest_node = self.instance.node # Inform compute that instance uses non-shared or shared storage _is_instance_storage_shared.return_value = is_shared request_spec = objects.RequestSpec() self.compute.revert_resize(context=self.context, migration=self.migration, instance=self.instance, request_spec=request_spec) _is_instance_storage_shared.assert_called_once_with( self.context, self.instance, host=self.migration.source_compute) # If instance storage is shared, driver destroy method # should not destroy disks otherwise it should destroy disks. destroy.assert_called_once_with(self.context, self.instance, mock.ANY, mock.ANY, not is_shared) mock_finish_revert.assert_called_once_with( self.context, self.instance, self.migration, self.migration.source_compute, request_spec) do_test() def test_revert_resize_instance_destroy_disks_shared_storage(self): self._test_revert_resize_instance_destroy_disks(is_shared=True) def test_revert_resize_instance_destroy_disks_non_shared_storage(self): self._test_revert_resize_instance_destroy_disks(is_shared=False) def test_finish_revert_resize_network_calls_order(self): self.nw_info = None def _migrate_instance_finish( context, instance, migration, provider_mappings): # The migration.dest_compute is temporarily set to source_compute. self.assertEqual(migration.source_compute, migration.dest_compute) self.nw_info = 'nw_info' def _get_instance_nw_info(context, instance): return self.nw_info reportclient = self.compute.reportclient @mock.patch('nova.objects.ComputeNode.get_by_host_and_nodename') @mock.patch.object(self.compute.driver, 'finish_revert_migration') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info', side_effect=_get_instance_nw_info) @mock.patch.object(self.compute.network_api, 'migrate_instance_finish', side_effect=_migrate_instance_finish) @mock.patch.object(self.compute.network_api, 'setup_networks_on_host') @mock.patch.object(self.instance, 'save') @mock.patch.object(self.compute, '_set_instance_info') @mock.patch.object(db, 'instance_fault_create') @mock.patch.object(db, 'instance_extra_update_by_uuid') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(compute_utils, 'notify_about_instance_usage') def do_test(notify_about_instance_usage, get_by_instance_uuid, extra_update, fault_create, set_instance_info, instance_save, setup_networks_on_host, migrate_instance_finish, get_instance_nw_info, finish_revert_migration, mock_get_cn): # Mock the resource tracker, but keep the report client self._mock_rt().reportclient = reportclient fault_create.return_value = ( test_instance_fault.fake_faults['fake-uuid'][0]) self.instance.migration_context = objects.MigrationContext() self.migration.uuid = uuids.migration self.migration.source_compute = self.instance['host'] self.migration.source_node = self.instance['host'] request_spec = objects.RequestSpec() self.compute.finish_revert_resize(context=self.context, migration=self.migration, instance=self.instance, request_spec=request_spec) finish_revert_migration.assert_called_with(self.context, self.instance, 'nw_info', self.migration, mock.ANY, mock.ANY) # Make sure the migration.dest_compute is not still set to the # source_compute value. self.assertNotEqual(self.migration.dest_compute, self.migration.source_compute) do_test() def test_finish_revert_resize_migration_context(self): request_spec = objects.RequestSpec() @mock.patch('nova.compute.resource_tracker.ResourceTracker.' 'drop_move_claim_at_dest') @mock.patch('nova.compute.rpcapi.ComputeAPI.finish_revert_resize') @mock.patch.object(self.instance, 'revert_migration_context') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info') @mock.patch.object(self.compute, '_is_instance_storage_shared') @mock.patch.object(self.compute, '_instance_update') @mock.patch.object(self.compute.driver, 'destroy') @mock.patch.object(self.compute.network_api, 'setup_networks_on_host') @mock.patch.object(self.compute.network_api, 'migrate_instance_start') @mock.patch.object(compute_utils, 'notify_usage_exists') @mock.patch.object(db, 'instance_extra_update_by_uuid') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def do_revert_resize(mock_get_by_instance_uuid, mock_extra_update, mock_notify_usage_exists, mock_migrate_instance_start, mock_setup_networks_on_host, mock_destroy, mock_instance_update, mock_is_instance_storage_shared, mock_get_instance_nw_info, mock_revert_migration_context, mock_finish_revert, mock_drop_move_claim): self.compute.rt.tracked_migrations[self.instance['uuid']] = ( self.migration, None) self.instance.migration_context = objects.MigrationContext() self.migration.source_compute = self.instance['host'] self.migration.source_node = self.instance['node'] self.compute.revert_resize(context=self.context, migration=self.migration, instance=self.instance, request_spec=request_spec) mock_drop_move_claim.assert_called_once_with( self.context, self.instance, self.migration) # Three fake BDMs: # 1. volume BDM with an attachment_id which will be updated/completed # 2. volume BDM without an attachment_id so it's not updated # 3. non-volume BDM so it's not updated fake_bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(destination_type='volume', attachment_id=uuids.attachment_id, device_name='/dev/vdb'), objects.BlockDeviceMapping(destination_type='volume', attachment_id=None), objects.BlockDeviceMapping(destination_type='local') ]) @mock.patch('nova.objects.Service.get_minimum_version', return_value=22) @mock.patch('nova.objects.ComputeNode.get_by_host_and_nodename') @mock.patch.object(self.compute, "_notify_about_instance_usage") @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(self.compute, "_set_instance_info") @mock.patch.object(self.instance, 'save') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(db, 'instance_fault_create') @mock.patch.object(db, 'instance_extra_update_by_uuid') @mock.patch.object(self.compute.network_api, 'setup_networks_on_host') @mock.patch.object(self.compute.network_api, 'migrate_instance_finish') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid', return_value=fake_bdms) @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute.driver, 'get_volume_connector') @mock.patch.object(self.compute.volume_api, 'attachment_update') @mock.patch.object(self.compute.volume_api, 'attachment_complete') def do_finish_revert_resize(mock_attachment_complete, mock_attachment_update, mock_get_vol_connector, mock_get_blk, mock_get_by_instance_uuid, mock_get_instance_nw_info, mock_instance_finish, mock_setup_network, mock_extra_update, mock_fault_create, mock_fault_from_exc, mock_inst_save, mock_set, mock_notify_about_instance_action, mock_notify, mock_get_cn, mock_version): self.migration.uuid = uuids.migration self.compute.finish_revert_resize(context=self.context, instance=self.instance, migration=self.migration, request_spec=request_spec) self.assertIsNone(self.instance.migration_context) # We should only have one attachment_update/complete call for the # volume BDM that had an attachment. mock_attachment_update.assert_called_once_with( self.context, uuids.attachment_id, mock_get_vol_connector.return_value, '/dev/vdb') mock_attachment_complete.assert_called_once_with( self.context, uuids.attachment_id) do_revert_resize() do_finish_revert_resize() @mock.patch.object(objects.Instance, 'drop_migration_context') @mock.patch('nova.network.neutron.API.migrate_instance_finish') @mock.patch('nova.scheduler.utils.' 'fill_provider_mapping_based_on_allocation') @mock.patch('nova.compute.manager.ComputeManager._revert_allocation') @mock.patch.object(objects.Instance, 'save') @mock.patch('nova.compute.manager.ComputeManager.' '_set_instance_info') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def test_finish_revert_resize_recalc_group_rp_mapping( self, mock_get_bdms, mock_notify_action, mock_notify_usage, mock_set_instance_info, mock_instance_save, mock_revert_allocation, mock_fill_provider_mapping, mock_migrate_instance_finish, mock_drop_migration_context): mock_get_bdms.return_value = objects.BlockDeviceMappingList() request_spec = objects.RequestSpec() mock_revert_allocation.return_value = mock.sentinel.allocation with mock.patch.object( self.compute.network_api, 'get_instance_nw_info'): self.compute.finish_revert_resize( self.context, self.instance, self.migration, request_spec) mock_fill_provider_mapping.assert_called_once_with( self.context, self.compute.reportclient, request_spec, mock.sentinel.allocation) @mock.patch.object(objects.Instance, 'drop_migration_context') @mock.patch('nova.network.neutron.API.migrate_instance_finish') @mock.patch('nova.scheduler.utils.' 'fill_provider_mapping_based_on_allocation') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_consumer') @mock.patch('nova.compute.manager.ComputeManager._revert_allocation') @mock.patch.object(objects.Instance, 'save') @mock.patch('nova.compute.manager.ComputeManager.' '_set_instance_info') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') def test_finish_revert_resize_recalc_group_rp_mapping_missing_request_spec( self, mock_get_bdms, mock_notify_action, mock_notify_usage, mock_set_instance_info, mock_instance_save, mock_revert_allocation, mock_get_allocations, mock_fill_provider_mapping, mock_migrate_instance_finish, mock_drop_migration_context): mock_get_bdms.return_value = objects.BlockDeviceMappingList() mock_get_allocations.return_value = mock.sentinel.allocation with mock.patch.object( self.compute.network_api, 'get_instance_nw_info'): # This is the case when the compute is pinned to use older than # RPC version 5.2 self.compute.finish_revert_resize( self.context, self.instance, self.migration, request_spec=None) mock_get_allocations.assert_not_called() mock_fill_provider_mapping.assert_not_called() mock_migrate_instance_finish.assert_called_once_with( self.context, self.instance, self.migration, None) def test_confirm_resize_deletes_allocations_and_update_scheduler(self): @mock.patch.object(self.compute, '_delete_scheduler_instance_info') @mock.patch('nova.objects.Instance.get_by_uuid') @mock.patch('nova.objects.Migration.get_by_id') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute.driver, 'confirm_migration') @mock.patch.object(self.compute, '_delete_allocation_after_move') @mock.patch.object(self.instance, 'drop_migration_context') @mock.patch.object(self.instance, 'save') def do_confirm_resize(mock_save, mock_drop, mock_delete, mock_confirm, mock_nwapi, mock_notify, mock_mig_get, mock_inst_get, mock_delete_scheduler_info): self._mock_rt() self.instance.migration_context = objects.MigrationContext( new_pci_devices=None, old_pci_devices=None) self.migration.source_compute = self.instance['host'] self.migration.source_node = self.instance['node'] self.migration.status = 'confirming' mock_mig_get.return_value = self.migration mock_inst_get.return_value = self.instance self.compute.confirm_resize(self.context, self.instance, self.migration) mock_delete.assert_called_once_with(self.context, self.instance, self.migration) mock_save.assert_has_calls([ mock.call( expected_task_state=[ None, task_states.DELETING, task_states.SOFT_DELETING, ], ), mock.call(), ]) mock_delete_scheduler_info.assert_called_once_with( self.context, self.instance.uuid) do_confirm_resize() @mock.patch('nova.objects.MigrationContext.get_pci_mapping_for_migration') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.objects.Instance.get_by_uuid') @mock.patch('nova.compute.utils.notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_instance_action') @mock.patch('nova.objects.Instance.save') def test_confirm_resize_driver_confirm_migration_fails( self, instance_save, notify_action, notify_usage, instance_get_by_uuid, add_fault, get_mapping): """Tests the scenario that driver.confirm_migration raises some error to make sure the error is properly handled, like the instance and migration status is set to 'error'. """ self.migration.status = 'confirming' instance_get_by_uuid.return_value = self.instance self.instance.migration_context = objects.MigrationContext() error = exception.HypervisorUnavailable( host=self.migration.source_compute) with test.nested( mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.compute.driver, 'confirm_migration', side_effect=error), mock.patch.object(self.compute, '_delete_allocation_after_move'), mock.patch.object(self.compute, '_get_updated_nw_info_with_pci_mapping') ) as ( network_api, confirm_migration, delete_allocation, pci_mapping ): self.assertRaises(exception.HypervisorUnavailable, self.compute.confirm_resize, self.context, self.instance, self.migration) # Make sure the instance is in ERROR status. self.assertEqual(vm_states.ERROR, self.instance.vm_state) # Make sure the migration is in error status. self.assertEqual('error', self.migration.status) # Instance.save is called twice, once to clear the resize metadata # and once to set the instance to ERROR status. self.assertEqual(2, instance_save.call_count) # The migration.status should have been saved. self.migration.save.assert_called_once_with() # Allocations should always be cleaned up even if cleaning up the # source host fails. delete_allocation.assert_called_once_with( self.context, self.instance, self.migration) # Assert other mocks we care less about. notify_usage.assert_called_once() notify_action.assert_called_once() add_fault.assert_called_once() confirm_migration.assert_called_once() network_api.setup_networks_on_host.assert_called_once() instance_get_by_uuid.assert_called_once() def test_confirm_resize_calls_virt_driver_with_old_pci(self): @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute.driver, 'confirm_migration') @mock.patch.object(self.compute, '_delete_allocation_after_move') @mock.patch.object(self.instance, 'drop_migration_context') @mock.patch.object(self.instance, 'save') def do_confirm_resize( mock_save, mock_drop, mock_delete, mock_confirm, mock_nwapi, mock_notify ): # Mock virt driver confirm_resize() to save the provided # network_info, we will check it later. updated_nw_info = [] def driver_confirm_resize(*args, **kwargs): if 'network_info' in kwargs: nw_info = kwargs['network_info'] else: nw_info = args[3] updated_nw_info.extend(nw_info) mock_confirm.side_effect = driver_confirm_resize self._mock_rt() old_devs = objects.PciDeviceList( objects=[objects.PciDevice( compute_node_id=1, address='0000:04:00.2', request_id=uuids.pcidev1)]) new_devs = objects.PciDeviceList( objects=[objects.PciDevice( compute_node_id=2, address='0000:05:00.3', request_id=uuids.pcidev1)]) self.instance.migration_context = objects.MigrationContext( new_pci_devices=new_devs, old_pci_devices=old_devs) # Create VIF with new_devs[0] PCI address. nw_info = network_model.NetworkInfo([ network_model.VIF( id=uuids.port1, vnic_type=network_model.VNIC_TYPE_DIRECT, profile={'pci_slot': new_devs[0].address})]) mock_nwapi.get_instance_nw_info.return_value = nw_info self.migration.source_compute = self.instance['host'] self.migration.source_node = self.instance['node'] self.compute._confirm_resize(self.context, self.instance, self.migration) # Assert virt driver confirm_migration() was called # with the updated nw_info object. self.assertEqual(old_devs[0].address, updated_nw_info[0]['profile']['pci_slot']) do_confirm_resize() def test_delete_allocation_after_move_confirm_by_migration(self): with mock.patch.object(self.compute, 'reportclient') as mock_report: mock_report.delete_allocation_for_instance.return_value = True self.compute._delete_allocation_after_move(self.context, self.instance, self.migration) mock_report.delete_allocation_for_instance.assert_called_once_with( self.context, self.migration.uuid, consumer_type='migration', force=False ) def test_revert_allocation_allocation_exists(self): """New-style migration-based allocation revert.""" @mock.patch('nova.compute.manager.LOG.info') @mock.patch.object(self.compute, 'reportclient') def doit(mock_report, mock_info): a = { uuids.node: {'resources': {'DISK_GB': 1}}, uuids.child_rp: {'resources': {'CUSTOM_FOO': 1}} } mock_report.get_allocations_for_consumer.return_value = a self.migration.uuid = uuids.migration r = self.compute._revert_allocation(mock.sentinel.ctx, self.instance, self.migration) self.assertTrue(r) mock_report.move_allocations.assert_called_once_with( mock.sentinel.ctx, self.migration.uuid, self.instance.uuid) mock_info.assert_called_once_with( 'Swapping old allocation on %(rp_uuids)s held by migration ' '%(mig)s for instance', {'rp_uuids': a.keys(), 'mig': self.migration.uuid}, instance=self.instance) doit() def test_revert_allocation_allocation_not_exist(self): """Test that we don't delete allocs for migration if none found.""" @mock.patch('nova.compute.manager.LOG.error') @mock.patch.object(self.compute, 'reportclient') def doit(mock_report, mock_error): mock_report.get_allocations_for_consumer.return_value = {} self.migration.uuid = uuids.migration r = self.compute._revert_allocation(mock.sentinel.ctx, self.instance, self.migration) self.assertFalse(r) self.assertFalse(mock_report.move_allocations.called) mock_error.assert_called_once_with( 'Did not find resource allocations for migration ' '%s on source node %s. Unable to revert source node ' 'allocations back to the instance.', self.migration.uuid, self.migration.source_node, instance=self.instance) doit() def test_consoles_enabled(self): self.flags(enabled=False, group='vnc') self.flags(enabled=False, group='spice') self.flags(enabled=False, group='rdp') self.flags(enabled=False, group='serial_console') self.assertFalse(self.compute._consoles_enabled()) self.flags(enabled=True, group='vnc') self.assertTrue(self.compute._consoles_enabled()) self.flags(enabled=False, group='vnc') for console in ['spice', 'rdp', 'serial_console']: self.flags(enabled=True, group=console) self.assertTrue(self.compute._consoles_enabled()) self.flags(enabled=False, group=console) @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.objects.ConsoleAuthToken') def test_get_mks_console(self, mock_console_obj, mock_elevated): self.flags(enabled=True, group='mks') instance = objects.Instance(uuid=uuids.instance) with mock.patch.object(self.compute.driver, 'get_mks_console') as mock_get_console: console = self.compute.get_mks_console(self.context, 'webmks', instance) driver_console = mock_get_console.return_value mock_console_obj.assert_called_once_with( context=mock_elevated.return_value, console_type='webmks', host=driver_console.host, port=driver_console.port, internal_access_path=driver_console.internal_access_path, instance_uuid=instance.uuid, access_url_base=CONF.mks.mksproxy_base_url) mock_console_obj.return_value.authorize.assert_called_once_with( CONF.consoleauth.token_ttl) self.assertEqual(driver_console.get_connection_info.return_value, console) @mock.patch('nova.context.RequestContext.elevated') @mock.patch('nova.objects.ConsoleAuthToken') def test_get_serial_console(self, mock_console_obj, mock_elevated): self.flags(enabled=True, group='serial_console') instance = objects.Instance(uuid=uuids.instance) with mock.patch.object(self.compute.driver, 'get_serial_console') as mock_get_console: console = self.compute.get_serial_console(self.context, 'serial', instance) driver_console = mock_get_console.return_value mock_console_obj.assert_called_once_with( context=mock_elevated.return_value, console_type='serial', host=driver_console.host, port=driver_console.port, internal_access_path=driver_console.internal_access_path, instance_uuid=instance.uuid, access_url_base=CONF.serial_console.base_url) mock_console_obj.return_value.authorize.assert_called_once_with( CONF.consoleauth.token_ttl) self.assertEqual(driver_console.get_connection_info.return_value, console) @mock.patch('nova.utils.pass_context') @mock.patch('nova.compute.manager.ComputeManager.' '_do_live_migration') def _test_max_concurrent_live(self, mock_lm, mock_pass_context): # pass_context wraps the function, which doesn't work with a mock # So we simply mock it too def _mock_pass_context(runner, func, *args, **kwargs): return runner(func, *args, **kwargs) mock_pass_context.side_effect = _mock_pass_context @mock.patch('nova.objects.Migration.save') def _do_it(mock_mig_save): instance = objects.Instance(uuid=uuids.fake) migration = objects.Migration(uuid=uuids.migration) self.compute.live_migration(self.context, mock.sentinel.dest, instance, mock.sentinel.block_migration, migration, mock.sentinel.migrate_data) self.assertEqual('queued', migration.status) migration.save.assert_called_once_with() with mock.patch.object(self.compute, '_live_migration_executor') as mock_exc: for i in (1, 2, 3): _do_it() self.assertEqual(3, mock_exc.submit.call_count) def test_max_concurrent_live_limited(self): self.flags(max_concurrent_live_migrations=2) self._test_max_concurrent_live() def test_max_concurrent_live_unlimited(self): self.flags(max_concurrent_live_migrations=0) self._test_max_concurrent_live() @mock.patch('futurist.GreenThreadPoolExecutor') def test_max_concurrent_live_semaphore_limited(self, mock_executor): self.flags(max_concurrent_live_migrations=123) manager.ComputeManager() mock_executor.assert_called_once_with(max_workers=123) @mock.patch('futurist.GreenThreadPoolExecutor') def test_max_concurrent_live_semaphore_unlimited(self, mock_executor): self.flags(max_concurrent_live_migrations=0) manager.ComputeManager() mock_executor.assert_called_once_with() @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_pre_live_migration_cinder_v3_api(self): # This tests that pre_live_migration with a bdm with an # attachment_id, will create a new attachment and update # attachment_id's in the bdm. compute = manager.ComputeManager() instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) volume_id = uuids.volume vol_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) # attach_create should not be called on this image_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'image', 'destination_type': 'local', 'volume_id': volume_id, 'device_name': '/dev/vda', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) orig_attachment_id = uuids.attachment1 vol_bdm.attachment_id = orig_attachment_id new_attachment_id = uuids.attachment2 image_bdm.attachment_id = uuids.attachment3 migrate_data = migrate_data_obj.LiveMigrateData() migrate_data.old_vol_attachment_ids = {} @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(compute.volume_api, 'attachment_complete') @mock.patch.object(vol_bdm, 'save') @mock.patch.object(compute, '_notify_about_instance_usage') @mock.patch.object(compute, 'network_api') @mock.patch.object(compute.driver, 'pre_live_migration') @mock.patch.object(compute, '_get_instance_block_device_info') @mock.patch.object(compute_utils, 'is_volume_backed_instance') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(compute.volume_api, 'attachment_create') def _test(mock_attach, mock_get_bdms, mock_ivbi, mock_gibdi, mock_plm, mock_nwapi, mock_notify, mock_bdm_save, mock_attach_complete, mock_notify_about_inst): mock_driver_bdm = mock.Mock(spec=driver_bdm_volume) mock_gibdi.return_value = { 'block_device_mapping': [mock_driver_bdm]} mock_get_bdms.return_value = [vol_bdm, image_bdm] mock_attach.return_value = {'id': new_attachment_id} mock_plm.return_value = migrate_data connector = compute.driver.get_volume_connector(instance) r = compute.pre_live_migration(self.context, instance, {}, migrate_data) mock_notify_about_inst.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', action='live_migration_pre', phase='start', bdms=mock_get_bdms.return_value), mock.call(self.context, instance, 'fake-mini', action='live_migration_pre', phase='end', bdms=mock_get_bdms.return_value)]) self.assertIsInstance(r, migrate_data_obj.LiveMigrateData) self.assertIsInstance(mock_plm.call_args_list[0][0][5], migrate_data_obj.LiveMigrateData) mock_attach.assert_called_once_with( self.context, volume_id, instance.uuid, connector=connector, mountpoint=vol_bdm.device_name) self.assertEqual(vol_bdm.attachment_id, new_attachment_id) self.assertEqual(migrate_data.old_vol_attachment_ids[volume_id], orig_attachment_id) # Initially save of the attachment_id mock_bdm_save.assert_called_once_with() # Later save via the driver bdm for connection_info updates etc. mock_driver_bdm.save.assert_called_once_with() mock_attach_complete.assert_called_once_with(self.context, new_attachment_id) _test() @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_pre_live_migration_exception_cinder_v3_api(self): # The instance in this test has 2 attachments. The second attach_create # will throw an exception. This will test that the first attachment # is restored after the exception is thrown. compute = manager.ComputeManager() instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) volume1_id = uuids.volume1 vol1_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume1_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) vol1_orig_attachment_id = uuids.attachment1 vol1_bdm.attachment_id = vol1_orig_attachment_id volume2_id = uuids.volume2 vol2_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume2_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) vol2_orig_attachment_id = uuids.attachment2 vol2_bdm.attachment_id = vol2_orig_attachment_id migrate_data = migrate_data_obj.LiveMigrateData() migrate_data.old_vol_attachment_ids = {} @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(vol1_bdm, 'save') @mock.patch.object(compute, '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_instance_action') @mock.patch.object(compute, 'network_api') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(compute.volume_api, 'attachment_delete') @mock.patch.object(compute.volume_api, 'attachment_create') def _test(mock_attach_create, mock_attach_delete, mock_get_bdms, mock_nwapi, mock_ver_notify, mock_notify, mock_bdm_save, mock_exception): new_attachment_id = uuids.attachment3 mock_attach_create.side_effect = [{'id': new_attachment_id}, test.TestingException] mock_get_bdms.return_value = [vol1_bdm, vol2_bdm] self.assertRaises(test.TestingException, compute.pre_live_migration, self.context, instance, {}, migrate_data) self.assertEqual(vol1_orig_attachment_id, vol1_bdm.attachment_id) self.assertEqual(vol2_orig_attachment_id, vol2_bdm.attachment_id) self.assertEqual(mock_attach_create.call_count, 2) mock_attach_delete.assert_called_once_with(self.context, new_attachment_id) # Meta: ensure un-asserted mocks are still required for m in (mock_nwapi, mock_get_bdms, mock_ver_notify, mock_notify, mock_bdm_save, mock_exception): # NOTE(artom) This is different from assert_called() because # mock_calls contains the calls to a mock's method as well # (which is what we want for network_api.get_instance_nw_info # for example), whereas assert_called() only asserts # calls to the mock itself. self.assertGreater(len(m.mock_calls), 0) _test() @mock.patch('nova.objects.InstanceGroup.get_by_instance_uuid', mock.Mock( side_effect=exception.InstanceGroupNotFound(group_uuid=''))) def test_pre_live_migration_exceptions_delete_attachments(self): # The instance in this test has 2 attachments. The call to # driver.pre_live_migration will raise an exception. This will test # that the attachments are restored after the exception is thrown. compute = manager.ComputeManager() instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) vol1_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.vol1, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) vol1_bdm.attachment_id = uuids.vol1_attach_orig vol2_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.vol2, 'device_name': '/dev/vdc', 'instance_uuid': instance.uuid, 'connection_info': '{"test": "test"}'}) vol2_bdm.attachment_id = uuids.vol2_attach_orig migrate_data = migrate_data_obj.LiveMigrateData() migrate_data.old_vol_attachment_ids = {} @mock.patch.object(manager, 'compute_utils', autospec=True) @mock.patch.object(compute, 'network_api', autospec=True) @mock.patch.object(compute, 'volume_api', autospec=True) @mock.patch.object(objects.BlockDeviceMapping, 'save') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(compute.driver, 'pre_live_migration', autospec=True) def _test(mock_plm, mock_bdms_get, mock_bdm_save, mock_vol_api, mock_net_api, mock_compute_utils): mock_vol_api.attachment_create.side_effect = [ {'id': uuids.vol1_attach_new}, {'id': uuids.vol2_attach_new}] mock_bdms_get.return_value = [vol1_bdm, vol2_bdm] mock_plm.side_effect = test.TestingException self.assertRaises(test.TestingException, compute.pre_live_migration, self.context, instance, {}, migrate_data) self.assertEqual(2, mock_vol_api.attachment_create.call_count) # Assert BDMs have original attachments restored self.assertEqual(uuids.vol1_attach_orig, vol1_bdm.attachment_id) self.assertEqual(uuids.vol2_attach_orig, vol2_bdm.attachment_id) # Assert attachment cleanup self.assertEqual(2, mock_vol_api.attachment_delete.call_count) mock_vol_api.attachment_delete.assert_has_calls( [mock.call(self.context, uuids.vol1_attach_new), mock.call(self.context, uuids.vol2_attach_new)], any_order=True) # Meta: ensure un-asserted mocks are still required for m in (mock_net_api, mock_compute_utils): self.assertGreater(len(m.mock_calls), 0) _test() def test_get_neutron_events_for_live_migration_empty(self): """Tests the various ways that _get_neutron_events_for_live_migration will return an empty list. """ migration = mock.Mock() migration.is_same_host = lambda: False self.assertFalse(migration.is_same_host()) # 1. no timeout self.flags(vif_plugging_timeout=0) with mock.patch.object(self.instance, 'get_network_info') as nw_info: nw_info.return_value = network_model.NetworkInfo( [network_model.VIF(uuids.port1, details={ network_model.VIF_DETAILS_OVS_HYBRID_PLUG: True})]) self.assertTrue(nw_info.return_value[0].is_hybrid_plug_enabled()) self.assertEqual( [], self.compute._get_neutron_events_for_live_migration( self.instance)) # 2. no VIFs self.flags(vif_plugging_timeout=300) with mock.patch.object(self.instance, 'get_network_info') as nw_info: nw_info.return_value = network_model.NetworkInfo([]) self.assertEqual( [], self.compute._get_neutron_events_for_live_migration( self.instance)) # 3. no plug time events with mock.patch.object(self.instance, 'get_network_info') as nw_info: nw_info.return_value = network_model.NetworkInfo( [network_model.VIF( uuids.port1, details={ network_model.VIF_DETAILS_OVS_HYBRID_PLUG: False})]) self.assertFalse(nw_info.return_value[0].is_hybrid_plug_enabled()) self.assertEqual( [], self.compute._get_neutron_events_for_live_migration( self.instance)) @mock.patch('nova.compute.rpcapi.ComputeAPI.pre_live_migration') @mock.patch('nova.compute.manager.ComputeManager._post_live_migration') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') def test_live_migration_wait_vif_plugged( self, mock_get_bdms, mock_post_live_mig, mock_pre_live_mig): """Tests the happy path of waiting for network-vif-plugged events from neutron when pre_live_migration returns a migrate_data object with wait_for_vif_plugged=True. """ migrate_data = objects.LibvirtLiveMigrateData( wait_for_vif_plugged=True) mock_get_bdms.return_value = objects.BlockDeviceMappingList(objects=[]) mock_pre_live_mig.return_value = migrate_data details = {network_model.VIF_DETAILS_OVS_HYBRID_PLUG: True} self.instance.info_cache = objects.InstanceInfoCache( network_info=network_model.NetworkInfo([ network_model.VIF(uuids.port1, details=details), network_model.VIF(uuids.port2, details=details) ])) self.compute._waiting_live_migrations[self.instance.uuid] = ( self.migration, mock.MagicMock() ) with mock.patch.object(self.compute.virtapi, 'wait_for_instance_event') as wait_for_event: self.compute._do_live_migration( self.context, 'dest-host', self.instance, None, self.migration, migrate_data) self.assertEqual(2, len(wait_for_event.call_args[0][1])) self.assertEqual(CONF.vif_plugging_timeout, wait_for_event.call_args[1]['deadline']) mock_pre_live_mig.assert_called_once_with( self.context, self.instance, None, None, 'dest-host', migrate_data) @mock.patch('nova.compute.rpcapi.ComputeAPI.pre_live_migration') @mock.patch('nova.compute.manager.ComputeManager._post_live_migration') @mock.patch('nova.compute.manager.LOG.debug') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') def test_live_migration_wait_vif_plugged_old_dest_host( self, mock_get_bdms, mock_log_debug, mock_post_live_mig, mock_pre_live_mig): """Tests the scenario that the destination compute returns a migrate_data with no wait_for_vif_plugged set because the dest compute doesn't have that code yet. In this case, we default to legacy behavior of not waiting. """ migrate_data = objects.LibvirtLiveMigrateData() details = {network_model.VIF_DETAILS_OVS_HYBRID_PLUG: True} mock_get_bdms.return_value = objects.BlockDeviceMappingList(objects=[]) mock_pre_live_mig.return_value = migrate_data self.instance.info_cache = objects.InstanceInfoCache( network_info=network_model.NetworkInfo([ network_model.VIF(uuids.port1, details=details)])) self.compute._waiting_live_migrations[self.instance.uuid] = ( self.migration, mock.MagicMock() ) with mock.patch.object( self.compute.virtapi, 'wait_for_instance_event'): self.compute._do_live_migration( self.context, 'dest-host', self.instance, None, self.migration, migrate_data) # This isn't awesome, but we need a way to assert that we # short-circuit'ed the wait_for_instance_event context manager. self.assertEqual(2, mock_log_debug.call_count) self.assertIn('Not waiting for events after pre_live_migration', mock_log_debug.call_args_list[0][0][0]) # first call/arg @mock.patch('nova.compute.rpcapi.ComputeAPI.pre_live_migration') @mock.patch('nova.compute.manager.ComputeManager._rollback_live_migration') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') def test_live_migration_wait_vif_plugged_vif_plug_error( self, mock_get_bdms, mock_rollback_live_mig, mock_pre_live_mig): """Tests the scenario where wait_for_instance_event fails with VirtualInterfacePlugException. """ migrate_data = objects.LibvirtLiveMigrateData( wait_for_vif_plugged=True) source_bdms = objects.BlockDeviceMappingList(objects=[]) mock_get_bdms.return_value = source_bdms mock_pre_live_mig.return_value = migrate_data self.instance.info_cache = objects.InstanceInfoCache( network_info=network_model.NetworkInfo([ network_model.VIF(uuids.port1)])) self.compute._waiting_live_migrations[self.instance.uuid] = ( self.migration, mock.MagicMock() ) with mock.patch.object( self.compute.virtapi, 'wait_for_instance_event') as wait_for_event: wait_for_event.return_value.__enter__.side_effect = ( exception.VirtualInterfacePlugException()) self.assertRaises( exception.VirtualInterfacePlugException, self.compute._do_live_migration, self.context, 'dest-host', self.instance, None, self.migration, migrate_data) self.assertEqual('error', self.migration.status) mock_rollback_live_mig.assert_called_once_with( self.context, self.instance, 'dest-host', migrate_data=migrate_data, source_bdms=source_bdms, pre_live_migration=True) @mock.patch('nova.compute.rpcapi.ComputeAPI.pre_live_migration') @mock.patch('nova.compute.manager.ComputeManager._rollback_live_migration') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') def test_live_migration_wait_vif_plugged_timeout_error( self, mock_get_bdms, mock_rollback_live_mig, mock_pre_live_mig): """Tests the scenario where wait_for_instance_event raises an eventlet Timeout exception and we're configured such that vif plugging failures are fatal (which is the default). """ migrate_data = objects.LibvirtLiveMigrateData( wait_for_vif_plugged=True) source_bdms = objects.BlockDeviceMappingList(objects=[]) mock_get_bdms.return_value = source_bdms mock_pre_live_mig.return_value = migrate_data self.instance.info_cache = objects.InstanceInfoCache( network_info=network_model.NetworkInfo([ network_model.VIF(uuids.port1)])) self.compute._waiting_live_migrations[self.instance.uuid] = ( self.migration, mock.MagicMock() ) with mock.patch.object( self.compute.virtapi, 'wait_for_instance_event') as wait_for_event: wait_for_event.return_value.__enter__.side_effect = ( eventlet_timeout.Timeout()) ex = self.assertRaises( exception.MigrationError, self.compute._do_live_migration, self.context, 'dest-host', self.instance, None, self.migration, migrate_data) self.assertIn('Timed out waiting for events', str(ex)) self.assertEqual('error', self.migration.status) mock_rollback_live_mig.assert_called_once_with( self.context, self.instance, 'dest-host', migrate_data=migrate_data, source_bdms=source_bdms, pre_live_migration=True) @mock.patch('nova.compute.rpcapi.ComputeAPI.pre_live_migration') @mock.patch('nova.compute.manager.ComputeManager._rollback_live_migration') @mock.patch('nova.compute.manager.ComputeManager._post_live_migration') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') def test_live_migration_wait_vif_plugged_timeout_non_fatal( self, mock_get_bdms, mock_post_live_mig, mock_rollback_live_mig, mock_pre_live_mig): """Tests the scenario where wait_for_instance_event raises an eventlet Timeout exception and we're configured such that vif plugging failures are NOT fatal. """ self.flags(vif_plugging_is_fatal=False) mock_get_bdms.return_value = objects.BlockDeviceMappingList(objects=[]) migrate_data = objects.LibvirtLiveMigrateData( wait_for_vif_plugged=True) mock_pre_live_mig.return_value = migrate_data self.instance.info_cache = objects.InstanceInfoCache( network_info=network_model.NetworkInfo([ network_model.VIF(uuids.port1)])) self.compute._waiting_live_migrations[self.instance.uuid] = ( self.migration, mock.MagicMock() ) with mock.patch.object( self.compute.virtapi, 'wait_for_instance_event') as wait_for_event: wait_for_event.return_value.__enter__.side_effect = ( eventlet_timeout.Timeout()) self.compute._do_live_migration( self.context, 'dest-host', self.instance, None, self.migration, migrate_data) self.assertEqual('running', self.migration.status) mock_rollback_live_mig.assert_not_called() @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_live_migration_submit_failed(self, mock_notify, mock_exc): migration = objects.Migration(self.context, uuid=uuids.migration) migration.save = mock.MagicMock() with mock.patch.object( self.compute._live_migration_executor, 'submit') as mock_sub: mock_sub.side_effect = RuntimeError self.assertRaises(exception.LiveMigrationNotSubmitted, self.compute.live_migration, self.context, 'fake', self.instance, True, migration, {}) self.assertEqual('error', migration.status) @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') @mock.patch( 'nova.compute.manager.ComputeManager._notify_about_instance_usage') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch('nova.compute.manager.ComputeManager._rollback_live_migration') @mock.patch('nova.compute.rpcapi.ComputeAPI.pre_live_migration') def test_live_migration_aborted_before_running(self, mock_rpc, mock_rollback, mock_action_notify, mock_usage_notify, mock_get_bdms): source_bdms = objects.BlockDeviceMappingList(objects=[]) mock_get_bdms.return_value = source_bdms migrate_data = objects.LibvirtLiveMigrateData( wait_for_vif_plugged=True) details = {network_model.VIF_DETAILS_OVS_HYBRID_PLUG: True} self.instance.info_cache = objects.InstanceInfoCache( network_info=network_model.NetworkInfo([ network_model.VIF(uuids.port1, details=details), network_model.VIF(uuids.port2, details=details) ])) self.compute._waiting_live_migrations = {} fake_migration = objects.Migration( uuid=uuids.migration, instance_uuid=self.instance.uuid) fake_migration.save = mock.MagicMock() with mock.patch.object(self.compute.virtapi, 'wait_for_instance_event') as wait_for_event: self.compute._do_live_migration( self.context, 'dest-host', self.instance, 'block_migration', fake_migration, migrate_data) self.assertEqual(2, len(wait_for_event.call_args[0][1])) mock_rpc.assert_called_once_with( self.context, self.instance, 'block_migration', None, 'dest-host', migrate_data) # Ensure that rollback is specifically called with the migrate_data # that came back from the call to pre_live_migration on the dest host # rather than the one passed to _do_live_migration. mock_rollback.assert_called_once_with( self.context, self.instance, 'dest-host', mock_rpc.return_value, 'cancelled', source_bdms=source_bdms) mock_usage_notify.assert_called_once_with( self.context, self.instance, 'live.migration.abort.end') mock_action_notify.assert_called_once_with( self.context, self.instance, self.compute.host, action=fields.NotificationAction.LIVE_MIGRATION_ABORT, phase=fields.NotificationPhase.END) def test_live_migration_force_complete_succeeded(self): migration = objects.Migration() migration.status = 'running' migration.id = 0 @mock.patch('nova.compute.utils.notify_about_instance_action') @mock.patch('nova.image.glance.API.generate_image_url', return_value='fake-url') @mock.patch.object(objects.Migration, 'get_by_id', return_value=migration) @mock.patch.object(self.compute.driver, 'live_migration_force_complete') def _do_test(force_complete, get_by_id, gen_img_url, mock_notify): self.compute.live_migration_force_complete( self.context, self.instance) force_complete.assert_called_once_with(self.instance) self.assertEqual(2, len(self.notifier.notifications)) self.assertEqual( 'compute.instance.live.migration.force.complete.start', self.notifier.notifications[0].event_type) self.assertEqual( self.instance.uuid, self.notifier.notifications[0].payload['instance_id']) self.assertEqual( 'compute.instance.live.migration.force.complete.end', self.notifier.notifications[1].event_type) self.assertEqual( self.instance.uuid, self.notifier.notifications[1].payload['instance_id']) self.assertEqual(2, mock_notify.call_count) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, self.compute.host, action='live_migration_force_complete', phase='start'), mock.call(self.context, self.instance, self.compute.host, action='live_migration_force_complete', phase='end')]) _do_test() def test_post_live_migration_at_destination_success(self): @mock.patch.object(objects.Instance, 'drop_migration_context') @mock.patch.object(objects.Instance, 'apply_migration_context') @mock.patch.object(self.compute, 'rt') @mock.patch.object(self.instance, 'save') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value='test_network') @mock.patch.object(self.compute.network_api, 'setup_networks_on_host') @mock.patch.object(self.compute.network_api, 'migrate_instance_finish') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute, '_get_power_state', return_value=1) @mock.patch.object(self.compute, '_get_compute_info') @mock.patch.object(self.compute.driver, 'post_live_migration_at_destination') @mock.patch('nova.compute.utils.notify_about_instance_action') def _do_test(mock_notify, post_live_migration_at_destination, _get_compute_info, _get_power_state, _get_instance_block_device_info, _notify_about_instance_usage, migrate_instance_finish, setup_networks_on_host, get_instance_nw_info, save, rt_mock, mock_apply_mig_ctxt, mock_drop_mig_ctxt): cn = mock.Mock(spec_set=['hypervisor_hostname']) cn.hypervisor_hostname = 'test_host' _get_compute_info.return_value = cn cn_old = self.instance.host instance_old = self.instance self.compute.post_live_migration_at_destination( self.context, self.instance, False) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, self.instance.host, action='live_migration_post_dest', phase='start'), mock.call(self.context, self.instance, self.instance.host, action='live_migration_post_dest', phase='end')]) setup_networks_calls = [ mock.call(self.context, self.instance, self.compute.host), mock.call(self.context, self.instance, cn_old, teardown=True), mock.call(self.context, self.instance, self.compute.host) ] setup_networks_on_host.assert_has_calls(setup_networks_calls) notify_usage_calls = [ mock.call(self.context, instance_old, "live_migration.post.dest.start", network_info='test_network'), mock.call(self.context, self.instance, "live_migration.post.dest.end", network_info='test_network') ] _notify_about_instance_usage.assert_has_calls(notify_usage_calls) migrate_instance_finish.assert_called_once_with( self.context, self.instance, test.MatchType(objects.Migration), provider_mappings=None) mig = migrate_instance_finish.call_args[0][2] self.assertTrue(base_obj.obj_equal_prims( objects.Migration(source_compute=cn_old, dest_compute=self.compute.host, migration_type='live-migration'), mig)) _get_instance_block_device_info.assert_called_once_with( self.context, self.instance ) get_instance_nw_info.assert_called_once_with(self.context, self.instance) _get_power_state.assert_called_once_with(self.instance) _get_compute_info.assert_called_once_with(self.context, self.compute.host) rt_mock.allocate_pci_devices_for_instance.assert_called_once_with( self.context, self.instance) self.assertEqual(self.compute.host, self.instance.host) self.assertEqual('test_host', self.instance.node) self.assertEqual(1, self.instance.power_state) self.assertEqual(0, self.instance.progress) self.assertIsNone(self.instance.task_state) save.assert_called_once_with( expected_task_state=task_states.MIGRATING) mock_apply_mig_ctxt.assert_called_once_with() mock_drop_mig_ctxt.assert_called_once_with() _do_test() def test_post_live_migration_at_destination_compute_not_found(self): @mock.patch.object(objects.Instance, 'drop_migration_context') @mock.patch.object(objects.Instance, 'apply_migration_context') @mock.patch.object(self.compute, 'rt') @mock.patch.object(self.instance, 'save') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute, '_get_power_state', return_value=1) @mock.patch.object(self.compute, '_get_compute_info', side_effect=exception.ComputeHostNotFound( host=uuids.fake_host)) @mock.patch.object(self.compute.driver, 'post_live_migration_at_destination') @mock.patch('nova.compute.utils.notify_about_instance_action') def _do_test(mock_notify, post_live_migration_at_destination, _get_compute_info, _get_power_state, _get_instance_block_device_info, _notify_about_instance_usage, network_api, save, rt_mock, mock_apply_mig_ctxt, mock_drop_mig_ctxt): cn = mock.Mock(spec_set=['hypervisor_hostname']) cn.hypervisor_hostname = 'test_host' _get_compute_info.return_value = cn self.compute.post_live_migration_at_destination( self.context, self.instance, False) mock_notify.assert_has_calls([ mock.call(self.context, self.instance, self.instance.host, action='live_migration_post_dest', phase='start'), mock.call(self.context, self.instance, self.instance.host, action='live_migration_post_dest', phase='end')]) self.assertIsNone(self.instance.node) mock_apply_mig_ctxt.assert_called_with() mock_drop_mig_ctxt.assert_called_once_with() _do_test() def test_post_live_migration_at_destination_unexpected_exception(self): @mock.patch.object(objects.Instance, 'drop_migration_context') @mock.patch.object(objects.Instance, 'apply_migration_context') @mock.patch.object(self.compute, 'rt') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(self.instance, 'save') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute, '_get_power_state', return_value=1) @mock.patch.object(self.compute, '_get_compute_info') @mock.patch.object(self.compute.driver, 'post_live_migration_at_destination', side_effect=exception.NovaException) def _do_test(post_live_migration_at_destination, _get_compute_info, _get_power_state, _get_instance_block_device_info, _notify_about_instance_usage, network_api, save, add_instance_fault_from_exc, rt_mock, mock_apply_mig_ctxt, mock_drop_mig_ctxt): cn = mock.Mock(spec_set=['hypervisor_hostname']) cn.hypervisor_hostname = 'test_host' _get_compute_info.return_value = cn self.assertRaises(exception.NovaException, self.compute.post_live_migration_at_destination, self.context, self.instance, False) self.assertEqual(vm_states.ERROR, self.instance.vm_state) mock_apply_mig_ctxt.assert_called_with() mock_drop_mig_ctxt.assert_called_once_with() _do_test() @mock.patch('nova.compute.manager.LOG.error') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_post_live_migration_at_destination_port_binding_delete_fails( self, mock_notify, mock_log_error): """Tests that neutron fails to delete the source host port bindings but we handle the error and just log it. """ @mock.patch.object(self.instance, 'drop_migration_context') @mock.patch.object(objects.Instance, 'apply_migration_context') @mock.patch.object(self.compute, 'rt') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute, '_get_power_state', return_value=power_state.RUNNING) @mock.patch.object(self.compute, '_get_compute_info', return_value=objects.ComputeNode( hypervisor_hostname='fake-dest-host')) @mock.patch.object(self.instance, 'save') def _do_test(instance_save, get_compute_node, get_power_state, get_bdms, network_api, legacy_notify, rt_mock, mock_apply_mig_ctxt, mock_drop_mig_ctxt): # setup_networks_on_host is called three times: # 1. set the migrating_to port binding profile value (no-op) # 2. delete the source host port bindings - we make this raise # 3. once more to update dhcp for nova-network (no-op for neutron) network_api.setup_networks_on_host.side_effect = [ None, exception.PortBindingDeletionFailed( port_id=uuids.port_id, host='fake-source-host'), None] self.compute.post_live_migration_at_destination( self.context, self.instance, block_migration=False) self.assertEqual(1, mock_log_error.call_count) self.assertIn('Network cleanup failed for source host', mock_log_error.call_args[0][0]) mock_apply_mig_ctxt.assert_called_once_with() mock_drop_mig_ctxt.assert_called_once_with() _do_test() @mock.patch('nova.objects.ConsoleAuthToken.' 'clean_console_auths_for_instance') def _call_post_live_migration(self, mock_clean, *args, **kwargs): @mock.patch.object(self.compute, 'update_available_resource') @mock.patch.object(self.compute, 'compute_rpcapi') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, 'network_api') def _do_call(nwapi, notify, rpc, update): bdms = objects.BlockDeviceMappingList(objects=[]) result = self.compute._post_live_migration( self.context, self.instance, 'foo', *args, source_bdms=bdms, **kwargs) return result mock_rt = self._mock_rt() result = _do_call() mock_clean.assert_called_once_with(self.context, self.instance.uuid) mock_rt.free_pci_device_allocations_for_instance.\ assert_called_once_with(self.context, self.instance) return result def test_post_live_migration_new_allocations(self): # We have a migrate_data with a migration... migration = objects.Migration(uuid=uuids.migration) migration.save = mock.MagicMock() md = objects.LibvirtLiveMigrateData(migration=migration, is_shared_instance_path=False, is_shared_block_storage=False) with test.nested( mock.patch.object(self.compute, 'reportclient'), mock.patch.object(self.compute, '_delete_allocation_after_move'), ) as ( mock_report, mock_delete, ): # ...and that migration has allocations... mock_report.get_allocations_for_consumer.return_value = ( mock.sentinel.allocs) self._call_post_live_migration(migrate_data=md) # ...so we should have called the new style delete mock_delete.assert_called_once_with(self.context, self.instance, migration) def test_post_live_migration_update_host(self): @mock.patch.object(self.compute, '_get_compute_info') def _test_post_live_migration(_get_compute_info): dest_host = 'dest' cn = objects.ComputeNode(hypervisor_hostname=dest_host) _get_compute_info.return_value = cn instance = fake_instance.fake_instance_obj(self.context, node='src', uuid=uuids.instance) with mock.patch.object(self.compute, "_post_live_migration" ) as plm, mock.patch.object(instance, "save") as save: error = ValueError("some failure") plm.side_effect = error self.assertRaises( ValueError, self.compute._post_live_migration_update_host, self.context, instance, dest_host) save.assert_called_once() self.assertEqual(instance.host, dest_host) _test_post_live_migration() def test_post_live_migration_cinder_pre_344_api(self): # Because live migration has # succeeded,_post_live_migration_remove_source_vol_connections() # should call terminate_connection() with the volume UUID. dest_host = 'test_dest_host' instance = fake_instance.fake_instance_obj(self.context, node='dest', uuid=uuids.instance) vol_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.volume, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'id': 42, 'connection_info': '{"connector": {"host": "%s"}}' % dest_host}) image_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'image', 'destination_type': 'local', 'volume_id': uuids.image_volume, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid}) @mock.patch.object(self.compute.driver, 'get_volume_connector') @mock.patch.object(self.compute.volume_api, 'terminate_connection') def _test(mock_term_conn, mock_get_vol_conn): bdms = objects.BlockDeviceMappingList(objects=[vol_bdm, image_bdm]) self.compute._post_live_migration_remove_source_vol_connections( self.context, instance, bdms) mock_term_conn.assert_called_once_with( self.context, uuids.volume, mock_get_vol_conn.return_value) _test() def test_post_live_migration_cinder_v3_api(self): # Because live migration has succeeded, _post_live_migration # should call attachment_delete with the original/old attachment_id dest_host = 'test_dest_host' instance = fake_instance.fake_instance_obj(self.context, node='dest', uuid=uuids.instance) bdm_id = 1 volume_id = uuids.volume vol_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'id': bdm_id, 'connection_info': '{"connector": {"host": "%s"}}' % dest_host}) image_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'image', 'destination_type': 'local', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid}) vol_bdm.attachment_id = uuids.attachment image_bdm.attachment_id = uuids.attachment3 @mock.patch.object(self.compute.volume_api, 'attachment_delete') def _test(mock_attach_delete): bdms = objects.BlockDeviceMappingList(objects=[vol_bdm, image_bdm]) self.compute._post_live_migration_remove_source_vol_connections( self.context, instance, bdms) mock_attach_delete.assert_called_once_with( self.context, uuids.attachment) _test() @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid', return_value=objects.BlockDeviceMappingList()) @mock.patch('nova.compute.utils.notify_about_instance_action') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_post_live_migration_unplug_with_stashed_source_vifs( self, mock_add_fault, mock_notify, mock_get_bdms): """Tests the scenario that migrate_data.vifs is set so we unplug using the stashed source vifs from that rather than the current instance network info cache. """ migrate_data = objects.LibvirtLiveMigrateData() source_vif = network_model.VIF(uuids.port_id, type='ovs') migrate_data.vifs = [objects.VIFMigrateData(source_vif=source_vif)] bdms = objects.BlockDeviceMappingList(objects=[]) nw_info = network_model.NetworkInfo( [network_model.VIF(uuids.port_id, type='ovn')]) def fake_post_live_migration_at_source( _context, _instance, network_info): # Make sure we got the source_vif for unplug. self.assertEqual(1, len(network_info)) self.assertEqual(source_vif, network_info[0]) def fake_driver_cleanup(_context, _instance, network_info, *a, **kw): # Make sure we got the source_vif for unplug. self.assertEqual(1, len(network_info)) self.assertEqual(source_vif, network_info[0]) # Based on the number of mocks here, clearly _post_live_migration is # too big at this point... @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute.network_api, 'get_instance_nw_info', return_value=nw_info) @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute.network_api, 'migrate_instance_start') @mock.patch.object(self.compute.driver, 'post_live_migration_at_source', side_effect=fake_post_live_migration_at_source) @mock.patch.object(self.compute.compute_rpcapi, 'post_live_migration_at_destination') @mock.patch.object(self.compute, '_live_migration_cleanup_flags', return_value=(True, False)) @mock.patch.object(self.compute.driver, 'cleanup', side_effect=fake_driver_cleanup) @mock.patch.object(self.compute, 'update_available_resource') @mock.patch.object(self.compute, '_update_scheduler_instance_info') @mock.patch.object(self.compute, '_clean_instance_console_tokens') def _test(_clean_instance_console_tokens, _update_scheduler_instance_info, update_available_resource, driver_cleanup, _live_migration_cleanup_flags, post_live_migration_at_destination, post_live_migration_at_source, migrate_instance_start, _notify_about_instance_usage, get_instance_nw_info, _get_instance_block_device_info): self._mock_rt() self.compute._post_live_migration( self.context, self.instance, 'fake-dest', migrate_data=migrate_data, source_bdms=bdms) post_live_migration_at_source.assert_called_once_with( self.context, self.instance, test.MatchType(network_model.NetworkInfo)) driver_cleanup.assert_called_once_with( self.context, self.instance, test.MatchType(network_model.NetworkInfo), destroy_disks=False, migrate_data=migrate_data, destroy_vifs=False) _test() def _generate_volume_bdm_list(self, instance, original=False): # TODO(lyarwood): There are various methods generating fake bdms within # this class, we should really look at writing a small number of # generic reusable methods somewhere to replace all of these. connection_info = "{'data': {'host': 'dest'}}" if original: connection_info = "{'data': {'host': 'original'}}" vol1_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.vol1, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid, 'connection_info': connection_info}) vol1_bdm.save = mock.Mock() vol2_bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': uuids.vol2, 'device_name': '/dev/vdc', 'instance_uuid': instance.uuid, 'connection_info': connection_info}) vol2_bdm.save = mock.Mock() if original: vol1_bdm.attachment_id = uuids.vol1_attach_original vol2_bdm.attachment_id = uuids.vol2_attach_original else: vol1_bdm.attachment_id = uuids.vol1_attach vol2_bdm.attachment_id = uuids.vol2_attach return objects.BlockDeviceMappingList(objects=[vol1_bdm, vol2_bdm]) @mock.patch('nova.compute.rpcapi.ComputeAPI.remove_volume_connection') def test_remove_remote_volume_connections(self, mock_remove_vol_conn): instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) bdms = self._generate_volume_bdm_list(instance) self.compute._remove_remote_volume_connections(self.context, 'fake', bdms, instance) mock_remove_vol_conn.assert_has_calls([ mock.call(self.context, instance, bdm.volume_id, 'fake') for bdm in bdms]) @mock.patch('nova.compute.rpcapi.ComputeAPI.remove_volume_connection') def test_remove_remote_volume_connections_exc(self, mock_remove_vol_conn): instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) bdms = self._generate_volume_bdm_list(instance) # Raise an exception for the second call to remove_volume_connections mock_remove_vol_conn.side_effect = [None, test.TestingException] # Assert that errors are ignored self.compute._remove_remote_volume_connections(self.context, 'fake', bdms, instance) @mock.patch('nova.volume.cinder.API.attachment_delete') def test_rollback_volume_bdms(self, mock_delete_attachment): instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) bdms = self._generate_volume_bdm_list(instance) original_bdms = self._generate_volume_bdm_list(instance, original=True) self.compute._rollback_volume_bdms(self.context, bdms, original_bdms, instance) # Assert that we delete the current attachments mock_delete_attachment.assert_has_calls([ mock.call(self.context, uuids.vol1_attach), mock.call(self.context, uuids.vol2_attach)]) # Assert that we switch the attachment ids and connection_info for each # bdm back to their original values self.assertEqual(uuids.vol1_attach_original, bdms[0].attachment_id) self.assertEqual("{'data': {'host': 'original'}}", bdms[0].connection_info) self.assertEqual(uuids.vol2_attach_original, bdms[1].attachment_id) self.assertEqual("{'data': {'host': 'original'}}", bdms[1].connection_info) # Assert that save is called for each bdm bdms[0].save.assert_called_once() bdms[1].save.assert_called_once() @mock.patch('nova.compute.manager.LOG') @mock.patch('nova.volume.cinder.API.attachment_delete') def test_rollback_volume_bdms_exc(self, mock_delete_attachment, mock_log): instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) bdms = self._generate_volume_bdm_list(instance) original_bdms = self._generate_volume_bdm_list(instance, original=True) # Assert that we ignore cinderclient exceptions and continue to attempt # to rollback any remaining bdms. mock_delete_attachment.side_effect = [ cinder_exception.ClientException(code=9001), None] self.compute._rollback_volume_bdms(self.context, bdms, original_bdms, instance) self.assertEqual(uuids.vol2_attach_original, bdms[1].attachment_id) self.assertEqual("{'data': {'host': 'original'}}", bdms[1].connection_info) bdms[0].save.assert_not_called() bdms[1].save.assert_called_once() mock_log.warning.assert_called_once() self.assertIn('Ignoring cinderclient exception', mock_log.warning.call_args[0][0]) # Assert that we raise unknown Exceptions mock_log.reset_mock() bdms[0].save.reset_mock() bdms[1].save.reset_mock() mock_delete_attachment.side_effect = test.TestingException self.assertRaises(test.TestingException, self.compute._rollback_volume_bdms, self.context, bdms, original_bdms, instance) bdms[0].save.assert_not_called() bdms[1].save.assert_not_called() mock_log.exception.assert_called_once() self.assertIn('Exception while attempting to rollback', mock_log.exception.call_args[0][0]) @mock.patch('nova.volume.cinder.API.attachment_delete') def test_rollback_volume_bdms_after_pre_failure( self, mock_delete_attachment): instance = fake_instance.fake_instance_obj( self.context, uuid=uuids.instance) original_bdms = bdms = self._generate_volume_bdm_list(instance) self.compute._rollback_volume_bdms( self.context, bdms, original_bdms, instance) # Assert that attachment_delete isn't called when the bdms have already # been rolled back by a failure in pre_live_migration to reference the # source bdms. mock_delete_attachment.assert_not_called() @mock.patch.object(objects.ComputeNode, 'get_first_node_by_host_for_old_compat') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'remove_provider_tree_from_instance_allocation') def test_rollback_live_migration_cinder_v3_api(self, mock_remove_allocs, mock_get_node): compute = manager.ComputeManager() dest_node = objects.ComputeNode(host='foo', uuid=uuids.dest_node) mock_get_node.return_value = dest_node instance = fake_instance.fake_instance_obj(self.context, uuid=uuids.instance) volume_id = uuids.volume orig_attachment_id = uuids.attachment1 new_attachment_id = uuids.attachment2 migrate_data = migrate_data_obj.LiveMigrateData() migrate_data.old_vol_attachment_ids = { volume_id: orig_attachment_id} def fake_bdm(): bdm = fake_block_device.fake_bdm_object( self.context, {'source_type': 'volume', 'destination_type': 'volume', 'volume_id': volume_id, 'device_name': '/dev/vdb', 'instance_uuid': instance.uuid}) bdm.save = mock.Mock() return bdm # NOTE(mdbooth): Use of attachment_id as connection_info is a # test convenience. It just needs to be a string. source_bdm = fake_bdm() source_bdm.attachment_id = orig_attachment_id source_bdm.connection_info = orig_attachment_id source_bdms = objects.BlockDeviceMappingList(objects=[source_bdm]) bdm = fake_bdm() bdm.attachment_id = new_attachment_id bdm.connection_info = new_attachment_id bdms = objects.BlockDeviceMappingList(objects=[bdm]) @mock.patch('nova.objects.InstancePCIRequests.get_by_instance_uuid') @mock.patch.object(compute.volume_api, 'attachment_delete') @mock.patch.object(compute_utils, 'notify_about_instance_action') @mock.patch.object(instance, 'save') @mock.patch.object(compute, '_notify_about_instance_usage') @mock.patch.object(compute.compute_rpcapi, 'remove_volume_connection') @mock.patch.object(compute, 'network_api') @mock.patch.object(objects.BlockDeviceMappingList, 'get_by_instance_uuid') @mock.patch.object(objects.Instance, 'drop_migration_context') def _test(mock_drop_mig_ctxt, mock_get_bdms, mock_net_api, mock_remove_conn, mock_usage, mock_instance_save, mock_action, mock_attach_delete, mock_get_pci): # this tests that _rollback_live_migration replaces the bdm's # attachment_id with the original attachment id that is in # migrate_data. mock_get_bdms.return_value = bdms mock_get_pci.return_value = objects.InstancePCIRequests() compute._rollback_live_migration(self.context, instance, None, migrate_data=migrate_data, source_bdms=source_bdms) mock_remove_conn.assert_called_once_with(self.context, instance, bdm.volume_id, None) mock_attach_delete.assert_called_once_with(self.context, new_attachment_id) self.assertEqual(bdm.attachment_id, orig_attachment_id) self.assertEqual(orig_attachment_id, bdm.connection_info) bdm.save.assert_called_once_with() mock_drop_mig_ctxt.assert_called_once_with() mock_get_pci.assert_called_once_with(self.context, instance.uuid) self.assertEqual(mock_get_pci.return_value, instance.pci_requests) _test() @mock.patch('nova.compute.manager.LOG.error') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid', return_value=objects.BlockDeviceMappingList()) @mock.patch('nova.compute.utils.notify_about_instance_action') def test_rollback_live_migration_port_binding_delete_fails( self, mock_notify, mock_get_bdms, mock_log_error): """Tests that neutron fails to delete the destination host port bindings but we handle the error and just log it. """ migrate_data = objects.LibvirtLiveMigrateData( migration=self.migration, is_shared_instance_path=True, is_shared_block_storage=True) @mock.patch('nova.objects.InstancePCIRequests.get_by_instance_uuid') @mock.patch.object(self.compute, '_revert_allocation') @mock.patch.object(self.instance, 'save') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(objects.Instance, 'drop_migration_context') def _do_test(drop_mig_ctxt, legacy_notify, network_api, instance_save, _revert_allocation, mock_get_pci): # setup_networks_on_host is called two times: # 1. set the migrating_to attribute in the port binding profile, # which is a no-op in this case for neutron # 2. delete the dest host port bindings - we make this raise network_api.setup_networks_on_host.side_effect = [ None, exception.PortBindingDeletionFailed( port_id=uuids.port_id, host='fake-dest-host')] mock_get_pci.return_value = objects.InstancePCIRequests() self.compute._rollback_live_migration( self.context, self.instance, 'fake-dest-host', migrate_data, source_bdms=objects.BlockDeviceMappingList()) self.assertEqual(1, mock_log_error.call_count) self.assertIn('Network cleanup failed for destination host', mock_log_error.call_args[0][0]) drop_mig_ctxt.assert_called_once_with() mock_get_pci.assert_called_once_with( self.context, self.instance.uuid) self.assertEqual( mock_get_pci.return_value, self.instance.pci_requests) _do_test() @mock.patch('nova.compute.manager.LOG.error') def test_rollback_live_migration_at_destination_port_binding_delete_fails( self, mock_log_error): """Tests that neutron fails to delete the destination host port bindings but we handle the error and just log it. """ @mock.patch.object(self.compute, 'rt') @mock.patch.object(self.compute, '_notify_about_instance_usage') @mock.patch.object(self.compute, 'network_api') @mock.patch.object(self.compute, '_get_instance_block_device_info') @mock.patch.object(self.compute.driver, 'rollback_live_migration_at_destination') def _do_test(driver_rollback, get_bdms, network_api, legacy_notify, rt_mock): self.compute.network_api.setup_networks_on_host.side_effect = ( exception.PortBindingDeletionFailed( port_id=uuids.port_id, host='fake-dest-host')) mock_md = mock.MagicMock() self.compute.rollback_live_migration_at_destination( self.context, self.instance, destroy_disks=False, migrate_data=mock_md) self.assertEqual(1, mock_log_error.call_count) self.assertIn('Network cleanup failed for destination host', mock_log_error.call_args[0][0]) driver_rollback.assert_called_once_with( self.context, self.instance, network_api.get_instance_nw_info.return_value, get_bdms.return_value, destroy_disks=False, migrate_data=mock_md) rt_mock.free_pci_device_claims_for_instance.\ assert_called_once_with(self.context, self.instance) _do_test() def _get_migration(self, migration_id, status, migration_type): migration = objects.Migration() migration.id = migration_id migration.status = status migration.migration_type = migration_type return migration @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') @mock.patch.object(objects.Migration, 'get_by_id') @mock.patch.object(nova.virt.fake.FakeDriver, 'live_migration_abort') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_live_migration_abort(self, mock_notify_action, mock_driver, mock_get_migration, mock_notify): instance = objects.Instance(id=123, uuid=uuids.instance) migration = self._get_migration(10, 'running', 'live-migration') mock_get_migration.return_value = migration self.compute.live_migration_abort(self.context, instance, migration.id) mock_driver.assert_called_with(instance) mock_notify.assert_has_calls( [mock.call(self.context, instance, 'live.migration.abort.start'), mock.call(self.context, instance, 'live.migration.abort.end')] ) mock_notify_action.assert_has_calls( [mock.call(self.context, instance, 'fake-mini', action='live_migration_abort', phase='start'), mock.call(self.context, instance, 'fake-mini', action='live_migration_abort', phase='end')] ) @mock.patch.object(objects.Instance, 'save') @mock.patch.object(manager.ComputeManager, '_revert_allocation') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') @mock.patch.object(objects.Migration, 'get_by_id') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_live_migration_abort_queued(self, mock_notify_action, mock_get_migration, mock_notify, mock_revert_allocations, mock_instance_save): instance = objects.Instance(id=123, uuid=uuids.instance) migration = self._get_migration(10, 'queued', 'live-migration') migration.dest_compute = uuids.dest migration.dest_node = uuids.dest migration.save = mock.MagicMock() mock_get_migration.return_value = migration fake_future = mock.MagicMock() self.compute._waiting_live_migrations[instance.uuid] = ( migration, fake_future) with mock.patch.object( self.compute.network_api, 'setup_networks_on_host') as mock_setup_net: self.compute.live_migration_abort( self.context, instance, migration.id) mock_setup_net.assert_called_once_with( self.context, instance, host=migration.dest_compute, teardown=True) mock_revert_allocations.assert_called_once_with( self.context, instance, migration) mock_notify.assert_has_calls( [mock.call(self.context, instance, 'live.migration.abort.start'), mock.call(self.context, instance, 'live.migration.abort.end')] ) mock_notify_action.assert_has_calls( [mock.call(self.context, instance, 'fake-mini', action='live_migration_abort', phase='start'), mock.call(self.context, instance, 'fake-mini', action='live_migration_abort', phase='end')] ) self.assertEqual('cancelled', migration.status) fake_future.cancel.assert_called_once_with() @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') @mock.patch.object(objects.Migration, 'get_by_id') @mock.patch.object(nova.virt.fake.FakeDriver, 'live_migration_abort') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_live_migration_abort_not_supported(self, mock_notify_action, mock_driver, mock_get_migration, mock_notify, mock_instance_fault): instance = objects.Instance(id=123, uuid=uuids.instance) migration = self._get_migration(10, 'running', 'live-migration') mock_get_migration.return_value = migration mock_driver.side_effect = NotImplementedError() self.assertRaises(NotImplementedError, self.compute.live_migration_abort, self.context, instance, migration.id) mock_notify_action.assert_called_once_with(self.context, instance, 'fake-mini', action='live_migration_abort', phase='start') @mock.patch.object(compute_utils, 'add_instance_fault_from_exc') @mock.patch.object(manager.ComputeManager, '_notify_about_instance_usage') @mock.patch.object(objects.Migration, 'get_by_id') @mock.patch('nova.compute.utils.notify_about_instance_action') def test_live_migration_abort_wrong_migration_state(self, mock_notify_action, mock_get_migration, mock_notify, mock_instance_fault): instance = objects.Instance(id=123, uuid=uuids.instance) migration = self._get_migration(10, 'completed', 'live-migration') mock_get_migration.return_value = migration self.assertRaises(exception.InvalidMigrationState, self.compute.live_migration_abort, self.context, instance, migration.id) def test_live_migration_cleanup_flags_shared_path_and_vpmem_libvirt(self): migrate_data = objects.LibvirtLiveMigrateData( is_shared_block_storage=False, is_shared_instance_path=True) migr_ctxt = objects.MigrationContext() vpmem_resource = objects.Resource( provider_uuid=uuids.rp_uuid, resource_class="CUSTOM_PMEM_NAMESPACE_4GB", identifier='ns_0', metadata=objects.LibvirtVPMEMDevice( label='4GB', name='ns_0', devpath='/dev/dax0.0', size=4292870144, align=2097152)) migr_ctxt.old_resources = objects.ResourceList( objects=[vpmem_resource]) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data, migr_ctxt) self.assertTrue(do_cleanup) self.assertTrue(destroy_disks) def test_live_migration_cleanup_flags_block_migrate_libvirt(self): migrate_data = objects.LibvirtLiveMigrateData( is_shared_block_storage=False, is_shared_instance_path=False) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertTrue(do_cleanup) self.assertTrue(destroy_disks) def test_live_migration_cleanup_flags_shared_block_libvirt(self): migrate_data = objects.LibvirtLiveMigrateData( is_shared_block_storage=True, is_shared_instance_path=False) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertTrue(do_cleanup) self.assertFalse(destroy_disks) def test_live_migration_cleanup_flags_shared_path_libvirt(self): migrate_data = objects.LibvirtLiveMigrateData( is_shared_block_storage=False, is_shared_instance_path=True) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertFalse(do_cleanup) self.assertTrue(destroy_disks) def test_live_migration_cleanup_flags_shared_libvirt(self): migrate_data = objects.LibvirtLiveMigrateData( is_shared_block_storage=True, is_shared_instance_path=True) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertFalse(do_cleanup) self.assertFalse(destroy_disks) def test_live_migration_cleanup_flags_live_migrate(self): do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( {}) self.assertFalse(do_cleanup) self.assertFalse(destroy_disks) def test_live_migration_cleanup_flags_block_migrate_hyperv(self): migrate_data = objects.HyperVLiveMigrateData( is_shared_instance_path=False) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertTrue(do_cleanup) self.assertTrue(destroy_disks) def test_live_migration_cleanup_flags_shared_hyperv(self): migrate_data = objects.HyperVLiveMigrateData( is_shared_instance_path=True) do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertTrue(do_cleanup) self.assertFalse(destroy_disks) def test_live_migration_cleanup_flags_other(self): migrate_data = mock.Mock() do_cleanup, destroy_disks = self.compute._live_migration_cleanup_flags( migrate_data) self.assertFalse(do_cleanup) self.assertFalse(destroy_disks) @mock.patch('nova.compute.utils.notify_about_resize_prep_instance') @mock.patch('nova.objects.ComputeNode.get_by_host_and_nodename') @mock.patch('nova.objects.InstanceFault.create') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.compute.utils.notify_about_instance_usage') @mock.patch('nova.compute.utils.is_volume_backed_instance', new=lambda *a: False) def test_prep_resize_errors_migration(self, mock_niu, mock_notify, mock_save, mock_if, mock_cn, mock_notify_resize): migration = mock.MagicMock() flavor = objects.Flavor(name='flavor', id=1) cn = objects.ComputeNode(uuid=uuids.compute) mock_cn.return_value = cn reportclient = self.compute.reportclient @mock.patch.object(self.compute, '_reschedule_resize_or_reraise') @mock.patch.object(self.compute, '_prep_resize') def doit(mock_pr, mock_r): # Mock the resource tracker, but keep the report client self._mock_rt().reportclient = reportclient mock_pr.side_effect = test.TestingException mock_r.side_effect = test.TestingException request_spec = objects.RequestSpec() instance = objects.Instance(uuid=uuids.instance, id=1, host='host', node='node', vm_state='active', task_state=None) self.assertRaises(test.TestingException, self.compute.prep_resize, self.context, mock.sentinel.image, instance, flavor, request_spec, {}, 'node', False, migration, []) # Make sure we set migration status to error self.assertEqual(migration.status, 'error') migration.save.assert_called_once_with() mock_r.assert_called_once_with( self.context, instance, mock.ANY, flavor, request_spec, {}, []) mock_pr.assert_called_once_with( self.context, mock.sentinel.image, instance, flavor, {}, 'node', migration, request_spec, False) mock_notify_resize.assert_has_calls([ mock.call(self.context, instance, 'fake-mini', 'start', flavor), mock.call(self.context, instance, 'fake-mini', 'end', flavor)]) doit() def test_prep_resize_fails_unable_to_migrate_to_self(self): """Asserts that _prep_resize handles UnableToMigrateToSelf when _prep_resize is called on the host on which the instance lives and the flavor is not changing. """ instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, expected_attrs=['system_metadata', 'flavor']) migration = mock.MagicMock(spec='nova.objects.Migration') with mock.patch.dict(self.compute.driver.capabilities, {'supports_migrate_to_same_host': False}): ex = self.assertRaises( exception.InstanceFaultRollback, self.compute._prep_resize, self.context, instance.image_meta, instance, instance.flavor, filter_properties={}, node=instance.node, migration=migration, request_spec=mock.sentinel) self.assertIsInstance( ex.inner_exception, exception.UnableToMigrateToSelf) @mock.patch('nova.compute.rpcapi.ComputeAPI.resize_instance') @mock.patch('nova.compute.resource_tracker.ResourceTracker.resize_claim') @mock.patch('nova.compute.manager.ComputeManager.' '_get_request_group_mapping') @mock.patch.object(objects.Instance, 'save') def test_prep_resize_handles_legacy_request_spec( self, mock_save, mock_get_mapping, mock_claim, mock_resize): instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, expected_attrs=['system_metadata', 'flavor']) request_spec = objects.RequestSpec(instance_uuid = instance.uuid) mock_get_mapping.return_value = {} self.compute._prep_resize( self.context, instance.image_meta, instance, instance.flavor, filter_properties={}, node=instance.node, migration=self.migration, request_spec=request_spec.to_legacy_request_spec_dict()) # we expect that the legacy request spec is transformed to object # before _prep_resize calls _get_request_group_mapping() mock_get_mapping.assert_called_once_with( test.MatchType(objects.RequestSpec)) @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_resize_prep_instance') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager._revert_allocation') @mock.patch('nova.compute.manager.ComputeManager.' '_reschedule_resize_or_reraise') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_prep_resize_fails_rollback( self, add_instance_fault_from_exc, _reschedule_resize_or_reraise, _revert_allocation, mock_instance_save, notify_about_resize_prep_instance, _notify_about_instance_usage, notify_usage_exists): """Tests that if _prep_resize raises InstanceFaultRollback, the instance.vm_state is reset properly in _error_out_instance_on_exception """ instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, vm_state=vm_states.STOPPED, node='fake-node', expected_attrs=['system_metadata', 'flavor']) migration = mock.MagicMock(spec='nova.objects.Migration') request_spec = mock.MagicMock(spec='nova.objects.RequestSpec') ex = exception.InstanceFaultRollback( inner_exception=exception.UnableToMigrateToSelf( instance_id=instance.uuid, host=instance.host)) def fake_reschedule_resize_or_reraise(*args, **kwargs): raise ex _reschedule_resize_or_reraise.side_effect = ( fake_reschedule_resize_or_reraise) with mock.patch.object(self.compute, '_prep_resize', side_effect=ex): self.assertRaises( # _error_out_instance_on_exception should reraise the # UnableToMigrateToSelf inside InstanceFaultRollback. exception.UnableToMigrateToSelf, self.compute.prep_resize, self.context, instance.image_meta, instance, instance.flavor, request_spec, filter_properties={}, node=instance.node, clean_shutdown=True, migration=migration, host_list=[]) # The instance.vm_state should remain unchanged # (_error_out_instance_on_exception will set to ACTIVE by default). self.assertEqual(vm_states.STOPPED, instance.vm_state) @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_resize_prep_instance') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager._revert_allocation') @mock.patch('nova.compute.manager.ComputeManager.' '_reschedule_resize_or_reraise') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') # this is almost copy-paste from test_prep_resize_fails_rollback def test_prep_resize_fails_group_validation( self, add_instance_fault_from_exc, _reschedule_resize_or_reraise, _revert_allocation, mock_instance_save, notify_about_resize_prep_instance, _notify_about_instance_usage, notify_usage_exists): """Tests that if _validate_instance_group_policy raises InstanceFaultRollback, the instance.vm_state is reset properly in _error_out_instance_on_exception """ instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, vm_state=vm_states.STOPPED, node='fake-node', expected_attrs=['system_metadata', 'flavor']) migration = mock.MagicMock(spec='nova.objects.Migration') request_spec = mock.MagicMock(spec='nova.objects.RequestSpec') ex = exception.RescheduledException( instance_uuid=instance.uuid, reason="policy violated") ex2 = exception.InstanceFaultRollback( inner_exception=ex) def fake_reschedule_resize_or_reraise(*args, **kwargs): raise ex2 _reschedule_resize_or_reraise.side_effect = ( fake_reschedule_resize_or_reraise) with mock.patch.object(self.compute, '_validate_instance_group_policy', side_effect=ex): self.assertRaises( # _error_out_instance_on_exception should reraise the # RescheduledException inside InstanceFaultRollback. exception.RescheduledException, self.compute.prep_resize, self.context, instance.image_meta, instance, instance.flavor, request_spec, filter_properties={}, node=instance.node, clean_shutdown=True, migration=migration, host_list=[]) # The instance.vm_state should remain unchanged # (_error_out_instance_on_exception will set to ACTIVE by default). self.assertEqual(vm_states.STOPPED, instance.vm_state) @mock.patch('nova.compute.rpcapi.ComputeAPI.resize_instance') @mock.patch('nova.compute.resource_tracker.ResourceTracker.resize_claim') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.utils.' 'update_pci_request_with_placement_allocations') @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_resize_prep_instance') def test_prep_resize_update_pci_request( self, mock_notify, mock_notify_usage, mock_notify_exists, mock_update_pci, mock_save, mock_claim, mock_resize): instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, vm_state=vm_states.STOPPED, node='fake-node', expected_attrs=['system_metadata', 'flavor']) instance.pci_requests = objects.InstancePCIRequests(requests=[]) request_spec = objects.RequestSpec() request_spec.requested_resources = [ objects.RequestGroup( requester_id=uuids.port_id, provider_uuids=[uuids.rp_uuid])] self.compute.prep_resize( self.context, instance.image_meta, instance, instance.flavor, request_spec, filter_properties={}, node=instance.node, clean_shutdown=True, migration=self.migration, host_list=[]) mock_update_pci.assert_called_once_with( self.context, self.compute.reportclient, [], {uuids.port_id: [uuids.rp_uuid]}) mock_save.assert_called_once_with() @mock.patch('nova.compute.manager.ComputeManager._revert_allocation') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.compute.rpcapi.ComputeAPI.resize_instance') @mock.patch('nova.compute.resource_tracker.ResourceTracker.resize_claim') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.utils.' 'update_pci_request_with_placement_allocations') @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.compute.manager.ComputeManager.' '_notify_about_instance_usage') @mock.patch('nova.compute.utils.notify_about_resize_prep_instance') def test_prep_resize_update_pci_request_fails( self, mock_notify, mock_notify_usage, mock_notify_exists, mock_update_pci, mock_save, mock_claim, mock_resize, mock_fault, mock_revert): instance = fake_instance.fake_instance_obj( self.context, host=self.compute.host, vm_state=vm_states.STOPPED, node='fake-node', expected_attrs=['system_metadata', 'flavor']) instance.pci_requests = objects.InstancePCIRequests(requests=[]) request_spec = objects.RequestSpec() request_spec.requested_resources = [ objects.RequestGroup( requester_id=uuids.port_id, provider_uuids=[uuids.rp_uuid])] migration = mock.MagicMock(spec='nova.objects.Migration') mock_update_pci.side_effect = exception.BuildAbortException( instance_uuid=instance.uuid, reason="") self.assertRaises( exception.BuildAbortException, self.compute.prep_resize, self.context, instance.image_meta, instance, instance.flavor, request_spec, filter_properties={}, node=instance.node, clean_shutdown=True, migration=migration, host_list=[]) mock_revert.assert_called_once_with(self.context, instance, migration) mock_notify.assert_has_calls([ mock.call( self.context, instance, 'fake-mini', 'start', instance.flavor), mock.call( self.context, instance, 'fake-mini', 'end', instance.flavor), ]) mock_notify_usage.assert_has_calls([ mock.call( self.context, instance, 'resize.prep.start', extra_usage_info=None), mock.call( self.context, instance, 'resize.prep.end', extra_usage_info={ 'new_instance_type_id': 1, 'new_instance_type': 'flavor1'}), ]) mock_notify_exists.assert_called_once_with( self.compute.notifier, self.context, instance, 'fake-mini', current_period=True) def test__claim_pci_for_instance_no_vifs(self): @mock.patch.object(self.compute, 'rt') @mock.patch.object(pci_request, 'get_instance_pci_request_from_vif') @mock.patch.object(self.instance, 'get_network_info') def _test(mock_get_network_info, mock_get_instance_pci_request_from_vif, rt_mock): # when no VIFS, expecting no pci devices to be claimed mock_get_network_info.return_value = [] port_id_to_pci = self.compute._claim_pci_for_instance_vifs( self.context, self.instance) rt_mock.claim_pci_devices.assert_not_called() self.assertEqual(0, len(port_id_to_pci)) _test() def test__claim_pci_for_instance_vifs(self): @mock.patch.object(self.compute, 'rt') @mock.patch.object(pci_request, 'get_instance_pci_request_from_vif') @mock.patch.object(self.instance, 'get_network_info') def _test(mock_get_network_info, mock_get_instance_pci_request_from_vif, rt_mock): # when there are VIFs, expect only ones with related PCI to be # claimed and their migrate vif profile to be updated. # Mock needed objects nw_vifs = network_model.NetworkInfo([ network_model.VIF( id=uuids.port0, vnic_type='direct', type=network_model.VIF_TYPE_HW_VEB, profile={'pci_slot': '0000:04:00.3', 'pci_vendor_info': '15b3:1018', 'physical_network': 'default'}), network_model.VIF( id=uuids.port1, vnic_type='normal', type=network_model.VIF_TYPE_OVS, profile={'some': 'attribute'})]) mock_get_network_info.return_value = nw_vifs pci_req = objects.InstancePCIRequest(request_id=uuids.pci_req) instance_pci_reqs = objects.InstancePCIRequests(requests=[pci_req]) instance_pci_devs = objects.PciDeviceList( objects=[objects.PciDevice(request_id=uuids.pci_req, address='0000:04:00.3', vendor_id='15b3', product_id='1018')]) def get_pci_req_side_effect(context, instance, vif): if vif['id'] == uuids.port0: return pci_req return None mock_get_instance_pci_request_from_vif.side_effect = \ get_pci_req_side_effect self.instance.pci_devices = instance_pci_devs self.instance.pci_requests = instance_pci_reqs self.instance.migration_context = objects.MigrationContext( new_numa_topology=objects.InstanceNUMATopology()) rt_mock.reset() claimed_pci_dev = objects.PciDevice(request_id=uuids.pci_req, address='0000:05:00.4', vendor_id='15b3', product_id='1018') rt_mock.claim_pci_devices.return_value = [claimed_pci_dev] # Do the actual test port_id_to_pci = self.compute._claim_pci_for_instance_vifs( self.context, self.instance) self.assertEqual(len(nw_vifs), mock_get_instance_pci_request_from_vif.call_count) rt_mock.claim_pci_devices.assert_called_once_with( self.context, test.MatchType(objects.InstancePCIRequests), self.instance.migration_context.new_numa_topology) self.assertEqual(len(port_id_to_pci), 1) _test() def test__update_migrate_vifs_profile_with_pci(self): # Define three migrate vifs with two pci devs that are required # to be updated, one VF and on PF. # Make sure method under test updated the correct devs with the correct # values. nw_vifs = network_model.NetworkInfo( [ network_model.VIF( id=uuids.port0, vnic_type='direct', type=network_model.VIF_TYPE_HW_VEB, profile={ 'pci_slot': '0000:04:00.3', 'pci_vendor_info': '15b3:1018', 'physical_network': 'default', }, ), network_model.VIF( id=uuids.port1, vnic_type='normal', type=network_model.VIF_TYPE_OVS, profile={'some': 'attribute'}, ), network_model.VIF( id=uuids.port2, vnic_type='direct-physical', type=network_model.VIF_TYPE_HOSTDEV, profile={ 'pci_slot': '0000:01:00', 'pci_vendor_info': '8086:154d', 'physical_network': 'physnet2', }, ), ] ) pci_vf_dev = objects.PciDevice( request_id=uuids.pci_req, address='0000:05:00.4', parent_addr='0000:05:00', vendor_id='15b3', product_id='1018', compute_node_id=13, dev_type=fields.PciDeviceType.SRIOV_VF, ) pci_pf_dev = objects.PciDevice( request_id=uuids.pci_req2, address='0000:01:00', parent_addr='0000:02:00', vendor_id='8086', product_id='154d', compute_node_id=13, dev_type=fields.PciDeviceType.SRIOV_PF, extra_info={'mac_address': 'b4:96:91:34:f4:36'}, ) port_id_to_pci_dev = { uuids.port0: pci_vf_dev, uuids.port2: pci_pf_dev, } mig_vifs = ( migrate_data_obj.VIFMigrateData.create_skeleton_migrate_vifs( nw_vifs) ) self.compute._update_migrate_vifs_profile_with_pci( mig_vifs, port_id_to_pci_dev) # Make sure method under test updated the correct one. changed_vf_mig_vif = mig_vifs[0] unchanged_mig_vif = mig_vifs[1] changed_pf_mig_vif = mig_vifs[2] # Migrate vifs profile was updated with pci_dev.address # for port ID uuids.port0. self.assertEqual(changed_vf_mig_vif.profile['pci_slot'], pci_vf_dev.address) # MAC is not added as this is a VF self.assertNotIn('device_mac_address', changed_vf_mig_vif.profile) # Migrate vifs profile was unchanged for port ID uuids.port1. # i.e 'profile' attribute does not exist. self.assertNotIn('profile', unchanged_mig_vif) # Migrate vifs profile was updated with pci_dev.address # for port ID uuids.port2. self.assertEqual(changed_pf_mig_vif.profile['pci_slot'], pci_pf_dev.address) # MAC is updated as this is a PF self.assertEqual( 'b4:96:91:34:f4:36', changed_pf_mig_vif.profile['device_mac_address'] ) def test_get_updated_nw_info_with_pci_mapping(self): old_dev = objects.PciDevice(address='0000:04:00.2') new_dev = objects.PciDevice(address='0000:05:00.3') pci_mapping = {old_dev.address: new_dev} nw_info = network_model.NetworkInfo([ network_model.VIF( id=uuids.port1, vnic_type=network_model.VNIC_TYPE_NORMAL), network_model.VIF( id=uuids.port2, vnic_type=network_model.VNIC_TYPE_DIRECT, profile={'pci_slot': old_dev.address})]) updated_nw_info = self.compute._get_updated_nw_info_with_pci_mapping( nw_info, pci_mapping) self.assertDictEqual(nw_info[0], updated_nw_info[0]) self.assertEqual(new_dev.address, updated_nw_info[1]['profile']['pci_slot']) @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocs_for_consumer') def test_prep_snapshot_based_resize_at_dest(self, get_allocs): """Tests happy path for prep_snapshot_based_resize_at_dest""" # Setup mocks. flavor = self.instance.flavor limits = objects.SchedulerLimits() # resize_claim normally sets instance.migration_context and returns # a MoveClaim which is a context manager. Rather than deal with # mocking a context manager we just set the migration_context on the # fake instance ahead of time to ensure it is returned as expected. self.instance.migration_context = objects.MigrationContext() with test.nested( mock.patch.object(self.compute, '_send_prep_resize_notifications'), mock.patch.object(self.compute.rt, 'resize_claim'), ) as ( _send_prep_resize_notifications, resize_claim, ): # Run the code. mc = self.compute.prep_snapshot_based_resize_at_dest( self.context, self.instance, flavor, 'nodename', self.migration, limits) self.assertIs(mc, self.instance.migration_context) # Assert the mock calls. _send_prep_resize_notifications.assert_has_calls([ mock.call(self.context, self.instance, fields.NotificationPhase.START, flavor), mock.call(self.context, self.instance, fields.NotificationPhase.END, flavor)]) resize_claim.assert_called_once_with( self.context, self.instance, flavor, 'nodename', self.migration, get_allocs.return_value['allocations'], image_meta=test.MatchType(objects.ImageMeta), limits=limits) @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocs_for_consumer') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_prep_snapshot_based_resize_at_dest_get_allocs_fails( self, add_fault, get_allocs): """Tests that getting allocations fails and ExpectedException is raised with the MigrationPreCheckError inside. """ # Setup mocks. flavor = self.instance.flavor limits = objects.SchedulerLimits() ex1 = exception.ConsumerAllocationRetrievalFailed( consumer_uuid=self.instance.uuid, error='oops') get_allocs.side_effect = ex1 with test.nested( mock.patch.object(self.compute, '_send_prep_resize_notifications'), mock.patch.object(self.compute.rt, 'resize_claim') ) as ( _send_prep_resize_notifications, resize_claim, ): # Run the code. ex2 = self.assertRaises( messaging.ExpectedException, self.compute.prep_snapshot_based_resize_at_dest, self.context, self.instance, flavor, 'nodename', self.migration, limits) wrapped_exc = ex2.exc_info[1] # The original error should be in the MigrationPreCheckError which # itself is in the ExpectedException. self.assertIn(ex1.format_message(), str(wrapped_exc)) # Assert the mock calls. _send_prep_resize_notifications.assert_has_calls([ mock.call(self.context, self.instance, fields.NotificationPhase.START, flavor), mock.call(self.context, self.instance, fields.NotificationPhase.END, flavor)]) resize_claim.assert_not_called() # Assert the decorators that are triggered on error add_fault.assert_called_once_with( self.context, self.instance, wrapped_exc, mock.ANY) # There would really be three notifications but because we mocked out # _send_prep_resize_notifications there is just the one error # notification from the wrap_exception decorator. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocs_for_consumer') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_prep_snapshot_based_resize_at_dest_claim_fails( self, add_fault, get_allocs): """Tests that the resize_claim fails and ExpectedException is raised with the MigrationPreCheckError inside. """ # Setup mocks. flavor = self.instance.flavor limits = objects.SchedulerLimits() ex1 = exception.ComputeResourcesUnavailable(reason='numa') with test.nested( mock.patch.object(self.compute, '_send_prep_resize_notifications'), mock.patch.object(self.compute.rt, 'resize_claim', side_effect=ex1) ) as ( _send_prep_resize_notifications, resize_claim, ): # Run the code. ex2 = self.assertRaises( messaging.ExpectedException, self.compute.prep_snapshot_based_resize_at_dest, self.context, self.instance, flavor, 'nodename', self.migration, limits) wrapped_exc = ex2.exc_info[1] # The original error should be in the MigrationPreCheckError which # itself is in the ExpectedException. self.assertIn(ex1.format_message(), str(wrapped_exc)) # Assert the mock calls. _send_prep_resize_notifications.assert_has_calls([ mock.call(self.context, self.instance, fields.NotificationPhase.START, flavor), mock.call(self.context, self.instance, fields.NotificationPhase.END, flavor)]) resize_claim.assert_called_once_with( self.context, self.instance, flavor, 'nodename', self.migration, get_allocs.return_value['allocations'], image_meta=test.MatchType(objects.ImageMeta), limits=limits) # Assert the decorators that are triggered on error add_fault.assert_called_once_with( self.context, self.instance, wrapped_exc, mock.ANY) # There would really be three notifications but because we mocked out # _send_prep_resize_notifications there is just the one error # notification from the wrap_exception decorator. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) def test_snapshot_for_resize(self): """Happy path test for _snapshot_for_resize.""" with mock.patch.object(self.compute.driver, 'snapshot') as snapshot: self.compute._snapshot_for_resize( self.context, uuids.image_id, self.instance) snapshot.assert_called_once_with( self.context, self.instance, uuids.image_id, mock.ANY) def test_snapshot_for_resize_delete_image_on_error(self): """Tests that any exception raised from _snapshot_for_resize will result in attempting to delete the image. """ with mock.patch.object(self.compute.driver, 'snapshot', side_effect=test.TestingException) as snapshot: with mock.patch.object(self.compute.image_api, 'delete') as delete: self.assertRaises(test.TestingException, self.compute._snapshot_for_resize, self.context, uuids.image_id, self.instance) snapshot.assert_called_once_with( self.context, self.instance, uuids.image_id, mock.ANY) delete.assert_called_once_with(self.context, uuids.image_id) @mock.patch('nova.objects.Instance.get_bdms', return_value=objects.BlockDeviceMappingList()) @mock.patch('nova.objects.Instance.save') @mock.patch( 'nova.compute.manager.InstanceEvents.clear_events_for_instance') def _test_prep_snapshot_based_resize_at_source( self, clear_events_for_instance, instance_save, get_bdms, snapshot_id=None): """Happy path test for prep_snapshot_based_resize_at_source.""" with test.nested( mock.patch.object( self.compute.network_api, 'get_instance_nw_info', return_value=network_model.NetworkInfo()), mock.patch.object( self.compute, '_send_resize_instance_notifications'), mock.patch.object(self.compute, '_power_off_instance'), mock.patch.object(self.compute, '_get_power_state', return_value=power_state.SHUTDOWN), mock.patch.object(self.compute, '_snapshot_for_resize'), mock.patch.object(self.compute, '_get_instance_block_device_info'), mock.patch.object(self.compute.driver, 'destroy'), mock.patch.object(self.compute, '_terminate_volume_connections'), mock.patch.object( self.compute.network_api, 'migrate_instance_start') ) as ( get_instance_nw_info, _send_resize_instance_notifications, _power_off_instance, _get_power_state, _snapshot_for_resize, _get_instance_block_device_info, destroy, _terminate_volume_connections, migrate_instance_start ): self.compute.prep_snapshot_based_resize_at_source( self.context, self.instance, self.migration, snapshot_id=snapshot_id) # Assert the mock calls. get_instance_nw_info.assert_called_once_with( self.context, self.instance) _send_resize_instance_notifications.assert_has_calls([ mock.call(self.context, self.instance, get_bdms.return_value, get_instance_nw_info.return_value, fields.NotificationPhase.START), mock.call(self.context, self.instance, get_bdms.return_value, get_instance_nw_info.return_value, fields.NotificationPhase.END)]) _power_off_instance.assert_called_once_with(self.instance) self.assertEqual(power_state.SHUTDOWN, self.instance.power_state) if snapshot_id is None: _snapshot_for_resize.assert_not_called() else: _snapshot_for_resize.assert_called_once_with( self.context, snapshot_id, self.instance) _get_instance_block_device_info.assert_called_once_with( self.context, self.instance, bdms=get_bdms.return_value) destroy.assert_called_once_with( self.context, self.instance, get_instance_nw_info.return_value, block_device_info=_get_instance_block_device_info.return_value, destroy_disks=False) _terminate_volume_connections.assert_called_once_with( self.context, self.instance, get_bdms.return_value) migrate_instance_start.assert_called_once_with( self.context, self.instance, self.migration) self.assertEqual('post-migrating', self.migration.status) self.assertEqual(2, self.migration.save.call_count) self.assertEqual(task_states.RESIZE_MIGRATED, self.instance.task_state) instance_save.assert_called_once_with( expected_task_state=task_states.RESIZE_MIGRATING) clear_events_for_instance.assert_called_once_with(self.instance) def test_prep_snapshot_based_resize_at_source_with_snapshot(self): self._test_prep_snapshot_based_resize_at_source( snapshot_id=uuids.snapshot_id) def test_prep_snapshot_based_resize_at_source_without_snapshot(self): self._test_prep_snapshot_based_resize_at_source() @mock.patch('nova.objects.Instance.get_bdms', return_value=objects.BlockDeviceMappingList()) def test_prep_snapshot_based_resize_at_source_power_off_failure( self, get_bdms): """Tests that the power off fails and raises InstancePowerOffFailure""" with test.nested( mock.patch.object( self.compute.network_api, 'get_instance_nw_info', return_value=network_model.NetworkInfo()), mock.patch.object( self.compute, '_send_resize_instance_notifications'), mock.patch.object(self.compute, '_power_off_instance', side_effect=test.TestingException), ) as ( get_instance_nw_info, _send_resize_instance_notifications, _power_off_instance, ): self.assertRaises( exception.InstancePowerOffFailure, self.compute._prep_snapshot_based_resize_at_source, self.context, self.instance, self.migration) _power_off_instance.assert_called_once_with(self.instance) @mock.patch('nova.objects.Instance.get_bdms', return_value=objects.BlockDeviceMappingList()) @mock.patch('nova.objects.Instance.save') def test_prep_snapshot_based_resize_at_source_destroy_error( self, instance_save, get_bdms): """Tests that the driver.destroy fails and _error_out_instance_on_exception sets the instance.vm_state='error'. """ with test.nested( mock.patch.object( self.compute.network_api, 'get_instance_nw_info', return_value=network_model.NetworkInfo()), mock.patch.object( self.compute, '_send_resize_instance_notifications'), mock.patch.object(self.compute, '_power_off_instance'), mock.patch.object(self.compute, '_get_power_state', return_value=power_state.SHUTDOWN), mock.patch.object(self.compute, '_snapshot_for_resize'), mock.patch.object(self.compute, '_get_instance_block_device_info'), mock.patch.object(self.compute.driver, 'destroy', side_effect=test.TestingException), ) as ( get_instance_nw_info, _send_resize_instance_notifications, _power_off_instance, _get_power_state, _snapshot_for_resize, _get_instance_block_device_info, destroy, ): self.assertRaises( test.TestingException, self.compute._prep_snapshot_based_resize_at_source, self.context, self.instance, self.migration) destroy.assert_called_once_with( self.context, self.instance, get_instance_nw_info.return_value, block_device_info=_get_instance_block_device_info.return_value, destroy_disks=False) instance_save.assert_called_once_with() self.assertEqual(vm_states.ERROR, self.instance.vm_state) @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.objects.Instance.save') def test_prep_snapshot_based_resize_at_source_general_error_handling( self, instance_save, add_fault): """Tests the general error handling and allocation rollback code when _prep_snapshot_based_resize_at_source raises an exception. """ ex1 = exception.InstancePowerOffFailure(reason='testing') with mock.patch.object( self.compute, '_prep_snapshot_based_resize_at_source', side_effect=ex1) as _prep_snapshot_based_resize_at_source: self.instance.task_state = task_states.RESIZE_MIGRATING ex2 = self.assertRaises( messaging.ExpectedException, self.compute.prep_snapshot_based_resize_at_source, self.context, self.instance, self.migration, snapshot_id=uuids.snapshot_id) # The InstancePowerOffFailure should be wrapped in the # ExpectedException. wrapped_exc = ex2.exc_info[1] self.assertIn('Failed to power off instance: testing', str(wrapped_exc)) # Assert the non-decorator mock calls. _prep_snapshot_based_resize_at_source.assert_called_once_with( self.context, self.instance, self.migration, snapshot_id=uuids.snapshot_id) # Assert wrap_instance_fault is called. add_fault.assert_called_once_with( self.context, self.instance, wrapped_exc, mock.ANY) # Assert wrap_exception is called. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) # Assert errors_out_migration is called. self.assertEqual('error', self.migration.status) self.migration.save.assert_called_once_with() # Assert reverts_task_state is called. self.assertIsNone(self.instance.task_state) instance_save.assert_called_once_with() @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.objects.Instance.save') def test_finish_snapshot_based_resize_at_dest_outer_error( self, instance_save, add_fault): """Tests the error handling on the finish_snapshot_based_resize_at_dest method. """ self.instance.task_state = task_states.RESIZE_MIGRATED with mock.patch.object( self.compute, '_finish_snapshot_based_resize_at_dest', side_effect=test.TestingException('oops')) as _finish: ex = self.assertRaises( test.TestingException, self.compute.finish_snapshot_based_resize_at_dest, self.context, self.instance, self.migration, uuids.snapshot_id) # Assert the non-decorator mock calls. _finish.assert_called_once_with( self.context, self.instance, self.migration, uuids.snapshot_id) # Assert _error_out_instance_on_exception is called. self.assertEqual(vm_states.ERROR, self.instance.vm_state) # Assert wrap_instance_fault is called. add_fault.assert_called_once_with( self.context, self.instance, ex, mock.ANY) # Assert wrap_exception is called. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) # Assert errors_out_migration is called. self.assertEqual('error', self.migration.status) self.migration.save.assert_called_once_with() # Assert reverts_task_state is called. self.assertIsNone(self.instance.task_state) # Instance.save is called twice: # 1. _error_out_instance_on_exception # 2. reverts_task_state self.assertEqual(2, instance_save.call_count) @mock.patch('nova.objects.Instance.get_bdms') @mock.patch('nova.objects.Instance.apply_migration_context') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager.' '_send_finish_resize_notifications') @mock.patch('nova.compute.manager.ComputeManager.' '_finish_snapshot_based_resize_at_dest_spawn') @mock.patch('nova.objects.ImageMeta.from_image_ref') @mock.patch('nova.compute.utils.delete_image') def _test_finish_snapshot_based_resize_at_dest( self, delete_image, from_image_ref, _finish_spawn, notify, inst_save, apply_migration_context, get_bdms, snapshot_id=None): """Happy path test for finish_snapshot_based_resize_at_dest.""" # Setup the fake instance. self.instance.task_state = task_states.RESIZE_MIGRATED nwinfo = network_model.NetworkInfo([ network_model.VIF(id=uuids.port_id)]) self.instance.info_cache = objects.InstanceInfoCache( network_info=nwinfo) self.instance.new_flavor = fake_flavor.fake_flavor_obj(self.context) old_flavor = self.instance.flavor # Mock out ImageMeta. if snapshot_id: from_image_ref.return_value = objects.ImageMeta() # Setup the fake migration. self.migration.migration_type = 'resize' self.migration.dest_compute = uuids.dest self.migration.dest_node = uuids.dest with mock.patch.object(self.compute, 'network_api') as network_api: network_api.get_instance_nw_info.return_value = nwinfo # Run that big beautiful code! self.compute.finish_snapshot_based_resize_at_dest( self.context, self.instance, self.migration, snapshot_id) # Check the changes to the instance and migration object. self.assertEqual(vm_states.RESIZED, self.instance.vm_state) self.assertIsNone(self.instance.task_state) self.assertIs(self.instance.flavor, self.instance.new_flavor) self.assertIs(self.instance.old_flavor, old_flavor) self.assertEqual(self.migration.dest_compute, self.instance.host) self.assertEqual(self.migration.dest_node, self.instance.node) self.assertEqual('finished', self.migration.status) # Assert the mock calls. if snapshot_id: from_image_ref.assert_called_once_with( self.context, self.compute.image_api, snapshot_id) delete_image.assert_called_once_with( self.context, self.instance, self.compute.image_api, snapshot_id) else: from_image_ref.assert_not_called() delete_image.assert_not_called() # The instance migration context was applied and changes were saved # to the instance twice. apply_migration_context.assert_called_once_with() inst_save.assert_has_calls([ mock.call(expected_task_state=task_states.RESIZE_MIGRATED), mock.call(expected_task_state=task_states.RESIZE_FINISH)]) self.migration.save.assert_called_once_with() # Start and end notifications were sent. notify.assert_has_calls([ mock.call(self.context, self.instance, get_bdms.return_value, nwinfo, fields.NotificationPhase.START), mock.call(self.context, self.instance, get_bdms.return_value, nwinfo, fields.NotificationPhase.END)]) # Volumes and networking were setup prior to calling driver spawn. spawn_image_meta = from_image_ref.return_value \ if snapshot_id else test.MatchType(objects.ImageMeta) _finish_spawn.assert_called_once_with( self.context, self.instance, self.migration, spawn_image_meta, get_bdms.return_value) def test_finish_snapshot_based_resize_at_dest_image_backed(self): """Happy path test for finish_snapshot_based_resize_at_dest with an image-backed server where snapshot_id is provided. """ self._test_finish_snapshot_based_resize_at_dest( snapshot_id=uuids.snapshot_id) def test_finish_snapshot_based_resize_at_dest_volume_backed(self): """Happy path test for finish_snapshot_based_resize_at_dest with a volume-backed server where snapshot_id is None. """ self._test_finish_snapshot_based_resize_at_dest(snapshot_id=None) @mock.patch('nova.compute.manager.ComputeManager._prep_block_device') @mock.patch('nova.compute.manager.ComputeManager.' '_remove_volume_connection') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_consumer') def _test_finish_snapshot_based_resize_at_dest_spawn_fails( self, get_allocs, remove_volume_connection, _prep_block_device, volume_backed=False): """Tests _finish_snapshot_based_resize_at_dest_spawn where spawn fails. """ nwinfo = network_model.NetworkInfo([ network_model.VIF(id=uuids.port_id)]) self.instance.system_metadata['old_vm_state'] = vm_states.STOPPED # Mock out BDMs. if volume_backed: # Single remote volume BDM. bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping( source_type='volume', destination_type='volume', volume_id=uuids.volume_id, boot_index=0)]) else: # Single local image BDM. bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping( source_type='image', destination_type='local', image_id=uuids.image_id, boot_index=0)]) self.migration.migration_type = 'migration' self.migration.dest_compute = uuids.dest self.migration.source_compute = uuids.source image_meta = self.instance.image_meta # Stub out migrate_instance_start so we can assert how it is called. def fake_migrate_instance_start(context, instance, migration): # Make sure the migration.dest_compute was temporarily changed # to the source_compute value. self.assertEqual(uuids.source, migration.dest_compute) with test.nested( mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.compute.driver, 'spawn', side_effect=test.TestingException('spawn fail')), ) as ( network_api, spawn, ): network_api.get_instance_nw_info.return_value = nwinfo network_api.migrate_instance_start.side_effect = \ fake_migrate_instance_start # Run that big beautiful code! self.assertRaises( test.TestingException, self.compute._finish_snapshot_based_resize_at_dest_spawn, self.context, self.instance, self.migration, image_meta, bdms) # Assert the mock calls. # Volumes and networking were setup prior to calling driver spawn. _prep_block_device.assert_called_once_with( self.context, self.instance, bdms) get_allocs.assert_called_once_with(self.context, self.instance.uuid) network_api.migrate_instance_finish.assert_called_once_with( self.context, self.instance, self.migration, provider_mappings=None) spawn.assert_called_once_with( self.context, self.instance, image_meta, injected_files=[], admin_password=None, allocations=get_allocs.return_value, network_info=nwinfo, block_device_info=_prep_block_device.return_value, power_on=False) # Port bindings were rolled back to the source host. network_api.migrate_instance_start.assert_called_once_with( self.context, self.instance, self.migration) if volume_backed: # Volume connections were deleted. remove_volume_connection.assert_called_once_with( self.context, bdms[0], self.instance, delete_attachment=True) else: remove_volume_connection.assert_not_called() def test_finish_snapshot_based_resize_at_dest_spawn_fails_image_back(self): """Tests _finish_snapshot_based_resize_at_dest_spawn failing with an image-backed server. """ self._test_finish_snapshot_based_resize_at_dest_spawn_fails( volume_backed=False) def test_finish_snapshot_based_resize_at_dest_spawn_fails_vol_backed(self): """Tests _finish_snapshot_based_resize_at_dest_spawn failing with a volume-backed server. """ self._test_finish_snapshot_based_resize_at_dest_spawn_fails( volume_backed=True) @mock.patch('nova.compute.manager.ComputeManager._prep_block_device') @mock.patch('nova.compute.manager.ComputeManager.' '_remove_volume_connection') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_consumer') def test_finish_snapshot_based_resize_at_dest_spawn_fail_graceful_rollback( self, get_allocs, remove_volume_connection, _prep_block_device): """Tests that the cleanup except block is graceful in that one failure does not prevent trying to cleanup the other resources. """ nwinfo = network_model.NetworkInfo([ network_model.VIF(id=uuids.port_id)]) self.instance.system_metadata['old_vm_state'] = vm_states.STOPPED # Three BDMs: two volume (one of which will fail rollback) and a local. bdms = objects.BlockDeviceMappingList(objects=[ # First volume BDM which fails rollback. objects.BlockDeviceMapping( destination_type='volume', volume_id=uuids.bad_volume), # Second volume BDM is rolled back. objects.BlockDeviceMapping( destination_type='volume', volume_id=uuids.good_volume), # Third BDM is a local image BDM so we do not try to roll it back. objects.BlockDeviceMapping( destination_type='local', image_id=uuids.image_id) ]) self.migration.migration_type = 'migration' self.migration.dest_compute = uuids.dest self.migration.source_compute = uuids.source image_meta = self.instance.image_meta with test.nested( mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.compute.driver, 'spawn', side_effect=test.TestingException( 'spawn fail')), ) as ( network_api, spawn, ): network_api.get_instance_nw_info.return_value = nwinfo # Mock migrate_instance_start to fail on rollback. network_api.migrate_instance_start.side_effect = \ exception.PortNotFound(port_id=uuids.port_id) # Mock remove_volume_connection to fail on the first call. remove_volume_connection.side_effect = [ exception.CinderConnectionFailed(reason='gremlins'), None] # Run that big beautiful code! self.assertRaises( test.TestingException, self.compute._finish_snapshot_based_resize_at_dest_spawn, self.context, self.instance, self.migration, image_meta, bdms) # Assert the mock calls. # Volumes and networking were setup prior to calling driver spawn. _prep_block_device.assert_called_once_with( self.context, self.instance, bdms) get_allocs.assert_called_once_with(self.context, self.instance.uuid) network_api.migrate_instance_finish.assert_called_once_with( self.context, self.instance, self.migration, provider_mappings=None) spawn.assert_called_once_with( self.context, self.instance, image_meta, injected_files=[], admin_password=None, allocations=get_allocs.return_value, network_info=nwinfo, block_device_info=_prep_block_device.return_value, power_on=False) # Port bindings were rolled back to the source host. network_api.migrate_instance_start.assert_called_once_with( self.context, self.instance, self.migration) # Volume connections were deleted. remove_volume_connection.assert_has_calls([ mock.call(self.context, bdms[0], self.instance, delete_attachment=True), mock.call(self.context, bdms[1], self.instance, delete_attachment=True)]) # Assert the expected errors to get logged. self.assertIn('Failed to activate port bindings on the source', self.stdlog.logger.output) self.assertIn('Failed to remove volume connection', self.stdlog.logger.output) @mock.patch('nova.compute.manager.ComputeManager._prep_block_device') @mock.patch('nova.scheduler.client.report.SchedulerReportClient.' 'get_allocations_for_consumer') def test_finish_snapshot_based_resize_at_dest_spawn( self, get_allocs, _prep_block_device): """Happy path test for test_finish_snapshot_based_resize_at_dest_spawn. """ nwinfo = network_model.NetworkInfo([ network_model.VIF(id=uuids.port_id)]) self.instance.system_metadata['old_vm_state'] = vm_states.ACTIVE self.migration.migration_type = 'migration' self.migration.dest_compute = uuids.dest self.migration.source_compute = uuids.source image_meta = self.instance.image_meta bdms = objects.BlockDeviceMappingList() with test.nested( mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.compute.driver, 'spawn') ) as ( network_api, spawn, ): network_api.get_instance_nw_info.return_value = nwinfo # Run that big beautiful code! self.compute._finish_snapshot_based_resize_at_dest_spawn( self.context, self.instance, self.migration, image_meta, bdms) # Assert the mock calls. _prep_block_device.assert_called_once_with( self.context, self.instance, bdms) get_allocs.assert_called_once_with(self.context, self.instance.uuid) network_api.migrate_instance_finish.assert_called_once_with( self.context, self.instance, self.migration, provider_mappings=None) spawn.assert_called_once_with( self.context, self.instance, image_meta, injected_files=[], admin_password=None, allocations=get_allocs.return_value, network_info=nwinfo, block_device_info=_prep_block_device.return_value, power_on=True) @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager.' '_delete_allocation_after_move') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_confirm_snapshot_based_resize_at_source_error_handling( self, mock_add_fault, mock_delete_allocs, mock_inst_save): """Tests the error handling in confirm_snapshot_based_resize_at_source when a failure occurs. """ error = test.TestingException('oops') with mock.patch.object( self.compute, '_confirm_snapshot_based_resize_at_source', side_effect=error) as confirm_mock: self.assertRaises( test.TestingException, self.compute.confirm_snapshot_based_resize_at_source, self.context, self.instance, self.migration) confirm_mock.assert_called_once_with( self.context, self.instance, self.migration) self.assertIn('Confirm resize failed on source host', self.stdlog.logger.output) # _error_out_instance_on_exception should set the instance to ERROR self.assertEqual(vm_states.ERROR, self.instance.vm_state) mock_inst_save.assert_called() mock_delete_allocs.assert_called_once_with( self.context, self.instance, self.migration) # wrap_instance_fault should record a fault mock_add_fault.assert_called_once_with( self.context, self.instance, error, test.MatchType(tuple)) # errors_out_migration should set the migration status to 'error' self.assertEqual('error', self.migration.status) self.migration.save.assert_called_once_with() # Assert wrap_exception is called. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager.' '_confirm_snapshot_based_resize_at_source') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_confirm_snapshot_based_resize_at_source_delete_alloc_fails( self, mock_add_fault, mock_confirm, mock_inst_save): """Tests the error handling in confirm_snapshot_based_resize_at_source when _delete_allocation_after_move fails. """ error = exception.AllocationDeleteFailed( consumer_uuid=self.migration.uuid, error='placement down') with mock.patch.object( self.compute, '_delete_allocation_after_move', side_effect=error) as mock_delete_allocs: self.assertRaises( exception.AllocationDeleteFailed, self.compute.confirm_snapshot_based_resize_at_source, self.context, self.instance, self.migration) mock_confirm.assert_called_once_with( self.context, self.instance, self.migration) # _error_out_instance_on_exception should set the instance to ERROR self.assertEqual(vm_states.ERROR, self.instance.vm_state) mock_inst_save.assert_called() mock_delete_allocs.assert_called_once_with( self.context, self.instance, self.migration) # wrap_instance_fault should record a fault mock_add_fault.assert_called_once_with( self.context, self.instance, error, test.MatchType(tuple)) # errors_out_migration should set the migration status to 'error' self.assertEqual('error', self.migration.status) self.migration.save.assert_called_once_with() # Assert wrap_exception is called. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) @mock.patch('nova.objects.Instance.get_bdms') @mock.patch('nova.compute.manager.ComputeManager.' '_delete_allocation_after_move') @mock.patch('nova.compute.manager.ComputeManager.' '_confirm_snapshot_based_resize_delete_port_bindings') @mock.patch('nova.compute.manager.ComputeManager.' '_delete_volume_attachments') def test_confirm_snapshot_based_resize_at_source( self, mock_delete_vols, mock_delete_bindings, mock_delete_allocs, mock_get_bdms): """Happy path test for confirm_snapshot_based_resize_at_source.""" self.instance.old_flavor = objects.Flavor() with test.nested( mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.compute.driver, 'cleanup'), mock.patch.object(self.compute.rt, 'drop_move_claim_at_source') ) as ( mock_network_api, mock_cleanup, mock_drop_claim, ): # Run the code. self.compute.confirm_snapshot_based_resize_at_source( self.context, self.instance, self.migration) # Assert the mocks. mock_network_api.get_instance_nw_info.assert_called_once_with( self.context, self.instance) # The guest was cleaned up. mock_cleanup.assert_called_once_with( self.context, self.instance, mock_network_api.get_instance_nw_info.return_value, block_device_info=None, destroy_disks=True, destroy_vifs=False) # Ports and volumes were cleaned up. mock_delete_bindings.assert_called_once_with( self.context, self.instance) mock_delete_vols.assert_called_once_with( self.context, mock_get_bdms.return_value) # Move claim and migration context were dropped. mock_drop_claim.assert_called_once_with( self.context, self.instance, self.migration) mock_delete_allocs.assert_called_once_with( self.context, self.instance, self.migration) def test_confirm_snapshot_based_resize_delete_port_bindings(self): """Happy path test for _confirm_snapshot_based_resize_delete_port_bindings. """ with mock.patch.object( self.compute.network_api, 'cleanup_instance_network_on_host') as cleanup_networks: self.compute._confirm_snapshot_based_resize_delete_port_bindings( self.context, self.instance) cleanup_networks.assert_called_once_with( self.context, self.instance, self.compute.host) def test_confirm_snapshot_based_resize_delete_port_bindings_errors(self): """Tests error handling for _confirm_snapshot_based_resize_delete_port_bindings. """ # PortBindingDeletionFailed will be caught and logged. with mock.patch.object( self.compute.network_api, 'cleanup_instance_network_on_host', side_effect=exception.PortBindingDeletionFailed( port_id=uuids.port_id, host=self.compute.host) ) as cleanup_networks: self.compute._confirm_snapshot_based_resize_delete_port_bindings( self.context, self.instance) cleanup_networks.assert_called_once_with( self.context, self.instance, self.compute.host) self.assertIn('Failed to delete port bindings from source host', self.stdlog.logger.output) # Anything else is re-raised. func = self.compute._confirm_snapshot_based_resize_delete_port_bindings with mock.patch.object( self.compute.network_api, 'cleanup_instance_network_on_host', side_effect=test.TestingException('neutron down') ) as cleanup_networks: self.assertRaises(test.TestingException, func, self.context, self.instance) cleanup_networks.assert_called_once_with( self.context, self.instance, self.compute.host) def test_delete_volume_attachments(self): """Happy path test for _delete_volume_attachments.""" bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(attachment_id=uuids.attachment1), objects.BlockDeviceMapping(attachment_id=None), # skipped objects.BlockDeviceMapping(attachment_id=uuids.attachment2) ]) with mock.patch.object(self.compute.volume_api, 'attachment_delete') as attachment_delete: self.compute._delete_volume_attachments(self.context, bdms) self.assertEqual(2, attachment_delete.call_count) attachment_delete.assert_has_calls([ mock.call(self.context, uuids.attachment1), mock.call(self.context, uuids.attachment2)]) def test_delete_volume_attachments_errors(self): """Tests error handling for _delete_volume_attachments.""" bdms = objects.BlockDeviceMappingList(objects=[ objects.BlockDeviceMapping(attachment_id=uuids.attachment1, instance_uuid=self.instance.uuid), objects.BlockDeviceMapping(attachment_id=uuids.attachment2) ]) # First attachment_delete call fails and is logged, second is OK. errors = [test.TestingException('cinder down'), None] with mock.patch.object(self.compute.volume_api, 'attachment_delete', side_effect=errors) as attachment_delete: self.compute._delete_volume_attachments(self.context, bdms) self.assertEqual(2, attachment_delete.call_count) attachment_delete.assert_has_calls([ mock.call(self.context, uuids.attachment1), mock.call(self.context, uuids.attachment2)]) self.assertIn('Failed to delete volume attachment with ID %s' % uuids.attachment1, self.stdlog.logger.output) @mock.patch('nova.compute.utils.notify_usage_exists') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') @mock.patch('nova.compute.manager.InstanceEvents.' 'clear_events_for_instance') def test_revert_snapshot_based_resize_at_dest_error_handling( self, mock_clear_events, mock_add_fault, mock_inst_save, mock_notify_usage): """Tests error handling in revert_snapshot_based_resize_at_dest when a failure occurs. """ self.instance.task_state = task_states.RESIZE_REVERTING error = test.TestingException('oops') with mock.patch.object( self.compute, '_revert_snapshot_based_resize_at_dest', side_effect=error) as mock_revert: self.assertRaises( test.TestingException, self.compute.revert_snapshot_based_resize_at_dest, self.context, self.instance, self.migration) mock_notify_usage.assert_called_once_with( self.compute.notifier, self.context, self.instance, self.compute.host, current_period=True) mock_revert.assert_called_once_with( self.context, self.instance, self.migration) mock_inst_save.assert_called() # _error_out_instance_on_exception sets the instance to ERROR. self.assertEqual(vm_states.ERROR, self.instance.vm_state) # reverts_task_state will reset the task_state to None. self.assertIsNone(self.instance.task_state) # Ensure wrap_instance_fault was called. mock_add_fault.assert_called_once_with( self.context, self.instance, error, test.MatchType(tuple)) # errors_out_migration should mark the migration as 'error' status self.assertEqual('error', self.migration.status) self.migration.save.assert_called_once_with() # Assert wrap_exception is called. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) # clear_events_for_instance should not have been called. mock_clear_events.assert_not_called() @mock.patch('nova.compute.utils.notify_usage_exists', new=mock.Mock()) @mock.patch('nova.compute.manager.ComputeManager.' '_revert_snapshot_based_resize_at_dest') def test_revert_snapshot_based_resize_at_dest_post_error_log(self, revert): """Tests when _revert_snapshot_based_resize_at_dest is OK but post-processing cleanup fails and is just logged. """ # First test _delete_scheduler_instance_info failing. with mock.patch.object( self.compute, '_delete_scheduler_instance_info', side_effect=( test.TestingException('scheduler'), None)) as mock_del: self.compute.revert_snapshot_based_resize_at_dest( self.context, self.instance, self.migration) revert.assert_called_once() mock_del.assert_called_once_with(self.context, self.instance.uuid) self.assertIn('revert_snapshot_based_resize_at_dest failed during ' 'post-processing. Error: scheduler', self.stdlog.logger.output) revert.reset_mock() mock_del.reset_mock() # Now test clear_events_for_instance failing. with mock.patch.object( self.compute.instance_events, 'clear_events_for_instance', side_effect=test.TestingException( 'events')) as mock_events: self.compute.revert_snapshot_based_resize_at_dest( self.context, self.instance, self.migration) revert.assert_called_once() mock_del.assert_called_once_with(self.context, self.instance.uuid) mock_events.assert_called_once_with(self.instance) self.assertIn('revert_snapshot_based_resize_at_dest failed during ' 'post-processing. Error: events', self.stdlog.logger.output) # Assert _error_out_instance_on_exception wasn't tripped somehow. self.assertNotEqual(vm_states.ERROR, self.instance.vm_state) @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') def test_revert_snapshot_based_resize_at_dest(self, mock_get_bdms): """Happy path test for _revert_snapshot_based_resize_at_dest""" # Setup more mocks. def stub_migrate_instance_start(ctxt, instance, migration): # The migration.dest_compute should have been mutated to point # at the source compute. self.assertEqual(migration.source_compute, migration.dest_compute) with test.nested( mock.patch.object(self.compute, 'network_api'), mock.patch.object(self.compute, '_get_instance_block_device_info'), mock.patch.object(self.compute.driver, 'destroy'), mock.patch.object(self.compute, '_delete_volume_attachments'), mock.patch.object(self.compute.rt, 'drop_move_claim_at_dest') ) as ( mock_network_api, mock_get_bdi, mock_destroy, mock_delete_attachments, mock_drop_claim ): mock_network_api.migrate_instance_start.side_effect = \ stub_migrate_instance_start # Raise PortBindingDeletionFailed to make sure it's caught and # logged but not fatal. mock_network_api.cleanup_instance_network_on_host.side_effect = \ exception.PortBindingDeletionFailed(port_id=uuids.port_id, host=self.compute.host) # Run the code. self.compute._revert_snapshot_based_resize_at_dest( self.context, self.instance, self.migration) # Assert the calls. mock_network_api.get_instance_nw_info.assert_called_once_with( self.context, self.instance) mock_get_bdi.assert_called_once_with( self.context, self.instance, bdms=mock_get_bdms.return_value) mock_destroy.assert_called_once_with( self.context, self.instance, mock_network_api.get_instance_nw_info.return_value, block_device_info=mock_get_bdi.return_value) mock_network_api.migrate_instance_start.assert_called_once_with( self.context, self.instance, self.migration) mock_network_api.cleanup_instance_network_on_host.\ assert_called_once_with( self.context, self.instance, self.compute.host) # Assert that even though setup_networks_on_host raised # PortBindingDeletionFailed it was handled and logged. self.assertIn('Failed to delete port bindings from target host.', self.stdlog.logger.output) mock_delete_attachments.assert_called_once_with( self.context, mock_get_bdms.return_value) mock_drop_claim.assert_called_once_with( self.context, self.instance, self.migration) @mock.patch('nova.compute.manager.ComputeManager.' '_finish_revert_snapshot_based_resize_at_source') @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager.' '_update_scheduler_instance_info') @mock.patch('nova.compute.utils.add_instance_fault_from_exc') def test_finish_revert_snapshot_based_resize_at_source_error_handling( self, mock_add_fault, mock_update_sched, mock_inst_save, mock_finish_revert): """Tests error handling (context managers, decorators, post-processing) in finish_revert_snapshot_based_resize_at_source. """ self.instance.task_state = task_states.RESIZE_REVERTING # First make _finish_revert_snapshot_based_resize_at_source fail. error = test.TestingException('oops') mock_finish_revert.side_effect = error self.assertRaises( test.TestingException, self.compute.finish_revert_snapshot_based_resize_at_source, self.context, self.instance, self.migration) mock_finish_revert.assert_called_once_with( self.context, self.instance, self.migration) # _error_out_instance_on_exception should have set the instance status # to ERROR. mock_inst_save.assert_called() self.assertEqual(vm_states.ERROR, self.instance.vm_state) # We should not have updated the scheduler since we failed. mock_update_sched.assert_not_called() # reverts_task_state should have set the task_state to None. self.assertIsNone(self.instance.task_state) # errors_out_migration should have set the migration status to error. self.migration.save.assert_called_once_with() self.assertEqual('error', self.migration.status) # wrap_instance_fault should have recorded a fault. mock_add_fault.assert_called_once_with( self.context, self.instance, error, test.MatchType(tuple)) # wrap_exception should have sent an error notification. self.assertEqual(1, len(self.notifier.versioned_notifications)) self.assertEqual( 'compute.%s' % fields.NotificationAction.EXCEPTION, self.notifier.versioned_notifications[0]['event_type']) # Now run it again but _finish_revert_snapshot_based_resize_at_source # will pass and _update_scheduler_instance_info will fail but not be # fatal (just logged). mock_finish_revert.reset_mock(side_effect=True) # reset side_effect mock_update_sched.side_effect = test.TestingException('scheduler oops') self.compute.finish_revert_snapshot_based_resize_at_source( self.context, self.instance, self.migration) mock_finish_revert.assert_called_once_with( self.context, self.instance, self.migration) mock_update_sched.assert_called_once_with(self.context, self.instance) self.assertIn('finish_revert_snapshot_based_resize_at_source failed ' 'during post-processing. Error: scheduler oops', self.stdlog.logger.output) @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager._revert_allocation', side_effect=exception.AllocationMoveFailed('placement down')) @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') @mock.patch('nova.compute.manager.ComputeManager.' '_update_volume_attachments') @mock.patch('nova.network.neutron.API.migrate_instance_finish') @mock.patch('nova.compute.manager.ComputeManager.' '_get_instance_block_device_info') @mock.patch('nova.objects.Instance.drop_migration_context') @mock.patch('nova.compute.manager.ComputeManager.' '_update_instance_after_spawn') @mock.patch('nova.compute.manager.ComputeManager.' '_complete_volume_attachments') def test_finish_revert_snapshot_based_resize_at_source( self, mock_complete_attachments, mock_update_after_spawn, mock_drop_mig_context, mock_get_bdi, mock_migrate_instance_finish, mock_update_attachments, mock_get_bdms, mock_revert_allocs, mock_inst_save): """Happy path test for finish_revert_snapshot_based_resize_at_source. Also makes sure some cleanups that are handled gracefully do not raise. """ # Make _update_scheduler_instance_info a no-op. self.flags(track_instance_changes=False, group='filter_scheduler') # Configure the instance with old_vm_state = STOPPED so the guest is # created but not powered on. self.instance.system_metadata['old_vm_state'] = vm_states.STOPPED # Configure the instance with an old_flavor for the revert. old_flavor = fake_flavor.fake_flavor_obj(self.context) self.instance.old_flavor = old_flavor with test.nested( mock.patch.object(self.compute.network_api, 'get_instance_nw_info'), mock.patch.object(self.compute.driver, 'finish_revert_migration') ) as ( mock_get_nw_info, mock_driver_finish ): # Run the code. self.compute.finish_revert_snapshot_based_resize_at_source( self.context, self.instance, self.migration) # Assert the instance host/node and flavor info was updated. self.assertEqual(self.migration.source_compute, self.instance.host) self.assertEqual(self.migration.source_node, self.instance.node) self.assertIs(self.instance.flavor, old_flavor) self.assertEqual(old_flavor.id, self.instance.instance_type_id) # Assert _revert_allocation was called, raised and logged the error. mock_revert_allocs.assert_called_once_with( self.context, self.instance, self.migration) self.assertIn('Reverting allocation in placement for migration %s ' 'failed.' % self.migration.uuid, self.stdlog.logger.output) # Assert that volume attachments were updated. mock_update_attachments.assert_called_once_with( self.context, self.instance, mock_get_bdms.return_value) # Assert that port bindings were updated to point at the source host. mock_migrate_instance_finish.assert_called_once_with( self.context, self.instance, self.migration, provider_mappings=None) # Assert the driver finished reverting the migration. mock_get_bdi.assert_called_once_with( self.context, self.instance, bdms=mock_get_bdms.return_value) mock_driver_finish.assert_called_once_with( self.context, self.instance, mock_get_nw_info.return_value, self.migration, block_device_info=mock_get_bdi.return_value, power_on=False) # Assert final DB cleanup for the instance. mock_drop_mig_context.assert_called_once_with() mock_update_after_spawn.assert_called_once_with( self.instance, vm_state=vm_states.STOPPED) mock_inst_save.assert_has_calls([ mock.call(expected_task_state=[task_states.RESIZE_REVERTING])] * 2) # And finally that the volume attachments were completed. mock_complete_attachments.assert_called_once_with( self.context, mock_get_bdms.return_value) @mock.patch('nova.objects.Instance.save') @mock.patch('nova.compute.manager.ComputeManager._revert_allocation') @mock.patch('nova.objects.BlockDeviceMappingList.get_by_instance_uuid') @mock.patch('nova.compute.manager.ComputeManager.' '_update_volume_attachments') @mock.patch('nova.network.neutron.API.migrate_instance_finish') @mock.patch('nova.compute.manager.ComputeManager.' '_get_instance_block_device_info') @mock.patch('nova.objects.Instance.drop_migration_context') @mock.patch('nova.compute.manager.ComputeManager.' '_update_instance_after_spawn') @mock.patch('nova.compute.manager.ComputeManager.' '_complete_volume_attachments', side_effect=test.TestingException('vol complete failed')) def test_finish_revert_snapshot_based_resize_at_source_driver_fails( self, mock_complete_attachments, mock_update_after_spawn, mock_drop_mig_context, mock_get_bdi, mock_migrate_instance_finish, mock_update_attachments, mock_get_bdms, mock_revert_allocs, mock_inst_save): """Test for _finish_revert_snapshot_based_resize_at_source where the driver call to finish_revert_migration fails. """ self.instance.system_metadata['old_vm_state'] = vm_states.ACTIVE # Configure the instance with an old_flavor for the revert. old_flavor = fake_flavor.fake_flavor_obj(self.context) self.instance.old_flavor = old_flavor with test.nested( mock.patch.object(self.compute.network_api, 'get_instance_nw_info'), mock.patch.object(self.compute.driver, 'finish_revert_migration', side_effect=test.TestingException('driver fail')) ) as ( mock_get_nw_info, mock_driver_finish, ): # Run the code. ex = self.assertRaises( test.TestingException, self.compute._finish_revert_snapshot_based_resize_at_source, self.context, self.instance, self.migration) # Assert the driver call (note power_on=True b/c old_vm_state=active) mock_get_bdi.assert_called_once_with( self.context, self.instance, bdms=mock_get_bdms.return_value) mock_driver_finish.assert_called_once_with( self.context, self.instance, mock_get_nw_info.return_value, self.migration, block_device_info=mock_get_bdi.return_value, power_on=True) # _complete_volume_attachments is called even though the driver call # failed. mock_complete_attachments.assert_called_once_with( self.context, mock_get_bdms.return_value) # finish_revert_migration failed but _complete_volume_attachments # is still called and in this case also fails so the resulting # exception should be the one from _complete_volume_attachments # but the finish_revert_migration error should have been logged. self.assertIn('vol complete failed', str(ex)) self.assertIn('driver fail', self.stdlog.logger.output) # Assert the migration status was not updated. self.migration.save.assert_not_called() # Assert the instance host/node and flavor info was updated. self.assertEqual(self.migration.source_compute, self.instance.host) self.assertEqual(self.migration.source_node, self.instance.node) self.assertIs(self.instance.flavor, old_flavor) self.assertEqual(old_flavor.id, self.instance.instance_type_id) # Assert allocations were reverted. mock_revert_allocs.assert_called_once_with( self.context, self.instance, self.migration) # Assert that volume attachments were updated. mock_update_attachments.assert_called_once_with( self.context, self.instance, mock_get_bdms.return_value) # Assert that port bindings were updated to point at the source host. mock_migrate_instance_finish.assert_called_once_with( self.context, self.instance, self.migration, provider_mappings=None) # Assert final DB cleanup for the instance did not happen. mock_drop_mig_context.assert_not_called() mock_update_after_spawn.assert_not_called() # _finish_revert_resize_update_instance_flavor_host_node updated the # instance. mock_inst_save.assert_called_once_with( expected_task_state=[task_states.RESIZE_REVERTING]) class ComputeManagerInstanceUsageAuditTestCase(test.TestCase): def setUp(self): super(ComputeManagerInstanceUsageAuditTestCase, self).setUp() self.flags(group='glance', api_servers=['http://localhost:9292']) self.flags(instance_usage_audit=True) @mock.patch('nova.objects.TaskLog') def test_deleted_instance(self, mock_task_log): mock_task_log.get.return_value = None compute = manager.ComputeManager() admin_context = context.get_admin_context() fake_db_flavor = fake_flavor.fake_db_flavor() flavor = objects.Flavor(admin_context, **fake_db_flavor) updates = {'host': compute.host, 'flavor': flavor, 'root_gb': 0, 'ephemeral_gb': 0} # fudge beginning and ending time by a second (backwards and forwards, # respectively) so they differ from the instance's launch and # termination times when sub-seconds are truncated and fall within the # audit period one_second = datetime.timedelta(seconds=1) begin = timeutils.utcnow() - one_second instance = objects.Instance(admin_context, **updates) instance.create() instance.launched_at = timeutils.utcnow() instance.save() instance.destroy() end = timeutils.utcnow() + one_second def fake_last_completed_audit_period(): return (begin, end) self.stub_out('nova.utils.last_completed_audit_period', fake_last_completed_audit_period) compute._instance_usage_audit(admin_context) self.assertEqual(1, mock_task_log().task_items, 'the deleted test instance was not found in the audit' ' period') self.assertEqual(0, mock_task_log().errors, 'an error was encountered processing the deleted test' ' instance') @ddt.ddt class ComputeManagerSetHostEnabledTestCase(test.NoDBTestCase): def setUp(self): super(ComputeManagerSetHostEnabledTestCase, self).setUp() self.compute = manager.ComputeManager() self.context = context.RequestContext(user_id=fakes.FAKE_USER_ID, project_id=fakes.FAKE_PROJECT_ID) @ddt.data(True, False) def test_set_host_enabled(self, enabled): """Happy path test for set_host_enabled""" with mock.patch.object(self.compute, '_update_compute_provider_status') as ucpt: retval = self.compute.set_host_enabled(self.context, enabled) expected_retval = 'enabled' if enabled else 'disabled' self.assertEqual(expected_retval, retval) ucpt.assert_called_once_with(self.context, enabled) @mock.patch('nova.compute.manager.LOG.warning') def test_set_host_enabled_compute_host_not_found(self, mock_warning): """Tests _update_compute_provider_status raising ComputeHostNotFound""" error = exception.ComputeHostNotFound(host=self.compute.host) with mock.patch.object(self.compute, '_update_compute_provider_status', side_effect=error) as ucps: retval = self.compute.set_host_enabled(self.context, False) self.assertEqual('disabled', retval) ucps.assert_called_once_with(self.context, False) # A warning should have been logged for the ComputeHostNotFound error. mock_warning.assert_called_once() self.assertIn('Unable to add/remove trait COMPUTE_STATUS_DISABLED. ' 'No ComputeNode(s) found for host', mock_warning.call_args[0][0]) def test_set_host_enabled_update_provider_status_error(self): """Tests _update_compute_provider_status raising some unexpected error """ error = messaging.MessagingTimeout with test.nested( mock.patch.object(self.compute, '_update_compute_provider_status', side_effect=error), mock.patch.object(self.compute.driver, 'set_host_enabled', # The driver is not called in this case. new_callable=mock.NonCallableMock), ) as ( ucps, driver_set_host_enabled, ): self.assertRaises(error, self.compute.set_host_enabled, self.context, False) ucps.assert_called_once_with(self.context, False) @ddt.data(True, False) def test_set_host_enabled_not_implemented_error(self, enabled): """Tests the driver raising NotImplementedError""" with test.nested( mock.patch.object(self.compute, '_update_compute_provider_status'), mock.patch.object(self.compute.driver, 'set_host_enabled', side_effect=NotImplementedError), ) as ( ucps, driver_set_host_enabled, ): retval = self.compute.set_host_enabled(self.context, enabled) expected_retval = 'enabled' if enabled else 'disabled' self.assertEqual(expected_retval, retval) ucps.assert_called_once_with(self.context, enabled) driver_set_host_enabled.assert_called_once_with(enabled) def test_set_host_enabled_driver_error(self): """Tests the driver raising some unexpected error""" error = exception.HypervisorUnavailable(host=self.compute.host) with test.nested( mock.patch.object(self.compute, '_update_compute_provider_status'), mock.patch.object(self.compute.driver, 'set_host_enabled', side_effect=error), ) as ( ucps, driver_set_host_enabled, ): self.assertRaises(exception.HypervisorUnavailable, self.compute.set_host_enabled, self.context, False) ucps.assert_called_once_with(self.context, False) driver_set_host_enabled.assert_called_once_with(False) @ddt.data(True, False) def test_update_compute_provider_status(self, enabled): """Happy path test for _update_compute_provider_status""" # Fake out some fake compute nodes (ironic driver case). self.compute.rt.compute_nodes = { uuids.node1: objects.ComputeNode(uuid=uuids.node1), uuids.node2: objects.ComputeNode(uuid=uuids.node2), } with mock.patch.object(self.compute.virtapi, 'update_compute_provider_status') as ucps: self.compute._update_compute_provider_status( self.context, enabled=enabled) self.assertEqual(2, ucps.call_count) ucps.assert_has_calls([ mock.call(self.context, uuids.node1, enabled), mock.call(self.context, uuids.node2, enabled), ], any_order=True) def test_update_compute_provider_status_no_nodes(self): """Tests the case that _update_compute_provider_status will raise ComputeHostNotFound if there are no nodes in the resource tracker. """ self.assertRaises(exception.ComputeHostNotFound, self.compute._update_compute_provider_status, self.context, enabled=True) @mock.patch('nova.compute.manager.LOG.warning') def test_update_compute_provider_status_expected_errors(self, m_warn): """Tests _update_compute_provider_status handling a set of expected errors from the ComputeVirtAPI and logging a warning. """ # Setup a fake compute in the resource tracker. self.compute.rt.compute_nodes = { uuids.node: objects.ComputeNode(uuid=uuids.node) } errors = ( exception.ResourceProviderTraitRetrievalFailed(uuid=uuids.node), exception.ResourceProviderUpdateConflict( uuid=uuids.node, generation=1, error='conflict'), exception.ResourceProviderUpdateFailed( url='https://placement', error='dogs'), exception.TraitRetrievalFailed(error='cats'), ) for error in errors: with mock.patch.object( self.compute.virtapi, 'update_compute_provider_status', side_effect=error) as ucps: self.compute._update_compute_provider_status( self.context, enabled=False) ucps.assert_called_once_with(self.context, uuids.node, False) # The expected errors are logged as a warning. m_warn.assert_called_once() self.assertIn('An error occurred while updating ' 'COMPUTE_STATUS_DISABLED trait on compute node', m_warn.call_args[0][0]) m_warn.reset_mock() @mock.patch('nova.compute.manager.LOG.exception') def test_update_compute_provider_status_unexpected_error(self, m_exc): """Tests _update_compute_provider_status handling an unexpected exception from the ComputeVirtAPI and logging it. """ # Use two fake nodes here to make sure we try updating each even when # an error occurs. self.compute.rt.compute_nodes = { uuids.node1: objects.ComputeNode(uuid=uuids.node1), uuids.node2: objects.ComputeNode(uuid=uuids.node2), } with mock.patch.object( self.compute.virtapi, 'update_compute_provider_status', side_effect=(TypeError, AttributeError)) as ucps: self.compute._update_compute_provider_status( self.context, enabled=False) self.assertEqual(2, ucps.call_count) ucps.assert_has_calls([ mock.call(self.context, uuids.node1, False), mock.call(self.context, uuids.node2, False), ], any_order=True) # Each exception should have been logged. self.assertEqual(2, m_exc.call_count) self.assertIn('An error occurred while updating ' 'COMPUTE_STATUS_DISABLED trait', m_exc.call_args_list[0][0][0])