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) })