migan/discordmuffin
1
0
Fork 0
forked from pothtonswer/discordmuffin
Code Pull requests Activity

Fix handler tests.

Browse source
This commit is contained in:
Chris Rhodes 2016-06-04 18:13:46 -07:00
parent 323216e171
commit 84d430918b
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
discord_test.go
View file

@ -3,6 +3,7 @@ package discordgo
import (
"os"
"runtime"
"sync/atomic"
"testing"
"time"
)
@ -226,14 +227,14 @@ func TestOpenClose(t *testing.T) {
}
func TestAddHandler(t *testing.T) {
testHandlerCalled := 0
testHandlerCalled := int32(0)
testHandler := func(s *Session, m *MessageCreate) {
testHandlerCalled++
atomic.AddInt32(&testHandlerCalled, 1)
}
interfaceHandlerCalled := 0
interfaceHandlerCalled := int32(0)
interfaceHandler := func(s *Session, i interface{}) {
interfaceHandlerCalled++
atomic.AddInt32(&interfaceHandlerCalled, 1)
}
bogusHandlerCalled := false
@ -251,6 +252,8 @@ func TestAddHandler(t *testing.T) {
d.handle(&MessageCreate{})
d.handle(&MessageDelete{})
<-time.After(100 * time.Millisecond)
// testHandler will be called twice because it was added twice.
if testHandlerCalled != 2 {
t.Fatalf("testHandler was not called twice.")
@ -267,9 +270,9 @@ func TestAddHandler(t *testing.T) {
}
func TestRemoveHandler(t *testing.T) {
testHandlerCalled := 0
testHandlerCalled := int32(0)
testHandler := func(s *Session, m *MessageCreate) {
testHandlerCalled++
atomic.AddInt32(&testHandlerCalled, 1)
}
d := Session{}
@ -281,6 +284,8 @@ func TestRemoveHandler(t *testing.T) {
d.handle(&MessageCreate{})
<-time.After(100 * time.Millisecond)
// testHandler will be called once, as it was removed in between calls.
if testHandlerCalled != 1 {
t.Fatalf("testHandler was not called once.")