From 8239bc4b6e0fcf8004061bb54902928b1acb3a66 Mon Sep 17 00:00:00 2001
From: Alex Bramley <a.bramley@gmail.com>
Date: Wed, 9 Nov 2011 21:54:14 +0000
Subject: [PATCH] Unit test adding and deleting handlers and events.

---
 event/registry_test.go | 123 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 122 insertions(+), 1 deletion(-)

diff --git a/event/registry_test.go b/event/registry_test.go
index 4d194e9..6b775d2 100644
--- a/event/registry_test.go
+++ b/event/registry_test.go
@@ -7,6 +7,127 @@ import (
 	"time"
 )
 
+func TestAddDelHandler(t *testing.T) {
+	r := NewRegistry()
+
+	if len(r.events) != 0 {
+		t.Errorf("New registry has non-zero-length event map.")
+	}
+
+	h1 := NewHandler(func(ev ...interface{}) {})
+	h2 := NewHandler(func(ev ...interface{}) {})
+
+	// Ensure that a handler with no events to handle doesn't get added
+	r.AddHandler(h1)
+	if len(r.events) != 0 {
+		t.Errorf("Adding handler with no events succeded.")
+	}
+
+	// Add h1 to a couple of events.
+	r.AddHandler(h1, "e1", "E2")
+	if len(r.events) != 2 {
+		t.Errorf("Adding handler h1 to events failed.")
+	}
+	if l, ok := r.events["e1"]; !ok || l.Len() != 1 ||
+		l.Front().Value != h1 {
+		t.Errorf("Handler h1 not added to event e1 correctly.")
+	}
+	if l, ok := r.events["e2"]; !ok || l.Len() != 1 ||
+		l.Front().Value != h1 {
+		t.Errorf("Handler h1 not added to event e2 correctly.")
+	}
+
+	// Add h2 to a couple of events.
+	r.AddHandler(h2, "e2", "E3")
+	if len(r.events) != 3 {
+		t.Errorf("Adding handler h2 to events failed.")
+	}
+	if l, ok := r.events["e2"]; !ok || l.Len() != 2 ||
+		l.Front().Next().Value.(Handler) != h2 {
+		t.Errorf("Handler h2 not added to event e2 correctly.")
+	}
+	if l, ok := r.events["e3"]; !ok || l.Len() != 1 ||
+		l.Front().Value.(Handler) != h2 {
+		t.Errorf("Handler h2 not added to event e3 correctly.")
+	}
+
+	// Add h1 to some more events, which it may be in already.
+	r.AddHandler(h1, "e2", "e3", "e4", "e5")
+	if len(r.events) != 5 {
+		t.Errorf("Adding handler h1 to more events failed.")
+		println(len(r.events))
+	}
+	if l, ok := r.events["e2"]; !ok || l.Len() != 2 {
+		t.Errorf("Handler h1 added twice to event e2.")
+	}
+	if l, ok := r.events["e3"]; !ok || l.Len() != 2 ||
+		l.Front().Next().Value.(Handler) != h1 {
+		t.Errorf("Handler h1 not added to event e3 correctly.")
+	}
+
+	// Add h2 to a few more events, for testing delete
+	r.AddHandler(h2, "e5", "e6")
+	if len(r.events) != 6 {
+		t.Errorf("Adding handler h2 to more events failed.")
+	}
+
+	// Currently, we have the following handlers set up:
+	// h1: e1 e2 e3 e4 e5
+	// h2:    e2 e3    e5 e6
+	// NOTE: for e3, h2 is first and h1 is second in the linked list
+
+	// Delete h1 from a few events. This should remove e4 completely.
+	r.DelHandler(h1, "e2", "E3", "e4")
+	if _, ok := r.events["e4"]; ok || len(r.events) != 5 {
+		t.Errorf("Deleting h1 from some events failed to remove e4.")
+	}
+	if l, ok := r.events["e2"]; !ok || l.Len() != 1 ||
+		l.Front().Value.(Handler) != h2 {
+		t.Errorf("Handler h1 not deleted from event e2 correctly.")
+	}
+	if l, ok := r.events["e3"]; !ok || l.Len() != 1 ||
+		l.Front().Value.(Handler) != h2 {
+		t.Errorf("Handler h1 not deleted from event e3 correctly.")
+	}
+
+	// Now, we have the following handlers set up:
+	// h1: e1          e5
+	// h2:    e2 e3    e5 e6
+
+	// Delete h2 from a couple of events, removing e2 and e3.
+	// Deleting h2 from a handler it is not in should not cause problems.
+	r.DelHandler(h2, "e1", "e2", "e3")
+	if len(r.events) != 3 {
+		t.Errorf("Deleting h2 from some events failed to remove e{2,3}.")
+	}
+	if l, ok := r.events["e1"]; !ok || l.Len() != 1 ||
+		l.Front().Value.(Handler) != h1 {
+		t.Errorf("Handler h1 deleted from event e1 incorrectly.")
+	}
+
+	// Delete h1 completely.
+	r.DelHandler(h1)
+	if _, ok := r.events["e1"]; ok || len(r.events) != 2 {
+		t.Errorf("Deleting h1 completely failed to remove e1.")
+	}
+	if l, ok := r.events["e5"]; !ok || l.Len() != 1 ||
+		l.Front().Value.(Handler) != h2 {
+		t.Errorf("Handler h1 deleted from event e5 incorrectly.")
+	}
+
+	// Clear e5 completely
+	r.ClearEvents("e5")
+	if _, ok := r.events["e5"]; ok || len(r.events) != 1 {
+		t.Errorf("Deleting e5 completely failed to remove it.")
+	}
+
+	// All that should be left is e6, with h2 as it's only handler.
+	if l, ok := r.events["e6"]; !ok || l.Len() != 1 ||
+		l.Front().Value.(Handler) != h2 {
+		t.Errorf("Remaining event and handler doesn't match expectations.")
+	}
+}
+
 func TestSimpleDispatch(t *testing.T) {
 	r := NewRegistry()
 	out := make(chan bool)
@@ -66,7 +187,7 @@ func TestParallelDispatch(t *testing.T) {
 
 func TestSerialDispatch(t *testing.T) {
 	r := NewRegistry()
-	r.(*registry).Serial()
+	r.Serial()
 	// ensure we have enough of a buffer that all sends complete
 	out := make(chan int, 5)
 	// handler factory :-)