Update notification types and add test for comment mentions