def test_return_handler(self):
"""Hard drive source setup result propagates back."""
task = self.module.set_up_with_tasks()[0]
task.get_result = Mock(return_value=SetupHardDriveResult(
iso_mount_location, "iso_name.iso"))
task.succeeded_signal.emit()
assert self.module.install_tree_path == iso_mount_location
assert self.module._iso_name == "iso_name.iso"
def test_success_find_iso(self, find_and_mount_iso_image_mock,
find_and_mount_device_mock):
"""Hard drive source setup iso found."""
task = _create_setup_task()
result = task.run()
find_and_mount_device_mock.assert_called_once_with(
device_spec, device_mount_location)
find_and_mount_iso_image_mock.assert_called_once_with(
device_mount_location + path_on_device, iso_mount_location)
assert result == SetupHardDriveResult(iso_mount_location, "skynet.iso")
def test_return_handler_without_iso(self):
"""Hard drive source setup result propagates back when no ISO is involved.
This is happening when installation tree is used instead of ISO image.
"""
task = self.module.set_up_with_tasks()[0]
task.get_result = Mock(
return_value=SetupHardDriveResult(iso_mount_location, ""))
task.succeeded_signal.emit()
assert self.module.install_tree_path == iso_mount_location
assert self.module._iso_name == ""
def test_success_find_dir(self, verify_valid_repository_mock,
find_and_mount_iso_image_mock,
find_and_mount_device_mock):
"""Hard drive source setup dir found."""
task = _create_setup_task()
result = task.run()
find_and_mount_device_mock.assert_called_once_with(
device_spec, device_mount_location)
find_and_mount_iso_image_mock.assert_called_once_with(
device_mount_location + path_on_device, iso_mount_location)
verify_valid_repository_mock.assert_called_once_with(
device_mount_location + path_on_device)
assert result == SetupHardDriveResult(
device_mount_location + path_on_device, "")
def test_ready_state(self, ismount):
"""Hard drive source ready state for set up."""
ismount.return_value = False
assert self.module.get_state() == SourceState.UNREADY
ismount.assert_called_once_with(self.module._device_mount)
ismount.reset_mock()
ismount.return_value = True
task = self.module.set_up_with_tasks()[0]
task.get_result = Mock(return_value=SetupHardDriveResult("/my/path", ""))
task.succeeded_signal.emit()
assert self.module.get_state() == SourceState.READY
ismount.assert_called_once_with(self.module._device_mount)
def success_find_dir_test(self, verify_valid_installtree_mock,
find_first_iso_image_mock,
find_and_mount_device_mock):
"""Hard drive source setup dir found."""
task = _create_setup_task()
result = task.run()
find_and_mount_device_mock.assert_called_once_with(
device_spec, device_mount_location)
find_first_iso_image_mock.assert_called_once_with(
device_mount_location + path_on_device)
verify_valid_installtree_mock.assert_called_once_with(
device_mount_location + path_on_device)
self.assertEqual(
result,
SetupHardDriveResult(device_mount_location + path_on_device, ""))
def success_find_iso_test(self, mount_mock, find_first_iso_image_mock,
find_and_mount_device_mock):
"""Hard drive source setup iso found."""
task = _create_setup_task()
result = task.run()
find_and_mount_device_mock.assert_called_once_with(
device_spec, device_mount_location)
find_first_iso_image_mock.assert_called_once_with(
device_mount_location + path_on_device)
mount_mock.assert_called_once_with(device_mount_location +
path_on_device + "/skynet.iso",
iso_mount_location,
fstype="iso9660",
options="ro")
self.assertEqual(
result, SetupHardDriveResult(iso_mount_location, "skynet.iso"))
