96 lines
2.0 KiB
Go
96 lines
2.0 KiB
Go
// Copyright 2009 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
package syslog
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
var serverAddr string
|
|
|
|
func runSyslog(c net.PacketConn, done chan<- string) {
|
|
var buf [4096]byte
|
|
var rcvd string = ""
|
|
for {
|
|
n, _, err := c.ReadFrom(buf[0:])
|
|
if err != nil || n == 0 {
|
|
break
|
|
}
|
|
rcvd += string(buf[0:n])
|
|
}
|
|
done <- rcvd
|
|
}
|
|
|
|
func startServer(done chan<- string) {
|
|
c, e := net.ListenPacket("udp", "127.0.0.1:0")
|
|
if e != nil {
|
|
log.Fatalf("net.ListenPacket failed udp :0 %v", e)
|
|
}
|
|
serverAddr = c.LocalAddr().String()
|
|
c.SetReadTimeout(100e6) // 100ms
|
|
go runSyslog(c, done)
|
|
}
|
|
|
|
func TestNew(t *testing.T) {
|
|
s, err := New(LOG_INFO, "")
|
|
if err != nil {
|
|
t.Fatalf("New() failed: %s", err)
|
|
}
|
|
// Don't send any messages.
|
|
s.Close()
|
|
}
|
|
|
|
func TestNewLogger(t *testing.T) {
|
|
f := NewLogger(LOG_INFO, 0)
|
|
if f == nil {
|
|
t.Error("NewLogger() failed")
|
|
}
|
|
}
|
|
|
|
func TestDial(t *testing.T) {
|
|
l, err := Dial("", "", LOG_ERR, "syslog_test")
|
|
if err != nil {
|
|
t.Fatalf("Dial() failed: %s", err)
|
|
}
|
|
l.Close()
|
|
}
|
|
|
|
func TestUDPDial(t *testing.T) {
|
|
done := make(chan string)
|
|
startServer(done)
|
|
l, err := Dial("udp", serverAddr, LOG_INFO, "syslog_test")
|
|
if err != nil {
|
|
t.Fatalf("syslog.Dial() failed: %s", err)
|
|
}
|
|
msg := "udp test"
|
|
l.Info(msg)
|
|
expected := "<6>syslog_test: udp test\n"
|
|
rcvd := <-done
|
|
if rcvd != expected {
|
|
t.Fatalf("s.Info() = '%q', but wanted '%q'", rcvd, expected)
|
|
}
|
|
}
|
|
|
|
func TestWrite(t *testing.T) {
|
|
done := make(chan string)
|
|
startServer(done)
|
|
l, err := Dial("udp", serverAddr, LOG_ERR, "syslog_test")
|
|
if err != nil {
|
|
t.Fatalf("syslog.Dial() failed: %s", err)
|
|
}
|
|
msg := "write test"
|
|
_, err = io.WriteString(l, msg)
|
|
if err != nil {
|
|
t.Fatalf("WriteString() failed: %s", err)
|
|
}
|
|
expected := "<3>syslog_test: write test\n"
|
|
rcvd := <-done
|
|
if rcvd != expected {
|
|
t.Fatalf("s.Info() = '%q', but wanted '%q'", rcvd, expected)
|
|
}
|
|
}
|