Adds DOM aware unit tests