Add tests for TF_ADD_MODULE_PATH.

Author: cgroschupp

parser_test.go

@@ -3,12 +3,56 @@ package main
 import (
 	"bytes"
 	"encoding/json"
+	"os"
 	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
+const exampleStateFileEnvModulePath = `
+{
+	"version": 1,
+	"serial": 1,
+	"modules": [
+		{
+			"path": [
+        "root",
+        "web"
+      ],
+			"resources": {
+				"libvirt_domain.fourteen": {
+					"type": "libvirt_domain",
+					"primary": {
+						"id": "824c29be-2164-44c8-83e0-787705571d95",
+						"attributes": {
+							"name": "fourteen",
+							"network_interface.#": "1",
+							"network_interface.0.addresses.#": "1",
+							"network_interface.0.addresses.0": "192.168.102.14",
+							"network_interface.0.mac": "96:EE:4D:BD:B2:45"
+						}
+					}
+				}
+			}
+		}
+	]
+}`
+
+const expectedListOutputEnvModulePath = `
+{
+	"all":	 {
+		"hosts": [
+			"192.168.102.14"
+		],
+		"vars": {
+		}
+	},
+	"root.web.fourteen":	 ["192.168.102.14"],
+	"root.web.fourteen.0":	 ["192.168.102.14"],
+	"type_libvirt_domain": ["192.168.102.14"]
+}`
+
 const exampleStateFile = `
 {
 	"version": 1,
@@ -665,6 +709,33 @@ func TestListCommand(t *testing.T) {
 	assert.Equal(t, exp, act)
 }
 
+func TestListCommandEnvModulePath(t *testing.T) {
+	var s state
+	r := strings.NewReader(exampleStateFileEnvModulePath)
+	err := s.read(r)
+	assert.NoError(t, err)
+
+	// Decode expectation as JSON
+	var exp interface{}
+	err = json.Unmarshal([]byte(expectedListOutputEnvModulePath), &exp)
+	assert.NoError(t, err)
+
+	// Run the command, capture the output
+	var stdout, stderr bytes.Buffer
+	os.Setenv("TF_ADD_MODULE_PATH", "true")
+	exitCode := cmdList(&stdout, &stderr, &s)
+	os.Unsetenv("TF_ADD_MODULE_PATH")
+	assert.Equal(t, 0, exitCode)
+	assert.Equal(t, "", stderr.String())
+
+	// Decode the output to compare
+	var act interface{}
+	err = json.Unmarshal([]byte(stdout.String()), &act)
+	assert.NoError(t, err)
+
+	assert.Equal(t, exp, act)
+}
+
 func TestHostCommand(t *testing.T) {
 	var s state
 	r := strings.NewReader(exampleStateFile)