fix benchmark failure.

4 years ago · d2c7fb9fb3
1 changed files with 4 additions and 2 deletions
--- a/handle_test.go
+++ b/handle_test.go
@ -43,8 +43,10 @@ func BenchmarkHandle(b *testing.B) {
 	event := irc.NewEvent("test", eventName)

 	wg := sync.WaitGroup{}
-	client.AddHandler(func(event2 *irc.Event, client *irc.Client) {
-		wg.Done()
+	client.AddHandler(func(event2 *irc.Event, _ *irc.Client) {
+		if event2.Kind() != "hook" {
+			wg.Done()
+		}
 	})

 	b.Run("Emit", func(b *testing.B) {