From 5f22bda67654f231a17d8ec23d1cbbd97e9f5655 Mon Sep 17 00:00:00 2001
From: Daisuke Maki <lestrrat+github@gmail.com>
Date: Wed, 27 Aug 2014 11:27:37 +0900
Subject: [PATCH] Add tests for hub

---
 hub_test.go | 87 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 hub_test.go

diff --git a/hub_test.go b/hub_test.go
new file mode 100644
index 0000000..c862f43
--- /dev/null
+++ b/hub_test.go
@@ -0,0 +1,87 @@
+package peco
+
+import (
+	"testing"
+	"time"
+)
+
+func TestHub(t *testing.T) {
+	h := NewHub()
+
+	done := make(map[string]time.Time)
+
+	go func() {
+		hr := <-h.QueryCh()
+		time.Sleep(100 * time.Millisecond)
+		done["query"]= time.Now()
+		hr.Done()
+	}()
+	go func() {
+		hr := <-h.DrawCh()
+		if hr.DataInterface() != nil {
+			t.Errorf("Expected data to be nil, got %s", hr.DataInterface())
+		}
+		time.Sleep(100 * time.Millisecond)
+		done["draw"]= time.Now()
+		hr.Done()
+	}()
+	go func() {
+		hr := <-h.StatusMsgCh()
+		if hr.DataString() != "Hello, World!" {
+			t.Errorf("Expected data to be 'Hello World!', got '%s'", hr.DataString())
+		}
+		time.Sleep(100 * time.Millisecond)
+		done["status"] = time.Now()
+		hr.Done()
+	}()
+	go func() {
+		hr := <-h.ClearStatusCh()
+		time.Sleep(100 * time.Millisecond)
+		done["clearStatus"]= time.Now()
+		hr.Done()
+	}()
+	go func() {
+		hr := <-h.PagingCh()
+		time.Sleep(100 * time.Millisecond)
+		done["paging"]= time.Now()
+		hr.Done()
+	}()
+
+	h.Batch(func() {
+		h.SendQuery("Hello World!")
+		h.SendDraw(nil)
+		h.SendStatusMsg("Hello, World!")
+		h.SendClearStatus(time.Second)
+		h.SendPaging(ToLineAbove)
+	})
+
+	phases := []string{
+		"query",
+		"draw",
+		"status",
+		"clearStatus",
+		"paging",
+	}
+
+	max := len(phases) - 1
+	for i := range phases {
+		if max == i {
+			break
+		}
+
+		cur := phases[i]
+		next := phases[i+1]
+
+		t.Logf("Checkin if %s was fired before %s", cur, next)
+		if done[next].Before(done[cur]) {
+			t.Errorf("%s executed before %s?!", next, cur)
+		}
+	}
+
+	h.Stop()
+
+	_, ok := <-h.LoopCh()
+	if !ok {
+		t.Errorf("LoopCh should be closed, but it is not")
+	}
+}
\ No newline at end of file