package fileuploadcleanup

import (
	"context"
	"testing"
	"time"

	"hfb_sys/backend/internal/database"
	"hfb_sys/backend/internal/model"

	"gorm.io/gorm"
)

func TestCleanupOnlyDeletesExpiredUnreferencedUploads(t *testing.T) {
	db := openTestDB(t)
	if err := database.AutoMigrateIfMissing(db, &model.FileUploadOwner{}); err != nil {
		t.Fatalf("AutoMigrate() error = %v", err)
	}
	now := time.Date(2026, 8, 16, 0, 0, 0, 0, time.UTC)
	old := now.Add(-31 * 24 * time.Hour)
	recent := now.Add(-29 * 24 * time.Hour)
	records := []model.FileUploadOwner{
		{UserID: 1, ObjectKey: "listing/orphan.jpg", CreatedAt: old},
		{UserID: 1, ObjectKey: "listing/referenced.jpg", CreatedAt: old},
		{UserID: 1, ObjectKey: "listing/recent.jpg", CreatedAt: recent},
	}
	if err := db.Create(&records).Error; err != nil {
		t.Fatalf("create upload owners error = %v", err)
	}
	if err := db.Exec("INSERT INTO game_accounts (id, owner_id, server_region, login_platform, title, screenshot_urls) "+
		"VALUES (1, 1, '国服', 'steam', '测试账号', ?)", `["/api/files/object?key=listing%2Freferenced.jpg"]`).Error; err != nil {
		t.Fatalf("create referenced account error = %v", err)
	}

	job := New(db, nil, nil)
	deleted, err := job.cleanup(context.Background(), now)
	if err != nil {
		t.Fatalf("cleanup() error = %v", err)
	}
	if deleted != 1 {
		t.Fatalf("cleanup() deleted = %d, want 1", deleted)
	}
	var remaining []model.FileUploadOwner
	if err := db.Order("id ASC").Find(&remaining).Error; err != nil {
		t.Fatalf("load remaining uploads error = %v", err)
	}
	if len(remaining) != 2 || remaining[0].ObjectKey != "listing/referenced.jpg" || remaining[1].ObjectKey != "listing/recent.jpg" {
		t.Fatalf("unexpected remaining uploads: %#v", remaining)
	}
}

func openTestDB(t *testing.T) *gorm.DB {
	t.Helper()
	db := database.NewTestDB()
	return db
}