MiFaZhan
2bec257289
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），属既有缺陷，本仓未引入亦未修复。
61 lines
2.3 KiB
Go
61 lines
2.3 KiB
Go
package rabbitmq
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/service/trafficusage"
|
|
amqp "github.com/rabbitmq/amqp091-go"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTrafficUsageRetryRetainsPayload(t *testing.T) {
|
|
stubTrafficPublish(t)
|
|
usage := 12.5
|
|
handler := &trafficUsageHandlerStub{dailyErr: &trafficusage.Failure{Cause: errors.New("database down"), UsageMb: &usage}}
|
|
consumer := New(handler, nil, nil)
|
|
message := model.TrafficUsageDailyRetryMessage{SimKey: &model.SimKey{ICCID: "card"}, Card: model.TrafficUsageSimCard{Iccid: "card"}, QueryDate: "20260831"}
|
|
calls := 0
|
|
trafficUsagePublishDelayed = func(_ context.Context, queue, target string, delay time.Duration, body []byte, _ amqp.Table) error {
|
|
calls++
|
|
var retry model.TrafficUsageDailyRetryMessage
|
|
if err := json.Unmarshal(body, &retry); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if retry.Attempt != 1 || retry.UsageMb == nil || *retry.UsageMb != usage || retry.QueryDate != message.QueryDate || retry.LastError == "" || delay != 30*time.Second || target != consts.QueueTrafficUsageDailyRetry || queue == target {
|
|
t.Fatalf("invalid retry: %+v", retry)
|
|
}
|
|
return nil
|
|
}
|
|
if err := consumer.handleTrafficUsageDailyRetry(context.Background(), message); err != nil || calls != 1 {
|
|
t.Fatalf("calls=%d err=%v", calls, err)
|
|
}
|
|
message.Attempt = 9
|
|
dlqCalls := 0
|
|
trafficUsagePublish = func(_ context.Context, queue string, body []byte, _ amqp.Table) error {
|
|
dlqCalls++
|
|
if queue != consts.QueueTrafficUsageDailyRetry+".dlq" {
|
|
t.Fatal(queue)
|
|
}
|
|
var retry model.TrafficUsageDailyRetryMessage
|
|
if err := json.Unmarshal(body, &retry); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if retry.Attempt != 10 || retry.UsageMb == nil {
|
|
t.Fatal("DLQ lost snapshot")
|
|
}
|
|
return nil
|
|
}
|
|
if err := consumer.handleTrafficUsageDailyRetry(context.Background(), message); err != nil || dlqCalls != 1 || calls != 1 {
|
|
t.Fatalf("DLQ not published: %v", err)
|
|
}
|
|
expected := errors.New("broker unavailable")
|
|
trafficUsagePublish = func(context.Context, string, []byte, amqp.Table) error { return expected }
|
|
if err := consumer.handleTrafficUsageDailyRetry(context.Background(), message); !errors.Is(err, expected) {
|
|
t.Fatal("failed publication must not ack original")
|
|
}
|
|
}
|