def test_run_image(self, call_mock): script = standby._path_to_script('shell/reboot.sh') command = ['/bin/bash', script] call_mock.return_value = 0 success_result = self.agent_mode.run_image('run_image') success_result.join() call_mock.assert_called_once_with(command) call_mock.reset_mock() call_mock.return_value = 1 failed_result = self.agent_mode.run_image('run_image') failed_result.join() call_mock.assert_called_once_with(command) self.assertEqual('FAILED', failed_result.command_status)
def test_copy_configdrive_to_disk(self, call_mock, open_mock): device = '/dev/sda' configdrive = 'configdrive' script = standby._path_to_script('shell/copy_configdrive_to_disk.sh') command = ['/bin/bash', script, configdrive, device] call_mock.return_value = 0 standby._copy_configdrive_to_disk(configdrive, device) call_mock.assert_called_once_with(command) call_mock.reset_mock() call_mock.return_value = 1 self.assertRaises(errors.ConfigDriveWriteError, standby._copy_configdrive_to_disk, configdrive, device) call_mock.assert_called_once_with(command)
def test_write_image(self, call_mock, open_mock): image_info = self._build_fake_image_info() device = '/dev/sda' location = standby._image_location(image_info) script = standby._path_to_script('shell/write_image.sh') command = ['/bin/bash', script, location, device] call_mock.return_value = 0 standby._write_image(image_info, device) call_mock.assert_called_once_with(command) call_mock.reset_mock() call_mock.return_value = 1 self.assertRaises(errors.ImageWriteError, standby._write_image, image_info, device) call_mock.assert_called_once_with(command)