Files
wangjia 03b00b9a0f merge: config + store(UTC DSN) + 自动迁移 [tsk_zRA6fGU1JuHj]
以 main 现有 config 设计为准(FromEnv/*Config/单 DSN/RS256),移植分支新增的 store 层:mysql.go 的 Open(*config.Config) 改用 cfg.DSN、结构化强制 UTC + 会话时区断言;migrate.go + migrations/embed.go 提供 golang-migrate 嵌入式迁移;cmd/migrate/main.go 实现 migrate CLI(读 DB_DSN)。未采用分支的 Load()/PANGOLIN_ 前缀/MySQL 拆分字段(与 main 设计冲突)。

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
2026-06-13 18:42:46 +08:00

44 lines
1.2 KiB
Go

package store
import (
"testing"
"time"
"github.com/go-sql-driver/mysql"
"github.com/wangjia/pangolin/server/internal/config"
)
// TestBuildDSN_ConflictingParamsOverridden verifies that a DSN that already
// contains conflicting timezone / locale settings is corrected by buildDSN:
// ParseTime, Loc, Collation and the time_zone param are all forced to UTC.
func TestBuildDSN_ConflictingParamsOverridden(t *testing.T) {
// A DSN with parseTime=false, the wrong timezone, and a local location.
cfg := &config.Config{
DSN: "root:pass@tcp(localhost:3306)/db?parseTime=false&time_zone=%27Asia%2FShanghai%27",
}
dsn, err := buildDSN(cfg)
if err != nil {
t.Fatalf("buildDSN: %v", err)
}
parsed, err := mysql.ParseDSN(dsn)
if err != nil {
t.Fatalf("ParseDSN on output: %v", err)
}
if !parsed.ParseTime {
t.Error("ParseTime should be overridden to true")
}
if parsed.Loc != time.UTC {
t.Errorf("Loc = %v, want time.UTC (override of Local)", parsed.Loc)
}
if got := parsed.Params["time_zone"]; got != "'+00:00'" {
t.Errorf("time_zone param = %q, want \"'+00:00'\" (override)", got)
}
if parsed.Collation != "utf8mb4_unicode_ci" {
t.Errorf("Collation = %q, want utf8mb4_unicode_ci", parsed.Collation)
}
}