@@ -21511,6 +21511,8 @@ def test_migrate_disk_and_power_off_exception(
21511
21511
context.get_admin_context(), ins_ref, '10.0.0.2',
21512
21512
flavor_obj, None)
21513
21513
21514
+ @mock.patch('nova.virt.libvirt.driver.LibvirtDriver.'
21515
+ '_cleanup_failed_instance_base')
21514
21516
@mock.patch('nova.virt.libvirt.driver.LibvirtDriver.unplug_vifs')
21515
21517
@mock.patch('nova.virt.libvirt.utils.save_and_migrate_vtpm_dir')
21516
21518
@mock.patch('nova.virt.libvirt.driver.LibvirtDriver.'
@@ -21527,7 +21529,7 @@ def _test_migrate_disk_and_power_off(
21527
21529
self, ctxt, flavor_obj, mock_execute, mock_exists, mock_rename,
21528
21530
mock_is_shared, mock_get_host_ip, mock_destroy,
21529
21531
mock_get_disk_info, mock_vtpm, mock_unplug_vifs,
21530
- block_device_info=None, params_for_instance=None):
21532
+ mock_cleanup, block_device_info=None, params_for_instance=None):
21531
21533
"""Test for nova.virt.libvirt.driver.LivirtConnection
21532
21534
.migrate_disk_and_power_off.
21533
21535
"""
@@ -21542,6 +21544,8 @@ def _test_migrate_disk_and_power_off(
21542
21544
ctxt, instance, '10.0.0.2', flavor_obj, None,
21543
21545
block_device_info=block_device_info)
21544
21546
21547
+ mock_cleanup.assert_called_once()
21548
+ mock_cleanup.reset_mock()
21545
21549
self.assertEqual(out, disk_info_text)
21546
21550
mock_vtpm.assert_called_with(
21547
21551
instance.uuid, mock.ANY, mock.ANY, '10.0.0.2', mock.ANY, mock.ANY)
@@ -21552,6 +21556,7 @@ def _test_migrate_disk_and_power_off(
21552
21556
ctxt, instance, '10.0.0.1', flavor_obj, None,
21553
21557
block_device_info=block_device_info)
21554
21558
21559
+ mock_cleanup.assert_called_once()
21555
21560
self.assertEqual(out, disk_info_text)
21556
21561
mock_vtpm.assert_called_with(
21557
21562
instance.uuid, mock.ANY, mock.ANY, '10.0.0.1', mock.ANY, mock.ANY)
@@ -22449,8 +22454,8 @@ def test_finish_revert_migration_snap_backend_image_does_not_exist(self):
22449
22454
self.assertFalse(drvr.image_backend.remove_snap.called)
22450
22455
22451
22456
@mock.patch.object(shutil, 'rmtree')
22452
- def test_cleanup_failed_migration (self, mock_rmtree):
22453
- self.drvr._cleanup_failed_migration ('/fake/inst')
22457
+ def test_cleanup_failed_instance_base (self, mock_rmtree):
22458
+ self.drvr._cleanup_failed_instance_base ('/fake/inst')
22454
22459
mock_rmtree.assert_called_once_with('/fake/inst')
22455
22460
22456
22461
@mock.patch.object(libvirt_driver.LibvirtDriver, '_cleanup_resize')
0 commit comments