diamondburned/arikawa
1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-11-27 17:23:00 +00:00
Code Issues Releases Activity

Added timeouts to gateway integration tests

Browse source
This commit is contained in:
diamondburned (Forefront) 2020-01-17 23:13:50 -08:00
parent 345ef44e6c
commit a3a914ac4a
1 changed files with 13 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
gateway/integration_test.go
View file

@ -6,6 +6,7 @@ import (
"log"
"os"
"testing"
"time"
)
func TestIntegration(t *testing.T) {
@ -27,7 +28,7 @@ func TestIntegration(t *testing.T) {
}
gateway = g
ready, ok := (<-gateway.Events).(*ReadyEvent)
ready, ok := wait(t, gateway.Events).(*ReadyEvent)
if !ok {
t.Fatal("Event received is not of type Ready:", ready)
}
@ -50,8 +51,18 @@ func TestIntegration(t *testing.T) {
}
*/
ready, ok = (<-gateway.Events).(*ReadyEvent)
ready, ok = wait(t, gateway.Events).(*ReadyEvent)
if !ok {
t.Fatal("Event received is not of type Ready:", ready)
}
}
func wait(t *testing.T, evCh chan interface{}) interface{} {
select {
case ev := <-evCh:
return ev
case <-time.After(10 * time.Second):
t.Fatal("Timed out waiting for event")
return nil
}
}