|
|
@@ -0,0 +1,71 @@
|
|
|
+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])
|
|
|
+ }
|
|
|
+}
|
