fix(cognito-userpool): include user attribute update settings

ekristen · Dec 26, 2024 · 0a4b89a · 0a4b89a
1 parent d7a7494
commit 0a4b89a
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/resources/cognito-userpool.go b/resources/cognito-userpool.go
@@ -99,13 +99,21 @@ type CognitoUserPool struct {
 
 func (r *CognitoUserPool) Remove(_ context.Context) error {
 	if r.settings.GetBool("DisableDeletionProtection") {
-		_, err := r.svc.UpdateUserPool(&cognitoidentityprovider.UpdateUserPoolInput{
-			UserPoolId:         r.ID,
-			DeletionProtection: ptr.String("INACTIVE"),
+		userPool, err := r.svc.DescribeUserPool(&cognitoidentityprovider.DescribeUserPoolInput{
+			UserPoolId: r.ID,
 		})
 		if err != nil {
 			return err
 		}
+
+		_, updateErr := r.svc.UpdateUserPool(&cognitoidentityprovider.UpdateUserPoolInput{
+			UserPoolId:                  r.ID,
+			DeletionProtection:          ptr.String("INACTIVE"),
+			UserAttributeUpdateSettings: userPool.UserPool.UserAttributeUpdateSettings,
+		})
+		if updateErr != nil {
+			return updateErr
+		}
 	}
 
 	_, err := r.svc.DeleteUserPool(&cognitoidentityprovider.DeleteUserPoolInput{