cicd/drone-ssh
1
0
Fork 0
mirror of https://github.com/appleboy/drone-ssh.git synced 2025-09-13 20:00:10 +08:00
Code Issues Packages Projects Releases 1 Wiki Activity

test: update time out

Browse Source 
Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu 2019-03-11 11:45:39 +08:00
parent 1098d26f7e
commit 600b185e6d
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
plugin_test.go
View File

@ -5,6 +5,7 @@ import (
"os" "os"
"strings" "strings"
"testing" "testing"
"time"
"github.com/appleboy/easyssh-proxy" "github.com/appleboy/easyssh-proxy"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
@ -59,7 +60,7 @@ func TestIncorrectPassword(t *testing.T) {
Port: 22, Port: 22,
Password: "123456", Password: "123456",
Script: []string{"whoami"}, Script: []string{"whoami"},
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
}, },
} }
@ -73,7 +74,7 @@ func TestSSHScriptFromRawKey(t *testing.T) {
Host: []string{"localhost"}, Host: []string{"localhost"},
Username: "drone-scp", Username: "drone-scp",
Port: 22, Port: 22,
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
Key: `-----BEGIN RSA PRIVATE KEY----- Key: `-----BEGIN RSA PRIVATE KEY-----
MIIEpAIBAAKCAQEA4e2D/qPN08pzTac+a8ZmlP1ziJOXk45CynMPtva0rtK/RB26 MIIEpAIBAAKCAQEA4e2D/qPN08pzTac+a8ZmlP1ziJOXk45CynMPtva0rtK/RB26
VbfAF0hIJji7ltvnYnqCU9oFfvEM33cTn7T96+od8ib/Vz25YU8ZbstqtIskPuwC VbfAF0hIJji7ltvnYnqCU9oFfvEM33cTn7T96+od8ib/Vz25YU8ZbstqtIskPuwC
@ -118,7 +119,7 @@ func TestSSHScriptFromKeyFile(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"whoami", "ls -al"}, Script: []string{"whoami", "ls -al"},
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
}, },
} }
@ -134,7 +135,7 @@ func TestStreamFromSSHCommand(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"whoami", "for i in {1..5}; do echo ${i}; sleep 1; done", "echo 'done'"}, Script: []string{"whoami", "for i in {1..5}; do echo ${i}; sleep 1; done", "echo 'done'"},
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
}, },
} }
@ -150,7 +151,7 @@ func TestSSHScriptWithError(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"exit 1"}, Script: []string{"exit 1"},
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
}, },
} }
@ -167,7 +168,7 @@ func TestSSHCommandTimeOut(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"sleep 5"}, Script: []string{"sleep 5"},
CommandTimeout: 1, CommandTimeout: 1 * time.Second,
}, },
} }
@ -183,7 +184,7 @@ func TestProxyCommand(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"whoami"}, Script: []string{"whoami"},
CommandTimeout: 1, CommandTimeout: 1 * time.Second,
Proxy: easyssh.DefaultConfig{ Proxy: easyssh.DefaultConfig{
Server: "localhost", Server: "localhost",
User: "drone-scp", User: "drone-scp",
@ -205,7 +206,7 @@ func TestSSHCommandError(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"mkdir a", "mkdir a"}, Script: []string{"mkdir a", "mkdir a"},
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
}, },
} }
@ -227,7 +228,7 @@ func TestSSHCommandExitCodeError(t *testing.T) {
"mkdir a", "mkdir a",
"echo 2", "echo 2",
}, },
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
}, },
} }
@ -247,7 +248,7 @@ func TestSetENV(t *testing.T) {
Envs: []string{"foo"}, Envs: []string{"foo"},
Debug: true, Debug: true,
Script: []string{"whoami; echo $FOO"}, Script: []string{"whoami; echo $FOO"},
CommandTimeout: 1, CommandTimeout: 1 * time.Second,
Proxy: easyssh.DefaultConfig{ Proxy: easyssh.DefaultConfig{
Server: "localhost", Server: "localhost",
User: "drone-scp", User: "drone-scp",
@ -274,7 +275,7 @@ func TestSetExistingENV(t *testing.T) {
Envs: []string{"foo", "bar", "baz"}, Envs: []string{"foo", "bar", "baz"},
Debug: true, Debug: true,
Script: []string{"export FOO", "export BAR", "export BAZ", "env | grep -q '^FOO=Value for foo$'", "env | grep -q '^BAR=$'", "if env | grep -q BAZ; then false; else true; fi"}, Script: []string{"export FOO", "export BAR", "export BAZ", "env | grep -q '^FOO=Value for foo$'", "env | grep -q '^BAR=$'", "if env | grep -q BAZ; then false; else true; fi"},
CommandTimeout: 1, CommandTimeout: 1 * time.Second,
Proxy: easyssh.DefaultConfig{ Proxy: easyssh.DefaultConfig{
Server: "localhost", Server: "localhost",
User: "drone-scp", User: "drone-scp",
@ -296,7 +297,7 @@ func TestSyncMode(t *testing.T) {
Port: 22, Port: 22,
KeyPath: "./tests/.ssh/id_rsa", KeyPath: "./tests/.ssh/id_rsa",
Script: []string{"whoami", "for i in {1..3}; do echo ${i}; sleep 1; done", "echo 'done'"}, Script: []string{"whoami", "for i in {1..3}; do echo ${i}; sleep 1; done", "echo 'done'"},
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
Sync: true, Sync: true,
}, },
} }
@ -371,7 +372,7 @@ func TestCommandOutput(t *testing.T) {
"whoami", "whoami",
"uname", "uname",
}, },
CommandTimeout: 60, CommandTimeout: 60 * time.Second,
Sync: true, Sync: true,
}, },
Writer: &buffer, Writer: &buffer,
@ -405,7 +406,7 @@ func TestScriptStop(t *testing.T) {
"mkdir a/b/c", "mkdir a/b/c",
"mkdir d/e/f", "mkdir d/e/f",
}, },
CommandTimeout: 10, CommandTimeout: 10 * time.Second,
ScriptStop: true, ScriptStop: true,
}, },
Writer: &buffer, Writer: &buffer,
@ -440,7 +441,7 @@ func TestNoneScriptStop(t *testing.T) {
"mkdir a/b/c", "mkdir a/b/c",
"mkdir d/e/f", "mkdir d/e/f",
}, },
CommandTimeout: 10, CommandTimeout: 10 * time.Second,
}, },
Writer: &buffer, Writer: &buffer,
} }
@ -508,7 +509,7 @@ func TestEnvOutput(t *testing.T) {
`echo "[${ENV_6}]"`, `echo "[${ENV_6}]"`,
`echo "[${ENV_7}]"`, `echo "[${ENV_7}]"`,
}, },
CommandTimeout: 10, CommandTimeout: 10 * time.Second,
Proxy: easyssh.DefaultConfig{ Proxy: easyssh.DefaultConfig{
Server: "localhost", Server: "localhost",
User: "drone-scp", User: "drone-scp",