mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-12-11 07:54:58 +00:00
75 lines
1.4 KiB
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
|
|
}
|
|
}
|