confirm correct writer is used

pterm · Mar 23, 2022 · bcf821b · bcf821b
1 parent c50e99c
commit bcf821b
Showing 1 changed file with 36 additions and 0 deletions.
diff --git a/print_test.go b/print_test.go
@@ -161,12 +161,24 @@ func TestFprint(t *testing.T) {
 		pterm.Output = true
 		for _, randomString := range internal.RandomStrings {
 			out := captureStdout(func(w io.Writer) {
+				// set default to null to confirm that its correctly using the provided writer
+				pterm.SetDefaultOutput(nil)
 				pterm.Fprint(w, randomString)
 			})
 			testza.AssertEqual(t, randomString, out)
 		}
 	})
 
+	t.Run("confirm defaults to default output when no writer provided", func(t *testing.T) {
+		pterm.Output = true
+		for _, randomString := range internal.RandomStrings {
+			out := captureStdout(func(w io.Writer) {
+				pterm.Fprint(nil, randomString)
+			})
+			testza.AssertEqual(t, randomString, out)
+		}
+	})
+
 	t.Run("disabled output", func(t *testing.T) {
 		pterm.Output = false
 		for _, randomString := range internal.RandomStrings {
@@ -184,12 +196,24 @@ func TestFprintln(t *testing.T) {
 		pterm.Output = true
 		for _, randomString := range internal.RandomStrings {
 			out := captureStdout(func(w io.Writer) {
+				// set default to null to confirm that its correctly using the provided writer
+				pterm.SetDefaultOutput(nil)
 				pterm.Fprintln(w, randomString)
 			})
 			testza.AssertEqual(t, randomString+"\n", out)
 		}
 	})
 
+	t.Run("confirm defaults to default output when no writer provided", func(t *testing.T) {
+		pterm.Output = true
+		for _, randomString := range internal.RandomStrings {
+			out := captureStdout(func(w io.Writer) {
+				pterm.Fprintln(nil, randomString)
+			})
+			testza.AssertEqual(t, randomString+"\n", out)
+		}
+	})
+
 	t.Run("disabled output", func(t *testing.T) {
 		pterm.Output = false
 		for _, randomString := range internal.RandomStrings {
@@ -230,12 +254,24 @@ func TestFprinto(t *testing.T) {
 		pterm.Output = true
 		for _, randomString := range internal.RandomStrings {
 			out := captureStdout(func(w io.Writer) {
+				// set default to null to confirm that its correctly using the provided writer
+				pterm.SetDefaultOutput(nil)
 				pterm.Fprinto(w, randomString)
 			})
 			testza.AssertEqual(t, "\r"+randomString, out)
 		}
 	})
 
+	t.Run("confirm defaults to default output when no writer provided", func(t *testing.T) {
+		pterm.Output = true
+		for _, randomString := range internal.RandomStrings {
+			out := captureStdout(func(w io.Writer) {
+				pterm.Fprinto(nil, randomString)
+			})
+			testza.AssertEqual(t, "\r"+randomString, out)
+		}
+	})
+
 	t.Run("disabled output", func(t *testing.T) {
 		pterm.Output = false
 		for _, randomString := range internal.RandomStrings {