mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-09 16:35:12 +00:00
258b6149d7
This commit adds the Connect and Wait methods into session. This gives the user a way to block the program until the session runs into an error or the given ctx is done. In most cases, Connect is useful when combined with signal.NotifyContext, and so Connect is preferred over Open.
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)
|
|
}
|
|
}
|