Handle multi-document YAML streams

Signed-off-by: Dave Henderson <[email protected]>

data/data_test.go

@@ -56,7 +56,7 @@ func TestUnmarshalArray(t *testing.T) {
 		assert.EqualValues(t, expected, actual)
 	}
 	test(JSONArray(`["foo","bar",{"baz":{"qux": true},"quux":{"42":18},"corge":{"false":"blah"}}]`))
-	test(YAMLArray(`
+	test(YAMLArray(`---
 - foo
 - bar
 - baz:

tests/integration/datasources_file_test.go

@@ -38,6 +38,14 @@ FOO.BAR = "values can be double-quoted, and shell\nescapes are supported"
 
 BAZ = "variable expansion: ${FOO}"
 QUX='single quotes ignore $variables'
+`,
+			"multidoc.yaml": `---
+# empty document
+---
+foo: bar
+---
+foo: baz
+...
 `,
 		}),
 		fs.WithDir("sortorder", fs.WithFiles(map[string]string{
@@ -185,4 +193,12 @@ bar`})
   "FOO.BAR": "values can be double-quoted, and shell\nescapes are supported",
   "QUX": "single quotes ignore $variables"
 }`})
+
+	result = icmd.RunCmd(icmd.Command(GomplateBin,
+		"-c=multidoc.yaml",
+		"-i", `{{ .multidoc.foo }}`,
+	), func(c *icmd.Cmd) {
+		c.Dir = s.tmpDir.Path()
+	})
+	result.Assert(c, icmd.Expected{ExitCode: 0, Out: "bar"})
 }