main_test.go

Rama master · réplica de lectura del repositorio original.

package main

import (
	"os"
	"testing"

	"github.com/robfig/cron/v3"
	log "github.com/sirupsen/logrus"

	"github.com/Luzifer/twitch-bot/v3/internal/service/access"
	"github.com/Luzifer/twitch-bot/v3/internal/service/timer"
	"github.com/Luzifer/twitch-bot/v3/pkg/database"
)

func TestMain(m *testing.M) {
	var (
		dbEngine = "sqlite"
		dbDSN    = "file::memory:?cache=shared"

		err error
	)

	if v := os.Getenv("GO_TEST_DB_ENGINE"); v != "" {
		dbEngine = v
	}

	if v := os.Getenv("GO_TEST_DB_DSN"); v != "" {
		dbDSN = v
	}

	if db, err = database.New(dbEngine, dbDSN, "go-test-static-encryption"); err != nil {
		log.WithError(err).Fatal("opening storage backend")
	}

	if accessService, err = access.New(db); err != nil {
		log.WithError(err).Fatal("applying access migration")
	}

	cronService = cron.New(cron.WithSeconds())

	if timerService, err = timer.New(db, cronService); err != nil {
		log.WithError(err).Fatal("applying timer migration")
	}

	if err = initCorePlugins(); err != nil {
		log.WithError(err).Fatal("Unable to load core plugins")
	}

	m.Run()
}