packer-cn/builder/amazon/common/ami_config_test.go

187 lines
5.0 KiB
Go
Raw Normal View History

package common
import (
"reflect"
"testing"
)
func testAMIConfig() *AMIConfig {
return &AMIConfig{
AMIName: "foo",
}
}
2017-11-30 17:08:48 -05:00
func getFakeAccessConfig(region string) *AccessConfig {
return &AccessConfig{
RawRegion: region,
}
}
func TestAMIConfigPrepare_name(t *testing.T) {
c := testAMIConfig()
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
t.Fatalf("shouldn't have err: %s", err)
}
c.AMIName = ""
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("should have error")
}
}
func TestAMIConfigPrepare_regions(t *testing.T) {
c := testAMIConfig()
c.AMIRegions = nil
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
t.Fatalf("shouldn't have err: %s", err)
}
c.AMIRegions = listEC2Regions()
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
2016-01-11 16:02:38 -05:00
t.Fatalf("shouldn't have err: %s", err)
}
c.AMIRegions = []string{"foo"}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("should have error")
}
c.AMIRegions = []string{"us-east-1", "us-west-1", "us-east-1"}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
t.Fatalf("bad: %s", err)
}
expected := []string{"us-east-1", "us-west-1"}
if !reflect.DeepEqual(c.AMIRegions, expected) {
t.Fatalf("bad: %#v", c.AMIRegions)
}
c.AMIRegions = []string{"custom"}
c.AMISkipRegionValidation = true
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
t.Fatal("shouldn't have error")
}
c.AMISkipRegionValidation = false
c.AMIRegions = []string{"us-east-1", "us-east-2", "us-west-1"}
2017-05-31 16:41:32 -04:00
c.AMIRegionKMSKeyIDs = map[string]string{
"us-east-1": "123-456-7890",
"us-west-1": "789-012-3456",
"us-east-2": "456-789-0123",
}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
t.Fatal("shouldn't have error")
}
2017-05-31 17:16:02 -04:00
c.AMIRegions = []string{"us-east-1", "us-east-2", "us-west-1"}
c.AMIRegionKMSKeyIDs = map[string]string{
"us-east-1": "123-456-7890",
"us-west-1": "789-012-3456",
"us-east-2": "",
}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
2017-05-31 17:16:02 -04:00
t.Fatal("should have passed; we are able to use default KMS key if not sharing")
}
2017-05-26 15:21:07 -04:00
c.SnapshotUsers = []string{"user-foo", "user-bar"}
c.AMIRegions = []string{"us-east-1", "us-east-2", "us-west-1"}
2017-05-31 16:41:32 -04:00
c.AMIRegionKMSKeyIDs = map[string]string{
2017-05-26 15:21:07 -04:00
"us-east-1": "123-456-7890",
"us-west-1": "789-012-3456",
"us-east-2": "",
}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
2017-05-26 15:21:07 -04:00
t.Fatal("should have an error b/c can't use default KMS key if sharing")
}
c.AMIRegions = []string{"us-east-1", "us-west-1"}
2017-05-31 16:41:32 -04:00
c.AMIRegionKMSKeyIDs = map[string]string{
"us-east-1": "123-456-7890",
"us-west-1": "789-012-3456",
"us-east-2": "456-789-0123",
}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("should have error b/c theres a region in the key map that isn't in ami_regions")
}
c.AMIRegions = []string{"us-east-1", "us-west-1", "us-east-2"}
2017-05-31 16:41:32 -04:00
c.AMIRegionKMSKeyIDs = map[string]string{
"us-east-1": "123-456-7890",
"us-west-1": "789-012-3456",
}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("should have error b/c theres a region in in ami_regions that isn't in the key map")
}
c.SnapshotUsers = []string{"foo", "bar"}
c.AMIKmsKeyId = "123-abc-456"
c.AMIEncryptBootVolume = true
c.AMIRegions = []string{"us-east-1", "us-west-1"}
2017-05-31 16:41:32 -04:00
c.AMIRegionKMSKeyIDs = map[string]string{
"us-east-1": "123-456-7890",
"us-west-1": "",
}
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("should have error b/c theres a region in in ami_regions that isn't in the key map")
}
2017-11-30 17:08:48 -05:00
// allow rawregion to exist in ami_regions list.
accessConf := getFakeAccessConfig("us-east-1")
c.AMIRegions = []string{"us-east-1", "us-west-1", "us-east-2"}
c.AMIRegionKMSKeyIDs = nil
if err := c.Prepare(accessConf, nil); err != nil {
t.Fatal("should allow user to have the raw region in ami_regions")
}
}
2016-05-12 12:21:44 -04:00
func TestAMIConfigPrepare_Share_EncryptedBoot(t *testing.T) {
2016-05-12 12:21:44 -04:00
c := testAMIConfig()
c.AMIUsers = []string{"testAccountID"}
c.AMIEncryptBootVolume = true
c.AMIKmsKeyId = ""
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("shouldn't be able to share ami with encrypted boot volume")
}
c.AMIKmsKeyId = "89c3fb9a-de87-4f2a-aedc-fddc5138193c"
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
t.Fatal("shouldn't be able to share ami with encrypted boot volume")
2016-05-12 12:21:44 -04:00
}
}
2017-04-04 20:46:44 -04:00
func TestAMINameValidation(t *testing.T) {
c := testAMIConfig()
c.AMIName = "aa"
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
2017-04-04 20:46:44 -04:00
t.Fatal("shouldn't be able to have an ami name with less than 3 characters")
}
var longAmiName string
for i := 0; i < 129; i++ {
longAmiName += "a"
}
c.AMIName = longAmiName
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
2017-04-04 20:46:44 -04:00
t.Fatal("shouldn't be able to have an ami name with great than 128 characters")
}
c.AMIName = "+aaa"
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err == nil {
2017-04-04 20:46:44 -04:00
t.Fatal("shouldn't be able to have an ami name with invalid characters")
}
2017-04-05 01:08:09 -04:00
2017-05-19 05:11:43 -04:00
c.AMIName = "fooBAR1()[] ./-'@_"
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
2017-05-19 05:11:43 -04:00
t.Fatal("should be able to use all of the allowed AMI characters")
2017-04-05 01:08:09 -04:00
}
c.AMIName = `xyz-base-2017-04-05-1934`
2017-10-30 17:34:16 -04:00
if err := c.Prepare(nil, nil); err != nil {
t.Fatalf("expected `xyz-base-2017-04-05-1934` to pass validation.")
}
2017-04-04 21:02:23 -04:00
}