Files
2026-07-13 13:00:08 +08:00

1373 lines
50 KiB
Go

package config
import (
"os"
"path/filepath"
"reflect"
"runtime"
"strconv"
"strings"
"testing"
"github.com/BurntSushi/toml"
)
func isolateUserConfigHome(t *testing.T) string {
t.Helper()
home := t.TempDir()
t.Setenv("HOME", home)
t.Setenv("REASONIX_CREDENTIALS_STORE", "file")
t.Setenv("USERPROFILE", home)
t.Setenv("AppData", filepath.Join(home, "AppData", "Roaming"))
return home
}
// setRuntimeGOOS overrides the package-level runtimeGOOS for one test. The
// t.Setenv call is a guard: it panics if the test also uses t.Parallel, which
// would otherwise race on the shared global.
func setRuntimeGOOS(t *testing.T, goos string) {
t.Helper()
t.Setenv("REASONIX_TEST_GOOS", goos)
old := runtimeGOOS
runtimeGOOS = goos
t.Cleanup(func() { runtimeGOOS = old })
}
func expectedDefaultReasonixHome(home string) string {
if runtime.GOOS == "windows" {
return filepath.Join(home, "AppData", "Roaming", "reasonix")
}
return filepath.Join(home, ".reasonix")
}
func TestUserConfigDisplayPathCollapsesHome(t *testing.T) {
home := isolateUserConfigHome(t)
got := userConfigDisplayPath()
if !strings.HasPrefix(got, "~/") {
t.Fatalf("display path = %q, want ~/ prefix", got)
}
if !strings.HasSuffix(got, "reasonix/config.toml") {
t.Fatalf("display path = %q, want reasonix/config.toml suffix", got)
}
if strings.Contains(got, home) {
t.Fatalf("display path %q must not embed the absolute home", got)
}
}
func TestUserConfigPathUsesReasonixHome(t *testing.T) {
home := isolateUserConfigHome(t)
want := filepath.Join(expectedDefaultReasonixHome(home), "config.toml")
if got := UserConfigPath(); filepath.Clean(got) != filepath.Clean(want) {
t.Fatalf("UserConfigPath() = %q, want %q", got, want)
}
}
func TestReasonixManagedConfigPathsAreConfigFilesOnly(t *testing.T) {
home := isolateUserConfigHome(t)
setRuntimeGOOS(t, "windows")
oldConfigDir := osUserConfigDir
osUserConfigDir = func() string { return filepath.Join(home, "AppData", "Roaming") }
t.Cleanup(func() { osUserConfigDir = oldConfigDir })
paths := ReasonixManagedConfigPaths()
for _, want := range []string{
filepath.Join(home, "AppData", "Roaming", "reasonix", "config.toml"),
filepath.Join(home, ".reasonix", "config.json"),
} {
found := false
for _, got := range paths {
if samePath(got, want) {
found = true
break
}
}
if !found {
t.Fatalf("managed config paths = %v, want %s", paths, want)
}
}
// The escape hatch is file-level by contract: no directories, and none of
// the sensitive Reasonix-home siblings (credentials, hooks, skills,
// sessions) may ride along.
for _, got := range paths {
if base := filepath.Base(got); base != "config.toml" && base != "config.json" {
t.Fatalf("managed config path %q is not a known config file (paths must be files, not directories): %v", got, paths)
}
for _, forbidden := range []string{
home,
ReasonixHomeDir(),
UserCredentialsPath(),
filepath.Join(ReasonixHomeDir(), "settings.json"),
filepath.Join(ReasonixHomeDir(), "skills"),
filepath.Join(ReasonixHomeDir(), "sessions"),
} {
if samePath(got, forbidden) {
t.Fatalf("managed config paths must not include %q: %v", forbidden, paths)
}
}
}
}
func TestUserConfigPathHonorsReasonixHome(t *testing.T) {
home := isolateUserConfigHome(t)
custom := filepath.Join(home, "custom-home")
t.Setenv("REASONIX_HOME", custom)
want := filepath.Join(custom, "config.toml")
if got := UserConfigPath(); filepath.Clean(got) != filepath.Clean(want) {
t.Fatalf("UserConfigPath() = %q, want %q", got, want)
}
}
func TestLoadForRootUsesWindowsHomeFallbackWhenConfigDirUnavailable(t *testing.T) {
home := t.TempDir()
project := t.TempDir()
oldGOOS := runtimeGOOS
oldConfigDir := osUserConfigDir
oldHomeDir := osUserHomeDir
runtimeGOOS = "windows"
osUserConfigDir = func() string { return "" }
osUserHomeDir = func() (string, error) { return home, nil }
t.Cleanup(func() {
runtimeGOOS = oldGOOS
osUserConfigDir = oldConfigDir
osUserHomeDir = oldHomeDir
})
t.Setenv("REASONIX_HOME", "")
configPath := filepath.Join(home, "AppData", "Roaming", "reasonix", "config.toml")
if err := os.MkdirAll(filepath.Dir(configPath), 0o755); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(configPath, []byte("default_model = \"custom/from-home\"\n"), 0o600); err != nil {
t.Fatal(err)
}
cfg, err := LoadForRoot(project)
if err != nil {
t.Fatalf("LoadForRoot() error = %v", err)
}
if cfg.DefaultModel != "custom/from-home" {
t.Fatalf("DefaultModel = %q, want %q", cfg.DefaultModel, "custom/from-home")
}
}
func TestRenderTOMLHeaderShowsResolvedConfigPath(t *testing.T) {
isolateUserConfigHome(t)
out := RenderTOML(Default())
want := "> " + userConfigDisplayPath() + " > built-in defaults."
if !strings.Contains(out, want) {
t.Fatalf("rendered header missing resolved config path %q", want)
}
}
func TestWriteRootsForRootExcludesUserConfigDirByDefault(t *testing.T) {
isolateUserConfigHome(t)
project := t.TempDir()
cfg := Default()
roots := cfg.WriteRootsForRoot(project)
want := filepath.Clean(filepath.Dir(UserConfigPath()))
for _, root := range roots {
if filepath.Clean(root) == want {
t.Fatalf("WriteRootsForRoot() = %v, must not include user config dir %q by default", roots, want)
}
}
if got := filepath.Clean(roots[0]); got != filepath.Clean(project) {
t.Fatalf("first write root = %q, want project %q", got, project)
}
}
// TestRenderTOMLRoundTrips ensures the annotated TOML we emit parses back into
// an equivalent config — i.e. the wizard never writes a file it can't read.
func TestRenderTOMLRoundTrips(t *testing.T) {
orig := Default()
orig.Providers = append(orig.Providers, legacyMimoCustomProvider("mimo-pro"))
orig.DefaultModel = "mimo-pro"
orig.Language = "zh"
orig.UI.Theme = "light"
orig.UI.ThemeStyle = "glacier"
orig.UI.ShortcutLayout = "desktop"
orig.UI.CursorShape = "bar"
orig.Desktop.Language = "en"
orig.Desktop.LayoutStyle = "workbench"
orig.Desktop.Theme = "dark"
orig.Desktop.ThemeStyle = "graphite"
orig.Desktop.CloseBehavior = "background"
orig.Desktop.DisplayMode = "compact"
orig.Desktop.StatusBarStyle = "text"
orig.Desktop.StatusBarItems = []string{"model", "balance", "cache"}
orig.Desktop.DefaultToolApprovalMode = "auto"
orig.Desktop.CheckUpdates = boolPtr(false)
orig.Desktop.Telemetry = boolPtr(false)
orig.Notifications.Enabled = true
orig.Notifications.TurnDone = true
orig.Notifications.ApprovalRequest = true
orig.Notifications.AskRequest = true
orig.Agent.MaxSteps = 30
orig.Agent.PlannerMaxSteps = 0
orig.Agent.AutoPlanClassifier = "deepseek-flash"
orig.Agent.ReasoningLanguage = "zh"
orig.Agent.ToolResultSnipRatio = 0.65
orig.Agent.SubagentModel = "mimo-pro"
orig.Agent.SubagentModels = map[string]string{"review": "deepseek-pro"}
orig.Agent.MaxSubagentDepth = 3
orig.Agent.Keep = []string{"errors", "user_marked"}
orig.Agent.RecentKeep = 4
orig.Tools.BashTimeoutSeconds = intPtr(900)
orig.Tools.BackgroundJobs.StalledWarningSeconds = intPtr(30)
orig.Tools.Shell.Prefer = "bash"
orig.Tools.Shell.Path = "/usr/local/bin/bash"
orig.Permissions = PermissionsConfig{
Mode: "deny",
Deny: []string{"Bash(rm -rf*)"},
Allow: []string{"Bash(go test:*)", "read_file"},
}
orig.Network = NetworkConfig{
ProxyMode: "custom",
NoProxy: "localhost,127.0.0.1",
Proxy: NetworkProxyConfig{
Type: "socks5",
Server: "127.0.0.1",
Port: 7890,
Username: "user",
Password: "${REASONIX_PROXY_PASSWORD}",
},
}
orig.Environment.Enabled = boolPtr(false)
orig.Environment.Tools = map[string]string{"go": "/opt/homebrew/bin/go", "python3": "~/.pyenv/shims/python3"}
orig.Skills.Paths = []string{"~/my-skills", "../shared/skills"}
orig.Skills.ExcludedPaths = []string{"~/.agents/skills"}
orig.Skills.DisabledSkills = []string{"review", "explore"}
orig.Skills.MaxDepth = 2
orig.Bot.ToolApprovalMode = "auto"
orig.Bot.Control = BotControlConfig{Enabled: true, Addr: "127.0.0.1:39001", TokenEnv: "BOT_CONTROL_TOKEN"}
orig.Bot.Feishu.OutboundMediaRoots = []string{"/tmp/reasonix-media", "/srv/shots"}
orig.Bot.Routes = []BotRouteConfig{{
ConnectionID: "feishu-lark",
ChatType: "group",
ChatID: "oc_group",
Model: "deepseek-pro",
ToolApprovalMode: "ask",
WorkspaceRoot: "/tmp/reasonix-route",
}}
orig.Bot.DesktopWatchers = []BotDesktopWatcherConfig{{
Platform: "feishu",
ConnectionID: "feishu-lark",
Domain: "lark",
ChatType: "dm",
ChatID: "oc_watcher",
}}
orig.Bot.Connections = []BotConnectionConfig{{
ID: "feishu-lark",
Provider: "feishu",
Domain: "lark",
Label: "Lark",
Enabled: true,
Status: "connected",
Model: "deepseek-pro",
ToolApprovalMode: "yolo",
WorkspaceRoot: "/tmp/reasonix-bot",
Credential: BotConnectionCredential{AppID: "cli_lark", AppSecretEnv: "LARK_BOT_APP_SECRET"},
SessionMappings: []BotConnectionSessionMapping{{
RemoteID: "ou_123",
SessionID: "topic:topic_bot",
Scope: "project",
WorkspaceRoot: "/tmp/reasonix-bot",
UpdatedAt: "2026-06-11T00:00:00Z",
}},
}}
orig.LSP = LSPConfig{
Enabled: true,
Servers: map[string]LSPServer{
"lua": {
Command: "lua-language-server",
Args: []string{"--stdio"},
Env: map[string]string{"LUA_PATH": "./?.lua"},
LanguageID: "lua",
Extensions: []string{".lua", ".script", ".gui_script"},
InstallHint: "install lua-language-server",
},
},
}
orig.Plugins = []PluginEntry{
{Name: "example", Command: "reasonix-plugin-example"},
{Name: "stripe", Type: "http", URL: "https://mcp.stripe.com", Headers: map[string]string{"Authorization": "Bearer x"}, TrustedReadOnlyTools: []string{"customer_read"}, AutoStart: boolPtr(false), Tier: "background"},
}
mm, _ := orig.Provider("mimo-pro")
mm.BaseURL = "http://localhost:8000/v1"
mm.ChatURL = "http://localhost:8000/v1/chat/completions"
mm.ModelsURL = "http://localhost:8000/v1/models"
mm.ReasoningProtocol = "openai"
mm.PresetID = "mimo-api"
mm.PresetVersion = ProviderPresetVersion
ds, _ := orig.Provider("deepseek-flash")
ds.Effort = "max"
rendered := RenderTOML(orig)
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n---\n%s", err, rendered)
}
if got.DefaultModel != "mimo-pro" {
t.Errorf("default_model = %q, want mimo-pro", got.DefaultModel)
}
if got.ConfigVersion != 4 {
t.Errorf("config_version = %d, want 4", got.ConfigVersion)
}
if got.Language != "zh" {
t.Errorf("language = %q, want zh", got.Language)
}
if got.UI.Theme != "light" {
t.Errorf("ui.theme = %q, want light", got.UI.Theme)
}
if got.UI.ThemeStyle != "glacier" {
t.Errorf("ui.theme_style = %q, want glacier", got.UI.ThemeStyle)
}
if got.UI.ShortcutLayout != "desktop" {
t.Errorf("ui.shortcut_layout = %q, want desktop", got.UI.ShortcutLayout)
}
if got.UICursorShape() != "bar" {
t.Errorf("ui.cursor_shape = %q, want bar", got.UICursorShape())
}
if got.Desktop.Language != "en" {
t.Errorf("desktop.language = %q, want en", got.Desktop.Language)
}
if got.Desktop.LayoutStyle != "workbench" {
t.Errorf("desktop.layout_style = %q, want workbench", got.Desktop.LayoutStyle)
}
if got.Desktop.Theme != "dark" {
t.Errorf("desktop.theme = %q, want dark", got.Desktop.Theme)
}
if got.Desktop.ThemeStyle != "graphite" {
t.Errorf("desktop.theme_style = %q, want graphite", got.Desktop.ThemeStyle)
}
if got.Desktop.CloseBehavior != "background" {
t.Errorf("desktop.close_behavior = %q, want background", got.Desktop.CloseBehavior)
}
if got.DesktopDisplayMode() != "compact" {
t.Errorf("desktop.display_mode = %q, want compact", got.DesktopDisplayMode())
}
if got.Desktop.StatusBarStyle != "text" {
t.Errorf("desktop.status_bar_style = %q, want text", got.Desktop.StatusBarStyle)
}
if want := []string{"model", "balance", "cache"}; !reflect.DeepEqual(got.Desktop.StatusBarItems, want) {
t.Errorf("desktop.status_bar_items = %v, want %v", got.Desktop.StatusBarItems, want)
}
if got.DesktopDefaultToolApprovalMode() != "auto" {
t.Errorf("desktop.default_tool_approval_mode = %q, want auto", got.DesktopDefaultToolApprovalMode())
}
if got.Desktop.CheckUpdates == nil || *got.Desktop.CheckUpdates {
t.Errorf("desktop.check_updates = %+v, want false", got.Desktop.CheckUpdates)
}
if !got.Notifications.Enabled || !got.Notifications.TurnDone || !got.Notifications.ApprovalRequest || !got.Notifications.AskRequest {
t.Errorf("notifications not preserved: %+v", got.Notifications)
}
if got.Agent.MaxSteps != orig.Agent.MaxSteps {
t.Errorf("max_steps = %d, want %d", got.Agent.MaxSteps, orig.Agent.MaxSteps)
}
if got.Agent.PlannerMaxSteps != orig.Agent.PlannerMaxSteps {
t.Errorf("planner_max_steps = %d, want %d", got.Agent.PlannerMaxSteps, orig.Agent.PlannerMaxSteps)
}
if len(got.Bot.Connections) != 1 || got.Bot.Connections[0].Model != "deepseek-pro" || got.Bot.Connections[0].WorkspaceRoot != "/tmp/reasonix-bot" {
t.Errorf("bot connection not preserved: %+v", got.Bot.Connections)
}
if got.Bot.ToolApprovalMode != "auto" || got.Bot.Connections[0].ToolApprovalMode != "yolo" {
t.Errorf("bot tool approval mode not preserved: bot=%q connection=%q", got.Bot.ToolApprovalMode, got.Bot.Connections[0].ToolApprovalMode)
}
if !got.Bot.Control.Enabled || got.Bot.Control.Addr != "127.0.0.1:39001" || got.Bot.Control.TokenEnv != "BOT_CONTROL_TOKEN" {
t.Errorf("bot control not preserved: %+v", got.Bot.Control)
}
if len(got.Bot.Feishu.OutboundMediaRoots) != 2 || got.Bot.Feishu.OutboundMediaRoots[0] != "/tmp/reasonix-media" {
t.Errorf("feishu outbound_media_roots not preserved: %+v", got.Bot.Feishu.OutboundMediaRoots)
}
if len(got.Bot.Routes) != 1 || got.Bot.Routes[0].WorkspaceRoot != "/tmp/reasonix-route" || got.Bot.Routes[0].ChatID != "oc_group" {
t.Errorf("bot routes not preserved: %+v", got.Bot.Routes)
}
if len(got.Bot.DesktopWatchers) != 1 || got.Bot.DesktopWatchers[0].ChatID != "oc_watcher" || got.Bot.DesktopWatchers[0].Platform != "feishu" || got.Bot.DesktopWatchers[0].Domain != "lark" {
t.Errorf("bot desktop watchers not preserved: %+v", got.Bot.DesktopWatchers)
}
if len(got.Bot.Connections[0].SessionMappings) != 1 || got.Bot.Connections[0].SessionMappings[0].Scope != "project" || got.Bot.Connections[0].SessionMappings[0].WorkspaceRoot != "/tmp/reasonix-bot" {
t.Errorf("bot session mapping scope not preserved: %+v", got.Bot.Connections[0].SessionMappings)
}
if got.Agent.Temperature != orig.Agent.Temperature {
t.Errorf("temperature = %v, want %v", got.Agent.Temperature, orig.Agent.Temperature)
}
if got.Agent.AutoPlan != "off" {
t.Errorf("auto_plan = %q, want off", got.Agent.AutoPlan)
}
if got.Agent.AutoPlanClassifier != "deepseek-flash" {
t.Errorf("auto_plan_classifier = %q, want deepseek-flash", got.Agent.AutoPlanClassifier)
}
if got.Agent.ReasoningLanguage != "zh" {
t.Errorf("reasoning_language = %q, want zh", got.Agent.ReasoningLanguage)
}
if got.Agent.SoftCompactRatio != orig.Agent.SoftCompactRatio {
t.Errorf("soft_compact_ratio = %v, want %v", got.Agent.SoftCompactRatio, orig.Agent.SoftCompactRatio)
}
if got.Agent.ToolResultSnipRatio != orig.Agent.ToolResultSnipRatio {
t.Errorf("tool_result_snip_ratio = %v, want %v", got.Agent.ToolResultSnipRatio, orig.Agent.ToolResultSnipRatio)
}
if got.Agent.CompactRatio != orig.Agent.CompactRatio {
t.Errorf("compact_ratio = %v, want %v", got.Agent.CompactRatio, orig.Agent.CompactRatio)
}
if got.Agent.CompactForceRatio != orig.Agent.CompactForceRatio {
t.Errorf("compact_force_ratio = %v, want %v", got.Agent.CompactForceRatio, orig.Agent.CompactForceRatio)
}
if strings.Join(got.Agent.Keep, ",") != strings.Join(orig.Agent.Keep, ",") {
t.Errorf("keep = %v, want %v", got.Agent.Keep, orig.Agent.Keep)
}
if got.Agent.RecentKeep != orig.Agent.RecentKeep {
t.Errorf("recent_keep = %d, want %d", got.Agent.RecentKeep, orig.Agent.RecentKeep)
}
if got.Agent.SystemPrompt != orig.Agent.SystemPrompt {
t.Errorf("system_prompt mismatch:\n got %q\nwant %q", got.Agent.SystemPrompt, orig.Agent.SystemPrompt)
}
if !got.LSP.Enabled {
t.Error("lsp.enabled = false, want true")
}
if got.Environment.Enabled == nil || *got.Environment.Enabled {
t.Errorf("environment.enabled = %+v, want false", got.Environment.Enabled)
}
if !reflect.DeepEqual(got.Environment.Tools, orig.Environment.Tools) {
t.Errorf("environment.tools = %v, want %v", got.Environment.Tools, orig.Environment.Tools)
}
lua := got.LSP.Servers["lua"]
if lua.Command != "lua-language-server" || lua.LanguageID != "lua" || lua.InstallHint != "install lua-language-server" {
t.Errorf("lsp.servers.lua scalar fields not preserved: %+v", lua)
}
if len(lua.Args) != 1 || lua.Args[0] != "--stdio" {
t.Errorf("lsp.servers.lua.args = %v, want [--stdio]", lua.Args)
}
if lua.Env["LUA_PATH"] != "./?.lua" {
t.Errorf("lsp.servers.lua.env = %v, want LUA_PATH", lua.Env)
}
if len(lua.Extensions) != 3 || lua.Extensions[2] != ".gui_script" {
t.Errorf("lsp.servers.lua.extensions = %v", lua.Extensions)
}
if got.Agent.SubagentModel != "mimo-pro" {
t.Errorf("subagent_model = %q, want mimo-pro", got.Agent.SubagentModel)
}
if got.Agent.SubagentModels["review"] != "deepseek-pro" {
t.Errorf("subagent_models.review = %q, want deepseek-pro", got.Agent.SubagentModels["review"])
}
if got.Agent.MaxSubagentDepth != 3 {
t.Errorf("max_subagent_depth = %d, want 3", got.Agent.MaxSubagentDepth)
}
if got.Tools.BashTimeoutSeconds == nil || *got.Tools.BashTimeoutSeconds != 900 {
t.Errorf("tools.bash_timeout_seconds = %v, want 900", got.Tools.BashTimeoutSeconds)
}
if got.Tools.BackgroundJobs.StalledWarningSeconds == nil || *got.Tools.BackgroundJobs.StalledWarningSeconds != 30 {
t.Errorf("tools.background_jobs.stalled_warning_seconds = %v, want 30", got.Tools.BackgroundJobs.StalledWarningSeconds)
}
if got.Tools.Shell.Prefer != "bash" {
t.Errorf("tools.shell.prefer = %q, want bash", got.Tools.Shell.Prefer)
}
if got.Tools.Shell.Path != "/usr/local/bin/bash" {
t.Errorf("tools.shell.path = %q, want /usr/local/bin/bash", got.Tools.Shell.Path)
}
if g, _ := got.Provider("mimo-pro"); g == nil || g.BaseURL != "http://localhost:8000/v1" || g.ChatURL != "http://localhost:8000/v1/chat/completions" || g.ModelsURL != "http://localhost:8000/v1/models" || g.ReasoningProtocol != "openai" {
t.Errorf("mimo-pro endpoint fields not preserved: %+v", g)
}
if g, _ := got.Provider("mimo-pro"); g == nil || g.PresetID != "mimo-api" || g.PresetVersion != ProviderPresetVersion {
t.Errorf("mimo-pro preset metadata not preserved: %+v", g)
}
if g, _ := got.Provider("deepseek-flash"); g == nil || g.Effort != "max" {
t.Errorf("deepseek-flash effort not preserved: %+v", g)
}
if len(got.Providers) != len(orig.Providers) {
t.Errorf("providers count = %d, want %d", len(got.Providers), len(orig.Providers))
}
if got.Permissions.Mode != "deny" {
t.Errorf("permissions.mode = %q, want deny", got.Permissions.Mode)
}
if len(got.Permissions.Deny) != 1 || got.Permissions.Deny[0] != "Bash(rm -rf*)" {
t.Errorf("permissions.deny = %v, want [Bash(rm -rf*)]", got.Permissions.Deny)
}
if len(got.Permissions.Allow) != 2 {
t.Errorf("permissions.allow = %v, want 2 entries", got.Permissions.Allow)
}
if got.Network.ProxyMode != "custom" || got.Network.Proxy.Type != "socks5" || got.Network.Proxy.Port != 7890 {
t.Errorf("network proxy not preserved: %+v", got.Network)
}
if len(got.Skills.Paths) != 2 || got.Skills.Paths[0] != "~/my-skills" {
t.Errorf("skills.paths = %v", got.Skills.Paths)
}
if len(got.Skills.ExcludedPaths) != 1 || got.Skills.ExcludedPaths[0] != "~/.agents/skills" {
t.Errorf("skills.excluded_paths = %v", got.Skills.ExcludedPaths)
}
if len(got.Skills.DisabledSkills) != 2 || got.Skills.DisabledSkills[0] != "review" || got.Skills.DisabledSkills[1] != "explore" {
t.Errorf("skills.disabled_skills = %v", got.Skills.DisabledSkills)
}
if got.SkillMaxDepth() != 2 {
t.Errorf("skills.max_depth = %d, want 2", got.SkillMaxDepth())
}
if len(got.Plugins) != 2 {
t.Fatalf("plugins count = %d, want 2", len(got.Plugins))
}
stripe := got.Plugins[1]
if stripe.Name != "stripe" || stripe.Type != "http" || stripe.URL != "https://mcp.stripe.com" {
t.Errorf("http plugin not preserved: %+v", stripe)
}
if stripe.Headers["Authorization"] != "Bearer x" {
t.Errorf("plugin headers not preserved: %v", stripe.Headers)
}
if len(stripe.TrustedReadOnlyTools) != 1 || stripe.TrustedReadOnlyTools[0] != "customer_read" {
t.Errorf("plugin trusted_read_only_tools not preserved: %+v", stripe.TrustedReadOnlyTools)
}
if stripe.AutoStart == nil || *stripe.AutoStart {
t.Errorf("auto_start should render and parse as false, got %+v", stripe.AutoStart)
}
if stripe.Tier != "" {
t.Errorf("plugin tier should be omitted from new config, got %q", stripe.Tier)
}
if strings.Contains(rendered, "\ntier") {
t.Errorf("rendered config should not contain MCP tier fields:\n%s", rendered)
}
}
func TestRenderTOMLDocumentsPlanModeAllowedTools(t *testing.T) {
cfg := Default()
cfg.Agent.PlanModeAllowedTools = []string{"custom_reader"}
cfg.Agent.PlanModeReadOnlyCommands = []string{"gh issue view"}
rendered := RenderTOML(cfg)
if !strings.Contains(rendered, `plan_mode_allowed_tools = ["custom_reader"]`) {
t.Fatalf("rendered config should preserve plan_mode_allowed_tools:\n%s", rendered)
}
if !strings.Contains(rendered, "extra read-only declarations") || !strings.Contains(rendered, "cannot unlock known blocked tools or unsafe bash") {
t.Fatalf("rendered config should document tightened plan_mode_allowed_tools semantics:\n%s", rendered)
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n%s", err, rendered)
}
if !reflect.DeepEqual(got.Agent.PlanModeAllowedTools, cfg.Agent.PlanModeAllowedTools) {
t.Fatalf("PlanModeAllowedTools round trip = %v, want %v", got.Agent.PlanModeAllowedTools, cfg.Agent.PlanModeAllowedTools)
}
if !strings.Contains(rendered, `plan_mode_read_only_commands = ["gh issue view"]`) {
t.Fatalf("rendered config should preserve plan_mode_read_only_commands:\n%s", rendered)
}
if !strings.Contains(rendered, "concrete read-only shell prefixes") {
t.Fatalf("rendered config should document plan_mode_read_only_commands semantics:\n%s", rendered)
}
if !reflect.DeepEqual(got.Agent.PlanModeReadOnlyCommands, cfg.Agent.PlanModeReadOnlyCommands) {
t.Fatalf("PlanModeReadOnlyCommands round trip = %v, want %v", got.Agent.PlanModeReadOnlyCommands, cfg.Agent.PlanModeReadOnlyCommands)
}
}
func TestRenderTOMLDocumentsPluginTrustedReadOnlyTools(t *testing.T) {
cfg := Default()
cfg.Plugins = []PluginEntry{{
Name: "github",
Command: "github-mcp",
TrustedReadOnlyTools: []string{"issue_read", "pull_request_read"},
}}
rendered := RenderTOML(cfg)
if !strings.Contains(rendered, `trusted_read_only_tools = ["issue_read", "pull_request_read"]`) {
t.Fatalf("rendered config should preserve trusted_read_only_tools:\n%s", rendered)
}
if !strings.Contains(rendered, "optional pre-seeded MCP read-only trust") {
t.Fatalf("rendered config should document trusted_read_only_tools semantics:\n%s", rendered)
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n%s", err, rendered)
}
if !reflect.DeepEqual(got.Plugins[0].TrustedReadOnlyTools, cfg.Plugins[0].TrustedReadOnlyTools) {
t.Fatalf("TrustedReadOnlyTools round trip = %v, want %v", got.Plugins[0].TrustedReadOnlyTools, cfg.Plugins[0].TrustedReadOnlyTools)
}
}
func TestRenderTOMLPreservesMCPCallTimeouts(t *testing.T) {
cfg := Default()
cfg.Tools.MCPCallTimeoutSeconds = intPtr(450)
cfg.Plugins = []PluginEntry{{
Name: "maker",
Command: "maker-mcp",
CallTimeoutSeconds: 600,
ToolTimeoutSeconds: map[string]int{
"generate/video": 1800,
"search": 120,
},
}}
rendered := RenderTOML(cfg)
for _, want := range []string{
"mcp_call_timeout_seconds = 450",
"call_timeout_seconds = 600",
`tool_timeout_seconds = { "generate/video" = 1800, "search" = 120 }`,
"Raw MCP tool names",
} {
if !strings.Contains(rendered, want) {
t.Fatalf("rendered config missing %q:\n%s", want, rendered)
}
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n%s", err, rendered)
}
if got.Tools.MCPCallTimeoutSeconds == nil || *got.Tools.MCPCallTimeoutSeconds != 450 {
t.Fatalf("MCPCallTimeoutSeconds round trip = %v, want 450", got.Tools.MCPCallTimeoutSeconds)
}
if got.Plugins[0].CallTimeoutSeconds != 600 {
t.Fatalf("CallTimeoutSeconds round trip = %d, want 600", got.Plugins[0].CallTimeoutSeconds)
}
if !reflect.DeepEqual(got.Plugins[0].ToolTimeoutSeconds, cfg.Plugins[0].ToolTimeoutSeconds) {
t.Fatalf("ToolTimeoutSeconds round trip = %v, want %v", got.Plugins[0].ToolTimeoutSeconds, cfg.Plugins[0].ToolTimeoutSeconds)
}
}
func TestRenderTOMLCreationLayoutStyle(t *testing.T) {
c := Default()
if err := c.SetDesktopLayoutStyle("creation"); err != nil {
t.Fatalf("SetDesktopLayoutStyle: %v", err)
}
rendered := RenderTOML(c)
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n---\n%s", err, rendered)
}
if got.Desktop.LayoutStyle != "creation" {
t.Errorf("desktop.layout_style = %q, want creation", got.Desktop.LayoutStyle)
}
if got.DesktopLayoutStyle() != "creation" {
t.Errorf("DesktopLayoutStyle() = %q, want creation", got.DesktopLayoutStyle())
}
}
func TestScopedRenderPreservesLSPConfig(t *testing.T) {
const src = `
config_version = 4
default_model = "mimo"
[lsp]
enabled = true
[lsp.servers.lua]
command = "lua-language-server"
args = ["--stdio"]
env = { LUA_PATH = "./?.lua" }
language_id = "lua"
extensions = [".lua", ".script", ".gui_script"]
install_hint = "install lua-language-server"
[lsp.servers."c++"]
command = "clangd"
extensions = [".cc", ".cpp", ".hpp"]
`
var cfg Config
if _, err := toml.Decode(src, &cfg); err != nil {
t.Fatalf("decode source TOML: %v", err)
}
for _, scope := range []RenderScope{RenderScopeFull, RenderScopeUser, RenderScopeProject} {
t.Run(string(scope), func(t *testing.T) {
rendered := RenderTOMLForScope(&cfg, scope)
if !strings.Contains(rendered, "[lsp]") {
t.Fatalf("render missing [lsp]:\n%s", rendered)
}
if !strings.Contains(rendered, "[lsp.servers.lua]") {
t.Fatalf("render missing [lsp.servers.lua]:\n%s", rendered)
}
if !strings.Contains(rendered, `[lsp.servers."c++"]`) {
t.Fatalf("render missing quoted c++ server key:\n%s", rendered)
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("decode rendered TOML: %v\n---\n%s", err, rendered)
}
if !got.LSP.Enabled {
t.Fatalf("lsp.enabled = false, want true")
}
lua, ok := got.LSP.Servers["lua"]
if !ok {
t.Fatalf("lsp.servers.lua missing after round-trip: %+v", got.LSP.Servers)
}
if lua.Command != "lua-language-server" || lua.LanguageID != "lua" || lua.InstallHint != "install lua-language-server" {
t.Fatalf("lsp.servers.lua scalar fields not preserved: %+v", lua)
}
if len(lua.Args) != 1 || lua.Args[0] != "--stdio" {
t.Fatalf("lsp.servers.lua.args = %v, want [--stdio]", lua.Args)
}
if lua.Env["LUA_PATH"] != "./?.lua" {
t.Fatalf("lsp.servers.lua.env = %v, want LUA_PATH", lua.Env)
}
if len(lua.Extensions) != 3 || lua.Extensions[0] != ".lua" || lua.Extensions[2] != ".gui_script" {
t.Fatalf("lsp.servers.lua.extensions = %v", lua.Extensions)
}
cpp, ok := got.LSP.Servers["c++"]
if !ok {
t.Fatalf("lsp.servers.c++ missing after round-trip: %+v", got.LSP.Servers)
}
if cpp.Command != "clangd" || len(cpp.Extensions) != 3 || cpp.Extensions[1] != ".cpp" {
t.Fatalf("lsp.servers.c++ not preserved: %+v", cpp)
}
})
}
}
func BenchmarkRenderTOMLWithLSPServers(b *testing.B) {
cfg := Default()
cfg.LSP.Servers = make(map[string]LSPServer, 64)
for i := 0; i < 64; i++ {
lang := "lang" + strconv.Itoa(i)
cfg.LSP.Servers[lang] = LSPServer{
Command: "server-" + strconv.Itoa(i),
Args: []string{"--stdio", "--flag"},
Env: map[string]string{"SERVER_MODE": "stdio", "SERVER_ROOT": "."},
LanguageID: lang,
Extensions: []string{"." + lang, "." + lang + "x"},
InstallHint: "install server-" + strconv.Itoa(i),
}
}
b.ReportAllocs()
for i := 0; i < b.N; i++ {
rendered := RenderTOML(cfg)
if len(rendered) == 0 {
b.Fatal("empty render")
}
}
}
func TestNotificationsDefaultsKeepEventSwitchesEnabled(t *testing.T) {
cfg := Default()
if cfg.Notifications.Enabled {
t.Fatal("notifications.enabled default = true, want false")
}
if !cfg.Notifications.TurnDone || !cfg.Notifications.ApprovalRequest || !cfg.Notifications.AskRequest {
t.Fatalf("notification event switches default off: %+v", cfg.Notifications)
}
if _, err := toml.Decode("[notifications]\nenabled = true\n", cfg); err != nil {
t.Fatalf("decode notifications: %v", err)
}
if !cfg.Notifications.Enabled || !cfg.Notifications.TurnDone || !cfg.Notifications.ApprovalRequest || !cfg.Notifications.AskRequest {
t.Fatalf("enabled-only config should keep event switches on: %+v", cfg.Notifications)
}
}
func TestScopedRenderSeparatesUserAndProjectConfig(t *testing.T) {
c := Default()
c.Language = "zh"
c.Desktop.Language = "zh"
c.Desktop.Theme = "dark"
c.Desktop.ThemeStyle = "graphite"
c.Desktop.CloseBehavior = "background"
c.Desktop.StatusBarStyle = "text"
c.Desktop.DefaultToolApprovalMode = "auto"
c.Desktop.CheckUpdates = boolPtr(false)
user := RenderTOMLForScope(c, RenderScopeUser)
for _, want := range []string{"config_version = 4", "[desktop]", `theme = "dark"`, `close_behavior = "background"`, `status_bar_style = "text"`, `default_tool_approval_mode = "auto"`, `check_updates = false`, "[notifications]", "[tools.shell]"} {
if !strings.Contains(user, want) {
t.Fatalf("user render missing %q:\n%s", want, user)
}
}
project := RenderTOMLForScope(c, RenderScopeProject)
for _, forbidden := range []string{"[desktop]", "[notifications]", "close_behavior =", "default_tool_approval_mode =", "check_updates =", "max_steps", "planner_max_steps"} {
if strings.Contains(project, forbidden) {
t.Fatalf("project render should not contain %q:\n%s", forbidden, project)
}
}
if strings.Contains(project, "\nsystem_prompt = \"\"\"") {
t.Fatalf("project render should not pin the built-in system prompt:\n%s", project)
}
if !strings.Contains(project, "# system_prompt =") {
t.Fatalf("project render should leave a system prompt hint:\n%s", project)
}
}
func TestProjectDeltaRendersToolsShellOverrides(t *testing.T) {
c := Default()
c.Tools.Shell.Prefer = "bash"
c.Tools.Shell.Path = "/usr/local/bin/bash"
delta := RenderTOMLProjectDelta(c)
for _, want := range []string{"[tools.shell]", `prefer = "bash"`, `path = "/usr/local/bin/bash"`} {
if !strings.Contains(delta, want) {
t.Fatalf("project delta missing %q:\n%s", want, delta)
}
}
if strings.Contains(delta, "[tools]\n\n") {
t.Fatalf("project delta should not emit an empty [tools] block:\n%s", delta)
}
got := Default()
if _, err := toml.Decode(delta, got); err != nil {
t.Fatalf("decode project delta: %v\n%s", err, delta)
}
if got.Tools.Shell.Prefer != "bash" || got.Tools.Shell.Path != "/usr/local/bin/bash" {
t.Fatalf("tools.shell = %+v, want bash with path", got.Tools.Shell)
}
}
func TestProjectDeltaRendersUICursorShape(t *testing.T) {
c := Default()
c.UI.CursorShape = "block"
delta := RenderTOMLProjectDelta(c)
for _, want := range []string{"[ui]", `cursor_shape = "block"`} {
if !strings.Contains(delta, want) {
t.Fatalf("project delta missing %q:\n%s", want, delta)
}
}
got := Default()
if _, err := toml.Decode(delta, got); err != nil {
t.Fatalf("decode project delta: %v\n%s", err, delta)
}
if got.UICursorShape() != "block" {
t.Fatalf("ui.cursor_shape = %q, want block", got.UICursorShape())
}
}
func TestProjectRenderPreservesNonDefaultLegacySections(t *testing.T) {
c := Default()
c.UI.Theme = "light"
c.UI.CloseBehavior = "quit"
c.Network.ProxyMode = "custom"
c.Network.Proxy.Server = "127.0.0.1"
c.Network.Proxy.Port = 7890
project := RenderTOMLForScope(c, RenderScopeProject)
for _, want := range []string{"[ui]", `theme = "light"`, `close_behavior = "quit"`, "[network]", `proxy_mode = "custom"`, `server = "127.0.0.1"`} {
if !strings.Contains(project, want) {
t.Fatalf("project render missing legacy/non-default %q:\n%s", want, project)
}
}
}
func TestRenderTOMLRoundTripsPerModelPrices(t *testing.T) {
orig := Default()
orig.Providers = []ProviderEntry{{
Name: "deepseek",
Kind: "openai",
BaseURL: "https://api.deepseek.com",
Models: []string{"deepseek-v4-flash", "deepseek-v4-pro"},
Default: "deepseek-v4-flash",
APIKeyEnv: "DEEPSEEK_API_KEY",
Prices: deepSeekV4Prices(),
}}
var got Config
if _, err := toml.Decode(RenderTOML(orig), &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v", err)
}
p, ok := got.Provider("deepseek")
if !ok {
t.Fatal("deepseek provider missing after round trip")
}
if p.Prices["deepseek-v4-flash"].Input != 1 || p.Prices["deepseek-v4-pro"].Output != 6 {
t.Fatalf("prices after round trip = %+v", p.Prices)
}
}
func TestRenderTOMLRoundTripsVisionModels(t *testing.T) {
orig := Default()
orig.Providers = []ProviderEntry{
{
Name: "custom",
Kind: "openai",
BaseURL: "https://proxy.example.com/v1",
Models: []string{"text-only", "qwen-vl-plus"},
Default: "text-only",
APIKeyEnv: "CUSTOM_API_KEY",
VisionModels: []string{"qwen-vl-plus"},
VisionDetail: "low",
},
{
Name: "disabled-vision",
Kind: "openai",
BaseURL: "https://proxy.example.com/v1",
Models: []string{"qwen-vl-plus"},
Default: "qwen-vl-plus",
APIKeyEnv: "CUSTOM_API_KEY",
VisionModels: []string{},
},
}
rendered := RenderTOML(orig)
if !strings.Contains(rendered, `vision_models = ["qwen-vl-plus"]`) {
t.Fatalf("rendered TOML missing vision_models:\n%s", rendered)
}
if !strings.Contains(rendered, `vision_models = []`) {
t.Fatalf("rendered TOML missing explicit empty vision_models:\n%s", rendered)
}
if !strings.Contains(rendered, `vision_detail = "low"`) {
t.Fatalf("rendered TOML missing vision_detail:\n%s", rendered)
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v", err)
}
p, ok := got.Provider("custom")
if !ok {
t.Fatal("custom provider missing after round trip")
}
if !reflect.DeepEqual(p.VisionModels, []string{"qwen-vl-plus"}) {
t.Fatalf("vision_models after round trip = %v, want [qwen-vl-plus]", p.VisionModels)
}
if p.VisionDetail != "low" {
t.Fatalf("vision_detail after round trip = %q, want low", p.VisionDetail)
}
disabled, ok := got.Provider("disabled-vision")
if !ok {
t.Fatal("disabled-vision provider missing after round trip")
}
if disabled.VisionModels == nil || len(disabled.VisionModels) != 0 {
t.Fatalf("disabled-vision vision_models after round trip = %#v, want explicit empty list", disabled.VisionModels)
}
}
func TestRenderTOMLRoundTripsProviderHeadersAndModelOverrides(t *testing.T) {
orig := Default()
orig.Providers = []ProviderEntry{{
Name: "gateway",
Kind: "openai",
BaseURL: "https://gateway.example/v1",
Models: []string{"deepseek-v4-flash", "plain-chat"},
Default: "plain-chat",
APIKeyEnv: "GATEWAY_API_KEY",
Headers: map[string]string{
"HTTP-Referer": "https://app.example",
"X-Title": "Reasonix",
},
ExtraBody: map[string]any{
"enable_thinking": true,
"top_p": 0.8,
"metadata": map[string]any{
"mode": "fast",
},
},
AuthHeader: true,
ModelOverrides: map[string]ProviderModelOverride{
"deepseek-v4-flash": {
ReasoningProtocol: ReasoningProtocolDeepSeek,
SupportedEfforts: []string{"high", "max"},
DefaultEffort: "high",
Vision: boolPtr(false),
},
},
}}
rendered := RenderTOML(orig)
if !strings.Contains(rendered, `headers = { HTTP-Referer = "https://app.example", X-Title = "Reasonix" }`) {
t.Fatalf("rendered TOML missing headers:\n%s", rendered)
}
if !strings.Contains(rendered, `extra_body`) || !strings.Contains(rendered, `"enable_thinking" = true`) {
t.Fatalf("rendered TOML missing extra_body:\n%s", rendered)
}
if !strings.Contains(rendered, `auth_header = true`) {
t.Fatalf("rendered TOML missing auth_header:\n%s", rendered)
}
if !strings.Contains(rendered, `model_overrides`) || !strings.Contains(rendered, `reasoning_protocol = "deepseek"`) {
t.Fatalf("rendered TOML missing model overrides:\n%s", rendered)
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n%s", err, rendered)
}
p, ok := got.Provider("gateway")
if !ok {
t.Fatal("gateway provider missing after round trip")
}
if p.Headers["HTTP-Referer"] != "https://app.example" || p.Headers["X-Title"] != "Reasonix" {
t.Fatalf("headers after round trip = %+v", p.Headers)
}
if p.ExtraBody["enable_thinking"] != true || p.ExtraBody["top_p"] != 0.8 {
t.Fatalf("extra_body after round trip = %+v", p.ExtraBody)
}
if !p.AuthHeader {
t.Fatal("auth_header after round trip = false, want true")
}
metadata, ok := p.ExtraBody["metadata"].(map[string]any)
if !ok || metadata["mode"] != "fast" {
t.Fatalf("extra_body metadata after round trip = %+v", p.ExtraBody["metadata"])
}
ov := p.ModelOverrides["deepseek-v4-flash"]
if ov.ReasoningProtocol != ReasoningProtocolDeepSeek || !reflect.DeepEqual(ov.SupportedEfforts, []string{"high", "max"}) || ov.DefaultEffort != "high" || ov.Vision == nil || *ov.Vision {
t.Fatalf("model override after round trip = %+v", ov)
}
}
func TestRenderStringMapQuotesNonBareTOMLKeys(t *testing.T) {
rendered := renderStringMap(map[string]string{
"github:gh-fix-ci": "deepseek-pro",
"review": "deepseek-flash",
})
if !strings.Contains(rendered, `"github:gh-fix-ci" = "deepseek-pro"`) {
t.Fatalf("non-bare key was not quoted: %s", rendered)
}
var got struct {
M map[string]string `toml:"m"`
}
if _, err := toml.Decode("m = "+rendered, &got); err != nil {
t.Fatalf("rendered inline map does not parse: %v (%s)", err, rendered)
}
if got.M["github:gh-fix-ci"] != "deepseek-pro" || got.M["review"] != "deepseek-flash" {
t.Fatalf("decoded map = %+v", got.M)
}
}
func TestRenderTOMLTablePathQuotesEachSegment(t *testing.T) {
got := renderTOMLTablePath("lsp", "servers", "c++", "github:gh-fix-ci")
want := `lsp.servers."c++"."github:gh-fix-ci"`
if got != want {
t.Fatalf("renderTOMLTablePath = %q, want %q", got, want)
}
}
func boolPtr(v bool) *bool { return &v }
func intPtr(v int) *int { return &v }
func TestRenderTOMLPreservesDesktopDisplayMode(t *testing.T) {
c := Default()
if err := c.SetDesktopDisplayMode("compact"); err != nil {
t.Fatalf("SetDesktopDisplayMode: %v", err)
}
rendered := RenderTOMLForScope(c, RenderScopeUser)
if !strings.Contains(rendered, `display_mode = "compact"`) {
t.Fatalf("rendered user config missing display_mode:\n%s", rendered)
}
var got Config
if _, err := toml.Decode(rendered, &got); err != nil {
t.Fatalf("rendered TOML does not parse: %v\n---\n%s", err, rendered)
}
if got.DesktopDisplayMode() != "compact" {
t.Fatalf("display_mode after round trip = %q, want compact", got.DesktopDisplayMode())
}
}
func TestRenderTOMLDefaultStepsCommentedOut(t *testing.T) {
isolateUserConfigHome(t)
out := RenderTOML(Default())
agentLines := extractSectionLines(out, "[agent]")
for _, line := range agentLines {
if strings.HasPrefix(line, "max_steps ") || strings.HasPrefix(line, "max_steps=") {
if !strings.HasPrefix(line, "#") {
t.Errorf("default max_steps should be commented out in [agent], got: %s", line)
}
}
if strings.HasPrefix(line, "planner_max_steps ") || strings.HasPrefix(line, "planner_max_steps=") {
if !strings.HasPrefix(line, "#") {
t.Errorf("default planner_max_steps should be commented out in [agent], got: %s", line)
}
}
}
}
func TestRenderTOMLWindowsSandboxDefaultAndExplicitEnforceDisabled(t *testing.T) {
isolateUserConfigHome(t)
setRuntimeGOOS(t, "windows")
defaultRendered := RenderTOMLForScope(Default(), RenderScopeUser)
if !strings.Contains(defaultRendered, `bash = "off"`) {
t.Fatalf("Windows default user config should render bash off:\n%s", defaultRendered)
}
cfg := Default()
cfg.Sandbox.Bash = "enforce"
delta := RenderTOMLProjectDelta(cfg)
if strings.Contains(delta, `[sandbox]`) || strings.Contains(delta, `bash = `) {
t.Fatalf("Windows explicit enforce should not render as an effective project delta:\n%s", delta)
}
}
func extractSectionLines(toml, section string) []string {
var lines []string
inSection := false
for _, line := range strings.Split(toml, "\n") {
trimmed := strings.TrimSpace(line)
if strings.HasPrefix(trimmed, section) {
inSection = true
continue
}
if inSection && strings.HasPrefix(trimmed, "[") && !strings.HasPrefix(trimmed, "[[") {
break
}
if inSection {
lines = append(lines, trimmed)
}
}
return lines
}
func TestRenderTOMLNonDefaultStepsWrittenExplicitly(t *testing.T) {
isolateUserConfigHome(t)
c := Default()
c.Agent.MaxSteps = 5
c.Agent.PlannerMaxSteps = 7
out := RenderTOML(c)
agentLines := extractSectionLines(out, "[agent]")
foundMax, foundPlanner := false, false
for _, line := range agentLines {
if !strings.HasPrefix(line, "#") && strings.HasPrefix(line, "max_steps ") {
foundMax = true
}
if !strings.HasPrefix(line, "#") && strings.HasPrefix(line, "planner_max_steps ") {
foundPlanner = true
}
}
if !foundMax {
t.Error("non-default max_steps should be written explicitly in [agent]")
}
if !foundPlanner {
t.Error("non-default planner_max_steps should be written explicitly in [agent]")
}
}
func TestRenderTOMLDefaultStepsDoNotOverrideGlobalConfig(t *testing.T) {
isolateUserConfigHome(t)
globalDir := filepath.Dir(UserConfigPath())
if err := os.MkdirAll(globalDir, 0o755); err != nil {
t.Fatal(err)
}
globalPath := filepath.Join(globalDir, "config.toml")
if err := os.WriteFile(globalPath, []byte("[agent]\nplanner_max_steps = 9\nmax_steps = 100\n"), 0o644); err != nil {
t.Fatal(err)
}
projectDir := t.TempDir()
projectTOML := RenderTOML(Default())
projectPath := filepath.Join(projectDir, "reasonix.toml")
if err := os.WriteFile(projectPath, []byte(projectTOML), 0o644); err != nil {
t.Fatal(err)
}
cfg := Default()
if err := mergeFile(cfg, globalPath); err != nil {
t.Fatalf("global merge failed: %v", err)
}
if cfg.Agent.PlannerMaxSteps != 9 {
t.Fatalf("after global: planner_max_steps = %d, want 9", cfg.Agent.PlannerMaxSteps)
}
if cfg.Agent.MaxSteps != 100 {
t.Fatalf("after global: max_steps = %d, want 100", cfg.Agent.MaxSteps)
}
if err := mergeFile(cfg, projectPath); err != nil {
t.Fatalf("project merge failed: %v", err)
}
if cfg.Agent.PlannerMaxSteps != 9 {
t.Errorf("after project: planner_max_steps = %d, want 9 (global should not be overridden by commented-out default)", cfg.Agent.PlannerMaxSteps)
}
if cfg.Agent.MaxSteps != 100 {
t.Errorf("after project: max_steps = %d, want 100 (global should not be overridden by commented-out default)", cfg.Agent.MaxSteps)
}
}
func TestIsolatedHomeDirEmptyByDefault(t *testing.T) {
t.Setenv("REASONIX_HOME", "")
if got := IsolatedHomeDir(); got != "" {
t.Fatalf("IsolatedHomeDir() = %q, want empty", got)
}
}
func TestIsolatedHomeDirReturnsCleanPath(t *testing.T) {
raw := filepath.Join(t.TempDir(), "isolated-reasonix")
t.Setenv("REASONIX_HOME", raw)
got := IsolatedHomeDir()
if filepath.Clean(got) != filepath.Clean(raw) {
t.Fatalf("IsolatedHomeDir() = %q, want %q", got, raw)
}
}
func TestLegacyOSSupportDirEmptyWhenIsolated(t *testing.T) {
isolateUserConfigHome(t)
t.Setenv("REASONIX_HOME", filepath.Join(t.TempDir(), "isolated-home"))
if got := legacyOSSupportDir(); got != "" {
t.Fatalf("legacyOSSupportDir() = %q, want empty when isolated", got)
}
}
func TestLegacyXDGConfigPathsEmptyWhenIsolated(t *testing.T) {
isolateUserConfigHome(t)
t.Setenv("REASONIX_HOME", filepath.Join(t.TempDir(), "isolated-home"))
if got := legacyXDGConfigPaths(); got != nil {
t.Fatalf("legacyXDGConfigPaths() = %v, want nil when isolated", got)
}
}
func TestCacheDirHonorsReasonixHome(t *testing.T) {
home := t.TempDir()
isolated := filepath.Join(home, "isolated-home")
t.Setenv("HOME", home)
t.Setenv("USERPROFILE", home)
t.Setenv("REASONIX_HOME", isolated)
got := CacheDir()
want := filepath.Join(isolated, "cache")
if filepath.Clean(got) != filepath.Clean(want) {
t.Fatalf("CacheDir() = %q, want %q", got, want)
}
}
func TestCacheDirHonorsReasonixCacheHomeOverReasonixHome(t *testing.T) {
home := t.TempDir()
cacheHome := filepath.Join(home, "custom-cache")
t.Setenv("HOME", home)
t.Setenv("USERPROFILE", home)
t.Setenv("REASONIX_HOME", filepath.Join(home, "isolated-home"))
t.Setenv("REASONIX_CACHE_HOME", cacheHome)
got := CacheDir()
want := cacheHome
if filepath.Clean(got) != filepath.Clean(want) {
t.Fatalf("CacheDir() = %q, want %q (REASONIX_CACHE_HOME must win)", got, want)
}
}
func TestUserConfigLoadPathNoLegacyFallbackWhenIsolated(t *testing.T) {
home := isolateUserConfigHome(t)
isolated := filepath.Join(home, "isolated-home")
t.Setenv("REASONIX_HOME", isolated)
// Create a legacy config at the OS production path — it must not be loaded.
productionHome := expectedDefaultReasonixHome(home)
if err := os.MkdirAll(productionHome, 0o755); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(filepath.Join(productionHome, "config.toml"), []byte("default_model = \"production/model\"\n"), 0o600); err != nil {
t.Fatal(err)
}
// The primary config under isolated home does not exist yet.
got := userConfigLoadPath()
want := filepath.Join(isolated, "config.toml")
if filepath.Clean(got) != filepath.Clean(want) {
t.Fatalf("userConfigLoadPath() = %q, want %q (must not fall back to production legacy config)", got, want)
}
}
func TestCredentialSourceCandidatesSkipHomeEnvWhenIsolated(t *testing.T) {
isolateUserConfigHome(t)
t.Setenv("REASONIX_HOME", filepath.Join(t.TempDir(), "isolated-home"))
// Write a key into the production home .env — it must not appear as a source.
if home, err := os.UserHomeDir(); err == nil {
if err := os.WriteFile(filepath.Join(home, ".env"), []byte("LEAKED_KEY=leaked-value\n"), 0o600); err != nil {
t.Fatal(err)
}
}
candidates := credentialSourceCandidates(".")
for _, c := range candidates {
if c.Kind == CredentialSourceHomeEnv {
t.Fatalf("credentialSourceCandidates includes CredentialSourceHomeEnv when isolated: %v", c)
}
}
}
func TestMigrateLegacyIfNeededSkipsWhenIsolated(t *testing.T) {
home := isolateUserConfigHome(t)
isolated := filepath.Join(home, "isolated-home")
t.Setenv("REASONIX_HOME", isolated)
// Create a legacy config.json in production home — migration must skip it.
legacyDir := filepath.Join(home, ".reasonix")
if err := os.MkdirAll(legacyDir, 0o755); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(filepath.Join(legacyDir, "config.json"), []byte(`{"model":"production-model","apiKey":"sk-legacy"}`), 0o600); err != nil {
t.Fatal(err)
}
res, err := MigrateLegacyIfNeeded()
if err != nil {
t.Fatalf("MigrateLegacyIfNeeded() error = %v", err)
}
if res != nil {
t.Fatalf("MigrateLegacyIfNeeded() = %+v, want nil when isolated", res)
}
}
// TestProjectConfigCannotOverrideSecrets pins [secrets] as a user-global
// security control: a cloned repository's reasonix.toml must not be able to
// switch off tool-output redaction or opt the user into subprocess env
// stripping / sensitive-path hiding.
func TestProjectConfigCannotOverrideSecrets(t *testing.T) {
isolateUserConfigHome(t)
t.Setenv("REASONIX_HOME", "")
globalDir := filepath.Dir(UserConfigPath())
if err := os.MkdirAll(globalDir, 0o755); err != nil {
t.Fatal(err)
}
globalTOML := "[secrets]\nredact_tool_output = true\nfilter_subprocess_env = false\nprotect_sensitive_files = false\n"
if err := os.WriteFile(filepath.Join(globalDir, "config.toml"), []byte(globalTOML), 0o644); err != nil {
t.Fatal(err)
}
project := t.TempDir()
projectTOML := "[secrets]\nredact_tool_output = false\nfilter_subprocess_env = true\nprotect_sensitive_files = true\n"
if err := os.WriteFile(filepath.Join(project, "reasonix.toml"), []byte(projectTOML), 0o644); err != nil {
t.Fatal(err)
}
cfg, err := LoadForRoot(project)
if err != nil {
t.Fatalf("LoadForRoot() error = %v", err)
}
if !cfg.SecretsRedactToolOutput() {
t.Error("project reasonix.toml disabled redact_tool_output; [secrets] must stay user-global")
}
if cfg.Secrets.FilterSubprocessEnv {
t.Error("project reasonix.toml enabled filter_subprocess_env; [secrets] must stay user-global")
}
if cfg.Secrets.ProtectSensitiveFiles {
t.Error("project reasonix.toml enabled protect_sensitive_files; [secrets] must stay user-global")
}
}
// TestRenderTOMLPersistsSecretsSection pins config-save round-tripping: the
// renderer must emit [secrets] for the user scope or every WriteFile would
// silently drop the user's security toggles.
func TestRenderTOMLPersistsSecretsSection(t *testing.T) {
cfg := Default()
off := false
cfg.Secrets.RedactToolOutput = &off
cfg.Secrets.FilterSubprocessEnv = true
cfg.Secrets.ProtectSensitiveFiles = true
out := RenderTOMLForScope(cfg, RenderScopeUser)
for _, want := range []string{"[secrets]", "redact_tool_output = false", "filter_subprocess_env = true", "protect_sensitive_files = true"} {
if !strings.Contains(out, want) {
t.Fatalf("user-scope render missing %q:\n%s", want, out)
}
}
path := filepath.Join(t.TempDir(), "config.toml")
if err := os.WriteFile(path, []byte(out), 0o644); err != nil {
t.Fatal(err)
}
back := Default()
if err := mergeFile(back, path); err != nil {
t.Fatalf("round-trip decode: %v", err)
}
if back.SecretsRedactToolOutput() {
t.Fatal("redact_tool_output=false lost in render round-trip")
}
if !back.Secrets.FilterSubprocessEnv || !back.Secrets.ProtectSensitiveFiles {
t.Fatalf("secrets toggles lost in render round-trip: %+v", back.Secrets)
}
// Project scope must not render the section — LoadForRoot ignores it there.
if proj := RenderTOMLForScope(cfg, RenderScopeProject); strings.Contains(proj, "[secrets]") {
t.Fatalf("project scope rendered [secrets]:\n%s", proj)
}
}