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