Exporting the writer used when templates are sent to Stdout

Signed-off-by: Dave Henderson <dhenderson@gmail.com>

1 files changed, 20 insertions, 0 deletions

diff --git a/gomplate_test.go b/gomplate_test.go
index a90a2b20..f97487bc 100644
--- a/gomplate_test.go
+++ b/gomplate_test.go
@@ -2,6 +2,7 @@ package gomplate
 
 import (
 	"bytes"
+	"io"
 	"net/http/httptest"
 	"os"
 	"testing"
@@ -15,6 +16,15 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+// like ioutil.NopCloser(), except for io.WriteClosers...
+type nopWCloser struct {
+	io.Writer
+}
+
+func (n *nopWCloser) Close() error {
+	return nil
+}
+
 func testTemplate(g *gomplate, tmpl string) string {
 	var out bytes.Buffer
 	err := g.runTemplate(&tplate{name: "testtemplate", contents: tmpl, target: &out})
@@ -148,3 +158,13 @@ func TestCustomDelim(t *testing.T) {
 	}
 	assert.Equal(t, "hi", testTemplate(g, `[print "hi"]`))
 }
+
+func TestRunTemplates(t *testing.T) {
+	defer func() { Stdout = os.Stdout }()
+	buf := &bytes.Buffer{}
+	Stdout = &nopWCloser{buf}
+	config := &Config{Input: "foo"}
+	err := RunTemplates(config)
+	assert.NoError(t, err)
+	assert.Equal(t, "foo", buf.String())
+}