fakeLRPDB.UnclaimActualLRPStub = func(_ lager.Logger, key *models.ActualLRPKey) (*models.ActualLRPGroup, *models.ActualLRPGroup, error) { if key.ProcessGuid == unclaimingActualLRP1.ProcessGuid { return &models.ActualLRPGroup{Instance: unclaimingActualLRP1}, &models.ActualLRPGroup{Instance: unclaimingActualLRP1}, nil } if key.ProcessGuid == unclaimingActualLRP2.ProcessGuid { return &models.ActualLRPGroup{Instance: unclaimingActualLRP2}, &models.ActualLRPGroup{Instance: unclaimingActualLRP2}, nil } return nil, nil, models.ErrResourceNotFound } fakeLRPDB.ConvergeLRPsReturns(keysToAuction, keysWithMissingCells, keysToRetire) logger.RegisterSink(lager.NewWriterSink(GinkgoWriter, lager.DEBUG)) fakeServiceClient = new(fake_bbs.FakeServiceClient) fakeRepClientFactory = new(repfakes.FakeClientFactory) fakeRepClient = new(repfakes.FakeClient) fakeRepClientFactory.CreateClientReturns(fakeRepClient, nil) fakeServiceClient.CellByIdReturns(nil, errors.New("hi")) cellPresence := models.NewCellPresence("cell-id", "1.1.1.1", "", "z1", models.CellCapacity{}, nil, nil, nil, nil) cellSet = models.CellSet{"cell-id": &cellPresence} fakeServiceClient.CellsReturns(cellSet, nil) actualHub = &eventfakes.FakeHub{} retirer := controllers.NewActualLRPRetirer(fakeLRPDB, actualHub, fakeRepClientFactory, fakeServiceClient) controller = controllers.NewLRPConvergenceController(logger, fakeLRPDB, actualHub, fakeAuctioneerClient, fakeServiceClient, retirer, 2) })
) var _ = Describe("TaskWorker", func() { var ( fakeServer *ghttp.Server logger *lagertest.TestLogger timeout time.Duration ) BeforeEach(func() { timeout = 1 * time.Second cfhttp.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 task *models.Task httpClient *http.Client )