Add assertDocument() and assertElement() and use where appropriate