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