package zset

import (
	"context"
	"reflect"
	"testing"

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

func TestMarkProducerAbandonedAtomicGuards(t *testing.T) {
	store, err := New(testutil.Redis(t), "abandoned:")
	if err != nil {
		t.Fatal(err)
	}
	ctx := context.Background()
	for _, test := range []struct {
		name, status, syncType string
		done, want             bool
	}{
		{name: "producing", status: "producing", want: true},
		{name: "blocked_running", status: "running", want: true},
		{name: "producing_done", status: "producing", done: true},
		{name: "running_done", status: "running", done: true},
		{name: "producing_replay", status: "producing", syncType: "replay"},
		{name: "running_replay", status: "running", syncType: "replay"},
		{name: "completed", status: "completed"},
		{name: "completed_with_failures", status: "completed_with_failures"},
		{name: "unknown_status", status: "paused"},
		{name: "missing"},
	} {
		t.Run(test.name, func(t *testing.T) {
			key, _ := store.RunMetaKey(test.name)
			if test.status != "" {
				if err := store.CreateRun(ctx, RunMeta{RunID: test.name, Status: test.status, ProducerDone: test.done, SyncType: test.syncType, ActiveCount: 3, InflightCount: 1, NextRoundCount: 2, FailedCount: 4}); err != nil {
					t.Fatal(err)
				}
				if _, err := store.RedisClient().Do(ctx, "HSET", key, "blocked:a", "1"); err != nil {
					t.Fatal(err)
				}
			}
			before, err := store.RedisClient().Do(ctx, "HGETALL", key)
			if err != nil {
				t.Fatal(err)
			}
			promoted, err := store.MarkProducerAbandoned(ctx, test.name)
			if err != nil || promoted != test.want {
				t.Fatalf("promoted=%t want=%t err=%v", promoted, test.want, err)
			}
			after, err := store.RedisClient().Do(ctx, "HGETALL", key)
			if err != nil {
				t.Fatal(err)
			}
			expected := before.MapStrStr()
			actual := after.MapStrStr()
			if test.want {
				if actual["abandonedAt"] == "" || actual["abandonedAt"] != actual["producerCompletedAt"] {
					t.Fatalf("missing abandonment timestamps: %v", actual)
				}
				expected["producerDone"] = "1"
				expected["status"] = "running"
				expected["abandonedAt"] = actual["abandonedAt"]
				expected["producerCompletedAt"] = actual["producerCompletedAt"]
			}
			if !reflect.DeepEqual(expected, actual) {
				t.Fatalf("unexpected metadata change: want=%v got=%v", expected, actual)
			}
			if again, err := store.MarkProducerAbandoned(ctx, test.name); err != nil || again {
				t.Fatalf("repeated promotion=%t err=%v", again, err)
			}
			unchanged, err := store.RedisClient().Do(ctx, "HGETALL", key)
			if err != nil || !reflect.DeepEqual(actual, unchanged.MapStrStr()) {
				t.Fatalf("idempotent promotion changed metadata: %v err=%v", unchanged, err)
			}
		})
	}
}