import for_each from data source

Author: jbardin

internal/terraform/context_plan_import_test.go

@@ -1281,3 +1281,103 @@ import {
 		}
 	}
 }
+
+func TestContext2Plan_importForEachFromData(t *testing.T) {
+	m := testModuleInline(t, map[string]string{
+		"main.tf": `
+data "test_object" "d" {
+}
+
+resource "test_object" "a" {
+  count = 2
+  test_string = "foo"
+}
+
+import {
+  for_each = data.test_object.d.objects
+  to = test_object.a[each.key]
+  id = each.value
+}
+`,
+	})
+
+	p := simpleMockProvider()
+	p.GetProviderSchemaResponse = &providers.GetProviderSchemaResponse{
+		Provider: providers.Schema{Block: simpleTestSchema()},
+		ResourceTypes: map[string]providers.Schema{
+			"test_object": providers.Schema{Block: simpleTestSchema()},
+		},
+		DataSources: map[string]providers.Schema{
+			"test_object": providers.Schema{
+				Block: &configschema.Block{
+					Attributes: map[string]*configschema.Attribute{
+						"objects": {
+							Type:     cty.List(cty.String),
+							Computed: true,
+						},
+					},
+				},
+			},
+		},
+	}
+
+	ctx := testContext2(t, &ContextOpts{
+		Providers: map[addrs.Provider]providers.Factory{
+			addrs.NewDefaultProvider("test"): testProviderFuncFixed(p),
+		},
+	})
+
+	p.ReadDataSourceResponse = &providers.ReadDataSourceResponse{
+		State: cty.ObjectVal(map[string]cty.Value{
+			"objects": cty.ListVal([]cty.Value{
+				cty.StringVal("first_id"), cty.StringVal("second_id"),
+			}),
+		}),
+	}
+
+	p.ReadResourceResponse = &providers.ReadResourceResponse{
+		NewState: cty.ObjectVal(map[string]cty.Value{
+			"test_string": cty.StringVal("foo"),
+		}),
+	}
+	p.ImportResourceStateResponse = &providers.ImportResourceStateResponse{
+		ImportedResources: []providers.ImportedResource{
+			{
+				TypeName: "test_object",
+				State: cty.ObjectVal(map[string]cty.Value{
+					"test_string": cty.StringVal("foo"),
+				}),
+			},
+		},
+	}
+
+	plan, diags := ctx.Plan(m, states.NewState(), DefaultPlanOpts)
+	if diags.HasErrors() {
+		t.Fatalf("unexpected errors\n%s", diags.Err().Error())
+	}
+
+	firstAddr := mustResourceInstanceAddr(`test_object.a[0]`)
+	secondAddr := mustResourceInstanceAddr(`test_object.a[1]`)
+
+	for _, instPlan := range plan.Changes.Resources {
+		switch {
+		case instPlan.Addr.Equal(firstAddr):
+			if instPlan.Importing.ID != "first_id" {
+				t.Errorf("expected import ID of \"first_id\", got %q", instPlan.Importing.ID)
+			}
+		case instPlan.Addr.Equal(secondAddr):
+			if instPlan.Importing.ID != "second_id" {
+				t.Errorf("expected import ID of \"second_id\", got %q", instPlan.Importing.ID)
+			}
+		default:
+			t.Errorf("unexpected change for %s", instPlan.Addr)
+		}
+
+		if got, want := instPlan.Action, plans.NoOp; got != want {
+			t.Errorf("wrong planned action\ngot:  %s\nwant: %s", got, want)
+		}
+		if got, want := instPlan.ActionReason, plans.ResourceInstanceChangeNoReason; got != want {
+			t.Errorf("wrong action reason\ngot:  %s\nwant: %s", got, want)
+		}
+	}
+}