|
|
|
|
@ -3,11 +3,8 @@ package remoteexec
|
|
|
|
|
import (
|
|
|
|
|
"bytes"
|
|
|
|
|
"io"
|
|
|
|
|
"strings"
|
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
|
|
"reflect"
|
|
|
|
|
|
|
|
|
|
"github.com/hashicorp/terraform/config"
|
|
|
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
|
|
|
"github.com/hashicorp/terraform/terraform"
|
|
|
|
|
@ -46,8 +43,6 @@ wget http://foobar
|
|
|
|
|
exit 0
|
|
|
|
|
`
|
|
|
|
|
|
|
|
|
|
var expectedInlineScriptsOut = strings.Split(expectedScriptOut, "\n")
|
|
|
|
|
|
|
|
|
|
func TestResourceProvider_generateScript(t *testing.T) {
|
|
|
|
|
p := Provisioner().(*schema.Provisioner)
|
|
|
|
|
conf := map[string]interface{}{
|
|
|
|
|
@ -63,7 +58,11 @@ func TestResourceProvider_generateScript(t *testing.T) {
|
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if reflect.DeepEqual(out, expectedInlineScriptsOut) {
|
|
|
|
|
if len(out) != 1 {
|
|
|
|
|
t.Fatal("expected 1 out")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if out[0] != expectedScriptOut {
|
|
|
|
|
t.Fatalf("bad: %v", out)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
@ -84,20 +83,18 @@ func TestResourceProvider_CollectScripts_inline(t *testing.T) {
|
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if len(scripts) != 3 {
|
|
|
|
|
if len(scripts) != 1 {
|
|
|
|
|
t.Fatalf("bad: %v", scripts)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
for i, script := range scripts {
|
|
|
|
|
var out bytes.Buffer
|
|
|
|
|
_, err = io.Copy(&out, script)
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
|
}
|
|
|
|
|
var out bytes.Buffer
|
|
|
|
|
_, err = io.Copy(&out, scripts[0])
|
|
|
|
|
if err != nil {
|
|
|
|
|
t.Fatalf("err: %v", err)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if out.String() != expectedInlineScriptsOut[i] {
|
|
|
|
|
t.Fatalf("bad: %v", out.String())
|
|
|
|
|
}
|
|
|
|
|
if out.String() != expectedScriptOut {
|
|
|
|
|
t.Fatalf("bad: %v", out.String())
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|