1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2024-12-11 07:54:58 +00:00
arikawa/gateway/integration_test.go

75 lines
1.4 KiB
Go

// +build integration
package gateway
import (
"log"
"os"
"testing"
"time"
)
func TestIntegration(t *testing.T) {
var token = os.Getenv("BOT_TOKEN")
if token == "" {
t.Fatal("Missing $BOT_TOKEN")
}
WSError = func(err error) {
log.Println(err)
}
var gateway *Gateway
// NewGateway should call Start for us.
g, err := NewGateway("Bot " + token)
if err != nil {
t.Fatal("Failed to make a Gateway:", err)
}
gateway = g
if err := g.Open(); err != nil {
t.Fatal("Failed to authenticate with Discord:", err)
}
ev := wait(t, gateway.Events)
ready, ok := ev.(*ReadyEvent)
if !ok {
t.Fatal("Event received is not of type Ready:", ev)
}
if gateway.SessionID == "" {
t.Fatal("Session ID is empty")
}
log.Println("Bot's username is", ready.User.Username)
// Try and reconnect
if err := gateway.Reconnect(); err != nil {
t.Fatal("Failed to reconnect:", err)
}
/* TODO: We're not testing this, as Discord will replay events before it
* sends the Resumed event.
resumed, ok := (<-gateway.Events).(*ResumedEvent)
if !ok {
t.Fatal("Event received is not of type Resumed:", resumed)
}
*/
if err := g.Close(); err != nil {
t.Fatal("Failed to close Gateway:", err)
}
}
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
}
}