From ac32d560614d2fbff910335d40cea3ac8cfae00f Mon Sep 17 00:00:00 2001 From: Dave Henderson Date: Thu, 21 May 2020 20:36:57 -0400 Subject: refactoring: extract stdin and boltdb readers Signed-off-by: Dave Henderson --- data/datasource_stdin_test.go | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 data/datasource_stdin_test.go (limited to 'data/datasource_stdin_test.go') diff --git a/data/datasource_stdin_test.go b/data/datasource_stdin_test.go new file mode 100644 index 00000000..7fc02318 --- /dev/null +++ b/data/datasource_stdin_test.go @@ -0,0 +1,22 @@ +package data + +import ( + "strings" + "testing" + + "github.com/stretchr/testify/assert" +) + +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) +} -- cgit v1.2.3