. "github.com/onsi/ginkgo" . "github.com/onsi/gomega" . "github.com/onsi/gomega/gbytes" ) var _ = Describe("TaskAuctionHandler", func() { var ( logger *lagertest.TestLogger runner *fake_auction_runner.FakeAuctionRunner responseRecorder *httptest.ResponseRecorder handler *auctionhandlers.TaskAuctionHandler ) BeforeEach(func() { logger = lagertest.NewTestLogger("test") logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.DEBUG)) runner = new(fake_auction_runner.FakeAuctionRunner) responseRecorder = httptest.NewRecorder() handler = auctionhandlers.NewTaskAuctionHandler(runner) }) Describe("Create", func() { Context("when the request body is a task", func() { var tasks []*models.Task BeforeEach(func() { tasks = []*models.Task{{ TaskGuid: "the-task-guid", Domain: "some-domain", RootFs: "some:rootfs", Action: models.WrapAction(&models.RunAction{
) var _ = Describe("TaskWorker", func() { var ( fakeServer *ghttp.Server logger *lagertest.TestLogger timeout time.Duration ) BeforeEach(func() { timeout = 1 * time.Second cf_http.Initialize(timeout) fakeServer = ghttp.NewServer() logger = lagertest.NewTestLogger("test") logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.INFO)) }) AfterEach(func() { fakeServer.Close() }) Describe("HandleCompletedTask", func() { var ( callbackURL string taskDB *dbfakes.FakeTaskDB statusCodes chan int reqCount chan struct{} task *models.Task httpClient *http.Client