Files
iotlcm/internal/queue/zset/producer_abandoned_test.go
T
MiFaZhan 2bec257289 chore: 初始开源发布
iotlcm-worker-svr:异步执行中枢(RabbitMQ 消费者 + 全量同步编排)。

- Go module:iotlcm-worker-svr → git.5iot.com/iotlcm-public/iotlcm-worker-svr(214 文件)
- 凭据与内部标识脱敏:MySQL/Redis 凭据、私网与生产 IP、内网域名、
  库名 xiujie_admin → iotlcm、哨兵名占位
- 159 处真实手机号 → 合法 11 位编造号(保留号段与后 4 位,维持互异,
  不破坏「校验 MSISDN 格式」等测试断言)
- 移除内部资产:CI 触发器、Jenkins 触发测试文档、预演门禁与镜像
  digest 记录
- Dockerfile 改用公共基础镜像(golang:1.25-alpine → alpine:3.23),
  移除写死的 git 凭据
- README 重写(架构/全量同步模型/配置/已知失败),补 Apache-2.0
  LICENSE,LF 行尾 + .gitattributes

已知:3 个测试包在源仓同一提交上同样失败(client/upstream、
service/allocate、service/inbound),属既有缺陷,本仓未引入亦未修复。
2026-10-22 16:37:56 +08:00

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)
}
})
}
}