github.phpd.cn/hashicorp/packer@v1.3.2/builder/amazon/common/step_source_ami_info_test.go (about)

     1  package common
     2  
     3  import (
     4  	"github.com/aws/aws-sdk-go/aws"
     5  	"github.com/aws/aws-sdk-go/service/ec2"
     6  	"github.com/stretchr/testify/assert"
     7  	"testing"
     8  )
     9  
    10  func TestStepSourceAmiInfo_PVImage(t *testing.T) {
    11  	err := new(StepSourceAMIInfo).canEnableEnhancedNetworking(&ec2.Image{
    12  		VirtualizationType: aws.String("paravirtual"),
    13  	})
    14  	assert.Error(t, err)
    15  }
    16  
    17  func TestStepSourceAmiInfo_HVMImage(t *testing.T) {
    18  	err := new(StepSourceAMIInfo).canEnableEnhancedNetworking(&ec2.Image{
    19  		VirtualizationType: aws.String("hvm"),
    20  	})
    21  	assert.NoError(t, err)
    22  }
    23  
    24  func TestStepSourceAmiInfo_PVImageWithAMIVirtPV(t *testing.T) {
    25  	stepSourceAMIInfo := StepSourceAMIInfo{
    26  		AMIVirtType: "paravirtual",
    27  	}
    28  	err := stepSourceAMIInfo.canEnableEnhancedNetworking(&ec2.Image{
    29  		VirtualizationType: aws.String("paravirtual"),
    30  	})
    31  	assert.Error(t, err)
    32  }
    33  
    34  func TestStepSourceAmiInfo_PVImageWithAMIVirtHVM(t *testing.T) {
    35  	stepSourceAMIInfo := StepSourceAMIInfo{
    36  		AMIVirtType: "hvm",
    37  	}
    38  	err := stepSourceAMIInfo.canEnableEnhancedNetworking(&ec2.Image{
    39  		VirtualizationType: aws.String("paravirtual"),
    40  	})
    41  	assert.NoError(t, err)
    42  }