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!") }