def setUp(self): self.data = get_data('folder_item.json') parent_task = get_sample_task_base() self.task = CreateDirTask(parent_task, '/', self.data['name']) self.mock_call = mock.Mock(return_value=OneDriveItem(drive=self.task.drive, data=self.data)) self.task.drive.create_dir = self.mock_call self._backup_isdir = os.path.isdir
class TestCreateDirTask(unittest.TestCase): def setUp(self): self.data = get_data('folder_item.json') parent_task = get_sample_task_base() self.task = CreateDirTask(parent_task, '/', self.data['name']) self.mock_call = mock.Mock(return_value=OneDriveItem(drive=self.task.drive, data=self.data)) self.task.drive.create_dir = self.mock_call self._backup_isdir = os.path.isdir def tearDown(self): os.path.isdir = self._backup_isdir def test_handle(self): os.path.isdir = lambda p: True self.task.handle() self.mock_call.assert_called_once_with(name=self.data['name'], parent_path=self.data['parentReference']['path']) self.assertEqual(1, len(self.task.items_store.get_items_by_id(item_id=self.data['id']))) def test_handle_OSError(self): os.path.isdir = mock.Mock(side_effect=OSError()) self.task.handle() def test_handle_APIError(self): os.path.isdir = lambda p: True self.task.drive.create_dir = mock.Mock(side_effect=OneDriveError(get_data('error_type1.json'))) self.task.handle()