69 lines
1.9 KiB
Go
69 lines
1.9 KiB
Go
// License: GPLv3 Copyright: 2023, Kovid Goyal, <kovid at kovidgoyal.net>
|
|
|
|
package ssh
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"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)
|
|
}
|
|
|
|
func TestRelevantKittyOpts(t *testing.T) {
|
|
tdir := t.TempDir()
|
|
path := filepath.Join(tdir, "kitty.conf")
|
|
os.WriteFile(path, []byte("term XXX\nshell_integration changed\nterm abcd"), 0o600)
|
|
rko := read_relevant_kitty_opts(path)
|
|
if rko.Term != "abcd" {
|
|
t.Fatalf("Unexpected TERM: %s", RelevantKittyOpts().Term)
|
|
}
|
|
if rko.Shell_integration != "changed" {
|
|
t.Fatalf("Unexpected shell_integration: %s", RelevantKittyOpts().Shell_integration)
|
|
}
|
|
}
|