diff --git a/pkg/client/connector/sshcli_test.go b/pkg/client/connector/sshcli_test.go index aca58375..d6062125 100644 --- a/pkg/client/connector/sshcli_test.go +++ b/pkg/client/connector/sshcli_test.go @@ -23,7 +23,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { Host: "example.com", }, controlSockPath: "/tmp/test.sock", - expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-T", "root@example.com"}, + expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "root@example.com"}, }, { name: "basic connection without control socket", @@ -32,7 +32,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { Host: "example.com", }, controlSockPath: "", - expected: []string{"-o", "ConnectTimeout=5", "-T", "root@example.com"}, + expected: []string{"-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "root@example.com"}, }, { name: "with custom port", @@ -42,7 +42,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { Port: 2222, }, controlSockPath: "/tmp/test.sock", - expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-T", "-p", "2222", "root@example.com"}, + expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "-p", "2222", "root@example.com"}, }, { name: "with identity file", @@ -52,7 +52,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { KeyPath: "/path/to/key", }, controlSockPath: "/tmp/test.sock", - expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-T", "-i", "/path/to/key", "root@example.com"}, + expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "-i", "/path/to/key", "root@example.com"}, }, { name: "all options combined", @@ -64,7 +64,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { SockPath: "/custom/path/uncloud.sock", }, controlSockPath: "/tmp/test.sock", - expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-T", "-p", "2222", "-i", "/path/to/key", "root@example.com"}, + expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "-p", "2222", "-i", "/path/to/key", "root@example.com"}, }, { name: "port 0 not included", @@ -74,7 +74,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { Port: 0, }, controlSockPath: "/tmp/test.sock", - expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-T", "root@example.com"}, + expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "root@example.com"}, }, { name: "port 22 included when explicit", @@ -84,7 +84,7 @@ func TestSSHCLIConnector_buildSSHArgs(t *testing.T) { Port: 22, }, controlSockPath: "/tmp/test.sock", - expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-T", "-p", "22", "root@example.com"}, + expected: []string{"-o", "ControlMaster=auto", "-o", "ControlPath=/tmp/test.sock", "-o", "ControlPersist=10m", "-o", "ConnectTimeout=5", "-o", "BatchMode=yes", "-T", "-p", "22", "root@example.com"}, }, }