func TestStoreAdapter(t *testing.T) { registerSignalHandler() RegisterFailHandler(Fail) etcdPort := 5000 + (config.GinkgoConfig.ParallelNode-1)*10 etcdRunner = storerunner.NewETCDClusterRunner(etcdPort, 5) zookeeperPort := 2181 + (config.GinkgoConfig.ParallelNode-1)*10 zookeeperRunner = storerunner.NewZookeeperClusterRunner(zookeeperPort, 1) etcdRunner.Start() zookeeperRunner.Start() RunSpecs(t, "Store Adapter Suite") stopStores() }
nodes := nodes for _, concurrency := range concurrencies { concurrency := concurrency Context(fmt.Sprintf("With %d %s nodes (%d concurrent requests at a time)", nodes, storeType, concurrency), func() { var storeAdapter storeadapter.StoreAdapter BeforeEach(func() { if storeType == "ETCD" { storeRunner = storerunner.NewETCDClusterRunner(5001, nodes) storeRunner.Start() storeAdapter = storeadapter.NewETCDStoreAdapter(storeRunner.NodeURLS(), workerpool.NewWorkerPool(concurrency)) err := storeAdapter.Connect() Ω(err).ShouldNot(HaveOccured()) } else if storeType == "Zookeeper" { storeRunner = storerunner.NewZookeeperClusterRunner(2181, nodes) storeRunner.Start() storeAdapter = storeadapter.NewZookeeperStoreAdapter(storeRunner.NodeURLS(), workerpool.NewWorkerPool(concurrency), &timeprovider.RealTimeProvider{}, time.Second) err := storeAdapter.Connect() Ω(err).ShouldNot(HaveOccured()) } }) AfterEach(func() { storeAdapter.Disconnect() storeRunner.Stop() storeRunner = nil }) randomBytes := func(sizeInBytes int) []byte {