🧹 linting

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

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/internal/service/cognitoidp/user_group_data_source_test.go b/internal/service/cognitoidp/user_group_data_source_test.go
@@ -19,7 +19,11 @@ func TestAccCognitoIDPUserGroupDataSource_basic(t *testing.T) {
 	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
@@ -19,17 +19,19 @@ func TestAccCognitoIDPUserGroupsDataSource_basic(t *testing.T) {
 	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"),
 				),
 			},
 		},