From b6c3a4ffbed817e5999c21115a5bb020d1eecb57 Mon Sep 17 00:00:00 2001
From: Martino Facchin <m.facchin@arduino.cc>
Date: Mon, 24 Jul 2017 19:22:41 +0200
Subject: [PATCH] Add test to fix coverage

The test should anaylize the output of the command to make sure it worked but I didn't find any suitable function
---
 plugin_test.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

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{