func TestBadWorkerCountAndCustomFormatFunc(t *testing.T) {
addr, err := net.ResolveUDPAddr("udp", ":2004")
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
conn, err := net.ListenUDP("udp", addr)
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
defer conn.Close()
sLog, err := New("udp", "127.0.0.1:2004", "", nil)
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
sLog.SetDisplayColor(true)
sLog.SetBuffersAndWorkers(3, 0)
sLog.SetTimestampFormat("2006")
sLog.SetFormatFunc(func(s *Syslog) Formatter {
return func(e *log.Entry) []byte {
return []byte(e.Message)
}
})
log.RegisterHandler(sLog, log.AllLevels...)
log.Debug("debug")
if s := hasString(conn); s != "debug" {
log.Errorf("Expected '%s' Got '%s'", "debug", s)
}
}
func TestSetFilenameColor(t *testing.T) {
addr, err := net.ResolveUDPAddr("udp", ":2006")
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
conn, err := net.ListenUDP("udp", addr)
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
defer conn.Close()
sLog, err := New("udp", "127.0.0.1:2006", "", nil)
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
sLog.SetDisplayColor(true)
sLog.SetBuffersAndWorkers(3, 1)
sLog.SetTimestampFormat("MST")
sLog.SetFilenameDisplay(log.Llongfile)
log.RegisterHandler(sLog, log.AllLevels...)
log.Error("error")
if s := hasString(conn); !strings.Contains(s, "log/handlers/syslog/syslog_test.go:367 error") {
t.Errorf("Expected '%s' Got '%s'", "log/handlers/syslog/syslog_test.go:367 error", s)
}
}
func TestBadAddress(t *testing.T) {
sLog, err := New("udp", "255.255.255.67", "", nil)
if err == nil {
log.Errorf("Expected '%v' Got '%v'", "not nil", err)
}
if sLog != nil {
log.Errorf("Expected '%v' Got '%v'", nil, sLog)
}
}
func TestCustomFormatFunc(t *testing.T) {
buff := new(bytes.Buffer)
cLog := New()
cLog.SetWriter(buff)
cLog.SetTimestampFormat("2006")
cLog.SetBuffersAndWorkers(3, 2)
cLog.SetFormatFunc(func(c *Console) Formatter {
var b []byte
return func(e *log.Entry) []byte {
b = b[0:0]
b = append(b, e.Message...)
return b
}
})
log.RegisterHandler(cLog, log.AllLevels...)
log.Debug("debug")
if buff.String() != "debug" {
log.Errorf("Expected '%s' Got '%s'", "debug", buff.String())
}
buff.Reset()
}
func TestSyslogTLS(t *testing.T) {
// setup server
addr, err := net.ResolveTCPAddr("tcp", ":2022")
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
cnn, err := net.ListenTCP("tcp", addr)
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
defer cnn.Close()
tlsConfig := &tls.Config{
Certificates: make([]tls.Certificate, 1),
}
tlsConfig.Certificates[0], err = tls.X509KeyPair([]byte(publicKey), []byte(privateKey))
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
conn := tls.NewListener(cnn, tlsConfig)
var msg string
var m sync.Mutex
go func() {
client, err := conn.Accept()
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
b := make([]byte, 1024)
read, err := client.Read(b)
if err != nil {
log.Errorf("Expected '%v' Got '%v'", nil, err)
}
m.Lock()
defer m.Unlock()
msg = string(b[0:read])
}()
// setup client log
pool := x509.NewCertPool()
pool.AppendCertsFromPEM([]byte(publicKey))
config := &tls.Config{
RootCAs: pool,
}
sLog, err := New("tcp+tls", "127.0.0.1:2022", "", config)
if err != nil {
log.Fatalf("Expected '%v' Got '%v'", nil, err)
}
//sLog.SetDisplayColor(true)
// sLog.SetBuffersAndWorkers(3, 0)
// sLog.SetTimestampFormat("2006")
sLog.SetFormatFunc(func(s *Syslog) Formatter {
return func(e *log.Entry) []byte {
return []byte(e.Message)
}
})
log.RegisterHandler(sLog, log.AllLevels...)
log.Debug("debug")
time.Sleep(500 * time.Millisecond)
m.Lock()
defer m.Unlock()
if msg != "debug" {
log.Errorf("Expected '%s' Got '%s'", "debug", msg)
}
}