package elasticilm

import (
	"fmt"
	// "math"
	"testing"
	"time"
)

// TestEnsureILMPolicy 测试 ensureILMPolicy 函数并生成退化阶段时间矩阵
func TestEnsureILMPolicy(t *testing.T) {
	// 定义时间框架
	timeFrames := []string{"1m", "3m", "5m", "15m", "30m", "1H", "2H", "4H", "6H", "12H", "1D", "2D", "5D", "1W"}

	// 定义日期范围
	startDate := time.Date(2018, 1, 1, 0, 0, 0, 0, time.UTC)
	endDate := time.Date(2025, 3, 1, 0, 0, 0, 0, time.UTC)
	currentDate := time.Date(2025, 4, 1, 0, 0, 0, 0, time.UTC)

	// 初始化结果矩阵
	results := make([][]string, 0)

	// 遍历每个时间框架
	for _, period := range timeFrames {
		row := make([]string, 0)
		// 遍历每个月
		for d := startDate; !d.After(endDate); d = d.AddDate(0, 1, 0) {
			daysDiff := int(currentDate.Sub(d).Hours() / 24)

			// 默认配置
			config := map[string]float64{
				"timeDecayFactor":         0.5,
				"periodGranularityFactor": 0.5,
				"warmPhaseMultiplier":     1.0,
				"coldPhaseMultiplier":     2.0,
				"deletePhaseMultiplier":   3.0,
			}

			// 使用非线性冷却模型计算各个阶段的时间
			warmDays, coldDays, deleteDays := NonLinearCoolingModel(daysDiff, period, config)

			// 格式化结果
			result := fmt.Sprintf("Warm: %d, Cold: %d, Delete: %d", warmDays, coldDays, deleteDays)
			row = append(row, result)
		}
		results = append(results, row)
	}

	// 打印结果矩阵
	for i, period := range timeFrames {
		fmt.Printf("Time Frame: %s\n", period)
		for j, result := range results[i] {
			fmt.Printf("  Month: %s - %s\n", startDate.AddDate(0, j, 0).Format("2006-01"), result)
		}
	}
}