mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
5299e18cd8
播种判据曾经是「config 表为空」。而发行包的 postinst 先跑 setup.sh → initconfig，后者会写一行 webui.listen_addr，于是**全新安装**被判定为 "已有配置"并整体跳过播种：没有 core.plugin.dir（装完 0 个插件）、没有 core.memory.* 路径、也没有随包模型对应的多模态 provider——754MB 产物与 24MB 运行库全成死重量。同样的机制此前已在协议 2 迁移演练里被观察到。 也不能改成"每次都补缺键"：老安装升级时被注进新默认值，会让它突然去加载 一个 1.8GB 的模型，那是刻意要避免的静默变重。 故改为显式标记 core.internal.seed_version： 有标记 → 已播种，返回 无标记但有 core.daemon.data_dir → 老安装，只补标记、不播种 两者都没有 → 全新安装，播种并打标记 测试：TestSeedDefaultsAfterInitconfigPrepopulate（精确复现 initconfig 的 那一行写入）、TestSeedDefaultsDoesNotInjectIntoLegacyInstall。
396 lines
11 KiB
Go
396 lines
11 KiB
Go
package config
|
||
|
||
import (
|
||
"path/filepath"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestRegistryBasic(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
r.Register("core.llm.model", "deepseek-v4-flash")
|
||
r.Register("plugin.qq.access_token", "abc123")
|
||
|
||
val, err := r.Get("core.llm.model")
|
||
if err != nil {
|
||
t.Fatalf("Get error: %v", err)
|
||
}
|
||
if v, ok := val.(string); !ok || v != "deepseek-v4-flash" {
|
||
t.Fatalf("expected deepseek-v4-flash, got %v", val)
|
||
}
|
||
|
||
keys := r.List("core")
|
||
if len(keys) != 1 || keys[0] != "core.llm.model" {
|
||
t.Fatalf("expected [core.llm.model], got %v", keys)
|
||
}
|
||
|
||
r.Set("core.llm.model", "gpt-4")
|
||
val, _ = r.Get("core.llm.model")
|
||
if v, _ := val.(string); v != "gpt-4" {
|
||
t.Fatalf("expected gpt-4, got %v", val)
|
||
}
|
||
}
|
||
|
||
func TestRegistryPersist(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
|
||
r := NewConfigRegistry(path)
|
||
r.Register("core.log_level", "debug")
|
||
r.Set("plugin.test.key", "42")
|
||
if err := r.Flush(); err != nil {
|
||
t.Fatalf("Flush: %v", err)
|
||
}
|
||
r.Close()
|
||
|
||
r2 := NewConfigRegistry(path)
|
||
val, err := r2.Get("plugin.test.key")
|
||
if err != nil {
|
||
t.Fatalf("Get after reload: %v", err)
|
||
}
|
||
if v, _ := val.(string); v != "42" {
|
||
t.Fatalf("expected 42, got %v", val)
|
||
}
|
||
r2.Close()
|
||
}
|
||
|
||
func TestRegistryDelete(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
r.Register("a.b", "1")
|
||
r.Register("a.c", "2")
|
||
r.Delete("a.b")
|
||
keys := r.List("a")
|
||
if len(keys) != 1 || keys[0] != "a.c" {
|
||
t.Fatalf("expected [a.c], got %v", keys)
|
||
}
|
||
}
|
||
|
||
func TestRegistryDump(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
r.Register("x", "1")
|
||
r.Register("y", "two")
|
||
dump := r.Dump()
|
||
if len(dump) != 2 {
|
||
t.Fatalf("expected 2 keys, got %d", len(dump))
|
||
}
|
||
}
|
||
|
||
func TestRegistryUnknownKey(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
_, err := r.Get("nonexistent")
|
||
if err == nil {
|
||
t.Fatal("expected error for unknown key")
|
||
}
|
||
}
|
||
|
||
func TestRegistryFlush(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
r := NewConfigRegistry(path)
|
||
r.Set("k", "v")
|
||
if err := r.Flush(); err != nil {
|
||
t.Fatalf("Flush: %v", err)
|
||
}
|
||
r.Close()
|
||
|
||
// Reopen and verify persistence
|
||
r2 := NewConfigRegistry(path)
|
||
val, err := r2.Get("k")
|
||
if err != nil {
|
||
t.Fatalf("Get after flush: %v", err)
|
||
}
|
||
if v, _ := val.(string); v != "v" {
|
||
t.Fatalf("expected v, got %v", val)
|
||
}
|
||
r2.Close()
|
||
}
|
||
|
||
func TestRegistryFlushIdempotent(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
r := NewConfigRegistry(path)
|
||
r.Set("k", "v")
|
||
r.Flush()
|
||
r.Flush() // second flush should not error
|
||
r.Close()
|
||
}
|
||
|
||
func TestPluginConfig(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
r := NewConfigRegistry(path)
|
||
|
||
ps := r.PluginConfig("test_deepseek")
|
||
ps.RegisterDef(ConfigDef{Key: "api_key", Default: "sk-test123"})
|
||
if err := ps.Set("api_key", "sk-test123"); err != nil {
|
||
t.Fatalf("PluginSettings.Set: %v", err)
|
||
}
|
||
|
||
val, err := ps.Get("api_key")
|
||
if err != nil {
|
||
t.Fatalf("PluginSettings.Get: %v", err)
|
||
}
|
||
if v, _ := val.(string); v != "sk-test123" {
|
||
t.Fatalf("expected sk-test123, got %v", val)
|
||
}
|
||
|
||
keys, err := ps.List("")
|
||
if err != nil {
|
||
t.Fatalf("PluginSettings.List: %v", err)
|
||
}
|
||
if len(keys) != 1 || keys[0] != "api_key" {
|
||
t.Fatalf("expected [api_key], got %v", keys)
|
||
}
|
||
|
||
// Core table should not contain plugin data
|
||
coreKeys := r.List("")
|
||
for _, k := range coreKeys {
|
||
if k == "api_key" {
|
||
t.Fatal("plugin key leaked into core config table")
|
||
}
|
||
}
|
||
|
||
r.Close()
|
||
}
|
||
|
||
func TestSeedDefaultsToConfig(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
|
||
r := NewConfigRegistry(path)
|
||
r.SeedDefaults(dir)
|
||
|
||
// Verify DB was seeded with expected number of keys
|
||
keys := r.List("")
|
||
if len(keys) == 0 {
|
||
t.Fatal("SeedDefaults produced empty DB")
|
||
}
|
||
|
||
// Reconstruct config from DB
|
||
cfg2 := r.ToConfig()
|
||
|
||
if v := r.GetString("webui.listen_addr", ""); v != ":8080" {
|
||
t.Fatalf("expected :8080, got %s", v)
|
||
}
|
||
if cfg2.LLM.Provider != "deepseek" {
|
||
t.Fatalf("expected deepseek, got %s", cfg2.LLM.Provider)
|
||
}
|
||
if len(cfg2.LLM.Sources) == 0 {
|
||
t.Fatal("expected at least 1 LLM source")
|
||
}
|
||
|
||
// Second SeedDefaults should be no-op (DB already has data)
|
||
r.SeedDefaults(dir)
|
||
if len(r.List("")) != len(keys) {
|
||
t.Fatal("second SeedDefaults changed DB count")
|
||
}
|
||
|
||
r.Close()
|
||
}
|
||
|
||
// 发行包全新安装：postinst 先跑 setup.sh → initconfig，而 initconfig 只写
|
||
// webui.listen_addr。于是 config 表已经非空，旧实现据此判定“已有配置”并整体
|
||
// 跳过播种——装完没有 core.plugin.dir（0 个插件）、也没有随包模型对应的
|
||
// 多模态 provider（754MB 产物 + 24MB 运行库全成死重量）。
|
||
func TestSeedDefaultsAfterInitconfigPrepopulate(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
|
||
r := NewConfigRegistry(path)
|
||
// 精确复现 initconfig 的唯一一笔写入
|
||
if _, err := r.db.Exec(`INSERT INTO config (key, value) VALUES ('webui.listen_addr', ':8080')`); err != nil {
|
||
t.Fatalf("预置 initconfig 行: %v", err)
|
||
}
|
||
|
||
r.SeedDefaults(dir)
|
||
|
||
for _, k := range []string{"core.daemon.data_dir", "core.plugin.dir", "core.memory.multimodal_space.provider"} {
|
||
if r.GetString(k, "") == "" {
|
||
t.Fatalf("全新安装（initconfig 已写 webui.listen_addr）后 %s 仍为空：默认值播种被跳过", k)
|
||
}
|
||
}
|
||
if got := r.GetString("core.memory.multimodal_space.provider", ""); got != "chineseclip" {
|
||
t.Fatalf("随包默认 provider 应为 chineseclip，实为 %q", got)
|
||
}
|
||
r.Close()
|
||
}
|
||
|
||
// 老安装升级：绝不能因为新版本加了默认值就把它注进现有 DB——那会让升级即
|
||
// 静默加载一个 1.8GB 的模型。判据是 core.daemon.data_dir 在场（老安装由播种
|
||
// 写入）而 seed 标记缺失。
|
||
func TestSeedDefaultsDoesNotInjectIntoLegacyInstall(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "config.db")
|
||
|
||
r := NewConfigRegistry(path)
|
||
if _, err := r.db.Exec(`INSERT INTO config (key, value) VALUES ('core.daemon.data_dir', ?)`, dir); err != nil {
|
||
t.Fatalf("预置老安装行: %v", err)
|
||
}
|
||
|
||
r.SeedDefaults(dir)
|
||
|
||
if got := r.GetString("core.memory.multimodal_space.provider", ""); got != "" {
|
||
t.Fatalf("老安装升级被注入新默认值 provider=%q（升级后会静默加载大模型）", got)
|
||
}
|
||
if got := r.GetString("core.plugin.dir", ""); got != "" {
|
||
t.Fatalf("老安装升级被注入新默认值 core.plugin.dir=%q", got)
|
||
}
|
||
|
||
// 但标记必须补上，否则每次启动都会重走判断
|
||
var n int
|
||
if err := r.db.QueryRow(`SELECT COUNT(*) FROM config WHERE key = 'core.internal.seed_version'`).Scan(&n); err != nil {
|
||
t.Fatalf("查 seed 标记: %v", err)
|
||
}
|
||
if n != 1 {
|
||
t.Fatalf("老安装应补上 seed 标记，实际 count=%d", n)
|
||
}
|
||
r.Close()
|
||
}
|
||
|
||
func TestGetHelpers(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
r.Set("str_key", "hello")
|
||
r.Set("int_key", "42")
|
||
r.Set("dur_key", "5m")
|
||
r.Set("bool_key", "true")
|
||
|
||
if got := r.GetString("str_key", ""); got != "hello" {
|
||
t.Fatalf("GetString: expected hello, got %s", got)
|
||
}
|
||
if got := r.GetString("nonexistent", "fallback"); got != "fallback" {
|
||
t.Fatalf("GetString fallback: expected fallback, got %s", got)
|
||
}
|
||
if got := r.GetInt("int_key", 0); got != 42 {
|
||
t.Fatalf("GetInt: expected 42, got %d", got)
|
||
}
|
||
if got := r.GetInt("nonexistent", 99); got != 99 {
|
||
t.Fatalf("GetInt fallback: expected 99, got %d", got)
|
||
}
|
||
if got := r.GetDuration("dur_key", 0); got != 5*time.Minute {
|
||
t.Fatalf("GetDuration: expected 5m, got %v", got)
|
||
}
|
||
r.Set("dur_key_days", "2d")
|
||
if got := r.GetDuration("dur_key_days", 0); got != 48*time.Hour {
|
||
t.Fatalf("GetDuration d-unit: expected 48h, got %v", got)
|
||
}
|
||
r.Set("dur_key_weeks", "1w")
|
||
if got := r.GetDuration("dur_key_weeks", 0); got != 168*time.Hour {
|
||
t.Fatalf("GetDuration w-unit: expected 168h, got %v", got)
|
||
}
|
||
if got := r.GetDuration("nonexistent", 30*time.Second); got != 30*time.Second {
|
||
t.Fatalf("GetDuration fallback: expected 30s, got %v", got)
|
||
}
|
||
if got := r.GetBool("bool_key", false); got != true {
|
||
t.Fatalf("GetBool: expected true, got %v", got)
|
||
}
|
||
if got := r.GetBool("nonexistent", true); got != true {
|
||
t.Fatalf("GetBool fallback: expected true, got %v", got)
|
||
}
|
||
}
|
||
|
||
func TestParseDurationExtended(t *testing.T) {
|
||
cases := []struct {
|
||
in string
|
||
want time.Duration
|
||
wantErr bool
|
||
}{
|
||
{"2d", 48 * time.Hour, false},
|
||
{"1w", 7 * 24 * time.Hour, false},
|
||
{"1d", 24 * time.Hour, false},
|
||
{"2d12h", 60 * time.Hour, false},
|
||
{"30m", 30 * time.Minute, false},
|
||
{"500ms", 500 * time.Millisecond, false},
|
||
{"1h30m", 90 * time.Minute, false},
|
||
{" 3d ", 72 * time.Hour, false},
|
||
{"2w", 336 * time.Hour, false},
|
||
{"", 0, true},
|
||
{"abc", 0, true},
|
||
}
|
||
for _, c := range cases {
|
||
got, err := parseDurationExtended(c.in)
|
||
if c.wantErr {
|
||
if err == nil {
|
||
t.Errorf("%q: expected error, got %v", c.in, got)
|
||
}
|
||
continue
|
||
}
|
||
if err != nil {
|
||
t.Errorf("%q: unexpected error: %v", c.in, err)
|
||
continue
|
||
}
|
||
if got != c.want {
|
||
t.Errorf("%q: expected %v, got %v", c.in, c.want, got)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestSnapshotRestoreCoreLLM(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
defer r.Close()
|
||
|
||
r.Set("core.llm.sources.main.base_url", "https://a")
|
||
r.Set("core.llm.sources.main.model", "m1")
|
||
r.Set("core.llm.sources.main.api_key", "k1")
|
||
|
||
snap := r.SnapshotCoreLLM()
|
||
if len(snap) != 3 {
|
||
t.Fatalf("expected 3 keys, got %d: %v", len(snap), snap)
|
||
}
|
||
|
||
// 模拟写坏
|
||
r.Set("core.llm.sources.main.base_url", "https://broken")
|
||
r.Set("core.llm.sources.main.api_key", "hacked")
|
||
r.Set("core.llm.sources.extra.model", "intruder")
|
||
|
||
if err := r.RestoreCoreLLM(snap); err != nil {
|
||
t.Fatalf("RestoreCoreLLM: %v", err)
|
||
}
|
||
got := r.SnapshotCoreLLM()
|
||
if len(got) != 3 {
|
||
t.Fatalf("after restore expected 3 keys, got %d: %v", len(got), got)
|
||
}
|
||
for k, v := range snap {
|
||
if got[k] != v {
|
||
t.Errorf("key %s: want %q got %q", k, v, got[k])
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestLLMSnapshotFile(t *testing.T) {
|
||
path := filepath.Join(t.TempDir(), "llm_snapshot.json")
|
||
snap := map[string]string{"core.llm.sources.main.base_url": "https://a", "core.llm.sources.main.model": "m1"}
|
||
if err := SaveLLMSnapshot(path, snap); err != nil {
|
||
t.Fatalf("Save: %v", err)
|
||
}
|
||
got, err := LoadLLMSnapshot(path)
|
||
if err != nil {
|
||
t.Fatalf("Load: %v", err)
|
||
}
|
||
if got["core.llm.sources.main.base_url"] != "https://a" || got["core.llm.sources.main.model"] != "m1" {
|
||
t.Fatalf("round-trip mismatch: %v", got)
|
||
}
|
||
}
|
||
|
||
func TestSetLLMSnapshotFile(t *testing.T) {
|
||
r := NewConfigRegistry("")
|
||
defer r.Close()
|
||
r.Set("core.llm.sources.main.base_url", "https://orig")
|
||
r.Set("core.llm.sources.main.model", "m0")
|
||
|
||
path := filepath.Join(t.TempDir(), "llm_pre.json")
|
||
r.SetLLMSnapshotFile(path)
|
||
// 再次写入：写前自动留档应记录当前值 orig/m0，随后才被覆盖
|
||
r.Set("core.llm.sources.main.base_url", "https://broken")
|
||
|
||
got, err := LoadLLMSnapshot(path)
|
||
if err != nil {
|
||
t.Fatalf("Load snapshot: %v", err)
|
||
}
|
||
if got["core.llm.sources.main.base_url"] != "https://orig" {
|
||
t.Fatalf("write-ahead snapshot should record pre-write value, got %q", got["core.llm.sources.main.base_url"])
|
||
}
|
||
if got["core.llm.sources.main.model"] != "m0" {
|
||
t.Fatalf("snapshot missing untouched key model: %v", got)
|
||
}
|
||
}
|