38 lines
968 B
Go
38 lines
968 B
Go
|
package ebssurrogate
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/aws/aws-sdk-go/service/ec2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func GetStringPointer() *string {
|
||
|
|
tmp := "/dev/name"
|
||
|
|
return &tmp
|
||
|
|
}
|
||
|
|
|
||
|
|
func GetTestDevice() *ec2.BlockDeviceMapping {
|
||
|
|
TestDev := ec2.BlockDeviceMapping{
|
||
|
|
DeviceName: GetStringPointer(),
|
||
|
|
}
|
||
|
|
return &TestDev
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStepRegisterAmi_DeduplicateRootVolume(t *testing.T) {
|
||
|
|
TestRootDevice := RootBlockDevice{}
|
||
|
|
TestRootDevice.SourceDeviceName = "/dev/name"
|
||
|
|
|
||
|
|
blockDevices := []*ec2.BlockDeviceMapping{}
|
||
|
|
blockDevicesExcludingRoot := DeduplicateRootVolume(blockDevices, TestRootDevice, "12342351")
|
||
|
|
if len(blockDevicesExcludingRoot) != 1 {
|
||
|
|
t.Fatalf("Unexpected length of block devices list")
|
||
|
|
}
|
||
|
|
|
||
|
|
TestBlockDevice := GetTestDevice()
|
||
|
|
blockDevices = append(blockDevices, TestBlockDevice)
|
||
|
|
blockDevicesExcludingRoot = DeduplicateRootVolume(blockDevices, TestRootDevice, "12342351")
|
||
|
|
if len(blockDevicesExcludingRoot) != 1 {
|
||
|
|
t.Fatalf("Unexpected length of block devices list")
|
||
|
|
}
|
||
|
|
}
|