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

worker-svr/internal/worker/monthlyusage_config_test.go

@@ -0,0 +1,39 @@
+package worker
+
+import (
+	"context"
+	"testing"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcfg"
+)
+
+func TestMonthlyUsageScanStartupConfig(t *testing.T) {
+	old := g.Cfg().GetAdapter()
+	defer g.Cfg().SetAdapter(old)
+	for _, test := range []struct {
+		name, yaml  string
+		concurrency int
+		invalid     bool
+	}{
+		{"omitted", "{}", 4, false},
+		{"channel", "trafficUsageFullSync:\n  producer:\n    scanMode: channel\n    scanConcurrency: 4", 4, false},
+		{"zero", "trafficUsageFullSync:\n  producer:\n    scanConcurrency: 0", 0, true},
+		{"negative", "trafficUsageFullSync:\n  producer:\n    scanConcurrency: -1", 0, true},
+	} {
+		t.Run(test.name, func(t *testing.T) {
+			adapter, err := gcfg.NewAdapterContent(test.yaml)
+			if err != nil {
+				t.Fatal(err)
+			}
+			g.Cfg().SetAdapter(adapter)
+			config, err := loadMonthlyUsageProducerConfig(context.Background())
+			if (err != nil) != test.invalid {
+				t.Fatalf("config=%+v err=%v", config, err)
+			}
+			if !test.invalid && config.ScanConcurrency != test.concurrency {
+				t.Fatalf("config=%+v", config)
+			}
+		})
+	}
+}