2013-09-08 01:27:25 -04:00
|
|
|
package puppetmasterless
|
2013-08-01 20:05:23 -04:00
|
|
|
|
|
|
|
import (
|
2013-09-08 02:31:28 -04:00
|
|
|
"io/ioutil"
|
|
|
|
"os"
|
2013-08-01 20:05:23 -04:00
|
|
|
"testing"
|
2015-11-03 14:36:04 -05:00
|
|
|
|
|
|
|
"github.com/mitchellh/packer/packer"
|
2013-08-01 20:05:23 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
func testConfig() map[string]interface{} {
|
2013-09-08 02:31:28 -04:00
|
|
|
tf, err := ioutil.TempFile("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
panic(err)
|
|
|
|
}
|
|
|
|
|
2013-08-01 20:05:23 -04:00
|
|
|
return map[string]interface{}{
|
2013-09-08 02:31:28 -04:00
|
|
|
"manifest_file": tf.Name(),
|
2013-08-01 20:05:23 -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-09-08 02:31:28 -04:00
|
|
|
|
2013-09-09 16:24:17 -04:00
|
|
|
func TestProvisionerPrepare_hieraConfigPath(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "hiera_config_path")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// 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["hiera_config_path"] = tf.Name()
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-08 02:31:28 -04:00
|
|
|
func TestProvisionerPrepare_manifestFile(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "manifest_file")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should be an 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["manifest_file"] = tf.Name()
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-11-25 02:52:41 -05:00
|
|
|
func TestProvisionerPrepare_manifestDir(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "manifestdir")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good one
|
2013-12-11 14:21:51 -05:00
|
|
|
td, err := ioutil.TempDir("", "packer")
|
2013-11-25 02:52:41 -05:00
|
|
|
if err != nil {
|
2013-12-11 14:21:51 -05:00
|
|
|
t.Fatalf("error: %s", err)
|
2013-11-25 02:52:41 -05:00
|
|
|
}
|
2013-12-11 14:21:51 -05:00
|
|
|
defer os.RemoveAll(td)
|
2013-11-25 02:52:41 -05:00
|
|
|
|
2013-12-11 14:21:51 -05:00
|
|
|
config["manifest_dir"] = td
|
2013-11-25 02:52:41 -05:00
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-09-08 02:31:28 -04:00
|
|
|
func TestProvisionerPrepare_modulePaths(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "module_paths")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with bad paths
|
|
|
|
config["module_paths"] = []string{"i-should-not-exist"}
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should be an error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good one
|
|
|
|
td, err := ioutil.TempDir("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error: %s", err)
|
|
|
|
}
|
|
|
|
defer os.RemoveAll(td)
|
|
|
|
|
|
|
|
config["module_paths"] = []string{td}
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|
2015-02-24 22:52:09 -05:00
|
|
|
|
|
|
|
func TestProvisionerPrepare_facterFacts(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "facter")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with malformed fact
|
|
|
|
config["facter"] = "fact=stringified"
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should be an error")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with a good one
|
|
|
|
td, err := ioutil.TempDir("", "packer")
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("error: %s", err)
|
|
|
|
}
|
|
|
|
defer os.RemoveAll(td)
|
|
|
|
|
2015-02-24 22:56:37 -05:00
|
|
|
facts := make(map[string]string)
|
2015-02-24 22:52:09 -05:00
|
|
|
facts["fact_name"] = "fact_value"
|
2015-02-24 22:56:37 -05:00
|
|
|
config["facter"] = facts
|
2015-02-24 22:52:09 -05:00
|
|
|
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
2015-02-24 22:56:37 -05:00
|
|
|
// Make sure the default facts are present
|
2015-02-24 22:52:09 -05:00
|
|
|
delete(config, "facter")
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if p.config.Facter == nil {
|
|
|
|
t.Fatalf("err: Default facts are not set in the Puppet provisioner!")
|
|
|
|
}
|
|
|
|
}
|
2015-11-03 14:36:04 -05:00
|
|
|
|
|
|
|
func TestProvisionerPrepare_options(t *testing.T) {
|
|
|
|
config := testConfig()
|
|
|
|
|
|
|
|
delete(config, "options")
|
|
|
|
p := new(Provisioner)
|
|
|
|
err := p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
// Test with malformed fact
|
|
|
|
config["options"] = "{{}}"
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err == nil {
|
|
|
|
t.Fatal("should be an error")
|
|
|
|
}
|
|
|
|
|
|
|
|
config["options"] = []string{
|
|
|
|
"arg",
|
|
|
|
}
|
|
|
|
|
|
|
|
p = new(Provisioner)
|
|
|
|
err = p.Prepare(config)
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("err: %s", err)
|
|
|
|
}
|
|
|
|
}
|