mirror of https://github.com/gissleh/irc.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.4 KiB
67 lines
1.4 KiB
package irc_test
|
|
|
|
import (
|
|
"context"
|
|
"math/rand"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gissleh/irc"
|
|
)
|
|
|
|
func TestHandle(t *testing.T) {
|
|
rng := rand.NewSource(time.Now().UnixNano())
|
|
eventName := strconv.FormatInt(rng.Int63(), 36) + strconv.FormatInt(rng.Int63(), 36) + strconv.FormatInt(rng.Int63(), 36)
|
|
|
|
client := irc.New(context.Background(), irc.Config{})
|
|
event := irc.NewEvent("test", eventName)
|
|
handled := false
|
|
|
|
client.AddHandler(func(event *irc.Event, client *irc.Client) {
|
|
if !handled {
|
|
t.Log("Got:", event.Kind(), event.Verb())
|
|
}
|
|
|
|
if event.Kind() == "test" && event.Verb() == eventName {
|
|
handled = true
|
|
}
|
|
})
|
|
|
|
client.EmitSync(context.Background(), event)
|
|
if !handled {
|
|
t.Error("Event wasn't handled")
|
|
}
|
|
}
|
|
|
|
func BenchmarkHandle(b *testing.B) {
|
|
rng := rand.NewSource(time.Now().UnixNano())
|
|
eventName := strconv.FormatInt(rng.Int63(), 36) + strconv.FormatInt(rng.Int63(), 36) + strconv.FormatInt(rng.Int63(), 36)
|
|
|
|
client := irc.New(context.Background(), irc.Config{})
|
|
event := irc.NewEvent("test_event", eventName)
|
|
|
|
wg := sync.WaitGroup{}
|
|
client.AddHandler(func(event2 *irc.Event, _ *irc.Client) {
|
|
if event2.Kind() == "test_event" {
|
|
wg.Done()
|
|
}
|
|
})
|
|
|
|
b.Run("Emit", func(b *testing.B) {
|
|
wg.Add(b.N)
|
|
for n := 0; n < b.N; n++ {
|
|
client.Emit(event)
|
|
}
|
|
wg.Wait()
|
|
})
|
|
|
|
b.Run("EmitSync", func(b *testing.B) {
|
|
wg.Add(b.N)
|
|
for n := 0; n < b.N; n++ {
|
|
client.EmitSync(context.Background(), event)
|
|
}
|
|
wg.Wait()
|
|
})
|
|
}
|