General refactoring & cleanup

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

1 files changed, 2 insertions, 5 deletions

diff --git a/data/datasource_stdin_test.go b/data/datasource_stdin_test.go
index 1b92914e..6cd34133 100644
--- a/data/datasource_stdin_test.go
+++ b/data/datasource_stdin_test.go
@@ -11,15 +11,12 @@ import (
 func TestReadStdin(t *testing.T) {
 	ctx := context.Background()
 
-	defer func() {
-		stdin = nil
-	}()
-	stdin = strings.NewReader("foo")
+	ctx = ContextWithStdin(ctx, strings.NewReader("foo"))
 	out, err := readStdin(ctx, nil)
 	assert.NoError(t, err)
 	assert.Equal(t, []byte("foo"), out)
 
-	stdin = errorReader{}
+	ctx = ContextWithStdin(ctx, errorReader{})
 	_, err = readStdin(ctx, nil)
 	assert.Error(t, err)
 }