Add test for ta_highlight{,_all} and fix implementation