From d969f97e730d527a568a0929ce52df70e062bca0 Mon Sep 17 00:00:00 2001
From: Jake Champlin <jake.champlin.27@gmail.com>
Date: Thu, 3 Aug 2017 17:53:07 -0400
Subject: [PATCH] update tests

---
 helper/schema/resource_data_test.go | 43 +++++++++++++++++------------
 1 file changed, 26 insertions(+), 17 deletions(-)

diff --git a/helper/schema/resource_data_test.go b/helper/schema/resource_data_test.go
index fea3df2ed4..09aefb8f12 100644
--- a/helper/schema/resource_data_test.go
+++ b/helper/schema/resource_data_test.go
@@ -1271,11 +1271,18 @@ func TestResourceDataGetOkExists(t *testing.T) {
 			},
 
 			State: nil,
-			Diff:  nil,
+			Diff: &terraform.InstanceDiff{
+				Attributes: map[string]*terraform.ResourceAttrDiff{
+					"availability_zone": {
+						Old: "",
+						New: "",
+					},
+				},
+			},
 
 			Key:   "availability_zone",
 			Value: false,
-			Ok:    false,
+			Ok:    true,
 		},
 
 		{
@@ -1305,23 +1312,25 @@ func TestResourceDataGetOkExists(t *testing.T) {
 		},
 	}
 
-	for _, tc := range cases {
-		d, err := schemaMap(tc.Schema).Data(tc.State, tc.Diff)
-		if err != nil {
-			t.Fatalf("%s err: %s", tc.Name, err)
-		}
+	for i, tc := range cases {
+		t.Run(fmt.Sprintf("%d-%s", i, tc.Name), func(t *testing.T) {
+			d, err := schemaMap(tc.Schema).Data(tc.State, tc.Diff)
+			if err != nil {
+				t.Fatalf("%s err: %s", tc.Name, err)
+			}
 
-		v, ok := d.GetOkExists(tc.Key)
-		if s, ok := v.(*Set); ok {
-			v = s.List()
-		}
+			v, ok := d.GetOkExists(tc.Key)
+			if s, ok := v.(*Set); ok {
+				v = s.List()
+			}
 
-		if !reflect.DeepEqual(v, tc.Value) {
-			t.Fatalf("Bad %s: \n%#v", tc.Name, v)
-		}
-		if ok != tc.Ok {
-			t.Fatalf("%s: expected ok: %t, got: %t", tc.Name, tc.Ok, ok)
-		}
+			if !reflect.DeepEqual(v, tc.Value) {
+				t.Fatalf("Bad %s: \n%#v", tc.Name, v)
+			}
+			if ok != tc.Ok {
+				t.Fatalf("%s: expected ok: %t, got: %t", tc.Name, tc.Ok, ok)
+			}
+		})
 	}
 }