Add tests for NotificationsGenerator