github.com/mmcquillan/packer@v1.1.1-0.20171009221028-c85cf0483a5d/builder/amazon/ebssurrogate/step_register_ami_test.go (about) 1 package ebssurrogate 2 3 import ( 4 "testing" 5 6 "github.com/aws/aws-sdk-go/service/ec2" 7 ) 8 9 func GetStringPointer() *string { 10 tmp := "/dev/name" 11 return &tmp 12 } 13 14 func GetTestDevice() *ec2.BlockDeviceMapping { 15 TestDev := ec2.BlockDeviceMapping{ 16 DeviceName: GetStringPointer(), 17 } 18 return &TestDev 19 } 20 21 func TestStepRegisterAmi_DeduplicateRootVolume(t *testing.T) { 22 TestRootDevice := RootBlockDevice{} 23 TestRootDevice.SourceDeviceName = "/dev/name" 24 25 blockDevices := []*ec2.BlockDeviceMapping{} 26 blockDevicesExcludingRoot := DeduplicateRootVolume(blockDevices, TestRootDevice, "12342351") 27 if len(blockDevicesExcludingRoot) != 1 { 28 t.Fatalf("Unexpected length of block devices list") 29 } 30 31 TestBlockDevice := GetTestDevice() 32 blockDevices = append(blockDevices, TestBlockDevice) 33 blockDevicesExcludingRoot = DeduplicateRootVolume(blockDevices, TestRootDevice, "12342351") 34 if len(blockDevicesExcludingRoot) != 1 { 35 t.Fatalf("Unexpected length of block devices list") 36 } 37 }