Added tests for NotificationController