package runtime

import (
	"context"
	"encoding/json"
	"testing"
	"time"

	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/queue/zset"
	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/testutil"
)

// Exercise the real task -> failure pool -> replayed task path for both
// TDengine usage chains; a producer-only assertion misses lost package codes.
func TestReplayDailyAndMonthlyFailurePoolPreservesPackageCode(t *testing.T) {
	for _, tc := range []struct {
		name, operation, period string
	}{
		{"daily", "previous_day_usage_full_sync", "20261009"},
		{"monthly", "previous_month_usage_full_sync", "202609"},
	} {
		t.Run(tc.name, func(t *testing.T) {
			redis := testutil.Redis(t)
			ctx := context.Background()
			store, err := zset.New(redis, tc.name+":")
			if err != nil {
				t.Fatal(err)
			}
			sourceID := tc.name + "-source"
			if err := store.CreateRun(ctx, zset.RunMeta{RunID: sourceID, Operation: tc.operation, QueryMonth: tc.period}); err != nil {
				t.Fatal(err)
			}
			task := zset.Task{TaskID: sourceID + ":card-42", RunID: sourceID, Operation: tc.operation, QueryMonth: tc.period, CardID: 42, ICCID: "89860042", ChanKey: "217", TempKey: "sim_yd_v5", PackageCode: "PKG-42"}
			if n, err := store.EnqueueRunBatch(ctx, sourceID, task.ChanKey, []zset.Task{task}, time.Now().Add(-time.Second)); err != nil || n != 1 {
				t.Fatalf("enqueue source task: %d, %v", n, err)
			}
			if err := store.MarkProducerDone(ctx, sourceID); err != nil {
				t.Fatal(err)
			}
			now := time.Now()
			claimed, err := store.ClaimRun(ctx, sourceID, task.ChanKey, now, now.Add(time.Minute), 1)
			if err != nil || len(claimed) != 1 {
				t.Fatalf("claim source task: %d, %v", len(claimed), err)
			}
			if done, err := store.FinalizeRunTask(ctx, claimed[0], zset.TaskOutcome{ErrorClass: "business", Code: 500, Message: "retry by pool"}); err != nil || !done {
				t.Fatalf("write failure pool: %v, %v", done, err)
			}
			failedKey, _ := store.FailedKey(sourceID)
			value, err := redis.Do(ctx, "HGET", failedKey, task.TaskID)
			if err != nil || value == nil {
				t.Fatalf("failure record missing: %v", err)
			}
			var failure zset.FailureRecord
			if err := json.Unmarshal([]byte(value.String()), &failure); err != nil || failure.PackageCode != task.PackageCode {
				t.Fatalf("failure pool package code: %+v, %v", failure, err)
			}
			if done, err := store.CompleteRunIfIdle(ctx, sourceID); err != nil || !done {
				t.Fatalf("complete failed source: %v, %v", done, err)
			}
			svc, err := NewSimpleReplay(store)
			if err != nil {
				t.Fatal(err)
			}
			receipt, err := svc.Request(ctx, sourceID, "42", 0)
			if err != nil {
				t.Fatal(err)
			}
			if got := awaitTargetMaxRound(t, svc, store, receipt.ReplayRunID); got != DefaultReplayMaxRounds {
				t.Fatalf("unexpected replay budget: %d", got)
			}
			now = time.Now()
			replayed, err := store.ClaimRun(ctx, receipt.ReplayRunID, task.ChanKey, now, now.Add(time.Minute), 1)
			if err != nil || len(replayed) != 1 {
				t.Fatalf("claim replayed task: %d, %v", len(replayed), err)
			}
			got := replayed[0].Task
			if got.PackageCode != task.PackageCode || got.QueryMonth != tc.period || got.Operation != tc.operation {
				t.Fatalf("replayed task lost package or period: %+v", got)
			}
		})
	}
}