arikawa/internal/testenv/testenv.go

// +build !uintonly

package testenv

import (
	"os"
	"strconv"
	"sync"
	"testing"
	"time"

	"github.com/diamondburned/arikawa/v3/discord"
	"github.com/pkg/errors"
)

const PerseveranceTime = 50 * time.Minute

type Env struct {
	BotToken   string
	ChannelID  discord.ChannelID
	VoiceChID  discord.ChannelID
	ShardCount int // default 3
}

var (
	globalEnv Env
	globalErr error
	once      sync.Once
)

func Must(t *testing.T) Env {
	e, err := GetEnv()
	if err != nil {
		t.Skip("integration test variables missing")
	}
	return e
}

func GetEnv() (Env, error) {
	once.Do(getEnv)
	return globalEnv, globalErr
}

func getEnv() {
	token := os.Getenv("BOT_TOKEN")
	if token == "" {
		globalErr = errors.New("missing $BOT_TOKEN")
		return
	}

	id, err := discord.ParseSnowflake(os.Getenv("CHANNEL_ID"))
	if err != nil {
		globalErr = errors.Wrap(err, "invalid $CHANNEL_ID")
		return
	}

	vid, err := discord.ParseSnowflake(os.Getenv("VOICE_ID"))
	if err != nil {
		globalErr = errors.Wrap(err, "invalid $VOICE_ID")
		return
	}

	shardCount := 3
	if c, err := strconv.Atoi(os.Getenv("SHARD_COUNT")); err == nil {
		shardCount = c
	}

	globalEnv = Env{
		BotToken:   token,
		ChannelID:  discord.ChannelID(id),
		VoiceChID:  discord.ChannelID(vid),
		ShardCount: shardCount,
	}
}