add testcase

Author: mikatong

tencentcloud/data_source_tc_user_info.go

@@ -100,29 +100,31 @@ func datasourceTencentCloudUserInfoRead(d *schema.ResourceData, meta interface{}
 	appId := strconv.FormatUint(*result.AppId, 10)
 	uin := *result.Uin
 	ownerUin := *result.OwnerUin
-
-	d.SetId(fmt.Sprintf("user-%s-%s-%d", uin, appId, rand.Intn(10000)))
-
-	_ = d.Set("app_id", appId)
-	_ = d.Set("uin", uin)
-	_ = d.Set("owner_uin", ownerUin)
 	accountInfoRequest := cam.NewDescribeSubAccountsRequest()
 	accountInfoRequest.FilterSubAccountUin = []*uint64{helper.Uint64(helper.StrToUInt64(uin))}
 	accountInfoResponse, err := client.UseCamClient().DescribeSubAccounts(accountInfoRequest)
 	if err != nil {
 		return err
 	}
 	subAccounts := accountInfoResponse.Response.SubAccounts
+	var name string
 	if len(subAccounts) > 0 {
-		name := *subAccounts[0].Name
-		_ = d.Set("name", name)
+		name = *subAccounts[0].Name
 	}
+	d.SetId(fmt.Sprintf("user-%s-%s-%d", uin, appId, rand.Intn(10000)))
+
+	_ = d.Set("app_id", appId)
+	_ = d.Set("uin", uin)
+	_ = d.Set("owner_uin", ownerUin)
+	_ = d.Set("name", name)
+
 	output, ok := d.GetOk("result_output_file")
 	if ok && output.(string) != "" {
 		if e := writeToFile(output.(string), map[string]interface{}{
 			"app_id":   appId,
 			"uin":      uin,
 			"ownerUin": ownerUin,
+			"name":     name,
 		}); e != nil {
 			return e
 		}

tencentcloud/data_source_tc_user_info_test.go

@@ -6,14 +6,14 @@ import (
 	"github.com/hashicorp/terraform-plugin-sdk/helper/resource"
 )
 
-func TestAccTencentCloudDataSourceUserInfo(t *testing.T) {
+func TestAccTencentCloudDataSourceUserInfoBasic(t *testing.T) {
 	t.Parallel()
 	resource.Test(t, resource.TestCase{
-		PreCheck:  func() { testAccPreCheck(t) },
 		Providers: testAccProviders,
 		Steps: []resource.TestStep{
 			{
-				Config: testAccDataUserInfoBasic,
+				PreConfig: func() { testAccStepPreConfigSetTempAKSK(t, ACCOUNT_TYPE_COMMON) },
+				Config:    testAccDataUserInfoBasic,
 				Check: resource.ComposeAggregateTestCheckFunc(
 					resource.TestCheckResourceAttrSet("data.tencentcloud_user_info.info", "app_id"),
 					resource.TestCheckResourceAttrSet("data.tencentcloud_user_info.info", "uin"),
@@ -24,6 +24,29 @@ func TestAccTencentCloudDataSourceUserInfo(t *testing.T) {
 	})
 }
 
+func TestAccTencentCloudDataSourceUserInfoSubAccount(t *testing.T) {
+	t.Parallel()
+	resource.Test(t, resource.TestCase{
+		Providers: testAccProviders,
+		Steps: []resource.TestStep{
+			{
+				// Need use subaccount aksk
+				PreConfig: func() { testAccStepPreConfigSetTempAKSK(t, ACCOUNT_TYPE_SUB_ACCOUNT) },
+				Config:    testAccDataUserInfoSubAccount,
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet("data.tencentcloud_user_info.info_sub_account", "app_id"),
+					resource.TestCheckResourceAttrSet("data.tencentcloud_user_info.info_sub_account", "uin"),
+					resource.TestCheckResourceAttrSet("data.tencentcloud_user_info.info_sub_account", "owner_uin"),
+					resource.TestCheckResourceAttrSet("data.tencentcloud_user_info.info_sub_account", "name"),
+				),
+			},
+		},
+	})
+}
+
 const testAccDataUserInfoBasic = `
 data "tencentcloud_user_info" "info" {}
 `
+const testAccDataUserInfoSubAccount = `
+data "tencentcloud_user_info" "info_sub_account" {}
+`

tencentcloud/provider_test.go

@@ -19,6 +19,7 @@ const (
 	ACCOUNT_TYPE_PREPAY             = "PREPAY"
 	ACCOUNT_TYPE_COMMON             = "COMMON"
 	ACCOUNT_TYPE_PRIVATE            = "PRIVATE"
+	ACCOUNT_TYPE_SUB_ACCOUNT        = "SUB_ACCOUNT"
 	INTERNATION_PROVIDER_SECRET_ID  = "TENCENTCLOUD_SECRET_ID_INTERNATION"
 	INTERNATION_PROVIDER_SECRET_KEY = "TENCENTCLOUD_SECRET_KEY_INTERNATION"
 	PREPAY_PROVIDER_SECRET_ID       = "TENCENTCLOUD_SECRET_ID_PREPAY"
@@ -27,6 +28,8 @@ const (
 	PRIVATE_PROVIDER_SECRET_KEY     = "TENCENTCLOUD_SECRET_KEY_PRIVATE"
 	COMMON_PROVIDER_SECRET_ID       = "TENCENTCLOUD_SECRET_ID_COMMON"
 	COMMON_PROVIDER_SECRET_KEY      = "TENCENTCLOUD_SECRET_KEY_COMMON"
+	SUB_ACCOUNT_PROVIDER_SECRET_ID  = "TENCENTCLOUD_SECRET_ID_SUB_ACCOUNT"
+	SUB_ACCOUNT_PROVIDER_SECRET_KEY = "TENCENTCLOUD_SECRET_KEY_SUB_ACCOUNT"
 )
 
 func init() {
@@ -109,6 +112,14 @@ func testAccPreCheckCommon(t *testing.T, accountType string) {
 		}
 		os.Setenv(PROVIDER_SECRET_ID, secretId)
 		os.Setenv(PROVIDER_SECRET_KEY, secretKey)
+	case accountType == ACCOUNT_TYPE_SUB_ACCOUNT:
+		secretId := os.Getenv(SUB_ACCOUNT_PROVIDER_SECRET_ID)
+		secretKey := os.Getenv(SUB_ACCOUNT_PROVIDER_SECRET_KEY)
+		if secretId == "" || secretKey == "" {
+			t.Fatalf("%v and %v must be set for acceptance tests\n", SUB_ACCOUNT_PROVIDER_SECRET_ID, SUB_ACCOUNT_PROVIDER_SECRET_KEY)
+		}
+		os.Setenv(PROVIDER_SECRET_ID, secretId)
+		os.Setenv(PROVIDER_SECRET_KEY, secretKey)
 	default:
 		if v := os.Getenv(PROVIDER_SECRET_ID); v == "" {
 			t.Fatalf("%v must be set for acceptance tests\n", PROVIDER_SECRET_ID)