mirror of https://github.com/fluffle/goirc
202 lines
5.7 KiB
Go
202 lines
5.7 KiB
Go
package client
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestHandlerSet(t *testing.T) {
|
|
// A Conn is needed here because the previous behaviour of passing nil to
|
|
// hset.dispatch causes a nil pointer dereference with panic recovery.
|
|
c, s := setUp(t)
|
|
defer s.tearDown()
|
|
|
|
hs := handlerSet()
|
|
if len(hs.set) != 0 {
|
|
t.Errorf("New set contains things!")
|
|
}
|
|
|
|
callcount := new(int32)
|
|
f := func(_ *Conn, _ *Line) {
|
|
atomic.AddInt32(callcount, 1)
|
|
}
|
|
|
|
// Add one
|
|
hn1 := hs.add("ONE", HandlerFunc(f)).(*hNode)
|
|
hl, ok := hs.set["one"]
|
|
if len(hs.set) != 1 || !ok {
|
|
t.Errorf("Set doesn't contain 'one' list after add().")
|
|
}
|
|
if hn1.set != hs || hn1.event != "one" || hn1.prev != nil || hn1.next != nil {
|
|
t.Errorf("First node for 'one' not created correctly")
|
|
}
|
|
if hl.start != hn1 || hl.end != hn1 {
|
|
t.Errorf("Node not added to empty 'one' list correctly.")
|
|
}
|
|
|
|
// Add another one...
|
|
hn2 := hs.add("one", HandlerFunc(f)).(*hNode)
|
|
if len(hs.set) != 1 {
|
|
t.Errorf("Set contains more than 'one' list after add().")
|
|
}
|
|
if hn2.set != hs || hn2.event != "one" {
|
|
t.Errorf("Second node for 'one' not created correctly")
|
|
}
|
|
if hn1.prev != nil || hn1.next != hn2 || hn2.prev != hn1 || hn2.next != nil {
|
|
t.Errorf("Nodes for 'one' not linked correctly.")
|
|
}
|
|
if hl.start != hn1 || hl.end != hn2 {
|
|
t.Errorf("Node not appended to 'one' list correctly.")
|
|
}
|
|
|
|
// Add a third one!
|
|
hn3 := hs.add("one", HandlerFunc(f)).(*hNode)
|
|
if len(hs.set) != 1 {
|
|
t.Errorf("Set contains more than 'one' list after add().")
|
|
}
|
|
if hn3.set != hs || hn3.event != "one" {
|
|
t.Errorf("Third node for 'one' not created correctly")
|
|
}
|
|
if hn1.prev != nil || hn1.next != hn2 ||
|
|
hn2.prev != hn1 || hn2.next != hn3 ||
|
|
hn3.prev != hn2 || hn3.next != nil {
|
|
t.Errorf("Nodes for 'one' not linked correctly.")
|
|
}
|
|
if hl.start != hn1 || hl.end != hn3 {
|
|
t.Errorf("Node not appended to 'one' list correctly.")
|
|
}
|
|
|
|
// And finally a fourth one!
|
|
hn4 := hs.add("one", HandlerFunc(f)).(*hNode)
|
|
if len(hs.set) != 1 {
|
|
t.Errorf("Set contains more than 'one' list after add().")
|
|
}
|
|
if hn4.set != hs || hn4.event != "one" {
|
|
t.Errorf("Fourth node for 'one' not created correctly.")
|
|
}
|
|
if hn1.prev != nil || hn1.next != hn2 ||
|
|
hn2.prev != hn1 || hn2.next != hn3 ||
|
|
hn3.prev != hn2 || hn3.next != hn4 ||
|
|
hn4.prev != hn3 || hn4.next != nil {
|
|
t.Errorf("Nodes for 'one' not linked correctly.")
|
|
}
|
|
if hl.start != hn1 || hl.end != hn4 {
|
|
t.Errorf("Node not appended to 'one' list correctly.")
|
|
}
|
|
|
|
// Dispatch should result in 4 additions.
|
|
if atomic.LoadInt32(callcount) != 0 {
|
|
t.Errorf("Something incremented call count before we were expecting it.")
|
|
}
|
|
hs.dispatch(c, &Line{Cmd: "One"})
|
|
<-time.After(time.Millisecond)
|
|
if atomic.LoadInt32(callcount) != 4 {
|
|
t.Errorf("Our handler wasn't called four times :-(")
|
|
}
|
|
|
|
// Remove node 3.
|
|
hn3.Remove()
|
|
if len(hs.set) != 1 {
|
|
t.Errorf("Set list count changed after remove().")
|
|
}
|
|
if hn3.set != nil || hn3.prev != nil || hn3.next != nil {
|
|
t.Errorf("Third node for 'one' not removed correctly.")
|
|
}
|
|
if hn1.prev != nil || hn1.next != hn2 ||
|
|
hn2.prev != hn1 || hn2.next != hn4 ||
|
|
hn4.prev != hn2 || hn4.next != nil {
|
|
t.Errorf("Third node for 'one' not unlinked correctly.")
|
|
}
|
|
if hl.start != hn1 || hl.end != hn4 {
|
|
t.Errorf("Third node for 'one' changed list pointers.")
|
|
}
|
|
|
|
// Dispatch should result in 3 additions.
|
|
hs.dispatch(c, &Line{Cmd: "One"})
|
|
<-time.After(time.Millisecond)
|
|
if atomic.LoadInt32(callcount) != 7 {
|
|
t.Errorf("Our handler wasn't called three times :-(")
|
|
}
|
|
|
|
// Remove node 1.
|
|
hs.remove(hn1)
|
|
if len(hs.set) != 1 {
|
|
t.Errorf("Set list count changed after remove().")
|
|
}
|
|
if hn1.set != nil || hn1.prev != nil || hn1.next != nil {
|
|
t.Errorf("First node for 'one' not removed correctly.")
|
|
}
|
|
if hn2.prev != nil || hn2.next != hn4 || hn4.prev != hn2 || hn4.next != nil {
|
|
t.Errorf("First node for 'one' not unlinked correctly.")
|
|
}
|
|
if hl.start != hn2 || hl.end != hn4 {
|
|
t.Errorf("First node for 'one' didn't change list pointers.")
|
|
}
|
|
|
|
// Dispatch should result in 2 additions.
|
|
hs.dispatch(c, &Line{Cmd: "One"})
|
|
<-time.After(time.Millisecond)
|
|
if atomic.LoadInt32(callcount) != 9 {
|
|
t.Errorf("Our handler wasn't called two times :-(")
|
|
}
|
|
|
|
// Remove node 4.
|
|
hn4.Remove()
|
|
if len(hs.set) != 1 {
|
|
t.Errorf("Set list count changed after remove().")
|
|
}
|
|
if hn4.set != nil || hn4.prev != nil || hn4.next != nil {
|
|
t.Errorf("Fourth node for 'one' not removed correctly.")
|
|
}
|
|
if hn2.prev != nil || hn2.next != nil {
|
|
t.Errorf("Fourth node for 'one' not unlinked correctly.")
|
|
}
|
|
if hl.start != hn2 || hl.end != hn2 {
|
|
t.Errorf("Fourth node for 'one' didn't change list pointers.")
|
|
}
|
|
|
|
// Dispatch should result in 1 addition.
|
|
hs.dispatch(c, &Line{Cmd: "One"})
|
|
<-time.After(time.Millisecond)
|
|
if atomic.LoadInt32(callcount) != 10 {
|
|
t.Errorf("Our handler wasn't called once :-(")
|
|
}
|
|
|
|
// Remove node 2.
|
|
hs.remove(hn2)
|
|
if len(hs.set) != 0 {
|
|
t.Errorf("Removing last node in 'one' didn't remove list.")
|
|
}
|
|
if hn2.set != nil || hn2.prev != nil || hn2.next != nil {
|
|
t.Errorf("Second node for 'one' not removed correctly.")
|
|
}
|
|
if hl.start != nil || hl.end != nil {
|
|
t.Errorf("Second node for 'one' didn't change list pointers.")
|
|
}
|
|
|
|
// Dispatch should result in NO additions.
|
|
hs.dispatch(c, &Line{Cmd: "One"})
|
|
<-time.After(time.Millisecond)
|
|
if atomic.LoadInt32(callcount) != 10 {
|
|
t.Errorf("Our handler was called?")
|
|
}
|
|
}
|
|
|
|
func TestPanicRecovery(t *testing.T) {
|
|
c, s := setUp(t)
|
|
defer s.tearDown()
|
|
|
|
recovered := callCheck(t)
|
|
c.cfg.Recover = func(conn *Conn, line *Line) {
|
|
if err, ok := recover().(string); ok && err == "panic!" {
|
|
recovered.call()
|
|
}
|
|
}
|
|
c.HandleFunc(PRIVMSG, func(conn *Conn, line *Line) {
|
|
panic("panic!")
|
|
})
|
|
c.in <- ParseLine(":nick!user@host.com PRIVMSG #channel :OH NO PIGEONS")
|
|
recovered.assertWasCalled("Failed to recover panic!")
|
|
}
|