tests_test.go

package clickhouse_test

import (
	"log"
	"math/rand"
	"os"
	"time"

	"gorm.io/driver/clickhouse"
	"gorm.io/gorm"
)

var DB *gorm.DB

const defaultDBDSN = "http://gorm:gorm@127.0.0.1:9941/gorm?dial_timeout=10s&read_timeout=20s"

func testDSN() string {
	if dsn := os.Getenv("CLICKHOUSE_TEST_DSN"); dsn != "" {
		return dsn
	}
	return defaultDBDSN
}

func init() {
	var (
		err error
	)

	if DB, err = gorm.Open(clickhouse.Open(testDSN()), &gorm.Config{}); err != nil {
		log.Printf("failed to connect database, got error %v", err)
		os.Exit(1)
	}

	RunMigrations()

	if os.Getenv("DEBUG") == "true" {
		DB = DB.Debug()
	}
}

func RunMigrations() {
	allModels := []interface{}{&User{}}
	rand.Seed(time.Now().UnixNano())
	rand.Shuffle(len(allModels), func(i, j int) { allModels[i], allModels[j] = allModels[j], allModels[i] })

	if err := DB.Migrator().DropTable(allModels...); err != nil {
		log.Printf("Failed to drop table, got error %v\n", err)
		os.Exit(1)
	}

	if err := DB.AutoMigrate(allModels...); err != nil {
		log.Printf("Failed to auto migrate, but got error %v\n", err)
		os.Exit(1)
	}

	for _, m := range allModels {
		if !DB.Migrator().HasTable(m) {
			log.Printf("Failed to create table for %#v\n", m)
			os.Exit(1)
		}
	}
}