Add unit tests for highlight features.