def test_get_workbook(self): """Tests if get workbook function will be able to return fake workbook document""" _get_abs_path_mock = self._patch_get_abs_path('dd') _split_ext_mock = self._patch_split_ext(('a', '.xmind')) _input_stream_mock = Mock() _input_stream_mock.namelist = MagicMock( return_value=['something', 'content.xml']) _input_stream_mock.read = MagicMock() _stream_mock = Mock() _stream_mock.__enter__ = MagicMock(return_value=_input_stream_mock) _stream_mock.__exit__ = MagicMock() _utils_extract_mock = self._init_patch_with_name( '_utils_extract', 'xmind.utils.extract', _stream_mock) _parse_dom_string_mock = self._init_patch_with_name( '_parse_dom_string', 'xmind.utils.parse_dom_string', 'something') _wb_mock = self._init_patch_with_name( '_wb', 'xmind.core.loader.WorkbookDocument', autospec=True) wb = WorkbookLoader('dd') wb.get_workbook() _get_abs_path_mock.assert_called_once() _split_ext_mock.assert_called_once_with('dd') _stream_mock.__enter__.assert_called_once() _stream_mock.__exit__.assert_called_once() _input_stream_mock.namelist.assert_called_once() _input_stream_mock.read.assert_called_once() _utils_extract_mock.assert_called_once() _parse_dom_string_mock.assert_called_once() _wb_mock.assert_called()
def load(path): """ Load XMind workbook from given path. If file no exist on given path then created new one. """ loader = WorkbookLoader(path) return loader.get_workbook()
def load_all(path): """ Load XMind workbook from given path. If file no exist on given path then created new one. (workbook, stylesbook) return """ loader = WorkbookLoader(path) return loader.get_workbook(), loader.get_stylesbook()
def load_all(path): """ Load XMind workbook from given path. If file no exist on given path then created new one. (workbook, stylesbook) return """ loader = WorkbookLoader(path) return loader.get_workbook(), loader.get_stylesbook()