sync: 同步 v1.0.4 开源代码并脱敏

internal/repository/inbound/repository_test.go

@@ -1,16 +1,96 @@
 package inbound
 
 import (
+	"context"
 	"os"
 	"path/filepath"
+	"strings"
 	"testing"
 
-	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
-	serviceinbound "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/service/inbound"
-
+	_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
+	"github.com/gogf/gf/v2/database/gdb"
 	"github.com/gogf/gf/v2/encoding/gjson"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gcfg"
+
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/library/batchfields"
+	serviceinbound "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/service/inbound"
 )
 
+func TestBatchDetailRowClampsInvalidIdentifiersAndReasons(t *testing.T) {
+	task := &serviceinbound.Task{Batch: serviceinbound.Batch{BatchNo: "INB-TEST", BatchType: 103}}
+	row := &serviceinbound.Row{RowNo: 2, ICCID: strings.Repeat("9", 100), Reason: strings.Repeat("错", 1000)}
+	detail := batchDetailRow(task, row, consts.BatchDetailStatusFailed)
+	if detail.ItemId != "" || len([]rune(detail.ErrorReason.(string))) != batchfields.DetailReasonLimit {
+		t.Fatalf("invalid detail: itemID=%v, reason=%v", detail.ItemId, detail.ErrorReason)
+	}
+	if row.ICCID != strings.Repeat("9", 100) || row.Reason != strings.Repeat("错", 1000) {
+		t.Fatal("source row was mutated; full value must remain available to result export")
+	}
+}
+
+func TestSaveValidationFailureHandlesOversizedDetailsAndFallback(t *testing.T) {
+	ctx := context.Background()
+	oldAdapter := g.Cfg().GetAdapter()
+	adapter, err := gcfg.NewAdapterContent("batch:\n  inbound:\n    writeBatchSize: 1000\n")
+	if err != nil {
+		t.Fatal(err)
+	}
+	g.Cfg().SetAdapter(adapter)
+	defer g.Cfg().SetAdapter(oldAdapter)
+	for _, test := range []struct {
+		name         string
+		constraint   string
+		wantFallback bool
+	}{
+		{name: "oversized invalid row is saved", constraint: "CHECK(length(item_id) <= 64 AND length(error_reason) <= 512)"},
+		{name: "detail write fails", constraint: "CHECK(error_reason = 'allowed')", wantFallback: true},
+	} {
+		t.Run(test.name, func(t *testing.T) {
+			db, err := gdb.New(gdb.ConfigNode{Type: "sqlite", Link: "sqlite::@file(:memory:)", MaxOpenConnCount: 1})
+			if err != nil {
+				t.Fatal(err)
+			}
+			t.Cleanup(func() { _ = db.Close(ctx) })
+			statements := []string{
+				"CREATE TABLE batch_main (batch_no TEXT PRIMARY KEY, batch_type INTEGER, batch_async_status INTEGER, total_count INTEGER, valid_count INTEGER, success_count INTEGER, fail_count INTEGER, run_result INTEGER, fail_reason TEXT, batch_info TEXT, end_time DATETIME, updated_by INTEGER)",
+				"CREATE TABLE batch_detail (id INTEGER PRIMARY KEY AUTOINCREMENT, batch_no TEXT, batch_type INTEGER, detail_status INTEGER, blank_out_status INTEGER, row_no INTEGER, item_id TEXT, error_reason TEXT " + test.constraint + ", snapshot TEXT, tenant_id TEXT, created_dept INTEGER, created_by INTEGER, updated_by INTEGER)",
+				"INSERT INTO batch_main (batch_no, batch_type, batch_async_status) VALUES ('INB-TEST', 103, 2)",
+			}
+			for _, statement := range statements {
+				if _, err := db.Exec(ctx, statement); err != nil {
+					t.Fatalf("fixture failed: %v", err)
+				}
+			}
+			task := &serviceinbound.Task{
+				Batch:         serviceinbound.Batch{BatchNo: "INB-TEST", BatchType: 103},
+				Rows:          []*serviceinbound.Row{{RowNo: 2, ICCID: strings.Repeat("9", 100), Reason: strings.Repeat("错", 1000)}},
+				FailureReason: "输入格式不正确",
+			}
+			if err := (&Repository{db: db}).SaveValidationFailure(ctx, task); err != nil {
+				t.Fatalf("failed batch must be finalized: %v", err)
+			}
+			status, err := db.Model("batch_main").Ctx(ctx).Where("batch_no", "INB-TEST").Value("batch_async_status")
+			if err != nil || status.Int() != consts.BatchAsyncStatusFailed {
+				t.Fatalf("status = %v, err = %v", status, err)
+			}
+			reason, err := db.Model("batch_main").Ctx(ctx).Where("batch_no", "INB-TEST").Value("fail_reason")
+			if err != nil || strings.Contains(reason.String(), "错误明细保存失败") != test.wantFallback {
+				t.Fatalf("fallback reason = %v, err = %v", reason, err)
+			}
+			detailCount, err := db.Model("batch_detail").Ctx(ctx).Where("batch_no", "INB-TEST").Count()
+			wantDetails := 1
+			if test.wantFallback {
+				wantDetails = 0
+			}
+			if err != nil || detailCount != wantDetails {
+				t.Fatalf("details = %d, want %d, err = %v", detailCount, wantDetails, err)
+			}
+		})
+	}
+}
+
 func TestResolveLocalOssPath(t *testing.T) {
 	uploadRoot := t.TempDir()
 	expectedPath := filepath.Join(uploadRoot, "000000", "user", "file", "20260721", "source.xlsx")