Example #1
0
func (app Application) StartWorkers() {
	zonedUAAClient := uaa.NewZonedUAAClient(app.env.UAAClientID, app.env.UAAClientSecret, app.env.VerifySSL, UAAPublicKey)

	WorkerGenerator{
		InstanceIndex: app.env.VCAPApplication.InstanceIndex,
		Count:         WorkerCount,
	}.Work(func(i int) Worker {
		cloak, err := conceal.NewCloak(app.env.EncryptionKey)
		if err != nil {
			panic(err)
		}
		v1Workflow := postal.NewV1Process(postal.V1ProcessConfig{
			DBTrace: app.env.DBLoggingEnabled,
			UAAHost: app.env.UAAHost,
			Sender:  app.env.Sender,
			Domain:  app.env.Domain,

			Packager:    postal.NewPackager(app.mother.TemplatesLoader(), cloak),
			MailClient:  app.mother.MailClient(),
			Database:    app.mother.Database(),
			TokenLoader: uaa.NewTokenLoader(zonedUAAClient),
			UserLoader:  postal.NewUserLoader(zonedUAAClient),

			KindsRepo:              app.mother.KindsRepo(),
			ReceiptsRepo:           app.mother.ReceiptsRepo(),
			UnsubscribesRepo:       app.mother.UnsubscribesRepo(),
			GlobalUnsubscribesRepo: app.mother.GlobalUnsubscribesRepo(),
			MessageStatusUpdater:   postal.NewMessageStatusUpdater(app.mother.MessagesRepo()),
			DeliveryFailureHandler: postal.NewDeliveryFailureHandler(),
		})

		database := v2models.NewDatabase(app.mother.SQLDatabase(), v2models.Config{})
		messageStatusUpdater := postal.NewV2MessageStatusUpdater(v2models.NewMessagesRepository(util.NewClock()))
		v2Workflow := postal.NewV2Workflow(app.mother.MailClient(), postal.NewPackager(app.mother.TemplatesLoader(), cloak), postal.NewUserLoader(zonedUAAClient), uaa.NewTokenLoader(zonedUAAClient), messageStatusUpdater, database, app.env.Sender, app.env.Domain, app.env.UAAHost)

		worker := postal.NewDeliveryWorker(v1Workflow, v2Workflow, postal.DeliveryWorkerConfig{
			ID:                     i,
			UAAHost:                app.env.UAAHost,
			Logger:                 app.mother.Logger(),
			Queue:                  app.mother.Queue(),
			DBTrace:                app.env.DBLoggingEnabled,
			Database:               database,
			StrategyDeterminer:     strategy.NewStrategyDeterminer(app.mother.UserStrategy(), app.mother.SpaceStrategy(), app.mother.OrganizationStrategy(), app.mother.EmailStrategy()),
			DeliveryFailureHandler: postal.NewDeliveryFailureHandler(),
			MessageStatusUpdater:   messageStatusUpdater,
		})
		return &worker
	})
}
		delivery = postal.Delivery{
			ClientID: "some-client",
			UserGUID: "user-123",
			Options: postal.Options{
				Subject:    "the subject",
				Text:       "body content",
				ReplyTo:    "*****@*****.**",
				KindID:     "some-kind",
				TemplateID: "some-template-id",
			},
			MessageID:     "randomly-generated-guid",
			VCAPRequestID: "some-request-id",
			CampaignID:    "some-campaign-id",
		}

		workflow = postal.NewV2Workflow(mailClient, packager, userLoader, tokenLoader, messageStatusUpdater, database, "*****@*****.**", "example.com", "uaa-host")
	})

	It("ensures message delivery", func() {
		err := workflow.Deliver(delivery, logger)
		Expect(err).NotTo(HaveOccurred())

		Expect(tokenLoader.LoadCall.Receives.UAAHost).To(Equal("uaa-host"))

		Expect(userLoader.LoadCall.Receives.UserGUIDs).To(Equal([]string{"user-123"}))
		Expect(userLoader.LoadCall.Receives.Token).To(Equal("some-token"))

		Expect(mailClient.SendCall.CallCount).To(Equal(1))
		msg := mailClient.SendCall.Receives.Message
		Expect(msg.From).To(Equal("*****@*****.**"))
		Expect(msg.ReplyTo).To(Equal("*****@*****.**"))