add TestResourceDataWithIdentityRaw test helper (#1475)

Resolves #1464

Author: ansgarm

helper/schema/testing.go

@@ -30,3 +30,20 @@ func TestResourceDataRaw(t testing.T, schema map[string]*Schema, raw map[string]
 
 	return result
 }
+
+// TestResourceDataWithIdentityRaw creates a ResourceData with an identity from a raw identity map.
+func TestResourceDataWithIdentityRaw(t testing.T, schema map[string]*Schema, identitySchema map[string]*Schema, raw map[string]string) *ResourceData {
+	t.Helper()
+
+	sm := schemaMapWithIdentity{schema, identitySchema}
+	state := terraform.InstanceState{
+		Identity: raw,
+	}
+
+	result, err := sm.Data(&state, nil)
+	if err != nil {
+		t.Fatalf("err: %s", err)
+	}
+
+	return result
+}