What if ... ImportStateBlockConfig

Author: bbasata

helper/resource/testing.go

@@ -633,6 +633,10 @@ type TestStep struct {
 	// ID of that resource.
 	ImportState bool
 
+	// ImportStateBlockConfig, if non-empty, supplies declarative import
+	// configuration. This is (?mutually exclusive of ImportStateID + ResourceName?).
+	ImportStateBlockConfig string
+
 	// ImportStateId is the ID to perform an ImportState operation with.
 	// This is optional. If it isn't set, then the resource ID is automatically
 	// determined by inspecting the state for ResourceName's ID.

helper/resource/testing_new_import_block_test.go

@@ -0,0 +1,96 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package resource
+
+import (
+	"testing"
+
+	"github.com/hashicorp/terraform-plugin-go/tfprotov6"
+	"github.com/hashicorp/terraform-plugin-go/tftypes"
+
+	"github.com/hashicorp/terraform-plugin-testing/internal/testing/testprovider"
+	"github.com/hashicorp/terraform-plugin-testing/internal/testing/testsdk/providerserver"
+	"github.com/hashicorp/terraform-plugin-testing/internal/testing/testsdk/resource"
+	"github.com/hashicorp/terraform-plugin-testing/tfversion"
+)
+
+func TestTest_TestStep_ImportBlockVerify(t *testing.T) {
+	t.Parallel()
+
+	UnitTest(t, TestCase{
+		TerraformVersionChecks: []tfversion.TerraformVersionCheck{
+			tfversion.SkipBelow(tfversion.Version1_5_0), // import blocks are only available in v1.5.0 and later
+		},
+		ProtoV6ProviderFactories: map[string]func() (tfprotov6.ProviderServer, error){
+			"examplecloud": providerserver.NewProviderServer(testprovider.Provider{
+				Resources: map[string]testprovider.Resource{
+					"examplecloud_thing": {
+						CreateResponse: &resource.CreateResponse{
+							NewState: tftypes.NewValue(
+								tftypes.Object{
+									AttributeTypes: map[string]tftypes.Type{
+										"id":    tftypes.String,
+										"other": tftypes.String,
+									},
+								},
+								map[string]tftypes.Value{
+									"id":    tftypes.NewValue(tftypes.String, "resource-test"),
+									"other": tftypes.NewValue(tftypes.String, "testvalue"),
+								},
+							),
+						},
+						ImportStateResponse: &resource.ImportStateResponse{
+							State: tftypes.NewValue(
+								tftypes.Object{
+									AttributeTypes: map[string]tftypes.Type{
+										"id":    tftypes.String,
+										"other": tftypes.String,
+									},
+								},
+								map[string]tftypes.Value{
+									"id":    tftypes.NewValue(tftypes.String, "resource-test"),
+									"other": tftypes.NewValue(tftypes.String, "testvalue"),
+								},
+							),
+						},
+						SchemaResponse: &resource.SchemaResponse{
+							Schema: &tfprotov6.Schema{
+								Block: &tfprotov6.SchemaBlock{
+									Attributes: []*tfprotov6.SchemaAttribute{
+										{
+											Name:     "id",
+											Type:     tftypes.String,
+											Computed: true,
+										},
+										{
+											Name:     "other",
+											Type:     tftypes.String,
+											Computed: true,
+										},
+									},
+								},
+							},
+						},
+					},
+				},
+			}),
+		},
+		Steps: []TestStep{
+			{
+				Config: `resource "examplecloud_thing" "test" {}`,
+			},
+			{
+				ImportState:       true,
+				ImportStateVerify: true,
+				ImportStateBlockConfig: `import {
+              			  to = examplecloud_thing.test
+				  identity = {
+				    hat = "derby"
+				    cat = "garfield"
+				  }
+				}`,
+			},
+		},
+	})
+}