From 999b0874cce88279a3d39dce441a342e7c5f5017 Mon Sep 17 00:00:00 2001
From: Clint Shryock <clint@ctshryock.com>
Date: Tue, 9 Jun 2015 11:38:53 -0500
Subject: [PATCH 1/2] Update AWS EBS builder to fix invalid params

---
 builder/amazon/common/block_device.go | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/builder/amazon/common/block_device.go b/builder/amazon/common/block_device.go
index 14add3276..c44d03561 100644
--- a/builder/amazon/common/block_device.go
+++ b/builder/amazon/common/block_device.go
@@ -29,13 +29,23 @@ func buildBlockDevices(b []BlockDevice) []*ec2.BlockDeviceMapping {
 
 	for _, blockDevice := range b {
 		ebsBlockDevice := &ec2.EBSBlockDevice{
-			SnapshotID:          &blockDevice.SnapshotId,
-			Encrypted:           &blockDevice.Encrypted,
-			IOPS:                &blockDevice.IOPS,
 			VolumeType:          &blockDevice.VolumeType,
 			VolumeSize:          &blockDevice.VolumeSize,
 			DeleteOnTermination: &blockDevice.DeleteOnTermination,
 		}
+
+		// IOPS is only valid for SSD Volumes
+		if blockDevice.VolumeType != "standard" && blockDevice.VolumeType != "gp2" {
+			ebsBlockDevice.IOPS = &blockDevice.IOPS
+		}
+
+		// You cannot specify Encrypted if you specify a Snapshot ID
+		if blockDevice.SnapshotId != "" {
+			ebsBlockDevice.SnapshotID = &blockDevice.SnapshotId
+		} else {
+			ebsBlockDevice.Encrypted = &blockDevice.Encrypted
+		}
+
 		mapping := &ec2.BlockDeviceMapping{
 			EBS:         ebsBlockDevice,
 			DeviceName:  &blockDevice.DeviceName,

From 4da118c64f22b950cb596c5c370751be5f59121b Mon Sep 17 00:00:00 2001
From: Clint Shryock <clint@ctshryock.com>
Date: Tue, 9 Jun 2015 11:56:40 -0500
Subject: [PATCH 2/2] fix up tests

---
 builder/amazon/common/block_device_test.go | 27 +++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/builder/amazon/common/block_device_test.go b/builder/amazon/common/block_device_test.go
index c4f644f67..aacd54cb4 100644
--- a/builder/amazon/common/block_device_test.go
+++ b/builder/amazon/common/block_device_test.go
@@ -5,6 +5,7 @@ import (
 	"testing"
 
 	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/awsutil"
 	"github.com/aws/aws-sdk-go/service/ec2"
 )
 
@@ -28,11 +29,31 @@ func TestBlockDevice(t *testing.T) {
 				DeviceName:  aws.String("/dev/sdb"),
 				VirtualName: aws.String("ephemeral0"),
 				EBS: &ec2.EBSBlockDevice{
-					Encrypted:           aws.Boolean(false),
 					SnapshotID:          aws.String("snap-1234"),
 					VolumeType:          aws.String("standard"),
 					VolumeSize:          aws.Long(8),
 					DeleteOnTermination: aws.Boolean(true),
+				},
+			},
+		},
+		{
+			Config: &BlockDevice{
+				DeviceName:          "/dev/sdb",
+				VirtualName:         "ephemeral0",
+				VolumeType:          "io1",
+				VolumeSize:          8,
+				DeleteOnTermination: true,
+				IOPS:                1000,
+			},
+
+			Result: &ec2.BlockDeviceMapping{
+				DeviceName:  aws.String("/dev/sdb"),
+				VirtualName: aws.String("ephemeral0"),
+				EBS: &ec2.EBSBlockDevice{
+					Encrypted:           aws.Boolean(false),
+					VolumeType:          aws.String("io1"),
+					VolumeSize:          aws.Long(8),
+					DeleteOnTermination: aws.Boolean(true),
 					IOPS:                aws.Long(1000),
 				},
 			},
@@ -48,11 +69,11 @@ func TestBlockDevice(t *testing.T) {
 		expected := []*ec2.BlockDeviceMapping{tc.Result}
 		got := blockDevices.BuildAMIDevices()
 		if !reflect.DeepEqual(expected, got) {
-			t.Fatalf("bad: %#v", expected)
+			t.Fatalf("Bad block device, \nexpected: %s\n\ngot: %s", awsutil.StringValue(expected), awsutil.StringValue(got))
 		}
 
 		if !reflect.DeepEqual(expected, blockDevices.BuildLaunchDevices()) {
-			t.Fatalf("bad: %#v", expected)
+			t.Fatalf("Bad block device, \nexpected: %s\n\ngot: %s", awsutil.StringValue(expected), awsutil.StringValue(blockDevices.BuildLaunchDevices()))
 		}
 	}
 }