【capacity 唤醒式核心(77)】internal/runtime/capacity:(Decider L1启发式+L2上下文+人格覆盖→三态静息/唤醒/全能→允许工具集+capacity_awaken事件)。loop 外围允许集过滤(不硬阻断=能力≠边界红线)。config 加 Capacity(Enabled/MinConfidence/ToolSets)。wire buildCapacity 接入 loop。 【L4 记忆驱动意图(80)】recall.go:从记忆召回用户习惯(搞一下xx=办事)→唤醒L1/L2漏掉/升置信;群聊陌生人不读owner习惯(隔离红线);RecordIntent闭环(判定写回记忆越用越准);默认保守静息。 【会话消息编排(78)】gateway/session_orchestrator.go:每会话(模式/pending有界5/tail上下文尾巴20);Submit非阻塞(idle立即/busy入队ack/满拒绝);完成回调按序续接(tail+新消息合并增量注入非全量重放);Interrupt中断;REST 202/409+WS session_busy/pending_ack事件;不重造任务引擎复用loop。 【前端集成(79)】ws.ts 加 capacity_awaken/session_busy/pending_ack 解析;store 加 awakenState/sessionBusy;TopBar/ChatShell 加唤醒态徽标+忙态提示(不阻塞输入框);状态组件 status.tsx;Miuix风格。 【管理面板写CRUD(81)】20个写端点(POST/PUT/DELETE /admin/*,owner+审计落库):personas(真持久化YAML)/users apikey(签发掩码+撤销)/iot控制(真下发)/automations(建/触发/删)/llm(test真调用+临时)/channels(connect/close/身份测试actor白名单)/plugins(卸载)/memories(search真检索)/sessions(export)。无存储如实501不伪造。拆4文件满足≤500红线。前端9模块接真写。 【wire收口】buildCapacity接入loop+OnCapacity事件;server.go newOrchestrator接hub。 实机验证:iot ac.living control 200 真实改变 on:true;observability/metrics 200、health ok;全仓 build/vet/test 47包绿+前端tsc/vite。
184 lines
4.9 KiB
Go
184 lines
4.9 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func writeEnvFile(t *testing.T, content string) string {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, ".env")
|
|
if err := os.WriteFile(path, []byte(content), 0o600); err != nil {
|
|
t.Fatalf("write env file: %v", err)
|
|
}
|
|
return path
|
|
}
|
|
|
|
func TestLoad_DefaultsWhenUnset(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
|
|
cfg, err := Load("")
|
|
if err != nil {
|
|
t.Fatalf("Load() error: %v", err)
|
|
}
|
|
|
|
if cfg.Env != "development" {
|
|
t.Errorf("Env = %q, want development", cfg.Env)
|
|
}
|
|
if cfg.Server.Host != "0.0.0.0" || cfg.Server.Port != 8080 {
|
|
t.Errorf("Server = %+v, want 0.0.0.0:8080", cfg.Server)
|
|
}
|
|
if cfg.LogLevel != "debug" {
|
|
t.Errorf("LogLevel = %q, want debug", cfg.LogLevel)
|
|
}
|
|
if cfg.DB.SQLitePath != "./data/cyrene.db" {
|
|
t.Errorf("SQLitePath = %q, want ./data/cyrene.db", cfg.DB.SQLitePath)
|
|
}
|
|
want := []string{"http://localhost:5173", "http://localhost:3000"}
|
|
if !reflect.DeepEqual(cfg.AllowedOrigins, want) {
|
|
t.Errorf("AllowedOrigins = %v, want %v", cfg.AllowedOrigins, want)
|
|
}
|
|
}
|
|
|
|
func TestLoad_EnvOverridesDefault(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
t.Setenv("PORT", "9090")
|
|
t.Setenv("LOG_LEVEL", "warn")
|
|
t.Setenv("IOT_PROVIDER", "none")
|
|
|
|
cfg, err := Load("")
|
|
if err != nil {
|
|
t.Fatalf("Load() error: %v", err)
|
|
}
|
|
if cfg.Server.Port != 9090 {
|
|
t.Errorf("Port = %d, want 9090", cfg.Server.Port)
|
|
}
|
|
if cfg.LogLevel != "warn" {
|
|
t.Errorf("LogLevel = %q, want warn", cfg.LogLevel)
|
|
}
|
|
if cfg.IoT.Provider != "none" {
|
|
t.Errorf("IoT.Provider = %q, want none", cfg.IoT.Provider)
|
|
}
|
|
}
|
|
|
|
func TestLoad_DotEnvThenProcessEnv(t *testing.T) {
|
|
path := writeEnvFile(t, `PORT=7000
|
|
LOG_LEVEL=error
|
|
# comment line
|
|
IOT_PROVIDER=none
|
|
`)
|
|
// Process env overrides .env for PORT but leaves LOG_LEVEL from .env.
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
t.Setenv("PORT", "8000")
|
|
|
|
cfg, err := Load(path)
|
|
if err != nil {
|
|
t.Fatalf("Load(%q) error: %v", path, err)
|
|
}
|
|
if cfg.Server.Port != 8000 {
|
|
t.Errorf("Port = %d, want 8000 (process env wins)", cfg.Server.Port)
|
|
}
|
|
if cfg.LogLevel != "error" {
|
|
t.Errorf("LogLevel = %q, want error from .env", cfg.LogLevel)
|
|
}
|
|
if cfg.IoT.Provider != "none" {
|
|
t.Errorf("IoT.Provider = %q, want none from .env", cfg.IoT.Provider)
|
|
}
|
|
}
|
|
|
|
func TestLoad_MissingRequiredFails(t *testing.T) {
|
|
// Do not set JWT_SECRET.
|
|
if _, err := Load(""); err == nil {
|
|
t.Fatal("expected error when JWT_SECRET missing, got nil")
|
|
} else if !strings.Contains(err.Error(), "JWT_SECRET") {
|
|
t.Errorf("error should mention JWT_SECRET, got: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestLoad_ValidationMinLength(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", "short")
|
|
if _, err := Load(""); err == nil {
|
|
t.Fatal("expected error for JWT_SECRET shorter than 32, got nil")
|
|
} else if !strings.Contains(err.Error(), "at least 32") {
|
|
t.Errorf("error should mention min length, got: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestLoad_InvalidPortFails(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
t.Setenv("PORT", "not-a-number")
|
|
if _, err := Load(""); err == nil {
|
|
t.Fatal("expected error for invalid PORT, got nil")
|
|
} else if !strings.Contains(err.Error(), "PORT") {
|
|
t.Errorf("error should mention PORT, got: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestLoad_MissingDotEnvIsTolerated(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
// A non-existent .env path must not error.
|
|
cfg, err := Load(filepath.Join(t.TempDir(), "does-not-exist.env"))
|
|
if err != nil {
|
|
t.Fatalf("Load(nonexistent) error: %v", err)
|
|
}
|
|
if cfg.Server.Port != 8080 {
|
|
t.Errorf("Port = %d, want default 8080", cfg.Server.Port)
|
|
}
|
|
}
|
|
|
|
func TestLoad_CapacityDefaults(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
|
|
cfg, err := Load("")
|
|
if err != nil {
|
|
t.Fatalf("Load() error: %v", err)
|
|
}
|
|
if cfg.Capacity.Enabled {
|
|
t.Errorf("Capacity.Enabled = true, want false (default, opt-in)")
|
|
}
|
|
if cfg.Capacity.MinConfidence != 0.6 {
|
|
t.Errorf("Capacity.MinConfidence = %v, want 0.6", cfg.Capacity.MinConfidence)
|
|
}
|
|
}
|
|
|
|
func TestLoad_CapacityEnvOverrides(t *testing.T) {
|
|
t.Setenv("JWT_SECRET", strings.Repeat("s", 40))
|
|
t.Setenv("CAPACITY_ENABLED", "true")
|
|
t.Setenv("CAPACITY_MIN_CONFIDENCE", "0.75")
|
|
|
|
cfg, err := Load("")
|
|
if err != nil {
|
|
t.Fatalf("Load() error: %v", err)
|
|
}
|
|
if !cfg.Capacity.Enabled {
|
|
t.Errorf("Capacity.Enabled = false, want true")
|
|
}
|
|
if cfg.Capacity.MinConfidence != 0.75 {
|
|
t.Errorf("Capacity.MinConfidence = %v, want 0.75", cfg.Capacity.MinConfidence)
|
|
}
|
|
}
|
|
|
|
func TestParseDotEnv_QuotedAndExport(t *testing.T) {
|
|
path := writeEnvFile(t, `export FOO=bar
|
|
BAZ='quoted value'
|
|
EMPTY=
|
|
`)
|
|
m, err := parseDotEnv(path)
|
|
if err != nil {
|
|
t.Fatalf("parseDotEnv error: %v", err)
|
|
}
|
|
if m["FOO"] != "bar" {
|
|
t.Errorf("FOO = %q, want bar", m["FOO"])
|
|
}
|
|
if m["BAZ"] != "quoted value" {
|
|
t.Errorf("BAZ = %q, want 'quoted value'", m["BAZ"])
|
|
}
|
|
if v, ok := m["EMPTY"]; !ok || v != "" {
|
|
t.Errorf("EMPTY = %q (present=%v), want empty", v, ok)
|
|
}
|
|
}
|