chore: 并入 iotlcm-admin-svr 历史到 admin-svr/（原提交 49818c6）

2026-11-02 13:27:59 +08:00
parent 976980a3f5 49818c6dbe
commit 193f7393c0
1250 changed files with 154111 additions and 0 deletions
@@ -0,0 +1,41 @@
+package trafficusage
+
+import (
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
+	"testing"
+	"time"
+)
+
+func TestPreviousUsageMonth(t *testing.T) {
+	for input, want := range map[string]string{"2026-03-31T10:00:00+08:00": "202602", "2024-03-31T10:00:00+08:00": "202402", "2026-01-31T00:00:00+08:00": "202512", "2026-08-31T16:00:00Z": "202608"} {
+		date, err := time.Parse(time.RFC3339, input)
+		if err != nil {
+			t.Fatal(err)
+		}
+		if got := previousUsageMonth(date).Format("200601"); got != want {
+			t.Fatalf("%s: got %s want %s", input, got, want)
+		}
+	}
+}
+
+func TestReconcileUsage(t *testing.T) {
+	keys := []*model.SimKey{{ICCID: "ok"}, {ICCID: "missing"}, {ICCID: "duplicate"}, {ICCID: "failed"}}
+	output := []model.SimBaseRes[model.SimDataUsageDailyOutputItem]{
+		{SimKey: keys[0]}, {SimKey: keys[2]}, {SimKey: keys[2]}, {SimKey: keys[3], Code: 1}, {}, {SimKey: &model.SimKey{ICCID: "unknown"}},
+	}
+	success, failed := reconcileUsage(keys, output, func(item model.SimDataUsageDailyOutputItem) float64 { return item.DataAmount })
+	if len(success) != 1 || success[0].SimKey != keys[0] || len(failed) != 3 {
+		t.Fatalf("success=%+v failed=%+v", success, failed)
+	}
+}
+
+func TestReconcileUsageWithPeriod(t *testing.T) {
+	key := &model.SimKey{ICCID: "period"}
+	output := []model.SimBaseRes[model.SimDataUsageDailyOutputItem]{
+		{SimKey: key, ResData: model.SimDataUsageDailyOutputItem{QueryDate: "2026-07-14", DataAmount: 1}},
+	}
+	success, failed := reconcileUsageWithPeriod([]*model.SimKey{key}, output, func(item model.SimDataUsageDailyOutputItem) float64 { return item.DataAmount }, func(item model.SimDataUsageDailyOutputItem) string { return item.QueryDate }, normalizeUsageDay, "20260713")
+	if len(success) != 0 || len(failed) != 1 {
+		t.Fatalf("expected period mismatch to fail, success=%+v failed=%+v", success, failed)
+	}
+}