chore: 并入 iotlcm-worker-svr 历史到 worker-svr/（原提交 96f0f4e）

worker-svr/internal/repository/trafficusage/repository_test.go

@@ -0,0 +1,149 @@
+package trafficusage
+
+import (
+	"fmt"
+	"hash/fnv"
+	"strings"
+	"testing"
+	"time"
+	"unicode"
+)
+
+// adminChildTableName is a verbatim copy of Admin's childTableNameWithPrefix
+// (iotlcm-admin-svr internal/logic/trafficusage/traffic_usage.go). It exists so
+// the worker rule is asserted against the production reader rule rather than
+// against itself: Admin's QueryDaily resolves child tables with this rule, so
+// any divergence here would silently hide a card's daily usage.
+func adminChildTableName(prefix, value string) string {
+	var builder strings.Builder
+	builder.WriteString(prefix)
+	for _, r := range strings.ToLower(value) {
+		if unicode.IsLetter(r) || unicode.IsDigit(r) || r == '_' {
+			builder.WriteRune(r)
+		}
+	}
+	name := builder.String()
+	if name == prefix || len(name) > 190 {
+		hash := fnv.New32a()
+		_, _ = hash.Write([]byte(value))
+		return fmt.Sprintf("%s%x", prefix, hash.Sum32())
+	}
+	return name
+}
+
+func adminChildTrafficTableName(stable, iccid string) string {
+	prefix := "sim_card_usage_m_"
+	if stable == "t_sim_card_usage_daily" {
+		prefix = "sim_card_usage_d_"
+	}
+	return adminChildTableName(prefix, iccid)
+}
+
+func TestChildTableNameMatchesAdminRule(t *testing.T) {
+	longICCID := strings.Repeat("8", 191)
+	veryLongICCID := strings.Repeat("89", 120)
+	cases := []string{
+		"",
+		"89860001123456789012",
+		"8986-0001",
+		"8986 0001",
+		"8986.0001",
+		"8986/0001",
+		"ICCID_8986",
+		"8986-0001-中文",
+		"中文卡号",
+		"____",
+		"8986+0001",
+		"8986'0001",
+		longICCID,
+		veryLongICCID,
+	}
+	for _, stable := range []string{"t_sim_card_usage_daily", "t_sim_card_usage_monthly"} {
+		for _, iccid := range cases {
+			want := adminChildTrafficTableName(stable, iccid)
+			got := childTableName(stable, iccid)
+			if got != want {
+				t.Errorf("childTableName(%q, %q) = %q, admin rule yields %q", stable, iccid, got, want)
+			}
+			if !strings.HasPrefix(got, "sim_card_usage_") {
+				t.Errorf("childTableName(%q, %q) = %q loses the table prefix", stable, iccid, got)
+			}
+		}
+	}
+}
+
+// TestChildTableNameDirtyICCIDSplitsOnce guards the original defect: the old
+// worker rule replaced illegal characters with "_" while Admin dropped them, so
+// a dirty ICCID could land in a different child table than the reader queried.
+func TestChildTableNameDirtyICCIDSplitsOnce(t *testing.T) {
+	for _, iccid := range []string{"8986-0001", "8986 0001", "8986.0001"} {
+		got := childTableName("t_sim_card_usage_daily", iccid)
+		if strings.Contains(got, "__") {
+			t.Errorf("childTableName(%q) = %q still substitutes instead of dropping", iccid, got)
+		}
+		if got != "sim_card_usage_d_89860001" {
+			t.Errorf("childTableName(%q) = %q, want the illegal characters dropped", iccid, got)
+		}
+	}
+}
+
+func TestChildTableNameLongIdentifierUsesHashFallback(t *testing.T) {
+	iccid := strings.Repeat("8", 191)
+	got := childTableName("t_sim_card_usage_daily", iccid)
+	if len(got) > 190 {
+		t.Fatalf("childTableName kept an over-long name: len=%d", len(got))
+	}
+	if got != adminChildTrafficTableName("t_sim_card_usage_daily", iccid) {
+		t.Fatalf("hash fallback diverged from the admin rule: %q", got)
+	}
+}
+
+func TestSnapshotDayUsesEndOfDay(t *testing.T) {
+	// The dashboard "yesterday top 5" and the diagnosis page both select the
+	// 23:59:59.000 row, so the timestamp contract must not drift.
+	day := snapshotDay(mustParse(t, "2026-10-09T13:45:00+08:00"))
+	if got := day.Format("2006-01-02 15:04:05.000"); got != "2026-10-09 23:59:59.000" {
+		t.Fatalf("snapshotDay = %s, want 2026-10-09 23:59:59.000", got)
+	}
+}
+
+func mustParse(t *testing.T, value string) time.Time {
+	t.Helper()
+	parsed, err := time.Parse(time.RFC3339, value)
+	if err != nil {
+		t.Fatalf("parse %q: %v", value, err)
+	}
+	return parsed
+}
+
+// TestSnapshotMonthMatchesAdminRule pins the monthly timestamp contract.
+//
+// The diagnosis page reads monthly history by ts and groups it with
+// ts.Format("2006-01"), while admin's monthStart wrote the end-of-month
+// 23:59:59.000 row. A migration that changed this would leave the page reading
+// a month that never gets written.
+func TestSnapshotMonthMatchesAdminRule(t *testing.T) {
+	// adminMonthStart is a verbatim copy of Admin's monthStart
+	// (iotlcm-admin-svr internal/logic/trafficusage/traffic_usage.go), despite
+	// its name returning the end of the month.
+	adminMonthStart := func(tm time.Time) time.Time {
+		if tm.IsZero() {
+			tm = time.Now()
+		}
+		return time.Date(tm.Year(), tm.Month()+1, 0, 23, 59, 59, 0, tm.Location())
+	}
+	location := time.FixedZone("UTC+8", 8*60*60)
+	for _, month := range []time.Time{
+		time.Date(2026, 9, 1, 0, 0, 0, 0, location),
+		time.Date(2026, 2, 1, 0, 0, 0, 0, location),
+		time.Date(2026, 12, 1, 0, 0, 0, 0, location),
+		// Leap February must land on the 29th.
+		time.Date(2024, 2, 1, 0, 0, 0, 0, location),
+	} {
+		want := adminMonthStart(month).Format("2006-01-02 15:04:05.000")
+		got := snapshotMonth(month).Format("2006-01-02 15:04:05.000")
+		if got != want {
+			t.Errorf("snapshotMonth(%s) = %s, admin rule yields %s", month.Format("2006-01"), got, want)
+		}
+	}
+}