AskaEth
27ee0a8149
守卫用例（非法值不落盘 / 坏文件不覆盖 / 同值不重写 / 热生效两种结果）移到 admin_memory_tuning_guard_test.go，helper 与读侧用例留在 admin_memory_tuning_test.go； 口径不变，两个文件都 ≤300 行。
280 lines
10 KiB
Go
280 lines
10 KiB
Go
package gateway
|
||
|
||
import (
|
||
"encoding/json"
|
||
"net/http"
|
||
"net/http/httptest"
|
||
"os"
|
||
"path/filepath"
|
||
"testing"
|
||
"time"
|
||
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/auth"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/config"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/config/runtimeconfig"
|
||
)
|
||
|
||
// 本文件锁定记忆整理「面板调优」读写端点（`GET|PUT /api/v1/admin/memories/tidy/tuning`）
|
||
// 的判据：
|
||
// ① 读侧如实回显**盘上值**（键缺位 ⇒ 该键不出现）+ 取值规则 specs + `.env` 兜底值，
|
||
// 且不回显同文件里不受管的键；
|
||
// ② 写侧**三态**：键缺位 ⇒ 逐字节不动（含「不重写文件」）；合法值 ⇒ 采用且保留
|
||
// 同文件其它键；非法值 ⇒ `400 invalid_payload` 且**盘上一个字节都不动**；
|
||
// ③ 盘上文件解不开 ⇒ 读/写都 `409 tuning_corrupt` 且绝不覆盖；
|
||
// ④ `null` 是清除哨兵（删键 ⇒ 回到下一档兜底）；
|
||
// ⑤ RBAC 走 writeAuth（owner|admin）、写落审计 `memory.tidy.tuning`；
|
||
// ⑥ 热生效**如实**：接线了 ⇒ hot=true 且真把值交给运行期；没接线/失败 ⇒ hot=false +
|
||
// hotError，绝不把「落盘了」说成「生效了」。
|
||
|
||
// tuneEnvT 是一个注入了数据目录（= 面板配置文件落点所在）的网关测试环境。
|
||
type tuneEnvT struct {
|
||
srv *Server
|
||
ts *httptest.Server
|
||
dir string
|
||
jwt *auth.JWTService
|
||
}
|
||
|
||
// newTuneEnv 构造环境。`cfg.MemoryTidy` 被显式填上，用来验证读侧下发的
|
||
// 「缺位时 `.env` 那一档」确实取自进程生效配置（而不是网关自己编的数）。
|
||
func newTuneEnv(t *testing.T, deps Deps) *tuneEnvT {
|
||
t.Helper()
|
||
dir := t.TempDir()
|
||
cfg := cfgDefault()
|
||
cfg.MemoryTidy = config.MemoryTidyConfig{IntervalSec: 86400, QuietStartHour: 23, QuietEndHour: 7}
|
||
deps.DataDir = dir
|
||
srv, ts, jwt := buildEnv(t, cfg, deps)
|
||
return &tuneEnvT{srv: srv, ts: ts, dir: dir, jwt: jwt}
|
||
}
|
||
|
||
func (e *tuneEnvT) url() string {
|
||
return e.ts.URL + "/api/v1/admin/memories/tidy/tuning"
|
||
}
|
||
|
||
func (e *tuneEnvT) path() string {
|
||
return filepath.Join(e.dir, filepath.FromSlash(runtimeconfig.MemoryTidyConfigRelPath))
|
||
}
|
||
|
||
func (e *tuneEnvT) token(t *testing.T, role auth.Role) string {
|
||
t.Helper()
|
||
return signAny(t, e.jwt, "acct-"+string(role), role)
|
||
}
|
||
|
||
// seed 写入一份面板配置并返回写后的字节（后续逐字节比对用）。
|
||
func (e *tuneEnvT) seed(t *testing.T, body string) []byte {
|
||
t.Helper()
|
||
if err := os.MkdirAll(filepath.Dir(e.path()), 0o755); err != nil {
|
||
t.Fatalf("mkdir: %v", err)
|
||
}
|
||
if err := os.WriteFile(e.path(), []byte(body), 0o600); err != nil {
|
||
t.Fatalf("seed: %v", err)
|
||
}
|
||
return e.bytes(t)
|
||
}
|
||
|
||
// bytes 读回盘上的原始字节：「非法值不落盘」的证据就是它与之前相等。
|
||
func (e *tuneEnvT) bytes(t *testing.T) []byte {
|
||
t.Helper()
|
||
raw, err := os.ReadFile(e.path())
|
||
if err != nil {
|
||
t.Fatalf("read tuning file: %v", err)
|
||
}
|
||
return raw
|
||
}
|
||
|
||
func (e *tuneEnvT) put(t *testing.T, token, raw string) (*http.Response, []byte) {
|
||
t.Helper()
|
||
return doRaw(t, http.MethodPut, e.url(), raw, token)
|
||
}
|
||
|
||
func (e *tuneEnvT) get(t *testing.T, token string) (*http.Response, []byte) {
|
||
t.Helper()
|
||
return doReq(t, http.MethodGet, e.url(), nil, token)
|
||
}
|
||
|
||
// view 解一次成功响应的 data。
|
||
func (e *tuneEnvT) view(t *testing.T, body []byte) AdminMemoryTidyTuningView {
|
||
t.Helper()
|
||
var view AdminMemoryTidyTuningView
|
||
if err := json.Unmarshal(decodeBody(t, body)["data"], &view); err != nil {
|
||
t.Fatalf("decode view: %v (body=%s)", err, body)
|
||
}
|
||
return view
|
||
}
|
||
|
||
// assertErrorCode 断言响应是错误信封且 code 匹配。
|
||
func assertErrorCode(t *testing.T, body []byte, want string) {
|
||
t.Helper()
|
||
var eobj struct {
|
||
Code string `json:"code"`
|
||
}
|
||
if err := json.Unmarshal(decodeBody(t, body)["error"], &eobj); err != nil {
|
||
t.Fatalf("decode error: %v (body=%s)", err, body)
|
||
}
|
||
if eobj.Code != want {
|
||
t.Fatalf("error code = %q, want %q (body=%s)", eobj.Code, want, body)
|
||
}
|
||
}
|
||
|
||
// diskDoc 解出盘上的文档（键 → 原样 JSON 值）。
|
||
func (e *tuneEnvT) diskDoc(t *testing.T) map[string]json.RawMessage {
|
||
t.Helper()
|
||
doc := map[string]json.RawMessage{}
|
||
if err := json.Unmarshal(e.bytes(t), &doc); err != nil {
|
||
t.Fatalf("decode disk: %v", err)
|
||
}
|
||
return doc
|
||
}
|
||
|
||
// TestAdminMemoryTidyTuningRead_EchoesDiskValues 读侧：回显盘上 7 键（缺位不下发）、
|
||
// 下发取值规则与 `.env` 兜底值，且**不回显**同文件里不受管的键。
|
||
func TestAdminMemoryTidyTuningRead_EchoesDiskValues(t *testing.T) {
|
||
t.Parallel()
|
||
env := newTuneEnv(t, Deps{})
|
||
env.seed(t, `{
|
||
"intervalSec": 3600,
|
||
"quietStartHour": 23,
|
||
"quietEndHour": 7,
|
||
"ingestGate": false,
|
||
"t1PerWindow": 120
|
||
}`)
|
||
resp, body := env.get(t, env.token(t, auth.RoleOwner))
|
||
if resp.StatusCode != http.StatusOK {
|
||
t.Fatalf("read status = %d, want 200 (body=%s)", resp.StatusCode, body)
|
||
}
|
||
view := env.view(t, body)
|
||
if view.Path != runtimeconfig.MemoryTidyConfigRelPath || !view.Present {
|
||
t.Fatalf("view path/present = %q/%v", view.Path, view.Present)
|
||
}
|
||
if got := view.Values["intervalSec"]; got != float64(3600) {
|
||
t.Fatalf("values.intervalSec = %#v, want 3600", got)
|
||
}
|
||
if got := view.Values["ingestGate"]; got != false {
|
||
t.Fatalf("values.ingestGate = %#v, want false", got)
|
||
}
|
||
// 键缺位 ⇒ 不出现（「没配」与「配成 0」必须分得开）。
|
||
for _, absent := range []string{"quietIntervalSec", "autoApply", "tempTtlSec"} {
|
||
if _, ok := view.Values[absent]; ok {
|
||
t.Fatalf("values 不该出现 %s（盘上没这个键）：%#v", absent, view.Values)
|
||
}
|
||
}
|
||
// 不受管的键（t1PerWindow）不进 values：这张卡改不了它，混进来会误导。
|
||
if _, ok := view.Values["t1PerWindow"]; ok {
|
||
t.Fatalf("values 不该出现不受管的 t1PerWindow：%#v", view.Values)
|
||
}
|
||
if len(view.Specs) != len(memoryTidyTuningManagedKeys()) {
|
||
t.Fatalf("specs = %d 条, want %d", len(view.Specs), len(memoryTidyTuningManagedKeys()))
|
||
}
|
||
if view.EnvValues["intervalSec"] != 86400 || view.EnvValues["quietStartHour"] != 23 {
|
||
t.Fatalf("envValues = %#v, want intervalSec=86400 / quietStartHour=23", view.EnvValues)
|
||
}
|
||
if view.Hot {
|
||
t.Fatal("只读响应里 hot 必须为 false（读不改变任何东西）")
|
||
}
|
||
}
|
||
|
||
// TestAdminMemoryTidyTuningRead_MissingFileIsNotAnError 文件还没播种时：正常回 200、
|
||
// present=false、values 为空 —— 「还没配过」不是错误（与 runtimeconfig 的读取口径一致）。
|
||
func TestAdminMemoryTidyTuningRead_MissingFileIsNotAnError(t *testing.T) {
|
||
t.Parallel()
|
||
env := newTuneEnv(t, Deps{})
|
||
resp, body := env.get(t, env.token(t, auth.RoleOwner))
|
||
if resp.StatusCode != http.StatusOK {
|
||
t.Fatalf("status = %d, want 200 (body=%s)", resp.StatusCode, body)
|
||
}
|
||
view := env.view(t, body)
|
||
if view.Present || len(view.Values) != 0 {
|
||
t.Fatalf("view = %+v, want present=false / 无值", view)
|
||
}
|
||
}
|
||
|
||
// TestAdminMemoryTidyTuningRead_ViewerForbidden RBAC：viewer 403、匿名 401。
|
||
func TestAdminMemoryTidyTuningRead_ViewerForbidden(t *testing.T) {
|
||
t.Parallel()
|
||
env := newTuneEnv(t, Deps{})
|
||
env.seed(t, `{"intervalSec": 3600}`)
|
||
resp, body := env.get(t, env.token(t, auth.RoleViewer))
|
||
if resp.StatusCode != http.StatusForbidden {
|
||
t.Fatalf("viewer read status = %d, want 403 (body=%s)", resp.StatusCode, body)
|
||
}
|
||
resp, _ = env.get(t, "")
|
||
if resp.StatusCode != http.StatusUnauthorized {
|
||
t.Fatalf("anonymous read status = %d, want 401", resp.StatusCode)
|
||
}
|
||
}
|
||
|
||
// TestAdminMemoryTidyTuningWrite_AdoptsValidValuesAndKeepsOtherKeys 写侧「合法值 ⇒ 采用」：
|
||
// 7 键都能落盘，同文件里不受管的键**原样保留**（一次面板保存绝不抹掉别的旋钮）。
|
||
func TestAdminMemoryTidyTuningWrite_AdoptsValidValuesAndKeepsOtherKeys(t *testing.T) {
|
||
t.Parallel()
|
||
env := newTuneEnv(t, Deps{})
|
||
env.seed(t, `{"t1Judge": false, "t1PerWindow": 120}`)
|
||
// 用 admin 角色也走同一道门（writeAuth = owner|admin）。
|
||
resp, body := env.put(t, env.token(t, auth.RoleAdmin), `{
|
||
"intervalSec": 10800,
|
||
"quietIntervalSec": 3600,
|
||
"quietStartHour": 23,
|
||
"quietEndHour": 7,
|
||
"ingestGate": true,
|
||
"autoApply": true,
|
||
"tempTtlSec": 604800
|
||
}`)
|
||
if resp.StatusCode != http.StatusOK {
|
||
t.Fatalf("write status = %d, want 200 (body=%s)", resp.StatusCode, body)
|
||
}
|
||
view := env.view(t, body)
|
||
for key, want := range map[string]any{
|
||
"intervalSec": float64(10800), "quietIntervalSec": float64(3600),
|
||
"quietStartHour": float64(23), "quietEndHour": float64(7),
|
||
"ingestGate": true, "autoApply": true, "tempTtlSec": float64(604800),
|
||
} {
|
||
if got := view.Values[key]; got != want {
|
||
t.Fatalf("回显 values[%s] = %#v, want %#v", key, got, want)
|
||
}
|
||
}
|
||
doc := env.diskDoc(t)
|
||
if string(doc["intervalSec"]) != "10800" || string(doc["t1PerWindow"]) != "120" {
|
||
t.Fatalf("disk = %s", string(env.bytes(t)))
|
||
}
|
||
if _, ok := doc["t1Judge"]; !ok {
|
||
t.Fatalf("不受管的 t1Judge 被抹掉了：%s", string(env.bytes(t)))
|
||
}
|
||
// 未接运行期 ⇒ 如实说「已落盘、重启后生效」。
|
||
if view.Hot || view.HotError == "" {
|
||
t.Fatalf("未接热生效时 hot/hotError = %v/%q, want false/非空", view.Hot, view.HotError)
|
||
}
|
||
}
|
||
|
||
// TestAdminMemoryTidyTuningWrite_AbsentKeyLeavesBytesUntouched 三态之一：键**缺位** ⇒
|
||
// 不改动。空补丁连文件都不重写（mtime 都不动）；只提交一个键时另一个键逐字节不动。
|
||
func TestAdminMemoryTidyTuningWrite_AbsentKeyLeavesBytesUntouched(t *testing.T) {
|
||
t.Parallel()
|
||
env := newTuneEnv(t, Deps{})
|
||
before := env.seed(t, `{"intervalSec": 86400, "quietStartHour": 23}`)
|
||
past := time.Now().Add(-time.Hour)
|
||
if err := os.Chtimes(env.path(), past, past); err != nil {
|
||
t.Fatalf("chtimes: %v", err)
|
||
}
|
||
tok := env.token(t, auth.RoleOwner)
|
||
resp, body := env.put(t, tok, `{}`)
|
||
if resp.StatusCode != http.StatusOK {
|
||
t.Fatalf("empty patch status = %d, want 200 (body=%s)", resp.StatusCode, body)
|
||
}
|
||
if after := env.bytes(t); string(after) != string(before) {
|
||
t.Fatalf("空补丁改动了盘上内容：\nbefore=%s\nafter=%s", before, after)
|
||
}
|
||
info, err := os.Stat(env.path())
|
||
if err != nil {
|
||
t.Fatalf("stat: %v", err)
|
||
}
|
||
if !info.ModTime().Equal(past) {
|
||
t.Fatalf("空补丁重写了文件（mtime %v ⇒ %v）", past, info.ModTime())
|
||
}
|
||
if resp, body := env.put(t, tok, `{"quietStartHour": 22}`); resp.StatusCode != http.StatusOK {
|
||
t.Fatalf("partial patch status = %d (body=%s)", resp.StatusCode, body)
|
||
}
|
||
doc := env.diskDoc(t)
|
||
if string(doc["intervalSec"]) != "86400" || string(doc["quietStartHour"]) != "22" {
|
||
t.Fatalf("partial patch 结果 = %s", string(env.bytes(t)))
|
||
}
|
||
}
|