MiFaZhan
56 lines
2.0 KiB
Go
56 lines
2.0 KiB
Go
package allocationunits
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
_ "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"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestSnapshotAndLegacyFallback(t *testing.T) {
|
|
ctx := context.Background()
|
|
if err := gdb.SetConfig(gdb.Config{"default": gdb.ConfigGroup{{Type: "sqlite", Link: "sqlite::@file(:memory:)", MaxOpenConnCount: 1}}}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = g.DB().Close(ctx) })
|
|
for _, sql := range []string{"CREATE TABLE sys_dept(dept_id INTEGER,tenant_id TEXT,dept_key TEXT,dept_name TEXT,deleted_at TEXT)", "INSERT INTO sys_dept VALUES(1,'A','a','划拨单位',NULL),(2,'A','b','接收单位',NULL),(3,'B','other','其他租户',NULL),(4,'A','gone','已删除单位','2026-01-01')"} {
|
|
if _, err := g.DB().Exec(ctx, sql); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
snapshot, err := Snapshot(ctx, "A", 1, 2)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := g.DB().Exec(ctx, "UPDATE sys_dept SET dept_name='新名称' WHERE dept_id=2"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
frozen := Resolve(snapshot, 0, 0)
|
|
legacy := Resolve(nil, 1, 2)
|
|
foreign := Resolve(nil, 3, 4)
|
|
if err := Fill(ctx, "A", []*Fields{&frozen, &legacy, &foreign}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if frozen.TargetDeptName != "接收单位" || legacy.TargetDeptName != "新名称" || foreign.SourceDeptName != "" || foreign.TargetDeptName != "" {
|
|
t.Fatalf("快照/回退/隔离错误: %+v %+v %+v", frozen, legacy, foreign)
|
|
}
|
|
if f := Resolve(gjson.New(map[string]any{"sourceDeptName": "历史名称"}), 1, 0); f.TargetDeptMode != "per_row" {
|
|
t.Fatal(f)
|
|
}
|
|
encoded, err := json.Marshal(frozen)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.Contains(string(encoded), "sourceDeptName") {
|
|
t.Fatal("JSON契约缺少单位字段")
|
|
}
|
|
merged := Merge(gjson.New(map[string]any{"jobRole": "allocate_parent"}), snapshot)
|
|
if merged.Get("jobRole").String() != "allocate_parent" || merged.Get("sourceDeptName").String() != "划拨单位" {
|
|
t.Fatal(merged)
|
|
}
|
|
}
|