Add tests for notification type