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

server/internal/logic/iot/simcardbatch/allocate/splitter_metadata_test.go

@@ -0,0 +1,93 @@
+package allocate
+
+import (
+	"context"
+	"reflect"
+	"strings"
+	"testing"
+
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchimport"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/batchmetadata"
+	"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model/entity"
+
+	"github.com/gogf/gf/contrib/drivers/sqlite/v2"
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+type allocateMetadataSQLiteDriver struct {
+	*sqlite.Driver
+}
+
+func (driver *allocateMetadataSQLiteDriver) New(core *gdb.Core, _ *gdb.ConfigNode) (gdb.DB, error) {
+	return &allocateMetadataSQLiteDriver{Driver: &sqlite.Driver{Core: core}}, nil
+}
+
+func (driver *allocateMetadataSQLiteDriver) DoSelect(ctx context.Context, link gdb.Link, query string, args ...any) (gdb.Result, error) {
+	return driver.Core.DoSelect(ctx, link, strings.TrimSuffix(query, " FOR UPDATE"), args...)
+}
+
+func TestAllocateSplitInheritsRequestMetadata(t *testing.T) {
+	if err := gdb.Register("allocate_metadata_sqlite", &allocateMetadataSQLiteDriver{}); err != nil {
+		t.Fatal(err)
+	}
+	ctx := context.Background()
+	if err := gdb.SetConfig(gdb.Config{"default": gdb.ConfigGroup{{Type: "allocate_metadata_sqlite", Name: ":memory:", MaxOpenConnCount: 1}}}); err != nil {
+		t.Fatal(err)
+	}
+	t.Cleanup(func() { _ = g.DB().Close(ctx) })
+	if err := g.DB().GetCore().ClearTableFieldsAll(ctx); err != nil {
+		t.Fatal(err)
+	}
+	for table, record := range map[string]any{"batch_main": entity.BatchMain{}, "batch_detail": entity.BatchDetail{}} {
+		recordType := reflect.TypeOf(record)
+		columns := make([]string, 0, recordType.NumField())
+		for fieldIndex := 0; fieldIndex < recordType.NumField(); fieldIndex++ {
+			field := recordType.Field(fieldIndex)
+			column := field.Tag.Get("orm")
+			if column == "id" {
+				columns = append(columns, "id INTEGER PRIMARY KEY AUTOINCREMENT")
+			} else {
+				columns = append(columns, column+" TEXT")
+			}
+		}
+		if _, err := g.DB().Exec(ctx, "CREATE TABLE "+table+" ("+strings.Join(columns, ",")+")"); err != nil {
+			t.Fatal(err)
+		}
+		if _, err := g.Model(table).Ctx(ctx).TableFields(table); err != nil {
+			t.Fatal(err)
+		}
+	}
+	if _, err := g.DB().GetCore().GetTablesWithCache(); err != nil {
+		t.Fatal(err)
+	}
+	for _, mode := range []int{batchmetadata.Single, batchmetadata.Bulk} {
+		parentNo := "META-SPLIT-SINGLE"
+		if mode == batchmetadata.Bulk {
+			parentNo = "META-SPLIT-BULK"
+		}
+		req := batchimport.SubmitRequest{SourceChannel: batchmetadata.Console, ExecutionMode: mode, SourceType: consts.BatchSourceTypeBatchInput, Operator: batchimport.OperatorContext{TenantId: "A", UserId: 7}}
+		processor := NewProcessor()
+		if err := createAllocateParent(ctx, req, processor, parentNo, nil, 100); err != nil {
+			t.Fatal(err)
+		}
+		rows := []batchimport.Row{&Row{RowNo: 1, Iccid: "card1"}}
+		result, err := createAllocateChunk(ctx, req, processor, parentNo, 1, rows, nil, nil)
+		if err != nil {
+			t.Fatal(err)
+		}
+		for _, batchNo := range []string{parentNo, result.BatchNo} {
+			var batch *entity.BatchMain
+			if err := g.Model("batch_main").Ctx(ctx).Where("batch_no", batchNo).Scan(&batch); err != nil {
+				t.Fatal(err)
+			}
+			if batch == nil || batch.SourceChannel != req.SourceChannel || batch.ExecutionMode != uint(mode) || batch.SourceType != req.SourceType || batch.TotalCount != 1 {
+				t.Fatalf("metadata not inherited: %+v", batch)
+			}
+			if batchNo != parentNo && batch.SourceBatchNo != parentNo {
+				t.Fatalf("child lost parent association: %+v", batch)
+			}
+		}
+	}
+}