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

2026-11-02 13:28:03 +08:00
parent 6de2e503cc 96f0f4eacd
commit 2923c78755
422 changed files with 80993 additions and 0 deletions
@@ -0,0 +1,124 @@
+package warningconfig
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/gogf/gf/v2/encoding/gjson"
+	"github.com/gogf/gf/v2/os/gtime"
+
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model/entity"
+)
+
+func TestCanExecuteProcessingAllowsStaleWarningConfigBatch(t *testing.T) {
+	executor := &executor{}
+	main := &entity.BatchMain{
+		BatchType:        consts.BatchTypeSimWarningConfig,
+		BatchAsyncStatus: consts.BatchAsyncStatusProcessing,
+		StartTime:        gtime.Now().Add(-warningConfigProcessingRetryAfter - time.Second),
+	}
+
+	canExecute, err := executor.CanExecuteProcessing(context.Background(), main)
+	if err != nil || !canExecute {
+		t.Fatalf("CanExecuteProcessing() = %t, %v; want true, nil", canExecute, err)
+	}
+}
+
+func TestConfigFromDetailSnapshotUsesRowValues(t *testing.T) {
+	fallback := Config{
+		ItemType:    warnItemTypeCard,
+		WarnUnit:    warnUnitMb,
+		WarnMethods: warnDefaultMethod,
+		IsEnabled:   1,
+	}
+	snapshot := gjson.New(map[string]any{
+		"configSet":      true,
+		"configItemType": warnItemTypeCard,
+		"warnUnit":       warnUnitMb,
+		"firstWarn":      100,
+		"secondWarn":     500,
+		"thirdWarn":      900,
+		"warnMethods":    "3,1,0",
+		"warnPhones":     "13800008000",
+		"warnEmails":     "xxx@example.com",
+		"autoAction":     warnAutoActionSuspend,
+		"isEnabled":      1,
+		"remark":         "row config",
+	})
+
+	got := configFromDetailSnapshot(snapshot, fallback)
+	if got.FirstWarn != 100 || got.SecondWarn != 500 || got.ThirdWarn != 900 {
+		t.Fatalf("unexpected thresholds: %#v", got)
+	}
+	if got.WarnMethods != "3,1,0" || got.AutoAction != warnAutoActionSuspend {
+		t.Fatalf("unexpected row notification config: %#v", got)
+	}
+}
+
+func TestCanExecuteProcessingSkipsFreshOrDifferentBatch(t *testing.T) {
+	executor := &executor{}
+	tests := []struct {
+		name string
+		main *entity.BatchMain
+	}{
+		{
+			name: "fresh processing batch",
+			main: &entity.BatchMain{
+				BatchType:        consts.BatchTypeSimWarningConfig,
+				BatchAsyncStatus: consts.BatchAsyncStatusProcessing,
+				StartTime:        gtime.Now(),
+			},
+		},
+		{
+			name: "different batch type",
+			main: &entity.BatchMain{
+				BatchType:        consts.BatchTypeSimCardStart,
+				BatchAsyncStatus: consts.BatchAsyncStatusProcessing,
+				StartTime:        gtime.Now().Add(-warningConfigProcessingRetryAfter - time.Second),
+			},
+		},
+		{
+			name: "pending warning config batch",
+			main: &entity.BatchMain{
+				BatchType:        consts.BatchTypeSimWarningConfig,
+				BatchAsyncStatus: 1,
+				StartTime:        gtime.Now().Add(-warningConfigProcessingRetryAfter - time.Second),
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			canExecute, err := executor.CanExecuteProcessing(context.Background(), tt.main)
+			if err != nil || canExecute {
+				t.Fatalf("CanExecuteProcessing() = %t, %v; want false, nil", canExecute, err)
+			}
+		})
+	}
+}
+
+func TestValidateTaskMarksWholeBatchFailed(t *testing.T) {
+	task := &Task{
+		Details: []*Detail{
+			{ICCID: "8986000000000000001"},
+			{ICCID: "8986000000000000002", Reason: "卡不存在"},
+		},
+	}
+
+	validateTask(task)
+
+	for _, detail := range task.Details {
+		if detail.Success() {
+			t.Fatalf("detail %s remained successful after atomic failure", detail.ICCID)
+		}
+	}
+	if task.FailureReason == "" {
+		t.Fatal("expected batch failure reason")
+	}
+	status, result := batchResult(countResult(task))
+	if status != consts.BatchAsyncStatusFailed || result != consts.BatchRunResultFailed {
+		t.Fatalf("batchResult() = %d, %d; want failed, failed", status, result)
+	}
+}