chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

2026-11-02 13:28:03 +08:00
parent 6de2e503cc 96f0f4eacd
commit 2923c78755
422 changed files with 80993 additions and 0 deletions
@@ -0,0 +1,77 @@
+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)
+			}
+		})
+	}
+}