🧹 linting

hashicorp · Feb 21, 2024 · a158964 · a158964
1 parent 4a70575
commit a158964
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/internal/service/cognitoidp/user_group_data_source_test.go b/internal/service/cognitoidp/user_group_data_source_test.go
@@ -13,13 +13,17 @@ import (
 	"github.com/hashicorp/terraform-provider-aws/internal/acctest"
 )
 
-func TestAccCognitoIDPUserGroupDataSource_basic(t *testing.T) {
+func TestAccCognitoIdentityPoolProviderUserGroupDataSource_basic(t *testing.T) {
 	ctx := acctest.Context(t)
 	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
 	dataSourceName := "data.aws_cognito_user_group.test"
 
 	resource.ParallelTest(t, resource.TestCase{
-		PreCheck:                 func() { acctest.PreCheck(ctx, t); testAccPreCheckIdentityProvider(ctx, t) },
+		PreCheck: func() {
+			acctest.PreCheck(ctx, t)
+			acctest.PreCheckPartitionHasService(t, cognitoidentityprovider.ServiceID)
+			testAccPreCheckIdentityProvider(ctx, t)
+		},
 		ErrorCheck:               acctest.ErrorCheck(t, cognitoidentityprovider.EndpointsID),
 		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
 		CheckDestroy:             testAccCheckUserGroupDestroy(ctx),

diff --git a/internal/service/cognitoidp/user_groups_data_source_test.go b/internal/service/cognitoidp/user_groups_data_source_test.go
@@ -13,23 +13,25 @@ import (
 	"github.com/hashicorp/terraform-provider-aws/internal/acctest"
 )
 
-func TestAccCognitoIDPUserGroupsDataSource_basic(t *testing.T) {
+func TestAccCognitoIdentityPoolProviderUserGroupsDataSource_basic(t *testing.T) {
 	ctx := acctest.Context(t)
 	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
 	dataSourceName := "data.aws_cognito_user_groups.test"
 
 	resource.ParallelTest(t, resource.TestCase{
-		PreCheck:                 func() { acctest.PreCheck(ctx, t); testAccPreCheckIdentityProvider(ctx, t) },
+		PreCheck: func() {
+			acctest.PreCheck(ctx, t)
+			acctest.PreCheckPartitionHasService(t, cognitoidentityprovider.ServiceID)
+			testAccPreCheckIdentityProvider(ctx, t)
+		},
 		ErrorCheck:               acctest.ErrorCheck(t, cognitoidentityprovider.EndpointsID),
 		ProtoV5ProviderFactories: acctest.ProtoV5ProviderFactories,
 		CheckDestroy:             testAccCheckUserGroupDestroy(ctx),
 		Steps: []resource.TestStep{
 			{
 				Config: testAccUserGroupsDataSourceConfig_basic(rName),
 				Check: resource.ComposeTestCheckFunc(
-					resource.TestCheckResourceAttrSet(dataSourceName, "groups.#"),
-					resource.TestCheckResourceAttr(dataSourceName, "groups.0.group_name", fmt.Sprintf("%s-1", rName)),
-					resource.TestCheckResourceAttr(dataSourceName, "groups.1.group_name", fmt.Sprintf("%s-2", rName)),
+					resource.TestCheckResourceAttr(dataSourceName, "groups.#", "2"),
 				),
 			},
 		},