func NewRouter(mx muxer, mom mother, config Config) http.Handler { registrar := mom.Registrar() notificationsFinder := mom.NotificationsFinder() emailStrategy := mom.EmailStrategy() userStrategy := mom.UserStrategy() spaceStrategy := mom.SpaceStrategy() organizationStrategy := mom.OrganizationStrategy() everyoneStrategy := mom.EveryoneStrategy() uaaScopeStrategy := mom.UAAScopeStrategy() notifyObj := notify.NewNotify(mom.NotificationsFinder(), registrar) preferencesFinder := mom.PreferencesFinder() preferenceUpdater := mom.PreferenceUpdater() templateCreator, templateFinder, templateUpdater, templateDeleter, templateLister, templateAssigner, templateAssociationLister := mom.TemplateServiceObjects() notificationsUpdater := mom.NotificationsUpdater() messageFinder := mom.MessageFinder() errorWriter := webutil.NewErrorWriter() requestCounter := middleware.NewRequestCounter(mx.GetRouter(), metrics.DefaultLogger) logging := middleware.NewRequestLogging(config.Logger) notificationsWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notifications.write") notificationsManageAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notifications.manage") notificationPreferencesReadAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notification_preferences.read") notificationPreferencesWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notification_preferences.write") notificationPreferencesAdminAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notification_preferences.admin") emailsWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "emails.write") notificationsTemplateWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notification_templates.write") notificationsTemplateReadAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notification_templates.read") notificationsWriteOrEmailsWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notifications.write", "emails.write") databaseAllocator := middleware.NewDatabaseAllocator(mom.SQLDatabase(), config.DBLoggingEnabled) cors := middleware.NewCORS(config.CORSOrigin) info.Routes{ RequestCounter: requestCounter, RequestLogging: logging, }.Register(mx) preferences.Routes{ CORS: cors, RequestCounter: requestCounter, RequestLogging: logging, DatabaseAllocator: databaseAllocator, NotificationPreferencesReadAuthenticator: notificationPreferencesReadAuthenticator, NotificationPreferencesWriteAuthenticator: notificationPreferencesWriteAuthenticator, NotificationPreferencesAdminAuthenticator: notificationPreferencesAdminAuthenticator, ErrorWriter: errorWriter, PreferencesFinder: preferencesFinder, PreferenceUpdater: preferenceUpdater, }.Register(mx) clients.Routes{ RequestCounter: requestCounter, RequestLogging: logging, DatabaseAllocator: databaseAllocator, NotificationsManageAuthenticator: notificationsManageAuthenticator, ErrorWriter: errorWriter, TemplateAssigner: templateAssigner, }.Register(mx) messages.Routes{ RequestCounter: requestCounter, RequestLogging: logging, DatabaseAllocator: databaseAllocator, NotificationsWriteOrEmailsWriteAuthenticator: notificationsWriteOrEmailsWriteAuthenticator, ErrorWriter: errorWriter, MessageFinder: messageFinder, }.Register(mx) templates.Routes{ RequestCounter: requestCounter, RequestLogging: logging, DatabaseAllocator: databaseAllocator, NotificationTemplatesReadAuthenticator: notificationsTemplateReadAuthenticator, NotificationTemplatesWriteAuthenticator: notificationsTemplateWriteAuthenticator, NotificationsManageAuthenticator: notificationsManageAuthenticator, ErrorWriter: errorWriter, TemplateFinder: templateFinder, TemplateUpdater: templateUpdater, TemplateCreator: templateCreator, TemplateDeleter: templateDeleter, TemplateLister: templateLister, TemplateAssociationLister: templateAssociationLister, }.Register(mx) notifications.Routes{ RequestCounter: requestCounter, RequestLogging: logging, DatabaseAllocator: databaseAllocator, NotificationsWriteAuthenticator: notificationsWriteAuthenticator, NotificationsManageAuthenticator: notificationsManageAuthenticator, ErrorWriter: errorWriter, Registrar: registrar, NotificationsFinder: notificationsFinder, NotificationsUpdater: notificationsUpdater, TemplateAssigner: templateAssigner, }.Register(mx) notify.Routes{ RequestCounter: requestCounter, RequestLogging: logging, NotificationsWriteAuthenticator: notificationsWriteAuthenticator, DatabaseAllocator: databaseAllocator, EmailsWriteAuthenticator: emailsWriteAuthenticator, ErrorWriter: errorWriter, Notify: notifyObj, UserStrategy: userStrategy, SpaceStrategy: spaceStrategy, OrganizationStrategy: organizationStrategy, EveryoneStrategy: everyoneStrategy, UAAScopeStrategy: uaaScopeStrategy, EmailStrategy: emailStrategy, }.Register(mx) return mx }
var _ = Describe("Database Allocator", func() { var ( ware middleware.DatabaseAllocator sqlDB *sql.DB writer *httptest.ResponseRecorder request *http.Request buffer *bytes.Buffer context stack.Context ) BeforeEach(func() { var err error sqlDB, err = sqlmock.New() Expect(err).NotTo(HaveOccurred()) ware = middleware.NewDatabaseAllocator(sqlDB, true) writer = httptest.NewRecorder() request = &http.Request{} buffer = bytes.NewBuffer([]byte{}) logger := lager.NewLogger("notifications") logger.RegisterSink(lager.NewWriterSink(buffer, lager.DEBUG)) context = stack.NewContext() context.Set("logger", logger) context.Set(middleware.VCAPRequestIDKey, "some-vcap-request-id") }) AfterEach(func() { err := sqlDB.Close()