From dee9740e68d66050dee84d95fd657b75e3b0ef8d Mon Sep 17 00:00:00 2001
From: Shivaram Lingamneni <slingamn@cs.stanford.edu>
Date: Mon, 27 May 2019 16:09:09 -0400
Subject: [PATCH] add some testing

---
 irc/modes/modes_test.go | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/irc/modes/modes_test.go b/irc/modes/modes_test.go
index e8b7dbc6..de1761c4 100644
--- a/irc/modes/modes_test.go
+++ b/irc/modes/modes_test.go
@@ -77,6 +77,16 @@ func TestSetMode(t *testing.T) {
 	}
 }
 
+func TestModeString(t *testing.T) {
+	set := NewModeSet()
+	set.SetMode('A', true)
+	set.SetMode('z', true)
+
+	if modeString := set.String(); !(modeString == "Az" || modeString == "Za") {
+		t.Errorf("unexpected modestring: %s", modeString)
+	}
+}
+
 func TestNilReceivers(t *testing.T) {
 	set := NewModeSet()
 	set = nil
@@ -113,3 +123,16 @@ func TestHighestChannelUserMode(t *testing.T) {
 		t.Errorf("nil modeset should have the zero mode as highest channel-user mode")
 	}
 }
+
+func BenchmarkModeString(b *testing.B) {
+	set := NewModeSet()
+	set.SetMode('A', true)
+	set.SetMode('N', true)
+	set.SetMode('b', true)
+	set.SetMode('i', true)
+	set.SetMode('x', true)
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		_ = set.String()
+	}
+}