| 1 | package config |
| 2 | |
| 3 | import ( |
| 4 | "os" |
| 5 | "path/filepath" |
| 6 | "strings" |
| 7 | "testing" |
| 8 | ) |
| 9 | |
| 10 | func TestFrozenModelCredentialsIncludeMissingValue(t *testing.T) { |
| 11 | t.Setenv("REASONIX_HOME", t.TempDir()) |
| 12 | c := Default() |
| 13 | c.Providers = []ProviderEntry{{Name: "test", APIKeyEnv: "MODEL_SNAPSHOT_KEY", Model: "m"}} |
| 14 | c.FreezeProviderCredentials() |
| 15 | before := c.ModelRuntimeFingerprint("test/m") |
| 16 | if err := os.WriteFile(UserCredentialsPath(), []byte("MODEL_SNAPSHOT_KEY=later-key\n"), 0600); err != nil { |
| 17 | t.Fatal(err) |
| 18 | } |
| 19 | entry, _ := c.ResolveModel("test/m") |
| 20 | if entry.APIKey() != "" { |
| 21 | t.Fatal("frozen absent credential read a newer value") |
| 22 | } |
| 23 | if before != c.ModelRuntimeFingerprint("test/m") { |
| 24 | t.Fatal("frozen fingerprint changed") |
| 25 | } |
| 26 | next := Default() |
| 27 | next.Providers = []ProviderEntry{{Name: "test", APIKeyEnv: "MODEL_SNAPSHOT_KEY", Model: "m"}} |
| 28 | next.FreezeProviderCredentials() |
| 29 | if next.ModelRuntimeFingerprint("test/m") == before { |
| 30 | t.Fatal("new snapshot did not observe rotation") |
| 31 | } |
| 32 | } |
| 33 | |
| 34 | func TestModelRuntimeFingerprintIgnoresNewSessionDefault(t *testing.T) { |
| 35 | c := Default() |
| 36 | c.FreezeProviderCredentials() |
| 37 | before := c.ModelRuntimeFingerprint("deepseek/a") |
| 38 | c.DefaultModel = "other/model" |
| 39 | if c.ModelRuntimeFingerprint("deepseek/a") != before { |
| 40 | t.Fatal("global default invalidated an existing session") |
| 41 | } |
| 42 | c.Agent.PlannerModel = "other/model" |
| 43 | if c.ModelRuntimeFingerprint("deepseek/a") == before { |
| 44 | t.Fatal("planner change not detected") |
| 45 | } |
| 46 | } |
| 47 | |
| 48 | func TestSaveModelSettingsPreservesUnknownFields(t *testing.T) { |
| 49 | t.Setenv("REASONIX_HOME", t.TempDir()) |
| 50 | path := UserConfigPath() |
| 51 | if err := os.MkdirAll(filepath.Dir(path), 0700); err != nil { |
| 52 | t.Fatal(err) |
| 53 | } |
| 54 | raw := `config_version = 5 |
| 55 | future_root = "keep" |
| 56 | [agent] |
| 57 | future_agent = "keep-agent" |
| 58 | planner_model = "" |
| 59 | [[providers]] |
| 60 | name = "custom" |
| 61 | kind = "openai" |
| 62 | base_url = "http://localhost:1234/v1" |
| 63 | model = "m" |
| 64 | future_provider = "keep-provider" |
| 65 | [providers.future_nested] |
| 66 | value = "keep-nested" |
| 67 | ` |
| 68 | if err := os.WriteFile(path, []byte(raw), 0600); err != nil { |
| 69 | t.Fatal(err) |
| 70 | } |
| 71 | unlock := LockUserConfigEdits() |
| 72 | defer unlock() |
| 73 | c, err := LoadForEditReadOnlyStrict(path) |
| 74 | if err != nil { |
| 75 | t.Fatal(err) |
| 76 | } |
| 77 | baseline := c.ModelSettingsBaseline() |
| 78 | c.Agent.PlannerModel = "custom/m" |
| 79 | for i := range c.Providers { |
| 80 | if c.Providers[i].Name == "custom" { |
| 81 | c.Providers[i].BaseURL = "http://localhost:5678/v1" |
| 82 | } |
| 83 | } |
| 84 | if err := c.SaveModelSettingsTo(path, baseline); err != nil { |
| 85 | t.Fatal(err) |
| 86 | } |
| 87 | got, err := os.ReadFile(path) |
| 88 | if err != nil { |
| 89 | t.Fatal(err) |
| 90 | } |
| 91 | for _, want := range []string{"keep", "keep-agent", "keep-provider", "keep-nested", "5678"} { |
| 92 | if !strings.Contains(string(got), want) { |
| 93 | t.Fatalf("lost %s", want) |
| 94 | } |
| 95 | } |
| 96 | } |
| 97 |