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()