3x-ui-agent/internal/database/seeder_fastpath_test.go

package database

import (
	"path/filepath"
	"testing"

	"github.com/mhsanaei/3x-ui/v3/internal/database/model"
)

func TestFreshInstallFastPathMarksResetIpLimitSeeder(t *testing.T) {
	if err := InitDB(filepath.Join(t.TempDir(), "x-ui.db")); err != nil {
		t.Fatalf("InitDB: %v", err)
	}
	t.Cleanup(func() { _ = CloseDB() })

	if err := db.Where("1 = 1").Delete(&model.HistoryOfSeeders{}).Error; err != nil {
		t.Fatalf("reset seeder history: %v", err)
	}
	if err := db.Where("1 = 1").Delete(&model.User{}).Error; err != nil {
		t.Fatalf("reset users: %v", err)
	}

	if err := runSeeders(true); err != nil {
		t.Fatalf("runSeeders: %v", err)
	}

	var cnt int64
	if err := db.Model(&model.HistoryOfSeeders{}).
		Where("seeder_name = ?", "ResetIpLimitNoFail2ban").
		Count(&cnt).Error; err != nil {
		t.Fatalf("count seeder history: %v", err)
	}
	if cnt != 1 {
		t.Fatal("fresh-install fast path must mark ResetIpLimitNoFail2ban done so it cannot wipe admin-set IP limits on the next boot")
	}
}