From cfffcf2a852b30cd55069abad99cf293ff61b3f4 Mon Sep 17 00:00:00 2001
From: Alex Bramley <a.bramley@gmail.com>
Date: Sat, 5 Nov 2011 02:25:25 +0000
Subject: [PATCH] Add a method to wipe a state tracker clean.

---
 state/tracker.go      | 11 ++++++++-
 state/tracker_test.go | 55 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+), 1 deletion(-)

diff --git a/state/tracker.go b/state/tracker.go
index 1b7065b..1810c6d 100644
--- a/state/tracker.go
+++ b/state/tracker.go
@@ -21,6 +21,7 @@ type StateTracker interface {
 	IsOn(channel, nick string) (*ChanPrivs, bool)
 	Associate(channel *Channel, nick *Nick) *ChanPrivs
 	Dissociate(channel *Channel, nick *Nick)
+	Wipe()
 }
 
 // ... and a struct to implement it ...
@@ -37,7 +38,7 @@ type stateTracker struct {
 	l logging.Logger
 }
 
-// ... and finally a constructor to make it.
+// ... and a constructor to make it ...
 func NewTracker(mynick string, l logging.Logger) *stateTracker {
 	st := &stateTracker{
 		chans: make(map[string]*Channel),
@@ -48,6 +49,14 @@ func NewTracker(mynick string, l logging.Logger) *stateTracker {
 	return st
 }
 
+// ... and a method to wipe the state clean.
+func (st *stateTracker) Wipe() {
+	// Deleting all the channels implicitly deletes every nick but me.
+	for _, ch := range st.chans {
+		st.delChannel(ch)
+	}
+}
+
 /******************************************************************************\
  * tracker methods to create/look up nicks/channels
 \******************************************************************************/
diff --git a/state/tracker_test.go b/state/tracker_test.go
index c63f542..c25e9df 100644
--- a/state/tracker_test.go
+++ b/state/tracker_test.go
@@ -477,3 +477,58 @@ func TestSTDissociate(t *testing.T) {
 		"StateTracker.Dissociate(): channel #test3 not found in (or differs "+
 		"from) internal state.")
 }
+
+func TestSTWipe(t *testing.T) {
+	l, m := logging.NewMock()
+	st := NewTracker("mynick", l)
+
+	nick1 := st.NewNick("test1")
+	nick2 := st.NewNick("test2")
+	nick3 := st.NewNick("test3")
+
+	chan1 := st.NewChannel("#test1")
+	chan2 := st.NewChannel("#test2")
+	chan3 := st.NewChannel("#test3")
+
+	// Some associations
+	st.Associate(chan1, st.me)
+	st.Associate(chan2, st.me)
+	st.Associate(chan3, st.me)
+
+	st.Associate(chan1, nick1)
+	st.Associate(chan2, nick2)
+	st.Associate(chan3, nick3)
+
+	st.Associate(chan1, nick2)
+	st.Associate(chan2, nick3)
+
+	st.Associate(chan1, nick3)
+
+	m.CheckNothingWritten(t)
+
+	// Check the state we have at this point is what we would expect.
+	if len(st.nicks) != 4 || len(st.chans) != 3 || len(st.me.chans) != 3 {
+		t.Errorf("Tracker nick/channel lists wrong length before wipe.")
+	}
+	if len(chan1.nicks) != 4 || len(chan2.nicks) != 3 || len(chan3.nicks) != 2 {
+		t.Errorf("Channel nick lists wrong length before wipe.")
+	}
+	if len(nick1.chans) != 1 || len(nick2.chans) != 2 || len(nick3.chans) != 3 {
+		t.Errorf("Nick chan lists wrong length before wipe.")
+	}
+
+	// Nuke *all* the state!
+	st.Wipe()
+	m.CheckNothingWritten(t)
+
+	// Check the state we have at this point is what we would expect.
+	if len(st.nicks) != 1 || len(st.chans) != 0 || len(st.me.chans) != 0 {
+		t.Errorf("Tracker nick/channel lists wrong length after wipe.")
+	}
+	if len(chan1.nicks) != 0 || len(chan2.nicks) != 0 || len(chan3.nicks) != 0 {
+		t.Errorf("Channel nick lists wrong length after wipe.")
+	}
+	if len(nick1.chans) != 0 || len(nick2.chans) != 0 || len(nick3.chans) != 0 {
+		t.Errorf("Nick chan lists wrong length after wipe.")
+	}
+}