wait for user profile update to finish

hashicorp · Feb 25, 2021 · 36dfddc · 36dfddc
1 parent ddf76f9
commit 36dfddc
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/.changelog/17251.txt b/.changelog/17251.txt
@@ -8,4 +8,8 @@ resource/aws_sagemaker_domain: Make `default_resource_spec` optional for the `te
 
 ```release-note:bug
 resource/aws_sagemaker_domain: Wait for update to finish.
+```
+
+```release-note:bug
+resource/aws_sagemaker_user_profile: Wait for update to finish.
 ```
diff --git a/aws/resource_aws_sagemaker_user_profile.go b/aws/resource_aws_sagemaker_user_profile.go
@@ -310,9 +310,12 @@ func resourceAwsSagemakerUserProfileUpdate(d *schema.ResourceData, meta interfac
 	conn := meta.(*AWSClient).sagemakerconn
 
 	if d.HasChange("user_settings") {
+		domainID := d.Get("domain_id").(string)
+		userProfileName := d.Get("user_profile_name").(string)
+
 		input := &sagemaker.UpdateUserProfileInput{
-			UserProfileName: aws.String(d.Get("user_profile_name").(string)),
-			DomainId:        aws.String(d.Get("domain_id").(string)),
+			UserProfileName: aws.String(userProfileName),
+			DomainId:        aws.String(domainID),
 			UserSettings:    expandSagemakerDomainDefaultUserSettings(d.Get("user_settings").([]interface{})),
 		}
 
@@ -321,6 +324,10 @@ func resourceAwsSagemakerUserProfileUpdate(d *schema.ResourceData, meta interfac
 		if err != nil {
 			return fmt.Errorf("error updating SageMaker User Profile: %w", err)
 		}
+
+		if _, err := waiter.UserProfileInService(conn, domainID, userProfileName); err != nil {
+			return fmt.Errorf("error waiting for SageMaker User Profile (%s) to update: %w", d.Id(), err)
+		}
 	}
 
 	if d.HasChange("tags") {