| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376 |
- // 分账 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)
- }
- }
|