}, nil)
fakeInfoFactory.GetReturns(info.Info{}, nil)
fakeCredFactory.GetReturns(credential.Credential{}, nil)
fakeSecureClient.DialStub = net.Dial
})
JustBeforeEach(func() {
connectErr := secureShell.Connect(opts)
Expect(connectErr).NotTo(HaveOccurred())
localForwardError = secureShell.LocalPortForward()
})
AfterEach(func() {
err := secureShell.Close()
Expect(err).NotTo(HaveOccurred())
echoServer.Shutdown()
realLocalListener.Close()
})
validateConnectivity := func(addr string) {
conn, err := net.Dial("tcp", addr)
Expect(err).NotTo(HaveOccurred())
msg := fmt.Sprintf("Hello from %s\n", addr)
n, err := conn.Write([]byte(msg))
Expect(err).NotTo(HaveOccurred())
Expect(n).To(Equal(len(msg)))
