def test_write_image(self, execute_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] execute_mock.return_value = ('', '') standby._write_image(image_info, 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.ImageWriteError, standby._write_image, image_info, device) execute_mock.assert_called_once_with(*command, check_exit_code=[0])
def test_power_off(self, execute_mock): script = standby._path_to_script('shell/shutdown.sh') command = ['/bin/bash', script, '-h'] execute_mock.return_value = ('', '') success_result = self.agent_extension.power_off() success_result.join() execute_mock.assert_called_once_with(*command, check_exit_code=[0]) self.assertEqual('SUCCEEDED', success_result.command_status) execute_mock.reset_mock() execute_mock.return_value = ('', '') execute_mock.side_effect = processutils.ProcessExecutionError failed_result = self.agent_extension.power_off() failed_result.join() execute_mock.assert_called_once_with(*command, check_exit_code=[0]) self.assertEqual('FAILED', failed_result.command_status)
def test_power_off(self, execute_mock): script = standby._path_to_script('shell/shutdown.sh') command = ['/bin/bash', script, '-h'] execute_mock.return_value = ('', '') success_result = self.agent_extension.power_off() success_result.join() execute_mock.assert_called_once_with(*command, check_exit_code=[0]) self.assertEqual('SUCCEEDED', success_result.command_status) execute_mock.reset_mock() execute_mock.return_value = ('', '') execute_mock.side_effect = processutils.ProcessExecutionError failed_result = self.agent_extension.power_off() failed_result.join() execute_mock.assert_called_once_with(*command, check_exit_code=[0]) self.assertEqual('FAILED', failed_result.command_status)
def test_write_image(self, execute_mock, open_mock): image_info = _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] execute_mock.return_value = ('', '') standby._write_image(image_info, 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.ImageWriteError, standby._write_image, image_info, 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])
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_path_to_script(self): script = standby._path_to_script('shell/reboot.sh') self.assertTrue(script.endswith('extensions/../shell/reboot.sh'))
def test_path_to_script(self): script = standby._path_to_script('shell/reboot.sh') self.assertTrue(script.endswith('extensions/../shell/reboot.sh'))