package zset

import (
	"context"
	"errors"
	"testing"

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

func TestProducerStateReadView(t *testing.T) {
	store, err := New(testutil.Redis(t), "producer-state:")
	if err != nil {
		t.Fatal(err)
	}
	ctx := context.Background()
	if _, err = store.ReadProducerState(ctx, "missing", "a"); !errors.Is(err, ErrRunNotFound) {
		t.Fatalf("missing run err=%v", err)
	}
	if err = store.CreateRun(ctx, RunMeta{RunID: "state", Status: "producing", ActiveCount: 3, InflightCount: 2}); err != nil {
		t.Fatal(err)
	}
	metaKey, _ := store.RunMetaKey("state")
	_, _ = store.RedisClient().Do(ctx, "HSET", metaKey, "blocked:a", "1")
	state, err := store.ReadProducerState(ctx, "state", "a")
	if err != nil || state.ActiveCount != 3 || state.ProducerDone || state.CurrentRound != 1 || !state.Blocked {
		t.Fatalf("state=%+v err=%v", state, err)
	}
	// Every state deduplicates admission, including terminal and leased tasks.
	statesKey, _ := store.TaskStateKey("state")
	_, _ = store.RedisClient().Do(ctx, "HSET", statesKey, "success", "succeeded", "leased", "claimed", "waiting", "waiting_next_round")
	exists, err := store.ExistingRunTasks(ctx, "state", []Task{{TaskID: "success", RunID: "state"}, {TaskID: "leased", RunID: "state"}, {TaskID: "waiting", RunID: "state"}, {TaskID: "new", RunID: "state"}})
	if err != nil || len(exists) != 4 || !exists[0] || !exists[1] || !exists[2] || exists[3] {
		t.Fatalf("exists=%v err=%v", exists, err)
	}
	if _, err = store.ExistingRunTasks(ctx, "state", []Task{{RunID: "foreign", TaskID: "x"}}); err == nil {
		t.Fatal("foreign task accepted")
	}
	for _, args := range [][]any{{"currentRound", 0}, {"currentRound", 1, "activeCount", -1}} {
		_, _ = store.RedisClient().Do(ctx, "HSET", append([]any{metaKey}, args...)...)
		if _, err = store.ReadProducerState(ctx, "state", "a"); err == nil {
			t.Fatal("invalid run counters accepted")
		}
	}
}