80 lines
2.3 KiB
Go
80 lines
2.3 KiB
Go
package database
|
||
|
||
import (
|
||
"context"
|
||
"database/sql"
|
||
"path/filepath"
|
||
"testing"
|
||
|
||
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
||
)
|
||
|
||
func openSQLiteTestPool(t *testing.T, cfg config.DatabaseConfig) *sql.DB {
|
||
t.Helper()
|
||
db, err := Open(t.Context(), cfg)
|
||
if err != nil {
|
||
t.Fatalf("打开隔离 SQLite 文件库失败：%v", err)
|
||
}
|
||
t.Cleanup(func() {
|
||
if err := db.Close(); err != nil {
|
||
t.Errorf("关闭隔离 SQLite 连接池失败：%v", err)
|
||
}
|
||
})
|
||
return db.DB
|
||
}
|
||
|
||
func checkSQLiteConnection(t *testing.T, ctx context.Context, conn *sql.Conn) {
|
||
t.Helper()
|
||
var timeout int
|
||
if err := conn.QueryRowContext(ctx, "PRAGMA busy_timeout").Scan(&timeout); err != nil {
|
||
t.Fatalf("读取连接忙等待配置失败：%v", err)
|
||
}
|
||
if timeout != 5000 {
|
||
t.Errorf("连接忙等待配置丢失：得到 %d，期望 5000", timeout)
|
||
}
|
||
var mode string
|
||
if err := conn.QueryRowContext(ctx, "PRAGMA journal_mode").Scan(&mode); err != nil {
|
||
t.Fatalf("读取连接日志模式失败：%v", err)
|
||
}
|
||
if mode != "wal" {
|
||
t.Errorf("文件库没有使用 WAL：%s", mode)
|
||
}
|
||
}
|
||
|
||
func TestSQLitePoolConfiguresEveryConnection(t *testing.T) {
|
||
for _, name := range []string{"parallel_connections", "no_idle_connections", "replacement_connections", "uri_options"} {
|
||
t.Run(name, func(t *testing.T) {
|
||
cfg := config.Default(t.TempDir()).Database
|
||
cfg.DSN = filepath.Join(t.TempDir(), "pool.db")
|
||
cfg.MaxOpenConns = 4
|
||
cfg.MaxIdleConns = 4
|
||
if name == "no_idle_connections" {
|
||
cfg.MaxIdleConns = 0
|
||
}
|
||
if name == "uri_options" {
|
||
cfg.DSN += "?_pragma=foreign_keys(1)&_pragma=busy_timeout(17)&_txlock=immediate"
|
||
}
|
||
db := openSQLiteTestPool(t, cfg)
|
||
if name == "replacement_connections" {
|
||
db.SetMaxIdleConns(0)
|
||
db.SetMaxIdleConns(4)
|
||
}
|
||
// 同时持有所有连接，避免顺序查询反复使用初始化时的同一条连接。
|
||
for i := 0; i < 4; i++ {
|
||
conn, err := db.Conn(t.Context())
|
||
if err != nil {
|
||
t.Fatalf("取得第 %d 条连接失败：%v", i+1, err)
|
||
}
|
||
t.Cleanup(func() { _ = conn.Close() })
|
||
checkSQLiteConnection(t, t.Context(), conn)
|
||
if name == "uri_options" {
|
||
var enabled int
|
||
if err := conn.QueryRowContext(t.Context(), "PRAGMA foreign_keys").Scan(&enabled); err != nil || enabled != 1 {
|
||
t.Fatalf("用户提供的连接选项丢失：foreign_keys=%d，错误=%v", enabled, err)
|
||
}
|
||
}
|
||
}
|
||
})
|
||
}
|
||
}
|