def test_createRelationship_implementations_not_equal(self):
_sheet1 = Mock()
_sheet1.getImplementation.return_value = "implementation1"
_end1 = Mock()
_end1.getOwnerSheet.return_value = _sheet1
_sheet2 = Mock()
_sheet2.getImplementation.return_value = "implementation2"
_end2 = Mock()
_end2.getOwnerSheet.return_value = _sheet2
_create_relationship = self._init_patch_with_name(
'_create_relationship',
'xmind.core.workbook.WorkbookDocument._create_relationship')
_workbook_document = WorkbookDocument()
with self.assertRaises(Exception) as _ex:
_workbook_document.createRelationship(_end1, _end2)
self.assertTrue(
_ex.exception.args[0].find("Topics not on the same sheet!") != -1)
_end1.getOwnerSheet.assert_called_once()
_end2.getOwnerSheet.assert_called_once()
_sheet1.getImplementation.assert_called_once()
_sheet2.getImplementation.assert_called_once()
_create_relationship.assert_not_called()
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def test_get_path_is_none(self):
_get_abs_path = self._init_patch_with_name(
'_get_abs_path', 'xmind.core.utils.get_abs_path')
_workbook_document = WorkbookDocument()
self.assertIsNone(_workbook_document.get_path())
_get_abs_path.assert_not_called()
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def test_get_path_is_not_none(self):
_get_abs_path = self._init_patch_with_name(
'_get_abs_path',
'xmind.core.utils.get_abs_path',
return_value="abs_path")
_workbook_document = WorkbookDocument(None, "path")
self.assertEqual(_workbook_document.get_path(), "abs_path")
_get_abs_path.assert_called_with("path")
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def test_create_topic(self):
_topic_element = self._init_patch_with_name(
'_relationship_element',
'xmind.core.workbook.TopicElement',
return_value="TopicElement")
_workbook_document = WorkbookDocument()
self.assertEqual(_workbook_document.createTopic(), "TopicElement")
_topic_element.assert_called_once_with(None, _workbook_document)
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def test_set_modified_time(self):
self._remove_patched_function("_workbook_element")
_obj = Mock()
_obj.setModifiedTime.return_value = None
self._workbook_element = self._init_patch_with_name(
'_workbook_element',
'xmind.core.workbook.WorkbookElement',
return_value=_obj)
_workbook_document = WorkbookDocument()
self.assertIsNone(_workbook_document.setModifiedTime())
_obj.setModifiedTime.assert_called_once()
self._append_child.assert_called_once_with(_obj)
self._assert_init_methods(_workbook_document)
def test_get_version(self):
self._remove_patched_function("_workbook_element")
_obj = Mock()
_obj.getVersion.return_value = "Version"
self._workbook_element = self._init_patch_with_name(
'_workbook_element',
'xmind.core.workbook.WorkbookElement',
return_value=_obj)
_workbook_document = WorkbookDocument()
self.assertEqual(_workbook_document.getVersion(), "Version")
_obj.getVersion.assert_called_once()
self._append_child.assert_called_once_with(_obj)
self._assert_init_methods(_workbook_document)
def test_add_sheet(self):
self._remove_patched_function("_workbook_element")
_obj = Mock()
_obj.addSheet.return_value = None
self._workbook_element = self._init_patch_with_name(
'_workbook_element',
'xmind.core.workbook.WorkbookElement',
return_value=_obj)
_workbook_document = WorkbookDocument()
self.assertIsNone(_workbook_document.addSheet("sheet1"))
_obj.addSheet.assert_called_once_with("sheet1", None)
self._append_child.assert_called_once_with(_obj)
self._assert_init_methods(_workbook_document)
def test_get_primary_sheets(self):
self._remove_patched_function("_workbook_element")
_obj = Mock()
_obj.getSheetByIndex.return_value = "sheet1"
self._workbook_element = self._init_patch_with_name(
'_workbook_element',
'xmind.core.workbook.WorkbookElement',
return_value=_obj)
_workbook_document = WorkbookDocument()
self.assertEqual(_workbook_document.getPrimarySheet(), "sheet1")
_obj.getSheetByIndex.assert_called_once_with(0)
self._append_child.assert_called_once_with(_obj)
self._assert_init_methods(_workbook_document)
def test_get_sheets(self):
self._remove_patched_function("_workbook_element")
_obj = Mock()
_obj.getSheets.return_value = ["sheet1", "sheet2"]
self._workbook_element = self._init_patch_with_name(
'_workbook_element',
'xmind.core.workbook.WorkbookElement',
return_value=_obj)
_workbook_document = WorkbookDocument()
self.assertListEqual(_workbook_document.getSheets(),
["sheet1", "sheet2"])
_obj.getSheets.assert_called_once()
self._append_child.assert_called_once_with(_obj)
self._assert_init_methods(_workbook_document)
def test_createRelationship_sheet_is_none(self):
_sheet1 = Mock()
_sheet1.getImplementation.return_value = None
_end1 = Mock()
_end1.getOwnerSheet.return_value = _sheet1
_end2 = Mock()
_end2.getOwnerSheet.return_value = None
_workbook_document = WorkbookDocument()
with self.assertRaises(Exception) as _ex:
_workbook_document.createRelationship(_end1, _end2)
self.assertTrue(
_ex.exception.args[0].find("Topics not on the same sheet!") != -1)
_end1.getOwnerSheet.assert_called_once()
_end2.getOwnerSheet.assert_called_once()
_sheet1.getImplementation.assert_not_called()
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def test_init_appendChild_is_not_called(self):
self._remove_patched_function("_get_first_child_node_by_tag_name")
_workbook_element = Mock()
self._get_first_child_node_by_tag_name = self._init_patch_with_name(
'_get_first_child_node_by_tag_name',
'xmind.core.workbook.WorkbookDocument.getFirstChildNodeByTagName',
return_value=_workbook_element)
_workbook_document = WorkbookDocument()
self._append_child.assert_not_called()
self._assert_init_methods(_workbook_document, _workbook_element)
def test_createRelationship(self):
_sheet1 = Mock()
_sheet1.getImplementation.return_value = "implementation"
_sheet1.addRelationships.return_value = None
_end1 = Mock()
_end1.getOwnerSheet.return_value = _sheet1
_end1.getID.return_value = "ID1"
_sheet2 = Mock()
_sheet2.getImplementation.return_value = "implementation"
_end2 = Mock()
_end2.getOwnerSheet.return_value = _sheet2
_end2.getID.return_value = "ID2"
_rel = Mock()
_rel.setEnd1ID.return_value = None
_rel.setEnd2ID.return_value = None
_create_relationship = self._init_patch_with_name(
'_create_relationship',
'xmind.core.workbook.WorkbookDocument._create_relationship',
return_value=_rel)
_workbook_document = WorkbookDocument()
self.assertEqual(_workbook_document.createRelationship(_end1, _end2),
_rel)
_end1.getOwnerSheet.assert_called_once()
_end2.getOwnerSheet.assert_called_once()
_sheet1.getImplementation.assert_called_once()
_sheet2.getImplementation.assert_called_once()
_create_relationship.assert_called_once()
_end1.getID.assert_called_once()
_end2.getID.assert_called_once()
_rel.setEnd1ID.assert_called_once_with("ID1")
_rel.setEnd2ID.assert_called_once_with("ID2")
_sheet1.addRelationships.assert_called_once_with(_rel)
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def test_excessive_parameters(self):
_workbook_document = WorkbookDocument()
_parameters = [('__init__', (2, False)), ('getWorkbookElement', 0),
('_create_relationship', 0), ('createRelationship', 2),
('createTopic', 0), ('getSheets', 0),
('getPrimarySheet', 0), ('createSheet', 0),
('addSheet', 2), ('removeSheet', 1), ('moveSheet', 2),
('getVersion', 0), ('getModifiedTime', 0),
('updateModifiedTime', 0), ('setModifiedTime', 0),
('get_path', 0), ('set_path', 1)]
for pair in _parameters:
with self.subTest(pair=pair):
self._test_method_by_excessive_parameters(
pair, _workbook_document)
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
def get_indent_out_file(self, workbook: WorkbookDocument):
out_file = self.indent_output
if not out_file:
out_file = workbook.getPrimarySheet().getRootTopic().getTitle(
) + '.md'
return out_file
def test_get_workbook_element(self):
_workbook_document = WorkbookDocument()
self._append_child.assert_called_once_with(None)
self.assertIsNone(_workbook_document.getWorkbookElement())
self._assert_init_methods(_workbook_document)
def test_init(self):
_workbook_document = WorkbookDocument()
self._append_child.assert_called_once_with(None)
self._assert_init_methods(_workbook_document)
