package store import ( "context" "encoding/json" "reflect" "strings" "testing" "time" "github.com/jmoiron/sqlx" _ "github.com/mattn/go-sqlite3" "github.com/kandev/kandev/internal/testutil" "github.com/kandev/kandev/internal/user/models" ) type settingsScanner struct { raw string } func upsertUserSettingsForTest(t *testing.T, repo *sqliteRepository, ctx context.Context, settings *models.UserSettings) { t.Helper() var patch *models.TaskCreateLastUsed if settings.TaskCreateLastUsed != (models.TaskCreateLastUsed{}) { patch = &settings.TaskCreateLastUsed } if _, err := repo.UpsertUserSettingsPreservingTaskCreateLastUsed(ctx, settings, patch); err != nil { t.Fatalf("upsert settings: %v", err) } } func (s settingsScanner) Scan(dest ...any) error { *(dest[0].(*string)) = s.raw *(dest[1].(*time.Time)) = time.Time{} return nil } func TestScanUserSettingsStartupPage(t *testing.T) { tests := []struct { name string raw string want string }{ {name: "empty settings default to task overview", raw: "{}", want: "task_overview"}, {name: "missing setting defaults to task overview", raw: `{"chat_submit_key":"cmd_enter"}`, want: "task_overview"}, {name: "unknown setting defaults to task overview", raw: `{"startup_page":"future_value"}`, want: "task_overview"}, {name: "last task is preserved", raw: `{"startup_page":"last_task"}`, want: "last_task"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } encoded, err := marshalUserSettingsPayload(settings) if err != nil { t.Fatalf("marshal settings payload: %v", err) } var payload map[string]any if err := json.Unmarshal(encoded, &payload); err != nil { t.Fatalf("decode normalized settings: %v", err) } if got := payload["startup_page"]; got != tt.want { t.Fatalf("startup_page = %#v, want %q", got, tt.want) } }) } } func TestScanUserSettingsSidebarDefaults(t *testing.T) { defaultView := models.SidebarView{ ID: "view-all-tasks", Name: "All tasks", Filters: []models.SidebarViewClause{}, Sort: models.SidebarViewSort{Key: "state", Direction: "asc"}, Group: "repository", CollapsedGroups: []string{}, } tests := []struct { name string raw string wantDefaults bool wantViewCount int }{ {name: "empty settings use canonical sidebar default", raw: "{}", wantDefaults: true, wantViewCount: 1}, {name: "unrelated settings retain canonical sidebar default", raw: `{"workspace_id":"workspace-1"}`, wantDefaults: true, wantViewCount: 1}, {name: "explicit sidebar settings are preserved", raw: `{"sidebar_views":[{"id":"custom","name":"Custom"}],"sidebar_active_view_id":"custom"}`, wantDefaults: false, wantViewCount: 1}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if len(settings.SidebarViews) != tt.wantViewCount { t.Fatalf("sidebar view count = %d, want %d", len(settings.SidebarViews), tt.wantViewCount) } if tt.wantDefaults { if !reflect.DeepEqual(settings.SidebarViews[0], defaultView) { t.Fatalf("sidebar default = %+v, want %+v", settings.SidebarViews[0], defaultView) } if settings.SidebarActiveViewID != defaultView.ID { t.Fatalf("active sidebar view = %q, want %q", settings.SidebarActiveViewID, defaultView.ID) } return } if settings.SidebarViews[0].ID != "custom" || settings.SidebarActiveViewID != "custom" { t.Fatalf("explicit sidebar settings were not preserved: views=%+v active=%q", settings.SidebarViews, settings.SidebarActiveViewID) } }) } } func TestScanUserSettingsPreservesExplicitEmptySidebarSettings(t *testing.T) { for _, raw := range []string{ `{"sidebar_views":[],"sidebar_active_view_id":""}`, `{"sidebar_views":null,"sidebar_active_view_id":null}`, } { t.Run(raw, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if len(settings.SidebarViews) != 0 { t.Fatalf("sidebar views = %+v, want an explicit empty list", settings.SidebarViews) } if settings.SidebarActiveViewID != "" { t.Fatalf("active sidebar view = %q, want an explicit empty ID", settings.SidebarActiveViewID) } }) } } func TestScanUserSettingsChangesPanelLayoutDefault(t *testing.T) { t.Run("empty settings default to tree", func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: "{}"}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if settings.ChangesPanelLayout != "tree" { t.Fatalf("expected ChangesPanelLayout=tree, got %q", settings.ChangesPanelLayout) } }) t.Run("missing layout defaults to tree", func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: `{"chat_submit_key":"cmd_enter"}`}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if settings.ChangesPanelLayout != "tree" { t.Fatalf("expected ChangesPanelLayout=tree, got %q", settings.ChangesPanelLayout) } }) t.Run("explicit flat is preserved", func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: `{"changes_panel_layout":"flat"}`}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if settings.ChangesPanelLayout != "flat" { t.Fatalf("expected ChangesPanelLayout=flat, got %q", settings.ChangesPanelLayout) } }) } func TestScanUserSettingsConfirmTaskArchiveDefault(t *testing.T) { tests := []struct { name string raw string want bool }{ {name: "empty settings require confirmation", raw: `{}`, want: true}, {name: "missing setting requires confirmation", raw: `{"chat_submit_key":"enter"}`, want: true}, {name: "explicit false skips confirmation", raw: `{"confirm_task_archive":false}`, want: false}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if settings.ConfirmTaskArchive != tt.want { t.Fatalf("ConfirmTaskArchive = %v, want %v", settings.ConfirmTaskArchive, tt.want) } }) } } func TestScanUserSettingsUnreadDividerDefault(t *testing.T) { tests := []struct { name string raw string want bool }{ {name: "empty settings disable the divider", raw: `{}`, want: false}, {name: "missing setting disables the divider", raw: `{"chat_submit_key":"enter"}`, want: false}, {name: "explicit false disables the divider", raw: `{"unread_divider":false}`, want: false}, {name: "explicit true enables the divider", raw: `{"unread_divider":true}`, want: true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if settings.UnreadDivider != tt.want { t.Fatalf("UnreadDivider = %v, want %v", settings.UnreadDivider, tt.want) } }) } } func TestScanUserSettingsAgentGeneratedTaskTitlesDefault(t *testing.T) { tests := []struct { name string raw string want bool }{ {name: "empty settings enable title generation", raw: `{}`, want: true}, {name: "missing setting enables title generation", raw: `{"chat_submit_key":"enter"}`, want: true}, {name: "explicit false disables title generation", raw: `{"agent_generated_task_titles":false}`, want: false}, {name: "explicit true enables title generation", raw: `{"agent_generated_task_titles":true}`, want: true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if settings.AgentGeneratedTaskTitles != tt.want { t.Fatalf("AgentGeneratedTaskTitles = %v, want %v", settings.AgentGeneratedTaskTitles, tt.want) } }) } } func TestScanUserSettingsMCPTaskAgentProfileDefault(t *testing.T) { tests := []struct { name string raw string want string }{ {name: "empty settings use current task", raw: `{}`, want: "current_task"}, {name: "missing setting uses current task", raw: `{"chat_submit_key":"enter"}`, want: "current_task"}, {name: "unknown setting uses current task", raw: `{"mcp_task_agent_profile_default":"future_value"}`, want: "current_task"}, {name: "workspace default is preserved", raw: `{"mcp_task_agent_profile_default":"workspace_default"}`, want: "workspace_default"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } raw, err := json.Marshal(settings) if err != nil { t.Fatalf("marshal normalized settings: %v", err) } var payload map[string]any if err := json.Unmarshal(raw, &payload); err != nil { t.Fatalf("decode normalized settings: %v", err) } if got := payload["mcp_task_agent_profile_default"]; got != tt.want { t.Fatalf("mcp_task_agent_profile_default = %#v, want %q", got, tt.want) } }) } } func TestScanUserSettingsShowAnchoredPromptBarDefault(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: "{}"}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if settings.ShowAnchoredPromptBar { t.Fatal("ShowAnchoredPromptBar = true, want false (default)") } settings, err = scanUserSettings(settingsScanner{raw: `{"show_anchored_prompt_bar":false}`}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if settings.ShowAnchoredPromptBar { t.Fatal("ShowAnchoredPromptBar = true, want false (stored)") } } func TestScanUserSettingsTranscriptNavigationDefaults(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: "{}"}, DefaultUserID) if err != nil { t.Fatalf("scan defaults: %v", err) } if !settings.ShowScrollToLastPrompt || settings.ShowScrollToStart { t.Fatalf( "transcript controls = (%t, %t), want (true, false)", settings.ShowScrollToLastPrompt, settings.ShowScrollToStart, ) } settings, err = scanUserSettings( settingsScanner{raw: `{"show_scroll_to_last_prompt":false,"show_scroll_to_start":false}`}, DefaultUserID, ) if err != nil { t.Fatalf("scan stored preferences: %v", err) } if settings.ShowScrollToLastPrompt || settings.ShowScrollToStart { t.Fatalf( "transcript controls = (%t, %t), want (false, false)", settings.ShowScrollToLastPrompt, settings.ShowScrollToStart, ) } if settings.ShowTranscriptAutoScrollControl { t.Fatal("ShowTranscriptAutoScrollControl = true, want false (default)") } settings, err = scanUserSettings( settingsScanner{raw: `{"show_transcript_auto_scroll_control":true}`}, DefaultUserID, ) if err != nil { t.Fatalf("scan stored auto-scroll-control preference: %v", err) } if !settings.ShowTranscriptAutoScrollControl { t.Fatal("ShowTranscriptAutoScrollControl = false, want true (stored)") } } func TestScanUserSettingsDefaultsTranscriptAutoScrollControlToHidden(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: "{}"}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if settings.ShowTranscriptAutoScrollControl { t.Fatal("ShowTranscriptAutoScrollControl = true, want false (default)") } } func TestTranscriptNavigationSettingsRoundTripThroughMarshalAndScan(t *testing.T) { raw, err := marshalUserSettingsPayload(&models.UserSettings{ ShowScrollToLastPrompt: false, ShowScrollToStart: false, ShowTranscriptAutoScrollControl: false, }) if err != nil { t.Fatalf("marshal settings: %v", err) } settings, err := scanUserSettings(settingsScanner{raw: string(raw)}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if settings.ShowScrollToLastPrompt || settings.ShowScrollToStart || settings.ShowTranscriptAutoScrollControl { t.Fatalf( "transcript controls = (%t, %t, %t), want (false, false, false)", settings.ShowScrollToLastPrompt, settings.ShowScrollToStart, settings.ShowTranscriptAutoScrollControl, ) } } func TestMarshalUserSettingsPersistsDisabledArchiveConfirmation(t *testing.T) { raw, err := marshalUserSettingsPayload(&models.UserSettings{ConfirmTaskArchive: false}) if err != nil { t.Fatalf("marshal settings: %v", err) } var payload map[string]any if err := json.Unmarshal(raw, &payload); err != nil { t.Fatalf("decode settings: %v", err) } if got, ok := payload["confirm_task_archive"].(bool); !ok || got { t.Fatalf("confirm_task_archive = %#v, want false", payload["confirm_task_archive"]) } } func TestShowAnchoredPromptBarRoundTripsThroughMarshalAndScan(t *testing.T) { raw, err := marshalUserSettingsPayload(&models.UserSettings{ShowAnchoredPromptBar: true}) if err != nil { t.Fatalf("marshal settings: %v", err) } settings, err := scanUserSettings(settingsScanner{raw: string(raw)}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if !settings.ShowAnchoredPromptBar { t.Fatal("ShowAnchoredPromptBar = false after round trip, want true") } } func TestMarshalUserSettingsPersistsTasksListShowDetails(t *testing.T) { raw, err := marshalUserSettingsPayload(&models.UserSettings{TasksListShowDetails: true}) if err != nil { t.Fatalf("marshal settings: %v", err) } var payload map[string]any if err := json.Unmarshal(raw, &payload); err != nil { t.Fatalf("decode settings: %v", err) } if got, ok := payload["tasks_list_show_details"].(bool); !ok || !got { t.Fatalf("tasks_list_show_details = %#v, want true", payload["tasks_list_show_details"]) } } func TestScanUserSettingsTasksListShowDetailsDefaultsAndLoads(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: "{}"}, DefaultUserID) if err != nil { t.Fatalf("scan defaults: %v", err) } if settings.TasksListShowDetails { t.Fatal("TasksListShowDetails = true, want false") } settings, err = scanUserSettings(settingsScanner{raw: `{"tasks_list_show_details":true}`}, DefaultUserID) if err != nil { t.Fatalf("scan stored settings: %v", err) } if !settings.TasksListShowDetails { t.Fatal("TasksListShowDetails = false, want true") } } func TestMarshalUserSettingsPersistsMCPTaskAgentProfileDefault(t *testing.T) { raw, err := marshalUserSettingsPayload(&models.UserSettings{ MCPTaskAgentProfileDefault: models.MCPTaskAgentProfileDefaultWorkspaceDefault, }) if err != nil { t.Fatalf("marshal settings: %v", err) } var payload map[string]any if err := json.Unmarshal(raw, &payload); err != nil { t.Fatalf("decode settings: %v", err) } if got := payload["mcp_task_agent_profile_default"]; got != models.MCPTaskAgentProfileDefaultWorkspaceDefault { t.Fatalf("mcp_task_agent_profile_default = %#v, want workspace_default", got) } } func TestSQLiteRepositoryMCPTaskAgentProfileDefaultRoundTrip(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } settings.MCPTaskAgentProfileDefault = models.MCPTaskAgentProfileDefaultWorkspaceDefault upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get saved settings: %v", err) } if got.MCPTaskAgentProfileDefault != models.MCPTaskAgentProfileDefaultWorkspaceDefault { t.Fatalf("MCPTaskAgentProfileDefault = %q, want workspace_default", got.MCPTaskAgentProfileDefault) } } func TestScanUserSettingsLspStatusLocationDefaultsAndLoads(t *testing.T) { tests := []struct { name string raw string want string }{ {name: "empty settings use toolbar", raw: `{}`, want: models.LspStatusLocationToolbar}, {name: "unknown setting uses toolbar", raw: `{"lsp_status_location":"sidebar"}`, want: models.LspStatusLocationToolbar}, {name: "toolbar is preserved", raw: `{"lsp_status_location":"toolbar"}`, want: models.LspStatusLocationToolbar}, {name: "status bar is preserved", raw: `{"lsp_status_location":"status_bar"}`, want: models.LspStatusLocationStatusBar}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: tt.raw}, DefaultUserID) if err != nil { t.Fatalf("scan settings: %v", err) } if settings.LspStatusLocation != tt.want { t.Fatalf("LspStatusLocation = %q, want %q", settings.LspStatusLocation, tt.want) } }) } } func TestMarshalUserSettingsLspStatusLocation(t *testing.T) { raw, err := marshalUserSettingsPayload(&models.UserSettings{ LspStatusLocation: models.LspStatusLocationStatusBar, }) if err != nil { t.Fatalf("marshal settings: %v", err) } var payload map[string]any if err := json.Unmarshal(raw, &payload); err != nil { t.Fatalf("decode settings: %v", err) } if got := payload["lsp_status_location"]; got != models.LspStatusLocationStatusBar { t.Fatalf("lsp_status_location = %#v, want status_bar", got) } } func TestSQLiteRepositoryLspStatusLocationRoundTrip(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } settings.LspStatusLocation = models.LspStatusLocationStatusBar upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get saved settings: %v", err) } if got.LspStatusLocation != models.LspStatusLocationStatusBar { t.Fatalf("LspStatusLocation = %q, want status_bar", got.LspStatusLocation) } } func TestScanUserSettingsSystemMetricsDisplayDefault(t *testing.T) { settings, err := scanUserSettings(settingsScanner{raw: "{}"}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if settings.SystemMetricsDisplay.ShowInTopbar { t.Fatal("system metrics display should default to disabled") } encoded, err := json.Marshal(settings.SystemMetricsDisplay) if err != nil { t.Fatalf("marshal system metrics display: %v", err) } if string(encoded) != `{"show_in_topbar":false,"simplified":false}` { t.Fatalf("default system metrics display = %s, want detailed preference", encoded) } settings, err = scanUserSettings(settingsScanner{raw: `{"system_metrics_display":{"show_in_topbar":true,"simplified":true}}`}, DefaultUserID) if err != nil { t.Fatalf("unexpected error: %v", err) } if !settings.SystemMetricsDisplay.ShowInTopbar { t.Fatal("expected stored system metrics display preference") } encoded, err = json.Marshal(settings.SystemMetricsDisplay) if err != nil { t.Fatalf("marshal stored system metrics display: %v", err) } if string(encoded) != `{"show_in_topbar":true,"simplified":true}` { t.Fatalf("stored system metrics display = %s, want simplified preference", encoded) } } func TestSQLiteRepositorySystemMetricsDisplayRoundTrip(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } settings.SystemMetricsDisplay = models.SystemMetricsDisplaySettings{ShowInTopbar: true, Simplified: true} upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get settings: %v", err) } if !got.SystemMetricsDisplay.ShowInTopbar || !got.SystemMetricsDisplay.Simplified { t.Fatal("expected system metrics display preference to round-trip") } } func TestSQLiteRepositoryAppStatusBarOrderDefaultAndRoundTrip(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } if settings.AppStatusBarOrder.LeftItemIDs == nil || settings.AppStatusBarOrder.RightItemIDs == nil { t.Fatalf("default AppStatusBarOrder = %#v, want non-nil empty arrays", settings.AppStatusBarOrder) } settings.AppStatusBarOrder = models.AppStatusBarOrder{ LeftItemIDs: []string{"builtin:connection", "plugin:left"}, RightItemIDs: []string{"builtin:metrics", "plugin:right"}, } upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get settings: %v", err) } if !reflect.DeepEqual(got.AppStatusBarOrder, settings.AppStatusBarOrder) { t.Fatalf("AppStatusBarOrder = %#v, want %#v", got.AppStatusBarOrder, settings.AppStatusBarOrder) } } func TestSQLiteRepositoryUpdateTaskCreateLastUsedPatchesNonEmptyFields(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } settings.SidebarActiveViewID = "view-1" settings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-1", Branch: "main", AgentProfileID: "agent-1", ExecutorProfileID: "exec-1", } upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.UpdateTaskCreateLastUsed(ctx, DefaultUserID, models.TaskCreateLastUsed{ Branch: "feature", AgentProfileID: "agent-2", }) if err != nil { t.Fatalf("update task-create last-used: %v", err) } if got.TaskCreateLastUsed.RepositoryID != "repo-1" { t.Fatalf("repository id should be preserved, got %q", got.TaskCreateLastUsed.RepositoryID) } if got.TaskCreateLastUsed.Branch != "feature" { t.Fatalf("branch should update, got %q", got.TaskCreateLastUsed.Branch) } if got.TaskCreateLastUsed.AgentProfileID != "agent-2" { t.Fatalf("agent profile should update, got %q", got.TaskCreateLastUsed.AgentProfileID) } if got.TaskCreateLastUsed.ExecutorProfileID != "exec-1" { t.Fatalf("executor profile should be preserved, got %q", got.TaskCreateLastUsed.ExecutorProfileID) } if got.SidebarActiveViewID != "view-1" { t.Fatalf("unrelated settings should be preserved, got active view %q", got.SidebarActiveViewID) } } func TestSQLiteRepositoryUpdateTaskCreateLastUsedClearsBranchOnRepositoryChange(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } settings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-before", Branch: "main", AgentProfileID: "agent-1", ExecutorProfileID: "exec-1", } upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.UpdateTaskCreateLastUsed(ctx, DefaultUserID, models.TaskCreateLastUsed{ RepositoryID: "repo-after", }) if err != nil { t.Fatalf("update task-create last-used: %v", err) } if got.TaskCreateLastUsed.RepositoryID != "repo-after" { t.Fatalf("repository id should update, got %q", got.TaskCreateLastUsed.RepositoryID) } if got.TaskCreateLastUsed.Branch != "" { t.Fatalf("branch should clear on repository change, got %q", got.TaskCreateLastUsed.Branch) } if got.TaskCreateLastUsed.AgentProfileID != "agent-1" { t.Fatalf("agent profile should be preserved, got %q", got.TaskCreateLastUsed.AgentProfileID) } if got.TaskCreateLastUsed.ExecutorProfileID != "exec-1" { t.Fatalf("executor profile should be preserved, got %q", got.TaskCreateLastUsed.ExecutorProfileID) } } func TestBuildPostgresTaskCreateLastUsedUpdatePatchesNonEmptyFields(t *testing.T) { query, args := buildPostgresTaskCreateLastUsedUpdate(models.TaskCreateLastUsed{ RepositoryID: "repo-1", Branch: "feature", AgentProfileID: "agent-1", ExecutorProfileID: "exec-1", }) if strings.Contains(query, "json(") || strings.Contains(query, "json_extract") { t.Fatalf("postgres update must not use sqlite JSON functions: %s", query) } if !strings.Contains(query, "jsonb_set") { t.Fatalf("postgres update should use jsonb_set: %s", query) } if !strings.Contains(query, "{task_create_last_used,repository_id}") || !strings.Contains(query, "{task_create_last_used,branch}") || !strings.Contains(query, "{task_create_last_used,agent_profile_id}") || !strings.Contains(query, "{task_create_last_used,executor_profile_id}") { t.Fatalf("postgres update should patch task-create fields: %s", query) } if len(args) != 4 { t.Fatalf("expected one arg per task-create field, got %d", len(args)) } } func TestBuildPostgresTaskCreateLastUsedUpdateClearsBranchOnRepositoryChange(t *testing.T) { query, args := buildPostgresTaskCreateLastUsedUpdate(models.TaskCreateLastUsed{ RepositoryID: "repo-after", }) if !strings.Contains(query, "{task_create_last_used,repository_id}") || !strings.Contains(query, "{task_create_last_used,branch}") { t.Fatalf("postgres update should patch repository and clear branch: %s", query) } if strings.Contains(query, "{task_create_last_used,agent_profile_id}") || strings.Contains(query, "{task_create_last_used,executor_profile_id}") { t.Fatalf("postgres update should not patch profile fields: %s", query) } if len(args) != 2 { t.Fatalf("expected repository and empty branch args, got %d", len(args)) } if args[0] != "repo-after" || args[1] != "" { t.Fatalf("expected repo-after and empty branch args, got %#v", args) } } func TestBuildPostgresUserSettingsPreservingTaskCreateLastUsedUpdateUsesJSONB(t *testing.T) { patch := models.TaskCreateLastUsed{ RepositoryID: "repo-1", Branch: "feature", AgentProfileID: "agent-1", ExecutorProfileID: "exec-1", } query, args := buildPostgresUserSettingsPreservingTaskCreateLastUsedUpdate(&patch) if strings.Contains(query, "json(") || strings.Contains(query, "json_extract") { t.Fatalf("postgres update must not use sqlite JSON functions: %s", query) } if !strings.Contains(query, "?::jsonb") || !strings.Contains(query, "jsonb_set") { t.Fatalf("postgres update should merge payload with jsonb_set: %s", query) } if !strings.Contains(query, "{task_create_last_used,repository_id}") || !strings.Contains(query, "{task_create_last_used,branch}") || !strings.Contains(query, "{task_create_last_used,agent_profile_id}") || !strings.Contains(query, "{task_create_last_used,executor_profile_id}") { t.Fatalf("postgres update missing task-create paths: %s", query) } if len(args) != 4 { t.Fatalf("expected one arg per task-create field, got %d", len(args)) } } func TestPostgresRepositoryTaskCreateLastUsedRoundTrip(t *testing.T) { conn := testutil.OpenIsolatedPostgres(t, testutil.PostgresDSNFromEnv(t)) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new postgres repo: %v", err) } ctx := context.Background() staleSettings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } staleSettings.SidebarActiveViewID = "view-before" staleSettings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-before", Branch: "main", AgentProfileID: "agent-before", ExecutorProfileID: "exec-before", } upsertUserSettingsForTest(t, repo, ctx, staleSettings) got, err := repo.UpdateTaskCreateLastUsed(ctx, DefaultUserID, models.TaskCreateLastUsed{ RepositoryID: "repo-after", Branch: "feature", AgentProfileID: "agent-after", }) if err != nil { t.Fatalf("update postgres task-create last-used: %v", err) } if got.SidebarActiveViewID != "view-before" { t.Fatalf("unrelated postgres setting should be preserved, got %q", got.SidebarActiveViewID) } if got.TaskCreateLastUsed.RepositoryID != "repo-after" || got.TaskCreateLastUsed.Branch != "feature" || got.TaskCreateLastUsed.AgentProfileID != "agent-after" || got.TaskCreateLastUsed.ExecutorProfileID != "exec-before" { t.Fatalf("postgres task-create update mismatch: %+v", got.TaskCreateLastUsed) } staleSettings.SidebarActiveViewID = "view-after" staleSettings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-stale", Branch: "stale", } got, err = repo.UpsertUserSettingsPreservingTaskCreateLastUsed(ctx, staleSettings, nil) if err != nil { t.Fatalf("upsert preserving postgres task-create last-used: %v", err) } if got.SidebarActiveViewID != "view-after" { t.Fatalf("expected unrelated postgres setting to update, got %q", got.SidebarActiveViewID) } if got.TaskCreateLastUsed.RepositoryID != "repo-after" || got.TaskCreateLastUsed.Branch != "feature" || got.TaskCreateLastUsed.AgentProfileID != "agent-after" || got.TaskCreateLastUsed.ExecutorProfileID != "exec-before" { t.Fatalf("postgres preserving upsert should keep current task-create values: %+v", got.TaskCreateLastUsed) } } func TestSQLiteRepositoryUpsertSettingsPreservesCurrentTaskCreateLastUsed(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() staleSettings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } staleSettings.SidebarActiveViewID = "view-before" staleSettings.TaskCreateLastUsed = models.TaskCreateLastUsed{RepositoryID: "repo-before"} upsertUserSettingsForTest(t, repo, ctx, staleSettings) // Simulate a task-create write that lands after another settings caller // read the old blob but before that caller writes its unrelated change. if _, err := repo.UpdateTaskCreateLastUsed(ctx, DefaultUserID, models.TaskCreateLastUsed{ RepositoryID: "repo-after", Branch: "feature", }); err != nil { t.Fatalf("update task-create last-used: %v", err) } staleSettings.SidebarActiveViewID = "view-after" got, err := repo.UpsertUserSettingsPreservingTaskCreateLastUsed(ctx, staleSettings, nil) if err != nil { t.Fatalf("upsert preserving task-create last-used: %v", err) } if got.SidebarActiveViewID != "view-after" { t.Fatalf("expected unrelated setting to update, got %q", got.SidebarActiveViewID) } if got.TaskCreateLastUsed.RepositoryID != "repo-after" { t.Fatalf("expected current task-create repository to survive stale write, got %q", got.TaskCreateLastUsed.RepositoryID) } if got.TaskCreateLastUsed.Branch != "feature" { t.Fatalf("expected current task-create branch to survive stale write, got %q", got.TaskCreateLastUsed.Branch) } } func TestSQLiteRepositoryUpsertSettingsPreservingTaskCreateLastUsedAppliesPatch(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() staleSettings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } staleSettings.SidebarActiveViewID = "view-before" staleSettings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-before", Branch: "main", AgentProfileID: "agent-before", ExecutorProfileID: "exec-before", } upsertUserSettingsForTest(t, repo, ctx, staleSettings) if _, err := repo.UpdateTaskCreateLastUsed(ctx, DefaultUserID, models.TaskCreateLastUsed{ RepositoryID: "repo-current", Branch: "current", ExecutorProfileID: "exec-current", }); err != nil { t.Fatalf("update current task-create last-used: %v", err) } staleSettings.SidebarActiveViewID = "view-after" staleSettings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-stale", Branch: "stale", AgentProfileID: "agent-stale", } patch := models.TaskCreateLastUsed{AgentProfileID: "agent-after"} got, err := repo.UpsertUserSettingsPreservingTaskCreateLastUsed(ctx, staleSettings, &patch) if err != nil { t.Fatalf("upsert preserving task-create last-used with patch: %v", err) } if got.SidebarActiveViewID != "view-after" { t.Fatalf("expected unrelated setting to update, got %q", got.SidebarActiveViewID) } if got.TaskCreateLastUsed.RepositoryID != "repo-current" { t.Fatalf("expected current repository to survive stale write, got %q", got.TaskCreateLastUsed.RepositoryID) } if got.TaskCreateLastUsed.Branch != "current" { t.Fatalf("expected current branch to survive stale write, got %q", got.TaskCreateLastUsed.Branch) } if got.TaskCreateLastUsed.AgentProfileID != "agent-after" { t.Fatalf("expected patch agent profile to apply, got %q", got.TaskCreateLastUsed.AgentProfileID) } if got.TaskCreateLastUsed.ExecutorProfileID != "exec-current" { t.Fatalf("expected current executor profile to survive stale write, got %q", got.TaskCreateLastUsed.ExecutorProfileID) } } func TestSQLiteRepositorySidebarViewStateRoundTrip(t *testing.T) { conn, err := sqlx.Open("sqlite3", ":memory:") if err != nil { t.Fatalf("open sqlite: %v", err) } conn.SetMaxOpenConns(1) t.Cleanup(func() { _ = conn.Close() }) repo, err := newSQLiteRepositoryWithDB(conn, conn) if err != nil { t.Fatalf("new repo: %v", err) } ctx := context.Background() settings, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get defaults: %v", err) } settings.SidebarActiveViewID = "view-1" settings.SidebarTaskPrefs = models.SidebarTaskPrefs{ PinnedTaskIDs: []string{"task-1"}, OrderedTaskIDs: []string{"task-2", "task-1"}, SubtaskOrderByParentID: map[string][]string{"task-1": {"sub-1"}}, } settings.TaskCreateLastUsed = models.TaskCreateLastUsed{ RepositoryID: "repo-1", Branch: "main", AgentProfileID: "agent-1", ExecutorProfileID: "exec-1", } settings.JiraSavedViews = json.RawMessage(`[{"id":"view-1"}]`) settings.GitLabSavedPresets = json.RawMessage(`[{"id":"preset-1"}]`) settings.SidebarDraft = &models.SidebarViewDraft{ BaseViewID: "view-1", Filters: []models.SidebarViewClause{{ ID: "clause-1", Dimension: "titleMatch", Op: "matches", Value: json.RawMessage(`"bug"`), }}, Sort: models.SidebarViewSort{Key: "updatedAt", Direction: "desc"}, Group: "workflow", } upsertUserSettingsForTest(t, repo, ctx, settings) got, err := repo.GetUserSettings(ctx, DefaultUserID) if err != nil { t.Fatalf("get settings: %v", err) } if got.SidebarActiveViewID != "view-1" { t.Fatalf("expected active view to round-trip, got %q", got.SidebarActiveViewID) } if got.SidebarDraft == nil || got.SidebarDraft.Group != "workflow" { t.Fatalf("expected sidebar draft to round-trip, got %+v", got.SidebarDraft) } if got.SidebarTaskPrefs.PinnedTaskIDs[0] != "task-1" { t.Fatalf("expected sidebar task prefs to round-trip, got %+v", got.SidebarTaskPrefs) } if got.TaskCreateLastUsed.Branch != "main" { t.Fatalf("expected task-create prefs to round-trip, got %+v", got.TaskCreateLastUsed) } if string(got.JiraSavedViews) != `[{"id":"view-1"}]` { t.Fatalf("expected Jira saved views to round-trip, got %s", string(got.JiraSavedViews)) } if string(got.GitLabSavedPresets) != `[{"id":"preset-1"}]` { t.Fatalf("expected GitLab presets to round-trip, got %s", string(got.GitLabSavedPresets)) } }