fakeAuctioneerClient = new(auctioneerfakes.FakeClient)
etcdRunner.Reset()
consulRunner.Reset()
consulSession = consulRunner.NewSession("a-session")
etcdClient := etcdRunner.Client()
etcdClient.SetConsistency(etcdclient.STRONG_CONSISTENCY)
storeClient = etcd.NewStoreClient(etcdClient)
fakeStoreClient = &fakes.FakeStoreClient{}
consulHelper = test_helpers.NewConsulHelper(consulSession)
serviceClient = bbs.NewServiceClient(consulSession, clock)
fakeTaskCompletionClient = new(faketaskworkpool.FakeTaskCompletionClient)
fakeRepClientFactory = new(repfakes.FakeClientFactory)
fakeRepClient = new(repfakes.FakeClient)
fakeRepClientFactory.CreateClientReturns(fakeRepClient)
etcdHelper = etcd_helpers.NewETCDHelper(format.ENCRYPTED_PROTO, cryptor, storeClient)
etcdDB = etcd.NewETCD(format.ENCRYPTED_PROTO, 100, 100, cryptor, storeClient, fakeAuctioneerClient, serviceClient, clock, fakeRepClientFactory, fakeTaskCompletionClient)
etcdDBWithFakeStore = etcd.NewETCD(format.ENCRYPTED_PROTO, 100, 100, cryptor, fakeStoreClient, fakeAuctioneerClient, serviceClient, clock, fakeRepClientFactory, fakeTaskCompletionClient)
})
func registerCell(cell models.CellPresence) {
var err error
jsonBytes, err := json.Marshal(cell)
Expect(err).NotTo(HaveOccurred())
_, err = consulSession.SetPresence(bbs.CellSchemaPath(cell.CellID), jsonBytes)
Expect(err).NotTo(HaveOccurred())
}
delegate *auctionrunnerdelegate.AuctionRunnerDelegate
bbsClient *fake_bbs.FakeInternalClient
metricSender *fake.FakeMetricSender
repClientFactory *repfakes.FakeClientFactory
repClient *repfakes.FakeClient
logger lager.Logger
)
BeforeEach(func() {
metricSender = fake.NewFakeMetricSender()
metrics.Initialize(metricSender, nil)
bbsClient = &fake_bbs.FakeInternalClient{}
repClientFactory = &repfakes.FakeClientFactory{}
repClient = &repfakes.FakeClient{}
repClientFactory.CreateClientReturns(repClient)
logger = lagertest.NewTestLogger("delegate")
delegate = auctionrunnerdelegate.New(repClientFactory, bbsClient, logger)
})
Describe("fetching cell reps", func() {
Context("when the BSS succeeds", func() {
BeforeEach(func() {
cellPresence1 := models.NewCellPresence("cell-A", "cell-a.url", "zone-1", models.NewCellCapacity(123, 456, 789), []string{}, []string{})
cellPresence2 := models.NewCellPresence("cell-B", "cell-b.url", "zone-1", models.NewCellCapacity(123, 456, 789), []string{}, []string{})
cells := []*models.CellPresence{&cellPresence1, &cellPresence2}
bbsClient.CellsReturns(cells, nil)
})
