MiFaZhan
79 lines
2.4 KiB
Go
79 lines
2.4 KiB
Go
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)
|
|
}
|
|
}
|