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)
	}
}