From cec23e66f9bd5022845162ae4dd3f2633b5236fa Mon Sep 17 00:00:00 2001 From: Dave Henderson Date: Sat, 28 May 2022 19:24:48 -0400 Subject: General refactoring & cleanup Signed-off-by: Dave Henderson --- data/datasource_stdin_test.go | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) (limited to 'data/datasource_stdin_test.go') 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) } -- cgit v1.2.3