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
}).Should(HaveOccurred())
})
Context("when we call to get multiple workers", func() {
JustBeforeEach(func() {
workers, workersErr = provider.Workers()
})
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()
Eventually(func() error {
conn, err := net.Dial("tcp", workerAAddr)
if err == nil {
conn.Close()
}
return err
}).Should(HaveOccurred())
Eventually(func() error {
conn, err := net.Dial("tcp", workerBAddr)
if err == nil {
conn.Close()
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)
}
})
Context("and the client sends a PingRequest", func() {
Context("and the backend ping succeeds", func() {
It("does not error", func() {
Ω(apiClient.Ping()).ShouldNot(HaveOccurred())
})
})
Context("when the backend ping fails", func() {
BeforeEach(func() {
"-forwardHost", forwardHost,
)
tsaRunner := ginkgomon.New(ginkgomon.Config{
Command: tsaCommand,
Name: "tsa",
StartCheck: "tsa.listening",
AnsiColorCode: "32m",
})
tsaProcess = ginkgomon.Invoke(tsaRunner)
})
AfterEach(func() {
atcServer.Close()
gardenServer.Stop()
ginkgomon.Interrupt(tsaProcess)
})
Describe("SSHing", func() {
var sshSess *gexec.Session
var sshStdin io.Writer
var sshArgv []string
BeforeEach(func() {
sshArgv = []string{
"127.0.0.1",
"-p", strconv.Itoa(tsaPort),
"-o", "UserKnownHostsFile=" + userKnownHostsFile,
}
})
