142 lines
3.5 KiB
Go
142 lines
3.5 KiB
Go
package ssh
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gotest.tools/v3/assert"
|
|
is "gotest.tools/v3/assert/cmp"
|
|
)
|
|
|
|
func TestParseURL(t *testing.T) {
|
|
testCases := []struct {
|
|
doc string
|
|
url string
|
|
remoteCommand []string
|
|
expectedArgs []string
|
|
expectedError string
|
|
expectedSpec Spec
|
|
}{
|
|
{
|
|
doc: "bare ssh URL",
|
|
url: "ssh://example.com",
|
|
expectedArgs: []string{
|
|
"--", "example.com",
|
|
},
|
|
expectedSpec: Spec{
|
|
Host: "example.com",
|
|
},
|
|
},
|
|
{
|
|
doc: "bare ssh URL and remote command",
|
|
url: "ssh://example.com",
|
|
remoteCommand: []string{
|
|
"docker", "system", "dial-stdio",
|
|
},
|
|
expectedArgs: []string{
|
|
"--", "example.com",
|
|
"docker", "system", "dial-stdio",
|
|
},
|
|
expectedSpec: Spec{
|
|
Host: "example.com",
|
|
},
|
|
},
|
|
{
|
|
doc: "ssh URL with path and remote command and flag",
|
|
url: "ssh://example.com/var/run/docker.sock",
|
|
remoteCommand: []string{
|
|
"docker", "--host", "unix:///var/run/docker.sock", "system", "dial-stdio",
|
|
},
|
|
expectedArgs: []string{
|
|
"--", "example.com",
|
|
"docker", "--host", "unix:///var/run/docker.sock", "system", "dial-stdio",
|
|
},
|
|
expectedSpec: Spec{
|
|
Host: "example.com",
|
|
Path: "/var/run/docker.sock",
|
|
},
|
|
},
|
|
{
|
|
doc: "ssh URL with username and port",
|
|
url: "ssh://me@example.com:10022",
|
|
expectedArgs: []string{
|
|
"-l", "me",
|
|
"-p", "10022",
|
|
"--", "example.com",
|
|
},
|
|
expectedSpec: Spec{
|
|
User: "me",
|
|
Host: "example.com",
|
|
Port: "10022",
|
|
},
|
|
},
|
|
{
|
|
doc: "ssh URL with username, port, and path",
|
|
url: "ssh://me@example.com:10022/var/run/docker.sock",
|
|
expectedArgs: []string{
|
|
"-l", "me",
|
|
"-p", "10022",
|
|
"--", "example.com",
|
|
},
|
|
expectedSpec: Spec{
|
|
User: "me",
|
|
Host: "example.com",
|
|
Port: "10022",
|
|
Path: "/var/run/docker.sock",
|
|
},
|
|
},
|
|
{
|
|
doc: "malformed URL",
|
|
url: "malformed %%url",
|
|
expectedError: `invalid SSH URL: invalid URL escape "%%u"`,
|
|
},
|
|
{
|
|
doc: "URL missing scheme",
|
|
url: "no-scheme.example.com",
|
|
expectedError: "invalid SSH URL: no scheme provided",
|
|
},
|
|
{
|
|
doc: "invalid URL with password",
|
|
url: "ssh://me:passw0rd@example.com",
|
|
expectedError: "invalid SSH URL: plain-text password is not supported",
|
|
},
|
|
{
|
|
doc: "invalid URL with query parameter",
|
|
url: "ssh://example.com?foo=bar&bar=baz",
|
|
expectedError: `invalid SSH URL: query parameters are not allowed: "foo=bar&bar=baz"`,
|
|
},
|
|
{
|
|
doc: "invalid URL with fragment",
|
|
url: "ssh://example.com#bar",
|
|
expectedError: `invalid SSH URL: fragments are not allowed: "bar"`,
|
|
},
|
|
{
|
|
doc: "invalid URL without hostname",
|
|
url: "ssh://",
|
|
expectedError: "invalid SSH URL: hostname is empty",
|
|
},
|
|
{
|
|
url: "ssh:///no-hostname",
|
|
expectedError: "invalid SSH URL: hostname is empty",
|
|
},
|
|
{
|
|
doc: "invalid URL with unsupported scheme",
|
|
url: "https://example.com",
|
|
expectedError: `invalid SSH URL: incorrect scheme: https`,
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.doc, func(t *testing.T) {
|
|
sp, err := ParseURL(tc.url)
|
|
if tc.expectedError == "" {
|
|
assert.NilError(t, err)
|
|
actualArgs := sp.Args(tc.remoteCommand...)
|
|
assert.Check(t, is.DeepEqual(actualArgs, tc.expectedArgs))
|
|
assert.Check(t, is.Equal(*sp, tc.expectedSpec))
|
|
} else {
|
|
assert.Check(t, is.Error(err, tc.expectedError))
|
|
assert.Check(t, is.Nil(sp))
|
|
}
|
|
})
|
|
}
|
|
}
|