95 lines
1.9 KiB
Go
95 lines
1.9 KiB
Go
package session
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/diamondburned/arikawa/v3/gateway"
|
|
"github.com/diamondburned/arikawa/v3/internal/testenv"
|
|
)
|
|
|
|
func TestSession(t *testing.T) {
|
|
attempts := 1
|
|
timeout := 15 * time.Second
|
|
|
|
if !testing.Short() {
|
|
attempts = 5
|
|
timeout = time.Minute // 5s-10s each reconnection
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
|
t.Cleanup(cancel)
|
|
|
|
env := testenv.Must(t)
|
|
|
|
readyCh := make(chan *gateway.ReadyEvent, 1)
|
|
|
|
s := NewWithIntents(env.BotToken, gateway.IntentGuilds)
|
|
s.AddHandler(readyCh)
|
|
|
|
for i := 0; i < attempts; i++ {
|
|
if err := s.Open(ctx); err != nil {
|
|
t.Fatal("failed to open:", err)
|
|
}
|
|
|
|
if ready, ok := <-readyCh; !ok {
|
|
t.Error("ready not received")
|
|
} else {
|
|
now := time.Now()
|
|
t.Logf("%s: logged in as %s", now.Format(time.StampMilli), ready.User.Username)
|
|
}
|
|
|
|
if err := s.Close(); err != nil {
|
|
t.Fatal("failed to close:", err)
|
|
}
|
|
|
|
// Hold for an additional one second.
|
|
time.Sleep(time.Second)
|
|
}
|
|
}
|
|
|
|
func TestSessionConnect(t *testing.T) {
|
|
attempts := 1
|
|
timeout := 15 * time.Second
|
|
|
|
if !testing.Short() {
|
|
attempts = 5
|
|
timeout = time.Minute // 5s-10s each reconnection
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), timeout)
|
|
t.Cleanup(cancel)
|
|
|
|
env := testenv.Must(t)
|
|
|
|
readyCh := make(chan *gateway.ReadyEvent, 1)
|
|
|
|
s := NewWithIntents(env.BotToken, gateway.IntentGuilds)
|
|
s.AddHandler(readyCh)
|
|
|
|
for i := 0; i < attempts; i++ {
|
|
ctx, cancel := context.WithCancel(ctx)
|
|
|
|
go func() {
|
|
select {
|
|
case ready := <-readyCh:
|
|
now := time.Now()
|
|
t.Logf("%s: logged in as %s", now.Format(time.StampMilli), ready.User.Username)
|
|
cancel()
|
|
case <-ctx.Done():
|
|
t.Error("ready not received")
|
|
}
|
|
}()
|
|
|
|
if err := s.Connect(ctx); err != nil {
|
|
t.Fatal("failed to open:", err)
|
|
}
|
|
|
|
cancel()
|
|
|
|
// Hold for an additional one second.
|
|
time.Sleep(time.Second)
|
|
}
|
|
}
|