36 lines
1.2 KiB
Go
36 lines
1.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/database"
|
|
)
|
|
|
|
func TestSQLAuditRepositoryCreatesLog(t *testing.T) {
|
|
dir := t.TempDir()
|
|
dbConfig := config.Default(dir).Database
|
|
dbConfig.DSN = filepath.Join(dir, "data", "audit.db")
|
|
db, err := database.Open(context.Background(), dbConfig)
|
|
if err != nil {
|
|
t.Fatalf("打开数据库失败：%v", err)
|
|
}
|
|
defer db.Close()
|
|
if err := database.Migrate(context.Background(), db.DB, "sqlite"); err != nil {
|
|
t.Fatalf("迁移数据库失败：%v", err)
|
|
}
|
|
repo := NewSQLAuditRepository(db.DB)
|
|
if err := repo.CreateAuditLog(context.Background(), AuditLog{Action: "admin_password_reset", ResourceType: "super_admin", ResourceID: "super-admin", Status: "success", Message: "密码重置成功"}); err != nil {
|
|
t.Fatalf("写入审计日志失败：%v", err)
|
|
}
|
|
var count int
|
|
if err := db.DB.QueryRow("SELECT COUNT(*) FROM audit_logs WHERE action = ? AND status = ?", "admin_password_reset", "success").Scan(&count); err != nil {
|
|
t.Fatalf("查询审计日志失败：%v", err)
|
|
}
|
|
if count != 1 {
|
|
t.Fatalf("审计日志数量不正确：%d", count)
|
|
}
|
|
}
|