1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- 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])
- }
- }
|