MiFaZhan
83 lines
3.2 KiB
Go
83 lines
3.2 KiB
Go
package runtime
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/queue/zset"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/testutil"
|
|
)
|
|
|
|
// Exercise the real task -> failure pool -> replayed task path for both
|
|
// TDengine usage chains; a producer-only assertion misses lost package codes.
|
|
func TestReplayDailyAndMonthlyFailurePoolPreservesPackageCode(t *testing.T) {
|
|
for _, tc := range []struct {
|
|
name, operation, period string
|
|
}{
|
|
{"daily", "previous_day_usage_full_sync", "20261009"},
|
|
{"monthly", "previous_month_usage_full_sync", "202609"},
|
|
} {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
redis := testutil.Redis(t)
|
|
ctx := context.Background()
|
|
store, err := zset.New(redis, tc.name+":")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sourceID := tc.name + "-source"
|
|
if err := store.CreateRun(ctx, zset.RunMeta{RunID: sourceID, Operation: tc.operation, QueryMonth: tc.period}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
task := zset.Task{TaskID: sourceID + ":card-42", RunID: sourceID, Operation: tc.operation, QueryMonth: tc.period, CardID: 42, ICCID: "89860042", ChanKey: "217", TempKey: "sim_yd_v5", PackageCode: "PKG-42"}
|
|
if n, err := store.EnqueueRunBatch(ctx, sourceID, task.ChanKey, []zset.Task{task}, time.Now().Add(-time.Second)); err != nil || n != 1 {
|
|
t.Fatalf("enqueue source task: %d, %v", n, err)
|
|
}
|
|
if err := store.MarkProducerDone(ctx, sourceID); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
now := time.Now()
|
|
claimed, err := store.ClaimRun(ctx, sourceID, task.ChanKey, now, now.Add(time.Minute), 1)
|
|
if err != nil || len(claimed) != 1 {
|
|
t.Fatalf("claim source task: %d, %v", len(claimed), err)
|
|
}
|
|
if done, err := store.FinalizeRunTask(ctx, claimed[0], zset.TaskOutcome{ErrorClass: "business", Code: 500, Message: "retry by pool"}); err != nil || !done {
|
|
t.Fatalf("write failure pool: %v, %v", done, err)
|
|
}
|
|
failedKey, _ := store.FailedKey(sourceID)
|
|
value, err := redis.Do(ctx, "HGET", failedKey, task.TaskID)
|
|
if err != nil || value == nil {
|
|
t.Fatalf("failure record missing: %v", err)
|
|
}
|
|
var failure zset.FailureRecord
|
|
if err := json.Unmarshal([]byte(value.String()), &failure); err != nil || failure.PackageCode != task.PackageCode {
|
|
t.Fatalf("failure pool package code: %+v, %v", failure, err)
|
|
}
|
|
if done, err := store.CompleteRunIfIdle(ctx, sourceID); err != nil || !done {
|
|
t.Fatalf("complete failed source: %v, %v", done, err)
|
|
}
|
|
svc, err := NewSimpleReplay(store)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
receipt, err := svc.Request(ctx, sourceID, "42", 0)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := awaitTargetMaxRound(t, svc, store, receipt.ReplayRunID); got != DefaultReplayMaxRounds {
|
|
t.Fatalf("unexpected replay budget: %d", got)
|
|
}
|
|
now = time.Now()
|
|
replayed, err := store.ClaimRun(ctx, receipt.ReplayRunID, task.ChanKey, now, now.Add(time.Minute), 1)
|
|
if err != nil || len(replayed) != 1 {
|
|
t.Fatalf("claim replayed task: %d, %v", len(replayed), err)
|
|
}
|
|
got := replayed[0].Task
|
|
if got.PackageCode != task.PackageCode || got.QueryMonth != tc.period || got.Operation != tc.operation {
|
|
t.Fatalf("replayed task lost package or period: %+v", got)
|
|
}
|
|
})
|
|
}
|
|
}
|