Merge pull request #233 from hairyhenderson/stdin-datasource-230

Adding support for stdin: scheme for datasources

1 files changed, 21 insertions, 0 deletions

diff --git a/data/datasource_test.go b/data/datasource_test.go
index a3b6f31e..ac7b884a 100644
--- a/data/datasource_test.go
+++ b/data/datasource_test.go
@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"net/url"
+	"strings"
 	"testing"
 
 	"github.com/blang/vfs"
@@ -307,3 +308,23 @@ func TestInclude(t *testing.T) {
 	actual := data.Include("foo")
 	assert.Equal(t, contents, actual)
 }
+
+type errorReader struct{}
+
+func (e errorReader) Read(p []byte) (n int, err error) {
+	return 0, fmt.Errorf("error")
+}
+
+func TestReadStdin(t *testing.T) {
+	defer func() {
+		stdin = nil
+	}()
+	stdin = strings.NewReader("foo")
+	out, err := readStdin(nil)
+	assert.NoError(t, err)
+	assert.Equal(t, []byte("foo"), out)
+
+	stdin = errorReader{}
+	_, err = readStdin(nil)
+	assert.Error(t, err)
+}