167 lines
3.2 KiB
Go
167 lines
3.2 KiB
Go
package remoteexec
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/config"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
)
|
|
|
|
func TestResourceProvisioner_impl(t *testing.T) {
|
|
var _ terraform.ResourceProvisioner = new(ResourceProvisioner)
|
|
}
|
|
|
|
func TestResourceProvider_Validate_good(t *testing.T) {
|
|
c := testConfig(t, map[string]interface{}{
|
|
"inline": "echo foo",
|
|
})
|
|
p := new(ResourceProvisioner)
|
|
warn, errs := p.Validate(c)
|
|
if len(warn) > 0 {
|
|
t.Fatalf("Warnings: %v", warn)
|
|
}
|
|
if len(errs) > 0 {
|
|
t.Fatalf("Errors: %v", errs)
|
|
}
|
|
}
|
|
|
|
func TestResourceProvider_Validate_bad(t *testing.T) {
|
|
c := testConfig(t, map[string]interface{}{
|
|
"invalid": "nope",
|
|
})
|
|
p := new(ResourceProvisioner)
|
|
warn, errs := p.Validate(c)
|
|
if len(warn) > 0 {
|
|
t.Fatalf("Warnings: %v", warn)
|
|
}
|
|
if len(errs) == 0 {
|
|
t.Fatalf("Should have errors")
|
|
}
|
|
}
|
|
|
|
func TestResourceProvider_generateScript(t *testing.T) {
|
|
p := new(ResourceProvisioner)
|
|
conf := testConfig(t, map[string]interface{}{
|
|
"inline": []string{
|
|
"cd /tmp",
|
|
"wget http://foobar",
|
|
"exit 0",
|
|
},
|
|
})
|
|
out, err := p.generateScript(conf)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if out != expectedScriptOut {
|
|
t.Fatalf("bad: %v", out)
|
|
}
|
|
}
|
|
|
|
var expectedScriptOut = `#!/bin/sh
|
|
cd /tmp
|
|
wget http://foobar
|
|
exit 0
|
|
`
|
|
|
|
func TestResourceProvider_CollectScripts_inline(t *testing.T) {
|
|
p := new(ResourceProvisioner)
|
|
conf := testConfig(t, map[string]interface{}{
|
|
"inline": []string{
|
|
"cd /tmp",
|
|
"wget http://foobar",
|
|
"exit 0",
|
|
},
|
|
})
|
|
|
|
scripts, err := p.collectScripts(conf)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if len(scripts) != 1 {
|
|
t.Fatalf("bad: %v", scripts)
|
|
}
|
|
|
|
var out bytes.Buffer
|
|
_, err = io.Copy(&out, scripts[0])
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if string(out.Bytes()) != expectedScriptOut {
|
|
t.Fatalf("bad: %v", out.Bytes())
|
|
}
|
|
}
|
|
|
|
func TestResourceProvider_CollectScripts_script(t *testing.T) {
|
|
p := new(ResourceProvisioner)
|
|
conf := testConfig(t, map[string]interface{}{
|
|
"script": "test-fixtures/script1.sh",
|
|
})
|
|
|
|
scripts, err := p.collectScripts(conf)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if len(scripts) != 1 {
|
|
t.Fatalf("bad: %v", scripts)
|
|
}
|
|
|
|
var out bytes.Buffer
|
|
_, err = io.Copy(&out, scripts[0])
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if string(out.Bytes()) != expectedScriptOut {
|
|
t.Fatalf("bad: %v", out.Bytes())
|
|
}
|
|
}
|
|
|
|
func TestResourceProvider_CollectScripts_scripts(t *testing.T) {
|
|
p := new(ResourceProvisioner)
|
|
conf := testConfig(t, map[string]interface{}{
|
|
"scripts": []interface{}{
|
|
"test-fixtures/script1.sh",
|
|
"test-fixtures/script1.sh",
|
|
"test-fixtures/script1.sh",
|
|
},
|
|
})
|
|
|
|
scripts, err := p.collectScripts(conf)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if len(scripts) != 3 {
|
|
t.Fatalf("bad: %v", scripts)
|
|
}
|
|
|
|
for idx := range scripts {
|
|
var out bytes.Buffer
|
|
_, err = io.Copy(&out, scripts[idx])
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if string(out.Bytes()) != expectedScriptOut {
|
|
t.Fatalf("bad: %v", out.Bytes())
|
|
}
|
|
}
|
|
}
|
|
|
|
func testConfig(
|
|
t *testing.T,
|
|
c map[string]interface{}) *terraform.ResourceConfig {
|
|
r, err := config.NewRawConfig(c)
|
|
if err != nil {
|
|
t.Fatalf("bad: %s", err)
|
|
}
|
|
|
|
return terraform.NewResourceConfig(r)
|
|
}
|