| from unittest import TestCase, mock |
| from unittest.mock import Mock |
| |
| from solrorbit.builder.utils.path_manager import PathManager |
| from solrorbit.exceptions import ExecutorError |
| |
| |
| class PathManagerTest(TestCase): |
| def setUp(self): |
| self.host = None |
| self.path = "fake" |
| |
| self.executor = Mock() |
| self.path_manager = PathManager(self.executor) |
| |
| @mock.patch('solrorbit.utils.io.ensure_dir') |
| def test_create_path(self, ensure_dir): |
| self.path_manager.create_path(self.host, self.path) |
| |
| ensure_dir.assert_has_calls([ |
| mock.call(self.path) |
| ]) |
| self.executor.execute.assert_has_calls([ |
| mock.call(self.host, f"mkdir -m 0777 -p {self.path}") |
| ]) |
| |
| @mock.patch('solrorbit.utils.io.ensure_dir') |
| def test_create_path_no_local_copy(self, ensure_dir): |
| self.path_manager.create_path(self.host, self.path) |
| |
| ensure_dir.assert_has_calls([]) |
| self.executor.execute.assert_has_calls([ |
| mock.call(self.host, f"mkdir -m 0777 -p {self.path}") |
| ]) |
| |
| def test_delete_valid_path(self): |
| self.path_manager.delete_path(self.host, self.path) |
| |
| self.executor.execute.assert_has_calls([ |
| mock.call(self.host, f"rm -r {self.path}") |
| ]) |
| |
| def test_delete_invalid_path(self): |
| self.path_manager.delete_path(self.host, "/") |
| |
| self.executor.execute.assert_has_calls([]) |
| |
| def test_path_is_present(self): |
| self.executor.execute.return_value = None |
| |
| is_path_present = self.path_manager.is_path_present(self.host, self.path) |
| self.assertEqual(is_path_present, True) |
| |
| def test_path_is_not_present(self): |
| self.executor.execute.side_effect = ExecutorError("fake") |
| |
| is_path_present = self.path_manager.is_path_present(self.host, self.path) |
| self.assertEqual(is_path_present, False) |
