BeforeEach(func() {
logger = lagertest.NewTestLogger("test")
})
Describe("HandleConnection", func() {
var (
proxyAuthenticator *fake_authenticators.FakePasswordAuthenticator
proxySSHConfig *ssh.ServerConfig
sshProxy *proxy.Proxy
daemonTargetConfig proxy.TargetConfig
daemonAuthenticator *fake_authenticators.FakePasswordAuthenticator
daemonSSHConfig *ssh.ServerConfig
daemonGlobalRequestHandlers map[string]handlers.GlobalRequestHandler
daemonNewChannelHandlers map[string]handlers.NewChannelHandler
sshDaemon *daemon.Daemon
proxyListener net.Listener
sshdListener net.Listener
proxyAddress string
daemonAddress string
proxyServer *server.Server
sshdServer *server.Server
)
BeforeEach(func() {
fakeLogSender = fake_logs.NewFakeLogSender()
logs.Initialize(fakeLogSender)
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/pivotal-golang/lager"
"github.com/pivotal-golang/lager/lagertest"
"golang.org/x/crypto/ssh"
)
var _ = Describe("DirectTcpipChannelHandler", func() {
var (
sshd *daemon.Daemon
client *ssh.Client
logger *lagertest.TestLogger
serverSSHConfig *ssh.ServerConfig
handler *fake_handlers.FakeNewChannelHandler
testHandler *handlers.DirectTcpipChannelHandler
testDialer *fakes.FakeDialer
echoHandler *fake_server.FakeConnectionHandler
echoServer *server.Server
echoAddress string
)
BeforeEach(func() {
logger = lagertest.NewTestLogger("test")
echoHandler = &fake_server.FakeConnectionHandler{}
echoHandler.HandleConnectionStub = func(conn net.Conn) {
io.Copy(conn, conn)
conn.Close()
"github.com/cloudfoundry-incubator/diego-ssh/test_helpers"
"github.com/cloudfoundry-incubator/diego-ssh/test_helpers/fake_net"
"github.com/pivotal-golang/lager"
"github.com/pivotal-golang/lager/lagertest"
"github.com/tedsuo/ifrit"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/onsi/gomega/gbytes"
)
var _ = Describe("Server", func() {
var (
logger lager.Logger
srv *server.Server
handler *fakes.FakeConnectionHandler
address string
)
BeforeEach(func() {
handler = &fakes.FakeConnectionHandler{}
address = fmt.Sprintf("127.0.0.1:%d", 7001+GinkgoParallelNode())
logger = lagertest.NewTestLogger("test")
})
Describe("Run", func() {
var process ifrit.Process
BeforeEach(func() {
srv = server.NewServer(logger, address, handler)