From cf2dd43f301a7eea751de0f5e7e444ce7ededb98 Mon Sep 17 00:00:00 2001
From: James Bardin <j.bardin@gmail.com>
Date: Tue, 5 May 2020 21:34:31 -0400
Subject: [PATCH] add data responses where they were missing

A few test with had data sources that were never read before, and needed
to get valid responses for the tests.
---
 terraform/context_plan_test.go    | 16 +++++++++++++++-
 terraform/context_refresh_test.go |  5 +++++
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/terraform/context_plan_test.go b/terraform/context_plan_test.go
index 4ff1dc5f27..282f07f1b1 100644
--- a/terraform/context_plan_test.go
+++ b/terraform/context_plan_test.go
@@ -5504,11 +5504,18 @@ func TestContext2Plan_invalidOutput(t *testing.T) {
 data "aws_data_source" "name" {}
 
 output "out" {
-  value = "${data.aws_data_source.name.missing}"
+  value = data.aws_data_source.name.missing
 }`,
 	})
 
 	p := testProvider("aws")
+	p.ReadDataSourceResponse = providers.ReadDataSourceResponse{
+		State: cty.ObjectVal(map[string]cty.Value{
+			"id":  cty.StringVal("data_id"),
+			"foo": cty.StringVal("foo"),
+		}),
+	}
+
 	ctx := testContext2(t, &ContextOpts{
 		Config: m,
 		Providers: map[addrs.Provider]providers.Factory{
@@ -5549,6 +5556,13 @@ resource "aws_instance" "foo" {
 	})
 
 	p := testProvider("aws")
+	p.ReadDataSourceResponse = providers.ReadDataSourceResponse{
+		State: cty.ObjectVal(map[string]cty.Value{
+			"id":  cty.StringVal("data_id"),
+			"foo": cty.StringVal("foo"),
+		}),
+	}
+
 	ctx := testContext2(t, &ContextOpts{
 		Config: m,
 		Providers: map[addrs.Provider]providers.Factory{
diff --git a/terraform/context_refresh_test.go b/terraform/context_refresh_test.go
index 4f739e1293..9ae8c55cc8 100644
--- a/terraform/context_refresh_test.go
+++ b/terraform/context_refresh_test.go
@@ -1612,6 +1612,11 @@ func TestContext2Refresh_dataResourceDependsOn(t *testing.T) {
 		},
 	}
 	p.DiffFn = testDiffFn
+	p.ReadDataSourceResponse = providers.ReadDataSourceResponse{
+		State: cty.ObjectVal(map[string]cty.Value{
+			"compute": cty.StringVal("value"),
+		}),
+	}
 
 	state := states.NewState()
 	root := state.EnsureModule(addrs.RootModuleInstance)