Files
iotlcm/internal/producer/monthlyusage/supply_observation_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

70 lines
2.5 KiB
Go

package monthlyusage
import (
"context"
"fmt"
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/fullsync/supply"
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/queue/zset"
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/testutil"
"github.com/gogf/gf/v2/container/gvar"
"strings"
"testing"
"time"
)
type telemetryFailRedis struct{ zset.RedisClient }
func (r telemetryFailRedis) Eval(ctx context.Context, script string, n int64, keys []string, args []any) (*gvar.Var, error) {
if strings.Contains(script, "'supplyObservation'") {
return nil, fmt.Errorf("injected telemetry write failure")
}
return r.RedisClient.Eval(ctx, script, n, keys, args)
}
func TestSupplyFlushFailureDoesNotBlockProducerDone(t *testing.T) {
client := testutil.Redis(t)
store, err := zset.New(telemetryFailRedis{client}, "flow:")
if err != nil {
t.Fatal(err)
}
scanner := scannerFor([]Card{{ID: 1, ChanKey: "a"}}, 1)
p := testProducer(t, scanner, store, Config{SupplyObservation: supply.NewTracker(true)})
if err = p.Produce(context.Background(), model.TrafficUsageMonthlyStartMessage{RunID: "telemetry-failure"}); err != nil {
t.Fatal(err)
}
requireMeta(t, store, "telemetry-failure", true, 1)
}
func TestSupplyObservesEmptyTargetAndFinalFlush(t *testing.T) {
store := testStore(t)
scanner := scannerFor([]Card{{ID: 1, ChanKey: "a"}}, 2)
scanner.channels = append(scanner.channels, channelMetadata{ChanKey: "empty", TempKey: "sim_yd_v5"})
tracker := supply.NewTracker(true)
tracker.ConsumerActive.Store(true)
scanner.hook = func(ctx context.Context, key string, last int64) error {
if key == "empty" {
return sleepContext(ctx, 2200*time.Millisecond)
}
return nil
}
p := testProducer(t, scanner, store, Config{SupplyObservation: tracker})
if err := p.Produce(context.Background(), model.TrafficUsageMonthlyStartMessage{RunID: "supply-end", QueryMonth: "202610"}); err != nil {
t.Fatal(err)
}
m, err := store.GetRunMeta(context.Background(), "supply-end")
if err != nil {
t.Fatal(err)
}
if !m.ProducerDone || m.SupplyObservation == nil {
t.Fatalf("missing final observation %+v", m)
}
snapshot := m.SupplyObservation
if snapshot.Summary.Targets != 2 || snapshot.Summary.Confirmed != 0 {
t.Fatalf("bad summary %+v", snapshot)
}
c := snapshot.Channels[1]
if c.Key != "empty" || c.FirstEnqueueMS != nil || c.Termination != "empty_result" || c.Unconfirmed.TotalMS < 900 {
t.Fatalf("bad empty target %+v", c)
}
}