From 2603e0984c6bcc1475a164b000f1057110553433 Mon Sep 17 00:00:00 2001
From: Alex Bramley <a.bramley@gmail.com>
Date: Thu, 27 Oct 2011 16:55:33 +0100
Subject: [PATCH] Test nick <-> channel association.

---
 state/tracker_test.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/state/tracker_test.go b/state/tracker_test.go
index 64efee1..818f5d1 100644
--- a/state/tracker_test.go
+++ b/state/tracker_test.go
@@ -357,3 +357,31 @@ func TestSTIsOn(t *testing.T) {
 	}
 	m.CheckNothingWritten(t)
 }
+
+func TestSTAssociate(t *testing.T) {
+	l, m := logging.NewMock()
+	l.SetLogLevel(logging.Debug)
+	st := NewTracker("mynick", l)
+
+	nick1 := st.NewNick("test1")
+	chan1 := st.NewChannel("#test1")
+
+	st.Associate(chan1, nick1)
+	m.CheckNothingWritten(t)
+	if !st.IsOn("#test1", "test1") {
+		t.Errorf("test1 was not associated with #test1.")
+	}
+
+	// Test error cases
+	st.Associate(nil, nick1)
+	m.CheckWrittenAtLevel(t, logging.Error,
+		"StateTracker.Associate(): passed nil values :-(")
+
+	st.Associate(chan1, nil)
+	m.CheckWrittenAtLevel(t, logging.Error,
+		"StateTracker.Associate(): passed nil values :-(")
+
+	st.Associate(chan1, nick1)
+	m.CheckWrittenAtLevel(t, logging.Warn,
+		"StateTracker.Associate(): test1 already on #test1.")
+}