It("removes the HostKeyCallback from the client config", func() {
Expect(fakeSecureDialer.DialCallCount()).To(Equal(1))
_, _, config := fakeSecureDialer.DialArgsForCall(0)
Expect(config.HostKeyCallback).To(BeNil())
})
})
Context("when dialing is successful", func() {
BeforeEach(func() {
fakeTerminalHelper.StdStreamsStub = terminalHelper.StdStreams
terminalHelper = fakeTerminalHelper
})
It("creates a new secure shell session", func() {
Expect(fakeSecureClient.NewSessionCallCount()).To(Equal(1))
})
It("closes the session", func() {
Expect(fakeSecureSession.CloseCallCount()).To(Equal(1))
})
It("allocates standard streams", func() {
Expect(fakeTerminalHelper.StdStreamsCallCount()).To(Equal(1))
})
It("gets a stdin pipe for the session", func() {
Expect(fakeSecureSession.StdinPipeCallCount()).To(Equal(1))
})
Context("when getting the stdin pipe fails", func() {
