fix(log_test.go): add more logging tests

Author: Aaron Schlesinger

log/log_test.go

@@ -27,15 +27,16 @@ func TestMsg(t *testing.T) {
 
 func TestErr(t *testing.T) {
 	stdout, _, stderr, _ := getWriters()
-	lg := NewLogger(stdout, stderr, true)
+	lg := NewLogger(stdout, stderr, false)
 	lg.Err("hello %s", world)
 }
 
 func TestInfo(t *testing.T) {
-	stdout, _, stderr, err := getWriters()
-	lg := NewLogger(stdout, stderr, true)
+	stdout, out, stderr, err := getWriters()
+	lg := NewLogger(stdout, stderr, false)
 	lg.Info("hello %s", world)
-	assert.Equal(t, err.Len(), 0, "stderr buffer length")
+	assert.Equal(t, string(err.Bytes()), addColor(InfoPrefix+" ", Green), "stderr output")
+	assert.Equal(t, string(out.Bytes()), "hello world\n", "stdout output")
 }
 
 func TestDebug(t *testing.T) {
@@ -46,6 +47,20 @@ func TestDebug(t *testing.T) {
 	assert.Equal(t, out.Len(), 0, "stdout buffer length")
 	assert.Equal(t, err.Len(), 0, "stderr buffer length")
 	lgOn.Debug("hello %s", world)
-	assert.Equal(t, err.Len(), 0, "stderr buffer length")
-	assert.True(t, out.Len() > 0, "stdout buffer was empty, expected it to have debug output")
+	assert.Equal(t, string(err.Bytes()), addColor(DebugPrefix+" ", Cyan), "stderr output")
+	assert.Equal(t, string(out.Bytes()), "hello world\n", "stdout output")
+}
+
+func TestWarn(t *testing.T) {
+	stdout, out, stderr, err := getWriters()
+	lg := NewLogger(stdout, stderr, false)
+	lg.Warn("hello %s", world)
+	assert.Equal(t, string(err.Bytes()), addColor(WarnPrefix+" ", Yellow), "stderr output")
+	assert.Equal(t, string(out.Bytes()), "hello world\n", "stdout output")
+}
+
+func TestAppendNewLine(t *testing.T) {
+	str := "abc"
+	newStr := appendNewLine(str)
+	assert.Equal(t, newStr, str+"\n", "new string")
 }