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
)
