zlog/syslog_test.go

109 lines
2.7 KiB
Go
Raw Permalink Normal View History

// +build !binary_log
2017-11-27 18:01:32 +00:00
// +build !windows
package zerolog
2019-11-19 00:31:13 +00:00
import (
2020-10-12 13:16:04 +00:00
"bytes"
2019-11-19 00:31:13 +00:00
"reflect"
2020-10-12 13:16:04 +00:00
"strings"
2019-11-19 00:31:13 +00:00
"testing"
)
type syslogEvent struct {
level string
msg string
}
type syslogTestWriter struct {
events []syslogEvent
}
func (w *syslogTestWriter) Write(p []byte) (int, error) {
return 0, nil
}
2019-11-04 19:39:22 +00:00
func (w *syslogTestWriter) Trace(m string) error {
w.events = append(w.events, syslogEvent{"Trace", m})
return nil
}
func (w *syslogTestWriter) Debug(m string) error {
w.events = append(w.events, syslogEvent{"Debug", m})
return nil
}
func (w *syslogTestWriter) Info(m string) error {
w.events = append(w.events, syslogEvent{"Info", m})
return nil
}
func (w *syslogTestWriter) Warning(m string) error {
w.events = append(w.events, syslogEvent{"Warning", m})
return nil
}
func (w *syslogTestWriter) Err(m string) error {
w.events = append(w.events, syslogEvent{"Err", m})
return nil
}
func (w *syslogTestWriter) Emerg(m string) error {
w.events = append(w.events, syslogEvent{"Emerg", m})
return nil
}
func (w *syslogTestWriter) Crit(m string) error {
w.events = append(w.events, syslogEvent{"Crit", m})
return nil
}
func TestSyslogWriter(t *testing.T) {
sw := &syslogTestWriter{}
log := New(SyslogLevelWriter(sw))
2019-11-04 19:39:22 +00:00
log.Trace().Msg("trace")
log.Debug().Msg("debug")
log.Info().Msg("info")
log.Warn().Msg("warn")
log.Error().Msg("error")
2017-12-01 17:52:37 +00:00
log.Log().Msg("nolevel")
want := []syslogEvent{
{"Debug", `{"level":"debug","message":"debug"}` + "\n"},
{"Info", `{"level":"info","message":"info"}` + "\n"},
{"Warning", `{"level":"warn","message":"warn"}` + "\n"},
{"Err", `{"level":"error","message":"error"}` + "\n"},
2017-12-01 17:52:37 +00:00
{"Info", `{"message":"nolevel"}` + "\n"},
}
if got := sw.events; !reflect.DeepEqual(got, want) {
t.Errorf("Invalid syslog message routing: want %v, got %v", want, got)
}
}
2020-10-12 13:16:04 +00:00
type testCEEwriter struct {
buf *bytes.Buffer
}
// Only implement one method as we're just testing the prefixing
func (c testCEEwriter) Debug(m string) error { return nil }
func (c testCEEwriter) Info(m string) error {
_, err := c.buf.Write([]byte(m))
return err
}
func (c testCEEwriter) Warning(m string) error { return nil }
func (c testCEEwriter) Err(m string) error { return nil }
func (c testCEEwriter) Emerg(m string) error { return nil }
func (c testCEEwriter) Crit(m string) error { return nil }
func (c testCEEwriter) Write(b []byte) (int, error) {
return c.buf.Write(b)
}
func TestSyslogWriter_WithCEE(t *testing.T) {
var buf bytes.Buffer
sw := testCEEwriter{&buf}
log := New(SyslogCEEWriter(sw))
log.Info().Str("key", "value").Msg("message string")
got := string(buf.Bytes())
want := "@cee:{"
if !strings.HasPrefix(got, want) {
t.Errorf("Bad CEE message start: want %v, got %v", want, got)
}
}