package zset

import (
	"context"
	"fmt"
	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/fullsync/supply"
	"testing"
	"time"
)

func TestSupplyBatchAndLeaseView(t *testing.T) {
	s := newLockStore(t)
	ctx := context.Background()
	run := "supply-test"
	if err := s.CreateRun(ctx, RunMeta{RunID: run, Status: "producing", CurrentRound: 1}); err != nil {
		t.Fatal(err)
	}
	keys := make([]string, 130)
	now := time.Now()
	for i := range keys {
		keys[i] = fmt.Sprintf("channel-%03d", i)
	}
	a, _ := s.RoundActiveKey(run, 1, keys[0])
	l, _ := s.RoundLeaseKey(run, 1, keys[1])
	if _, err := s.redis.Do(ctx, "ZADD", a, now.UnixMilli(), "ready", now.Add(time.Hour).UnixMilli(), "future"); err != nil {
		t.Fatal(err)
	}
	if _, err := s.redis.Do(ctx, "ZADD", l, now.Add(-time.Hour).UnixMilli(), "expired"); err != nil {
		t.Fatal(err)
	}
	view, err := s.ReadSupply(ctx, run, keys, now)
	if err != nil {
		t.Fatal(err)
	}
	if len(view) != 130 || view[keys[0]].Ready != 1 || view[keys[1]].Leases != 1 || !view[keys[129]].Eligible {
		t.Fatalf("bad view %+v", view)
	}
	meta, _ := s.RunMetaKey(run)
	s.redis.Do(ctx, "HSET", meta, "blocked:"+keys[2], 1, "currentRound", 2)
	view, err = s.ReadSupply(ctx, run, keys, now)
	if err != nil {
		t.Fatal(err)
	}
	for _, v := range view {
		if v.Eligible {
			t.Fatal("retry round eligible")
		}
	}
}
func TestSupplyOwnerAndTerminalGuard(t *testing.T) {
	s := newLockStore(t)
	ctx := context.Background()
	run := "supply-guard"
	s.CreateRun(ctx, RunMeta{RunID: run, Status: "producing", CurrentRound: 1})
	lock := "flow:producer:active"
	s.AcquireLockOrTakeover(ctx, lock, "old", run, time.Minute)
	encoded, _ := supply.New("old", time.Now(), []string{"a"}).Encode()
	ok, err := s.SaveSupply(ctx, run, lock, "old", encoded)
	if err != nil || !ok {
		t.Fatalf("save %v %v", ok, err)
	}
	s.AcquireLockOrTakeover(ctx, lock, "new", run, time.Minute)
	ok, err = s.SaveSupply(ctx, run, lock, "old", encoded)
	if err != nil || ok {
		t.Fatalf("old owner save %v %v", ok, err)
	}
	meta, _ := s.RunMetaKey(run)
	s.redis.Do(ctx, "HSET", meta, "status", "completed")
	ok, err = s.SaveSupply(ctx, run, lock, "new", encoded)
	if err != nil || ok {
		t.Fatalf("terminal save %v %v", ok, err)
	}
	s.redis.Do(ctx, "HSET", meta, "supplyObservation", "{broken")
	m, err := s.GetRunMeta(ctx, run)
	if err != nil || m.SupplyObservation == nil || !m.SupplyObservation.Partial {
		t.Fatalf("invalid telemetry poisoned business %+v %v", m, err)
	}
}