mirror of
https://github.com/gravitational/teleport.git
synced 2026-10-31 03:27:49 +08:00
Andrew Burke
491130b20c
* Add remote port forwarding for Teleport nodes This change adds support for remote port forwarding (ssh -R) for Teleport nodes. * Fix windows build
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
/*
|
|
* Teleport
|
|
* Copyright (C) 2024 Gravitational, Inc.
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package sshutils
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestJoinHostPort(t *testing.T) {
|
|
t.Parallel()
|
|
assert.Equal(t, "example.com:1234", JoinHostPort("example.com", 1234))
|
|
}
|
|
|
|
func TestSplitHostPort(t *testing.T) {
|
|
t.Parallel()
|
|
tests := []struct {
|
|
name string
|
|
addr string
|
|
expectHost string
|
|
expectPort uint32
|
|
assertErr assert.ErrorAssertionFunc
|
|
}{
|
|
{
|
|
name: "empty",
|
|
assertErr: assert.Error,
|
|
},
|
|
{
|
|
name: "full host and port",
|
|
addr: "example.com:1234",
|
|
expectHost: "example.com",
|
|
expectPort: 1234,
|
|
assertErr: assert.NoError,
|
|
},
|
|
{
|
|
name: "without port",
|
|
addr: "example.com",
|
|
expectHost: "example.com",
|
|
assertErr: assert.NoError,
|
|
},
|
|
{
|
|
name: "ipv6 addr",
|
|
addr: "[::1]:80",
|
|
expectHost: "::1",
|
|
expectPort: 80,
|
|
assertErr: assert.NoError,
|
|
},
|
|
{
|
|
name: "ipv6 addr without port",
|
|
addr: "[::1]",
|
|
expectHost: "::1",
|
|
assertErr: assert.NoError,
|
|
},
|
|
}
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
host, port, err := SplitHostPort(tc.addr)
|
|
tc.assertErr(t, err)
|
|
assert.Equal(t, tc.expectHost, host)
|
|
assert.Equal(t, tc.expectPort, port)
|
|
})
|
|
}
|
|
}
|