fix: add e2e test for tke gpu args

Author: xawei

.changelog/1593.txt

@@ -1,9 +1,9 @@
 ```release-note:enhancement
-resource/tencentcloud_kubernetes_scale_worker: feat: support setting tke gpu args
+resource/tencentcloud_kubernetes_scale_worker: support setting tke gpu args
 ```
 ```release-note:enhancement
-resource/tencentcloud_kubernetes_cluster_attachment: feat: support setting tke gpu args
+resource/tencentcloud_kubernetes_cluster_attachment: support setting tke gpu args
 ```
 ```release-note:enhancement
-resource/tencentcloud_kubernetes_node_pool: feat: support setting tke gpu args
+resource/tencentcloud_kubernetes_node_pool: support setting tke gpu args
 ```

tencentcloud/resource_tc_kubernetes_node_pool_test.go

@@ -167,6 +167,25 @@ func TestAccTencentCloudKubernetesNodePoolResource_DiskEncrypt(t *testing.T) {
 	})
 }
 
+func TestAccTencentCloudKubernetesNodePoolResource_GPUInstance(t *testing.T) {
+	t.Parallel()
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckTkeNodePoolDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: testAccTkeNodePoolClusterGpu,
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckTkeNodePoolExists,
+					resource.TestCheckResourceAttrSet(testTkeClusterNodePoolResourceKey, "cluster_id"),
+					resource.TestCheckResourceAttrSet(testTkeClusterNodePoolResourceKey, "node_config.0.gpu_args.#"),
+				),
+			},
+		},
+	})
+}
+
 func testAccCheckTkeNodePoolDestroy(s *terraform.State) error {
 	logId := getLogId(contextNil)
 	ctx := context.WithValue(context.TODO(), logIdKey, logId)
@@ -436,3 +455,93 @@ resource "tencentcloud_kubernetes_node_pool" "np_test" {
   unschedulable = 0
 }
 `
+
+const testAccTkeNodePoolClusterGpu string = testAccTkeNodePoolClusterBasic + `
+resource "tencentcloud_kubernetes_node_pool" "np_test" {
+  name = "gpu_args_node_pool"
+  cluster_id = local.cluster_id
+  max_size = 1
+  min_size = 0
+  vpc_id               = data.tencentcloud_vpc_subnets.vpc.instance_list.0.vpc_id
+  subnet_ids           = [data.tencentcloud_vpc_subnets.vpc.instance_list.0.subnet_id]
+  retry_policy         = "INCREMENTAL_INTERVALS"
+  desired_capacity     = 1
+  enable_auto_scale    = false
+  node_os = "ubuntu20.04x86_64"
+  scaling_group_project_id = var.default_project
+  delete_keep_instance = false
+  scaling_group_name 	   = "asg_np_test_changed"
+  default_cooldown 		   = 350
+  termination_policies 	   = ["NEWEST_INSTANCE"]
+  multi_zone_subnet_policy = "EQUALITY"
+
+  auto_scaling_config {
+    instance_type      = "GN6S.LARGE20"
+    system_disk_type   = "CLOUD_PREMIUM"
+    system_disk_size   = "100"
+    security_group_ids = [data.tencentcloud_security_groups.sg.security_groups[0].security_group_id, data.tencentcloud_security_groups.sg_as.security_groups[0].security_group_id]
+	instance_charge_type = "SPOTPAID"
+    spot_instance_type = "one-time"
+    spot_max_price = "1000"
+    cam_role_name = "TCB_QcsRole"
+
+    data_disk {
+      disk_type = "CLOUD_PREMIUM"
+      disk_size = 50
+      delete_with_instance = true
+    }
+    data_disk {
+      disk_type = "CLOUD_PREMIUM"
+      disk_size = 100
+      delete_with_instance = true
+    }
+
+    public_ip_assigned         = false
+    password                   = "test123#"
+    enhanced_security_service  = true
+    enhanced_monitor_service   = false
+	host_name                  = "12.123.1.1"
+	host_name_style            = "UNIQUE"
+
+  }
+  unschedulable = 0
+  labels = {
+    "test3" = "test3",
+    "test2" = "test2",
+  }
+  
+  taints {
+	key = "test_taint"
+    value = "taint_value"
+    effect = "PreferNoSchedule"
+  }
+
+  tags = {
+    keep-test-np1 = "testI"
+    keep-test-np3 = "testIII"
+  }
+
+  node_config {
+    extra_args = [
+      "root-dir=/var/lib/kubelet"
+    ]
+	gpu_args {
+      mig_enable = false
+      driver = {
+        name = "NVIDIA-Linux-x86_64-450.102.04.run"
+        version = "450.102.04"
+      }
+      cuda = {
+        name = "cuda_11.0.3_450.51.06_linux.run"
+        version = "11.0.3"
+      }
+      cudnn = {
+        name = "libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb"
+        version = "8.1.0"
+        doc_name = "libcudnn8-samples_8.1.0.77-1+cuda11.2_amd64.deb"
+        dev_name = "libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb"
+      }
+    }
+  }
+}
+`

tencentcloud/resource_tc_kubernetes_scale_worker_test.go

@@ -95,6 +95,15 @@ func TestAccTencentCloudKubernetesScaleWorkerResource(t *testing.T) {
 					resource.TestCheckResourceAttrSet(testTkeScaleWorkerResourceKey, "unschedulable"),
 				),
 			},
+			{
+				Config: testAccTkeScaleWorkerInstanceGpuInsTypeUpdate,
+				Check: resource.ComposeTestCheckFunc(
+					testAccCheckTkeScaleWorkerExists(testTkeScaleWorkerResourceKey),
+					resource.TestCheckResourceAttrSet(testTkeScaleWorkerResourceKey, "cluster_id"),
+					resource.TestCheckResourceAttr(testTkeScaleWorkerResourceKey, "worker_instances_list.0.instance_type", "GN6S.LARGE20"),
+					resource.TestCheckResourceAttrSet(testTkeScaleWorkerResourceKey, "gpu_args.#"),
+				),
+			},
 		},
 	})
 }
@@ -200,6 +209,44 @@ const testAccTkeScaleWorkerInstanceBasic = TkeExclusiveNetwork + TkeDataSource +
 
 const testAccTkeScaleWorkerInstance string = testAccTkeScaleWorkerInstanceBasic + `
 
+resource tencentcloud_kubernetes_scale_worker test_scale {
+  cluster_id = local.cluster_id
+  
+  extra_args = [
+ 	"root-dir=/var/lib/kubelet"
+  ]	
+
+  labels = {
+    "test1" = "test1",
+    "test2" = "test2",
+  }
+  unschedulable = 0
+
+  worker_config {
+    count                      				= 1
+    availability_zone          				= var.default_az
+    instance_type              				= "GN6S.LARGE20"
+    subnet_id                  				= local.subnet_id
+    system_disk_type           				= "CLOUD_SSD"
+    system_disk_size           				= 50
+    internet_charge_type       				= "TRAFFIC_POSTPAID_BY_HOUR"
+    security_group_ids                      = [local.sg_id]
+
+    data_disk {
+      disk_type = "CLOUD_PREMIUM"
+      disk_size = 50
+    }
+
+    enhanced_security_service 				= false
+    enhanced_monitor_service  				= false
+    user_data                 				= "dGVzdA=="
+    password                  				= "AABBccdd1122"
+  }
+}
+`
+
+const testAccTkeScaleWorkerInstanceGpuInsTypeUpdate string = testAccTkeScaleWorkerInstanceBasic + `
+
 resource tencentcloud_kubernetes_scale_worker test_scale {
   cluster_id = local.cluster_id
   
@@ -222,6 +269,7 @@ resource tencentcloud_kubernetes_scale_worker test_scale {
     system_disk_size           				= 50
     internet_charge_type       				= "TRAFFIC_POSTPAID_BY_HOUR"
     security_group_ids                      = [local.sg_id]
+	img_id 								    = "img-22trbn9x"
 
     data_disk {
       disk_type = "CLOUD_PREMIUM"
@@ -233,5 +281,23 @@ resource tencentcloud_kubernetes_scale_worker test_scale {
     user_data                 				= "dGVzdA=="
     password                  				= "AABBccdd1122"
   }
+
+  gpu_args {
+    mig_enable = false
+    driver = {
+      name = "NVIDIA-Linux-x86_64-450.102.04.run"
+      version = "450.102.04"
+    }
+    cuda = {
+      name = "cuda_11.0.3_450.51.06_linux.run"
+      version = "11.0.3"
+    }
+    cudnn = {
+      name = "libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb"
+      version = "8.1.0"
+      doc_name = "libcudnn8-samples_8.1.0.77-1+cuda11.2_amd64.deb"
+      dev_name = "libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb"
+    }
+  }
 }
 `