@@ -21457,6 +21457,8 @@ def test_migrate_disk_and_power_off_exception(
21457
21457
context.get_admin_context(), ins_ref, '10.0.0.2',
21458
21458
flavor_obj, None)
21459
21459
21460
+ @mock.patch('nova.virt.libvirt.driver.LibvirtDriver.'
21461
+ '_cleanup_failed_instance_base')
21460
21462
@mock.patch('nova.virt.libvirt.driver.LibvirtDriver.unplug_vifs')
21461
21463
@mock.patch('nova.virt.libvirt.utils.save_and_migrate_vtpm_dir')
21462
21464
@mock.patch('nova.virt.libvirt.driver.LibvirtDriver.'
@@ -21473,7 +21475,7 @@ def _test_migrate_disk_and_power_off(
21473
21475
self, ctxt, flavor_obj, mock_execute, mock_exists, mock_rename,
21474
21476
mock_is_shared, mock_get_host_ip, mock_destroy,
21475
21477
mock_get_disk_info, mock_vtpm, mock_unplug_vifs,
21476
- block_device_info=None, params_for_instance=None):
21478
+ mock_cleanup, block_device_info=None, params_for_instance=None):
21477
21479
"""Test for nova.virt.libvirt.driver.LivirtConnection
21478
21480
.migrate_disk_and_power_off.
21479
21481
"""
@@ -21488,6 +21490,8 @@ def _test_migrate_disk_and_power_off(
21488
21490
ctxt, instance, '10.0.0.2', flavor_obj, None,
21489
21491
block_device_info=block_device_info)
21490
21492
21493
+ mock_cleanup.assert_called_once()
21494
+ mock_cleanup.reset_mock()
21491
21495
self.assertEqual(out, disk_info_text)
21492
21496
mock_vtpm.assert_called_with(
21493
21497
instance.uuid, mock.ANY, mock.ANY, '10.0.0.2', mock.ANY, mock.ANY)
@@ -21498,6 +21502,7 @@ def _test_migrate_disk_and_power_off(
21498
21502
ctxt, instance, '10.0.0.1', flavor_obj, None,
21499
21503
block_device_info=block_device_info)
21500
21504
21505
+ mock_cleanup.assert_called_once()
21501
21506
self.assertEqual(out, disk_info_text)
21502
21507
mock_vtpm.assert_called_with(
21503
21508
instance.uuid, mock.ANY, mock.ANY, '10.0.0.1', mock.ANY, mock.ANY)
@@ -22395,8 +22400,8 @@ def test_finish_revert_migration_snap_backend_image_does_not_exist(self):
22395
22400
self.assertFalse(drvr.image_backend.remove_snap.called)
22396
22401
22397
22402
@mock.patch.object(shutil, 'rmtree')
22398
- def test_cleanup_failed_migration (self, mock_rmtree):
22399
- self.drvr._cleanup_failed_migration ('/fake/inst')
22403
+ def test_cleanup_failed_instance_base (self, mock_rmtree):
22404
+ self.drvr._cleanup_failed_instance_base ('/fake/inst')
22400
22405
mock_rmtree.assert_called_once_with('/fake/inst')
22401
22406
22402
22407
@mock.patch.object(libvirt_driver.LibvirtDriver, '_cleanup_resize')
0 commit comments