Files
iotlcm/internal/service/trafficusage/snapshot_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

55 lines
2.1 KiB
Go

package trafficusage
import (
"context"
"errors"
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
"testing"
)
func TestPersistenceFailureRetainsSnapshot(t *testing.T) {
upstream := &upstreamStub{dailyOutput: validDailyOutput()}
repository := &repositoryStub{dailyErr: errors.New("database offline")}
service := New(upstream, repository)
message := validDailyMessage()
var failure *Failure
if err := service.RetryDaily(context.Background(), message); !errors.As(err, &failure) || failure.UsageMb == nil || *failure.UsageMb != 1 {
t.Fatalf("snapshot missing: %v", err)
}
message.UsageMb = failure.UsageMb
repository.dailyErr = nil
if err := service.RetryDaily(context.Background(), message); err != nil {
t.Fatal(err)
}
if upstream.dailyCalls != 1 || repository.dailyCalls != 2 {
t.Fatal("persistence retry queried upstream again")
}
}
func TestZeroSnapshotBypassesUpstream(t *testing.T) {
upstream, repository := &upstreamStub{}, &repositoryStub{}
service := New(upstream, repository)
zero := 0.0
message := validDailyMessage()
message.UsageMb = &zero
if err := service.RetryDaily(context.Background(), message); err != nil {
t.Fatal(err)
}
monthly := model.TrafficUsageMonthlyRetryMessage{SimKey: message.SimKey, Card: message.Card, QueryDate: "202607", UsageMb: &zero}
if err := service.RetryMonthly(context.Background(), monthly); err != nil {
t.Fatal(err)
}
if upstream.dailyCalls != 0 || upstream.monthlyCalls != 0 || repository.dailyCalls != 1 || repository.monthlyCalls != 1 {
t.Fatal("zero snapshot was not persisted directly")
}
}
func TestRejectMismatchedOrMissingResult(t *testing.T) {
for _, output := range [][]model.SimBaseRes[model.SimDataUsageDailyOutputItem]{nil, {{}}, {{SimKey: &model.SimKey{ICCID: "wrong"}}}, {validDailyOutput()[0], validDailyOutput()[0]}} {
upstream, repository := &upstreamStub{dailyOutput: output}, &repositoryStub{}
if err := New(upstream, repository).RetryDaily(context.Background(), validDailyMessage()); err == nil || repository.dailyCalls != 0 {
t.Fatal("invalid result accepted")
}
}
}