chat_timestamps_test.go RawHistoryFolder

1package agent
2
3import "testing"
4
5func TestChatTimestampSettingsPersistence(t *testing.T) {
6	t.Setenv("ZOT_HOME", t.TempDir())
7	if err := SaveConfig(Config{Theme: "dark"}); err != nil {
8		t.Fatal(err)
9	}
10	cfg, err := LoadConfig()
11	if err != nil || cfg.ChatTimestamps || cfg.ChatTimestampIntervalMinutes != nil {
12		t.Fatalf("unexpected defaults: %+v, %v", cfg, err)
13	}
14	store := configSettingsStore{}
15	for _, mode := range []string{"every_message", "day_start", "", "invalid"} {
16		if err := store.SetChatTimestampDate(mode); err != nil {
17			t.Fatal(err)
18		}
19		cfg, err := LoadConfig()
20		want := "day_start"
21		if mode == "every_message" {
22			want = mode
23		}
24		if err != nil || cfg.ChatTimestampDate != want || cfg.Theme != "dark" {
25			t.Fatalf("date mode not persisted: %+v, %v", cfg, err)
26		}
27	}
28	for _, enabled := range []bool{true, false} {
29		if err := store.SetChatTimestamps(enabled); err != nil {
30			t.Fatal(err)
31		}
32		for _, minutes := range []int{0, 1, 5, 7, -1} {
33			if err := store.SetChatTimestampInterval(minutes); err != nil {
34				t.Fatal(err)
35			}
36			cfg, err := LoadConfig()
37			if err != nil {
38				t.Fatal(err)
39			}
40			want := minutes
41			if want < 0 {
42				want = 0
43			}
44			if cfg.ChatTimestamps != enabled || cfg.ChatTimestampIntervalMinutes == nil || *cfg.ChatTimestampIntervalMinutes != want || cfg.Theme != "dark" {
45				t.Fatalf("settings not preserved: %+v", cfg)
46			}
47		}
48	}
49}