forked from boyska/circolog
71 lines
1.1 KiB
Go
71 lines
1.1 KiB
Go
package circolog
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gopkg.in/mcuadros/go-syslog.v2/format"
|
|
)
|
|
|
|
func msg(s string) format.LogParts {
|
|
return format.LogParts{"text": s}
|
|
}
|
|
|
|
func hubCount(h Hub) int {
|
|
cl := Client{Nofollow: true}
|
|
cl.Messages = make(chan format.LogParts)
|
|
h.Register <- cl
|
|
cnt := 0
|
|
for range cl.Messages {
|
|
cnt++
|
|
}
|
|
return cnt
|
|
}
|
|
|
|
func hubToArray(h Hub) []string {
|
|
r := make([]string, 0)
|
|
cl := Client{Nofollow: true}
|
|
cl.Messages = make(chan format.LogParts)
|
|
h.Register <- cl
|
|
for m := range cl.Messages {
|
|
r = append(r, m["text"].(string))
|
|
}
|
|
return r
|
|
}
|
|
|
|
func TestSingle(t *testing.T) {
|
|
h := NewHub(5)
|
|
go h.Run()
|
|
h.LogMessages <- msg("hi")
|
|
|
|
if hubCount(h) != 1 {
|
|
t.Error("wrong message number")
|
|
}
|
|
}
|
|
|
|
func TestMany(t *testing.T) {
|
|
h := NewHub(5)
|
|
go h.Run()
|
|
for i := 0; i < 10; i++ {
|
|
h.LogMessages <- msg("hi")
|
|
}
|
|
|
|
if hubCount(h) != 5 {
|
|
t.Error("wrong message number")
|
|
}
|
|
}
|
|
|
|
func TestOrder(t *testing.T) {
|
|
h := NewHub(5)
|
|
go h.Run()
|
|
h.LogMessages <- msg("one")
|
|
h.LogMessages <- msg("two")
|
|
|
|
r := hubToArray(h)
|
|
|
|
if r[0] != "one" {
|
|
t.Error("wrong first message", r[0])
|
|
}
|
|
if r[1] != "two" {
|
|
t.Error("wrong last message", r[1])
|
|
}
|
|
}
|