sync: 对齐小程序 v1.0.2 生产代码并脱敏

server/internal/logic/system/sys_dept_test.go

@@ -1,47 +1,82 @@
 // package system_test
 // @Link  https://github.com/xiujiecn/xiu-admin
 // @Copyright  Copyright (c) 2025 LiXiujie
-// @Author  Lxj <li@xiujie.cn>
+// @Author  Lxj <li@example.com>
 // @License  https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
 package system_test
 
 import (
 	"context"
-	"fmt"
-	"git.5iot.com/iotlcm-public/iotlcm-app/internal/consts"
-	"git.5iot.com/iotlcm-public/iotlcm-app/internal/model"
-	"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
 	"testing"
 
-	_ "github.com/gogf/gf/contrib/drivers/mysql/v2"
+	"git.5iot.com/iotlcm-public/iotlcm-app/internal/library/contexts"
+	_ "git.5iot.com/iotlcm-public/iotlcm-app/internal/logic/memorydb"
+	"git.5iot.com/iotlcm-public/iotlcm-app/internal/model"
+	"git.5iot.com/iotlcm-public/iotlcm-app/internal/service"
+
+	_ "github.com/gogf/gf/contrib/drivers/sqlite/v2"
+	"github.com/gogf/gf/v2/database/gdb"
 	"github.com/gogf/gf/v2/frame/g"
-	"github.com/gogf/gf/v2/os/gctx"
 )
 
-func init() {
-	// 数据库只读，不执行
-	// g.DB().SetDryRun(true)
-	g.DB().SetDebug(true)
-	testing.Init()
+type deptTestMemoryDB struct {
+	service.IMemoryDB
+	database gdb.DB
+}
+
+func (memory *deptTestMemoryDB) DB(ctx context.Context) gdb.DB {
+	return memory.database
 }
 
 func TestSysDept_RefreshDeptAncestors(t *testing.T) {
-
-	// 设置上下文用户身份为用户
-	ctx := context.WithValue(gctx.New(), consts.ContextKey, &model.Context{
-		// 为了测试只设置了hook中需要用到的数据
-		User: &model.Identity{
-			BaseClaims: model.BaseClaims{
-				ID:       1,
-				TenantId: "000000",
-				DeptId:   1,
-			},
-		},
-	})
-
-	err := service.SysDept().RefreshDeptAncestors(ctx)
-	if err != nil {
+	var (
+		ctx = contexts.Set(context.Background(), &model.Context{
+			User: &model.Identity{BaseClaims: model.BaseClaims{ID: 1, TenantId: "A", DeptId: 1}},
+		})
+		previous       = gdb.GetConfig("default")
+		previousMemory = service.MemoryDB()
+	)
+	if err := gdb.SetConfigGroup("default", gdb.ConfigGroup{{Type: "sqlite", Link: "sqlite::@file(:memory:)", MaxOpenConnCount: 1}}); err != nil {
 		t.Fatal(err)
 	}
-	fmt.Println("RefreshDeptAncestors success")
+	database := g.DB()
+	service.RegisterMemoryDB(&deptTestMemoryDB{database: database})
+	t.Cleanup(func() {
+		service.RegisterMemoryDB(previousMemory)
+		if err := database.GetCore().ClearCacheAll(ctx); err != nil {
+			t.Error(err)
+		}
+		if err := database.Close(ctx); err != nil {
+			t.Error(err)
+		}
+		if err := gdb.SetConfigGroup("default", previous); err != nil {
+			t.Error(err)
+		}
+	})
+	if err := database.GetCore().ClearCacheAll(ctx); err != nil {
+		t.Fatal(err)
+	}
+	statements := []string{
+		"CREATE TABLE sys_dept (dept_id INTEGER PRIMARY KEY, parent_id INTEGER, tenant_id TEXT, ancestors TEXT, updated_at DATETIME, deleted_at DATETIME)",
+		"INSERT INTO sys_dept VALUES (1,0,'A','wrong',NULL,NULL),(2,1,'A','wrong',NULL,NULL),(3,2,'A','wrong',NULL,NULL),(4,1,'B','other-tenant',NULL,NULL),(5,1,'A','deleted',NULL,'2026-01-01')",
+	}
+	for _, statement := range statements {
+		if _, err := database.Exec(ctx, statement); err != nil {
+			t.Fatal(err)
+		}
+	}
+	for attempt := 0; attempt < 2; attempt++ {
+		if err := service.SysDept().RefreshDeptAncestors(ctx); err != nil {
+			t.Fatal(err)
+		}
+		for deptID, expected := range map[int]string{1: "0,", 2: "0,1,", 3: "0,1,2,", 4: "other-tenant", 5: "deleted"} {
+			actual, err := database.GetValue(ctx, "SELECT ancestors FROM sys_dept WHERE dept_id = ?", deptID)
+			if err != nil {
+				t.Fatal(err)
+			}
+			if actual.String() != expected {
+				t.Fatalf("attempt=%d, dept=%d, ancestors=%q, expected=%q", attempt, deptID, actual.String(), expected)
+			}
+		}
+	}
 }