sync: 同步 v1.0.4 并脱敏配置与镜像

2026-10-29 14:36:49 +08:00
parent 743e8e6e6e
commit 2206a393f7
516 changed files with 18119 additions and 2253 deletions
@@ -0,0 +1,74 @@
+package common
+
+import (
+	"testing"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/request"
+)
+
+func TestInboundBatchChannelFiltering(t *testing.T) {
+	ctx := setupBatchExecutionLogListDB(t)
+	for _, statement := range []string{
+		"ALTER TABLE batch_main ADD target_dept_id INTEGER DEFAULT 0",
+		"CREATE TABLE t_sim_card_stock(batch_no TEXT,tenant_id TEXT)",
+		"INSERT INTO t_sim_card_stock VALUES('A-NESTED',''),('A-NESTED','')",
+	} {
+		if _, err := g.DB().Exec(ctx, statement); err != nil {
+			t.Fatal(err)
+		}
+	}
+	fixtures := []struct {
+		name      string
+		info      string
+		batchType int
+		runResult int
+	}{
+		{"A-NESTED", `{"inboundOptions":{"chanKey":"A"}}`, 1, consts.BatchRunResultSuccess},
+		{"A-LEGACY", `{"chanKey":"A"}`, 1, consts.BatchRunResultSuccess},
+		{"B-NESTED", `{"inboundOptions":{"chanKey":"B"},"chanKey":"A"}`, 1, consts.BatchRunResultSuccess},
+		{"A-EMPTY-NESTED", `{"inboundOptions":{"chanKey":""},"chanKey":"A"}`, 1, consts.BatchRunResultSuccess},
+		{"NO-CHANNEL", "{}", 1, consts.BatchRunResultSuccess},
+		{"INVALID-JSON", "broken", 1, consts.BatchRunResultSuccess},
+		{"OTHER-TYPE", `{"chanKey":"A"}`, 2, consts.BatchRunResultSuccess},
+		{"NOT-SUCCESS", `{"chanKey":"A"}`, 1, consts.BatchRunResultFailed},
+	}
+	for _, fixture := range fixtures {
+		if _, err := g.DB().Exec(ctx, "INSERT INTO batch_main(batch_no,batch_type,batch_info,total_count,run_result) VALUES(?,?,?,?,?)", fixture.name, fixture.batchType, fixture.info, 2, fixture.runResult); err != nil {
+			t.Fatal(err)
+		}
+	}
+	for _, test := range []struct {
+		name      string
+		channel   string
+		page      int
+		size      int
+		wantTotal int
+		wantRows  int
+	}{
+		{"渠道A兼容新旧格式", "A", 1, 20, 3, 3},
+		{"渠道B优先使用嵌套编码", "B", 1, 20, 1, 1},
+		{"清空恢复全部入库批次", "", 1, 20, 6, 6},
+		{"前后空白不影响筛选", " A ", 1, 20, 3, 3},
+		{"总数按渠道过滤后分页", "A", 2, 2, 3, 1},
+		{"不存在的渠道返回空", "missing", 1, 20, 0, 0},
+		{"特殊字符作为参数而不是SQL", "A' OR 1=1 --", 1, 20, 0, 0},
+	} {
+		t.Run(test.name, func(t *testing.T) {
+			rows, total, err := NewCommonBatchMain().SuccessList(ctx, &model.BatchMainSuccessListParam{BatchType: consts.BatchTypeInbound, ChanKey: test.channel, PageInfo: request.PageInfo{Page: test.page, PageSize: test.size}})
+			if err != nil || total != test.wantTotal || len(rows) != test.wantRows {
+				t.Fatalf("total=%d rows=%d err=%v", total, len(rows), err)
+			}
+			for _, row := range rows {
+				if row.BatchNo == "A-NESTED" && row.StockCount != 2 {
+					t.Fatalf("stock count changed: %+v", row)
+				}
+				if test.channel == "B" && row.BatchNo != "B-NESTED" {
+					t.Fatalf("nested channel priority violated: %+v", row)
+				}
+			}
+		})
+	}
+}