diff options
Diffstat (limited to 'data/datasource_stdin_test.go')
| -rw-r--r-- | data/datasource_stdin_test.go | 7 |
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) } |