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()
