def set_up_device_tree(num_cdroms): """Set up a mock device tree with a specified amount of CD-ROMs. Mocks FindOpticalMedia() and GetDeviceData() suitable for testing SetUpCdromSourceTask.run() :param int num_cdroms: Amount od CD-ROMs :return: mock for the device tree :rtype: unittest.mock.Mock """ devices = [] for n in range(num_cdroms): device = DeviceData() device.name = "test{}".format(n) device.path = "/dev/cdrom-test{}".format(n) devices.append(device) device_tree = Mock() device_tree.FindOpticalMedia = Mock() device_tree.FindOpticalMedia.return_value = [dev.name for dev in devices] device_tree.GetDeviceData = Mock() device_tree.GetDeviceData.side_effect = [DeviceData.to_structure(dev) for dev in devices] return device_tree
def GetDeviceData(self, name: Str) -> Structure: """Get the device data. :param name: a device name :return: a structure with device data :raise: UnknownDeviceError if the device is not found """ return DeviceData.to_structure(self.implementation.get_device_data(name))
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 test_setup_install_source_task_invalid_block_dev( self, proxy_getter, os_stat, stat_mock): """Test Live OS Source setup installation source task with invalid block device 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 = DeviceData.to_structure( device) stat_mock.S_ISBLK = Mock() stat_mock.S_ISBLK.return_value = False with pytest.raises(SourceSetupError): SetUpLiveOSSourceTask("/path/to/base/image", "/path/to/mount/source/image").run()
def test_setup_install_source_task_failed_to_mount(self, proxy_getter, os_stat, stat, mount): """Test Live OS Source setup installation source task mount error.""" device_tree = Mock() proxy_getter.return_value = device_tree device_tree.ResolveDevice = Mock() device_tree.ResolveDevice.return_value = "dev1" device = DeviceData() device.path = "/dev/dev1" device_tree.GetDeviceData = Mock() device_tree.GetDeviceData.return_value = DeviceData.to_structure( device) mount.return_value = -20 with pytest.raises(SourceSetupError) as cm: SetUpLiveOSSourceTask("/path/to/base/image", "/path/to/mount/source/image").run() assert str(cm.value) == "Failed to mount the Live OS image."
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 _find_device_by_name(name): for dev in devices: if dev.name == name: return DeviceData.to_structure(dev) return None