alphago9
682aa3907a
- refine config loading with --config and current-directory .env - improve sqlite busy handling and runtime DB tuning - restore telegram payment notification and wallet input flow - add behavior-based config and callback recovery tests
97 lines
2.1 KiB
Go
97 lines
2.1 KiB
Go
package testutil
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/assimon/luuu/config"
|
|
"github.com/assimon/luuu/model/dao"
|
|
"github.com/assimon/luuu/model/mdb"
|
|
appLog "github.com/assimon/luuu/util/log"
|
|
"github.com/libtnb/sqlite"
|
|
"github.com/spf13/viper"
|
|
"go.uber.org/zap"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
func SetupTestDatabases(t testing.TB) func() {
|
|
t.Helper()
|
|
|
|
viper.Reset()
|
|
viper.Set("forced_usdt_rate", 1.0)
|
|
viper.Set("app_uri", "https://example.com")
|
|
viper.Set("order_expiration_time", 10)
|
|
viper.Set("order_notice_max_retry", 2)
|
|
viper.Set("callback_retry_base_seconds", 1)
|
|
viper.Set("queue_concurrency", 4)
|
|
viper.Set("queue_poll_interval_ms", 50)
|
|
viper.Set("api_auth_token", "test-token")
|
|
|
|
config.HTTPAccessLog = false
|
|
config.SQLDebug = false
|
|
config.LogLevel = "error"
|
|
config.UsdtRate = 0
|
|
appLog.Sugar = zap.NewNop().Sugar()
|
|
|
|
mainDB := mustOpenSQLite(t, filepath.Join(t.TempDir(), "main.db"))
|
|
runtimeDB := mustOpenSQLite(t, filepath.Join(t.TempDir(), "runtime.db"))
|
|
|
|
mustMigrate(t, mainDB, &mdb.Orders{}, &mdb.WalletAddress{})
|
|
mustMigrate(t, runtimeDB, &mdb.TransactionLock{})
|
|
|
|
dao.Mdb = mainDB
|
|
dao.RuntimeDB = runtimeDB
|
|
|
|
return func() {
|
|
closeDB(t, runtimeDB)
|
|
closeDB(t, mainDB)
|
|
dao.Mdb = nil
|
|
dao.RuntimeDB = nil
|
|
viper.Reset()
|
|
}
|
|
}
|
|
|
|
func mustOpenSQLite(t testing.TB, path string) *gorm.DB {
|
|
t.Helper()
|
|
|
|
db, err := gorm.Open(sqlite.Open(path), &gorm.Config{
|
|
NamingStrategy: schema.NamingStrategy{SingularTable: true},
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("open sqlite %s: %v", path, err)
|
|
}
|
|
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
t.Fatalf("db handle for %s: %v", path, err)
|
|
}
|
|
sqlDB.SetMaxOpenConns(1)
|
|
sqlDB.SetMaxIdleConns(1)
|
|
|
|
return db
|
|
}
|
|
|
|
func mustMigrate(t testing.TB, db *gorm.DB, models ...interface{}) {
|
|
t.Helper()
|
|
if err := db.AutoMigrate(models...); err != nil {
|
|
t.Fatalf("automigrate: %v", err)
|
|
}
|
|
}
|
|
|
|
func closeDB(t testing.TB, db *gorm.DB) {
|
|
t.Helper()
|
|
if db == nil {
|
|
return
|
|
}
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
t.Fatalf("db handle close: %v", err)
|
|
}
|
|
if err = sqlDB.Close(); err != nil {
|
|
t.Fatalf("close db: %v", err)
|
|
}
|
|
}
|