131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package saltmasterless
|
|
|
|
import (
|
|
"github.com/mitchellh/packer/packer"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func testConfig() map[string]interface{} {
|
|
return map[string]interface{}{
|
|
"local_state_tree": os.TempDir(),
|
|
}
|
|
}
|
|
|
|
func TestProvisioner_Impl(t *testing.T) {
|
|
var raw interface{}
|
|
raw = &Provisioner{}
|
|
if _, ok := raw.(packer.Provisioner); !ok {
|
|
t.Fatalf("must be a Provisioner")
|
|
}
|
|
}
|
|
|
|
func TestProvisionerPrepare_Defaults(t *testing.T) {
|
|
var p Provisioner
|
|
config := testConfig()
|
|
|
|
err := p.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
if p.config.TempConfigDir != DefaultTempConfigDir {
|
|
t.Errorf("unexpected temp config dir: %s", p.config.TempConfigDir)
|
|
}
|
|
}
|
|
|
|
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")
|
|
}
|
|
}
|
|
|
|
func TestProvisionerPrepare_MinionConfig(t *testing.T) {
|
|
var p Provisioner
|
|
config := testConfig()
|
|
|
|
config["minion_config"] = "/i/dont/exist/i/think"
|
|
err := p.Prepare(config)
|
|
if err == nil {
|
|
t.Fatal("should have error")
|
|
}
|
|
|
|
tf, err := ioutil.TempFile("", "minion")
|
|
if err != nil {
|
|
t.Fatalf("error tempfile: %s", err)
|
|
}
|
|
defer os.Remove(tf.Name())
|
|
|
|
config["minion_config"] = tf.Name()
|
|
err = p.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestProvisionerPrepare_LocalStateTree(t *testing.T) {
|
|
var p Provisioner
|
|
config := testConfig()
|
|
|
|
config["local_state_tree"] = "/i/dont/exist/i/think"
|
|
err := p.Prepare(config)
|
|
if err == nil {
|
|
t.Fatal("should have error")
|
|
}
|
|
|
|
config["local_state_tree"] = os.TempDir()
|
|
err = p.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestProvisionerPrepare_LocalPillarRoots(t *testing.T) {
|
|
var p Provisioner
|
|
config := testConfig()
|
|
|
|
config["local_pillar_roots"] = "/i/dont/exist/i/think"
|
|
err := p.Prepare(config)
|
|
if err == nil {
|
|
t.Fatal("should have error")
|
|
}
|
|
|
|
config["local_pillar_roots"] = os.TempDir()
|
|
err = p.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestProvisionerSudo(t *testing.T) {
|
|
var p Provisioner
|
|
config := testConfig()
|
|
|
|
err := p.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
withSudo := p.sudo("echo hello")
|
|
if withSudo != "sudo echo hello" {
|
|
t.Fatalf("sudo command not generated correctly")
|
|
}
|
|
|
|
config["disable_sudo"] = true
|
|
err = p.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
withoutSudo := p.sudo("echo hello")
|
|
if withoutSudo != "echo hello" {
|
|
t.Fatalf("sudo-less command not generated correctly")
|
|
}
|
|
}
|