provider WorkerProvider
workers []Worker
workersErr error
)
BeforeEach(func() {
fakeDB = new(fakes.FakeWorkerDB)
logger = lagertest.NewTestLogger("test")
worker = new(gfakes.FakeBackend)
workerAddr = fmt.Sprintf("0.0.0.0:%d", 8888+GinkgoParallelNode())
workerBaggageclaimURL = "http://1.2.3.4:7788"
workerServer = server.New("tcp", workerAddr, 0, worker, logger)
err := workerServer.Start()
Expect(err).NotTo(HaveOccurred())
provider = NewDBWorkerProvider(logger, fakeDB, nil, immediateRetryPolicy{})
})
AfterEach(func() {
workerServer.Stop()
Eventually(func() error {
conn, err := net.Dial("tcp", workerAddr)
if err == nil {
conn.Close()
}
return err
tmpdir, err = ioutil.TempDir(os.TempDir(), "api-server-test")
Ω(err).ShouldNot(HaveOccurred())
socketPath = path.Join(tmpdir, "api.sock")
serverBackend = new(fakes.FakeBackend)
serverContainerGraceTime = 42 * time.Second
apiServer = server.New(
"unix",
socketPath,
serverContainerGraceTime,
serverBackend,
logger,
)
err = apiServer.Start()
Ω(err).ShouldNot(HaveOccurred())
isRunning = true
Eventually(ErrorDialing("unix", socketPath)).ShouldNot(HaveOccurred())
apiClient = client.New(connection.New("unix", socketPath))
})
AfterEach(func() {
if isRunning {
apiServer.Stop()
}
if tmpdir != "" {
os.RemoveAll(tmpdir)
userKnownHostsFile string
userKey string
userKeyPub string
)
BeforeEach(func() {
tsaPort = 9800 + GinkgoParallelNode()
gardenPort := 9001 + GinkgoParallelNode()
gardenAddr = fmt.Sprintf("127.0.0.1:%d", gardenPort)
fakeBackend = new(gfakes.FakeBackend)
gardenServer = gserver.New("tcp", gardenAddr, 0, fakeBackend, lagertest.NewTestLogger("garden"))
err := gardenServer.Start()
Ω(err).ShouldNot(HaveOccurred())
atcServer = ghttp.NewServer()
hostKey, hostKeyPub = generateSSHKeypair()
userKnownHosts, err := ioutil.TempFile("", "known-hosts")
Ω(err).ShouldNot(HaveOccurred())
defer userKnownHosts.Close()
userKnownHostsFile = userKnownHosts.Name()
_, err = fmt.Fprintf(userKnownHosts, "[127.0.0.1]:%d ", tsaPort)
Ω(err).ShouldNot(HaveOccurred())
BeforeEach(func() {
fakeDB = new(fakes.FakeWorkerDB)
logger = lagertest.NewTestLogger("test")
workerA = new(gfakes.FakeBackend)
workerB = new(gfakes.FakeBackend)
workerAAddr = fmt.Sprintf("0.0.0.0:%d", 8888+GinkgoParallelNode())
workerBAddr = fmt.Sprintf("0.0.0.0:%d", 9999+GinkgoParallelNode())
workerAServer = server.New("tcp", workerAAddr, 0, workerA, logger)
workerBServer = server.New("tcp", workerBAddr, 0, workerB, logger)
err := workerAServer.Start()
Ω(err).ShouldNot(HaveOccurred())
err = workerBServer.Start()
Ω(err).ShouldNot(HaveOccurred())
provider = NewDBWorkerProvider(fakeDB, logger)
})
JustBeforeEach(func() {
workers, workersErr = provider.Workers()
})
AfterEach(func() {
workerAServer.Stop()
workerBServer.Stop()
var err error
tmpdir, err = ioutil.TempDir(os.TempDir(), "api-server-test")
Ω(err).ShouldNot(HaveOccurred())
socketPath = path.Join(tmpdir, "api.sock")
fakeBackend = new(fakes.FakeBackend)
serverBackend = fakeBackend
apiClient = client.New(connection.New("unix", socketPath))
})
JustBeforeEach(func() {
apiServer = server.New("unix", socketPath, 0, serverBackend, logger)
err := apiServer.Start()
Ω(err).ShouldNot(HaveOccurred())
Eventually(ErrorDialing("unix", socketPath)).ShouldNot(HaveOccurred())
})
It("stops accepting new connections", func() {
go apiServer.Stop()
Eventually(ErrorDialing("unix", socketPath)).Should(HaveOccurred())
})
It("stops the backend", func() {
apiServer.Stop()
Ω(fakeBackend.StopCallCount()).Should(Equal(1))