It("closes the listener when the client is closed", func() {
Eventually(fakeListenerFactory.ListenCallCount).Should(Equal(1))
Eventually(fakeLocalListener.AcceptCallCount).Should(Equal(1))
originalCloseCount := fakeLocalListener.CloseCallCount()
err := secureShell.Close()
Expect(err).NotTo(HaveOccurred())
Expect(fakeLocalListener.CloseCallCount()).Should(Equal(originalCloseCount + 1))
})
})
Context("when accept fails", func() {
var fakeConn *fake_net.FakeConn
BeforeEach(func() {
fakeConn = &fake_net.FakeConn{}
fakeConn.ReadReturns(0, io.EOF)
fakeListenerFactory.ListenReturns(fakeLocalListener, nil)
})
Context("with a permanent error", func() {
BeforeEach(func() {
fakeLocalListener.AcceptReturns(nil, errors.New("boom"))
})
It("stops trying to accept connections", func() {
Eventually(fakeLocalListener.AcceptCallCount).Should(Equal(1))
Consistently(fakeLocalListener.AcceptCallCount).Should(Equal(1))
Expect(fakeLocalListener.CloseCallCount()).To(Equal(1))
})
})
BeforeEach(func() {
logger = lagertest.NewTestLogger("test")
serverSSHConfig = &ssh.ServerConfig{
NoClientAuth: true,
}
serverSSHConfig.AddHostKey(TestHostKey)
})
Describe("HandleConnection", func() {
var fakeConn *fake_net.FakeConn
Context("when the function returns", func() {
BeforeEach(func() {
fakeConn = &fake_net.FakeConn{}
fakeConn.ReadReturns(0, errors.New("oops"))
sshd = daemon.New(logger, serverSSHConfig, nil, nil)
})
It("closes the connection", func() {
sshd.HandleConnection(fakeConn)
Expect(fakeConn.CloseCallCount()).To(BeNumerically(">=", 1))
})
})
Context("when an ssh client connects", func() {
var (
serverNetConn net.Conn
clientNetConn net.Conn