}) var _ = BeforeEach(func() { logger = lagertest.NewTestLogger("test") 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())
AuctioneerAddress: auctioneerServer.URL(), ConsulCluster: consulRunner.ConsulCluster(), DropsondePort: port, EtcdCluster: etcdUrl, MetricsReportInterval: 10 * time.Millisecond, HealthAddress: bbsHealthAddress, EncryptionKeys: []string{"label:key"}, ActiveKeyLabel: "label", } if test_helpers.UseSQL() { bbsArgs.DatabaseDriver = sqlRunner.DriverName() bbsArgs.DatabaseConnectionString = sqlRunner.ConnectionString() } storeClient = etcd.NewStoreClient(etcdClient) consulHelper = test_helpers.NewConsulHelper(logger, consulClient) }) var _ = AfterEach(func() { ginkgomon.Kill(bbsProcess) // Make sure the healthcheck server is really gone before trying to start up // the bbs again in another test. Eventually(func() error { conn, err := net.Dial("tcp", bbsHealthAddress) if err == nil { conn.Close() return nil } return err