tamakyi
b4c2a8452f
audit-2026-09-21.md 里的后端条目落地。每条都配了回归守卫，并逐条做过变异验证 （注入原 bug → 确认按正确理由变红 → 还原 → 变绿）。 安全 - CSRF 只加同源 Origin/Referer 守卫（csrf.go），不动模板 - 会话 cookie 的 Secure 按请求判断；两步验证两闸（loginGuard + 验证码一次性） - 跳转收口到 safeInternalPath；clientIP 只认 XFF 最后一段 - GPX 按属主圈定且查 RowsAffected；下载名走 downloadDisposition - 上传口统一 MaxBytesReader；落盘后的失败路径清掉半成品 数据一致性 - 区域判定两端各过一遍再 AND（apiTrack 与 movementsWhere 对齐） - Filter.Verdict 全域一致（下推到 visitsWhere）；聚合刷新必须返回 error - 附属数据改挂坐标键，Go 与 SQL 两套表达式逐位一致 - 脏备份的伪时间/越界坐标不再进领域模型；rows 循环统一 drainRows - 合并与导出：writeRawRow 上抛错误、悬空引用清理、原生列补全 性能 - placeCols 相关子查询带 dataset_id 前导列；findSourceArchive 先按大小筛 - 批量勾选改 ANY($3::bigint[])；fillupCache 加上限；同一查询不再查两遍 另：新增「全新库」冒烟测试 —— 空库跑完 Migrate 再用应用侧读查询真打一遍， 迁移块排到依赖表之前这种错只有新库会炸，老库毫无症状。
55 lines
2.0 KiB
Go
55 lines
2.0 KiB
Go
package web
|
|
|
|
import (
|
|
"context"
|
|
"math"
|
|
"testing"
|
|
)
|
|
|
|
// D2 回归：留档序列化失败必须把错误上抛，让整个合并事务回滚。
|
|
// 以前是只 log 后 return nil —— 调用方以为写成功了，合并照常提交，
|
|
// 这一行的原始列就永久缺失（下次跨备份比对认不出 UUID，导出也还原不回去）。
|
|
func TestWriteRawRowFailsLoudlyOnMarshalError(t *testing.T) {
|
|
f := newLoginFixture(t)
|
|
ctx := context.Background()
|
|
|
|
var ds int64
|
|
if err := f.pg.QueryRowContext(ctx, `SELECT id FROM datasets ORDER BY id LIMIT 1`).Scan(&ds); err != nil {
|
|
t.Skipf("库里没有数据集：%v", err)
|
|
}
|
|
const pk = 987654321
|
|
t.Cleanup(func() {
|
|
ctx := context.Background()
|
|
for _, e := range []string{"ZVISIT", "ZLOCATION"} {
|
|
f.pg.ExecContext(ctx, `DELETE FROM entity_raw WHERE dataset_id=$1 AND entity=$2 AND src_pk=$3`, ds, e, pk)
|
|
}
|
|
})
|
|
|
|
// NaN 是 json.Marshal 明确拒绝的值，拿来触发真实的序列化失败
|
|
err := writeRawRow(ctx, f.pg, ds, "ZVISIT", pk, map[string]any{"ZBAD": math.NaN()})
|
|
if err == nil {
|
|
t.Fatal("序列化失败必须返回错误；返回 nil 等于让留档静默丢失")
|
|
}
|
|
var n int
|
|
if err := f.pg.QueryRowContext(ctx, `SELECT count(*) FROM entity_raw
|
|
WHERE dataset_id=$1 AND entity='ZVISIT' AND src_pk=$2`, ds, pk).Scan(&n); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if n != 0 {
|
|
t.Errorf("失败时不应留下 entity_raw 行，实际 %d 行", n)
|
|
}
|
|
|
|
// 正常值必须写得进去（否则上面的断言可能只是「这张表根本写不动」）
|
|
if err := writeRawRow(ctx, f.pg, ds, "ZLOCATION", pk, map[string]any{"ZNAME": "回归测试"}); err != nil {
|
|
t.Fatalf("正常留档不应失败: %v", err)
|
|
}
|
|
var got string
|
|
if err := f.pg.QueryRowContext(ctx, `SELECT raw->>'ZNAME' FROM entity_raw
|
|
WHERE dataset_id=$1 AND entity='ZLOCATION' AND src_pk=$2`, ds, pk).Scan(&got); err != nil {
|
|
t.Fatalf("读取刚写入的留档失败: %v", err)
|
|
}
|
|
if got != "回归测试" {
|
|
t.Errorf("留档内容对不上：%q", got)
|
|
}
|
|
}
|