sync: 对齐小程序 v1.0.2 生产代码并脱敏

2026-10-29 15:15:12 +08:00
parent a7abd476f0
commit 5fcf74deb0
423 changed files with 15001 additions and 6382 deletions
@@ -0,0 +1,111 @@
+package app
+
+import (
+	"context"
+	"testing"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcfg"
+)
+
+func TestAppDiagnosisTrafficDatabaseFor(t *testing.T) {
+	tests := []struct {
+		name   string
+		config appDiagnosisTrafficConfig
+		stable string
+		want   string
+	}{
+		{
+			name:   "daily usage uses daily database",
+			config: appDiagnosisTrafficConfig{DbName: "iotlcm", DailyDbName: "iotlcm_daily", MonthlyDbName: "iotlcm_monthly"},
+			stable: appDiagnosisDailyStable,
+			want:   "iotlcm_daily",
+		},
+		{
+			name:   "monthly usage uses monthly database",
+			config: appDiagnosisTrafficConfig{DbName: "iotlcm", DailyDbName: "iotlcm_daily", MonthlyDbName: "iotlcm_monthly"},
+			stable: appDiagnosisMonthlyStable,
+			want:   "iotlcm_monthly",
+		},
+		{
+			name:   "daily usage falls back to legacy database",
+			config: appDiagnosisTrafficConfig{DbName: "iotlcm"},
+			stable: appDiagnosisDailyStable,
+			want:   "iotlcm",
+		},
+		{
+			name:   "monthly usage falls back to legacy database",
+			config: appDiagnosisTrafficConfig{DbName: "iotlcm"},
+			stable: appDiagnosisMonthlyStable,
+			want:   "iotlcm",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			if got := test.config.databaseFor(test.stable); got != test.want {
+				t.Fatalf("databaseFor(%q) = %q, want %q", test.stable, got, test.want)
+			}
+		})
+	}
+}
+
+func TestAppDiagnosisGetTrafficConfig(t *testing.T) {
+	tests := []struct {
+		name        string
+		content     string
+		wantDaily   string
+		wantMonthly string
+	}{
+		{
+			name:        "explicit split databases",
+			content:     `{"tsdb":{"tdengine":{"dbName":"iotlcm","dailyDbName":"iotlcm_daily","monthlyDbName":"iotlcm_monthly"}}}`,
+			wantDaily:   "iotlcm_daily",
+			wantMonthly: "iotlcm_monthly",
+		},
+		{
+			name:        "custom database names are trimmed",
+			content:     `{"tsdb":{"tdengine":{"dbName":" legacy ","dailyDbName":" custom_daily ","monthlyDbName":" custom_monthly "}}}`,
+			wantDaily:   "custom_daily",
+			wantMonthly: "custom_monthly",
+		},
+		{
+			name:        "missing split settings use configured database",
+			content:     `{"tsdb":{"tdengine":{"dbName":"iotlcm"}}}`,
+			wantDaily:   "iotlcm",
+			wantMonthly: "iotlcm",
+		},
+		{
+			name:        "blank split settings use configured database",
+			content:     `{"tsdb":{"tdengine":{"dbName":"iotlcm","dailyDbName":" ","monthlyDbName":""}}}`,
+			wantDaily:   "iotlcm",
+			wantMonthly: "iotlcm",
+		},
+		{
+			name:        "missing settings match console default database",
+			content:     `{"tsdb":{"tdengine":{}}}`,
+			wantDaily:   "iotlcm_daily",
+			wantMonthly: "iotlcm_daily",
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			adapter, err := gcfg.NewAdapterContent(test.content)
+			if err != nil {
+				t.Fatalf("configure traffic database: %v", err)
+			}
+			previous := g.Cfg().GetAdapter()
+			g.Cfg().SetAdapter(adapter)
+			t.Cleanup(func() { g.Cfg().SetAdapter(previous) })
+
+			config := appDiagnosisGetTrafficConfig(context.Background())
+			if got := config.databaseFor(appDiagnosisDailyStable); got != test.wantDaily {
+				t.Fatalf("daily database = %q, want %q", got, test.wantDaily)
+			}
+			if got := config.databaseFor(appDiagnosisMonthlyStable); got != test.wantMonthly {
+				t.Fatalf("monthly database = %q, want %q", got, test.wantMonthly)
+			}
+		})
+	}
+}