forked from mirror/logrus
44 lines
753 B
Go
44 lines
753 B
Go
|
package logrus_papertrail
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/Sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWritingToUDP(t *testing.T) {
|
||
|
|
log := logrus.New()
|
||
|
|
port := 16661
|
||
|
|
|
||
|
|
addr := net.UDPAddr{
|
||
|
|
Port: port,
|
||
|
|
IP: net.ParseIP("127.0.0.1"),
|
||
|
|
}
|
||
|
|
|
||
|
|
c, err := net.ListenUDP("udp", &addr)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("ListenUDP failed: %v", err)
|
||
|
|
}
|
||
|
|
defer c.Close()
|
||
|
|
|
||
|
|
hook, err := NewPapertrailHook("localhost", port, "test")
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("Unable to connect to local UDP server.")
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Hooks.Add(hook)
|
||
|
|
log.Info("Today was a good day.")
|
||
|
|
|
||
|
|
var buf = make([]byte, 1500)
|
||
|
|
n, _, err := c.ReadFromUDP(buf)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Error reading data from local UDP server")
|
||
|
|
}
|
||
|
|
|
||
|
|
if n <= 0 {
|
||
|
|
t.Errorf("Nothing written to local UDP server.")
|
||
|
|
}
|
||
|
|
}
|