def test_execute_rollback(self, prep_resize_mock, sel_dest_mock, sig_mock,
az_mock, gmv_mock, cm_mock, sm_mock, cn_mock,
rc_mock, mock_ra, mock_gbf):
sel_dest_mock.return_value = self.host_lists
az_mock.return_value = 'myaz'
task = self._generate_task()
gmv_mock.return_value = 23
mock_gbf.return_value = objects.MigrationList()
mock_get_resources = \
self.mock_network_api.get_requested_resource_for_instance
mock_get_resources.return_value = ([], objects.RequestLevelParams())
# We just need this hook point to set a uuid on the
# migration before we use it for teardown
def set_migration_uuid(*a, **k):
task._migration.uuid = uuids.migration
return mock.MagicMock()
# NOTE(danms): It's odd to do this on cn_mock, but it's just because
# of when we need to have it set in the flow and where we have an easy
# place to find it via self.migration.
cn_mock.side_effect = set_migration_uuid
prep_resize_mock.side_effect = test.TestingException
task._held_allocations = mock.sentinel.allocs
self.assertRaises(test.TestingException, task.execute)
self.assertIsNotNone(task._migration)
task._migration.create.assert_called_once_with()
task._migration.save.assert_called_once_with()
self.assertEqual('error', task._migration.status)
mock_ra.assert_called_once_with(task.context, task._source_cn,
task.instance, task._migration)
mock_get_resources.assert_called_once_with(self.context,
self.instance.uuid)
def test_find_destination_with_resource_request(self, mock_setup,
mock_reset, mock_select,
mock_check, mock_call,
mock_fill_provider_mapping,
mock_update_pci_req):
resource_req = [objects.RequestGroup(requester_id=uuids.port_id)]
self.mock_get_res_req.return_value = (resource_req,
objects.RequestLevelParams())
self.instance.pci_requests = objects.InstancePCIRequests(requests=[])
self.assertEqual(("host1", "node1", fake_limits1),
self.task._find_destination())
# Make sure the request_spec was updated to include the cell
# mapping.
self.assertIsNotNone(self.fake_spec.requested_destination.cell)
# Make sure the spec was updated to include the project_id.
self.assertEqual(self.fake_spec.project_id, self.instance.project_id)
# Make sure that requested_resources are added to the request spec
self.assertEqual(resource_req,
self.task.request_spec.requested_resources)
mock_setup.assert_called_once_with(self.context, self.fake_spec)
mock_reset.assert_called_once_with()
self.ensure_network_information_mock.assert_called_once_with(
self.instance)
self.heal_reqspec_is_bfv_mock.assert_called_once_with(
self.context, self.fake_spec, self.instance)
mock_select.assert_called_once_with(self.context,
self.fake_spec,
[self.instance.uuid],
return_objects=True,
return_alternates=False)
mock_check.assert_called_once_with('host1')
mock_call.assert_called_once_with('host1', {uuids.port_id: []})
mock_fill_provider_mapping.assert_called_once_with(
self.task.request_spec, fake_selection1)
mock_update_pci_req.assert_called_once_with(self.context,
self.task.report_client,
[], {uuids.port_id: []})
def setUp(self):
super(LiveMigrationTaskTestCase, self).setUp()
self.context = nova_context.get_admin_context()
self.instance_host = "host"
self.instance_uuid = uuids.instance
self.instance_image = "image_ref"
db_instance = fake_instance.fake_db_instance(
host=self.instance_host,
uuid=self.instance_uuid,
power_state=power_state.RUNNING,
vm_state=vm_states.ACTIVE,
memory_mb=512,
image_ref=self.instance_image)
self.instance = objects.Instance._from_db_object(
self.context, objects.Instance(), db_instance)
self.instance.system_metadata = {'image_hw_disk_bus': 'scsi'}
self.instance.numa_topology = None
self.instance.pci_requests = None
self.instance.resources = None
self.destination = "destination"
self.block_migration = "bm"
self.disk_over_commit = "doc"
self.migration = objects.Migration()
self.fake_spec = objects.RequestSpec()
self._generate_task()
_p = mock.patch('nova.compute.utils.heal_reqspec_is_bfv')
self.heal_reqspec_is_bfv_mock = _p.start()
self.addCleanup(_p.stop)
_p = mock.patch('nova.objects.RequestSpec.ensure_network_information')
self.ensure_network_information_mock = _p.start()
self.addCleanup(_p.stop)
_p = mock.patch(
'nova.network.neutron.API.'
'get_requested_resource_for_instance',
return_value=([], objects.RequestLevelParams()))
self.mock_get_res_req = _p.start()
self.addCleanup(_p.stop)
def create_resource_requests(self,
context,
requested_networks,
pci_requests=None,
affinity_policy=None):
return None, [], objects.RequestLevelParams()
def _test_execute(self,
cross_cell_exec_mock,
prep_resize_mock,
sel_dest_mock,
sig_mock,
az_mock,
gmv_mock,
cm_mock,
sm_mock,
cn_mock,
rc_mock,
gbf_mock,
requested_destination=False,
same_cell=True):
sel_dest_mock.return_value = self.host_lists
az_mock.return_value = 'myaz'
gbf_mock.return_value = objects.MigrationList()
mock_get_resources = \
self.mock_network_api.get_requested_resource_for_instance
mock_get_resources.return_value = ([], objects.RequestLevelParams())
if requested_destination:
self.request_spec.requested_destination = objects.Destination(
host='target_host',
node=None,
allow_cross_cell_move=not same_cell)
self.request_spec.retry = objects.SchedulerRetries.from_dict(
self.context, self.filter_properties['retry'])
self.filter_properties.pop('retry')
self.filter_properties['requested_destination'] = (
self.request_spec.requested_destination)
task = self._generate_task()
gmv_mock.return_value = 23
# We just need this hook point to set a uuid on the
# migration before we use it for teardown
def set_migration_uuid(*a, **k):
task._migration.uuid = uuids.migration
return mock.MagicMock()
# NOTE(danms): It's odd to do this on cn_mock, but it's just because
# of when we need to have it set in the flow and where we have an easy
# place to find it via self.migration.
cn_mock.side_effect = set_migration_uuid
selection = self.host_lists[0][0]
with mock.patch.object(task,
'_is_selected_host_in_source_cell',
return_value=same_cell) as _is_source_cell_mock:
task.execute()
_is_source_cell_mock.assert_called_once_with(selection)
self.ensure_network_information_mock.assert_called_once_with(
self.instance)
self.heal_reqspec_is_bfv_mock.assert_called_once_with(
self.context, self.request_spec, self.instance)
sig_mock.assert_called_once_with(self.context, self.request_spec)
task.query_client.select_destinations.assert_called_once_with(
self.context,
self.request_spec, [self.instance.uuid],
return_objects=True,
return_alternates=True)
if same_cell:
prep_resize_mock.assert_called_once_with(
self.context,
self.instance,
self.request_spec.image,
self.flavor,
selection.service_host,
task._migration,
request_spec=self.request_spec,
filter_properties=self.filter_properties,
node=selection.nodename,
clean_shutdown=self.clean_shutdown,
host_list=[])
az_mock.assert_called_once_with(self.context, 'host1')
cross_cell_exec_mock.assert_not_called()
else:
cross_cell_exec_mock.assert_called_once_with()
az_mock.assert_not_called()
prep_resize_mock.assert_not_called()
self.assertIsNotNone(task._migration)
old_flavor = self.instance.flavor
new_flavor = self.flavor
self.assertEqual(old_flavor.id, task._migration.old_instance_type_id)
self.assertEqual(new_flavor.id, task._migration.new_instance_type_id)
self.assertEqual('pre-migrating', task._migration.status)
self.assertEqual(self.instance.uuid, task._migration.instance_uuid)
self.assertEqual(self.instance.host, task._migration.source_compute)
self.assertEqual(self.instance.node, task._migration.source_node)
if old_flavor.id != new_flavor.id:
self.assertEqual('resize', task._migration.migration_type)
else:
self.assertEqual('migration', task._migration.migration_type)
task._migration.create.assert_called_once_with()
if requested_destination:
self.assertIsNone(self.request_spec.retry)
self.assertIn('cell', self.request_spec.requested_destination)
self.assertIsNotNone(self.request_spec.requested_destination.cell)
self.assertEqual(
not same_cell,
self.request_spec.requested_destination.allow_cross_cell_move)
mock_get_resources.assert_called_once_with(self.context,
self.instance.uuid)
self.assertEqual([], self.request_spec.requested_resources)
self.assertEqual(
mock_get_resources.return_value[1],
self.request_spec.request_level_params,
)