r/sagemaker_workteam

.changelog/20122.txt

@@ -0,0 +1,3 @@
+```release-note:new-resource
+aws_sagemaker_workteam
+```

aws/aws_sagemaker_test.go

@@ -0,0 +1,60 @@
+package aws
+
+import (
+	"testing"
+)
+
+// Tests are serialized as SagmMaker Domain resources are limited to 1 per account by default.
+// SageMaker UserProfile and App depend on the Domain resources and as such are also part of the serialized test suite.
+// Sagemaker Workteam tests must also be serialized
+func TestAccAWSSagemaker_serial(t *testing.T) {
+	testCases := map[string]map[string]func(t *testing.T){
+		"App": {
+			"basic":        testAccAWSSagemakerApp_basic,
+			"disappears":   testAccAWSSagemakerApp_tags,
+			"tags":         testAccAWSSagemakerApp_disappears,
+			"resourceSpec": testAccAWSSagemakerApp_resourceSpec,
+		},
+		"Domain": {
+			"basic":                                testAccAWSSagemakerDomain_basic,
+			"disappears":                           testAccAWSSagemakerDomain_tags,
+			"tags":                                 testAccAWSSagemakerDomain_disappears,
+			"tensorboardAppSettings":               testAccAWSSagemakerDomain_tensorboardAppSettings,
+			"tensorboardAppSettingsWithImage":      testAccAWSSagemakerDomain_tensorboardAppSettingsWithImage,
+			"kernelGatewayAppSettings":             testAccAWSSagemakerDomain_kernelGatewayAppSettings,
+			"kernelGatewayAppSettings_customImage": testAccAWSSagemakerDomain_kernelGatewayAppSettings_customImage,
+			"jupyterServerAppSettings":             testAccAWSSagemakerDomain_jupyterServerAppSettings,
+			"kms":                                  testAccAWSSagemakerDomain_kms,
+			"securityGroup":                        testAccAWSSagemakerDomain_securityGroup,
+			"sharingSettings":                      testAccAWSSagemakerDomain_sharingSettings,
+		},
+		"UserProfile": {
+			"basic":                           testAccAWSSagemakerUserProfile_basic,
+			"disappears":                      testAccAWSSagemakerUserProfile_tags,
+			"tags":                            testAccAWSSagemakerUserProfile_disappears,
+			"tensorboardAppSettings":          testAccAWSSagemakerUserProfile_tensorboardAppSettings,
+			"tensorboardAppSettingsWithImage": testAccAWSSagemakerUserProfile_tensorboardAppSettingsWithImage,
+			"kernelGatewayAppSettings":        testAccAWSSagemakerUserProfile_kernelGatewayAppSettings,
+			"jupyterServerAppSettings":        testAccAWSSagemakerUserProfile_jupyterServerAppSettings,
+		},
+		"Workteam": {
+			"disappears":         testAccAWSSagemakerWorkteam_disappears,
+			"CognitoConfig":      testAccAWSSagemakerWorkteam_cognitoConfig,
+			"NotificationConfig": testAccAWSSagemakerWorkteam_notificationConfig,
+			"OidcConfig":         testAccAWSSagemakerWorkteam_oidcConfig,
+			"Tags":               testAccAWSSagemakerWorkteam_tags,
+		},
+	}
+
+	for group, m := range testCases {
+		m := m
+		t.Run(group, func(t *testing.T) {
+			for name, tc := range m {
+				tc := tc
+				t.Run(name, func(t *testing.T) {
+					tc(t)
+				})
+			}
+		})
+	}
+}

aws/internal/service/sagemaker/finder/finder.go

@@ -209,3 +209,31 @@ func WorkforceByName(conn *sagemaker.SageMaker, name string) (*sagemaker.Workfor
 
 	return output.Workforce, nil
 }
+
+func WorkteamByName(conn *sagemaker.SageMaker, name string) (*sagemaker.Workteam, error) {
+	input := &sagemaker.DescribeWorkteamInput{
+		WorkteamName: aws.String(name),
+	}
+
+	output, err := conn.DescribeWorkteam(input)
+
+	if tfawserr.ErrMessageContains(err, "ValidationException", "The work team") {
+		return nil, &resource.NotFoundError{
+			LastError:   err,
+			LastRequest: input,
+		}
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	if output == nil || output.Workteam == nil {
+		return nil, &resource.NotFoundError{
+			Message:     "Empty result",
+			LastRequest: input,
+		}
+	}
+
+	return output.Workteam, nil
+}

aws/provider.go

@@ -1014,6 +1014,7 @@ func Provider() *schema.Provider {
 			"aws_sagemaker_notebook_instance":                         resourceAwsSagemakerNotebookInstance(),
 			"aws_sagemaker_user_profile":                              resourceAwsSagemakerUserProfile(),
 			"aws_sagemaker_workforce":                                 resourceAwsSagemakerWorkforce(),
+			"aws_sagemaker_workteam":                                  resourceAwsSagemakerWorkteam(),
 			"aws_schemas_discoverer":                                  resourceAwsSchemasDiscoverer(),
 			"aws_schemas_registry":                                    resourceAwsSchemasRegistry(),
 			"aws_schemas_schema":                                      resourceAwsSchemasSchema(),

aws/resource_aws_sagemaker_domain_test.go

@@ -17,53 +17,6 @@ import (
 	"github.com/terraform-providers/terraform-provider-aws/aws/internal/service/sagemaker/finder"
 )
 
-// Tests are serialized as SagmMaker Domain resources are limited to 1 per account by default.
-// SageMaker UserProfile and App depend on the Domain resources and as such are also part of the serialized test suite.
-func TestAccAWSSagemakerDomain_serial(t *testing.T) {
-	testCases := map[string]map[string]func(t *testing.T){
-		"Domain": {
-			"basic":                                testAccAWSSagemakerDomain_basic,
-			"disappears":                           testAccAWSSagemakerDomain_tags,
-			"tags":                                 testAccAWSSagemakerDomain_disappears,
-			"tensorboardAppSettings":               testAccAWSSagemakerDomain_tensorboardAppSettings,
-			"tensorboardAppSettingsWithImage":      testAccAWSSagemakerDomain_tensorboardAppSettingsWithImage,
-			"kernelGatewayAppSettings":             testAccAWSSagemakerDomain_kernelGatewayAppSettings,
-			"kernelGatewayAppSettings_customImage": testAccAWSSagemakerDomain_kernelGatewayAppSettings_customImage,
-			"jupyterServerAppSettings":             testAccAWSSagemakerDomain_jupyterServerAppSettings,
-			"kms":                                  testAccAWSSagemakerDomain_kms,
-			"securityGroup":                        testAccAWSSagemakerDomain_securityGroup,
-			"sharingSettings":                      testAccAWSSagemakerDomain_sharingSettings,
-		},
-		"UserProfile": {
-			"basic":                           testAccAWSSagemakerUserProfile_basic,
-			"disappears":                      testAccAWSSagemakerUserProfile_tags,
-			"tags":                            testAccAWSSagemakerUserProfile_disappears,
-			"tensorboardAppSettings":          testAccAWSSagemakerUserProfile_tensorboardAppSettings,
-			"tensorboardAppSettingsWithImage": testAccAWSSagemakerUserProfile_tensorboardAppSettingsWithImage,
-			"kernelGatewayAppSettings":        testAccAWSSagemakerUserProfile_kernelGatewayAppSettings,
-			"jupyterServerAppSettings":        testAccAWSSagemakerUserProfile_jupyterServerAppSettings,
-		},
-		"App": {
-			"basic":        testAccAWSSagemakerApp_basic,
-			"disappears":   testAccAWSSagemakerApp_tags,
-			"tags":         testAccAWSSagemakerApp_disappears,
-			"resourceSpec": testAccAWSSagemakerApp_resourceSpec,
-		},
-	}
-
-	for group, m := range testCases {
-		m := m
-		t.Run(group, func(t *testing.T) {
-			for name, tc := range m {
-				tc := tc
-				t.Run(name, func(t *testing.T) {
-					tc(t)
-				})
-			}
-		})
-	}
-}
-
 func init() {
 	resource.AddTestSweepers("aws_sagemaker_domain", &resource.Sweeper{
 		Name: "aws_sagemaker_domain",

aws/resource_aws_sagemaker_workforce.go

@@ -194,7 +194,6 @@ func resourceAwsSagemakerWorkforceRead(d *schema.ResourceData, meta interface{})
 
 	if err != nil {
 		return fmt.Errorf("error reading SageMaker Workforce (%s): %w", d.Id(), err)
-
 	}
 
 	d.Set("arn", workforce.WorkforceArn)

aws/resource_aws_sagemaker_workforce_test.go

@@ -20,6 +20,9 @@ func init() {
 	resource.AddTestSweepers("aws_sagemaker_workforce", &resource.Sweeper{
 		Name: "aws_sagemaker_workforce",
 		F:    testSweepSagemakerWorkforces,
+		Dependencies: []string{
+			"aws_sagemaker_workteam",
+		},
 	})
 }