2013-05-24 01:38:40 -04:00
|
|
|
package shell
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/mitchellh/packer/packer"
|
2013-06-11 16:42:15 -04:00
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
2013-05-24 01:38:40 -04:00
|
|
|
"testing"
|
|
|
|
)
|
|
|
|
|
2013-06-11 16:42:15 -04:00
|
|
|
func testConfig() map[string]interface{} {
|
|
|
|
return map[string]interface{}{
|
|
|
|
"inline": []interface{}{"foo", "bar"},
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-05-24 01:38:40 -04:00
|
|
|
func TestProvisioner_Impl(t *testing.T) {
|
|
|
|
var raw interface{}
|
|
|
|
raw = &Provisioner{}
|
|
|
|
if _, ok := raw.(packer.Provisioner); !ok {
|
|
|
|
t.Fatalf("must be a Provisioner")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-05-27 18:06:34 -04:00
|
|
|
func TestProvisionerPrepare_Defaults(t *testing.T) {
|
2013-06-11 16:42:15 -04:00
|
|
|
var p Provisioner
|
|
|
|
config := testConfig()
|
2013-05-27 18:06:34 -04:00
|
|
|
|
2013-06-11 16:42:15 -04:00
|
|
|
err := p.Prepare(config)
|
2013-06-06 20:07:42 -04:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
2013-05-27 18:06:34 -04:00
|
|
|
|
|
|
|
if p.config.RemotePath != DefaultRemotePath {
|
|
|
|
t.Errorf("unexpected remote path: %s", p.config.RemotePath)
|
|
|
|
}
|
|
|
|
}
|
2013-06-11 16:42:15 -04:00
|
|
|
|
2013-07-07 20:52:20 -04:00
|
|
|
func TestProvisionerPrepare_InlineShebang(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "inline_shebang")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("should not have error: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if p.config.InlineShebang != "/bin/sh" {
|
|
|
|
t.Fatalf("bad value: %s", p.config.InlineShebang)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good one
|
|
|
|
config["inline_shebang"] = "foo"
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("should not have error: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if p.config.InlineShebang != "foo" {
|
|
|
|
t.Fatalf("bad value: %s", p.config.InlineShebang)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-07-13 20:28:56 -04:00
|
|
|
func TestProvisionerPrepare_InvalidKey(t *testing.T) {
|
|
|
|
var p Provisioner
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
// Add a random key
|
|
|
|
config["i_should_not_be_valid"] = true
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-27 13:56:46 -04:00
|
|
|
func TestProvisionerPrepare_Script(t *testing.T) {
|
2013-06-11 16:42:15 -04:00
|
|
|
config := testConfig()
|
|
|
|
delete(config, "inline")
|
|
|
|
|
2013-06-27 13:56:46 -04:00
|
|
|
config["script"] = "/this/should/not/exist"
|
2013-06-20 16:45:54 -04:00
|
|
|
p := new(Provisioner)
|
2013-06-11 16:42:15 -04:00
|
|
|
err := p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good one
|
|
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error tempfile: %s", err)
|
|
|
|
}
|
|
|
|
defer os.Remove(tf.Name())
|
|
|
|
|
2013-06-27 13:56:46 -04:00
|
|
|
config["script"] = tf.Name()
|
2013-06-20 16:45:54 -04:00
|
|
|
p = new(Provisioner)
|
2013-06-11 16:42:15 -04:00
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("should not have error: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-06-27 13:56:46 -04:00
|
|
|
func TestProvisionerPrepare_ScriptAndInline(t *testing.T) {
|
2013-06-11 16:42:15 -04:00
|
|
|
var p Provisioner
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "inline")
|
2013-06-27 13:56:46 -04:00
|
|
|
delete(config, "script")
|
2013-06-11 16:42:15 -04:00
|
|
|
err := p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with both
|
|
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error tempfile: %s", err)
|
|
|
|
}
|
|
|
|
defer os.Remove(tf.Name())
|
|
|
|
|
|
|
|
config["inline"] = []interface{}{"foo"}
|
2013-06-27 13:56:46 -04:00
|
|
|
config["script"] = tf.Name()
|
2013-06-11 16:42:15 -04:00
|
|
|
err = p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
}
|
2013-06-20 16:45:54 -04:00
|
|
|
|
2013-06-27 13:56:46 -04:00
|
|
|
func TestProvisionerPrepare_ScriptAndScripts(t *testing.T) {
|
2013-06-20 16:45:54 -04:00
|
|
|
var p Provisioner
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
// Test with both
|
|
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error tempfile: %s", err)
|
|
|
|
}
|
|
|
|
defer os.Remove(tf.Name())
|
|
|
|
|
|
|
|
config["inline"] = []interface{}{"foo"}
|
|
|
|
config["scripts"] = []string{tf.Name()}
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestProvisionerPrepare_Scripts(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
delete(config, "inline")
|
|
|
|
|
|
|
|
config["scripts"] = []string{}
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good one
|
|
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error tempfile: %s", err)
|
|
|
|
}
|
|
|
|
defer os.Remove(tf.Name())
|
|
|
|
|
|
|
|
config["scripts"] = []string{tf.Name()}
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("should not have error: %s", err)
|
|
|
|
}
|
|
|
|
}
|
2013-06-27 08:39:16 -04:00
|
|
|
|
|
|
|
func TestProvisionerPrepare_EnvironmentVars(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
// Test with a bad case
|
|
|
|
config["environment_vars"] = []string{"badvar", "good=var"}
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a trickier case
|
|
|
|
config["environment_vars"] = []string{"=bad"}
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should have error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good case
|
|
|
|
// Note: baz= is a real env variable, just empty
|
|
|
|
config["environment_vars"] = []string{"FOO=bar", "baz="}
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("should not have error: %s", err)
|
|
|
|
}
|
|
|
|
}
|