Add test utils

knqyf263 · Aug 28, 2022 · f38e79f · f38e79f
1 parent f5b4dae
commit f38e79f
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/tests/stdout.go b/tests/stdout.go
@@ -0,0 +1,30 @@
+package tests
+
+import (
+	"bytes"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestStdout(t *testing.T, fn func()) string {
+	t.Helper()
+	old := os.Stdout
+	t.Cleanup(func() {
+		os.Stdout = old
+	})
+	r, w, err := os.Pipe()
+	require.NoError(t, err)
+
+	os.Stdout = w
+	fn()
+	w.Close()
+
+	var buffer bytes.Buffer
+	_, err = buffer.ReadFrom(r)
+	require.NoError(t, err)
+
+	s := buffer.String()
+	return s[:len(s)-1]
+}