86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
|
package database
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"carrotskin/pkg/config"
|
|||
|
|
"testing"
|
|||
|
|
|
|||
|
|
"go.uber.org/zap/zaptest"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// TestGetDB_NotInitialized 测试未初始化时获取数据库实例
|
|||
|
|
func TestGetDB_NotInitialized(t *testing.T) {
|
|||
|
|
_, err := GetDB()
|
|||
|
|
if err == nil {
|
|||
|
|
t.Error("未初始化时应该返回错误")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
expectedError := "数据库未初始化，请先调用 database.Init()"
|
|||
|
|
if err.Error() != expectedError {
|
|||
|
|
t.Errorf("错误消息 = %q, want %q", err.Error(), expectedError)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestMustGetDB_Panic 测试MustGetDB在未初始化时panic
|
|||
|
|
func TestMustGetDB_Panic(t *testing.T) {
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r == nil {
|
|||
|
|
t.Error("MustGetDB 应该在未初始化时panic")
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
_ = MustGetDB()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestInit_Database 测试数据库初始化逻辑
|
|||
|
|
func TestInit_Database(t *testing.T) {
|
|||
|
|
cfg := config.DatabaseConfig{
|
|||
|
|
Driver: "postgres",
|
|||
|
|
Host: "localhost",
|
|||
|
|
Port: 5432,
|
|||
|
|
Username: "postgres",
|
|||
|
|
Password: "password",
|
|||
|
|
Database: "testdb",
|
|||
|
|
SSLMode: "disable",
|
|||
|
|
Timezone: "Asia/Shanghai",
|
|||
|
|
MaxIdleConns: 10,
|
|||
|
|
MaxOpenConns: 100,
|
|||
|
|
ConnMaxLifetime: 0,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
logger := zaptest.NewLogger(t)
|
|||
|
|
|
|||
|
|
// 验证Init函数存在且可调用
|
|||
|
|
// 注意：实际连接可能失败，这是可以接受的
|
|||
|
|
err := Init(cfg, logger)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Logf("Init() 返回错误（可能正常，如果数据库未运行）: %v", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestAutoMigrate_ErrorHandling 测试AutoMigrate的错误处理逻辑
|
|||
|
|
func TestAutoMigrate_ErrorHandling(t *testing.T) {
|
|||
|
|
logger := zaptest.NewLogger(t)
|
|||
|
|
|
|||
|
|
// 测试未初始化时的错误处理
|
|||
|
|
err := AutoMigrate(logger)
|
|||
|
|
if err == nil {
|
|||
|
|
// 如果数据库已初始化，这是正常的
|
|||
|
|
t.Log("AutoMigrate() 成功（数据库可能已初始化）")
|
|||
|
|
} else {
|
|||
|
|
// 如果数据库未初始化，应该返回错误
|
|||
|
|
if err.Error() == "" {
|
|||
|
|
t.Error("AutoMigrate() 应该返回有意义的错误消息")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestClose_NotInitialized 测试未初始化时关闭数据库
|
|||
|
|
func TestClose_NotInitialized(t *testing.T) {
|
|||
|
|
// 未初始化时关闭应该不返回错误
|
|||
|
|
err := Close()
|
|||
|
|
if err != nil {
|
|||
|
|
t.Errorf("Close() 在未初始化时应该返回nil，实际返回: %v", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|