bedrock: Use 'testing.Short()' guards.

hashicorp · Feb 1, 2024 · 81ddd4a · 81ddd4a
1 parent f5c1165
commit 81ddd4a
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/internal/service/bedrock/custom_model_data_source_test.go b/internal/service/bedrock/custom_model_data_source_test.go
@@ -15,6 +15,10 @@ import (
 
 func TestAccBedrockCustomModelDataSource_basic(t *testing.T) {
 	ctx := acctest.Context(t)
+	if testing.Short() {
+		t.Skip("skipping long-running test in short mode")
+	}
+
 	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
 	resourceName := "aws_bedrock_custom_model.test"
 	datasourceName := "data.aws_bedrock_custom_model.test"

diff --git a/internal/service/bedrock/custom_model_test.go b/internal/service/bedrock/custom_model_test.go
@@ -205,6 +205,10 @@ func TestAccBedrockCustomModel_validationDataConfig(t *testing.T) {
 
 func TestAccBedrockCustomModel_validationDataConfigWaitForCompletion(t *testing.T) {
 	ctx := acctest.Context(t)
+	if testing.Short() {
+		t.Skip("skipping long-running test in short mode")
+	}
+
 	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
 	resourceName := "aws_bedrock_custom_model.test"
 	var v bedrock.GetModelCustomizationJobOutput

diff --git a/internal/service/bedrock/custom_models_data_source_test.go b/internal/service/bedrock/custom_models_data_source_test.go
@@ -15,6 +15,10 @@ import (
 
 func TestAccBedrockCustomModelsDataSource_basic(t *testing.T) {
 	ctx := acctest.Context(t)
+	if testing.Short() {
+		t.Skip("skipping long-running test in short mode")
+	}
+
 	rName := sdkacctest.RandomWithPrefix(acctest.ResourcePrefix)
 	datasourceName := "data.aws_bedrock_custom_models.test"
 	resourceName := "aws_bedrock_custom_model.test"