builder/amazonebs: Switch from json to mapstructure for decode

This commit is contained in:
Mitchell Hashimoto 2013-05-20 16:39:43 -07:00
parent bee266008e
commit 2efab467a8
2 changed files with 9 additions and 18 deletions

View File

@ -6,21 +6,21 @@
package amazonebs package amazonebs
import ( import (
"encoding/json"
"fmt" "fmt"
"github.com/mitchellh/goamz/aws" "github.com/mitchellh/goamz/aws"
"github.com/mitchellh/goamz/ec2" "github.com/mitchellh/goamz/ec2"
"github.com/mitchellh/mapstructure"
"github.com/mitchellh/packer/packer" "github.com/mitchellh/packer/packer"
"log" "log"
"time" "time"
) )
type config struct { type config struct {
AccessKey string `json:"access_key"` AccessKey string `mapstructure:"access_key"`
AMIName string `json:"ami_name"` AMIName string `mapstructure:"ami_name"`
Region string Region string
SecretKey string `json:"secret_key"` SecretKey string `mapstructure:"secret_key"`
SourceAmi string `json:"source_ami"` SourceAmi string `mapstructure:"source_ami"`
} }
type Builder struct { type Builder struct {
@ -28,16 +28,7 @@ type Builder struct {
} }
func (b *Builder) Prepare(raw interface{}) (err error) { func (b *Builder) Prepare(raw interface{}) (err error) {
// Marshal and unmarshal the raw configuration as a way to get it err = mapstructure.Decode(raw, &b.config)
// into our "config" struct.
// TODO: Use the reflection package and provide this as an API for
// better error messages
jsonBytes, err := json.Marshal(raw)
if err != nil {
return
}
err = json.Unmarshal(jsonBytes, &b.config)
if err != nil { if err != nil {
return return
} }

View File

@ -14,15 +14,15 @@ func TestBuilder_ImplementsBuilder(t *testing.T) {
} }
func TestBuilder_Prepare_BadType(t *testing.T) { func TestBuilder_Prepare_BadType(t *testing.T) {
assert := asserts.NewTestingAsserts(t, true)
b := &Builder{} b := &Builder{}
c := map[string]interface{}{ c := map[string]interface{}{
"access_key": []string{}, "access_key": []string{},
} }
err := b.Prepare(c) err := b.Prepare(c)
assert.NotNil(err, "should have an error") if err == nil {
t.Fatalf("prepare should fail")
}
} }
func TestBuilder_Prepare_Good(t *testing.T) { func TestBuilder_Prepare_Good(t *testing.T) {