forked from alterminal/alterminal
61 lines
2.2 KiB
Go
61 lines
2.2 KiB
Go
// Package testdb 準備整合測試專用的測試資料庫（與開發資料庫
|
||
// alterminal 隔離），供需要完整資料表（users、sessions、
|
||
// signing_keys、applications、authorization_codes、refresh_tokens、
|
||
// consents）的套件測試使用。go test 以套件為單位並行執行，各套件
|
||
// 取得各自的資料庫（alterminal_test_<套件名>），避免並行測試相互
|
||
// TRUNCATE。oidc 套件的測試須以外部測試套件（package oidc_test）
|
||
// 匯入本套件——store 為遷移而匯入 oidc 模型，內部測試套件匯入本
|
||
// 套件會形成循環。
|
||
package testdb
|
||
|
||
import (
|
||
"fmt"
|
||
"path/filepath"
|
||
"runtime"
|
||
"strings"
|
||
"testing"
|
||
|
||
"gorm.io/driver/postgres"
|
||
"gorm.io/gorm"
|
||
|
||
"alterminal/internal/store"
|
||
)
|
||
|
||
// New 連線本機 PostgreSQL，建立（若不存在）並遷移呼叫方套件專屬的
|
||
// 測試資料庫、清空所有資料表後回傳連線；本機 PostgreSQL 不可用時
|
||
// 跳過測試。
|
||
func New(t *testing.T) *gorm.DB {
|
||
t.Helper()
|
||
// 以呼叫方（測試檔）所在目錄為套件識別，各套件一個資料庫。
|
||
_, file, _, ok := runtime.Caller(1)
|
||
dbName := "alterminal_test"
|
||
if ok {
|
||
dbName += "_" + filepath.Base(filepath.Dir(file))
|
||
}
|
||
admin, err := gorm.Open(postgres.Open(fmt.Sprintf(
|
||
"host=%s port=%s user=%s password=%s dbname=postgres sslmode=disable TimeZone=UTC",
|
||
store.EnvOr("DB_HOST", "localhost"), store.EnvOr("DB_PORT", "5432"),
|
||
store.EnvOr("DB_USER", "postgres"), store.EnvOr("DB_PASSWORD", "postgres"),
|
||
)), &gorm.Config{})
|
||
if err != nil {
|
||
t.Skipf("本機 PostgreSQL 不可用，跳過整合測試：%v", err)
|
||
}
|
||
if err := admin.Exec(fmt.Sprintf("CREATE DATABASE %s", dbName)).Error; err != nil && !strings.Contains(err.Error(), "already exists") {
|
||
t.Skipf("無法建立測試資料庫：%v", err)
|
||
}
|
||
t.Setenv("DB_NAME", dbName)
|
||
db, err := store.Open()
|
||
if err != nil {
|
||
t.Skipf("連線測試資料庫失敗：%v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
if sqlDB, err := db.DB(); err == nil {
|
||
sqlDB.Close()
|
||
}
|
||
})
|
||
if err := db.Exec("TRUNCATE users, sessions, signing_keys, applications, authorization_codes, refresh_tokens, consents RESTART IDENTITY CASCADE").Error; err != nil {
|
||
t.Fatalf("清空測試資料失敗：%v", err)
|
||
}
|
||
return db
|
||
}
|