55 lines
1.5 KiB
Go
55 lines
1.5 KiB
Go
// License: GPLv3 Copyright: 2023, Kovid Goyal, <kovid at kovidgoyal.net>
|
|
|
|
package ssh
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"kitty/tools/utils/shlex"
|
|
|
|
"github.com/google/go-cmp/cmp"
|
|
)
|
|
|
|
var _ = fmt.Print
|
|
|
|
func TestGetSSHOptions(t *testing.T) {
|
|
m := SSHOptions()
|
|
if m["w"] != "local_tun[:remote_tun]" {
|
|
t.Fatalf("Unexpected set of SSH options: %#v", m)
|
|
}
|
|
}
|
|
|
|
func TestParseSSHArgs(t *testing.T) {
|
|
split := func(x string) []string {
|
|
ans, err := shlex.Split(x)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return ans
|
|
}
|
|
|
|
p := func(args, expected_ssh_args, expected_server_args, expected_extra_args string, expected_passthrough bool) {
|
|
ssh_args, server_args, passthrough, extra_args, err := ParseSSHArgs(split(args), "--kitten")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
check := func(a, b any) {
|
|
diff := cmp.Diff(a, b)
|
|
if diff != "" {
|
|
t.Fatalf("Unexpected value for args: %s\n%s", args, diff)
|
|
}
|
|
}
|
|
check(split(expected_ssh_args), ssh_args)
|
|
check(split(expected_server_args), server_args)
|
|
check(split(expected_extra_args), extra_args)
|
|
check(expected_passthrough, passthrough)
|
|
}
|
|
p(`localhost`, ``, `localhost`, ``, false)
|
|
p(`-- localhost`, ``, `localhost`, ``, false)
|
|
p(`-46p23 localhost sh -c "a b"`, `-4 -6 -p 23`, `localhost sh -c "a b"`, ``, false)
|
|
p(`-46p23 -S/moose -W x:6 -- localhost sh -c "a b"`, `-4 -6 -p 23 -S /moose -W x:6`, `localhost sh -c "a b"`, ``, false)
|
|
p(`--kitten=abc -np23 --kitten xyz host`, `-n -p 23`, `host`, `--kitten abc --kitten xyz`, true)
|
|
|
|
}
|