Add Azure VM and AKS integration tests for the default OTel config (#730)

Author: movence

environment/computetype/compute_type.go

@@ -11,13 +11,20 @@ const (
 	EC2 ComputeType = "EC2"
 	ECS ComputeType = "ECS"
 	EKS ComputeType = "EKS"
+	// AzureVM is a non-AWS host authenticating to AWS via the Azure web-identity credential chain.
+	AzureVM ComputeType = "AZUREVM"
+	// AKS is an Azure Kubernetes Service cluster authenticating to AWS via the projected
+	// service-account web-identity credential chain.
+	AKS ComputeType = "AKS"
 )
 
 var (
 	computeTypes = map[string]ComputeType{
-		"EC2": EC2,
-		"ECS": ECS,
-		"EKS": EKS,
+		"EC2":     EC2,
+		"ECS":     ECS,
+		"EKS":     EKS,
+		"AZUREVM": AzureVM,
+		"AKS":     AKS,
 	}
 )
 

environment/metadata.go

@@ -39,6 +39,7 @@ type MetaData struct {
 	CwaCommitSha                                string
 	CaCertPath                                  string
 	EKSClusterName                              string
+	AKSClusterName                              string
 	ProxyUrl                                    string
 	AssumeRoleArn                               string
 	InstanceArn                                 string
@@ -86,6 +87,7 @@ type MetaDataStrings struct {
 	CwaCommitSha                                string
 	CaCertPath                                  string
 	EKSClusterName                              string
+	AKSClusterName                              string
 	ProxyUrl                                    string
 	AssumeRoleArn                               string
 	InstanceArn                                 string
@@ -119,7 +121,7 @@ type MetaDataStrings struct {
 }
 
 func registerComputeType(dataString *MetaDataStrings) {
-	flag.StringVar(&(dataString.ComputeType), "computeType", "", "EC2/ECS/EKS")
+	flag.StringVar(&(dataString.ComputeType), "computeType", "", "EC2/ECS/EKS/AZUREVM/AKS")
 }
 func registerBucket(dataString *MetaDataStrings) {
 	flag.StringVar(&(dataString.Bucket), "bucket", "", "s3 bucket ex cloudwatch-agent-integration-bucket")
@@ -142,6 +144,10 @@ func registerECSData(dataString *MetaDataStrings) {
 	flag.StringVar(&(dataString.EcsServiceName), "cwagentECSServiceName", "", "Used to restart ecs task to apply new agent config")
 }
 
+func registerAKSData(d *MetaDataStrings) {
+	flag.StringVar(&(d.AKSClusterName), "aksClusterName", "", "AKS cluster name")
+}
+
 func registerEKSData(d *MetaDataStrings) {
 	flag.StringVar(&(d.EKSClusterName), "eksClusterName", "", "EKS cluster name")
 	flag.StringVar(&(d.EksDeploymentStrategy), "eksDeploymentStrategy", "", "Daemon/Replica/Sidecar")
@@ -187,7 +193,7 @@ func registerProxyUrl(dataString *MetaDataStrings) {
 func fillComputeType(e *MetaData, data *MetaDataStrings) {
 	computeType, ok := computetype.FromString(data.ComputeType)
 	if !ok {
-		log.Panic("Invalid compute type. Needs to be EC2/ECS/EKS. Compute Type is a required flag. :" + data.ComputeType)
+		log.Panic("Invalid compute type. Needs to be EC2/ECS/EKS/AZUREVM/AKS. Compute Type is a required flag. :" + data.ComputeType)
 	}
 	e.ComputeType = computeType
 }
@@ -321,6 +327,7 @@ func RegisterEnvironmentMetaDataFlags() *MetaDataStrings {
 	registerComputeType(registeredMetaDataStrings)
 	registerECSData(registeredMetaDataStrings)
 	registerEKSData(registeredMetaDataStrings)
+	registerAKSData(registeredMetaDataStrings)
 	registerEKSE2ETestData(registeredMetaDataStrings)
 	registerBucket(registeredMetaDataStrings)
 	registerS3Key(registeredMetaDataStrings)
@@ -359,6 +366,7 @@ func GetEnvironmentMetaData() *MetaData {
 	metaDataStorage.AssumeRoleArn = registeredMetaDataStrings.AssumeRoleArn
 	metaDataStorage.InstanceArn = registeredMetaDataStrings.InstanceArn
 	metaDataStorage.InstanceId = registeredMetaDataStrings.InstanceId
+	metaDataStorage.AKSClusterName = registeredMetaDataStrings.AKSClusterName
 	metaDataStorage.InstancePlatform = registeredMetaDataStrings.InstancePlatform
 	metaDataStorage.AgentStartCommand = registeredMetaDataStrings.AgentStartCommand
 	metaDataStorage.EksGpuType = registeredMetaDataStrings.EksGpuType

terraform/azure/aks/.gitignore

@@ -0,0 +1 @@
+kubeconfig