Fix bulk import test

hashicorp · Mar 15, 2022 · 21394f9 · 21394f9
1 parent 34c0d39
commit 21394f9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/internal/command/import_test.go b/internal/command/import_test.go
@@ -1059,11 +1059,13 @@ func TestImport_bulk(t *testing.T) {
 			},
 		}
 	}
-	p.GetSchemaReturn = &terraform.ProviderSchema{
-		ResourceTypes: map[string]*configschema.Block{
+	p.GetProviderSchemaResponse = &providers.GetProviderSchemaResponse{
+		ResourceTypes: map[string]providers.Schema{
 			"test_instance": {
-				Attributes: map[string]*configschema.Attribute{
-					"id": {Type: cty.String, Optional: true, Computed: true},
+				Block: &configschema.Block{
+					Attributes: map[string]*configschema.Attribute{
+						"id": {Type: cty.String, Optional: true, Computed: true},
+					},
 				},
 			},
 		},