module Test::Unit::NotificationHandler

def handle_notified_error(exception)
  return false unless exception.is_a?(NotifiedError)
  notification = Notification.new(name,
                                  filter_backtrace(exception.backtrace),
                                  exception.message)
  add_notification(notification)
  true
end

def included(base)
  base.exception_handler(:handle_notified_error)
end