Subversion Repositories openrisc

// 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"
        "time"
)

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.SetReadDeadline(time.Now().Add(100 * time.Millisecond))
        go runSyslog(c, done)
}

func skipNetTest(t *testing.T) bool {
        if testing.Short() {
                // Depends on syslog daemon running, and sometimes it's not.
                t.Logf("skipping syslog test during -short")
                return true
        }
        return false
}

func TestNew(t *testing.T) {
        if skipNetTest(t) {
                return
        }
        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) {
        if skipNetTest(t) {
                return
        }
        f := NewLogger(LOG_INFO, 0)
        if f == nil {
                t.Error("NewLogger() failed")
        }
}

func TestDial(t *testing.T) {
        if skipNetTest(t) {
                return
        }
        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)
        }
}