64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package common
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
|
"github.com/hashicorp/packer/helper/multistep"
|
|
)
|
|
|
|
func testImage() *ec2.Image {
|
|
return &ec2.Image{
|
|
ImageId: aws.String("ami-abcd1234"),
|
|
Name: aws.String("ami_test_name"),
|
|
Tags: []*ec2.Tag{
|
|
{
|
|
Key: aws.String("key-1"),
|
|
Value: aws.String("value-1"),
|
|
},
|
|
{
|
|
Key: aws.String("key-2"),
|
|
Value: aws.String("value-2"),
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func testState() multistep.StateBag {
|
|
state := new(multistep.BasicStateBag)
|
|
return state
|
|
}
|
|
|
|
func TestInterpolateBuildInfo_extractBuildInfo_noSourceImage(t *testing.T) {
|
|
state := testState()
|
|
buildInfo := extractBuildInfo("foo", state)
|
|
|
|
expected := BuildInfoTemplate{
|
|
BuildRegion: "foo",
|
|
}
|
|
if !reflect.DeepEqual(*buildInfo, expected) {
|
|
t.Fatalf("Unexpected BuildInfoTemplate: expected %#v got %#v\n", expected, *buildInfo)
|
|
}
|
|
}
|
|
|
|
func TestInterpolateBuildInfo_extractBuildInfo_withSourceImage(t *testing.T) {
|
|
state := testState()
|
|
state.Put("source_image", testImage())
|
|
buildInfo := extractBuildInfo("foo", state)
|
|
|
|
expected := BuildInfoTemplate{
|
|
BuildRegion: "foo",
|
|
SourceAMI: "ami-abcd1234",
|
|
SourceAMIName: "ami_test_name",
|
|
SourceAMITags: map[string]string{
|
|
"key-1": "value-1",
|
|
"key-2": "value-2",
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(*buildInfo, expected) {
|
|
t.Fatalf("Unexpected BuildInfoTemplate: expected %#v got %#v\n", expected, *buildInfo)
|
|
}
|
|
}
|