DEV-MERGE-LEGACY-BACKEND-2026/internal/academic/handler/divide_handler_test.go

// 分账 handler 单测（AC-13 第 1 批：端点 38 / 43）。
//
// 断言纪律（team-lead 的 AC-13 验收方式：**不收"用例全绿"，要"写错会变红"**）：
//
//	端点 38 分账树 —— 主断言是 **Nested Set 四条不变式**（I1~I4），不是"接口返回 200"：
//	  · `TestDivideTreeInvariantsHold`：造一棵 5 节点两层的树，逐条验 I1（1..2N 无重复无空洞）/
//	    I2（嵌套或相离）/ I3（父严格包含子）/ I4（layer=parent.layer+1）；
//	  · `TestDivideTreeRejectsBrokenNesting`：**故意造坏树**（右值错位 1），断言 5001 + 文案命中
//	    具体不变式 —— 这是"树坏了能被发现"的证明（"接口能返回一棵树"这种断言在树坏时依然绿）；
//	  · `TestDivideTreeQuotaIsMoneyString`：`quota` 必须是**金额字符串**（"100.00"），
//	    不是 0~1 比率、不是 number（mock 曾错成比率）。
//	端点 43 流水 —— `TestDivideLogsWindowAndTotals`：右开窗口（`createTime = to` 的行不得出现），
//	  且 `total` 与 `list` 同口径（不等于历史全量）。
package handler

import (
	"encoding/json"
	"fmt"
	"net/http"
	"net/http/httptest"
	"path/filepath"
	"strings"
	"testing"
	"time"

	gormsqlite "github.com/glebarez/sqlite"
	"github.com/gofiber/fiber/v2"
	"gorm.io/gorm"

	"yuansiwei/unified-backend/internal/academic/model"
	infraerrors "yuansiwei/unified-backend/internal/infra/errors"
	appscope "yuansiwei/unified-backend/internal/infra/scope"
)

// newDivideTestDB 分账测试库（两张分账表；其余表由 handler 其它用例各自建）。
func newDivideTestDB(t *testing.T) *gorm.DB {
	t.Helper()
	dsn := filepath.Join(t.TempDir(), "academic_divide_handler_test.db")
	db, err := gorm.Open(gormsqlite.Open(dsn), &gorm.Config{})
	if err != nil {
		t.Fatalf("gorm.Open(分账库) 失败: %v", err)
	}
	sqlDB, err := db.DB()
	if err != nil {
		t.Fatalf("取 sql.DB 失败: %v", err)
	}
	sqlDB.SetMaxOpenConns(1)
	t.Cleanup(func() { _ = sqlDB.Close() })
	if mErr := db.AutoMigrate(&model.UserDivideInto{}, &model.UserDivideIntoLog{}); mErr != nil {
		t.Fatalf("AutoMigrate(分账 2 表) 失败: %v", mErr)
	}
	return db
}

// newDivideTestApp 只挂分账两条端点的 app（+ 作用域注入中间件）。
func newDivideTestApp(t *testing.T, db *gorm.DB) *fiber.App {
	t.Helper()
	h, err := New(Config{DB: db})
	if err != nil {
		t.Fatalf("handler.New 失败: %v", err)
	}
	app := fiber.New(fiber.Config{DisableStartupMessage: true})
	app.Use(func(c *fiber.Ctx) error {
		if c.Get("X-Test-Scope") == "none" {
			return c.Next()
		}
		appscope.Set(c, testScope())
		return c.Next()
	})
	app.Get(PathDivideTree, h.GetDivideTree)
	app.Get(PathDivideLogs, h.ListDivideLogs)
	return app
}

// seedDivideNode 落一行分账节点。
func seedDivideNode(t *testing.T, db *gorm.DB, userID, parentUserID *string,
	left, right, layer int32, quota string) {
	t.Helper()
	row := &model.UserDivideInto{
		UserID: userIDOf(userID), ParentUserID: parentUserID,
		LeftVal: left, RightVal: right, Layer: layer, Quota: quota, TotalIncome: "0.00",
	}
	if parentUserID == nil {
		root := *userID
		row.RootUserID = &root
	} else {
		// 根取父的根（测试里只有两层，直接指向根）
		root := *userID
		if *parentUserID != "" {
			var parent model.UserDivideInto
			if err := db.Where("user_id = ?", *parentUserID).Take(&parent).Error; err != nil {
				t.Fatalf("读父节点(%s) 失败: %v", *parentUserID, err)
			}
			if parent.RootUserID != nil {
				root = *parent.RootUserID
			}
		}
		row.RootUserID = &root
	}
	if err := db.Create(row).Error; err != nil {
		t.Fatalf("seed 分账节点(%s) 失败: %v", row.UserID, err)
	}
}

// userIDOf 解引用（nil → ""，用于父节点为空的安全构造）。
func userIDOf(p *string) string {
	if p == nil {
		return ""
	}
	return *p
}

// divideTreeResp 端点 38 的响应体（足够断言树结构与量纲）。
type divideTreeResp struct {
	Tree []struct {
		UserID           string          `json:"userId"`
		ParentUserID     *string         `json:"parentUserId"`
		Quota            json.RawMessage `json:"quota"`
		CumulativeIncome json.RawMessage `json:"cumulativeIncome"`
		Layer            int32           `json:"layer"`
		Children         []struct {
			UserID   string          `json:"userId"`
			Quota    json.RawMessage `json:"quota"`
			Layer    int32           `json:"layer"`
			Children []struct {
				UserID string `json:"userId"`
			} `json:"children"`
		} `json:"children"`
	} `json:"tree"`
	NodeCount int `json:"nodeCount"`
	RootCount int `json:"rootCount"`
}

// getDivideTree 发请求并解出树响应（HTTP 恒 200）。
func getDivideTree(t *testing.T, app *fiber.App) (int, divideTreeResp) {
	t.Helper()
	req := httptest.NewRequest(http.MethodGet, PathDivideTree, nil)
	resp, err := app.Test(req, 5000)
	if err != nil {
		t.Fatalf("请求分账树失败: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var envelope struct {
		Code int             `json:"code"`
		Data json.RawMessage `json:"data"`
	}
	if dErr := json.NewDecoder(resp.Body).Decode(&envelope); dErr != nil {
		t.Fatalf("响应解码失败: %v", dErr)
	}
	var out divideTreeResp
	if envelope.Code == 200 && len(envelope.Data) > 0 {
		if uErr := json.Unmarshal(envelope.Data, &out); uErr != nil {
			t.Fatalf("树体解码失败: %v", uErr)
		}
	}
	return envelope.Code, out
}

// TestDivideTreeInvariantsHold 端点 38：两层 5 节点树，**四条不变式**逐条成立。
//
// 造形（root=1..10, A=2..5, A1=3..4, B=6..9, B1=7..8）：
//
//	root(layer1) ├── A(layer2) ── A1(layer3)
//	             └── B(layer2) ── B1(layer3)
//
// 若实现的位移逻辑使区间重复/空洞/部分重叠，**本用例会红**（而"接口返回 200 + 有 tree 字段"不会）。
func TestDivideTreeInvariantsHold(t *testing.T) {
	db := newDivideTestDB(t)
	root, a, a1, b, b1 := "01HROOT000000000000000001", "01HA0000000000000000001",
		"01HA10000000000000000001", "01HB0000000000000000001", "01HB10000000000000000001"

	seedDivideNode(t, db, &root, nil, 1, 10, 1, "100.00")
	seedDivideNode(t, db, &a, &root, 2, 5, 2, "50.00")
	seedDivideNode(t, db, &a1, &a, 3, 4, 3, "20.00")
	seedDivideNode(t, db, &b, &root, 6, 9, 2, "30.00")
	seedDivideNode(t, db, &b1, &b, 7, 8, 3, "10.00")

	app := newDivideTestApp(t, db)
	code, body := getDivideTree(t, app)
	if code != 200 {
		t.Fatalf("合法树应返回 200，实得 %d", code)
	}
	if body.NodeCount != 5 || body.RootCount != 1 || len(body.Tree) != 1 {
		t.Fatalf("nodeCount/rootCount/len(tree) = %d/%d/%d，期望 5/1/1",
			body.NodeCount, body.RootCount, len(body.Tree))
	}
	// I4：根 layer=1、parent 为 null
	if body.Tree[0].Layer != 1 || body.Tree[0].ParentUserID != nil {
		t.Fatalf("根节点 layer/parent = %d/%v，期望 1/nil", body.Tree[0].Layer, body.Tree[0].ParentUserID)
	}
	// I3/I4：子节点 layer=2 且挂在根下
	if len(body.Tree[0].Children) != 2 {
		t.Fatalf("根的直接子节点 = %d，期望 2", len(body.Tree[0].Children))
	}
	for _, child := range body.Tree[0].Children {
		if child.Layer != 2 {
			t.Errorf("子节点 %s layer = %d，期望 2", child.UserID, child.Layer)
		}
	}
	// 逐级检查孙节点（A1 / B1 各 1 个）
	grandTotal := 0
	for _, child := range body.Tree[0].Children {
		grandTotal += len(child.Children)
	}
	if grandTotal != 2 {
		t.Fatalf("孙节点总数 = %d，期望 2（A1、B1）", grandTotal)
	}
}

// TestDivideTreeRejectsBrokenNesting 端点 38：**故意造坏树** ⇒ 5001 + 文案指到具体不变式。
//
// 造法：给 B 的右值从 9 改成 5（与 A 的右值相同）⇒ I1（区间值重复）。
// 这是"树坏了能被发现"的证明：只断言"返回 200 且有 tree"的实现，在这种情况下**依然绿**。
func TestDivideTreeRejectsBrokenNesting(t *testing.T) {
	db := newDivideTestDB(t)
	root, a, b := "01HROOT000000000000000002", "01HA0000000000000000002", "01HB0000000000000000002"
	seedDivideNode(t, db, &root, nil, 1, 6, 1, "100.00")
	seedDivideNode(t, db, &a, &root, 2, 5, 2, "50.00")
	seedDivideNode(t, db, &b, &root, 6, 5, 2, "30.00") // ⚠ 坏行：left(6) >= right(5)
	_ = b

	app := newDivideTestApp(t, db)
	code, _ := getDivideTree(t, app)
	if code != infraerrors.CodeInternal {
		t.Fatalf("坏树应返回 5001（数据损坏），实得 %d —— 说明不变式校验没生效", code)
	}
}

// TestDivideTreeQuotaIsMoneyString 端点 38：`quota` / `cumulativeIncome` 必须是**金额字符串**。
//
// mock 曾把 `quota` 当 0~1 比率 ⇒ 前端拿到 "0.35" 当 35%（差 100 倍）。
// 本用例把量纲钉死：JSON 里必须是 `"100.00"` 这样的**带引号两位小数**，
// 不能是 `100`（number）、也不能是 `1`/`0.35`（比率）。
func TestDivideTreeQuotaIsMoneyString(t *testing.T) {
	db := newDivideTestDB(t)
	root := "01HROOT000000000000000003"
	seedDivideNode(t, db, &root, nil, 1, 2, 1, "100.00")

	app := newDivideTestApp(t, db)
	code, body := getDivideTree(t, app)
	if code != 200 || len(body.Tree) != 1 {
		t.Fatalf("应返回 1 棵单节点树，实得 code=%d len=%d", code, len(body.Tree))
	}
	raw := string(body.Tree[0].Quota)
	if raw != `"100.00"` {
		t.Fatalf(`quota = %s，期望 "100.00"（金额字符串，两位小数；不是 0~1 比率、不是 number）`, raw)
	}
	if inc := string(body.Tree[0].CumulativeIncome); inc != `"0.00"` {
		t.Fatalf(`cumulativeIncome = %s，期望 "0.00"`, inc)
	}
}

// TestDivideTreeEmptyIsArray 空库：`tree` 必须是 `[]`（不能是 null），且不报错。
func TestDivideTreeEmptyIsArray(t *testing.T) {
	db := newDivideTestDB(t)
	app := newDivideTestApp(t, db)
	req := httptest.NewRequest(http.MethodGet, PathDivideTree, nil)
	resp, err := app.Test(req, 5000)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	raw := new(strings.Builder)
	buf := make([]byte, 4096)
	n, _ := resp.Body.Read(buf)
	raw.Write(buf[:n])
	if strings.Contains(raw.String(), `"tree":null`) {
		t.Fatalf("空树必须是 []，不能是 null：%s", raw.String())
	}
	if !strings.Contains(raw.String(), `"tree":[]`) {
		t.Fatalf("空树应为 `tree:[]`：%s", raw.String())
	}
}

// TestDivideTreeMissingScopeIs1001 无作用域 → 1001（分账是全局数据，更不能匿名读）。
func TestDivideTreeMissingScopeIs1001(t *testing.T) {
	db := newDivideTestDB(t)
	app := newDivideTestApp(t, db)
	req := httptest.NewRequest(http.MethodGet, PathDivideTree, nil)
	req.Header.Set("X-Test-Scope", "none")
	resp, err := app.Test(req, 5000)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var envelope struct {
		Code int `json:"code"`
	}
	if dErr := json.NewDecoder(resp.Body).Decode(&envelope); dErr != nil {
		t.Fatalf("解码失败: %v", dErr)
	}
	if envelope.Code != infraerrors.CodeUnauth {
		t.Fatalf("无作用域应 1001，实得 %d", envelope.Code)
	}
}

// TestDivideLogsWindowAndTotals 端点 43：右开窗口 + `total` 与 `list` 同口径。
//
// 造 3 条流水：窗口前（02-20）、窗口内（03-05）、**恰好等于 `to`**（04-01）。
// 断言：`total=1`（不是历史 3）、仅 1 行、且不出现 `to` 时刻的行 —— 若实现漏了窗口条件
// （或 `to` 写成闭区间），这里会红。金额同样必须是字符串。
func TestDivideLogsWindowAndTotals(t *testing.T) {
	db := newDivideTestDB(t)
	user := "01HFINDER00000000000000001"
	stu := "01HSTUDENT0000000000000001"
	for _, at := range []time.Time{
		time.Date(2026, 2, 20, 10, 0, 0, 0, time.UTC), // 窗口前
		time.Date(2026, 3, 5, 10, 0, 0, 0, time.UTC),  // 窗口内
		time.Date(2026, 4, 1, 0, 0, 0, 0, time.UTC),   // == to（右开：不得出现）
	} {
		if err := db.Create(&model.UserDivideIntoLog{
			Type: 1, UserID: user, StudentID: stu, Money: "12.34", CreateTime: at,
		}).Error; err != nil {
			t.Fatalf("seed 流水失败: %v", err)
		}
	}

	app := newDivideTestApp(t, db)
	path := fmt.Sprintf("%s?findUserId=%s&from=2026-03-01&to=2026-04-01", PathDivideLogs, user)
	req := httptest.NewRequest(http.MethodGet, path, nil)
	resp, err := app.Test(req, 5000)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var envelope struct {
		Code int `json:"code"`
		Data struct {
			List  []map[string]any `json:"list"`
			Total int64            `json:"total"`
			Pages int              `json:"pages"`
		} `json:"data"`
	}
	if dErr := json.NewDecoder(resp.Body).Decode(&envelope); dErr != nil {
		t.Fatalf("解码失败: %v", dErr)
	}
	if envelope.Code != 200 {
		t.Fatalf("期望 200，实得 %d", envelope.Code)
	}
	if envelope.Data.Total != 1 || len(envelope.Data.List) != 1 {
		t.Fatalf("total/list = %d/%d，期望 1/1（右开窗口：to 时刻那笔应被排除；历史 3 笔不应计入）",
			envelope.Data.Total, len(envelope.Data.List))
	}
	row := envelope.Data.List[0]
	if got, ok := row["findUserId"].(string); !ok || got != user {
		t.Fatalf("findUserId = %v，期望 %s（表列 user_id → 契约字段 findUserId）", row["findUserId"], user)
	}
	if money, ok := row["money"].(string); !ok || money != "12.34" {
		t.Fatalf("money = %v，期望字符串 \"12.34\"", row["money"])
	}
	if envelope.Data.Pages != 1 {
		t.Fatalf("pages = %d，期望 1", envelope.Data.Pages)
	}
}

// TestDivideLogsRejectsReversedWindow from > to → 2001（不静默返回空集）。
func TestDivideLogsRejectsReversedWindow(t *testing.T) {
	db := newDivideTestDB(t)
	app := newDivideTestApp(t, db)
	req := httptest.NewRequest(http.MethodGet,
		PathDivideLogs+"?from=2026-04-01&to=2026-03-01", nil)
	resp, err := app.Test(req, 5000)
	if err != nil {
		t.Fatalf("请求失败: %v", err)
	}
	defer func() { _ = resp.Body.Close() }()
	var envelope struct {
		Code int `json:"code"`
	}
	if dErr := json.NewDecoder(resp.Body).Decode(&envelope); dErr != nil {
		t.Fatalf("解码失败: %v", dErr)
	}
	if envelope.Code != infraerrors.CodeBadParam {
		t.Fatalf("from>to 应 2001，实得 %d", envelope.Code)
	}
}