diff --git a/plugin_test.go b/plugin_test.go index 8fe9709..3d3653d 100644 --- a/plugin_test.go +++ b/plugin_test.go @@ -1,6 +1,7 @@ package main import ( + "os" "testing" "github.com/appleboy/easyssh-proxy" @@ -137,6 +138,23 @@ func TestStreamFromSSHCommand(t *testing.T) { assert.Nil(t, err) } +func TestDroneEnvSubstitution(t *testing.T) { + os.Setenv("DRONE_DEPLOY_TO", "dev") + plugin := Plugin{ + Config: Config{ + Host: []string{"localhost", "127.0.0.1"}, + UserName: "drone-scp", + Port: 22, + KeyPath: "./tests/.ssh/id_rsa", + Script: []string{"echo $DRONE_DEPLOY_TO $PATH"}, + CommandTimeout: 60, + }, + } + + err := plugin.Exec() + assert.Nil(t, err) +} + func TestSSHScriptWithError(t *testing.T) { plugin := Plugin{ Config: Config{