pothtonswer/discordmuffin
1
0
Fork 1
Code Issues 2 Pull requests 1 Projects 2 Releases Packages Wiki Activity Actions

Merge pull request #213 from iopred/fixtest

Browse source 
Fix handler tests.
This commit is contained in:
Bruce 2016-06-04 20:19:28 -05:00
commit 04c0add4fd
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 ( import (
"os" "os"
"runtime" "runtime"
"sync/atomic"
"testing" "testing"
"time" "time"
) )
@ -226,14 +227,14 @@ func TestOpenClose(t *testing.T) {
} }
func TestAddHandler(t *testing.T) { func TestAddHandler(t *testing.T) {
testHandlerCalled := 0 testHandlerCalled := int32(0)
testHandler := func(s *Session, m *MessageCreate) { testHandler := func(s *Session, m *MessageCreate) {
testHandlerCalled++ atomic.AddInt32(&testHandlerCalled, 1)
} }
interfaceHandlerCalled := 0 interfaceHandlerCalled := int32(0)
interfaceHandler := func(s *Session, i interface{}) { interfaceHandler := func(s *Session, i interface{}) {
interfaceHandlerCalled++ atomic.AddInt32(&interfaceHandlerCalled, 1)
} }
bogusHandlerCalled := false bogusHandlerCalled := false
@ -251,6 +252,8 @@ func TestAddHandler(t *testing.T) {
d.handle(&MessageCreate{}) d.handle(&MessageCreate{})
d.handle(&MessageDelete{}) d.handle(&MessageDelete{})
<-time.After(100 * time.Millisecond)
// testHandler will be called twice because it was added twice. // testHandler will be called twice because it was added twice.
if testHandlerCalled != 2 { if testHandlerCalled != 2 {
t.Fatalf("testHandler was not called twice.") t.Fatalf("testHandler was not called twice.")
@ -267,9 +270,9 @@ func TestAddHandler(t *testing.T) {
} }
func TestRemoveHandler(t *testing.T) { func TestRemoveHandler(t *testing.T) {
testHandlerCalled := 0 testHandlerCalled := int32(0)
testHandler := func(s *Session, m *MessageCreate) { testHandler := func(s *Session, m *MessageCreate) {
testHandlerCalled++ atomic.AddInt32(&testHandlerCalled, 1)
} }
d := Session{} d := Session{}
@ -281,6 +284,8 @@ func TestRemoveHandler(t *testing.T) {
d.handle(&MessageCreate{}) d.handle(&MessageCreate{})
<-time.After(100 * time.Millisecond)
// testHandler will be called once, as it was removed in between calls. // testHandler will be called once, as it was removed in between calls.
if testHandlerCalled != 1 { if testHandlerCalled != 1 {
t.Fatalf("testHandler was not called once.") t.Fatalf("testHandler was not called once.")