21423b914f
use golangci/golangci-lint-action instead of reviewdog/action-golangci-lint as the second was not reporting any failures even if there was some. Report code coverage with codecov/codecov-action I have set some flags per dialect and go version Several linters has been fixed, some disabled so the build can pass
114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"log"
|
|
"math/rand"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"gorm.io/driver/mysql"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/driver/sqlserver"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
. "gorm.io/gorm/utils/tests"
|
|
)
|
|
|
|
var DB *gorm.DB
|
|
|
|
func init() {
|
|
var err error
|
|
if DB, err = OpenTestConnection(); err != nil {
|
|
log.Printf("failed to connect database, got error %v", err)
|
|
os.Exit(1)
|
|
} else {
|
|
sqlDB, err := DB.DB()
|
|
if err == nil {
|
|
err = sqlDB.Ping()
|
|
}
|
|
|
|
if err != nil {
|
|
log.Printf("failed to connect database, got error %v", err)
|
|
}
|
|
|
|
RunMigrations()
|
|
if DB.Dialector.Name() == "sqlite" {
|
|
DB.Exec("PRAGMA foreign_keys = ON")
|
|
}
|
|
}
|
|
}
|
|
|
|
func OpenTestConnection() (db *gorm.DB, err error) {
|
|
dbDSN := os.Getenv("GORM_DSN")
|
|
switch os.Getenv("GORM_DIALECT") {
|
|
case "mysql":
|
|
log.Println("testing mysql...")
|
|
if dbDSN == "" {
|
|
dbDSN = "gorm:gorm@tcp(localhost:9910)/gorm?charset=utf8&parseTime=True&loc=Local"
|
|
}
|
|
db, err = gorm.Open(mysql.Open(dbDSN), &gorm.Config{})
|
|
case "postgres":
|
|
log.Println("testing postgres...")
|
|
if dbDSN == "" {
|
|
dbDSN = "user=gorm password=gorm dbname=gorm host=localhost port=9920 sslmode=disable TimeZone=Asia/Shanghai"
|
|
}
|
|
db, err = gorm.Open(postgres.New(postgres.Config{
|
|
DSN: dbDSN,
|
|
PreferSimpleProtocol: true,
|
|
}), &gorm.Config{})
|
|
case "sqlserver":
|
|
// CREATE LOGIN gorm WITH PASSWORD = 'LoremIpsum86';
|
|
// CREATE DATABASE gorm;
|
|
// USE gorm;
|
|
// CREATE USER gorm FROM LOGIN gorm;
|
|
// sp_changedbowner 'gorm';
|
|
// npm install -g sql-cli
|
|
// mssql -u gorm -p LoremIpsum86 -d gorm -o 9930
|
|
log.Println("testing sqlserver...")
|
|
if dbDSN == "" {
|
|
dbDSN = "sqlserver://gorm:LoremIpsum86@localhost:9930?database=gorm"
|
|
}
|
|
db, err = gorm.Open(sqlserver.Open(dbDSN), &gorm.Config{})
|
|
default:
|
|
log.Println("testing sqlite3...")
|
|
db, err = gorm.Open(sqlite.Open(filepath.Join(os.TempDir(), "gorm.db")), &gorm.Config{})
|
|
}
|
|
|
|
if debug := os.Getenv("DEBUG"); debug == "true" {
|
|
db.Logger = db.Logger.LogMode(logger.Info)
|
|
} else if debug == "false" {
|
|
db.Logger = db.Logger.LogMode(logger.Silent)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func RunMigrations() {
|
|
var err error
|
|
allModels := []interface{}{&User{}, &Account{}, &Pet{}, &Company{}, &Toy{}, &Language{}, &Coupon{}, &CouponProduct{}, &Order{}}
|
|
rand.Seed(time.Now().UnixNano())
|
|
rand.Shuffle(len(allModels), func(i, j int) { allModels[i], allModels[j] = allModels[j], allModels[i] })
|
|
|
|
DB.Migrator().DropTable("user_friends", "user_speaks")
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|