109 lines
2.0 KiB
Go
109 lines
2.0 KiB
Go
package amazonebs
|
|
|
|
import (
|
|
"github.com/mitchellh/packer/packer"
|
|
"testing"
|
|
)
|
|
|
|
func testConfig() map[string]interface{} {
|
|
return map[string]interface{}{
|
|
"access_key": "foo",
|
|
"secret_key": "bar",
|
|
}
|
|
}
|
|
|
|
func TestBuilder_ImplementsBuilder(t *testing.T) {
|
|
var raw interface{}
|
|
raw = &Builder{}
|
|
if _, ok := raw.(packer.Builder); !ok {
|
|
t.Fatalf("Builder should be a builder")
|
|
}
|
|
}
|
|
|
|
func TestBuilder_Prepare_BadType(t *testing.T) {
|
|
b := &Builder{}
|
|
c := map[string]interface{}{
|
|
"access_key": []string{},
|
|
}
|
|
|
|
err := b.Prepare(c)
|
|
if err == nil {
|
|
t.Fatalf("prepare should fail")
|
|
}
|
|
}
|
|
|
|
func TestBuilderPrepare_AccessKey(t *testing.T) {
|
|
var b Builder
|
|
config := testConfig()
|
|
|
|
// Test good
|
|
config["access_key"] = "foo"
|
|
err := b.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("should not have error: %s", err)
|
|
}
|
|
|
|
if b.config.AccessKey != "foo" {
|
|
t.Errorf("access key invalid: %s", b.config.AccessKey)
|
|
}
|
|
|
|
// Test bad
|
|
delete(config, "access_key")
|
|
b = Builder{}
|
|
err = b.Prepare(config)
|
|
if err == nil {
|
|
t.Fatal("should have error")
|
|
}
|
|
}
|
|
|
|
func TestBuilderPrepare_SecretKey(t *testing.T) {
|
|
var b Builder
|
|
config := testConfig()
|
|
|
|
// Test good
|
|
config["secret_key"] = "foo"
|
|
err := b.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("should not have error: %s", err)
|
|
}
|
|
|
|
if b.config.SecretKey != "foo" {
|
|
t.Errorf("secret key invalid: %s", b.config.SecretKey)
|
|
}
|
|
|
|
// Test bad
|
|
delete(config, "secret_key")
|
|
b = Builder{}
|
|
err = b.Prepare(config)
|
|
if err == nil {
|
|
t.Fatal("should have error")
|
|
}
|
|
}
|
|
|
|
func TestBuilderPrepare_SSHPort(t *testing.T) {
|
|
var b Builder
|
|
config := testConfig()
|
|
|
|
// Test default
|
|
err := b.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("should not have error: %s", err)
|
|
}
|
|
|
|
if b.config.SSHPort != 22 {
|
|
t.Errorf("invalid: %d", b.config.SSHPort)
|
|
}
|
|
|
|
// Test set
|
|
config["ssh_port"] = 35
|
|
b = Builder{}
|
|
err = b.Prepare(config)
|
|
if err != nil {
|
|
t.Fatalf("should not have error: %s", err)
|
|
}
|
|
|
|
if b.config.SSHPort != 35 {
|
|
t.Errorf("invalid: %d", b.config.SSHPort)
|
|
}
|
|
}
|