def test_write_configdrive_too_large(self, execute_mock, open_mock, configdrive_mock, stat_mock): device = '/dev/sda' configdrive = standby._configdrive_location() stat_mock.return_value.st_size = 65 * 1024 * 1024 self.assertRaises(errors.ConfigDriveTooLargeError, standby._write_configdrive_to_partition, configdrive, device)
def test_write_configdrive_too_large(self, execute_mock, open_mock, configdrive_mock, stat_mock): device = '/dev/sda' configdrive = standby._configdrive_location() stat_mock.return_value.st_size = 65 * 1024 * 1024 self.assertRaises(errors.ConfigDriveTooLargeError, standby._write_configdrive_to_partition, configdrive, device)
def test_write_configdrive_to_file(self, b64_mock, open_mock, gzip_mock): open_mock.return_value.__enter__ = lambda s: s open_mock.return_value.__exit__ = mock.Mock() write_mock = open_mock.return_value.write gzip_read_mock = gzip_mock.return_value.read gzip_read_mock.return_value = 'ungzipped' b64_mock.return_value = 'configdrive_data' filename = standby._configdrive_location() standby._write_configdrive_to_file('b64data', filename) open_mock.assert_called_once_with(filename, 'wb') gzip_read_mock.assert_called_once_with() write_mock.assert_called_once_with('ungzipped')
def test_write_configdrive_to_file(self, b64_mock, open_mock, gzip_mock): open_mock.return_value.__enter__ = lambda s: s open_mock.return_value.__exit__ = mock.Mock() write_mock = open_mock.return_value.write gzip_read_mock = gzip_mock.return_value.read gzip_read_mock.return_value = 'ungzipped' b64_mock.return_value = 'configdrive_data' filename = standby._configdrive_location() standby._write_configdrive_to_file('b64data', filename) open_mock.assert_called_once_with(filename, 'wb') gzip_read_mock.assert_called_once_with() write_mock.assert_called_once_with('ungzipped')
def test_write_configdrive_to_partition(self, execute_mock, open_mock, configdrive_mock, stat_mock): device = '/dev/sda' configdrive = standby._configdrive_location() script = standby._path_to_script('shell/copy_configdrive_to_disk.sh') command = ['/bin/bash', script, configdrive, device] execute_mock.return_value = ('', '') stat_mock.return_value.st_size = 5 standby._write_configdrive_to_partition(configdrive, device) execute_mock.assert_called_once_with(*command, check_exit_code=[0]) execute_mock.reset_mock() execute_mock.return_value = ('', '') execute_mock.side_effect = processutils.ProcessExecutionError self.assertRaises(errors.ConfigDriveWriteError, standby._write_configdrive_to_partition, configdrive, device) execute_mock.assert_called_once_with(*command, check_exit_code=[0])
def test_write_configdrive_to_partition(self, execute_mock, open_mock, configdrive_mock, stat_mock): device = '/dev/sda' configdrive = standby._configdrive_location() script = standby._path_to_script('shell/copy_configdrive_to_disk.sh') command = ['/bin/bash', script, configdrive, device] execute_mock.return_value = ('', '') stat_mock.return_value.st_size = 5 standby._write_configdrive_to_partition(configdrive, device) execute_mock.assert_called_once_with(*command, check_exit_code=[0]) execute_mock.reset_mock() execute_mock.return_value = ('', '') execute_mock.side_effect = processutils.ProcessExecutionError self.assertRaises(errors.ConfigDriveWriteError, standby._write_configdrive_to_partition, configdrive, device) execute_mock.assert_called_once_with(*command, check_exit_code=[0])