ImportBlockWithResourceIdentity

Author: bbasata

helper/resource/importstate/examplecloud_test.go

@@ -75,6 +75,18 @@ func examplecloudResource() testprovider.Resource {
 				},
 			),
 		},
+		IdentitySchemaResponse: &resource.IdentitySchemaResponse{
+			Schema: &tfprotov6.ResourceIdentitySchema{
+				Version: 1,
+				IdentityAttributes: []*tfprotov6.ResourceIdentitySchemaAttribute{
+					{
+						Name:              "id",
+						Type:              tftypes.String,
+						RequiredForImport: true,
+					},
+				},
+			},
+		},
 		ImportStateResponse: &resource.ImportStateResponse{
 			State: tftypes.NewValue(
 				tftypes.Object{

helper/resource/importstate/resource_identity_test.go

@@ -7,7 +7,6 @@ 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"
@@ -28,18 +27,6 @@ func Test_ResourceIdentity(t *testing.T) {
 				Resources: map[string]testprovider.Resource{
 					"examplecloud_container": examplecloudResource(),
 				},
-				ResourceIdentitySchemas: map[string]*tfprotov6.ResourceIdentitySchema{
-					"examplecloud_container": {
-						Version: 1,
-						IdentityAttributes: []*tfprotov6.ResourceIdentitySchemaAttribute{
-							{
-								Name:              "id",
-								Type:              tftypes.String,
-								RequiredForImport: true,
-							},
-						},
-					},
-				},
 			}),
 		},
 		Steps: []r.TestStep{