SSH Proxy support (#2736)

* initial commit
This commit is contained in:
Irena Rindos authored and GitHub committed 2023-01-09 13:43:02 -05:00
1 parent 1fb960575f
commit beb97ac9f2
9 files changed
+70 -36

No files matched your search

+6 -9
View File
@@ -35,9 +35,8 @@ func TestRegisterHandler(t *testing.T) {
require.NoError(err)
}
func TestGetHandler(t *testing.T) {
func TestAlwaysTcpGetHandler(t *testing.T) {
assert, require := assert.New(t), require.New(t)
fn := func(context.Context, net.Conn, *ProxyDialer, string, *anypb.Any) (ProxyConnFn, error) {
return nil, nil
}
@@ -47,15 +46,13 @@ func TestGetHandler(t *testing.T) {
})
handlers = sync.Map{}
err := RegisterHandler("fn", fn)
require.NoError(err)
gotFn, err := GetHandler("fake")
_, err := tcpOnly("wid", nil)
require.Error(err)
assert.ErrorIs(err, ErrUnknownProtocol)
assert.Nil(gotFn)
gotFn, err = GetHandler("fn")
require.NoError(RegisterHandler("tcp", fn))
handler, err := tcpOnly("wid", nil)
require.NoError(err)
assert.NotNil(gotFn)
require.NotNil(handler)
}