Inject stdin/out/err instead of always using os.Stdin/out/err

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

1 files changed, 4 insertions, 4 deletions

diff --git a/data/datasource_stdin.go b/data/datasource_stdin.go
index f23905d2..8216573b 100644
--- a/data/datasource_stdin.go
+++ b/data/datasource_stdin.go
@@ -1,16 +1,16 @@
 package data
 
 import (
+	"io"
 	"io/ioutil"
-	"os"
 
 	"github.com/pkg/errors"
 )
 
+// stdin - for overriding in tests
+var stdin io.Reader
+
 func readStdin(source *Source, args ...string) ([]byte, error) {
-	if stdin == nil {
-		stdin = os.Stdin
-	}
 	b, err := ioutil.ReadAll(stdin)
 	if err != nil {
 		return nil, errors.Wrapf(err, "Can't read %s", stdin)