MiFaZhan
78 lines
2.7 KiB
Go
78 lines
2.7 KiB
Go
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)
|
|
}
|
|
})
|
|
}
|
|
}
|