| import unittest.mock as mock |
| from unittest import TestCase |
| |
| from solrorbit.builder.executors.exception_handling_shell_executor import ExceptionHandlingShellExecutor |
| from solrorbit.exceptions import ExecutorError |
| |
| |
| class ExceptionHandlingShellExecutorTests(TestCase): |
| def setUp(self): |
| self.executor_impl = mock.Mock() |
| self.executor_impl.execute.return_value = None |
| self.executor_impl.copy.return_value = None |
| |
| self.executor = ExceptionHandlingShellExecutor(self.executor_impl) |
| self.host = None |
| self.command = None |
| self.source = "/path/to/source" |
| self.destination = "/path/to/dest" |
| |
| def test_success_executing_command(self): |
| self.executor.execute(self.host, self.command) |
| |
| def test_failure_executing_command(self): |
| self.executor_impl.execute.side_effect = Exception("error") |
| |
| with self.assertRaises(ExecutorError): |
| self.executor.execute(self.host, self.command) |
