AUI-1170 - Use inDoc() to check if node is highlighted
AUI-1170 - Fix tests