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()
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Routes", func() {
var (
logging middleware.RequestLogging
dbAllocator middleware.DatabaseAllocator
auth middleware.Authenticator
muxer web.Muxer
)
BeforeEach(func() {
logging = middleware.NewRequestLogging(lager.NewLogger("log-prefix"))
auth = middleware.NewAuthenticator("some-public-key", "notifications.write")
dbAllocator = middleware.NewDatabaseAllocator(&sql.DB{}, false)
muxer = web.NewMuxer()
campaigntypes.Routes{
RequestLogging: logging,
Authenticator: auth,
DatabaseAllocator: dbAllocator,
CampaignTypesCollection: collections.CampaignTypesCollection{},
}.Register(muxer)
})
It("routes POST /senders/{sender_id}/campaign_types", func() {
request, err := http.NewRequest("POST", "/senders/some-sender-id/campaign_types", nil)
Expect(err).NotTo(HaveOccurred())
s := muxer.Match(request).(stack.Stack)
Expect(s.Handler).To(BeAssignableToTypeOf(campaigntypes.CreateHandler{}))
func NewRouter(mx muxer, config Config) http.Handler {
requestCounter := middleware.NewRequestCounter(mx.GetRouter(), metrics.DefaultLogger)
logging := middleware.NewRequestLogging(config.Logger)
notificationsWriteAuthenticator := middleware.NewAuthenticator(config.UAAPublicKey, "notifications.write")
databaseAllocator := middleware.NewDatabaseAllocator(config.SQLDB, config.DBLoggingEnabled)
warrantConfig := warrant.Config{
Host: config.UAAHost,
SkipVerifySSL: config.SkipVerifySSL,
}
warrantUsersService := warrant.NewUsersService(warrantConfig)
warrantClientsService := warrant.NewClientsService(warrantConfig)
rainmakerConfig := rainmaker.Config{
Host: config.CCHost,
SkipVerifySSL: config.SkipVerifySSL,
}
rainmakerSpacesService := rainmaker.NewSpacesService(rainmakerConfig)
rainmakerOrganizationsService := rainmaker.NewOrganizationsService(rainmakerConfig)
userFinder := uaa.NewUserFinder(config.UAAClientID, config.UAAClientSecret, warrantUsersService, warrantClientsService)
spaceFinder := cf.NewSpaceFinder(config.UAAClientID, config.UAAClientSecret, warrantClientsService, rainmakerSpacesService)
orgFinder := cf.NewOrgFinder(config.UAAClientID, config.UAAClientSecret, warrantClientsService, rainmakerOrganizationsService)
campaignEnqueuer := queue.NewCampaignEnqueuer(config.Queue)
sendersRepository := models.NewSendersRepository(uuid.NewV4)
campaignTypesRepository := models.NewCampaignTypesRepository(uuid.NewV4)
templatesRepository := models.NewTemplatesRepository(uuid.NewV4)
campaignsRepository := models.NewCampaignsRepository(uuid.NewV4)
messagesRepository := models.NewMessagesRepository(util.NewClock())
sendersCollection := collections.NewSendersCollection(sendersRepository, campaignTypesRepository)
templatesCollection := collections.NewTemplatesCollection(templatesRepository)
campaignTypesCollection := collections.NewCampaignTypesCollection(campaignTypesRepository, sendersRepository, templatesRepository)
campaignsCollection := collections.NewCampaignsCollection(campaignEnqueuer, campaignsRepository, campaignTypesRepository, templatesRepository, sendersRepository, userFinder, spaceFinder, orgFinder)
campaignStatusesCollection := collections.NewCampaignStatusesCollection(campaignsRepository, messagesRepository)
info.Routes{
RequestCounter: requestCounter,
RequestLogging: logging,
}.Register(mx)
senders.Routes{
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
SendersCollection: sendersCollection,
}.Register(mx)
campaigntypes.Routes{
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
CampaignTypesCollection: campaignTypesCollection,
}.Register(mx)
templates.Routes{
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
TemplatesCollection: templatesCollection,
}.Register(mx)
campaigns.Routes{
Clock: util.NewClock(),
RequestLogging: logging,
Authenticator: notificationsWriteAuthenticator,
DatabaseAllocator: databaseAllocator,
CampaignsCollection: campaignsCollection,
CampaignStatusesCollection: campaignStatusesCollection,
}.Register(mx)
return mx
}