From ec9a3b135c187f6e0f8c2b0e7fa11d2b3cb540b1 Mon Sep 17 00:00:00 2001
From: Tom Wilkie <tom.wilkie@gmail.com>
Date: Tue, 8 Dec 2015 17:59:24 +0000
Subject: [PATCH] Add a little test for collector.go

---
 app/collector_test.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/app/collector_test.go b/app/collector_test.go
index 00a18077e..9e77fcb9e 100644
--- a/app/collector_test.go
+++ b/app/collector_test.go
@@ -38,3 +38,21 @@ func TestCollector(t *testing.T) {
 		t.Error(test.Diff(want, have))
 	}
 }
+
+func TestCollectorWait(t *testing.T) {
+	window := time.Millisecond
+	c := app.NewCollector(window)
+
+	waiter := make(chan struct{}, 1)
+	c.WaitOn(waiter)
+	defer c.UnWait(waiter)
+	c.(interface {
+		Broadcast()
+	}).Broadcast()
+
+	select {
+	case <-waiter:
+	default:
+		t.Fatal("Didn't unblock")
+	}
+}