def test_addIdAttribute_no_op(self): """Tests addIdAttribute when node has attribute already""" _node = MagicMock() _node.hasAttribute.return_value = True _element = Element(_node) _result = _element.addIdAttribute('some') self.assertIsNone(_result) _node.hasAttribute.assert_called_once_with('some') _node.setAttribute.assert_not_called() _node.setIdAttribute.assert_not_called()
def test_addIdAttribute_has_document_owner(self): """Tests addIdAttribute when node has attribute already and has owner of document""" _node = MagicMock() _node.hasAttribute.return_value = False _generate_id_mock = self._init_patch_with_name( '_generate_id', 'xmind.core.utils.generate_id', return_value=2, autospec=True) _get_owner_document = self._init_patch_with_name( '_getOwnerDocument', 'xmind.core.Element.getOwnerDocument', return_value=True, autospec=True) _element = Element(_node) _result = _element.addIdAttribute('some') self.assertIsNone(_result) _generate_id_mock.assert_called_once() _node.hasAttribute.assert_called_once_with('some') _node.setAttribute.assert_called_once_with('some', 2) _node.setIdAttribute.assert_called_once_with('some') _get_owner_document.assert_called_once()