mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-10-02 23:06:36 +08:00
ryan
416603b616
- Replace db.DB(ctx) with database.DB(ctx) from plugins/infra/database
- Replace db.Redis/db.PrefixedKey/db.GetJSON/db.SetJSON with cachepkg.* from plugins/infra/cache
- Replace pkg/persistence/idgen with pkg/idgen (already exists)
- Replace pkg/persistence/batchwriter with pkg/batchwriter (already exists)
- Replace pkg/persistence/migrator with pkg/migrator (already exists)
- Replace pkg/persistence/logstore with plugins/domain/risk_control/logstore
- Delete defunct pkg/{persistence,cap,message_gateway,push,shared,task}
- Fix vet issues: db alias in domain_test.go, driver_asynq_worker.TaskHandler reference
- Update Makefile architecture guard
- Update docs and skill references
- Update go.mod: gorilla/sessions promotion to direct dependency
70 lines
2.0 KiB
Go
70 lines
2.0 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package task
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
database "github.com/Rain-kl/Wavelet/plugins/infra/database"
|
|
"github.com/Rain-kl/Wavelet/pkg/testhelper"
|
|
"github.com/Rain-kl/Wavelet/plugins/domain/upload/models"
|
|
)
|
|
|
|
func TestRebuildUploadStatsHandler_Execute(t *testing.T) {
|
|
_, _, cleanup := testhelper.SetupTestEnvironment(t)
|
|
defer cleanup()
|
|
|
|
ctx := context.Background()
|
|
now := time.Now()
|
|
|
|
uploads := []models.Upload{
|
|
{
|
|
UserID: 1001, FileName: "a.jpg", FilePath: "uploads/a.jpg",
|
|
FileSize: 100, MimeType: "image/jpeg", Extension: "jpg", Hash: "hash-a",
|
|
Type: "pixez_mirror", Status: models.UploadStatusUsed, CreatedAt: now,
|
|
},
|
|
{
|
|
UserID: 1001, FileName: "b.png", FilePath: "uploads/b.png",
|
|
FileSize: 200, MimeType: "image/png", Extension: "png", Hash: "hash-b",
|
|
Type: "attachment", Status: models.UploadStatusUsed, CreatedAt: now,
|
|
},
|
|
}
|
|
for i := range uploads {
|
|
if err := database.DB(ctx).Create(&uploads[i]).Error; err != nil {
|
|
t.Fatalf("seed upload failed: %v", err)
|
|
}
|
|
}
|
|
|
|
// Corrupt stats to ensure rebuild recalculates from uploads.
|
|
if err := database.DB(ctx).Create(&models.UploadStat{
|
|
Dimension: models.UploadStatDimensionTotal,
|
|
StatKey: "",
|
|
FileCount: 0,
|
|
FileSize: 0,
|
|
}).Error; err != nil {
|
|
t.Fatalf("seed broken total stat failed: %v", err)
|
|
}
|
|
|
|
handler := &RebuildUploadStatsHandler{}
|
|
result, err := handler.Execute(ctx, nil)
|
|
if err != nil {
|
|
t.Fatalf("Execute() error = %v", err)
|
|
}
|
|
if result == nil || result.Message == "" {
|
|
t.Fatalf("Execute() returned empty result: %+v", result)
|
|
}
|
|
|
|
var totalStat models.UploadStat
|
|
if err := database.DB(ctx).
|
|
Where("dimension = ? AND stat_key = ?", models.UploadStatDimensionTotal, "").
|
|
First(&totalStat).Error; err != nil {
|
|
t.Fatalf("load total stat failed: %v", err)
|
|
}
|
|
if totalStat.FileCount != 2 || totalStat.FileSize != 300 {
|
|
t.Fatalf("total stat = count %d size %d, want 2 / 300", totalStat.FileCount, totalStat.FileSize)
|
|
}
|
|
}
|