def setup_install_source_task_missing_image_test(self, proxy_getter):
"""Test Live OS setup installation source task missing image error."""
device_tree = Mock()
proxy_getter.return_value = device_tree
device_tree.ResolveDevice = Mock()
device_tree.ResolveDevice.return_value = ""
with self.assertRaises(SourceSetupError):
SetupInstallationSourceTask("/path/to/base/image",
"/path/to/mount/source/image").run()
def setup_install_source_task_failed_to_mount_test(self, proxy_getter, os_stat, stat, mount):
"""Test Live OS setup installation source task mount error."""
device_tree = Mock()
proxy_getter.return_value = device_tree
device_tree.ResolveDevice = Mock()
device_tree.ResolveDevice.return_value = "resolvedDeviceName"
device = DeviceData()
device.path = "/resolved/path/to/base/image"
device_tree.GetDeviceData = Mock()
device_tree.GetDeviceData.return_value = get_native(DeviceData.to_structure(device))
mount.return_value = -20
with self.assertRaises(SourceSetupError):
SetupInstallationSourceTask(
"/path/to/base/image",
"/path/to/mount/source/image"
).run()
def setup_install_source_task_test(self, proxy_getter, os_stat, stat, mount):
"""Test Live OS setup installation source task."""
device_tree = Mock()
proxy_getter.return_value = device_tree
device_tree.ResolveDevice = Mock()
device_tree.ResolveDevice.return_value = "resolvedDeviceName"
device = DeviceData()
device.path = "/resolved/path/to/base/image"
device_tree.GetDeviceData = Mock()
device_tree.GetDeviceData.return_value = get_native(DeviceData.to_structure(device))
mount.return_value = 0
SetupInstallationSourceTask(
"/path/to/base/image",
"/path/to/mount/source/image"
).run()
device_tree.ResolveDevice.assert_called_once_with("/path/to/base/image")
os_stat.assert_called_once_with("/resolved/path/to/base/image")
def setup_installation_source_with_task(self):
"""Setup installation source device."""
task = SetupInstallationSourceTask(self.image_path, INSTALL_TREE)
return self.publish_task(LIVE_OS_HANDLER.namespace, task)
def setup_installation_source_with_task(self):
"""Setup installation source device."""
return SetupInstallationSourceTask(self.image_path, INSTALL_TREE)