From 539e4da03f4a55ae795d9247837c6dcbc8ff5ebf Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Sun, 17 Aug 2014 20:48:50 -0700
Subject: [PATCH] helper/schema: more tests

---
 helper/schema/resource_data_test.go | 34 +++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/helper/schema/resource_data_test.go b/helper/schema/resource_data_test.go
index c5b818feef..7828e3690c 100644
--- a/helper/schema/resource_data_test.go
+++ b/helper/schema/resource_data_test.go
@@ -796,3 +796,37 @@ func TestResourceDataState(t *testing.T) {
 		}
 	}
 }
+
+func TestResourceDataSetId(t *testing.T) {
+	d := &ResourceData{}
+	d.SetId("foo")
+
+	actual := d.State()
+	if actual.ID != "foo" {
+		t.Fatalf("bad: %#v", actual)
+	}
+}
+
+func TestResourceDataSetId_clear(t *testing.T) {
+	d := &ResourceData{
+		state: &terraform.ResourceState{ID: "bar"},
+	}
+	d.SetId("")
+
+	actual := d.State()
+	if actual.ID != "" {
+		t.Fatalf("bad: %#v", actual)
+	}
+}
+
+func TestResourceDataSetId_override(t *testing.T) {
+	d := &ResourceData{
+		state: &terraform.ResourceState{ID: "bar"},
+	}
+	d.SetId("foo")
+
+	actual := d.State()
+	if actual.ID != "foo" {
+		t.Fatalf("bad: %#v", actual)
+	}
+}