Add unit test for hostvars

jtopjian · Apr 12, 2018 · 5aa4e88 · 5aa4e88
1 parent f27ad05
commit 5aa4e88
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 6 deletions.
diff --git a/fixtures/main.tf b/fixtures/main.tf
@@ -18,5 +18,17 @@ resource "ansible_host" "host_1" {
   vars {
     ansible_user = "ubuntu"
     ansible_host = "1.2.3.4"
+    test         = "host_1"
+  }
+}
+
+resource "ansible_host" "host_2" {
+  inventory_hostname = "host_2"
+  groups             = ["group_1"]
+
+  vars {
+    ansible_user = "ubuntu"
+    ansible_host = "1.2.3.5"
+    test         = "host_2"
   }
 }
diff --git a/fixtures/terraform.tfstate b/fixtures/terraform.tfstate
@@ -54,9 +54,31 @@
                             "groups.0": "group_1",
                             "id": "host_1",
                             "inventory_hostname": "host_1",
-                            "vars.%": "2",
+                            "vars.%": "3",
                             "vars.ansible_host": "1.2.3.4",
-                            "vars.ansible_user": "ubuntu"
+                            "vars.ansible_user": "ubuntu",
+                            "vars.test": "host_1"
+                        },
+                        "meta": {},
+                        "tainted": false
+                    },
+                    "deposed": [],
+                    "provider": "provider.ansible"
+                },
+                "ansible_host.host_2": {
+                    "type": "ansible_host",
+                    "depends_on": [],
+                    "primary": {
+                        "id": "host_2",
+                        "attributes": {
+                            "groups.#": "1",
+                            "groups.0": "group_1",
+                            "id": "host_2",
+                            "inventory_hostname": "host_2",
+                            "vars.%": "3",
+                            "vars.ansible_host": "1.2.3.5",
+                            "vars.ansible_user": "ubuntu",
+                            "vars.test": "host_2"
                         },
                         "meta": {},
                         "tainted": false

diff --git a/state_test.go b/state_test.go
@@ -19,9 +19,26 @@ var expectedState = State{
 							"inventory_hostname": "host_1",
 							"groups.#":           "1",
 							"groups.0":           "group_1",
-							"vars.%":             "2",
+							"vars.%":             "3",
 							"vars.ansible_host":  "1.2.3.4",
 							"vars.ansible_user":  "ubuntu",
+							"vars.test":          "host_1",
+						},
+					},
+				},
+				"ansible_host.host_2": Resource{
+					Type: "ansible_host",
+					Primary: Primary{
+						ID: "host_2",
+						Attributes: map[string]string{
+							"id":                 "host_2",
+							"inventory_hostname": "host_2",
+							"groups.#":           "1",
+							"groups.0":           "group_1",
+							"vars.%":             "3",
+							"vars.ansible_host":  "1.2.3.5",
+							"vars.ansible_user":  "ubuntu",
+							"vars.test":          "host_2",
 						},
 					},
 				},
@@ -56,7 +73,7 @@ var expectedState = State{
 
 var expectedInventory = map[string]interface{}{
 	"group_1": map[string]interface{}{
-		"hosts":    []string{"host_1"},
+		"hosts":    []string{"host_1", "host_2"},
 		"children": []string{"group_2"},
 		"vars": map[string]interface{}{
 			"foo": "bar",
@@ -70,6 +87,12 @@ var expectedInventory = map[string]interface{}{
 			"host_1": map[string]interface{}{
 				"ansible_host": "1.2.3.4",
 				"ansible_user": "ubuntu",
+				"test":         "host_1",
+			},
+			"host_2": map[string]interface{}{
+				"ansible_host": "1.2.3.5",
+				"ansible_user": "ubuntu",
+				"test":         "host_2",
 			},
 		},
 	},
@@ -91,7 +114,7 @@ func TestState_basic(t *testing.T) {
 
 	assert.Equal(t, expectedGroups, actualGroups)
 
-	expectedHosts := []string{"host_1"}
+	expectedHosts := []string{"host_1", "host_2"}
 	actualHosts, err := actual.GetHostsForGroup(expectedGroups[0])
 	if err != nil {
 		t.Fatal(err)
@@ -102,9 +125,10 @@ func TestState_basic(t *testing.T) {
 	expectedVars := map[string]interface{}{
 		"ansible_host": "1.2.3.4",
 		"ansible_user": "ubuntu",
+		"test":         "host_1",
 	}
 
-	actualVars, err := actual.GetVarsForHost(expectedHosts[0])
+	actualVars, err := actual.GetVarsForHost("host_1")
 	if err != nil {
 		t.Fatal(err)
 	}