148 lines
3.2 KiB
Go
148 lines
3.2 KiB
Go
package msg_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"git.wntrmute.dev/kyle/goutils/msg"
|
|
)
|
|
|
|
func checkExpected(buf *bytes.Buffer, expected string) bool {
|
|
return buf.String() == expected
|
|
}
|
|
|
|
func resetBuf() *bytes.Buffer {
|
|
buf := &bytes.Buffer{}
|
|
msg.SetWriter(buf)
|
|
|
|
return buf
|
|
}
|
|
|
|
func TestVerbosePrint(t *testing.T) {
|
|
buf := resetBuf()
|
|
|
|
msg.SetVerbose(false) // ensure verbose is explicitly not set
|
|
|
|
msg.Vprint("hello, world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.Vprintf("hello, %s", "world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.Vprintln("hello, world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.SetVerbose(true)
|
|
msg.Vprint("hello, world")
|
|
if !checkExpected(buf, "hello, world") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.Vprintf("hello, %s", "world")
|
|
if !checkExpected(buf, "hello, world") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.Vprintln("hello, world")
|
|
if !checkExpected(buf, "hello, world\n") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world\n", buf.String())
|
|
}
|
|
}
|
|
|
|
func TestQuietPrint(t *testing.T) {
|
|
buf := resetBuf()
|
|
|
|
msg.SetQuiet(true)
|
|
|
|
msg.Qprint("hello, world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.Qprintf("hello, %s", "world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.Qprintln("hello, world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.SetQuiet(false)
|
|
msg.Qprint("hello, world")
|
|
if !checkExpected(buf, "hello, world") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.Qprintf("hello, %s", "world")
|
|
if !checkExpected(buf, "hello, world") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.Qprintln("hello, world")
|
|
if !checkExpected(buf, "hello, world\n") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world\n", buf.String())
|
|
}
|
|
}
|
|
|
|
func TestDebugPrint(t *testing.T) {
|
|
buf := resetBuf()
|
|
|
|
msg.SetDebug(false) // ensure debug is explicitly not set
|
|
|
|
msg.Dprint("hello, world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.Dprintf("hello, %s", "world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.Dprintln("hello, world")
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.StackTrace()
|
|
if buf.Len() != 0 {
|
|
t.Fatalf("expected no output, have %s", buf.String())
|
|
}
|
|
|
|
msg.SetDebug(true)
|
|
msg.Dprint("hello, world")
|
|
if !checkExpected(buf, "hello, world") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.Dprintf("hello, %s", "world")
|
|
if !checkExpected(buf, "hello, world") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.Dprintln("hello, world")
|
|
if !checkExpected(buf, "hello, world\n") {
|
|
t.Fatalf("expected output %q, have %q", "hello, world\n", buf.String())
|
|
}
|
|
buf.Reset()
|
|
|
|
msg.StackTrace()
|
|
if buf.Len() == 0 {
|
|
t.Fatal("expected stack trace output, received no output")
|
|
}
|
|
}
|