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  }