use super::activity_detail::*;
use super::*;
use crate::config::{
ApiProvider, Config, DEFAULT_OPENROUTER_MODEL, DEFAULT_TEXT_MODEL, DEFAULT_ZAI_MODEL,
ProviderConfig, ProvidersConfig,
};
use crate::config_ui::{self, WebConfigSession, WebConfigSessionEvent};
use crate::core::engine::mock_engine_handle;
use crate::tui::active_cell::ActiveCell;
use crate::tui::app::{ReasoningEffort, SidebarHoverRow, SidebarHoverSection, ToolDetailRecord};
use crate::tui::file_mention::{
apply_mention_menu_selection, find_file_mention_completions, partial_file_mention_at_cursor,
try_autocomplete_file_mention, user_request_with_file_mentions, visible_mention_menu_entries,
};
use crate::tui::footer_ui::{
active_tool_status_label, footer_auxiliary_spans, footer_balance_spans, footer_cache_spans,
footer_session_tokens_spans, footer_state_label, footer_status_line_spans,
format_context_budget, format_token_count_compact, friendly_subagent_progress,
render_footer_from,
};
use crate::tui::history::{
ExecCell, ExecSource, GenericToolCell, HistoryCell, SubAgentCell, ToolCell, ToolStatus,
};
use crate::tui::hotbar::actions::{HotbarActionCategory, HotbarDispatch};
use crate::tui::ui_text::truncate_line_to_width;
use crate::tui::views::{HelpView, ModalView, ViewAction};
use crate::working_set::Workspace;
use crossterm::event::{KeyEvent, MouseButton, MouseEvent, MouseEventKind};
use ratatui::{Terminal, backend::TestBackend, text::Span};
use std::collections::{HashMap, HashSet, VecDeque};
use std::ffi::OsString;
use std::path::PathBuf;
use std::process::Command;
use std::sync::MutexGuard;
use std::time::{Duration, Instant};
use unicode_width::UnicodeWidthStr;
use crate::tui::selection::{SelectionAutoscroll, TranscriptSelectionPoint};
use tempfile::TempDir;
#[test]
fn permission_cycle_shortcut_accepts_both_shift_tab_encodings() {
assert!(is_permission_cycle_shortcut(&KeyEvent::new(
KeyCode::BackTab,
KeyModifiers::NONE,
)));
assert!(is_permission_cycle_shortcut(&KeyEvent::new(
KeyCode::BackTab,
KeyModifiers::SHIFT,
)));
assert!(is_permission_cycle_shortcut(&KeyEvent::new(
KeyCode::Tab,
KeyModifiers::SHIFT,
)));
assert!(!is_permission_cycle_shortcut(&KeyEvent::new(
KeyCode::Tab,
KeyModifiers::NONE,
)));
assert!(!is_permission_cycle_shortcut(&KeyEvent::new(
KeyCode::BackTab,
KeyModifiers::CONTROL,
)));
}
#[test]
fn underwater_motion_keeps_its_smoother_cadence_during_live_status() {
let mut app = create_test_app();
app.low_motion = false;
app.fancy_animations = true;
assert_eq!(
animation_interval_ms(&app, true, false),
UI_STATUS_ANIMATION_MS
);
assert_eq!(
animation_interval_ms(&app, false, true),
UI_UNDERWATER_ANIMATION_MS
);
assert_eq!(
animation_interval_ms(&app, true, true),
UI_UNDERWATER_ANIMATION_MS,
"the slower status spinner must not throttle ambient fish"
);
}
#[test]
fn underwater_motion_ticks_only_for_visible_unobscured_owners() {
assert!(!underwater_motion_surface_visible(None, true, true, false));
assert!(!underwater_motion_surface_visible(
Some(Rect::new(0, 0, 0, 24)),
true,
true,
false,
));
assert!(underwater_motion_surface_visible(
Some(Rect::new(0, 0, 40, 12)),
true,
false,
false,
));
assert!(underwater_motion_surface_visible(
Some(Rect::new(0, 0, 60, 16)),
false,
true,
false,
));
assert!(!underwater_motion_surface_visible(
Some(Rect::new(0, 0, 60, 16)),
false,
false,
false,
));
assert!(underwater_motion_surface_visible(
Some(Rect::new(0, 0, 80, 24)),
false,
false,
false,
));
assert!(!underwater_motion_surface_visible(
Some(Rect::new(0, 0, 100, 32)),
true,
true,
true,
));
}
#[test]
fn live_transcript_command_open_path_is_idempotent() {
let mut app = create_test_app();
open_live_transcript_overlay(&mut app);
open_live_transcript_overlay(&mut app);
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::LiveTranscript));
app.view_stack.pop();
assert_eq!(
app.view_stack.top_kind(),
None,
"opening twice must not stack duplicate transcript overlays"
);
}
#[test]
fn approval_prompt_keeps_transcript_page_navigation_live() {
let mut app = create_test_app();
app.viewport.last_transcript_visible = 12;
app.view_stack.push(ApprovalView::new(ApprovalRequest::new(
"approval-scroll",
"exec_shell",
"Review command",
&serde_json::json!({"command": "git status"}),
"approval-scroll-key",
)));
assert!(handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::PageUp, KeyModifiers::NONE),
));
assert_eq!(app.viewport.pending_scroll_delta, -12);
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::Approval));
assert!(handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::PageDown, KeyModifiers::NONE),
));
assert_eq!(app.viewport.pending_scroll_delta, 0);
assert!(handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::Up, KeyModifiers::CONTROL),
));
assert_eq!(app.viewport.pending_scroll_delta, -3);
assert!(handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::Home, KeyModifiers::NONE),
));
assert!(app.viewport.pending_scroll_delta < -1_000_000);
assert!(app.user_scrolled_during_stream);
assert!(handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::End, KeyModifiers::NONE),
));
assert_eq!(app.viewport.pending_scroll_delta, 0);
assert!(!app.user_scrolled_during_stream);
assert!(handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::Down, KeyModifiers::SHIFT),
));
assert_eq!(app.viewport.pending_scroll_delta, 3);
assert!(app.user_scrolled_during_stream);
assert!(
!handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::Char('j'), KeyModifiers::NONE),
),
"ordinary approval selection keys must stay with the decision card"
);
}
#[test]
fn transcript_navigation_does_not_capture_keys_for_other_modals() {
let mut app = create_test_app();
app.view_stack.push(HelpView::new_for_locale(app.ui_locale));
assert!(!handle_approval_transcript_key(
&mut app,
&KeyEvent::new(KeyCode::PageUp, KeyModifiers::NONE),
));
assert_eq!(app.viewport.pending_scroll_delta, 0);
}
#[test]
fn approval_mouse_wheel_reviews_transcript_without_closing_card() {
let mut app = create_test_app();
app.view_stack.push(ApprovalView::new(ApprovalRequest::new(
"approval-scroll",
"exec_shell",
"Review command",
&serde_json::json!({"command": "git status"}),
"approval-scroll-key",
)));
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::ScrollUp,
column: 4,
row: 4,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert!(app.viewport.pending_scroll_delta < 0);
assert!(app.user_scrolled_during_stream);
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::Approval));
}
#[test]
fn approval_wheel_preserves_sidebar_and_work_surface_ownership() {
let mut app = create_test_app();
app.view_stack.push(ApprovalView::new(ApprovalRequest::new(
"approval-scroll",
"exec_shell",
"Review command",
&serde_json::json!({"command": "git status"}),
"approval-scroll-key",
)));
app.viewport.last_sidebar_area = Some(Rect::new(60, 0, 20, 20));
app.work_surface.last_area = Some(Rect::new(0, 0, 30, 20));
app.viewport.last_approval_area = Some(Rect::new(0, 12, 80, 8));
for (column, row) in [(65, 4), (10, 4)] {
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::ScrollUp,
column,
row,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert_eq!(
app.viewport.pending_scroll_delta, 0,
"approval wheel leaked from side surface at ({column}, {row})"
);
}
for (column, row) in [(65, 15), (10, 15)] {
let before = app.viewport.pending_scroll_delta;
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::ScrollUp,
column,
row,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert!(
app.viewport.pending_scroll_delta < before,
"visible approval did not outrank underlying side surface at ({column}, {row})"
);
}
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::Approval));
}
#[test]
fn config_update_preview_suppresses_only_success_message_not_action_or_errors() {
let preview = prepare_config_update_result(
commands::CommandResult::with_message_and_action(
"preview confirmation",
AppAction::UpdateStreamChunkTimeout(42),
),
false,
);
assert!(preview.message.is_none());
assert!(matches!(
preview.action,
Some(AppAction::UpdateStreamChunkTimeout(42))
));
let persisted =
prepare_config_update_result(commands::CommandResult::message("saved confirmation"), true);
assert_eq!(persisted.message.as_deref(), Some("saved confirmation"));
let error =
prepare_config_update_result(commands::CommandResult::error("invalid value"), false);
assert!(error.is_error);
assert!(
error
.message
.as_deref()
.is_some_and(|msg| msg.contains("invalid value"))
);
}
#[test]
fn config_refresh_preserves_active_search_filter() {
let mut app = create_test_app();
let mut view = ConfigView::new_for_app(&app);
for ch in "model".chars() {
let _ = view.handle_key(KeyEvent::new(
crossterm::event::KeyCode::Char(ch),
crossterm::event::KeyModifiers::NONE,
));
}
app.view_stack.push(view);
refresh_config_view_if_open(&mut app, "default_model");
let mut view = app.view_stack.pop().expect("refreshed config view");
let config = view
.as_any_mut()
.downcast_mut::<ConfigView>()
.expect("config view type");
assert_eq!(config.filter_query(), "model");
}
#[test]
fn workflow_panel_plain_letters_return_to_composer() {
let mut app = create_test_app();
app.workflow_panel = Some(crate::tui::widgets::workflow_panel::WorkflowPanel::new(
"workflow_typing",
"typing regression",
0,
));
for ch in ['t', 'c', 'j', 'k'] {
app.workflow_panel
.as_mut()
.expect("workflow panel")
.keyboard_focus = true;
let key = KeyEvent::new(KeyCode::Char(ch), KeyModifiers::NONE);
assert!(
!handle_workflow_panel_key(&mut app, &key),
"plain {ch:?} must fall through to the composer"
);
assert!(
!app.workflow_panel
.as_ref()
.expect("workflow panel")
.keyboard_focus,
"typing releases panel focus"
);
app.insert_char(ch);
}
assert_eq!(app.input, "tcjk");
}
#[test]
fn workflow_panel_uses_non_text_keys_for_controls() {
let mut app = create_test_app();
let mut panel = crate::tui::widgets::workflow_panel::WorkflowPanel::new(
"workflow_keys",
"keyboard controls",
0,
);
panel.keyboard_focus = true;
let was_expanded = panel.expanded;
app.workflow_panel = Some(panel);
assert!(handle_workflow_panel_key(
&mut app,
&KeyEvent::new(KeyCode::Enter, KeyModifiers::NONE)
));
assert_ne!(
app.workflow_panel
.as_ref()
.expect("workflow panel")
.expanded,
was_expanded
);
}
struct ConfigPathEnvGuard {
_tmp: TempDir,
previous: Option<OsString>,
_lock: MutexGuard<'static, ()>,
}
impl ConfigPathEnvGuard {
fn new() -> Self {
let lock = crate::test_support::lock_test_env();
let tmp = TempDir::new().expect("config tempdir");
let config_path = tmp.path().join(".deepseek").join("config.toml");
std::fs::create_dir_all(config_path.parent().expect("config parent")).expect("config dir");
let previous = std::env::var_os("DEEPSEEK_CONFIG_PATH");
unsafe {
std::env::set_var("DEEPSEEK_CONFIG_PATH", &config_path);
}
Self {
_tmp: tmp,
previous,
_lock: lock,
}
}
}
impl Drop for ConfigPathEnvGuard {
fn drop(&mut self) {
unsafe {
if let Some(previous) = self.previous.take() {
std::env::set_var("DEEPSEEK_CONFIG_PATH", previous);
} else {
std::env::remove_var("DEEPSEEK_CONFIG_PATH");
}
}
}
}
struct SettingsHomeGuard {
_tmp: TempDir,
previous_home: Option<OsString>,
previous_userprofile: Option<OsString>,
previous_codewhale_home: Option<OsString>,
previous_deepseek_config_path: Option<OsString>,
previous_codewhale_provider: Option<OsString>,
previous_deepseek_provider: Option<OsString>,
previous_xdg_config_home: Option<OsString>,
previous_appdata: Option<OsString>,
previous_localappdata: Option<OsString>,
_lock: MutexGuard<'static, ()>,
}
impl SettingsHomeGuard {
fn new() -> Self {
let lock = crate::test_support::lock_test_env();
let tmp = TempDir::new().expect("settings tempdir");
let previous_home = std::env::var_os("HOME");
let previous_userprofile = std::env::var_os("USERPROFILE");
let previous_codewhale_home = std::env::var_os("CODEWHALE_HOME");
let previous_deepseek_config_path = std::env::var_os("DEEPSEEK_CONFIG_PATH");
let previous_codewhale_provider = std::env::var_os("CODEWHALE_PROVIDER");
let previous_deepseek_provider = std::env::var_os("DEEPSEEK_PROVIDER");
let previous_xdg_config_home = std::env::var_os("XDG_CONFIG_HOME");
let previous_appdata = std::env::var_os("APPDATA");
let previous_localappdata = std::env::var_os("LOCALAPPDATA");
let codewhale_home = tmp.path().join(".codewhale");
unsafe {
std::env::set_var("HOME", tmp.path());
std::env::set_var("USERPROFILE", tmp.path());
std::env::set_var("CODEWHALE_HOME", &codewhale_home);
std::env::set_var("DEEPSEEK_CONFIG_PATH", codewhale_home.join("config.toml"));
std::env::remove_var("CODEWHALE_PROVIDER");
std::env::remove_var("DEEPSEEK_PROVIDER");
std::env::set_var("XDG_CONFIG_HOME", tmp.path().join("xdg-config"));
std::env::set_var("APPDATA", tmp.path().join("appdata"));
std::env::set_var("LOCALAPPDATA", tmp.path().join("localappdata"));
}
Self {
_tmp: tmp,
previous_home,
previous_userprofile,
previous_codewhale_home,
previous_deepseek_config_path,
previous_codewhale_provider,
previous_deepseek_provider,
previous_xdg_config_home,
previous_appdata,
previous_localappdata,
_lock: lock,
}
}
}
impl Drop for SettingsHomeGuard {
fn drop(&mut self) {
fn restore(key: &str, previous: Option<OsString>) {
unsafe {
match previous {
Some(previous) => std::env::set_var(key, previous),
None => std::env::remove_var(key),
}
}
}
restore("HOME", self.previous_home.take());
restore("USERPROFILE", self.previous_userprofile.take());
restore("CODEWHALE_HOME", self.previous_codewhale_home.take());
restore(
"DEEPSEEK_CONFIG_PATH",
self.previous_deepseek_config_path.take(),
);
restore(
"CODEWHALE_PROVIDER",
self.previous_codewhale_provider.take(),
);
restore("DEEPSEEK_PROVIDER", self.previous_deepseek_provider.take());
restore("XDG_CONFIG_HOME", self.previous_xdg_config_home.take());
restore("APPDATA", self.previous_appdata.take());
restore("LOCALAPPDATA", self.previous_localappdata.take());
}
}
#[test]
fn resume_hint_uses_canonical_resume_command() {
assert_eq!(
resume_hint_text(),
"To continue this session, execute codewhale run --continue"
);
assert!(should_show_resume_hint(Some(
"019dd9d6-4f44-7c83-9863-59674a12b827"
)));
}
#[test]
fn resume_hint_omits_missing_session_id() {
assert!(!should_show_resume_hint(None));
assert!(!should_show_resume_hint(Some(" ")));
}
#[test]
fn plain_mcp_show_refreshes_discovery_counts() {
use crate::tui::app::McpUiAction;
assert!(mcp_ui_action_refreshes_discovery(&McpUiAction::Show));
assert!(mcp_ui_action_refreshes_discovery(&McpUiAction::Validate));
assert!(mcp_ui_action_refreshes_discovery(&McpUiAction::Reload));
assert!(!mcp_ui_action_refreshes_discovery(&McpUiAction::Init {
force: false,
}));
}
#[test]
fn focus_gained_forces_terminal_viewport_recapture() {
assert!(terminal_event_needs_viewport_recapture(&Event::FocusGained));
assert!(!terminal_event_needs_viewport_recapture(&Event::FocusLost));
}
#[cfg(not(windows))]
#[test]
fn recover_terminal_modes_emits_expected_csi_sequences_with_gating() {
let mut all_on: Vec<u8> = Vec::new();
let mut all_off: Vec<u8> = Vec::new();
recover_terminal_modes(&mut all_on, true, true);
recover_terminal_modes(&mut all_off, false, false);
let on = String::from_utf8_lossy(&all_on);
let off = String::from_utf8_lossy(&all_off);
assert!(
on.contains("\x1b[?1004h") && off.contains("\x1b[?1004h"),
"EnableFocusChange must be re-armed regardless of gating"
);
assert!(
on.contains("\x1b[>1u") && off.contains("\x1b[>1u"),
"Kitty keyboard disambiguation flag must be re-pushed regardless of gating"
);
assert!(
on.contains("\x1b[?1007h"),
"alternate-scroll mode must be re-armed when mouse capture is active"
);
assert!(
!off.contains("\x1b[?1007h"),
"alternate-scroll mode must stay off when mouse capture is disabled"
);
assert!(
off.contains("\x1b[?1007l"),
"alternate-scroll mode must be reset when mouse capture is disabled"
);
assert!(
on.contains("\x1b[?1000h"),
"EnableMouseCapture missing when use_mouse_capture=true"
);
assert!(
!off.contains("\x1b[?1000h"),
"EnableMouseCapture must be gated by use_mouse_capture"
);
assert!(
on.contains("\x1b[?2004h"),
"EnableBracketedPaste missing when use_bracketed_paste=true"
);
assert!(
!off.contains("\x1b[?2004h"),
"EnableBracketedPaste must be gated by use_bracketed_paste"
);
}
#[cfg(not(windows))]
#[test]
fn bracketed_paste_mode_helpers_ignore_writer_errors() {
struct FailingWriter;
impl std::io::Write for FailingWriter {
fn write(&mut self, _buf: &[u8]) -> std::io::Result<usize> {
Err(std::io::Error::other("terminal mode unsupported"))
}
fn flush(&mut self) -> std::io::Result<()> {
Err(std::io::Error::other("terminal mode unsupported"))
}
}
let mut writer = FailingWriter;
assert!(
!try_enable_bracketed_paste_mode(&mut writer),
"unsupported bracketed paste must be reported without bubbling an error"
);
disable_bracketed_paste_mode(&mut writer);
}
#[cfg(windows)]
#[test]
fn recover_terminal_modes_runs_without_panic_on_windows() {
let mut buf: Vec<u8> = Vec::new();
recover_terminal_modes(&mut buf, true, true);
recover_terminal_modes(&mut buf, false, false);
}
#[test]
fn alternate_scroll_mode_disable_emits_xterm_reset() {
let mut buf: Vec<u8> = Vec::new();
disable_alternate_scroll_mode(&mut buf);
let seq = String::from_utf8_lossy(&buf);
assert!(
seq.contains("\x1b[?1007l"),
"disable_alternate_scroll_mode must emit the xterm alternate-scroll reset"
);
}
#[cfg(windows)]
#[test]
fn push_keyboard_flags_writes_kitty_push_sequence_on_windows() {
let mut buf: Vec<u8> = Vec::new();
push_keyboard_enhancement_flags(&mut buf);
let seq = String::from_utf8_lossy(&buf);
assert!(
seq.contains("\x1b[>0u"),
"push_keyboard_enhancement_flags must write kitty probe (\\x1b[>0u) on Windows (#1599); got: {seq:?}"
);
}
#[cfg(windows)]
#[test]
fn pop_keyboard_flags_writes_kitty_pop_sequence_on_windows() {
let mut buf: Vec<u8> = Vec::new();
pop_keyboard_enhancement_flags(&mut buf);
let seq = String::from_utf8_lossy(&buf);
assert!(
seq.contains("\x1b[<1u"),
"pop_keyboard_enhancement_flags must write kitty pop (\\x1b[<1u) on Windows (#1359); got: {seq:?}"
);
}
#[test]
fn terminal_origin_reset_resets_scroll_region_origin_without_destructive_clear() {
assert!(
TERMINAL_ORIGIN_RESET.starts_with(b"\x1b[r\x1b[?6l"),
"must reset scroll margins and origin mode before repaint"
);
assert!(
TERMINAL_ORIGIN_RESET.ends_with(b"\x1b[H"),
"must home the cursor at the end of the reset sequence"
);
assert!(
!TERMINAL_ORIGIN_RESET
.windows(b"\x1b[2J".len())
.any(|sequence| sequence == b"\x1b[2J"),
"must not emit destructive CSI 2J — causes visible flicker"
);
assert!(
!TERMINAL_ORIGIN_RESET
.windows(b"\x1b[3J".len())
.any(|sequence| sequence == b"\x1b[3J"),
"must not emit destructive CSI 3J — causes visible flicker"
);
}
#[test]
fn composer_newline_shortcuts_do_not_steal_ctrl_enter() {
assert!(is_composer_newline_key(KeyEvent::new(
KeyCode::Char('j'),
KeyModifiers::CONTROL,
)));
assert!(is_composer_newline_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::ALT,
)));
assert!(is_composer_newline_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::SHIFT,
)));
assert!(!is_composer_newline_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::NONE,
)));
assert!(!is_composer_newline_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::CONTROL,
)));
assert!(!is_composer_newline_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::CONTROL | KeyModifiers::SHIFT,
)));
}
#[test]
fn forced_submit_accepts_ctrl_enter_and_ctrl_j_encodings() {
assert!(is_forced_submit_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::CONTROL,
)));
assert!(is_forced_submit_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::CONTROL | KeyModifiers::SHIFT,
)));
assert!(is_forced_submit_key(KeyEvent::new(
KeyCode::Char('j'),
KeyModifiers::CONTROL,
)));
assert!(is_forced_submit_key(KeyEvent::new(
KeyCode::Char('J'),
KeyModifiers::CONTROL | KeyModifiers::SHIFT,
)));
assert!(!is_forced_submit_key(KeyEvent::new(
KeyCode::Char('j'),
KeyModifiers::ALT | KeyModifiers::CONTROL,
)));
assert!(!is_forced_submit_key(KeyEvent::new(
KeyCode::Enter,
KeyModifiers::ALT,
)));
}
#[cfg(target_os = "macos")]
#[test]
fn cmd_enter_normalizes_to_control_enter_not_newline() {
use crate::tui::composer_ui::normalize_macos_modifiers;
let normalized = normalize_macos_modifiers(KeyModifiers::SUPER);
assert!(normalized.contains(KeyModifiers::CONTROL));
assert!(!is_composer_newline_key(KeyEvent::new(
KeyCode::Enter,
normalized,
)));
}
#[test]
fn word_cursor_modifier_accepts_control_and_alt() {
assert!(is_word_cursor_modifier(KeyModifiers::CONTROL));
assert!(is_word_cursor_modifier(KeyModifiers::ALT));
assert!(is_word_cursor_modifier(
KeyModifiers::CONTROL | KeyModifiers::SHIFT
));
assert!(!is_word_cursor_modifier(KeyModifiers::NONE));
assert!(!is_word_cursor_modifier(KeyModifiers::SHIFT));
}
#[cfg(target_os = "macos")]
#[test]
fn normalize_macos_modifiers_maps_super_to_control() {
use crate::tui::composer_ui::normalize_macos_modifiers;
let normalized = normalize_macos_modifiers(KeyModifiers::SUPER);
assert!(normalized.contains(KeyModifiers::CONTROL));
assert!(!normalized.contains(KeyModifiers::SUPER));
}
#[cfg(target_os = "macos")]
#[test]
fn normalize_macos_modifiers_preserves_existing_control() {
use crate::tui::composer_ui::normalize_macos_modifiers;
let normalized = normalize_macos_modifiers(KeyModifiers::CONTROL | KeyModifiers::SUPER);
assert!(normalized.contains(KeyModifiers::CONTROL));
assert!(!normalized.contains(KeyModifiers::SUPER));
}
#[test]
fn normalize_macos_modifiers_leaves_alt_unchanged() {
use crate::tui::composer_ui::normalize_macos_modifiers;
let normalized = normalize_macos_modifiers(KeyModifiers::ALT);
assert!(normalized.contains(KeyModifiers::ALT));
assert!(!normalized.contains(KeyModifiers::SUPER));
}
#[test]
fn alt_f_and_alt_b_move_by_word_without_inserting_text() {
let mut app = create_test_app();
app.input = "alpha beta gamma".to_string();
app.cursor_position = 0;
assert!(handle_composer_alt_word_motion_key(
&mut app,
KeyEvent::new(KeyCode::Char('f'), KeyModifiers::ALT),
));
assert_eq!(app.input, "alpha beta gamma");
assert_eq!(app.cursor_position, "alpha ".chars().count());
app.selection_anchor = Some(0);
assert!(handle_composer_alt_word_motion_key(
&mut app,
KeyEvent::new(KeyCode::Char('b'), KeyModifiers::ALT),
));
assert_eq!(app.input, "alpha beta gamma");
assert_eq!(app.cursor_position, 0);
assert!(app.selection_anchor.is_none());
}
#[test]
fn alt_word_motion_helper_ignores_altgr_style_control_alt() {
let mut app = create_test_app();
app.input = "alpha beta".to_string();
app.cursor_position = 0;
assert!(!handle_composer_alt_word_motion_key(
&mut app,
KeyEvent::new(
KeyCode::Char('f'),
KeyModifiers::CONTROL | KeyModifiers::ALT
),
));
assert_eq!(app.cursor_position, 0);
}
fn select_full_transcript(app: &mut App) {
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: app
.viewport
.transcript_cache
.total_lines()
.saturating_sub(1),
column: 80,
});
}
#[test]
fn selection_point_from_position_ignores_top_padding() {
let area = Rect {
x: 10,
y: 20,
width: 30,
height: 5,
};
let padding_top = 3;
let transcript_top = 0;
let transcript_total = 2;
assert!(
selection_point_from_position(
area,
area.x + 1,
area.y,
transcript_top,
transcript_total,
padding_top,
)
.is_none()
);
let p0 = selection_point_from_position(
area,
area.x + 2,
area.y + u16::try_from(padding_top).expect("padding should fit"),
transcript_top,
transcript_total,
padding_top,
)
.expect("point");
assert_eq!(p0.line_index, 0);
assert_eq!(p0.column, 2);
let p1 = selection_point_from_position(
area,
area.x,
area.y + u16::try_from(padding_top + 1).expect("padding should fit"),
transcript_top,
transcript_total,
padding_top,
)
.expect("point");
assert_eq!(p1.line_index, 1);
assert_eq!(p1.column, 0);
}
#[test]
fn selection_to_text_excludes_transcript_rail_decorations() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "fix the flaky pty test".to_string(),
},
HistoryCell::Assistant {
content: "Looking at the failing test first.".to_string(),
streaming: false,
},
];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
let last = app
.viewport
.transcript_cache
.lines()
.len()
.saturating_sub(1);
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: last,
column: 80,
});
let copied = selection_to_text(&app).expect("selection yields text");
for rail in ['▎', '▏', '╎', '│', '┃', '╭', '╰', '●'] {
assert!(
!copied.contains(rail),
"copied text must exclude the {rail:?} rail: {copied:?}"
);
}
assert!(
copied.contains("fix the flaky pty test"),
"content survives rail stripping: {copied:?}"
);
assert!(
copied.contains("Looking at the failing test first."),
"assistant content survives: {copied:?}"
);
}
#[test]
fn selection_to_text_handles_multiline_and_reversed_endpoints() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta\ngamma delta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 1,
column: 5,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: 0,
column: 6,
});
assert_eq!(selection_to_text(&app).as_deref(), Some("a beta\ngam"));
}
#[test]
fn selection_to_text_removes_visual_wrap_breaks_from_paragraphs() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta gamma delta epsilon".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
14,
app.transcript_render_options(),
);
select_full_transcript(&mut app);
let selected = selection_to_text(&app).expect("selection text");
assert!(
!selected.contains('\n'),
"soft-wrapped paragraph copied with visual newlines: {selected:?}"
);
assert!(selected.contains("alpha beta gamma delta epsilon"));
}
#[test]
fn selection_to_text_preserves_wrapped_long_words() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "abcdefghijklmnop".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
10,
app.transcript_render_options(),
);
select_full_transcript(&mut app);
let selected = selection_to_text(&app).expect("selection text");
assert_eq!(selected, "abcdefghijklmnop");
}
#[test]
fn selection_to_text_strips_code_block_visual_wrap_prefixes() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "```\nlet example = abcdefghijklmnop;\n```".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
14,
app.transcript_render_options(),
);
select_full_transcript(&mut app);
let selected = selection_to_text(&app).expect("selection text");
assert_eq!(selected, "let example = abcdefghijklmnop;");
}
#[test]
fn selection_to_text_strips_list_continuation_prefixes() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "- alpha beta gamma delta epsilon".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
14,
app.transcript_render_options(),
);
select_full_transcript(&mut app);
let selected = selection_to_text(&app).expect("selection text");
assert_eq!(selected, "- alpha beta gamma delta epsilon");
}
#[test]
fn selection_to_text_copies_rendered_transcript_block() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::System {
content: "copy system".to_string(),
},
HistoryCell::User {
content: "copy user".to_string(),
},
HistoryCell::Thinking {
content: "copy thinking".to_string(),
streaming: false,
duration_secs: Some(1.0),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Success,
input_summary: Some("cargo check".to_string()),
output: Some("tool output line".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
HistoryCell::Assistant {
content: "copy assistant".to_string(),
streaming: false,
},
];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: app
.viewport
.transcript_cache
.total_lines()
.saturating_sub(1),
column: 80,
});
let selected = selection_to_text(&app).expect("selection text");
assert!(selected.contains("Note copy system"), "{selected:?}");
assert!(selected.contains("copy user"), "{selected:?}");
assert!(
selected.contains("copy thinking"),
"short completed thinking should be visible inline: {selected:?}"
);
assert!(
!selected.contains("Ctrl+O"),
"short completed thinking should not show the detail affordance: {selected:?}"
);
assert!(selected.contains("run done · cargo check"), "{selected:?}");
assert!(selected.contains("copy assistant"), "{selected:?}");
for (idx, line) in selected.lines().enumerate() {
assert!(
!line.starts_with("\u{2502} "),
"line {idx} retained tool-card rail prefix: {line:?}"
);
}
}
#[test]
fn selection_has_content_rejects_zero_width_selection() {
let mut app = create_test_app();
let point = TranscriptSelectionPoint {
line_index: 0,
column: 3,
};
app.viewport.transcript_selection.anchor = Some(point);
app.viewport.transcript_selection.head = Some(point);
assert!(!selection_has_content(&app));
}
#[test]
fn mouse_selection_autocopies_on_release_without_ctrl_c() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.last_transcript_top = 0;
app.viewport.last_transcript_total = app.viewport.transcript_cache.total_lines();
app.viewport.last_transcript_padding_top = 0;
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 0,
row: 0,
modifiers: KeyModifiers::NONE,
},
);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 8,
row: 0,
modifiers: KeyModifiers::NONE,
},
);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Up(MouseButton::Left),
column: 8,
row: 0,
modifiers: KeyModifiers::NONE,
},
);
assert_eq!(app.status_message.as_deref(), Some("Selection copied"));
assert!(
app.clipboard
.last_written_text()
.is_some_and(|text| text.contains("alpha")),
"selection should be written to clipboard"
);
}
#[test]
fn loading_mouse_filter_keeps_active_drags() {
let mut app = create_test_app();
app.is_loading = true;
let moved = MouseEvent {
kind: MouseEventKind::Moved,
column: 3,
row: 2,
modifiers: KeyModifiers::NONE,
};
let drag = MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 5,
row: 2,
modifiers: KeyModifiers::NONE,
};
assert!(should_drop_loading_mouse_motion(&app, moved));
assert!(should_drop_loading_mouse_motion(&app, drag));
app.viewport.transcript_selection.dragging = true;
assert!(!should_drop_loading_mouse_motion(&app, drag));
app.viewport.transcript_selection.dragging = false;
app.viewport.transcript_scrollbar_dragging = true;
assert!(!should_drop_loading_mouse_motion(&app, drag));
app.viewport.transcript_scrollbar_dragging = false;
app.sidebar_resizing = true;
assert!(!should_drop_loading_mouse_motion(&app, drag));
}
#[test]
fn loading_mouse_filter_allows_sidebar_resize_down_drag_up() {
let mut app = create_test_app();
app.is_loading = true;
setup_resize_handle(&mut app, 80, 33, 120);
let down = MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 80,
row: 5,
modifiers: KeyModifiers::NONE,
};
assert!(!should_drop_loading_mouse_motion(&app, down));
handle_mouse_event(&mut app, down);
assert!(app.sidebar_resizing, "down on handle starts resize");
let drag = MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 76,
row: 5,
modifiers: KeyModifiers::NONE,
};
assert!(
!should_drop_loading_mouse_motion(&app, drag),
"resize drag must not be dropped while loading"
);
handle_mouse_event(&mut app, drag);
let expected = ((37u32 * 100) / 120) as u16;
assert_eq!(app.sidebar_width_percent, expected);
let up = MouseEvent {
kind: MouseEventKind::Up(MouseButton::Left),
column: 76,
row: 5,
modifiers: KeyModifiers::NONE,
};
assert!(!should_drop_loading_mouse_motion(&app, up));
handle_mouse_event(&mut app, up);
assert!(!app.sidebar_resizing);
assert!(app.sidebar_width_dirty);
}
#[test]
fn loading_mouse_filter_allows_sidebar_hover_popovers() {
let mut app = create_test_app();
app.is_loading = true;
app.viewport.last_sidebar_area = Some(Rect::new(60, 4, 20, 6));
app.sidebar_hover.sections.push(SidebarHoverSection {
content_area: Rect::new(60, 4, 20, 6),
lines: vec!["Visible row".to_string()],
rows: vec![SidebarHoverRow {
row_y: 5,
display_text: "Truncated".to_string(),
full_text: "Full sidebar task label".to_string(),
detail: Some("Detailed context".to_string()),
is_truncated: true,
click_action: None,
stop_action: None,
stop_zone_start_col: None,
stop_zone_end_col: None,
}],
});
let moved = MouseEvent {
kind: MouseEventKind::Moved,
column: 65,
row: 5,
modifiers: KeyModifiers::NONE,
};
assert!(!should_drop_loading_mouse_motion(&app, moved));
handle_mouse_event(&mut app, moved);
assert_eq!(
app.sidebar_hover_tooltip.as_deref(),
Some("Full sidebar task label\nDetailed context")
);
assert_eq!(app.last_mouse_pos, Some((65, 5)));
}
#[test]
fn loading_mouse_filter_allows_sidebar_hover_to_clear() {
let mut app = create_test_app();
app.is_loading = true;
app.viewport.last_sidebar_area = Some(Rect::new(60, 4, 20, 6));
app.sidebar_hover_tooltip = Some("Stale sidebar tooltip".to_string());
let moved = MouseEvent {
kind: MouseEventKind::Moved,
column: 12,
row: 5,
modifiers: KeyModifiers::NONE,
};
assert!(!should_drop_loading_mouse_motion(&app, moved));
handle_mouse_event(&mut app, moved);
assert_eq!(app.sidebar_hover_tooltip, None);
assert_eq!(app.last_mouse_pos, Some((12, 5)));
}
#[test]
fn loading_mouse_filter_allows_sidebar_exit_to_clear_highlight() {
let mut app = create_test_app();
app.is_loading = true;
app.viewport.last_sidebar_area = Some(Rect::new(60, 4, 20, 6));
app.last_mouse_pos = Some((60, 5));
let exit_left = MouseEvent {
kind: MouseEventKind::Moved,
column: 59,
row: 5,
modifiers: KeyModifiers::NONE,
};
assert!(
!should_drop_loading_mouse_motion(&app, exit_left),
"first move out of the sidebar must clear stale sidebar hover state"
);
handle_mouse_event(&mut app, exit_left);
assert_eq!(app.last_mouse_pos, Some((59, 5)));
assert!(should_drop_loading_mouse_motion(
&app,
MouseEvent {
kind: MouseEventKind::Moved,
column: 58,
row: 5,
modifiers: KeyModifiers::NONE,
}
));
}
#[test]
fn jump_to_latest_button_click_scrolls_to_tail() {
let mut app = create_test_app();
app.viewport.transcript_scroll = TranscriptScroll::at_line(7);
app.viewport.jump_to_latest_button_area = Some(Rect {
x: 10,
y: 5,
width: 3,
height: 3,
});
app.user_scrolled_during_stream = true;
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 11,
row: 6,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert!(app.viewport.transcript_scroll.is_at_tail());
assert!(app.viewport.jump_to_latest_button_area.is_none());
assert!(!app.user_scrolled_during_stream);
assert!(!app.viewport.transcript_selection.dragging);
}
#[test]
fn transcript_scrollbar_gutter_starts_scrollbar_drag() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.last_transcript_area = Some(Rect {
x: 2,
y: 5,
width: 20,
height: 10,
});
app.viewport.last_transcript_visible = 10;
app.viewport.last_transcript_total = 110;
app.viewport.transcript_scroll = TranscriptScroll::to_bottom();
app.user_scrolled_during_stream = false;
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 21,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert!(
app.viewport.transcript_scrollbar_dragging,
"gutter click should start scrollbar drag"
);
assert!(
!app.viewport.transcript_selection.dragging,
"gutter click should NOT start text selection"
);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 21,
row: 14,
modifiers: KeyModifiers::NONE,
},
);
assert!(app.viewport.transcript_scrollbar_dragging);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Up(MouseButton::Left),
column: 21,
row: 14,
modifiers: KeyModifiers::NONE,
},
);
assert!(!app.viewport.transcript_scrollbar_dragging);
}
#[test]
fn left_down_inside_transcript_starts_selection() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.last_transcript_area = Some(Rect {
x: 2,
y: 5,
width: 20,
height: 10,
});
app.viewport.last_transcript_visible = 10;
app.viewport.last_transcript_total = 110;
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 3,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert!(app.viewport.transcript_selection.dragging);
}
#[test]
fn drag_below_viewport_arms_autoscroll_down() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.last_transcript_total = app.viewport.transcript_cache.total_lines();
app.viewport.transcript_selection.dragging = true;
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 4,
row: 12, modifiers: KeyModifiers::NONE,
},
);
let state = app.viewport.selection_autoscroll.expect("autoscroll armed");
assert_eq!(state.direction, 1);
assert_eq!(state.column, 4);
}
#[test]
fn drag_above_viewport_arms_autoscroll_up() {
let mut app = create_test_app();
app.viewport.last_transcript_area = Some(Rect {
x: 5,
y: 4,
width: 40,
height: 6,
});
app.viewport.transcript_selection.dragging = true;
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 5,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: 5,
column: 0,
});
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 50, row: 1, modifiers: KeyModifiers::NONE,
},
);
let state = app.viewport.selection_autoscroll.expect("autoscroll armed");
assert_eq!(state.direction, -1);
assert_eq!(state.column, 5 + 40 - 1);
}
#[test]
fn drag_back_inside_disarms_autoscroll() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.last_transcript_total = app.viewport.transcript_cache.total_lines();
app.viewport.transcript_selection.dragging = true;
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.selection_autoscroll = Some(SelectionAutoscroll {
direction: 1,
column: 4,
next_tick: Instant::now(),
});
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 6,
row: 0, modifiers: KeyModifiers::NONE,
},
);
assert!(app.viewport.selection_autoscroll.is_none());
let head = app
.viewport
.transcript_selection
.head
.expect("head present");
assert_eq!(head.column, 6);
}
#[test]
fn mouse_up_clears_selection_autoscroll() {
let mut app = create_test_app();
app.viewport.transcript_selection.dragging = true;
app.viewport.selection_autoscroll = Some(SelectionAutoscroll {
direction: -1,
column: 0,
next_tick: Instant::now(),
});
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Up(MouseButton::Left),
column: 0,
row: 0,
modifiers: KeyModifiers::NONE,
},
);
assert!(app.viewport.selection_autoscroll.is_none());
assert!(!app.viewport.transcript_selection.dragging);
}
#[test]
fn tick_selection_autoscroll_advances_pending_scroll_when_due() {
let mut app = create_test_app();
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.last_transcript_total = 200;
app.viewport.transcript_selection.dragging = true;
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
let earlier = Instant::now() - Duration::from_millis(100);
app.viewport.selection_autoscroll = Some(SelectionAutoscroll {
direction: 1,
column: 10,
next_tick: earlier,
});
tick_selection_autoscroll(&mut app);
assert_eq!(app.viewport.pending_scroll_delta, 1);
assert!(app.user_scrolled_during_stream);
let next_tick = app
.viewport
.selection_autoscroll
.expect("still armed")
.next_tick;
assert!(next_tick > earlier);
let head = app
.viewport
.transcript_selection
.head
.expect("head extended");
assert_eq!(head.line_index, 7);
assert_eq!(head.column, 10);
}
#[test]
fn tick_selection_autoscroll_respects_cadence() {
let mut app = create_test_app();
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.transcript_selection.dragging = true;
let future = Instant::now() + Duration::from_secs(60);
app.viewport.selection_autoscroll = Some(SelectionAutoscroll {
direction: 1,
column: 0,
next_tick: future,
});
tick_selection_autoscroll(&mut app);
assert_eq!(app.viewport.pending_scroll_delta, 0);
assert_eq!(
app.viewport
.selection_autoscroll
.expect("still armed")
.next_tick,
future,
"next_tick must not advance before its deadline"
);
}
#[test]
fn tick_selection_autoscroll_clears_when_drag_ended() {
let mut app = create_test_app();
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.transcript_selection.dragging = false;
app.viewport.selection_autoscroll = Some(SelectionAutoscroll {
direction: 1,
column: 0,
next_tick: Instant::now() - Duration::from_millis(100),
});
tick_selection_autoscroll(&mut app);
assert!(app.viewport.selection_autoscroll.is_none());
assert_eq!(app.viewport.pending_scroll_delta, 0);
}
#[test]
fn right_click_opens_context_menu() {
let mut app = create_test_app();
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Right),
column: 4,
row: 4,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::ContextMenu));
}
#[test]
fn right_click_menu_includes_selection_and_clicked_cell_actions() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Assistant {
content: "alpha beta".to_string(),
streaming: false,
}];
app.resync_history_revisions();
app.viewport.transcript_cache.ensure(
&app.history,
&app.history_revisions,
80,
app.transcript_render_options(),
);
app.viewport.last_transcript_area = Some(Rect {
x: 0,
y: 0,
width: 80,
height: 8,
});
app.viewport.last_transcript_top = 0;
app.viewport.last_transcript_total = app.viewport.transcript_cache.total_lines();
app.viewport.transcript_selection.anchor = Some(TranscriptSelectionPoint {
line_index: 0,
column: 0,
});
app.viewport.transcript_selection.head = Some(TranscriptSelectionPoint {
line_index: 0,
column: 5,
});
let entries = build_context_menu_entries(
&app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Right),
column: 2,
row: 0,
modifiers: KeyModifiers::NONE,
},
);
let labels = entries
.iter()
.map(|entry| entry.label.as_str())
.collect::<Vec<_>>();
assert!(labels.contains(&"Copy selection"));
assert!(labels.contains(&"Open selection"));
assert!(labels.contains(&"Open details"));
assert!(labels.contains(&"Paste"));
}
#[test]
fn mouse_events_do_not_mutate_transcript_behind_modal() {
let mut app = create_test_app();
app.view_stack.push(HelpView::new_for_locale(app.ui_locale));
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::ScrollUp,
column: 4,
row: 4,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert_eq!(app.viewport.pending_scroll_delta, 0);
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::Help));
}
#[test]
fn composer_mouse_wheel_scrolls_wrapped_draft_not_transcript() {
let mut app = create_test_app();
app.input = "alpha beta gamma delta epsilon".to_string();
app.cursor_position = 0;
app.viewport.last_composer_area = Some(Rect {
x: 0,
y: 10,
width: 12,
height: 5,
});
app.viewport.last_composer_content = Some(Rect {
x: 1,
y: 11,
width: 5,
height: 3,
});
let events = handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::ScrollDown,
column: 2,
row: 12,
modifiers: KeyModifiers::NONE,
},
);
assert!(events.is_empty());
assert_eq!(app.viewport.pending_scroll_delta, 0);
assert!(app.cursor_position > 0);
}
#[test]
fn composer_mouse_wheel_up_moves_within_wrapped_draft() {
let mut app = create_test_app();
app.input = "alpha beta gamma delta epsilon".to_string();
app.cursor_position = app.input.chars().count();
app.viewport.last_composer_area = Some(Rect {
x: 0,
y: 10,
width: 12,
height: 5,
});
app.viewport.last_composer_content = Some(Rect {
x: 1,
y: 11,
width: 5,
height: 3,
});
assert!(handle_composer_mouse(
&mut app,
MouseEvent {
kind: MouseEventKind::ScrollUp,
column: 2,
row: 12,
modifiers: KeyModifiers::NONE,
},
));
assert!(app.cursor_position < app.input.chars().count());
}
#[test]
fn composer_mouse_first_visible_character_maps_to_zero_after_prompt_gutter() {
let mut app = create_test_app();
app.input = "abcdef".to_string();
app.cursor_position = app.input.chars().count();
app.viewport.last_composer_area = Some(Rect {
x: 10,
y: 10,
width: 12,
height: 4,
});
app.viewport.last_composer_content = Some(Rect {
x: 10,
y: 11,
width: 12,
height: 3,
});
app.viewport.last_composer_scroll_offset = 0;
app.viewport.last_composer_top_padding = 0;
assert!(handle_composer_mouse(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 12,
row: 11,
modifiers: KeyModifiers::NONE,
},
));
assert_eq!(
app.cursor_position, 0,
"the first rendered character must not inherit the prompt's two-column inset"
);
assert!(handle_composer_mouse(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 13,
row: 11,
modifiers: KeyModifiers::NONE,
},
));
assert_eq!(app.cursor_position, 1);
}
#[test]
fn copy_shortcut_accepts_cmd_and_ctrl_shift_only() {
assert!(crate::tui::key_shortcuts::is_copy_shortcut(&KeyEvent::new(
KeyCode::Char('c'),
KeyModifiers::SUPER,
)));
assert!(crate::tui::key_shortcuts::is_copy_shortcut(&KeyEvent::new(
KeyCode::Char('c'),
KeyModifiers::CONTROL | KeyModifiers::SHIFT,
)));
assert!(!crate::tui::key_shortcuts::is_copy_shortcut(
&KeyEvent::new(KeyCode::Char('c'), KeyModifiers::CONTROL,)
));
}
#[test]
fn control_like_modifier_accepts_super_only_on_macos() {
use crate::tui::key_shortcuts::has_control_like_modifier_for_platform;
assert!(has_control_like_modifier_for_platform(
KeyModifiers::CONTROL,
false
));
assert!(has_control_like_modifier_for_platform(
KeyModifiers::CONTROL,
true
));
assert!(!has_control_like_modifier_for_platform(
KeyModifiers::SUPER,
false
));
assert!(has_control_like_modifier_for_platform(
KeyModifiers::SUPER,
true
));
assert!(has_control_like_modifier_for_platform(
KeyModifiers::SUPER | KeyModifiers::ALT,
true
));
}
#[test]
fn file_tree_shortcut_does_not_steal_plain_ctrl_e() {
assert!(!crate::tui::key_shortcuts::is_file_tree_toggle_shortcut(
&KeyEvent::new(KeyCode::Char('e'), KeyModifiers::CONTROL,)
));
assert!(crate::tui::key_shortcuts::is_file_tree_toggle_shortcut(
&KeyEvent::new(KeyCode::Char('E'), KeyModifiers::CONTROL,)
));
assert!(crate::tui::key_shortcuts::is_file_tree_toggle_shortcut(
&KeyEvent::new(
KeyCode::Char('e'),
KeyModifiers::CONTROL | KeyModifiers::SHIFT,
)
));
assert!(crate::tui::key_shortcuts::is_file_tree_toggle_shortcut(
&KeyEvent::new(
KeyCode::Char('E'),
KeyModifiers::SUPER | KeyModifiers::SHIFT,
)
));
}
#[test]
fn parse_plan_choice_accepts_numbers() {
assert_eq!(parse_plan_choice("1"), Some(PlanChoice::AcceptAgent));
assert_eq!(parse_plan_choice("2"), Some(PlanChoice::AcceptYolo));
assert_eq!(parse_plan_choice("3"), Some(PlanChoice::RevisePlan));
assert_eq!(parse_plan_choice("4"), Some(PlanChoice::ExitPlan));
}
#[test]
fn parse_plan_choice_rejects_aliases_and_extra_text() {
assert_eq!(parse_plan_choice("accept"), None);
assert_eq!(parse_plan_choice("agent"), None);
assert_eq!(parse_plan_choice("yolo"), None);
assert_eq!(parse_plan_choice("3 revise"), None);
assert_eq!(parse_plan_choice("unknown"), None);
}
#[test]
fn plan_choice_from_option_maps_expected_values() {
assert_eq!(plan_choice_from_option(1), Some(PlanChoice::AcceptAgent));
assert_eq!(plan_choice_from_option(2), Some(PlanChoice::AcceptYolo));
assert_eq!(plan_choice_from_option(3), Some(PlanChoice::RevisePlan));
assert_eq!(plan_choice_from_option(4), Some(PlanChoice::ExitPlan));
assert_eq!(plan_choice_from_option(5), None);
}
#[test]
fn plan_prompt_view_escape_emits_dismiss_event() {
let mut view = PlanPromptView::new(None);
let action = view.handle_key(KeyEvent::new(KeyCode::Esc, KeyModifiers::NONE));
assert!(matches!(
action,
ViewAction::EmitAndClose(ViewEvent::PlanPromptDismissed)
));
}
#[test]
fn transcript_scroll_percent_is_clamped_and_relative() {
assert_eq!(transcript_scroll_percent(0, 20, 120), Some(0));
assert_eq!(transcript_scroll_percent(50, 20, 120), Some(50));
assert_eq!(transcript_scroll_percent(200, 20, 120), Some(100));
assert_eq!(transcript_scroll_percent(0, 20, 20), None);
}
#[test]
fn parse_git_status_path_handles_simple_and_renamed_entries() {
assert_eq!(
crate::tui::file_picker_relevance::parse_git_status_path(" M crates/tui/src/tui/ui.rs"),
Some("crates/tui/src/tui/ui.rs".to_string())
);
assert_eq!(
crate::tui::file_picker_relevance::parse_git_status_path(
"R old name.rs -> crates/tui/src/tui/file_picker.rs"
),
Some("crates/tui/src/tui/file_picker.rs".to_string())
);
}
#[test]
fn workspace_file_candidate_normalizes_absolute_and_line_suffixed_paths() {
let dir = TempDir::new().expect("tempdir");
let root = dir.path();
std::fs::create_dir_all(root.join("src")).unwrap();
let path = root.join("src/lib.rs");
std::fs::write(&path, "").unwrap();
let raw = format!("\"{}:42\",", path.display());
assert_eq!(
crate::tui::file_picker_relevance::workspace_file_candidate(&raw, root),
Some("src/lib.rs".to_string())
);
}
#[test]
fn tool_path_relevance_extracts_paths_from_command_text() {
let dir = TempDir::new().expect("tempdir");
let root = dir.path();
std::fs::create_dir_all(root.join("src")).unwrap();
std::fs::write(root.join("src/alpha.rs"), "").unwrap();
std::fs::write(root.join("src/zeta.rs"), "").unwrap();
let mut relevance = crate::tui::file_picker::FilePickerRelevance::default();
let mut seen = HashSet::new();
let mut budget = 16;
crate::tui::file_picker_relevance::mark_tool_paths_from_text(
"sed -n '1,20p' src/zeta.rs",
root,
&mut seen,
&mut relevance,
&mut budget,
);
let view = crate::tui::file_picker::FilePickerView::new_with_relevance(root, relevance);
assert_eq!(view.selected_for_test(), Some("src/zeta.rs"));
}
fn create_test_app() -> App {
let options = TuiOptions {
model: "deepseek-v4-pro".to_string(),
workspace: PathBuf::from("."),
config_path: None,
config_profile: None,
allow_shell: false,
use_alt_screen: true,
use_mouse_capture: false,
use_bracketed_paste: true,
max_subagents: 1,
skills_dir: PathBuf::from("."),
memory_path: PathBuf::from("memory.md"),
notes_path: PathBuf::from("notes.txt"),
mcp_config_path: PathBuf::from("mcp.json"),
use_memory: false,
start_in_agent_mode: true,
skip_onboarding: false,
yolo: false,
resume_session_id: None,
initial_input: None,
};
let mut app = App::new(options, &Config::default());
app.cost_currency = crate::pricing::CostCurrency::Usd;
app.ui_locale = crate::localization::Locale::En;
app.show_thinking = true;
app.set_provider_identity(crate::config::ApiProvider::Deepseek, "deepseek");
app.billing_presentation = crate::route_billing::BillingPresentation::Metered;
app.model = "deepseek-v4-pro".to_string();
app.auto_model = false;
app.last_effective_model = None;
app.active_route_limits = None;
app.active_context_window_override = None;
app.composer.mention_cwd = None;
app
}
#[test]
fn hotbar_setup_save_persists_bindings_to_config_path() {
let tmp = TempDir::new().expect("config tempdir");
let config_path = tmp.path().join("config.toml");
std::fs::write(
&config_path,
r#"# keep model note
model = "deepseek-v4-pro"
[providers.deepseek]
api_key = "test-key"
"#,
)
.expect("write config");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
let mut config = Config::load(Some(config_path.clone()), None).expect("load config");
let bindings = vec![codewhale_config::HotbarBindingToml {
slot: 1,
action: "mode.agent".to_string(),
label: Some("Agent".to_string()),
}];
apply_hotbar_setup_saved(&mut app, &mut config, bindings.clone());
assert_eq!(config.hotbar, Some(bindings.clone()));
assert!(app.needs_redraw);
assert!(
app.status_message
.as_deref()
.is_some_and(|message| message.contains("Hotbar bindings saved to"))
);
let body = std::fs::read_to_string(&config_path).expect("read saved config");
assert!(body.contains("# keep model note"), "comment lost: {body}");
assert!(
body.contains("[providers.deepseek]"),
"provider section lost: {body}"
);
assert!(body.contains("[[hotbar]]"), "hotbar table missing: {body}");
let parsed: codewhale_config::ConfigToml =
toml::from_str(&body).expect("saved config should parse");
assert_eq!(parsed.hotbar, Some(bindings));
}
#[test]
fn hotbar_setup_save_error_leaves_live_config_and_file_unchanged() {
let tmp = TempDir::new().expect("config tempdir");
let config_path = tmp.path().join("config.toml");
let invalid_body = "model = [\n";
std::fs::write(&config_path, invalid_body).expect("write malformed config");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
let original_bindings = vec![codewhale_config::HotbarBindingToml {
slot: 2,
action: "mode.plan".to_string(),
label: None,
}];
let mut config = Config {
hotbar: Some(original_bindings.clone()),
..Default::default()
};
let attempted_bindings = vec![codewhale_config::HotbarBindingToml {
slot: 1,
action: "mode.agent".to_string(),
label: None,
}];
apply_hotbar_setup_saved(&mut app, &mut config, attempted_bindings);
assert_eq!(config.hotbar, Some(original_bindings));
assert_eq!(
std::fs::read_to_string(&config_path).expect("read malformed config"),
invalid_body
);
assert!(
app.status_message
.as_deref()
.is_some_and(|message| message.contains("Failed to save Hotbar bindings"))
);
assert!(app.needs_redraw);
let last_system_message = app
.history
.iter()
.rev()
.find_map(|cell| match cell {
HistoryCell::System { content } => Some(content.as_str()),
_ => None,
})
.expect("failed save should add a system message");
assert!(last_system_message.contains("Failed to save Hotbar bindings"));
}
#[test]
fn app_system_prompt_includes_configured_instructions() {
let tmp = tempfile::tempdir().expect("tempdir");
let instructions = tmp.path().join("extra-instructions.md");
std::fs::write(&instructions, "CONFIGURED_INSTRUCTIONS_MARKER").expect("write instructions");
let mut app = create_test_app();
app.workspace = tmp.path().to_path_buf();
let config = Config {
instructions: Some(vec![instructions.display().to_string()]),
..Config::default()
};
let prompt = crate::prompts::system_prompt_flat_text(&build_app_system_prompt(&app, &config));
assert!(prompt.contains("CONFIGURED_INSTRUCTIONS_MARKER"));
assert!(prompt.contains(&instructions.display().to_string()));
}
#[test]
fn session_denied_cache_matches_only_approval_key() {
let mut app = create_test_app();
app.approval_session_denied.insert("edit_file".to_string());
assert!(
!is_session_denied_for_key(&app, "file:edit_file:fresh"),
"a legacy tool-name entry must not deny a later fresh call"
);
app.approval_session_denied
.insert("file:edit_file:retry".to_string());
assert!(is_session_denied_for_key(&app, "file:edit_file:retry"));
}
fn render_underwater_test_app(app: &mut App, width: u16, height: u16) -> String {
app.onboarding_workspace_trust_gate = false;
app.onboarding = OnboardingState::None;
let config = Config::default();
let mut terminal =
Terminal::new(TestBackend::new(width, height)).expect("underwater test terminal");
terminal
.draw(|frame| render(frame, app, &config))
.expect("render underwater shell");
terminal
.backend()
.buffer()
.content()
.iter()
.map(|cell| cell.symbol())
.collect::<String>()
}
#[tokio::test]
async fn session_denied_cache_auto_deny_explains_the_cached_rejection() {
let home = SettingsHomeGuard::new();
let audit_path = home._tmp.path().join(".codewhale").join("audit.log");
let mut app = create_test_app();
let mut engine = mock_engine_handle();
let approval_key = "shell:exec_shell:git push secret-token";
auto_deny_session_approval(
&mut app,
&engine.handle,
"tool-retry",
"exec_shell",
approval_key,
)
.await;
assert_eq!(
engine.recv_approval_event().await,
Some(crate::core::engine::MockApprovalEvent::Denied {
id: "tool-retry".to_string(),
})
);
let toast = app.status_toasts.back().expect("auto-deny warning toast");
assert_eq!(toast.level, StatusToastLevel::Warning);
assert_eq!(toast.ttl_ms, Some(12_000));
assert!(toast.text.contains("matching request was denied earlier"));
assert!(toast.text.contains("during this CodeWhale run"));
assert!(toast.text.contains("Restart CodeWhale"));
assert!(toast.text.contains("exec_shell"));
let history_notice = app
.history
.iter()
.rev()
.find_map(|cell| match cell {
HistoryCell::System { content } => Some(content.as_str()),
_ => None,
})
.expect("persistent auto-deny explanation");
assert_eq!(history_notice, toast.text);
for visible_text in [&toast.text, history_notice] {
assert!(
!visible_text.contains(approval_key)
&& !visible_text.contains("git push")
&& !visible_text.contains("secret-token"),
"the user-facing notice must not expose the approval key, command, or arguments"
);
}
let audit = std::fs::read_to_string(&audit_path).expect("isolated approval audit log");
assert!(audit.contains(approval_key));
let rendered = render_underwater_test_app(&mut app, 40, 12);
assert!(rendered.contains("Auto-denied"), "{rendered:?}");
assert!(
rendered.contains("Restart") && rendered.contains("CodeWhale"),
"{rendered:?}"
);
}
#[tokio::test]
async fn session_denied_cache_notice_preserves_active_tool_index() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
let mut engine = mock_engine_handle();
let tool_id = "tool-retry";
let tool_name = "exec_shell";
handle_tool_call_started(
&mut app,
tool_id,
tool_name,
&serde_json::json!({"command": "git push secret-token"}),
);
let history_len = app.history.len();
let tool_index = app.tool_cells[tool_id];
assert_eq!(tool_index, history_len);
auto_deny_session_approval(
&mut app,
&engine.handle,
tool_id,
tool_name,
"shell:exec_shell:git push secret-token",
)
.await;
assert_eq!(
app.history.len(),
history_len,
"the notice must not shift virtual indices while a tool is active"
);
assert_eq!(app.tool_cells.get(tool_id), Some(&tool_index));
let active = app.active_cell.as_ref().expect("active tool and notice");
assert!(matches!(
active.entries().first(),
Some(HistoryCell::Tool(ToolCell::Exec(exec))) if exec.status == ToolStatus::Running
));
assert!(matches!(
active.entries().get(1),
Some(HistoryCell::System { content }) if content.contains("Auto-denied")
));
assert_eq!(
engine.recv_approval_event().await,
Some(crate::core::engine::MockApprovalEvent::Denied {
id: tool_id.to_string(),
})
);
let denied_result = Ok(crate::tools::spec::ToolResult::error(
"request denied by cached approval",
));
handle_tool_call_complete(&mut app, tool_id, tool_name, &denied_result);
let active = app.active_cell.as_ref().expect("completed tool and notice");
assert!(matches!(
active.entries().first(),
Some(HistoryCell::Tool(ToolCell::Exec(exec))) if exec.status == ToolStatus::Failed
));
assert!(matches!(
active.entries().get(1),
Some(HistoryCell::System { .. })
));
app.flush_active_cell();
assert!(matches!(
app.history.get(history_len),
Some(HistoryCell::Tool(ToolCell::Exec(exec))) if exec.status == ToolStatus::Failed
));
assert!(matches!(
app.history.get(history_len + 1),
Some(HistoryCell::System { content }) if content.contains("Auto-denied")
));
let detail = app
.tool_detail_record_for_cell(history_len)
.expect("tool detail remains bound to the completed tool cell");
assert_eq!(detail.tool_id, tool_id);
assert!(
detail
.output
.as_deref()
.is_some_and(|output| output.contains("cached approval"))
);
}
#[tokio::test]
async fn session_denied_cache_notice_preserves_parallel_tool_indices() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
let mut engine = mock_engine_handle();
handle_tool_call_started(
&mut app,
"tool-first",
"exec_shell",
&serde_json::json!({"command": "echo first"}),
);
handle_tool_call_started(
&mut app,
"tool-denied",
"exec_shell",
&serde_json::json!({"command": "git push secret-token"}),
);
let history_len = app.history.len();
let first_index = app.tool_cells["tool-first"];
let denied_index = app.tool_cells["tool-denied"];
assert_eq!(first_index, history_len);
assert_eq!(denied_index, history_len + 1);
auto_deny_session_approval(
&mut app,
&engine.handle,
"tool-denied",
"exec_shell",
"shell:exec_shell:git push secret-token",
)
.await;
assert_eq!(app.history.len(), history_len);
assert_eq!(app.tool_cells.get("tool-first"), Some(&first_index));
assert_eq!(app.tool_cells.get("tool-denied"), Some(&denied_index));
assert_eq!(
engine.recv_approval_event().await,
Some(crate::core::engine::MockApprovalEvent::Denied {
id: "tool-denied".to_string(),
})
);
let denied_result = Ok(crate::tools::spec::ToolResult::error(
"request denied by cached approval",
));
handle_tool_call_complete(&mut app, "tool-denied", "exec_shell", &denied_result);
let active = app.active_cell.as_ref().expect("parallel tools and notice");
assert!(matches!(
active.entries().first(),
Some(HistoryCell::Tool(ToolCell::Exec(exec)))
if exec.status == ToolStatus::Running && exec.command == "echo first"
));
assert!(matches!(
active.entries().get(1),
Some(HistoryCell::Tool(ToolCell::Exec(exec)))
if exec.status == ToolStatus::Failed
&& exec.command == "git push secret-token"
&& exec.output.as_deref().is_some_and(|output| output.contains("cached approval"))
));
assert!(matches!(
active.entries().get(2),
Some(HistoryCell::System { content }) if content.contains("Auto-denied")
));
handle_tool_call_complete(
&mut app,
"tool-first",
"exec_shell",
&ok_result("first output"),
);
app.flush_active_cell();
assert!(matches!(
app.history.get(history_len),
Some(HistoryCell::Tool(ToolCell::Exec(exec)))
if exec.status == ToolStatus::Success
&& exec.command == "echo first"
&& exec.output.as_deref() == Some("first output")
));
assert!(matches!(
app.history.get(history_len + 1),
Some(HistoryCell::Tool(ToolCell::Exec(exec)))
if exec.status == ToolStatus::Failed && exec.command == "git push secret-token"
));
assert!(matches!(
app.history.get(history_len + 2),
Some(HistoryCell::System { content }) if content.contains("Auto-denied")
));
}
#[tokio::test]
async fn session_denied_cache_notice_renders_host_scope_in_zh_hans() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.ui_locale = crate::localization::Locale::ZhHans;
let mut engine = mock_engine_handle();
auto_deny_session_approval(
&mut app,
&engine.handle,
"fetch-retry",
"fetch_url",
"net:example.com",
)
.await;
assert_eq!(
engine.recv_approval_event().await,
Some(crate::core::engine::MockApprovalEvent::Denied {
id: "fetch-retry".to_string(),
})
);
let notice = app
.history
.iter()
.rev()
.find_map(|cell| match cell {
HistoryCell::System { content } => Some(content.as_str()),
_ => None,
})
.expect("localized persistent auto-deny explanation");
assert!(notice.contains("本次 CodeWhale 运行期间"));
assert!(notice.contains("匹配请求"));
assert!(!notice.contains("example.com"));
let rendered = render_underwater_test_app(&mut app, 60, 16);
let rendered_compact = rendered
.chars()
.filter(|ch| !ch.is_whitespace())
.collect::<String>();
assert!(rendered_compact.contains("已自动拒绝"), "{rendered:?}");
assert!(rendered_compact.contains("匹配请求"), "{rendered:?}");
assert!(
rendered_compact.contains("重启") && rendered_compact.contains("CodeWhale"),
"{rendered:?}"
);
}
#[test]
fn session_denied_notice_explains_cached_decision_and_recovery() {
let app = create_test_app();
let notice = session_denied_notice(&app, "exec_shell");
assert!(notice.contains("exec_shell"));
assert!(notice.contains("matching request was denied earlier"));
assert!(notice.contains("during this CodeWhale run"));
assert!(notice.contains("Restart CodeWhale"));
}
#[tokio::test]
async fn cached_denial_explanation_survives_tool_completion_and_done_render() {
use crate::core::engine::MockApprovalEvent;
use crate::tools::spec::ToolError;
use crate::tui::ocean::OceanTreatment;
use ratatui::{Terminal, backend::TestBackend};
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
app.launch.visible = false;
app.ocean_treatment = OceanTreatment::Ombre;
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
let tool_id = "cached-shell-denial";
let tool_name = "exec_shell";
handle_tool_call_started(
&mut app,
tool_id,
tool_name,
&serde_json::json!({"command": "printf cached-denial"}),
);
let mut engine = mock_engine_handle();
engine
.handle
.deny_tool_call(tool_id)
.await
.expect("send cached denial");
surface_session_denied_notice(&mut app, tool_name);
assert!(matches!(
engine.recv_approval_event().await,
Some(MockApprovalEvent::Denied { id }) if id == tool_id
));
let result = Err(ToolError::permission_denied(
"Tool 'exec_shell' denied by user".to_string(),
));
handle_tool_call_complete(&mut app, tool_id, tool_name, &result);
app.flush_active_cell();
let denied_tool_index = app
.history
.iter()
.position(|cell| {
matches!(
cell,
HistoryCell::Tool(ToolCell::Exec(exec))
if exec.command == "printf cached-denial"
&& exec.status == ToolStatus::Failed
)
})
.expect("cached denial must settle the pending tool as failed");
let recovery_receipt_index = app
.history
.iter()
.position(|cell| {
matches!(
cell,
HistoryCell::System { content }
if content.contains("Auto-denied exec_shell")
&& content.contains("Restart CodeWhale")
)
})
.expect("cached denial must leave a durable recovery receipt");
assert!(
denied_tool_index < recovery_receipt_index,
"recovery receipt must follow the tool it explains"
);
app.is_loading = false;
app.runtime_turn_status = Some("completed".to_string());
app.status_message = Some("Tool 'exec_shell' denied by user".to_string());
app.sync_status_message_to_toasts();
let config = Config::default();
let backend = TestBackend::new(89, 24);
let mut terminal = Terminal::new(backend).expect("test terminal");
terminal
.draw(|frame| render(frame, &mut app, &config))
.expect("render completed denial sequence");
let buffer = terminal.backend().buffer();
let rendered = (0..buffer.area.height)
.map(|y| {
(0..buffer.area.width)
.map(|x| buffer[(x, y)].symbol())
.collect::<String>()
})
.collect::<Vec<_>>()
.join("\n");
assert!(
rendered.contains("Auto-denied exec_shell"),
"cached-decision explanation disappeared after completion:\n{rendered}"
);
assert!(
rendered.contains("Restart CodeWhale"),
"cached-denial recovery path disappeared after completion:\n{rendered}"
);
assert_eq!(
app.view_stack.top_kind(),
None,
"cache hit must not re-prompt"
);
}
#[test]
fn session_approved_cache_keeps_tool_name_session_grants() {
let mut app = create_test_app();
app.approval_session_approved
.insert("edit_file".to_string());
assert!(
is_session_approved_for_tool(&app, "edit_file", "file:edit_file:fresh"),
"approve-for-session should still cover future calls of the same tool"
);
}
#[test]
fn forced_approval_prompt_bypasses_auto_mode_shortcut() {
let mut app = create_test_app();
app.approval_mode = ApprovalMode::Auto;
assert!(!should_auto_approve_approval_request(
&app,
"exec_shell",
"shell:exec_shell:cargo test",
true,
));
}
#[test]
fn forced_approval_prompt_bypasses_session_approval_shortcut() {
let mut app = create_test_app();
app.approval_session_approved
.insert("shell:exec_shell:cargo test".to_string());
assert!(!should_auto_approve_approval_request(
&app,
"exec_shell",
"shell:exec_shell:cargo test",
true,
));
}
#[test]
fn approval_request_uses_session_cache_not_current_mode_shortcut() {
let mut app = create_test_app();
app.approval_mode = ApprovalMode::Auto;
assert!(!should_auto_approve_approval_request(
&app,
"exec_shell",
"shell:exec_shell:cargo test",
false,
));
app.approval_mode = ApprovalMode::Bypass;
assert!(!should_auto_approve_approval_request(
&app,
"exec_shell",
"shell:exec_shell:cargo test",
false,
));
app.approval_mode = ApprovalMode::Suggest;
app.mode = AppMode::Yolo;
assert!(!should_auto_approve_approval_request(
&app,
"exec_shell",
"shell:exec_shell:cargo test",
false,
));
app.mode = AppMode::Agent;
app.approval_session_approved
.insert("shell:exec_shell:cargo test".to_string());
assert!(should_auto_approve_approval_request(
&app,
"exec_shell",
"shell:exec_shell:cargo test",
false,
));
}
#[test]
fn app_auto_approval_helper_covers_yolo_and_bypass_only() {
let mut app = create_test_app();
app.mode = AppMode::Agent;
app.approval_mode = ApprovalMode::Suggest;
assert!(!app_auto_approve_enabled(&app));
app.approval_mode = ApprovalMode::Auto;
assert!(!app_auto_approve_enabled(&app));
app.approval_mode = ApprovalMode::Bypass;
assert!(app_auto_approve_enabled(&app));
app.approval_mode = ApprovalMode::Suggest;
app.mode = AppMode::Yolo;
assert!(app_auto_approve_enabled(&app));
}
fn create_test_options() -> TuiOptions {
TuiOptions {
model: "deepseek-v4-pro".to_string(),
workspace: PathBuf::from("."),
config_path: None,
config_profile: None,
allow_shell: false,
use_alt_screen: true,
use_mouse_capture: false,
use_bracketed_paste: true,
max_subagents: 1,
skills_dir: PathBuf::from("."),
memory_path: PathBuf::from("memory.md"),
notes_path: PathBuf::from("notes.txt"),
mcp_config_path: PathBuf::from("mcp.json"),
use_memory: false,
start_in_agent_mode: true,
skip_onboarding: false,
yolo: false,
resume_session_id: None,
initial_input: None,
}
}
#[test]
fn setup_checkpoint_opens_after_onboarding_when_due() {
let _home = SettingsHomeGuard::new();
let config = Config::default();
let mut app = App::new(create_test_options(), &config);
app.onboarding = OnboardingState::None;
assert!(open_setup_checkpoint_if_due(&mut app, &config, false));
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::SetupWizard));
assert!(
!open_setup_checkpoint_if_due(&mut app, &config, false),
"setup wizard should not be stacked twice"
);
}
#[test]
fn setup_checkpoint_waits_for_onboarding_and_skip_flag() {
let _home = SettingsHomeGuard::new();
let config = Config::default();
let mut app = App::new(create_test_options(), &config);
app.onboarding = OnboardingState::Tips;
assert!(!open_setup_checkpoint_if_due(&mut app, &config, false));
assert!(app.view_stack.is_empty());
app.onboarding = OnboardingState::None;
assert!(!open_setup_checkpoint_if_due(&mut app, &config, true));
assert!(app.view_stack.is_empty());
let state = codewhale_config::SetupState::load()
.expect("load setup state")
.expect("setup state");
assert_eq!(
state.constitution_checkpoint_completed_for.as_deref(),
Some(crate::tui::setup::CONSTITUTION_CHECKPOINT_VERSION)
);
assert_eq!(
state.constitution_choice,
codewhale_config::ConstitutionChoice::Deferred
);
assert_eq!(
state.status(codewhale_config::SetupStep::Constitution),
codewhale_config::StepStatus::Deferred
);
assert!(
!open_setup_checkpoint_if_due(&mut app, &config, false),
"deferred skip record should suppress the update checkpoint on the next launch"
);
}
#[test]
fn setup_runtime_preset_apply_persists_settings_config_and_state() {
let _home = SettingsHomeGuard::new();
let config_path = crate::config_persistence::config_toml_path(None).expect("config path");
std::fs::create_dir_all(config_path.parent().expect("config parent"))
.expect("config parent exists");
std::fs::write(
&config_path,
"# preserve this comment\nmodel = \"deepseek-v4-pro\"\n",
)
.expect("seed config");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
let mut config = Config::default();
let preset = crate::tui::setup::SetupRuntimePreset::AskFirst;
let mut state = codewhale_config::SetupState {
runtime_posture_source: codewhale_config::RuntimePostureSource::Confirmed,
..Default::default()
};
state.set_step(
codewhale_config::SetupStep::TrustSandbox,
codewhale_config::StepEntry::new(
codewhale_config::StepStatus::Verified,
true,
crate::tui::setup::CONSTITUTION_CHECKPOINT_VERSION,
)
.with_result(preset.result_summary()),
);
let summary =
apply_setup_runtime_preset(&mut app, &mut config, preset, state).expect("apply preset");
assert!(summary.contains("preset=ask-first"));
let settings = Settings::load().expect("load saved settings");
assert_eq!(settings.default_mode, "plan");
assert_eq!(settings.permission_posture.as_deref(), Some("ask"));
assert_eq!(app.mode, AppMode::Plan);
assert!(!app.allow_shell);
assert_eq!(app.approval_mode, ApprovalMode::Suggest);
assert_eq!(config.allow_shell, Some(false));
assert_eq!(config.approval_policy.as_deref(), Some("on-request"));
assert_eq!(config.sandbox_mode.as_deref(), Some("read-only"));
let body = std::fs::read_to_string(&config_path).expect("read saved config");
assert!(
body.contains("# preserve this comment"),
"comment lost: {body}"
);
assert!(body.contains("approval_policy = \"on-request\""));
assert!(body.contains("allow_shell = false"));
assert!(body.contains("sandbox_mode = \"read-only\""));
let saved_state = codewhale_config::SetupState::load()
.expect("load setup state")
.expect("setup state exists");
assert_eq!(
saved_state.status(codewhale_config::SetupStep::TrustSandbox),
codewhale_config::StepStatus::Verified
);
assert_eq!(
saved_state.runtime_posture_source,
codewhale_config::RuntimePostureSource::Confirmed
);
}
#[test]
fn setup_runtime_preset_rolls_back_durable_and_live_state_when_state_save_fails() {
let _home = SettingsHomeGuard::new();
let config_path = crate::config_persistence::config_toml_path(None).expect("config path");
std::fs::create_dir_all(config_path.parent().expect("config parent"))
.expect("config parent exists");
std::fs::write(
&config_path,
"# keep exact bytes\napproval_policy = \"auto\"\nallow_shell = true\nsandbox_mode = \"workspace-write\"\n",
)
.expect("seed config");
let original_settings = Settings {
default_mode: "agent".to_string(),
permission_posture: Some("auto-review".to_string()),
..Settings::default()
};
original_settings.save().expect("seed settings");
let config_before = std::fs::read(&config_path).expect("read config snapshot");
let settings_path = Settings::path().expect("settings path");
let settings_before = std::fs::read(&settings_path).expect("read settings snapshot");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
app.set_agent_runtime_baseline(true, false, ApprovalMode::Auto);
app.set_mode(AppMode::Agent);
let app_before = (
app.mode,
app.allow_shell,
app.trust_mode,
app.approval_mode,
app.approval_policy_locked(),
);
let mut config = Config {
approval_policy: Some("auto".to_string()),
allow_shell: Some(true),
sandbox_mode: Some("workspace-write".to_string()),
..Config::default()
};
let config_before_live = (
config.approval_policy.clone(),
config.allow_shell,
config.sandbox_mode.clone(),
);
let state_path = codewhale_config::SetupState::path().expect("state path");
std::fs::create_dir_all(&state_path).expect("state path directory");
let error = apply_setup_runtime_preset(
&mut app,
&mut config,
crate::tui::setup::SetupRuntimePreset::AskFirst,
codewhale_config::SetupState::default(),
)
.expect_err("state persistence must fail");
assert!(
error
.to_string()
.contains("failed to persist setup runtime posture state"),
"{error:#}"
);
assert_eq!(std::fs::read(config_path).unwrap(), config_before);
assert_eq!(std::fs::read(settings_path).unwrap(), settings_before);
assert_eq!(
(
app.mode,
app.allow_shell,
app.trust_mode,
app.approval_mode,
app.approval_policy_locked(),
),
app_before
);
assert_eq!(
(
config.approval_policy,
config.allow_shell,
config.sandbox_mode,
),
config_before_live
);
}
#[test]
fn setup_high_trust_persists_full_access_without_legacy_yolo_mode() {
let _home = SettingsHomeGuard::new();
let config_path = crate::config_persistence::config_toml_path(None).expect("config path");
std::fs::create_dir_all(config_path.parent().expect("config parent"))
.expect("config parent exists");
std::fs::write(
&config_path,
"# keep me\napproval_policy = \"on-request\"\nallow_shell = false\n",
)
.expect("seed config");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
let mut config = Config {
approval_policy: Some("on-request".to_string()),
..Config::default()
};
let preset = crate::tui::setup::SetupRuntimePreset::HighTrustLocal;
apply_setup_runtime_preset(
&mut app,
&mut config,
preset,
codewhale_config::SetupState::default(),
)
.expect("apply high trust");
let settings = Settings::load().expect("load saved settings");
assert_eq!(settings.default_mode, "agent");
assert_eq!(settings.permission_posture.as_deref(), Some("full-access"));
assert_eq!(config.approval_policy, None);
assert_eq!(app.mode, AppMode::Agent);
assert_eq!(app.approval_mode, ApprovalMode::Bypass);
assert!(app.allow_shell);
assert!(app.trust_mode);
app.set_mode(AppMode::Plan);
assert!(!app.allow_shell);
assert_eq!(app.approval_mode, ApprovalMode::Suggest);
app.set_mode(AppMode::Agent);
assert!(app.allow_shell, "High Trust shell must survive Plan → Act");
assert!(app.trust_mode, "High Trust trust must survive Plan → Act");
assert_eq!(app.approval_mode, ApprovalMode::Bypass);
let body = std::fs::read_to_string(&config_path).expect("read saved config");
assert!(body.contains("# keep me"), "comment lost: {body}");
assert!(
!body.contains("approval_policy"),
"top-level policy would override the saved Full Access posture: {body}"
);
}
#[test]
fn setup_high_trust_cannot_override_project_approval_policy() {
let _home = SettingsHomeGuard::new();
let config_path = crate::config_persistence::config_toml_path(None).expect("config path");
std::fs::create_dir_all(config_path.parent().expect("config parent"))
.expect("config parent exists");
std::fs::write(&config_path, "approval_policy = \"on-request\"\n").expect("root config");
let workspace = config_path
.parent()
.and_then(std::path::Path::parent)
.expect("temporary home")
.join("project");
let project_dir = workspace.join(codewhale_config::CODEWHALE_APP_DIR);
std::fs::create_dir_all(&project_dir).expect("project config dir");
std::fs::write(
project_dir.join("config.toml"),
"approval_policy = \"never\"\n",
)
.expect("project config");
let mut app = create_test_app();
app.workspace = workspace;
app.config_path = Some(config_path.clone());
let mut config = Config {
approval_policy: Some("never".to_string()),
..Config::default()
};
let error = apply_setup_runtime_preset(
&mut app,
&mut config,
crate::tui::setup::SetupRuntimePreset::HighTrustLocal,
codewhale_config::SetupState::default(),
)
.expect_err("project policy must control the live session");
assert!(
error.to_string().contains("project runtime configuration"),
"{error:#}"
);
assert_eq!(config.approval_policy.as_deref(), Some("never"));
assert!(
std::fs::read_to_string(config_path)
.expect("root config remains")
.contains("approval_policy = \"on-request\"")
);
}
#[test]
fn project_runtime_provenance_only_blocks_values_startup_would_accept() {
let _home = SettingsHomeGuard::new();
let settings = Settings {
permission_posture: Some("ask".to_string()),
..Settings::default()
};
settings.save().expect("save settings");
let workspace = Settings::path()
.expect("settings path")
.parent()
.expect("Codewhale home")
.parent()
.expect("temporary home")
.join("project");
let project_dir = workspace.join(codewhale_config::CODEWHALE_APP_DIR);
std::fs::create_dir_all(&project_dir).expect("project config dir");
let project_path = project_dir.join("config.toml");
std::fs::write(
&project_path,
"approval_policy = \"auto\"\nsandbox_mode = \"danger-full-access\"\nallow_shell = true\n",
)
.expect("project config");
let config = Config {
sandbox_mode: Some("read-only".to_string()),
..Config::default()
};
assert_eq!(
config.approval_policy_control(None, None, &workspace),
crate::config::ApprovalPolicyControl::Unset,
"project Auto would loosen saved Ask and must not claim provenance"
);
assert_eq!(
config.allow_shell_control(None, None, &workspace),
crate::config::ShellAccessControl::Unset,
"project allow_shell=true is ignored by startup"
);
assert_eq!(
config.runtime_preset_blocker(None, None, &workspace),
None,
"ignored project escalations must not create a phantom preset blocker"
);
std::fs::write(
&project_path,
"approval_policy = \"never\"\nsandbox_mode = \"read-only\"\nallow_shell = false\n",
)
.expect("tightening project config");
assert_eq!(
config.approval_policy_control(None, None, &workspace),
crate::config::ApprovalPolicyControl::ProjectConfig
);
assert_eq!(
config.allow_shell_control(None, None, &workspace),
crate::config::ShellAccessControl::ProjectConfig
);
assert_eq!(
config.runtime_preset_blocker(None, None, &workspace),
Some("project runtime configuration")
);
}
#[test]
fn saved_full_access_baseline_allows_project_approval_tightening() {
let _home = SettingsHomeGuard::new();
let settings = Settings {
permission_posture: Some("full-access".to_string()),
..Settings::default()
};
settings.save().expect("save settings");
let workspace = Settings::path()
.expect("settings path")
.parent()
.expect("Codewhale home")
.parent()
.expect("temporary home")
.join("project");
let project_dir = workspace.join(codewhale_config::CODEWHALE_APP_DIR);
std::fs::create_dir_all(&project_dir).expect("project config dir");
std::fs::write(
project_dir.join("config.toml"),
"approval_policy = \"on-request\"\n",
)
.expect("project config");
assert_eq!(
Config::default().approval_policy_control(None, None, &workspace),
crate::config::ApprovalPolicyControl::ProjectConfig,
"Ask is a tightening from saved Full Access and owns the effective policy"
);
}
#[test]
fn setup_presets_cannot_override_managed_runtime_requirements() {
let _home = SettingsHomeGuard::new();
let config_path = crate::config_persistence::config_toml_path(None).expect("config path");
std::fs::create_dir_all(config_path.parent().expect("config parent"))
.expect("config parent exists");
std::fs::write(&config_path, "# managed posture stays intact\n").expect("root config");
let requirements_path = config_path
.parent()
.expect("config parent")
.join("requirements.toml");
std::fs::write(
&requirements_path,
"allowed_approval_policies = [\"never\"]\nallowed_sandbox_modes = [\"read-only\"]\n",
)
.expect("requirements");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
app.allow_shell = false;
app.approval_mode = ApprovalMode::Never;
let mut config = Config {
approval_policy: Some("never".to_string()),
sandbox_mode: Some("read-only".to_string()),
allow_shell: Some(false),
requirements_path: Some(requirements_path.to_string_lossy().into_owned()),
..Config::default()
};
for preset in [
crate::tui::setup::SetupRuntimePreset::AskFirst,
crate::tui::setup::SetupRuntimePreset::NormalAgent,
crate::tui::setup::SetupRuntimePreset::HighTrustLocal,
] {
let error = apply_setup_runtime_preset(
&mut app,
&mut config,
preset,
codewhale_config::SetupState::default(),
)
.expect_err("managed requirements must win");
assert!(
error.to_string().contains("managed runtime requirements"),
"{error:#}"
);
}
assert_eq!(config.approval_policy.as_deref(), Some("never"));
assert_eq!(config.sandbox_mode.as_deref(), Some("read-only"));
assert_eq!(config.allow_shell, Some(false));
assert_eq!(app.approval_mode, ApprovalMode::Never);
assert!(!app.allow_shell);
assert_eq!(
std::fs::read_to_string(config_path).expect("root config"),
"# managed posture stays intact\n"
);
}
#[tokio::test]
#[allow(clippy::await_holding_lock)]
async fn tool_result_api_content_receipts_large_live_output() {
let _guard = crate::tools::truncate::TEST_SPILLOVER_GUARD
.lock()
.unwrap_or_else(|err| err.into_inner());
let tmp = TempDir::new().expect("spillover tempdir");
let prior = crate::tools::truncate::set_test_spillover_root(Some(
tmp.path().join(".deepseek").join("tool_outputs"),
));
struct Restore(Option<PathBuf>);
impl Drop for Restore {
fn drop(&mut self) {
crate::tools::truncate::set_test_spillover_root(self.0.take());
}
}
let _restore = Restore(prior);
let mut app = App::new(create_test_options(), &Config::default());
app.api_messages.push(Message {
role: "assistant".to_string(),
content: vec![ContentBlock::ToolUse {
id: "call-live-big".to_string(),
name: "exec_shell".to_string(),
input: serde_json::json!({"command": "cargo test"}),
caller: None,
}],
});
let raw = "LIVE_RAW_SENTINEL\n".repeat(900);
let output = crate::tools::spec::ToolResult::success(raw.clone());
let content =
tool_result_content_for_api_message(&app, "call-live-big", "exec_shell", &output).await;
assert!(content.contains("[TOOL_OUTPUT_RECEIPT]"));
assert!(content.contains("tool: exec_shell"));
assert!(content.contains("tool_call_id: call-live-big"));
assert!(content.contains("detail_handle: sha:"));
assert!(content.contains("retrieve: retrieve_tool_result ref=sha:"));
assert!(!content.contains(&raw));
assert!(
content.chars().count()
< crate::tool_output_receipts::RAW_TOOL_OUTPUT_RECEIPT_THRESHOLD_CHARS
);
}
#[test]
fn live_tool_receipt_messages_clones_only_matching_tool_use() {
let mut app = App::new(create_test_options(), &Config::default());
app.api_messages.push(Message {
role: "assistant".to_string(),
content: vec![ContentBlock::ToolUse {
id: "call-old".to_string(),
name: "exec_shell".to_string(),
input: serde_json::json!({"command": "old"}),
caller: None,
}],
});
app.api_messages.push(Message {
role: "user".to_string(),
content: vec![ContentBlock::ToolResult {
tool_use_id: "call-old".to_string(),
content: "OLD_RAW\n".repeat(2_000),
is_error: None,
content_blocks: None,
}],
});
app.api_messages.push(Message {
role: "assistant".to_string(),
content: vec![ContentBlock::ToolUse {
id: "call-new".to_string(),
name: "read_file".to_string(),
input: serde_json::json!({"path": "src/main.rs"}),
caller: None,
}],
});
let messages = live_tool_receipt_messages(&app, "call-new", "NEW_RAW", true);
assert_eq!(messages.len(), 2);
assert!(matches!(
&messages[0].content[0],
ContentBlock::ToolUse { id, name, .. } if id == "call-new" && name == "read_file"
));
assert!(matches!(
&messages[1].content[0],
ContentBlock::ToolResult { tool_use_id, content, .. }
if tool_use_id == "call-new" && content == "NEW_RAW"
));
}
fn text_message(role: &str, text: &str) -> Message {
Message {
role: role.to_string(),
content: vec![ContentBlock::Text {
text: text.to_string(),
cache_control: None,
}],
}
}
fn saved_session_with_messages(messages: Vec<Message>) -> SavedSession {
SavedSession {
schema_version: 1,
metadata: crate::session_manager::SessionMetadata {
id: "resume-recovery-session".to_string(),
title: "resume recovery".to_string(),
created_at: chrono::Utc::now(),
updated_at: chrono::Utc::now(),
message_count: messages.len(),
total_tokens: 0,
model: "deepseek-v4-pro".to_string(),
model_provider: "deepseek".to_string(),
model_provider_id: None,
workspace: PathBuf::from("/tmp/resume-recovery"),
mode: Some("yolo".to_string()),
cost: crate::session_manager::SessionCostSnapshot::default(),
parent_session_id: None,
forked_from_message_count: None,
cumulative_turn_secs: 0,
},
messages,
system_prompt: None,
context_references: Vec::new(),
artifacts: Vec::new(),
work_state: None,
}
}
fn named_custom_session_config(name: &str, base_url: &str, model: &str) -> Config {
let mut custom = HashMap::new();
custom.insert(
name.to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some(base_url.to_string()),
model: Some(model.to_string()),
..ProviderConfig::default()
},
);
Config {
provider: Some(name.to_string()),
providers: Some(ProvidersConfig {
custom,
..ProvidersConfig::default()
}),
..Config::default()
}
}
#[test]
fn apply_loaded_session_restores_dangling_user_tail_as_retry_draft() {
let mut app = create_test_app();
let session = saved_session_with_messages(vec![text_message(
"user",
"finish the Qthresh proof bundle",
)]);
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(recovered);
assert!(app.api_messages.is_empty());
assert_eq!(app.input, "finish the Qthresh proof bundle");
assert_eq!(
app.queued_draft
.as_ref()
.map(|draft| draft.display.as_str()),
Some("finish the Qthresh proof bundle")
);
assert!(
app.history
.iter()
.all(|cell| !matches!(cell, HistoryCell::User { .. }))
);
assert!(
app.status_message
.as_deref()
.is_some_and(|msg| msg.contains("Recovered interrupted prompt")),
"status was {:?}",
app.status_message
);
}
#[test]
fn apply_loaded_session_does_not_restore_slash_command_tail_as_retry_draft() {
let mut app = create_test_app();
let session = saved_session_with_messages(vec![text_message("user", "/sessions")]);
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(!recovered);
assert_eq!(app.input, "");
assert!(app.queued_draft.is_none());
assert_eq!(app.api_messages.len(), 1);
assert!(
app.history
.iter()
.any(|cell| matches!(cell, HistoryCell::User { .. }))
);
}
#[test]
fn apply_loaded_session_resets_unpersisted_telemetry() {
let mut app = create_test_app();
app.session.session_cost = 1.25;
app.session.session_cost_cny = 9.13;
app.session.subagent_cost = 0.75;
app.session.subagent_cost_cny = 5.48;
app.session.subagent_cost_event_seqs.insert(42);
app.session.displayed_cost_high_water = 2.0;
app.session.displayed_cost_high_water_cny = 14.61;
app.session.last_prompt_tokens = Some(120);
app.session.last_completion_tokens = Some(35);
app.session.last_prompt_cache_hit_tokens = Some(80);
app.session.last_prompt_cache_miss_tokens = Some(40);
app.session.last_reasoning_replay_tokens = Some(12);
app.push_turn_cache_record(crate::tui::app::TurnCacheRecord {
provider: None,
provider_identity: None,
model: None,
auto_model: false,
input_tokens: 120,
output_tokens: 35,
cache_hit_tokens: Some(80),
cache_miss_tokens: Some(40),
reasoning_replay_tokens: Some(12),
recorded_at: Instant::now(),
});
let mut session = saved_session_with_messages(vec![text_message("assistant", "ready")]);
session.metadata.total_tokens = 500;
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(!recovered);
assert_eq!(app.session.total_tokens, 500);
assert_eq!(app.session.total_conversation_tokens, 500);
assert_eq!(app.session.session_cost, 0.0);
assert_eq!(app.session.session_cost_cny, 0.0);
assert_eq!(app.session.subagent_cost, 0.0);
assert_eq!(app.session.subagent_cost_cny, 0.0);
assert!(app.session.subagent_cost_event_seqs.is_empty());
assert_eq!(app.session.displayed_cost_high_water, 0.0);
assert_eq!(app.session.displayed_cost_high_water_cny, 0.0);
assert_eq!(app.session.last_prompt_tokens, None);
assert_eq!(app.session.last_completion_tokens, None);
assert_eq!(app.session.last_prompt_cache_hit_tokens, None);
assert_eq!(app.session.last_prompt_cache_miss_tokens, None);
assert_eq!(app.session.last_reasoning_replay_tokens, None);
assert!(app.session.turn_cache_history.is_empty());
}
#[tokio::test]
async fn apply_loaded_session_resets_workspace_runtime_state() {
let mut app = create_test_app();
let mut config = Config::default();
let old_shell_manager = app
.runtime_services
.shell_manager
.as_ref()
.expect("shell manager")
.clone();
let old_context_cell = app.workspace_context_cell.clone();
app.workspace_context = Some("old workspace context".to_string());
if let Ok(mut cell) = old_context_cell.lock() {
*cell = Some("old workspace context".to_string());
}
app.workspace_context_refreshed_at = Some(Instant::now());
app.file_tree = Some(crate::tui::file_tree::FileTreeState::new(
PathBuf::from(".").as_path(),
));
let mut session = saved_session_with_messages(vec![text_message("assistant", "ready")]);
session.metadata.workspace = TempDir::new().expect("temp dir").path().to_path_buf();
let recovered = apply_loaded_session(&mut app, &mut config, &session).expect("restore session");
assert!(!recovered);
assert_eq!(app.workspace, session.metadata.workspace);
assert!(app.workspace_context.is_none());
assert!(app.workspace_context_refreshed_at.is_none());
assert!(app.file_tree.is_none());
assert!(old_context_cell.lock().expect("context cell").is_none());
let new_shell_manager = app
.runtime_services
.shell_manager
.as_ref()
.expect("shell manager")
.clone();
assert!(!std::sync::Arc::ptr_eq(
&old_shell_manager,
&new_shell_manager
));
assert_eq!(
new_shell_manager
.lock()
.expect("shell manager")
.default_workspace(),
session.metadata.workspace.as_path()
);
assert!(app.runtime_services.hook_executor.is_some());
}
#[test]
fn shell_live_output_refresh_does_not_block_on_contended_lock() {
let mut app = create_test_app();
let shell_mgr = app
.runtime_services
.shell_manager
.as_ref()
.expect("shell manager")
.clone();
let guard = shell_mgr.lock().expect("hold shell lock");
let changed = refresh_shell_exec_live_output(&mut app);
assert!(
!changed,
"contended live-output refresh should skip this frame, not block or update"
);
drop(guard);
assert!(!refresh_shell_exec_live_output(&mut app));
}
#[test]
fn apply_loaded_session_updates_current_workspace_display() {
let mut app = create_test_app();
let mut config = Config::default();
let workspace = TempDir::new().expect("temp dir");
let mut session = saved_session_with_messages(vec![text_message("assistant", "ready")]);
session.metadata.workspace = workspace.path().to_path_buf();
let recovered = apply_loaded_session(&mut app, &mut config, &session).expect("restore session");
let result = commands::execute("/workspace", &mut app);
assert!(!recovered);
assert_eq!(
result.message,
Some(format!("Current workspace: {}", workspace.path().display()))
);
assert!(result.action.is_none());
}
#[tokio::test]
async fn drain_web_config_events_applies_draft_without_closing_session() {
let mut app = create_test_app();
let mut config = Config::default();
let engine = mock_engine_handle();
let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
let doc = config_ui::build_document(&app, &config).expect("document");
tx.send(WebConfigSessionEvent::Draft(doc))
.expect("send draft");
let mut session = Some(WebConfigSession::for_test(rx));
let keep = drain_web_config_events(&mut session, &mut app, &mut config, &engine.handle).await;
assert!(keep);
assert!(session.is_some());
}
#[tokio::test]
async fn drain_web_config_events_closes_session_after_commit() {
let _config_env = ConfigPathEnvGuard::new();
let mut app = create_test_app();
let mut config = Config::default();
let engine = mock_engine_handle();
let (tx, rx) = tokio::sync::mpsc::unbounded_channel();
let doc = config_ui::build_document(&app, &config).expect("document");
tx.send(WebConfigSessionEvent::Committed(doc))
.expect("send commit");
let mut session = Some(WebConfigSession::for_test(rx));
let keep = drain_web_config_events(&mut session, &mut app, &mut config, &engine.handle).await;
assert!(!keep);
}
#[test]
fn backtrack_prefill_rehydrates_attachment_rows() {
let mut app = create_test_app();
let user_text = "inspect this\n[Attached image: /tmp/pasted.png]";
app.add_message(HistoryCell::User {
content: user_text.to_string(),
});
app.api_messages.push(Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: user_text.to_string(),
cache_control: None,
}],
});
app.add_message(HistoryCell::Assistant {
content: "done".to_string(),
streaming: false,
});
app.api_messages.push(Message {
role: "assistant".to_string(),
content: vec![ContentBlock::Text {
text: "done".to_string(),
cache_control: None,
}],
});
apply_backtrack(&mut app, 0);
assert_eq!(app.input, user_text);
assert_eq!(app.composer_attachment_count(), 1);
}
#[test]
fn active_tool_status_label_summarizes_live_tool_group() {
let mut app = create_test_app();
app.turn_started_at = Some(Instant::now() - Duration::from_secs(5));
let mut active = ActiveCell::new();
active.push_tool(
"exec-1",
HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test --workspace --all-features".to_string(),
status: ToolStatus::Running,
output: None,
live_output: None,
shell_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
started_at: app.turn_started_at,
duration_ms: None,
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})),
);
active.push_tool(
"tool-2",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "grep_files".to_string(),
status: ToolStatus::Success,
input_summary: Some("pattern: TODO".to_string()),
output: Some("done".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
let label = active_tool_status_label(&app, true).expect("status label");
assert!(label.contains("cargo test"));
assert!(label.contains("1 active"));
assert!(label.contains("1 done"));
assert!(label.contains(crate::tui::key_shortcuts::tool_details_shortcut_label()));
assert!(label.contains("opens details"));
}
#[test]
fn shell_live_output_update_matches_exact_task_id_only() {
let mut app = create_test_app();
app.push_history_cell(HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test --workspace".to_string(),
status: ToolStatus::Running,
output: None,
live_output: None,
shell_task_id: Some("shell_a".to_string()),
owner_agent_id: None,
owner_agent_name: None,
started_at: None,
duration_ms: None,
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})));
app.push_history_cell(HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test --workspace".to_string(),
status: ToolStatus::Running,
output: None,
live_output: Some("previous".to_string()),
shell_task_id: Some("shell_b".to_string()),
owner_agent_id: None,
owner_agent_name: None,
started_at: None,
duration_ms: None,
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})));
let mut jobs = std::collections::HashMap::new();
jobs.insert(
"shell_b".to_string(),
ShellJobSnapshot {
id: "shell_b".to_string(),
job_id: "shell_b".to_string(),
command: "cargo test --workspace".to_string(),
cwd: PathBuf::from("/tmp/repo"),
status: ShellStatus::Running,
exit_code: None,
elapsed_ms: 777,
stdout_tail: "stdout tail\n".to_string(),
stderr_tail: "stderr tail\n".to_string(),
stdout_len: 12,
stderr_len: 12,
stdin_available: false,
stale: false,
elapsed_since_output_ms: None,
linked_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
},
);
assert!(shell_exec_live_update(&app, 0, &jobs).is_none());
let (_task_id, status, output, duration) =
shell_exec_live_update(&app, 1, &jobs).expect("matching task id updates");
assert_eq!(status, ToolStatus::Running);
assert_eq!(duration, 777);
assert_eq!(
output.as_deref(),
Some("stdout tail\n\n\nSTDERR:\nstderr tail\n")
);
}
#[test]
fn shell_live_output_update_skips_finalized_exec_cell() {
let mut app = create_test_app();
app.push_history_cell(HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test --workspace".to_string(),
status: ToolStatus::Success,
output: Some("final output".to_string()),
live_output: Some("old live output".to_string()),
shell_task_id: Some("shell_a".to_string()),
owner_agent_id: None,
owner_agent_name: None,
started_at: None,
duration_ms: Some(10),
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})));
let mut jobs = std::collections::HashMap::new();
jobs.insert(
"shell_a".to_string(),
ShellJobSnapshot {
id: "shell_a".to_string(),
job_id: "shell_a".to_string(),
command: "cargo test --workspace".to_string(),
cwd: PathBuf::from("/tmp/repo"),
status: ShellStatus::Completed,
exit_code: Some(0),
elapsed_ms: 999,
stdout_tail: "new live output".to_string(),
stderr_tail: String::new(),
stdout_len: 15,
stderr_len: 0,
stdin_available: false,
stale: false,
elapsed_since_output_ms: None,
linked_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
},
);
assert!(shell_exec_live_update(&app, 0, &jobs).is_none());
}
#[test]
fn active_tool_status_label_strips_shell_wrappers_from_ci_polling() {
let mut app = create_test_app();
app.turn_started_at = Some(Instant::now() - Duration::from_secs(5));
let mut active = ActiveCell::new();
active.push_tool(
"exec-1",
HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cd /tmp/repo && sleep 15 && gh pr checks 1611 --repo Hmbown/CodeWhale"
.to_string(),
status: ToolStatus::Running,
output: None,
live_output: None,
shell_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
started_at: app.turn_started_at,
duration_ms: None,
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})),
);
app.active_cell = Some(active);
let label = active_tool_status_label(&app, true).expect("status label");
assert!(label.contains("gh pr checks 1611"), "label: {label}");
assert!(!label.contains("cd /tmp"), "label: {label}");
assert!(!label.contains("sleep 15"), "label: {label}");
}
#[test]
fn active_tool_status_label_counts_foreground_rlm_work() {
let mut app = create_test_app();
app.turn_started_at = Some(Instant::now() - Duration::from_secs(5));
let mut active = ActiveCell::new();
active.push_tool(
"rlm-1",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "rlm".to_string(),
status: ToolStatus::Running,
input_summary: Some("task: compare projects".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
let label = active_tool_status_label(&app, true).expect("status label");
assert!(label.contains("rlm"), "label: {label}");
assert!(!label.contains("tool rlm"), "label: {label}");
assert!(label.contains("1 active"), "label: {label}");
}
#[test]
fn terminal_probe_timeout_defaults_to_500ms() {
let config = Config::default();
assert_eq!(terminal_probe_timeout(&config), Duration::from_millis(500));
}
#[test]
fn terminal_probe_timeout_uses_tui_config_and_clamps() {
let mut config = Config {
tui: Some(crate::config::TuiConfig {
alternate_screen: None,
mouse_capture: None,
terminal_probe_timeout_ms: Some(750),
stream_chunk_timeout_secs: None,
status_items: None,
osc8_links: None,
notification_condition: None,
composer_arrows_scroll: None,
}),
..Config::default()
};
assert_eq!(terminal_probe_timeout(&config), Duration::from_millis(750));
config
.tui
.as_mut()
.expect("tui config")
.terminal_probe_timeout_ms = Some(0);
assert_eq!(terminal_probe_timeout(&config), Duration::from_millis(100));
config
.tui
.as_mut()
.expect("tui config")
.terminal_probe_timeout_ms = Some(60_000);
assert_eq!(
terminal_probe_timeout(&config),
Duration::from_millis(5_000)
);
}
#[test]
fn file_mentions_add_local_text_context_to_model_payload() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(
tmpdir.path().join("guide.md"),
"# Guide\nUse the fast path.\n",
)
.expect("write file");
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
let message = QueuedMessage::new("Summarize @guide.md".to_string(), None);
let content = queued_message_content_for_app(&app, &message, None);
assert!(content.starts_with("Summarize @guide.md"));
assert!(content.contains("Local context from @mentions:"));
assert!(content.contains("<file mention=\"@guide.md\""));
assert!(content.contains("# Guide\nUse the fast path."));
assert_eq!(message.display, "Summarize @guide.md");
}
#[test]
fn compact_user_context_display_hides_persisted_mention_block() {
let content = "Summarize @guide.md\n\n---\n\nLocal context from @mentions:\n<file>large</file>";
assert_eq!(compact_user_context_display(content), "Summarize @guide.md");
}
#[test]
fn file_mentions_do_not_trigger_inside_email_addresses() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(tmpdir.path().join("example.com"), "not a mention").expect("write file");
let content = user_request_with_file_mentions("email me@example.com", tmpdir.path(), None);
assert_eq!(content, "email me@example.com");
}
#[test]
fn media_file_mentions_point_to_attach_instead_of_inlining_bytes() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(tmpdir.path().join("photo.png"), b"\0png").expect("write image");
let content = user_request_with_file_mentions("inspect @photo.png", tmpdir.path(), None);
assert!(content.contains("<media-file mention=\"@photo.png\""));
assert!(content.contains("Use /attach photo.png"));
assert!(!content.contains("\0png"));
}
#[tokio::test]
async fn model_change_update_syncs_engine_model_before_compaction() {
let mut app = create_test_app();
app.model = "deepseek-v4-flash".to_string();
let compaction = app.compaction_config();
let mut engine = crate::core::engine::mock_engine_handle();
apply_model_and_compaction_update(
&engine.handle,
compaction,
app.mode,
app.active_route_limits,
)
.await;
match engine.rx_op.recv().await.expect("set model op") {
crate::core::ops::Op::SetModel {
model,
mode,
route_limits: _,
} => {
assert_eq!(model, "deepseek-v4-flash");
assert_eq!(mode, app.mode);
}
other => panic!("expected SetModel, got {other:?}"),
}
match engine.rx_op.recv().await.expect("set compaction op") {
crate::core::ops::Op::SetCompaction { config } => {
assert_eq!(config.model, "deepseek-v4-flash");
}
other => panic!("expected SetCompaction, got {other:?}"),
}
}
#[tokio::test]
async fn mode_change_update_notifies_engine() {
let mut app = create_test_app();
let _ = app.set_mode(crate::tui::app::AppMode::Plan);
let mut engine = crate::core::engine::mock_engine_handle();
assert!(apply_mode_update(&mut app, &engine.handle, crate::tui::app::AppMode::Yolo).await);
match engine.rx_op.recv().await.expect("change mode op") {
crate::core::ops::Op::ChangeMode {
mode,
allow_shell,
trust_mode,
auto_approve,
approval_mode,
} => {
assert_eq!(mode, crate::tui::app::AppMode::Agent);
assert!(allow_shell);
assert!(trust_mode);
assert!(auto_approve);
assert_eq!(approval_mode, crate::tui::approval::ApprovalMode::Bypass);
}
other => panic!("expected ChangeMode, got {other:?}"),
}
}
#[tokio::test]
async fn mode_change_update_sends_restored_agent_policy() {
let mut app = create_test_app();
app.allow_shell = true;
app.trust_mode = false;
app.approval_mode = crate::tui::approval::ApprovalMode::Never;
let _ = app.set_mode(crate::tui::app::AppMode::Plan);
let mut engine = crate::core::engine::mock_engine_handle();
assert!(apply_mode_update(&mut app, &engine.handle, crate::tui::app::AppMode::Agent).await);
match engine.rx_op.recv().await.expect("change mode op") {
crate::core::ops::Op::ChangeMode {
mode,
allow_shell,
trust_mode,
auto_approve,
approval_mode,
} => {
assert_eq!(mode, crate::tui::app::AppMode::Agent);
assert!(allow_shell);
assert!(!trust_mode);
assert!(!auto_approve);
assert_eq!(approval_mode, crate::tui::approval::ApprovalMode::Never);
}
other => panic!("expected ChangeMode, got {other:?}"),
}
}
#[test]
fn saved_default_provider_syncs_back_to_runtime_config() {
let _home = SettingsHomeGuard::new();
let settings = crate::settings::Settings {
default_provider: Some("ollama".to_string()),
..Default::default()
};
settings.save().expect("save settings");
let mut config = Config::default();
assert_eq!(config.api_provider(), ApiProvider::Deepseek);
let app = App::new(create_test_options(), &config);
assert_eq!(app.api_provider, ApiProvider::Ollama);
sync_config_provider_from_app(&mut config, &app);
assert_eq!(config.api_provider(), ApiProvider::Ollama);
}
#[test]
fn provider_picker_reselecting_active_provider_preserves_current_model() {
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Ollama, "ollama");
app.model = "deepseek-coder-v2:16b".to_string();
let config = Config {
provider: Some("ollama".to_string()),
..Config::default()
};
assert_eq!(
provider_picker_model_override(&app, &config, ApiProvider::Ollama).as_deref(),
Some("deepseek-coder-v2:16b")
);
assert_eq!(
provider_picker_model_override(&app, &config, ApiProvider::Deepseek),
None
);
}
#[tokio::test]
async fn provider_switch_clears_turn_cache_history() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.push_turn_cache_record(crate::tui::app::TurnCacheRecord {
provider: None,
provider_identity: None,
model: None,
auto_model: false,
input_tokens: 100,
output_tokens: 25,
cache_hit_tokens: Some(70),
cache_miss_tokens: Some(30),
reasoning_replay_tokens: Some(12),
recorded_at: Instant::now(),
});
let mut engine = mock_engine_handle();
let mut config = Config::default();
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Ollama,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Ollama);
assert!(app.session.turn_cache_history.is_empty());
}
#[tokio::test]
async fn provider_switch_to_deepseek_canonicalizes_openrouter_default_model() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::Openrouter;
app.model = DEFAULT_OPENROUTER_MODEL.to_string();
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("openrouter".to_string()),
api_key: Some("test-key".to_string()),
default_text_model: Some(DEFAULT_OPENROUTER_MODEL.to_string()),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Deepseek,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert!(!app.model_ids_passthrough);
assert_eq!(app.model, DEFAULT_TEXT_MODEL);
}
#[tokio::test]
async fn provider_switch_to_deepseek_drops_stale_xiaomi_root_base_url() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::XiaomiMimo;
app.model = "mimo-v2.5-pro".to_string();
app.model_ids_passthrough = true;
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("xiaomi-mimo".to_string()),
api_key: Some("deepseek-key".to_string()),
base_url: Some("https://token-plan-sgp.xiaomimimo.com/v1".to_string()),
default_text_model: Some("mimo-v2.5-pro".to_string()),
providers: Some(ProvidersConfig {
xiaomi_mimo: ProviderConfig {
api_key: Some("mimo-key".to_string()),
model: Some("mimo-v2.5-pro".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Deepseek,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert!(!app.model_ids_passthrough);
assert_eq!(app.model, DEFAULT_TEXT_MODEL);
assert_eq!(config.provider.as_deref(), Some("deepseek"));
assert_eq!(config.base_url, None);
}
#[tokio::test]
async fn provider_switch_from_mimo_to_openrouter_without_key_fails_before_dispatch() {
let _home = SettingsHomeGuard::new();
let _openrouter_key = crate::test_support::EnvVarGuard::remove("OPENROUTER_API_KEY");
let mut app = create_test_app();
app.api_provider = ApiProvider::XiaomiMimo;
app.model = "mimo-v2.5-pro".to_string();
app.model_ids_passthrough = true;
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("xiaomi-mimo".to_string()),
api_key: Some("deepseek-key".to_string()),
base_url: Some("https://token-plan-sgp.xiaomimimo.com/v1".to_string()),
default_text_model: Some("mimo-v2.5-pro".to_string()),
providers: Some(ProvidersConfig {
xiaomi_mimo: ProviderConfig {
api_key: Some("mimo-key".to_string()),
model: Some("mimo-v2.5-pro".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Openrouter,
Some(crate::config::OPENROUTER_NEMOTRON_3_ULTRA_MODEL.to_string()),
)
.await;
assert_eq!(app.api_provider, ApiProvider::XiaomiMimo);
assert_eq!(app.model, "mimo-v2.5-pro");
assert!(app.model_ids_passthrough);
assert_eq!(config.provider.as_deref(), Some("xiaomi-mimo"));
assert_eq!(
config
.providers
.as_ref()
.and_then(|providers| providers.openrouter.api_key.as_deref()),
None
);
assert!(app.pending_provider_switch.is_none());
let last_system_message = app
.history
.iter()
.rev()
.find_map(|cell| match cell {
HistoryCell::System { content } => Some(content.as_str()),
_ => None,
})
.expect("failed provider switch should add a system message");
assert!(last_system_message.contains("OpenRouter API key not found"));
assert!(last_system_message.contains("Provider unchanged (xiaomi-mimo)"));
}
#[tokio::test]
async fn provider_switch_persists_provider_to_config_for_restart() {
let _home = SettingsHomeGuard::new();
let tmp = TempDir::new().expect("config tempdir");
let config_path = tmp.path().join("config.toml");
std::fs::write(
&config_path,
r#"provider = "arcee"
[providers.xiaomi_mimo]
base_url = "https://token-plan-sgp.xiaomimimo.com/v1"
model = "mimo-v2.5-pro"
api_key = "mimo-key"
[providers.arcee]
api_key = "arcee-key"
"#,
)
.expect("write config");
let mut app = create_test_app();
app.api_provider = ApiProvider::Arcee;
app.model = "auto".to_string();
app.config_path = Some(config_path.clone());
let mut engine = mock_engine_handle();
let mut config = Config::load(Some(config_path.clone()), None).expect("load config");
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::XiaomiMimo,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::XiaomiMimo);
assert_eq!(config.provider.as_deref(), Some("xiaomi-mimo"));
let reloaded = Config::load(Some(config_path.clone()), None).expect("reload config");
assert_eq!(reloaded.api_provider(), ApiProvider::XiaomiMimo);
assert_eq!(
reloaded.deepseek_base_url(),
"https://token-plan-sgp.xiaomimimo.com/v1"
);
let settings = crate::settings::Settings::load().expect("load settings");
assert_eq!(settings.default_provider.as_deref(), Some("xiaomi-mimo"));
}
#[tokio::test]
async fn provider_switch_model_override_updates_target_provider_model_slot() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::XiaomiMimo;
app.model = "mimo-v2.5-pro".to_string();
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("xiaomi-mimo".to_string()),
api_key: Some("deepseek-key".to_string()),
default_text_model: Some("mimo-v2.5-pro".to_string()),
providers: Some(ProvidersConfig {
xiaomi_mimo: ProviderConfig {
api_key: Some("mimo-key".to_string()),
model: Some("mimo-v2.5-pro".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Deepseek,
Some("deepseek-v4-flash".to_string()),
)
.await;
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert_eq!(app.model, "deepseek-v4-flash");
assert_eq!(
config
.providers
.as_ref()
.and_then(|providers| providers.deepseek.model.as_deref()),
Some("deepseek-v4-flash")
);
assert_eq!(
config
.providers
.as_ref()
.and_then(|providers| providers.xiaomi_mimo.model.as_deref()),
Some("mimo-v2.5-pro")
);
let state = codewhale_config::SetupState::load()
.expect("load setup state")
.expect("setup state");
assert_eq!(
state.status(codewhale_config::SetupStep::ProviderModel),
codewhale_config::StepStatus::Verified
);
let provider_model_result = state
.steps
.get(&codewhale_config::SetupStep::ProviderModel)
.and_then(|entry| entry.result.as_deref())
.expect("provider/model result");
assert!(provider_model_result.contains("provider=deepseek"));
assert!(provider_model_result.contains("model=deepseek-v4-flash"));
assert!(provider_model_result.contains("auth=key saved · not checked"));
assert!(provider_model_result.contains("health=attemptable"));
assert!(!provider_model_result.contains("deepseek-key"));
}
#[tokio::test]
async fn provider_switch_skips_setup_receipt_when_route_persistence_fails() {
let _home = SettingsHomeGuard::new();
let tmp = TempDir::new().expect("config tempdir");
let mut app = create_test_app();
app.api_provider = ApiProvider::XiaomiMimo;
app.model = "mimo-v2.5-pro".to_string();
app.config_path = Some(tmp.path().to_path_buf());
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("xiaomi-mimo".to_string()),
api_key: Some("deepseek-key".to_string()),
default_text_model: Some("mimo-v2.5-pro".to_string()),
providers: Some(ProvidersConfig {
xiaomi_mimo: ProviderConfig {
api_key: Some("mimo-key".to_string()),
model: Some("mimo-v2.5-pro".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Deepseek,
Some("deepseek-v4-flash".to_string()),
)
.await;
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert!(
app.status_message
.as_deref()
.is_some_and(|message| message.contains("not fully persisted"))
);
assert!(
codewhale_config::SetupState::load()
.expect("load setup state")
.is_none(),
"failed route persistence must not create a ProviderModel setup receipt"
);
}
#[tokio::test]
async fn provider_switch_without_model_uses_target_default_not_previous_provider_model() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::Openrouter;
app.model = "deepseek/deepseek-v4-pro".to_string();
app.model_ids_passthrough = true;
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("openrouter".to_string()),
api_key: Some("deepseek-key".to_string()),
providers: Some(ProvidersConfig {
openrouter: ProviderConfig {
api_key: Some("openrouter-key".to_string()),
model: Some("deepseek/deepseek-v4-pro".to_string()),
..Default::default()
},
zai: ProviderConfig {
api_key: Some("zai-key".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Zai,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Zai);
assert_eq!(app.model, DEFAULT_ZAI_MODEL);
assert_eq!(config.provider.as_deref(), Some("zai"));
assert_eq!(
config
.providers
.as_ref()
.and_then(|providers| providers.zai.model.as_deref()),
Some(DEFAULT_ZAI_MODEL)
);
assert_eq!(
config
.providers
.as_ref()
.and_then(|providers| providers.openrouter.model.as_deref()),
Some("deepseek/deepseek-v4-pro")
);
}
#[tokio::test]
async fn provider_switch_foreign_direct_model_rejected_before_mutation() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.model = DEFAULT_TEXT_MODEL.to_string();
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("deepseek".to_string()),
api_key: Some("deepseek-key".to_string()),
providers: Some(ProvidersConfig {
deepseek: ProviderConfig {
api_key: Some("deepseek-key".to_string()),
model: Some(DEFAULT_TEXT_MODEL.to_string()),
..Default::default()
},
zai: ProviderConfig {
api_key: Some("zai-key".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Zai,
Some("deepseek-v4-pro".to_string()),
)
.await;
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert_eq!(app.model, DEFAULT_TEXT_MODEL);
assert_eq!(config.provider.as_deref(), Some("deepseek"));
assert_eq!(
config
.providers
.as_ref()
.and_then(|providers| providers.zai.model.as_deref()),
None
);
assert!(app.pending_provider_switch.is_none());
assert!(
app.status_message
.as_deref()
.unwrap_or_default()
.contains("Route rejected before provider switch")
);
}
#[tokio::test]
async fn provider_switch_to_openai_codex_normalizes_deepseek_off_effort() {
let _home = SettingsHomeGuard::new();
let _token = crate::test_support::EnvVarGuard::set("OPENAI_CODEX_ACCESS_TOKEN", "test-token");
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.model = DEFAULT_TEXT_MODEL.to_string();
app.reasoning_effort = ReasoningEffort::Off;
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("deepseek".to_string()),
default_text_model: Some(DEFAULT_TEXT_MODEL.to_string()),
providers: Some(ProvidersConfig {
openai_codex: ProviderConfig {
model: Some(crate::config::DEFAULT_OPENAI_CODEX_MODEL.to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::OpenaiCodex,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::OpenaiCodex);
assert_eq!(app.model, crate::config::DEFAULT_OPENAI_CODEX_MODEL);
assert_eq!(app.reasoning_effort, ReasoningEffort::Low);
assert_eq!(app.reasoning_effort_display_label(), "low");
}
#[tokio::test]
async fn provider_switch_to_openrouter_canonicalizes_deepseek_default_model() {
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.model = DEFAULT_TEXT_MODEL.to_string();
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("deepseek".to_string()),
default_text_model: Some(DEFAULT_TEXT_MODEL.to_string()),
providers: Some(ProvidersConfig {
openrouter: ProviderConfig {
api_key: Some("test-key".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Openrouter,
None,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Openrouter);
assert_eq!(app.model, DEFAULT_OPENROUTER_MODEL);
}
#[tokio::test]
async fn dispatch_user_message_failed_send_clears_loading_state() {
let mut app = create_test_app();
let engine = mock_engine_handle();
let config = Config::default();
drop(engine.rx_op);
let result = dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("hello".to_string(), None),
)
.await;
assert!(
result.is_err(),
"dispatch should fail when engine channel is closed"
);
assert!(
!app.is_loading,
"failed dispatch must not leave the composer in a permanent busy state"
);
assert!(app.last_send_at.is_none());
assert!(app.dispatch_started_at.is_none());
assert!(
app.pending_turn_route.is_none(),
"failed dispatch must not leave stale route telemetry"
);
}
#[tokio::test]
async fn failed_paused_dispatch_preserves_app_checkpoint_state_and_engine_gate() {
let mut app = create_test_app();
app.pausable = true;
app.paused = true;
app.paused_quarry = Some("finish the paused audit".to_string());
app.hunt.quarry = None;
app.hunt.tokens_used = 7;
app.hunt.time_used_seconds = 11;
app.hunt.continuation_count = 2;
app.api_messages
.push(text_message("assistant", "existing conversation"));
app.add_message(HistoryCell::System {
content: "existing transcript".to_string(),
});
let before_messages = app.api_messages.clone();
let before_history = format!("{:?}", app.history);
let before_context_references = app.session_context_references.clone();
let before_system_prompt = app.system_prompt.clone();
let before_last_prompt = app.last_submitted_prompt.clone();
let engine = mock_engine_handle();
engine.handle.set_paused(true);
drop(engine.rx_op);
dispatch_user_message(
&mut app,
&Config::default(),
&engine.handle,
QueuedMessage::new("please continue".to_string(), None),
)
.await
.expect_err("closed engine mailbox must reject the paused dispatch");
assert!(app.paused);
assert!(app.pausable);
assert_eq!(
app.paused_quarry.as_deref(),
Some("finish the paused audit")
);
assert!(app.hunt.quarry.is_none());
assert_eq!(app.hunt.tokens_used, 7);
assert_eq!(app.hunt.time_used_seconds, 11);
assert_eq!(app.hunt.continuation_count, 2);
assert!(engine.handle.is_paused());
assert_eq!(app.api_messages, before_messages);
assert_eq!(format!("{:?}", app.history), before_history);
assert_eq!(app.session_context_references, before_context_references);
assert_eq!(app.system_prompt, before_system_prompt);
assert_eq!(app.last_submitted_prompt, before_last_prompt);
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
assert!(app.last_send_at.is_none());
assert!(app.pending_turn_route.is_none());
}
#[tokio::test]
async fn paused_dispatch_at_compaction_threshold_enqueues_one_atomic_send() {
let mut app = create_test_app();
app.pausable = true;
app.paused = true;
app.paused_quarry = Some("finish the paused audit".to_string());
app.hunt.quarry = None;
app.auto_compact_user_configured = true;
app.auto_compact = true;
app.auto_compact_threshold_percent = 10.0;
app.api_messages = vec![text_message("assistant", &"context ".repeat(240_000))];
let planned_compaction =
app.compaction_config_for_route(app.api_provider, &app.model, app.active_route_limits);
assert!(
should_auto_compact_before_send_with_config(&app, &planned_compaction),
"fixture must already require compaction"
);
let mut engine = mock_engine_handle();
engine.handle.set_paused(true);
dispatch_user_message(
&mut app,
&Config::default(),
&engine.handle,
QueuedMessage::new("please continue".to_string(), None),
)
.await
.expect("atomic paused dispatch");
match engine.rx_op.recv().await.expect("single send operation") {
Op::SendMessage {
compaction,
goal_objective,
..
} => {
assert!(compaction.enabled);
assert_eq!(goal_objective.as_deref(), Some("finish the paused audit"));
}
other => panic!("expected SendMessage, got {other:?}"),
}
assert!(
engine.rx_op.try_recv().is_err(),
"route-bound compaction must travel inside the one SendMessage op"
);
assert!(!app.paused);
assert!(!engine.handle.is_paused());
}
#[tokio::test]
async fn real_engine_client_preflight_failure_leaves_dispatch_state_atomic() {
let mut config =
named_custom_session_config("lm-studio", "http://127.0.0.1:1234/v1", "local-model");
config
.providers
.as_mut()
.expect("providers")
.custom
.get_mut("lm-studio")
.expect("lm-studio")
.insecure_skip_tls_verify = Some(true);
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "lm-studio");
app.set_model_selection("local-model".to_string());
let (_engine, handle) = crate::core::engine::Engine::new(EngineConfig::default(), &config);
let err = dispatch_user_message(
&mut app,
&config,
&handle,
QueuedMessage::new("keep this retryable".to_string(), None),
)
.await
.expect_err("real engine routes must preflight their concrete client");
assert!(
err.to_string()
.contains("Failed to configure provider route")
);
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
assert!(app.last_send_at.is_none());
assert!(app.last_submitted_prompt.is_none());
assert!(app.api_messages.is_empty());
assert!(app.history.is_empty());
assert!(app.pending_turn_route.is_none());
}
#[tokio::test]
async fn immediate_submit_closed_mailbox_restores_composer_and_skill() {
let mut app = create_test_app();
app.input = "retry this exactly".to_string();
app.cursor_position = app.input.chars().count();
app.active_skill = Some("keep the selected skill".to_string());
let input = app
.handle_composer_enter()
.expect("non-empty composer should submit");
let queued = build_queued_message(&mut app, input);
assert!(app.input.is_empty());
assert!(app.active_skill.is_none());
let engine = mock_engine_handle();
drop(engine.rx_op);
submit_or_steer_message(&mut app, &Config::default(), &engine.handle, queued)
.await
.expect("UI submit failures must remain inside the TUI");
assert_eq!(app.input, "retry this exactly");
assert_eq!(app.cursor_position, app.input.chars().count());
assert_eq!(app.active_skill.as_deref(), Some("keep the selected skill"));
assert!(!app.is_loading);
assert!(app.status_message.as_deref().is_some_and(|status| {
status.contains("Message not sent") && status.contains("restored to composer")
}));
let sticky = app
.sticky_status
.as_ref()
.expect("dispatch failure should stay visible");
assert_eq!(sticky.level, StatusToastLevel::Error);
assert!(sticky.ttl_ms.is_none());
}
#[tokio::test]
async fn immediate_submit_custom_provider_preflight_restores_exact_message() {
let mut config =
named_custom_session_config("lm-studio", "http://127.0.0.1:1234/v1", "local-model");
config
.providers
.as_mut()
.expect("providers")
.custom
.get_mut("lm-studio")
.expect("lm-studio")
.insecure_skip_tls_verify = Some(true);
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "lm-studio");
app.set_model_selection("local-model".to_string());
app.input = "preserve 用户 input".to_string();
app.cursor_position = app.input.chars().count();
let input = app
.handle_composer_enter()
.expect("non-empty composer should submit");
let queued = build_queued_message(&mut app, input);
let (_engine, handle) = crate::core::engine::Engine::new(EngineConfig::default(), &config);
submit_or_steer_message(&mut app, &config, &handle, queued)
.await
.expect("provider preflight failures must remain inside the TUI");
assert_eq!(app.input, "preserve 用户 input");
assert_eq!(app.cursor_position, app.input.chars().count());
assert!(app.api_messages.is_empty());
assert!(app.history.is_empty());
assert!(app.last_submitted_prompt.is_none());
assert!(app.status_message.as_deref().is_some_and(|status| {
status.contains("Failed to configure provider route")
&& status.contains("restored to composer")
}));
}
#[tokio::test]
async fn dispatch_uses_app_owned_exact_custom_identity_when_config_selector_drifts() {
let mut custom = HashMap::new();
for (name, base_url, model) in [
("custom-a", "http://127.0.0.1:18181/v1", "model-a"),
("custom-b", "http://127.0.0.1:18182/v1", "model-b"),
] {
custom.insert(
name.to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some(base_url.to_string()),
model: Some(model.to_string()),
api_key: Some(format!("{name}-test-key")),
..Default::default()
},
);
}
let config = Config {
provider: Some("custom-b".to_string()),
providers: Some(ProvidersConfig {
custom,
..Default::default()
}),
..Default::default()
};
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
let mut engine = mock_engine_handle();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("stay on A".to_string(), None),
)
.await
.expect("dispatch exact App-owned route");
match engine.rx_op.recv().await.expect("send message op") {
Op::SendMessage { route, .. } => {
assert_eq!(route.identity.provider, ApiProvider::Custom);
assert_eq!(route.identity.key, "custom-a");
assert_eq!(route.identity.exact_id.as_deref(), Some("custom-a"));
assert_eq!(route.model, "model-a");
assert_eq!(
route.config.deepseek_base_url(),
"http://127.0.0.1:18181/v1"
);
}
other => panic!("expected SendMessage, got {other:?}"),
}
assert_eq!(config.provider.as_deref(), Some("custom-b"));
}
#[tokio::test]
async fn dispatch_idless_custom_identity_keeps_legacy_root_over_literal_table() {
let config = Config {
provider: Some("custom".to_string()),
api_key: Some("legacy-root-test-key".to_string()),
base_url: Some("http://127.0.0.1:18180/v1".to_string()),
default_text_model: Some("legacy-root-model".to_string()),
providers: Some(ProvidersConfig {
custom: HashMap::from([(
"custom".to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
api_key: Some("literal-table-test-key".to_string()),
base_url: Some("http://127.0.0.1:18181/v1".to_string()),
model: Some("literal-table-model".to_string()),
..Default::default()
},
)]),
..Default::default()
}),
..Default::default()
};
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "custom");
app.set_model_selection("legacy-root-model".to_string());
let mut engine = mock_engine_handle();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("keep the released root route".to_string(), None),
)
.await
.expect("dispatch idless legacy root route");
match engine.rx_op.recv().await.expect("send message op") {
Op::SendMessage { route, .. } => {
assert_eq!(route.identity.provider, ApiProvider::Custom);
assert_eq!(route.identity.key, "custom");
assert_eq!(route.identity.exact_id, None);
assert_eq!(route.model, "legacy-root-model");
assert_eq!(
route.config.deepseek_base_url(),
"http://127.0.0.1:18180/v1"
);
assert!(
route
.config
.providers
.as_ref()
.is_none_or(|providers| !providers.custom.contains_key("custom"))
);
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[tokio::test]
async fn failed_real_preflight_preserves_paused_command_state_and_engine_gate() {
let mut config =
named_custom_session_config("lm-studio", "http://127.0.0.1:1234/v1", "local-model");
config
.providers
.as_mut()
.expect("providers")
.custom
.get_mut("lm-studio")
.expect("lm-studio")
.insecure_skip_tls_verify = Some(true);
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "lm-studio");
app.set_model_selection("local-model".to_string());
app.paused = true;
app.pausable = true;
app.paused_quarry = Some("finish the paused audit".to_string());
app.hunt.quarry = None;
app.hunt.tokens_used = 7;
app.hunt.time_used_seconds = 11;
app.hunt.continuation_count = 2;
let (_engine, handle) = crate::core::engine::Engine::new(EngineConfig::default(), &config);
handle.set_paused(true);
dispatch_user_message(
&mut app,
&config,
&handle,
QueuedMessage::new("please continue".to_string(), None),
)
.await
.expect_err("invalid client must fail before changing pause state");
assert!(app.paused);
assert!(app.pausable);
assert_eq!(
app.paused_quarry.as_deref(),
Some("finish the paused audit")
);
assert!(app.hunt.quarry.is_none());
assert_eq!(app.hunt.tokens_used, 7);
assert_eq!(app.hunt.time_used_seconds, 11);
assert_eq!(app.hunt.continuation_count, 2);
assert!(handle.is_paused());
assert!(app.api_messages.is_empty());
assert!(app.history.is_empty());
}
#[test]
fn logout_memory_clear_respects_named_and_legacy_custom_scopes() {
let mut named_app = create_test_app();
named_app.set_provider_identity(ApiProvider::Custom, "lm-studio");
let mut named_config = Config {
provider: Some("lm-studio".to_string()),
api_key: Some("deepseek-root-key".to_string()),
providers: Some(ProvidersConfig {
custom: HashMap::from([(
"lm-studio".to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
api_key: Some("named-key".to_string()),
base_url: Some("http://127.0.0.1:18181/v1".to_string()),
model: Some("local-model".to_string()),
..Default::default()
},
)]),
..Default::default()
}),
..Default::default()
};
clear_active_provider_api_key_from_memory(&named_app, &mut named_config);
assert_eq!(named_config.api_key.as_deref(), Some("deepseek-root-key"));
assert_eq!(
named_config
.providers
.as_ref()
.and_then(|providers| providers.custom.get("lm-studio"))
.and_then(|provider| provider.api_key.as_deref()),
None
);
let mut legacy_app = create_test_app();
legacy_app.set_provider_identity(ApiProvider::Custom, "custom");
let mut legacy_config = Config {
provider: Some("custom".to_string()),
api_key: Some("legacy-key".to_string()),
base_url: Some("http://127.0.0.1:18180/v1".to_string()),
default_text_model: Some("legacy-model".to_string()),
..Default::default()
};
clear_active_provider_api_key_from_memory(&legacy_app, &mut legacy_config);
assert_eq!(legacy_config.api_key, None);
assert!(legacy_config.providers.is_none());
}
#[test]
fn auto_routed_turn_compaction_uses_selected_route_not_stale_app_route() {
let mut app = create_test_app();
app.auto_model = true;
app.model = "auto".to_string();
app.api_provider = ApiProvider::Deepseek;
app.active_route_limits = Some(codewhale_config::route::RouteLimits {
context_tokens: Some(32_000),
..Default::default()
});
app.auto_compact_threshold_percent = 75.0;
let config = Config {
provider: Some("openrouter".to_string()),
providers: Some(ProvidersConfig {
openrouter: ProviderConfig {
api_key: Some("test-openrouter-key".to_string()),
model: Some("vendor/model-b".to_string()),
context_window: Some(196_000),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
let route = resolve_runtime_route(&config, ApiProvider::Openrouter, Some("vendor/model-b"))
.expect("resolve auto-selected route")
.validate()
.expect("preflight auto-selected route");
let route_limits = crate::route_budget::known_route_limits(route.candidate.limits);
let compaction =
app.compaction_config_for_route(route.identity.provider, &route.model, route_limits);
assert_eq!(compaction.model, "vendor/model-b");
assert_eq!(compaction.effective_context_window, Some(196_000));
assert_eq!(
compaction.token_threshold,
crate::route_budget::compaction_threshold_for_route_at_percent(
ApiProvider::Openrouter,
"vendor/model-b",
route_limits,
75.0,
)
);
assert_ne!(compaction, app.compaction_config());
let pre_send_compact = crate::core::ops::Op::CompactContext {
route: Box::new(route.into_resolved()),
compaction: Box::new(compaction.clone()),
};
match pre_send_compact {
crate::core::ops::Op::CompactContext { route, compaction } => {
assert_eq!(route.identity.provider, ApiProvider::Openrouter);
assert_eq!(route.model, "vendor/model-b");
assert_eq!(compaction.model, "vendor/model-b");
assert_eq!(compaction.effective_context_window, Some(196_000));
}
other => panic!("expected route-bound compact op, got {other:?}"),
}
}
#[cfg(not(windows))]
fn write_message_submit_hook(dir: &TempDir, name: &str, body: &str) -> String {
let path = dir.path().join(name);
std::fs::write(&path, body).expect("write message_submit hook");
format!("sh {}", path.display())
}
#[cfg(not(windows))]
fn configure_single_message_submit_hook(app: &mut App, dir: &TempDir, command: String) {
configure_message_submit_hooks(app, dir, vec![command]);
}
#[cfg(not(windows))]
fn configure_message_submit_hooks(app: &mut App, dir: &TempDir, commands: Vec<String>) {
app.hooks = crate::hooks::HookExecutor::new(
crate::hooks::HooksConfig {
enabled: true,
hooks: commands
.iter()
.map(|command| {
crate::hooks::Hook::new(crate::hooks::HookEvent::MessageSubmit, command)
})
.collect(),
working_dir: Some(dir.path().to_path_buf()),
..crate::hooks::HooksConfig::default()
},
dir.path().to_path_buf(),
);
}
#[cfg(not(windows))]
#[tokio::test]
async fn dispatch_user_message_surfaces_continued_message_submit_timeout() {
let dir = TempDir::new().expect("tempdir");
let slow = write_message_submit_hook(
&dir,
"slow.sh",
r#"#!/bin/sh
sleep 2
"#,
);
let replacing = write_message_submit_hook(
&dir,
"replace.sh",
r#"#!/bin/sh
printf '%s\n' '{"text":"after timeout"}'
"#,
);
let mut app = create_test_app();
app.hooks = crate::hooks::HookExecutor::new(
crate::hooks::HooksConfig {
enabled: true,
hooks: vec![
crate::hooks::Hook::new(crate::hooks::HookEvent::MessageSubmit, &slow)
.with_timeout(1),
crate::hooks::Hook::new(crate::hooks::HookEvent::MessageSubmit, &replacing),
],
working_dir: Some(dir.path().to_path_buf()),
..crate::hooks::HooksConfig::default()
},
dir.path().to_path_buf(),
);
let mut engine = crate::core::engine::mock_engine_handle();
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("hello".to_string(), None),
)
.await
.expect("dispatch user message");
assert_eq!(
app.status_message.as_deref(),
Some("Hook timed out after 1s")
);
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage { content, .. } => {
assert_eq!(content, "after timeout");
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[cfg(not(windows))]
#[tokio::test]
async fn dispatch_user_message_surfaces_continued_message_submit_stderr() {
let dir = TempDir::new().expect("tempdir");
let failing = write_message_submit_hook(
&dir,
"fail.sh",
r#"#!/bin/sh
printf '%s\n' 'soft failure' >&2
exit 9
"#,
);
let replacing = write_message_submit_hook(
&dir,
"replace.sh",
r#"#!/bin/sh
printf '%s\n' '{"text":"after soft failure"}'
"#,
);
let mut app = create_test_app();
configure_message_submit_hooks(&mut app, &dir, vec![failing, replacing]);
let mut engine = crate::core::engine::mock_engine_handle();
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("hello".to_string(), None),
)
.await
.expect("dispatch user message");
assert_eq!(app.status_message.as_deref(), Some("soft failure"));
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage { content, .. } => {
assert_eq!(content, "after soft failure");
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[tokio::test]
async fn dispatch_route_failure_leaves_loading_and_transcript_unchanged() {
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "lm-studio");
app.set_model_selection("local-model".to_string());
app.api_messages
.push(text_message("assistant", "existing conversation"));
app.add_message(HistoryCell::System {
content: "existing receipt".to_string(),
});
let mut custom = HashMap::new();
custom.insert(
"lm-studio".to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some("ftp://invalid.example/v1".to_string()),
model: Some("local-model".to_string()),
api_key: Some("local-test-key".to_string()),
..ProviderConfig::default()
},
);
let config = Config {
provider: Some("lm-studio".to_string()),
providers: Some(ProvidersConfig {
custom,
..ProvidersConfig::default()
}),
..Config::default()
};
let mut engine = mock_engine_handle();
let err = dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("do not duplicate me".to_string(), None),
)
.await
.expect_err("malformed named custom route must fail closed");
assert!(err.to_string().contains("http"), "{err}");
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
assert!(app.last_send_at.is_none());
assert_eq!(app.api_messages.len(), 1);
assert_eq!(app.history.len(), 1);
assert!(matches!(
&app.history[0],
HistoryCell::System { content } if content == "existing receipt"
));
assert!(engine.rx_op.try_recv().is_err());
}
#[cfg(not(windows))]
#[tokio::test]
async fn dispatch_user_message_uses_transformed_message_submit_text() {
let dir = TempDir::new().expect("tempdir");
let command = write_message_submit_hook(
&dir,
"replace.sh",
r#"#!/bin/sh
printf '%s\n' '{"text":"[hooked] hello"}'
"#,
);
let mut app = create_test_app();
configure_single_message_submit_hook(&mut app, &dir, command);
let mut engine = crate::core::engine::mock_engine_handle();
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("hello".to_string(), None),
)
.await
.expect("dispatch user message");
assert_eq!(app.last_submitted_prompt.as_deref(), Some("[hooked] hello"));
assert!(app.history.iter().any(|cell| matches!(
cell,
HistoryCell::User { content } if content == "[hooked] hello"
)));
assert_eq!(app.api_messages.len(), 1);
assert!(matches!(
&app.api_messages[0].content[0],
ContentBlock::Text { text, .. } if text == "[hooked] hello"
));
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage { content, .. } => {
assert_eq!(content, "[hooked] hello");
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[cfg(not(windows))]
#[tokio::test]
async fn dispatch_user_message_blocked_by_message_submit_hook_does_not_start_turn() {
let dir = TempDir::new().expect("tempdir");
let command = write_message_submit_hook(
&dir,
"block.sh",
r#"#!/bin/sh
printf '%s\n' '{"reason":"blocked by test hook"}'
exit 2
"#,
);
let mut app = create_test_app();
configure_single_message_submit_hook(&mut app, &dir, command);
let mut engine = crate::core::engine::mock_engine_handle();
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("hello".to_string(), None),
)
.await
.expect("blocked submit is handled locally");
assert_eq!(app.status_message.as_deref(), Some("blocked by test hook"));
assert!(app.api_messages.is_empty());
assert!(
app.history
.iter()
.all(|cell| !matches!(cell, HistoryCell::User { .. }))
);
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
assert!(app.runtime_turn_status.is_none());
assert!(
engine.rx_op.try_recv().is_err(),
"blocked submit must not send any engine operation"
);
}
#[test]
fn resume_message_helper_is_strict() {
for message in [
"continue",
"resume",
"please continue",
"continue the paused command",
"can you resume the paused task",
"go ahead and resume",
] {
assert!(is_resume_message(message), "expected resume: {message}");
}
for message in [
"don't continue yet",
"do not resume yet",
"I will resume tomorrow",
"we can continue tomorrow",
"continue later",
"how do I resume a git cherry-pick?",
"please do not continue",
] {
assert!(
!is_resume_message(message),
"expected not resume: {message}"
);
}
}
#[tokio::test]
async fn dispatch_non_resume_message_preserves_paused_command_state() {
let mut app = create_test_app();
app.pausable = true;
app.paused = true;
app.paused_quarry = Some("Scan nested git repositories".to_string());
app.hunt.quarry = Some("Scan nested git repositories".to_string());
let mut engine = mock_engine_handle();
engine.handle.set_paused(true);
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("how are you?".to_string(), None),
)
.await
.expect("dispatch user message");
assert!(!app.paused);
assert!(app.pausable);
assert_eq!(
app.paused_quarry.as_deref(),
Some("Scan nested git repositories")
);
assert!(app.hunt.quarry.is_none());
assert!(!engine.handle.is_paused());
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage {
content,
goal_objective,
..
} => {
assert!(goal_objective.is_none());
assert!(content.contains("Paused custom slash command: Scan nested git repositories"));
assert!(content.contains("do not continue the paused command"));
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[tokio::test]
async fn dispatch_resume_message_restores_paused_command_goal() {
let mut app = create_test_app();
app.pausable = true;
app.paused = true;
app.paused_quarry = Some("Scan nested git repositories".to_string());
let mut engine = mock_engine_handle();
engine.handle.set_paused(true);
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("please continue the paused command".to_string(), None),
)
.await
.expect("dispatch user message");
assert!(!app.paused);
assert!(app.pausable);
assert!(app.paused_quarry.is_none());
assert_eq!(
app.hunt.quarry.as_deref(),
Some("Scan nested git repositories")
);
assert!(!engine.handle.is_paused());
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage {
content,
goal_objective,
..
} => {
assert_eq!(
goal_objective.as_deref(),
Some("Scan nested git repositories")
);
assert!(content.contains("Paused custom slash command: Scan nested git repositories"));
assert!(content.contains("Continue the paused command"));
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[tokio::test]
async fn dispatch_user_message_keeps_auto_review_separate_from_bypass() {
let mut app = create_test_app();
app.mode = AppMode::Agent;
app.approval_mode = ApprovalMode::Auto;
app.allow_shell = true;
app.trust_mode = true;
let mut engine = mock_engine_handle();
let config = Config::default();
dispatch_user_message(
&mut app,
&config,
&engine.handle,
QueuedMessage::new("run the local verification".to_string(), None),
)
.await
.expect("dispatch user message");
let pending_route = app
.pending_turn_route
.as_ref()
.expect("successful dispatch records route telemetry");
assert_eq!(pending_route.0, app.api_provider);
assert_eq!(pending_route.1, app.model);
assert!(!pending_route.2);
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage {
mode,
auto_approve,
approval_mode,
..
} => {
assert_eq!(mode, AppMode::Agent);
assert!(!auto_approve);
assert_eq!(approval_mode, ApprovalMode::Auto);
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[test]
fn apply_goal_snapshot_updates_visible_goal_status() {
let mut app = create_test_app();
app.hunt.quarry = Some("Ship the release lane".to_string());
app.hunt.token_budget = Some(10_000);
app.hunt.verdict = crate::tui::app::HuntVerdict::Hunting;
let started_at = Instant::now();
app.hunt.started_at = Some(started_at);
let completed = crate::tools::goal::GoalSnapshot {
objective: Some("Ship the release lane".to_string()),
status: "complete".to_string(),
token_budget: Some(10_000),
tokens_used: 12_345,
time_used_seconds: 12,
continuation_count: 2,
elapsed_seconds: Some(12),
evidence: Some("focused tests passed".to_string()),
blocker: None,
completion_verification: Some(crate::tools::goal::GoalCompletionVerification {
status: "passed".to_string(),
check: "cargo test".to_string(),
summary: "focused tests passed".to_string(),
}),
};
assert!(apply_goal_snapshot_to_app(&mut app, &completed));
assert_eq!(app.hunt.quarry.as_deref(), Some("Ship the release lane"));
assert_eq!(app.hunt.token_budget, Some(10_000));
assert_eq!(app.hunt.tokens_used, 12_345);
assert_eq!(app.hunt.time_used_seconds, 12);
assert_eq!(app.hunt.continuation_count, 2);
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Hunted);
assert_eq!(app.hunt.started_at, Some(started_at));
assert!(
app.hunt.finished_at.is_some(),
"terminal verdict should set finished_at so the timer freezes"
);
let blocked = crate::tools::goal::GoalSnapshot {
objective: Some("Different objective".to_string()),
status: "blocked".to_string(),
token_budget: None,
tokens_used: 12_345,
time_used_seconds: 13,
continuation_count: 3,
elapsed_seconds: Some(1),
evidence: None,
blocker: Some("needs user approval".to_string()),
completion_verification: None,
};
assert!(apply_goal_snapshot_to_app(&mut app, &blocked));
assert_eq!(app.hunt.quarry.as_deref(), Some("Different objective"));
assert_eq!(app.hunt.token_budget, None);
assert_eq!(app.hunt.tokens_used, 12_345);
assert_eq!(app.hunt.time_used_seconds, 13);
assert_eq!(app.hunt.continuation_count, 3);
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Escaped);
assert!(app.hunt.started_at.is_some());
assert!(
app.hunt.finished_at.is_some(),
"blocked verdict should also freeze the elapsed timer"
);
}
#[test]
fn apply_goal_snapshot_resume_clears_frozen_timer() {
let mut app = create_test_app();
app.hunt.quarry = Some("Ship the release lane".to_string());
app.hunt.verdict = crate::tui::app::HuntVerdict::Hunting;
app.hunt.started_at = Some(Instant::now());
let completed = crate::tools::goal::GoalSnapshot {
objective: Some("Ship the release lane".to_string()),
status: "complete".to_string(),
token_budget: None,
tokens_used: 0,
time_used_seconds: 0,
continuation_count: 0,
elapsed_seconds: Some(0),
evidence: Some("done".to_string()),
blocker: None,
completion_verification: Some(crate::tools::goal::GoalCompletionVerification {
status: "passed".to_string(),
check: "cargo test".to_string(),
summary: "ok".to_string(),
}),
};
assert!(apply_goal_snapshot_to_app(&mut app, &completed));
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Hunted);
assert!(app.hunt.finished_at.is_some());
let resumed = crate::tools::goal::GoalSnapshot {
objective: Some("Ship the release lane".to_string()),
status: "active".to_string(),
token_budget: None,
tokens_used: 0,
time_used_seconds: 0,
continuation_count: 0,
elapsed_seconds: Some(0),
evidence: None,
blocker: None,
completion_verification: None,
};
assert!(apply_goal_snapshot_to_app(&mut app, &resumed));
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Hunting);
assert!(
app.hunt.finished_at.is_none(),
"resume should re-arm the elapsed timer"
);
}
#[test]
fn apply_goal_snapshot_keeps_paused_timer_frozen_across_usage_updates() {
let mut app = create_test_app();
app.hunt.quarry = Some("Ship the release lane".to_string());
app.hunt.verdict = crate::tui::app::HuntVerdict::Hunting;
app.hunt.started_at = Some(Instant::now());
let paused = crate::tools::goal::GoalSnapshot {
objective: Some("Ship the release lane".to_string()),
status: "paused".to_string(),
token_budget: None,
tokens_used: 100,
time_used_seconds: 10,
continuation_count: 0,
elapsed_seconds: Some(10),
evidence: None,
blocker: None,
completion_verification: None,
};
assert!(apply_goal_snapshot_to_app(&mut app, &paused));
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Wounded);
let frozen_at = app
.hunt
.finished_at
.expect("pausing must freeze the elapsed timer");
let paused_with_usage = crate::tools::goal::GoalSnapshot {
objective: Some("Ship the release lane".to_string()),
status: "paused".to_string(),
token_budget: None,
tokens_used: 250,
time_used_seconds: 25,
continuation_count: 0,
elapsed_seconds: Some(25),
evidence: None,
blocker: None,
completion_verification: None,
};
assert!(apply_goal_snapshot_to_app(&mut app, &paused_with_usage));
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Wounded);
assert_eq!(
app.hunt.finished_at,
Some(frozen_at),
"a paused goal's frozen timer must stay frozen when usage updates arrive"
);
let resumed = crate::tools::goal::GoalSnapshot {
objective: Some("Ship the release lane".to_string()),
status: "active".to_string(),
token_budget: None,
tokens_used: 250,
time_used_seconds: 25,
continuation_count: 1,
elapsed_seconds: Some(25),
evidence: None,
blocker: None,
completion_verification: None,
};
assert!(apply_goal_snapshot_to_app(&mut app, &resumed));
assert_eq!(app.hunt.verdict, crate::tui::app::HuntVerdict::Hunting);
assert!(
app.hunt.finished_at.is_none(),
"resuming a paused goal should re-arm the elapsed timer"
);
}
#[test]
fn turn_liveness_watchdog_clears_stale_dispatch() {
let mut app = create_test_app();
app.is_loading = true;
app.dispatch_started_at =
Some(Instant::now() - DISPATCH_WATCHDOG_TIMEOUT - Duration::from_millis(1));
app.turn_started_at = Some(Instant::now());
app.pending_turn_route = Some((ApiProvider::Deepseek, "pending-model".to_string(), false));
app.suppress_stream_events_until_turn_complete = true;
let recovered = reconcile_turn_liveness(&mut app, Instant::now(), false);
assert!(recovered);
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
assert!(app.turn_started_at.is_none());
assert!(app.pending_turn_route.is_none());
assert!(app.active_turn.is_none());
assert!(!app.suppress_stream_events_until_turn_complete);
let toast = app.status_toasts.back().expect("watchdog toast");
assert_eq!(toast.level, StatusToastLevel::Error);
assert!(toast.text.contains("Turn dispatch timed out"));
}
#[test]
fn turn_liveness_reconciles_completed_busy_state() {
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_status = Some("completed".to_string());
app.dispatch_started_at = Some(Instant::now());
app.turn_started_at = Some(Instant::now());
let recovered = reconcile_turn_liveness(&mut app, Instant::now(), false);
assert!(recovered);
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
assert!(app.turn_started_at.is_none());
let toast = app.status_toasts.back().expect("reconciliation toast");
assert_eq!(toast.level, StatusToastLevel::Warning);
assert!(
toast
.text
.contains("Recovered from an inconsistent busy state")
);
}
#[test]
fn turn_liveness_leaves_active_turn_running() {
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.turn_started_at = Some(Instant::now() - Duration::from_secs(60));
let recovered = reconcile_turn_liveness(&mut app, Instant::now(), false);
assert!(!recovered);
assert!(app.is_loading);
assert!(app.turn_started_at.is_some());
assert!(app.status_toasts.is_empty());
}
#[test]
fn turn_liveness_uses_recent_turn_activity_not_turn_start() {
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.turn_started_at = Some(Instant::now());
app.turn_last_activity_at =
Some(app.turn_started_at.unwrap() + TURN_STALL_WATCHDOG_TIMEOUT + Duration::from_secs(29));
let now = app.turn_last_activity_at.unwrap() + Duration::from_secs(1);
let recovered = reconcile_turn_liveness(&mut app, now, false);
assert!(!recovered);
assert!(app.is_loading);
assert!(app.runtime_turn_status.is_some());
assert!(app.status_toasts.is_empty());
}
#[test]
fn turn_liveness_does_not_abort_running_tool() {
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.turn_started_at = Some(Instant::now());
app.turn_last_activity_at = app.turn_started_at;
let now = app.turn_started_at.unwrap()
+ TURN_STALL_WATCHDOG_TIMEOUT
+ Duration::from_secs(30)
+ Duration::from_secs(1);
let mut active = ActiveCell::new();
active.push_tool(
"tool-1",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "edit_file".to_string(),
status: ToolStatus::Running,
input_summary: Some("path: CHANGELOG.md".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
let recovered = reconcile_turn_liveness(&mut app, now, false);
assert!(!recovered);
assert!(app.is_loading);
assert!(app.active_cell.is_some());
assert!(app.status_toasts.is_empty());
}
#[test]
fn turn_liveness_does_not_abort_running_tool_with_recent_heartbeat() {
let mut app = create_test_app();
let started_at = Instant::now();
let now = started_at + TOOL_HANG_WATCHDOG_TIMEOUT + Duration::from_secs(30);
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.turn_started_at = Some(started_at);
app.turn_last_activity_at = Some(now - Duration::from_secs(10));
let mut active = ActiveCell::new();
active.push_tool(
"tool-1",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Running,
input_summary: Some("command: cargo test".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
let recovered = reconcile_turn_liveness(&mut app, now, false);
assert!(!recovered);
assert!(app.is_loading);
assert!(app.active_cell.is_some());
assert!(app.status_toasts.is_empty());
}
#[test]
fn turn_liveness_respects_stream_idle_budget_for_quiet_model_waits() {
let mut app = create_test_app();
let started_at = Instant::now();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.stream_chunk_timeout_secs = 900;
app.turn_started_at = Some(started_at);
app.turn_last_activity_at = Some(started_at);
let now = started_at + TURN_STALL_WATCHDOG_TIMEOUT + Duration::from_secs(31);
let recovered = reconcile_turn_liveness(&mut app, now, false);
assert!(!recovered);
assert!(app.is_loading);
assert!(app.status_toasts.is_empty());
}
#[test]
fn turn_liveness_recovers_running_tool_without_heartbeat() {
let mut app = create_test_app();
let started_at = Instant::now();
let now = started_at + TOOL_HANG_WATCHDOG_TIMEOUT + Duration::from_secs(1);
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.runtime_turn_id = Some("stale-tool-turn".to_string());
app.turn_started_at = Some(started_at);
app.turn_last_activity_at = app.turn_started_at;
app.user_scrolled_during_stream = true;
let mut active = ActiveCell::new();
active.push_tool(
"tool-1",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Running,
input_summary: Some("command: cargo test".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
let recovered = reconcile_turn_liveness(&mut app, now, false);
assert!(recovered);
assert!(!app.is_loading);
assert!(app.turn_started_at.is_none());
assert!(app.runtime_turn_status.is_none());
assert!(app.runtime_turn_id.is_none());
assert!(!app.user_scrolled_during_stream);
let toast = app.status_toasts.back().expect("tool hang toast");
assert_eq!(toast.level, StatusToastLevel::Error);
assert!(toast.text.contains("Tool stalled with no progress"));
}
#[test]
fn turn_liveness_recovers_stalled_in_progress_turn() {
let mut app = create_test_app();
let now = Instant::now();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.runtime_turn_id = Some("stale-turn-id".to_string());
app.stream_chunk_timeout_secs = 300;
app.turn_started_at = Some(now - turn_stall_watchdog_timeout(&app) - Duration::from_millis(1));
app.streaming_message_index = Some(0);
app.user_scrolled_during_stream = true;
app.pending_turn_route = Some((ApiProvider::Deepseek, "pending-model".to_string(), false));
app.active_turn = Some(crate::tui::app::ActiveTurnMetadata {
turn_id: "stale-turn-id".to_string(),
created_at: chrono::Utc::now(),
route: Some(crate::core::events::TurnRoute {
provider: ApiProvider::Openai,
provider_identity: "openai".to_string(),
model: "gpt-5.5".to_string(),
auto_model: false,
}),
});
let recovered = reconcile_turn_liveness(&mut app, now, false);
assert!(recovered);
assert!(!app.is_loading);
assert!(app.turn_started_at.is_none());
assert!(app.runtime_turn_status.is_none());
assert!(app.runtime_turn_id.is_none());
assert!(app.dispatch_started_at.is_none());
assert!(app.streaming_message_index.is_none());
assert!(app.streaming_thinking_active_entry.is_none());
assert!(!app.user_scrolled_during_stream);
assert!(app.pending_turn_route.is_none());
assert!(app.active_turn.is_none());
assert!(!app.suppress_stream_events_until_turn_complete);
let toast = app.status_toasts.back().expect("stall toast");
assert_eq!(toast.level, StatusToastLevel::Error);
assert!(toast.text.contains("Turn stalled"));
}
#[test]
fn engine_event_disconnect_recovers_live_turn_immediately() {
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.runtime_turn_id = Some("turn_dead".to_string());
app.turn_started_at = Some(Instant::now());
app.dispatch_started_at = Some(Instant::now());
app.user_scrolled_during_stream = true;
app.pending_turn_route = Some((ApiProvider::Deepseek, "pending-model".to_string(), false));
app.active_turn = Some(crate::tui::app::ActiveTurnMetadata {
turn_id: "turn_dead".to_string(),
created_at: chrono::Utc::now(),
route: Some(crate::core::events::TurnRoute {
provider: ApiProvider::Openai,
provider_identity: "openai".to_string(),
model: "gpt-5.5".to_string(),
auto_model: false,
}),
});
let thinking_idx = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
crate::tui::streaming_thinking::append(&mut app, thinking_idx, "partial reasoning");
app.push_pending_steer(crate::tui::app::QueuedMessage::new(
"please continue after recovery".to_string(),
None,
));
let recovered = recover_engine_event_disconnect(&mut app);
assert!(recovered);
assert!(!app.is_loading);
assert!(app.runtime_turn_status.is_none());
assert!(app.runtime_turn_id.is_none());
assert!(app.dispatch_started_at.is_none());
assert!(app.turn_started_at.is_none());
assert!(app.streaming_thinking_active_entry.is_none());
assert!(!app.user_scrolled_during_stream);
assert!(app.pending_turn_route.is_none());
assert!(app.active_turn.is_none());
assert!(!app.suppress_stream_events_until_turn_complete);
assert_eq!(app.queued_message_count(), 1);
assert_eq!(
app.queued_messages
.front()
.map(crate::tui::app::QueuedMessage::content),
Some("please continue after recovery".to_string())
);
assert!(
app.history.iter().any(|cell| matches!(
cell,
HistoryCell::Error { message, .. }
if message.contains("Engine stopped before completing the turn")
)),
"disconnect recovery should add a visible transcript error"
);
let toast = app.status_toasts.back().expect("disconnect toast");
assert_eq!(toast.level, StatusToastLevel::Error);
}
#[test]
fn engine_event_disconnect_while_idle_is_noop() {
let mut app = create_test_app();
let recovered = recover_engine_event_disconnect(&mut app);
assert!(!recovered);
assert!(app.history.is_empty());
assert!(app.status_toasts.is_empty());
}
#[test]
fn engine_event_disconnect_cleans_cancelled_turn_metadata() {
let mut app = create_test_app();
app.suppress_stream_events_until_turn_complete = true;
app.active_turn = Some(crate::tui::app::ActiveTurnMetadata {
turn_id: "cancelled-turn".to_string(),
created_at: chrono::Utc::now(),
route: Some(crate::core::events::TurnRoute {
provider: ApiProvider::Openai,
provider_identity: "openai".to_string(),
model: "gpt-5.5".to_string(),
auto_model: false,
}),
});
let recovered = recover_engine_event_disconnect(&mut app);
assert!(recovered);
assert!(app.active_turn.is_none());
assert!(!app.suppress_stream_events_until_turn_complete);
assert!(app.history.iter().any(|cell| matches!(
cell,
HistoryCell::Error { message, .. }
if message.contains("Engine stopped before completing the turn")
)));
}
#[test]
fn fixed_model_auto_thinking_skips_auto_model_router() {
let mut app = create_test_app();
app.auto_model = false;
app.model = "deepseek-v4-pro".to_string();
app.reasoning_effort = ReasoningEffort::Auto;
assert!(
!crate::tui::auto_router::should_resolve_auto_model_selection(&app),
"fixed-model auto thinking must stay local instead of starting a hidden router request"
);
}
#[test]
fn auto_model_still_uses_auto_model_router() {
let mut app = create_test_app();
app.auto_model = true;
app.reasoning_effort = ReasoningEffort::Auto;
assert!(
crate::tui::auto_router::should_resolve_auto_model_selection(&app),
"auto model still needs the router to choose the concrete model"
);
}
fn init_git_repo() -> TempDir {
let dir = tempfile::tempdir().expect("tempdir");
let init = Command::new("git")
.arg("init")
.current_dir(dir.path())
.output()
.expect("git init should run");
assert!(
init.status.success(),
"git init failed: {}",
String::from_utf8_lossy(&init.stderr)
);
let autocrlf = Command::new("git")
.args(["config", "core.autocrlf", "false"])
.current_dir(dir.path())
.output()
.expect("git config core.autocrlf should run");
assert!(
autocrlf.status.success(),
"git config core.autocrlf failed: {}",
String::from_utf8_lossy(&autocrlf.stderr)
);
let commit = Command::new("git")
.args([
"-c",
"user.name=codewhale Tests",
"-c",
"user.email=tests@example.com",
"-c",
"commit.gpgsign=false",
"commit",
"--allow-empty",
"-m",
"init",
])
.current_dir(dir.path())
.output()
.expect("git commit should run");
assert!(
commit.status.success(),
"git commit failed: {}",
String::from_utf8_lossy(&commit.stderr)
);
dir
}
fn spans_text(spans: &[Span<'_>]) -> String {
spans
.iter()
.map(|span| span.content.as_ref())
.collect::<String>()
}
#[test]
fn ctrl_alt_4_focuses_agents_sidebar_without_switching_modes() {
let mut app = create_test_app();
app.mode = AppMode::Agent;
app.sidebar_focus = SidebarFocus::Auto;
apply_alt_4_shortcut(&mut app, KeyModifiers::ALT | KeyModifiers::CONTROL);
assert_eq!(app.mode, AppMode::Agent);
assert_eq!(app.sidebar_focus, SidebarFocus::Agents);
assert_eq!(app.status_message.as_deref(), Some("Sidebar focus: agents"));
}
#[test]
fn hotbar_bare_digit_inserts_text_even_when_composer_empty() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
let bare_four = KeyEvent::new(KeyCode::Char('4'), KeyModifiers::NONE);
assert_eq!(hotbar_slot_from_key(&app, &bare_four), None);
app.input = "draft".to_string();
assert_eq!(hotbar_slot_from_key(&app, &bare_four), None);
app.input = " ".to_string();
assert_eq!(hotbar_slot_from_key(&app, &bare_four), None);
}
#[test]
fn hotbar_alt_digit_fires_from_composer_and_sidebar_states() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
let alt_four = KeyEvent::new(KeyCode::Char('4'), KeyModifiers::ALT);
assert_eq!(hotbar_slot_from_key(&app, &alt_four), Some(4));
app.input = "draft".to_string();
assert_eq!(hotbar_slot_from_key(&app, &alt_four), Some(4));
app.input = " ".to_string();
assert_eq!(hotbar_slot_from_key(&app, &alt_four), Some(4));
app.sidebar_focus = SidebarFocus::Hidden;
assert_eq!(hotbar_slot_from_key(&app, &alt_four), Some(4));
app.sidebar_focus = SidebarFocus::Agents;
assert_eq!(hotbar_slot_from_key(&app, &alt_four), Some(4));
}
#[test]
fn hotbar_alt_digit_requires_plain_alt_one_through_eight() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
assert_eq!(
hotbar_slot_from_key(
&app,
&KeyEvent::new(
KeyCode::Char('4'),
KeyModifiers::ALT | KeyModifiers::CONTROL
)
),
None
);
assert_eq!(
hotbar_slot_from_key(
&app,
&KeyEvent::new(KeyCode::Char('4'), KeyModifiers::ALT | KeyModifiers::SUPER)
),
None
);
assert_eq!(
hotbar_slot_from_key(&app, &KeyEvent::new(KeyCode::Char('0'), KeyModifiers::ALT)),
None
);
assert_eq!(
hotbar_slot_from_key(&app, &KeyEvent::new(KeyCode::Char('9'), KeyModifiers::ALT)),
None
);
}
#[test]
fn hotbar_digits_are_blocked_while_modal_or_onboarding_is_active() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
app.view_stack.push(HelpView::new());
let bare_four = KeyEvent::new(KeyCode::Char('4'), KeyModifiers::NONE);
let alt_four = KeyEvent::new(KeyCode::Char('4'), KeyModifiers::ALT);
assert_eq!(hotbar_slot_from_key(&app, &bare_four), None);
assert_eq!(hotbar_slot_from_key(&app, &alt_four), None);
let mut app = create_test_app();
app.onboarding = OnboardingState::Language;
assert_eq!(hotbar_slot_from_key(&app, &bare_four), None);
assert_eq!(hotbar_slot_from_key(&app, &alt_four), None);
}
#[test]
fn hotbar_alt_digit_is_blocked_while_inline_selectors_are_open() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
app.input = "/".to_string();
app.cursor_position = app.input.chars().count();
app.slash_menu_hidden = false;
assert!(
!visible_slash_menu_entries(&app, SLASH_MENU_LIMIT).is_empty(),
"precondition: slash menu should be visible"
);
let alt_four = KeyEvent::new(KeyCode::Char('4'), KeyModifiers::ALT);
assert_eq!(hotbar_slot_from_key(&app, &alt_four), None);
app.input = "draft".to_string();
app.cursor_position = app.input.chars().count();
app.start_history_search();
assert!(app.is_history_search_active());
assert_eq!(hotbar_slot_from_key(&app, &alt_four), None);
}
#[test]
fn decision_card_numeric_shortcuts_accept_bare_digits_only() {
assert_eq!(
decision_card_number_from_key(&KeyEvent::new(KeyCode::Char('4'), KeyModifiers::NONE)),
Some(4)
);
assert_eq!(
decision_card_number_from_key(&KeyEvent::new(KeyCode::Char('4'), KeyModifiers::ALT)),
None
);
assert_eq!(
decision_card_number_from_key(&KeyEvent::new(KeyCode::Char('4'), KeyModifiers::CONTROL)),
None
);
}
#[test]
fn hotbar_alt_digit_is_blocked_while_decision_card_is_active() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
app.decision_card = Some(crate::tui::widgets::decision_card::DecisionCard::new(
"Pick one".to_string(),
vec![
crate::tui::widgets::decision_card::DecisionOption {
label: "First".to_string(),
description: None,
},
crate::tui::widgets::decision_card::DecisionOption {
label: "Second".to_string(),
description: None,
},
],
0,
));
assert_eq!(
hotbar_slot_from_key(&app, &KeyEvent::new(KeyCode::Char('1'), KeyModifiers::ALT)),
None
);
}
#[test]
fn hotbar_dispatches_bound_slot_and_ignores_empty_slot() {
let mut app = create_test_app();
let config = Config {
hotbar: Some(codewhale_config::default_hotbar_bindings_toml()),
..Config::default()
};
app.onboarding = OnboardingState::None;
app.mode = AppMode::Plan;
app.needs_redraw = false;
let dispatch = dispatch_hotbar_slot(&mut app, &config, 4).expect("hotbar dispatch");
assert!(matches!(
dispatch,
Some(HotbarDispatch::AppAction(AppAction::ModeChanged(
AppMode::Agent
)))
));
assert_eq!(app.mode, AppMode::Agent);
assert!(
app.needs_redraw,
"mode-changing hotbar actions should leave the app ready to redraw"
);
let empty_config = Config {
hotbar: Some(Vec::new()),
..Config::default()
};
assert_eq!(
dispatch_hotbar_slot(&mut app, &empty_config, 1).expect("empty slot is ok"),
None
);
}
#[test]
fn hotbar_dispatches_slash_command_slot() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
let config = Config {
hotbar: Some(vec![codewhale_config::HotbarBindingToml {
slot: 1,
label: Some("mode".to_string()),
action: "slash.mode".to_string(),
}]),
..Config::default()
};
assert_eq!(
dispatch_hotbar_slot(&mut app, &config, 1).expect("slash slot dispatch"),
Some(HotbarDispatch::AppAction(AppAction::OpenModePicker))
);
assert!(app.input.is_empty());
}
#[test]
fn hotbar_dispatches_route_switch_slot() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
let route_metadata = app
.hotbar_actions
.iter()
.map(|action| action.metadata(crate::localization::Locale::En))
.find(|metadata| metadata.category == HotbarActionCategory::Route)
.expect("test app should register at least the active provider route");
let route_id = route_metadata.id.clone();
let route_suffix = route_metadata
.id
.strip_prefix("route.")
.expect("route id prefix");
let (provider_key, model) = route_suffix.split_once('.').expect("route id shape");
let provider = ApiProvider::parse(provider_key).expect("provider key parses");
let model = model.to_string();
let config = Config {
hotbar: Some(vec![codewhale_config::HotbarBindingToml {
slot: 1,
label: Some(route_metadata.compact_label),
action: route_id,
}]),
..Config::default()
};
assert_eq!(
dispatch_hotbar_slot(&mut app, &config, 1).expect("route slot dispatch"),
Some(HotbarDispatch::AppAction(AppAction::SwitchModelRoute {
provider,
model,
}))
);
}
#[test]
fn hotbar_bound_disabled_action_reports_reason_without_dispatching() {
let mut app = create_test_app();
app.onboarding = OnboardingState::None;
app.auto_model = true;
app.reasoning_effort = ReasoningEffort::Off;
app.needs_redraw = false;
let config = Config {
hotbar: Some(vec![codewhale_config::HotbarBindingToml {
slot: 1,
label: Some("reason".to_string()),
action: "reasoning.cycle".to_string(),
}]),
..Config::default()
};
assert_eq!(
dispatch_hotbar_slot(&mut app, &config, 1).expect("disabled slot dispatch"),
Some(HotbarDispatch::Handled)
);
assert_eq!(app.reasoning_effort, ReasoningEffort::Off);
assert_eq!(
app.status_message.as_deref(),
Some(
"Hotbar slot 1 action is not available: Reasoning effort is controlled by auto model routing."
)
);
assert!(app.needs_redraw);
}
#[test]
fn alt_0_restores_auto_sidebar_focus() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Hidden;
apply_alt_0_shortcut(&mut app, KeyModifiers::ALT);
assert_eq!(app.sidebar_focus, SidebarFocus::Auto);
assert_eq!(app.status_message.as_deref(), Some("Sidebar focus: auto"));
}
#[test]
fn ctrl_alt_0_hides_sidebar() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Tasks;
apply_alt_0_shortcut(&mut app, KeyModifiers::ALT | KeyModifiers::CONTROL);
assert_eq!(app.sidebar_focus, SidebarFocus::Hidden);
assert!(app.sidebar_focus_dirty);
assert_eq!(app.status_message.as_deref(), Some("Sidebar hidden"));
}
#[test]
fn ctrl_alt_0_restores_pinned_sidebar_when_already_hidden() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Hidden;
apply_alt_0_shortcut(&mut app, KeyModifiers::ALT | KeyModifiers::CONTROL);
assert_eq!(app.sidebar_focus, SidebarFocus::Pinned);
assert_eq!(app.status_message.as_deref(), Some("Sidebar focus: pinned"));
}
#[test]
fn sidebar_focus_dirty_persists_saved_focus() {
let _guard = ConfigPathEnvGuard::new();
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Hidden;
app.sidebar_focus_dirty = true;
persist_sidebar_settings_if_dirty(&mut app);
assert!(!app.sidebar_focus_dirty);
let settings = crate::settings::Settings::load().expect("load settings");
assert_eq!(settings.sidebar_focus, "hidden");
}
#[test]
fn hidden_sidebar_focus_suppresses_sidebar_split_even_when_wide() {
let mut app = create_test_app();
app.sidebar_width_percent = 28;
app.sidebar_focus = SidebarFocus::Pinned;
assert_eq!(sidebar_width_for_chat_area(&app, 120), Some(33));
app.sidebar_focus = SidebarFocus::Hidden;
assert_eq!(sidebar_width_for_chat_area(&app, 120), None);
}
#[test]
fn compact_sidebar_split_survives_eighty_column_file_tree_host() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Pinned;
assert_eq!(sidebar_width_for_chat_area(&app, 60), Some(20));
assert_eq!(sidebar_width_for_chat_area(&app, 59), None);
}
#[test]
fn sidebar_width_gate_uses_compact_sixty_column_boundary() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Pinned;
app.last_sidebar_host_width = Some(SIDEBAR_VISIBLE_MIN_WIDTH - 1);
assert_eq!(
sidebar_render_state(&mut app),
SidebarRenderState::SuppressedByWidth {
available_width: SIDEBAR_VISIBLE_MIN_WIDTH - 1,
min_width: SIDEBAR_VISIBLE_MIN_WIDTH,
}
);
app.last_sidebar_host_width = Some(SIDEBAR_VISIBLE_MIN_WIDTH);
assert_eq!(sidebar_render_state(&mut app), SidebarRenderState::Visible);
}
#[test]
fn pinned_sidebar_is_visible_when_idle_and_wide() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Pinned;
app.last_sidebar_host_width = Some(120);
assert_eq!(sidebar_render_state(&mut app), SidebarRenderState::Visible);
}
#[test]
fn auto_sidebar_status_reports_idle_collapse_when_wide() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Auto;
app.last_sidebar_host_width = Some(120);
assert_eq!(
sidebar_render_state(&mut app),
SidebarRenderState::AutoCollapsed
);
}
#[test]
fn sidebar_auto_idle_collapses_when_nothing_active() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Auto;
assert!(crate::tui::sidebar::sidebar_auto_idle(&mut app));
}
#[test]
fn sidebar_auto_idle_false_when_fleet_active() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Auto;
app.agent_progress
.insert("agent_1".to_string(), "running".to_string());
assert!(!crate::tui::sidebar::sidebar_auto_idle(&mut app));
}
#[test]
fn sidebar_auto_idle_false_for_explicit_focus() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Agents;
assert!(!crate::tui::sidebar::sidebar_auto_idle(&mut app));
}
#[test]
fn jobs_panel_ignores_completed_history_but_shows_for_real_jobs() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Auto;
app.task_panel.push(crate::tui::app::TaskPanelEntry {
id: "shell_1".to_string(),
status: "completed".to_string(),
prompt_summary: "shell: cargo fmt".to_string(),
duration_ms: Some(10),
kind: crate::tui::app::TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
});
assert!(
crate::tui::sidebar::sidebar_auto_idle(&mut app),
"completed background jobs must not reopen the auto jobs panel"
);
app.task_panel.push(crate::tui::app::TaskPanelEntry {
id: "shell_2".to_string(),
status: "running".to_string(),
prompt_summary: "shell: cargo test".to_string(),
duration_ms: Some(10),
kind: crate::tui::app::TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
});
assert!(
!crate::tui::sidebar::sidebar_auto_idle(&mut app),
"a live background job must surface the jobs panel"
);
}
#[test]
fn ctrl_x_jobs_prefill_only_catches_running_shell_jobs_in_tasks_sidebar() {
let mut app = create_test_app();
app.sidebar_focus = SidebarFocus::Tasks;
app.input = "draft".to_string();
app.cursor_position = app.input.len();
app.task_panel.push(TaskPanelEntry {
id: "shell_active".to_string(),
status: "running".to_string(),
prompt_summary: "shell: cargo test".to_string(),
duration_ms: Some(10),
kind: TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
});
assert!(prefill_jobs_cancel_all_if_tasks_sidebar(&mut app));
assert_eq!(app.input, "/jobs cancel-all");
assert_eq!(app.cursor_position, app.input.len());
assert_eq!(
app.status_message.as_deref(),
Some("Press Enter to cancel all running commands")
);
}
#[test]
fn ctrl_x_jobs_prefill_falls_through_outside_tasks_sidebar_shell_jobs() {
let mut non_shell = create_test_app();
non_shell.sidebar_focus = SidebarFocus::Tasks;
non_shell.input = "draft".to_string();
non_shell.cursor_position = non_shell.input.len();
non_shell.task_panel.push(TaskPanelEntry {
id: "task_active".to_string(),
status: "running".to_string(),
prompt_summary: "summarize the release notes".to_string(),
duration_ms: Some(10),
kind: TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
});
assert!(!prefill_jobs_cancel_all_if_tasks_sidebar(&mut non_shell));
assert_eq!(non_shell.input, "draft");
let mut other_sidebar = create_test_app();
other_sidebar.sidebar_focus = SidebarFocus::Agents;
other_sidebar.input = "draft".to_string();
other_sidebar.cursor_position = other_sidebar.input.len();
other_sidebar.task_panel.push(TaskPanelEntry {
id: "shell_active".to_string(),
status: "running".to_string(),
prompt_summary: "shell: cargo test".to_string(),
duration_ms: Some(10),
kind: TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
});
assert!(!prefill_jobs_cancel_all_if_tasks_sidebar(
&mut other_sidebar
));
assert_eq!(other_sidebar.input, "draft");
}
fn setup_resize_handle(app: &mut App, handle_x: u16, sidebar_width: u16, total_width: u16) {
let y = 2;
let h = 10;
app.last_sidebar_handle_area = Some(Rect {
x: handle_x,
y,
width: 1,
height: h,
});
app.last_sidebar_area = Some(Rect {
x: handle_x,
y,
width: sidebar_width,
height: h,
});
app.sidebar_resize_total_width = total_width;
app.sidebar_width_percent = 28;
}
#[test]
fn sidebar_resize_down_on_handle_starts_resizing() {
let mut app = create_test_app();
setup_resize_handle(&mut app, 80, 33, 120);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 80,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(
app.sidebar_resizing,
"should start resizing on handle click"
);
assert_eq!(app.sidebar_resize_anchor_x, 80);
assert_eq!(app.sidebar_resize_anchor_width, 33);
}
#[test]
fn sidebar_resize_down_outside_handle_does_not_start_resizing() {
let mut app = create_test_app();
setup_resize_handle(&mut app, 80, 33, 120);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Down(MouseButton::Left),
column: 79, row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(
!app.sidebar_resizing,
"should not resize on non-handle click"
);
}
#[test]
fn sidebar_resize_drag_adjusts_width_percent() {
let mut app = create_test_app();
setup_resize_handle(&mut app, 80, 33, 120);
app.sidebar_width_percent = 28;
app.sidebar_resizing = true;
app.sidebar_resize_anchor_x = 80;
app.sidebar_resize_anchor_width = 33;
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 76,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
let expected = ((37u32 * 100) / 120) as u16; assert_eq!(app.sidebar_width_percent, expected);
}
#[test]
fn sidebar_resize_drag_clamps_to_10_50_range() {
let mut app = create_test_app();
setup_resize_handle(&mut app, 80, 33, 120);
app.sidebar_resizing = true;
app.sidebar_resize_anchor_x = 80;
app.sidebar_resize_anchor_width = 33;
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 200,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(app.sidebar_width_percent >= 10);
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Drag(MouseButton::Left),
column: 0,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(app.sidebar_width_percent <= 50);
}
#[test]
fn sidebar_resize_up_ends_resizing_and_marks_dirty() {
let mut app = create_test_app();
setup_resize_handle(&mut app, 80, 33, 120);
app.sidebar_resizing = true;
app.sidebar_resize_anchor_x = 80;
app.sidebar_resize_anchor_width = 33;
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Up(MouseButton::Left),
column: 76,
row: 5,
modifiers: KeyModifiers::NONE,
},
);
assert!(!app.sidebar_resizing, "should stop resizing on mouse up");
assert!(
app.sidebar_width_dirty,
"should mark width dirty for persistence"
);
}
#[test]
fn sidebar_resize_up_outside_handle_still_ends_resizing() {
let mut app = create_test_app();
setup_resize_handle(&mut app, 80, 33, 120);
app.sidebar_resizing = true;
app.sidebar_resize_anchor_x = 80;
app.sidebar_resize_anchor_width = 33;
handle_mouse_event(
&mut app,
MouseEvent {
kind: MouseEventKind::Up(MouseButton::Left),
column: 5,
row: 20,
modifiers: KeyModifiers::NONE,
},
);
assert!(
!app.sidebar_resizing,
"mouse up must clear resize state even outside the handle"
);
assert!(app.sidebar_width_dirty);
}
fn make_subagent(
id: &str,
status: crate::tools::subagent::SubAgentStatus,
) -> crate::tools::subagent::SubAgentResult {
crate::tools::subagent::SubAgentResult {
name: id.to_string(),
agent_id: id.to_string(),
context_mode: "fresh".to_string(),
fork_context: false,
workspace: None,
git_branch: None,
agent_type: crate::tools::subagent::SubAgentType::General,
assignment: crate::tools::subagent::SubAgentAssignment {
objective: format!("objective-{id}"),
role: Some("worker".to_string()),
},
model: "deepseek-v4-flash".to_string(),
nickname: None,
status,
worker_status: None,
parent_run_id: None,
spawn_depth: 0,
result: None,
steps_taken: 0,
checkpoint: None,
needs_input: None,
duration_ms: 0,
from_prior_session: false,
}
}
#[test]
fn sort_subagents_orders_running_before_terminal_statuses() {
let mut agents = vec![
make_subagent("agent_c", crate::tools::subagent::SubAgentStatus::Completed),
make_subagent("agent_a", crate::tools::subagent::SubAgentStatus::Running),
make_subagent(
"agent_b",
crate::tools::subagent::SubAgentStatus::Failed("boom".to_string()),
),
];
sort_subagents_in_place(&mut agents);
assert_eq!(agents[0].agent_id, "agent_a");
assert_eq!(agents[1].agent_id, "agent_b");
assert_eq!(agents[2].agent_id, "agent_c");
}
#[test]
fn subagent_hook_preview_is_bounded_on_char_boundaries() {
let text = format!("{}{}", "鲸".repeat(900), "tail");
let (preview, truncated) = bounded_subagent_hook_preview(&text);
assert!(truncated);
assert!(preview.ends_with("...[truncated]"));
assert!(preview.len() <= SUBAGENT_HOOK_PREVIEW_LIMIT + "...[truncated]".len());
assert!(preview.is_char_boundary(preview.len()));
}
#[test]
fn subagent_completion_status_reads_done_sentinel() {
let result = r#"done
<codewhale:subagent.done>{"agent_id":"agent_x","status":"completed"}</codewhale:subagent.done>"#;
assert_eq!(
subagent_completion_status(result).as_deref(),
Some("completed")
);
assert_eq!(subagent_completion_status("no sentinel"), None);
}
#[test]
fn subagent_completion_status_reads_summary_fallbacks() {
assert_eq!(
subagent_completion_status("Cancelled").as_deref(),
Some("cancelled")
);
assert_eq!(
subagent_completion_status("Failed: tool timed out").as_deref(),
Some("failed")
);
assert_eq!(
subagent_completion_status("Interrupted: process restarted").as_deref(),
Some("interrupted")
);
}
#[test]
fn subagent_status_from_completion_result_maps_terminal_sentinels() {
let failed = r#"Tool timed out
<codewhale:subagent.done>{"agent_id":"agent_x","status":"failed"}</codewhale:subagent.done>"#;
match subagent_status_from_completion_result(failed) {
crate::tools::subagent::SubAgentStatus::Failed(reason) => {
assert_eq!(reason, "Tool timed out")
}
status => panic!("expected failed status, got {status:?}"),
}
let interrupted = r#"Waiting for follow-up
<codewhale:subagent.done>{"agent_id":"agent_x","status":"interrupted"}</codewhale:subagent.done>"#;
match subagent_status_from_completion_result(interrupted) {
crate::tools::subagent::SubAgentStatus::Interrupted(reason) => {
assert_eq!(reason, "Waiting for follow-up")
}
status => panic!("expected interrupted status, got {status:?}"),
}
let budget = r#"Token budget exhausted
<codewhale:subagent.done>{"agent_id":"agent_x","status":"budget_exhausted"}</codewhale:subagent.done>"#;
assert_eq!(
subagent_status_from_completion_result(budget),
crate::tools::subagent::SubAgentStatus::BudgetExhausted
);
let cancelled = r#"Cancelled
<codewhale:subagent.done>{"agent_id":"agent_x","status":"cancelled"}</codewhale:subagent.done>"#;
assert_eq!(
subagent_status_from_completion_result(cancelled),
crate::tools::subagent::SubAgentStatus::Cancelled
);
assert_eq!(
subagent_status_from_completion_result("plain successful summary"),
crate::tools::subagent::SubAgentStatus::Completed
);
}
#[test]
fn subagent_terminal_projection_from_mailbox_maps_terminal_messages() {
let completed = crate::tools::subagent::MailboxMessage::Completed {
agent_id: "agent_done".to_string(),
summary: "all set".to_string(),
};
let (agent_id, status, result) =
subagent_terminal_projection_from_mailbox(&completed).expect("completed projection");
assert_eq!(agent_id, "agent_done");
assert_eq!(status, crate::tools::subagent::SubAgentStatus::Completed);
assert_eq!(result.as_deref(), Some("all set"));
let failed = crate::tools::subagent::MailboxMessage::Failed {
agent_id: "agent_fail".to_string(),
error: "tool failed".to_string(),
};
let (_, status, result) =
subagent_terminal_projection_from_mailbox(&failed).expect("failed projection");
assert_eq!(result.as_deref(), Some("tool failed"));
assert!(matches!(
status,
crate::tools::subagent::SubAgentStatus::Failed(ref reason)
if reason == "tool failed"
));
let interrupted = crate::tools::subagent::MailboxMessage::Interrupted {
agent_id: "agent_wait".to_string(),
reason: "needs input".to_string(),
};
let (_, status, result) =
subagent_terminal_projection_from_mailbox(&interrupted).expect("interrupted projection");
assert_eq!(result.as_deref(), Some("needs input"));
assert!(matches!(
status,
crate::tools::subagent::SubAgentStatus::Interrupted(ref reason)
if reason == "needs input"
));
let cancelled = crate::tools::subagent::MailboxMessage::Cancelled {
agent_id: "agent_stop".to_string(),
};
let (_, status, result) =
subagent_terminal_projection_from_mailbox(&cancelled).expect("cancelled projection");
assert_eq!(status, crate::tools::subagent::SubAgentStatus::Cancelled);
assert_eq!(result.as_deref(), Some("cancelled"));
assert!(
subagent_terminal_projection_from_mailbox(
&crate::tools::subagent::MailboxMessage::progress("agent_live", "step 1/2")
)
.is_none()
);
}
#[test]
fn running_agent_count_unions_cache_and_progress() {
let mut app = create_test_app();
app.subagent_cache = vec![
make_subagent("agent_a", crate::tools::subagent::SubAgentStatus::Running),
make_subagent("agent_b", crate::tools::subagent::SubAgentStatus::Completed),
];
app.agent_progress
.insert("agent_c".to_string(), "planning".to_string());
assert_eq!(running_agent_count(&app), 2);
}
#[test]
fn reconcile_subagent_activity_state_trims_stale_progress_and_sets_anchor() {
let mut app = create_test_app();
app.subagent_cache = vec![
make_subagent("agent_a", crate::tools::subagent::SubAgentStatus::Running),
make_subagent("agent_b", crate::tools::subagent::SubAgentStatus::Completed),
];
app.agent_progress
.insert("agent_pending".to_string(), "old".to_string());
reconcile_subagent_activity_state(&mut app);
assert!(app.agent_progress.contains_key("agent_a"));
assert!(app.agent_progress.contains_key("agent_pending"));
assert!(app.agent_activity_started_at.is_some());
app.subagent_cache = vec![
make_subagent("agent_a", crate::tools::subagent::SubAgentStatus::Completed),
make_subagent(
"agent_pending",
crate::tools::subagent::SubAgentStatus::Completed,
),
];
reconcile_subagent_activity_state(&mut app);
assert!(app.agent_progress.is_empty());
assert!(app.agent_activity_started_at.is_none());
}
#[test]
fn reconcile_subagent_activity_state_expires_terminal_cards_but_keeps_running() {
let mut app = create_test_app();
let old_seen_at = Instant::now();
let now = old_seen_at + Duration::from_secs(10 * 60);
let recent_seen_at = now - Duration::from_secs(30);
app.subagent_cache = vec![
make_subagent(
"agent_running",
crate::tools::subagent::SubAgentStatus::Running,
),
make_subagent(
"agent_old",
crate::tools::subagent::SubAgentStatus::Completed,
),
make_subagent(
"agent_recent",
crate::tools::subagent::SubAgentStatus::Failed("boom".to_string()),
),
];
app.subagent_terminal_seen_at
.insert("agent_old".to_string(), old_seen_at);
app.subagent_terminal_seen_at
.insert("agent_recent".to_string(), recent_seen_at);
reconcile_subagent_activity_state_at(&mut app, now);
let ids: HashSet<&str> = app
.subagent_cache
.iter()
.map(|agent| agent.agent_id.as_str())
.collect();
assert!(ids.contains("agent_running"));
assert!(ids.contains("agent_recent"));
assert!(!ids.contains("agent_old"));
assert!(!app.subagent_terminal_seen_at.contains_key("agent_old"));
assert!(app.subagent_terminal_seen_at.contains_key("agent_recent"));
}
#[test]
fn reconcile_subagent_activity_state_caps_terminal_card_bursts() {
let mut app = create_test_app();
let oldest_seen_at = Instant::now();
let now = oldest_seen_at + Duration::from_secs(30);
for idx in 0..30 {
let id = format!("agent_{idx:02}");
app.subagent_cache.push(make_subagent(
&id,
crate::tools::subagent::SubAgentStatus::Completed,
));
app.subagent_terminal_seen_at
.insert(id, now - Duration::from_secs(idx));
}
reconcile_subagent_activity_state_at(&mut app, now);
let terminal_count = app
.subagent_cache
.iter()
.filter(|agent| {
!matches!(
agent.status,
crate::tools::subagent::SubAgentStatus::Running
)
})
.count();
assert_eq!(terminal_count, 24);
assert!(
app.subagent_cache
.iter()
.any(|agent| agent.agent_id == "agent_00")
);
assert!(
!app.subagent_cache
.iter()
.any(|agent| agent.agent_id == "agent_29")
);
}
#[test]
fn subagent_token_usage_updates_live_cost_counter_without_card_change() {
let mut app = create_test_app();
handle_subagent_mailbox(
&mut app,
1,
&crate::tools::subagent::MailboxMessage::TokenUsage {
agent_id: "agent-a".to_string(),
provider: ApiProvider::Deepseek,
model: "deepseek-v4-flash".to_string(),
usage: crate::models::Usage {
input_tokens: 10_000,
output_tokens: 1_000,
..Default::default()
},
},
);
assert!(app.session.subagent_cost > 0.0);
assert!(
app.history.is_empty(),
"usage-only mailbox messages should not allocate a sub-agent card"
);
}
#[test]
fn subagent_token_usage_prices_the_child_route_not_the_parent_route() {
let mut app = create_test_app();
assert_eq!(app.api_provider, ApiProvider::Deepseek);
handle_subagent_mailbox(
&mut app,
2,
&crate::tools::subagent::MailboxMessage::TokenUsage {
agent_id: "agent-codex".to_string(),
provider: ApiProvider::OpenaiCodex,
model: "gpt-5.5".to_string(),
usage: crate::models::Usage {
input_tokens: 10_000,
output_tokens: 1_000,
..Default::default()
},
},
);
assert_eq!(
app.session.subagent_cost, 0.0,
"ChatGPT/Codex child usage must not inherit the DeepSeek parent's pricing"
);
}
#[test]
fn subagent_token_usage_is_deduped_by_mailbox_sequence() {
let mut app = create_test_app();
let usage = crate::tools::subagent::MailboxMessage::TokenUsage {
agent_id: "agent-a".to_string(),
provider: ApiProvider::Deepseek,
model: "deepseek-v4-flash".to_string(),
usage: crate::models::Usage {
input_tokens: 10_000,
output_tokens: 1_000,
..Default::default()
},
};
handle_subagent_mailbox(&mut app, 7, &usage);
let first = app.session.subagent_cost;
handle_subagent_mailbox(&mut app, 7, &usage);
assert_eq!(app.session.subagent_cost, first);
handle_subagent_mailbox(&mut app, 8, &usage);
assert!(app.session.subagent_cost > first);
}
#[test]
fn fanout_started_sibling_bumps_existing_card_revision() {
let mut app = create_test_app();
app.pending_subagent_dispatch = Some("rlm".to_string());
handle_subagent_mailbox(
&mut app,
1,
&crate::tools::subagent::MailboxMessage::Started {
agent_id: "fanout-a".to_string(),
agent_type: "default".to_string(),
},
);
let fanout_idx = app.last_fanout_card_index.expect("fanout card index");
let initial_revision = app.history_revisions[fanout_idx];
handle_subagent_mailbox(
&mut app,
2,
&crate::tools::subagent::MailboxMessage::Started {
agent_id: "fanout-b".to_string(),
agent_type: "default".to_string(),
},
);
assert_eq!(app.history.len(), 1, "sibling should reuse fanout card");
assert_ne!(
app.history_revisions[fanout_idx], initial_revision,
"reused fanout card must invalidate its cached transcript rows"
);
match &app.history[fanout_idx] {
HistoryCell::SubAgent(SubAgentCell::Fanout(card)) => {
assert_eq!(card.worker_count(), 2);
}
cell => panic!("expected fanout card, got {cell:?}"),
}
}
#[test]
fn fanout_interrupted_mailbox_drops_running_count() {
let mut app = create_test_app();
app.pending_subagent_dispatch = Some("rlm".to_string());
for (seq, id) in ["fanout-a", "fanout-b"].iter().enumerate() {
handle_subagent_mailbox(
&mut app,
seq as u64 + 1,
&crate::tools::subagent::MailboxMessage::Started {
agent_id: (*id).to_string(),
agent_type: "default".to_string(),
},
);
}
assert_eq!(
crate::tui::subagent_routing::active_fanout_counts(&app),
Some((2, 2))
);
handle_subagent_mailbox(
&mut app,
3,
&crate::tools::subagent::MailboxMessage::Interrupted {
agent_id: "fanout-a".to_string(),
reason: "API call timed out after 120000ms".to_string(),
},
);
assert_eq!(
crate::tui::subagent_routing::active_fanout_counts(&app),
Some((1, 2)),
"interrupted worker must no longer count as running"
);
}
#[test]
fn stall_reason_provider_wait_includes_route_and_idle_budget() {
let mut app = create_test_app();
app.is_loading = true;
app.stream_chunk_timeout_secs = 300;
app.turn_started_at = Some(Instant::now() - Duration::from_secs(70));
app.turn_last_activity_at = Some(Instant::now() - Duration::from_secs(65));
let reason = crate::tui::footer_ui::stall_reason(&app).expect("stalled turn has a reason");
assert!(reason.contains("waiting for model"), "{reason}");
assert!(reason.contains("65s"), "{reason}");
assert!(!reason.contains("/300s"), "{reason}");
}
#[test]
fn reconcile_syncs_stale_running_cards_with_interrupted_snapshots() {
let mut app = create_test_app();
app.pending_subagent_dispatch = Some("rlm".to_string());
for (seq, id) in ["fanout-a", "fanout-b"].iter().enumerate() {
handle_subagent_mailbox(
&mut app,
seq as u64 + 1,
&crate::tools::subagent::MailboxMessage::Started {
agent_id: (*id).to_string(),
agent_type: "default".to_string(),
},
);
}
let fanout_idx = app.last_fanout_card_index.expect("fanout card index");
let initial_revision = app.history_revisions[fanout_idx];
app.subagent_cache = vec![
make_subagent(
"fanout-a",
crate::tools::subagent::SubAgentStatus::Interrupted("API call timed out".to_string()),
),
make_subagent("fanout-b", crate::tools::subagent::SubAgentStatus::Running),
];
reconcile_subagent_activity_state(&mut app);
assert_eq!(
crate::tui::subagent_routing::active_fanout_counts(&app),
Some((1, 2)),
"snapshot reconciliation must clear the stale running slot"
);
assert_ne!(
app.history_revisions[fanout_idx], initial_revision,
"reconciled card must invalidate cached transcript rows"
);
assert_eq!(running_agent_count(&app), 1);
}
#[test]
fn stall_reason_provider_wait_reports_zero_running_for_planned_fanout() {
let mut app = create_test_app();
app.is_loading = true;
app.turn_started_at = Some(Instant::now() - Duration::from_secs(45));
let card = crate::tui::widgets::agent_card::FanoutCard::new("rlm")
.with_workers(["task:a", "task:b", "task:c", "task:d"]);
app.history
.push(HistoryCell::SubAgent(SubAgentCell::Fanout(card)));
app.history_revisions.push(0);
app.last_fanout_card_index = Some(app.history.len() - 1);
let reason = crate::tui::footer_ui::stall_reason(&app).expect("stalled turn has a reason");
assert!(reason.contains("fanout 0/4"), "{reason}");
if let Some(HistoryCell::SubAgent(SubAgentCell::Fanout(card))) = app
.last_fanout_card_index
.and_then(|idx| app.history.get_mut(idx))
{
card.upsert_worker(
"task:a",
crate::tui::widgets::agent_card::AgentLifecycle::Running,
);
}
let reason = crate::tui::footer_ui::stall_reason(&app).expect("stalled turn has a reason");
assert!(!reason.contains("fanout 0/4"), "{reason}");
}
#[test]
fn stall_reason_provider_wait_flags_pending_dispatch() {
let mut app = create_test_app();
app.is_loading = true;
app.turn_started_at = Some(Instant::now() - Duration::from_secs(31));
app.pending_subagent_dispatch = Some("agent".to_string());
let reason = crate::tui::footer_ui::stall_reason(&app).expect("stalled turn has a reason");
assert!(reason.contains("dispatch pending"), "{reason}");
}
#[test]
fn provider_wait_incident_logs_once_per_turn() {
let mut app = create_test_app();
app.is_loading = true;
app.turn_started_at = Some(Instant::now() - Duration::from_secs(150));
app.pending_subagent_dispatch = Some("rlm".to_string());
assert!(!app.provider_wait_incident_logged);
crate::tui::footer_ui::maybe_log_provider_wait_incident(&mut app);
assert!(app.provider_wait_incident_logged, "incident logged once");
let mut quiet = create_test_app();
quiet.is_loading = true;
quiet.turn_started_at = Some(Instant::now() - Duration::from_secs(150));
crate::tui::footer_ui::maybe_log_provider_wait_incident(&mut quiet);
assert!(!quiet.provider_wait_incident_logged);
let mut early = create_test_app();
early.is_loading = true;
early.turn_started_at = Some(Instant::now() - Duration::from_secs(60));
early.pending_subagent_dispatch = Some("rlm".to_string());
crate::tui::footer_ui::maybe_log_provider_wait_incident(&mut early);
assert!(!early.provider_wait_incident_logged);
}
#[test]
fn format_token_count_compact_formats_units() {
assert_eq!(format_token_count_compact(999), "999");
assert_eq!(format_token_count_compact(1_200), "1.2k");
assert_eq!(format_token_count_compact(1_000_000), "1.0M");
}
#[test]
fn footer_session_tokens_chip_uses_single_compact_total() {
let mut app = create_test_app();
app.session.total_input_tokens = 900_000;
app.session.total_cache_hit_tokens = 700_000;
app.session.total_cache_miss_tokens = 200_000;
app.session.total_output_tokens = 600_000;
let text = spans_text(&footer_session_tokens_spans(&app));
assert_eq!(text, "tok 1.5M");
assert!(!text.contains(" cch "));
assert!(!text.contains(" out"));
}
#[test]
fn footer_session_tokens_chip_shows_token_count_without_throughput() {
let mut app = create_test_app();
app.is_loading = true;
app.turn_started_at = Some(Instant::now() - Duration::from_secs(10));
app.streaming_output_token_estimate = 125;
let text = spans_text(&footer_session_tokens_spans(&app));
assert_eq!(text, "");
}
#[test]
fn footer_session_tokens_chip_shows_compact_total() {
let mut app = create_test_app();
app.session.total_input_tokens = 1_000;
app.session.total_output_tokens = 240;
app.session.last_output_throughput =
crate::resource_telemetry::TokenThroughput::new(240, Duration::from_secs(12));
let text = spans_text(&footer_session_tokens_spans(&app));
assert_eq!(text, "tok 1.2k");
assert!(!text.contains("out"));
assert!(!text.contains("/s"));
}
#[test]
fn format_context_budget_caps_overflow_display() {
assert_eq!(format_context_budget(5_000, 128_000), "5.0k/128.0k");
assert_eq!(format_context_budget(250_000, 128_000), ">128.0k/128.0k");
}
#[test]
fn footer_state_label_shows_idle_ready_and_prefers_compacting() {
let mut app = create_test_app();
assert_eq!(footer_state_label(&app).0, "idle");
app.is_loading = true;
assert_eq!(footer_state_label(&app).0, "ready");
assert_ne!(footer_state_label(&app).0, "thinking");
app.is_compacting = true;
assert!(footer_state_label(&app).0.starts_with("compacting"));
}
#[test]
fn event_poll_timeout_has_nonzero_floor() {
assert_eq!(
clamp_event_poll_timeout(Duration::ZERO),
Duration::from_millis(1)
);
assert_eq!(
clamp_event_poll_timeout(Duration::from_micros(250)),
Duration::from_millis(1)
);
assert_eq!(
clamp_event_poll_timeout(Duration::from_millis(24)),
Duration::from_millis(24)
);
}
#[tokio::test]
async fn bang_shell_input_dispatches_shell_op_instead_of_model_message() {
let mut app = create_test_app();
app.mode = AppMode::Agent;
app.trust_mode = false;
app.approval_mode = ApprovalMode::Suggest;
let mut engine = mock_engine_handle();
let handled = handle_bang_shell_input(&mut app, &engine.handle, "! pwd")
.await
.expect("bang shell handler");
assert!(handled);
assert_eq!(
app.status_message.as_deref(),
Some("Shell command submitted: pwd")
);
let op = engine.rx_op.recv().await.expect("engine op");
match op {
Op::RunShellCommand {
command,
mode,
allow_shell,
trust_mode,
auto_approve,
approval_mode,
} => {
assert_eq!(command, "pwd");
assert_eq!(mode, AppMode::Agent);
assert!(!allow_shell);
assert!(!trust_mode);
assert!(!auto_approve);
assert_eq!(approval_mode, ApprovalMode::Suggest);
}
other => panic!("expected RunShellCommand, got {other:?}"),
}
}
#[tokio::test]
async fn bang_shell_input_keeps_auto_review_separate_from_bypass() {
let mut app = create_test_app();
app.mode = AppMode::Agent;
app.approval_mode = ApprovalMode::Auto;
app.trust_mode = true;
let mut engine = mock_engine_handle();
let handled = handle_bang_shell_input(&mut app, &engine.handle, "! pwd")
.await
.expect("bang shell handler");
assert!(handled);
let op = engine.rx_op.recv().await.expect("engine op");
match op {
Op::RunShellCommand {
command,
mode,
allow_shell,
trust_mode,
auto_approve,
approval_mode,
} => {
assert_eq!(command, "pwd");
assert_eq!(mode, AppMode::Agent);
assert!(!allow_shell);
assert!(trust_mode);
assert!(!auto_approve);
assert_eq!(approval_mode, ApprovalMode::Auto);
}
other => panic!("expected RunShellCommand, got {other:?}"),
}
}
#[tokio::test]
async fn bang_shell_input_dispatches_even_while_turn_is_loading() {
let mut app = create_test_app();
app.mode = AppMode::Agent;
app.is_loading = true;
let mut engine = mock_engine_handle();
let handled = handle_bang_shell_input(&mut app, &engine.handle, "! echo steer-safe")
.await
.expect("bang shell handler");
assert!(handled);
let op = engine.rx_op.recv().await.expect("engine op");
match op {
Op::RunShellCommand { command, mode, .. } => {
assert_eq!(command, "echo steer-safe");
assert_eq!(mode, AppMode::Agent);
}
other => panic!("expected RunShellCommand, got {other:?}"),
}
}
#[tokio::test]
async fn empty_bang_shell_input_is_consumed_with_usage_error() {
let mut app = create_test_app();
let engine = mock_engine_handle();
let handled = handle_bang_shell_input(&mut app, &engine.handle, "! ")
.await
.expect("bang shell handler");
assert!(handled);
assert_eq!(
app.status_message.as_deref(),
Some("Error: Usage: ! <shell command>")
);
}
#[test]
fn local_bang_shell_tool_ids_are_not_model_visible() {
assert!(!is_model_visible_tool_call("user_shell_1"));
assert!(is_model_visible_tool_call("toolu_01abc"));
}
fn complete_release_json(tag: &str) -> serde_json::Value {
let assets = REQUIRED_RELEASE_ASSETS
.iter()
.map(|name| serde_json::json!({ "name": name, "state": "uploaded" }))
.collect::<Vec<_>>();
serde_json::json!({
"tag_name": tag,
"draft": false,
"prerelease": false,
"assets": assets,
})
}
#[test]
fn version_hint_requires_complete_release_assets() {
let complete = complete_release_json("v0.8.47");
let hint = version_hint_from_release_json(&complete, "0.8.46").expect("newer complete release");
assert!(hint.contains("v0.8.47 available"));
let mut missing_manifest = complete_release_json("v0.8.47");
missing_manifest["assets"] = serde_json::Value::Array(
missing_manifest["assets"]
.as_array()
.expect("assets")
.iter()
.filter(|asset| {
asset.get("name").and_then(serde_json::Value::as_str)
!= Some("codewhale-artifacts-sha256.txt")
})
.cloned()
.collect(),
);
assert!(
version_hint_from_release_json(&missing_manifest, "0.8.46").is_none(),
"do not advertise a release before checksums are uploaded"
);
let mut pending_asset = complete_release_json("v0.8.47");
pending_asset["assets"].as_array_mut().expect("assets")[0]["state"] = serde_json::json!("open");
assert!(
version_hint_from_release_json(&pending_asset, "0.8.46").is_none(),
"do not advertise a release before every asset is uploaded"
);
let mut missing_state = complete_release_json("v0.8.47");
missing_state["assets"].as_array_mut().expect("assets")[0]
.as_object_mut()
.expect("asset object")
.remove("state");
assert!(
version_hint_from_release_json(&missing_state, "0.8.46").is_none(),
"do not accept malformed asset state as uploaded"
);
}
#[test]
fn version_hint_ignores_draft_prerelease_and_current_versions() {
let mut draft = complete_release_json("v0.8.47");
draft["draft"] = serde_json::Value::Bool(true);
assert!(version_hint_from_release_json(&draft, "0.8.46").is_none());
let mut prerelease = complete_release_json("v0.8.47");
prerelease["prerelease"] = serde_json::Value::Bool(true);
assert!(version_hint_from_release_json(&prerelease, "0.8.46").is_none());
let current = complete_release_json("v0.8.46");
assert!(version_hint_from_release_json(¤t, "0.8.46").is_none());
}
#[test]
fn startup_version_check_source_respects_update_config() {
assert_eq!(
startup_version_check_source(&UpdateConfig {
check_for_updates: false,
update_uri: Some("https://mirror.example/releases/latest".to_string()),
}),
StartupVersionCheckSource::Disabled
);
assert_eq!(
startup_version_check_source(&UpdateConfig {
check_for_updates: true,
update_uri: Some(" https://mirror.example/releases/latest ".to_string()),
}),
StartupVersionCheckSource::ConfiguredUrl(
"https://mirror.example/releases/latest".to_string()
)
);
assert_eq!(
startup_version_check_source(&UpdateConfig::default()),
StartupVersionCheckSource::ReleaseResolver
);
}
#[test]
fn custom_update_uri_accepts_tag_only_release_json() {
let json = serde_json::json!({
"tag_name": "v0.8.47",
"draft": false,
"prerelease": false,
});
let hint = version_hint_from_custom_release_json(&json, "0.8.46")
.expect("tag-only custom metadata should be enough for mirrors");
assert!(hint.contains("v0.8.47 available"));
}
#[test]
#[cfg(any(unix, windows))]
fn external_url_launcher_does_not_wait_for_browser_process() {
let command = slow_external_url_command();
let start = Instant::now();
spawn_external_url_command(command).expect("spawn external URL command");
assert!(
start.elapsed() < Duration::from_millis(750),
"opening a feedback URL must not wait for the browser command to exit"
);
}
#[cfg(unix)]
fn slow_external_url_command() -> Command {
let mut command = Command::new("sh");
command.args(["-c", "sleep 1"]);
command
}
#[cfg(windows)]
fn slow_external_url_command() -> Command {
let mut command = Command::new("cmd");
command.args(["/C", "ping -n 2 127.0.0.1 >NUL"]);
command
}
#[test]
fn footer_status_line_spans_show_model_idle_and_active() {
let mut app = create_test_app();
app.model = "deepseek-v4-flash".to_string();
let idle = spans_text(&footer_status_line_spans(&app, 60));
assert!(idle.contains("deepseek-v4-flash"));
assert!(idle.contains("\u{00B7}"));
assert!(idle.contains("idle"));
assert!(!idle.contains("act"));
assert!(!idle.contains("agent"));
app.is_loading = true;
let active = spans_text(&footer_status_line_spans(&app, 60));
assert!(active.contains("deepseek-v4-flash"));
assert!(
!active.contains("busy"),
"footer must not repeat coarse busy when header streams: {active}"
);
assert!(
!active.contains("thinking"),
"footer must not show a `thinking` text label while loading"
);
}
#[test]
fn footer_status_line_spans_truncate_long_model_names() {
let mut app = create_test_app();
app.model = "deepseek-v4-pro-with-an-extremely-long-model-name".to_string();
app.is_loading = true;
let line = spans_text(&footer_status_line_spans(&app, 40));
assert!(line.contains("..."));
assert!(UnicodeWidthStr::width(line.as_str()) <= 40);
}
#[test]
fn footer_auxiliary_spans_show_cache_when_compact() {
let mut app = create_test_app();
app.is_loading = true;
app.session.last_prompt_tokens = Some(48_000);
app.session.last_prompt_cache_hit_tokens = Some(36_000);
app.session.last_prompt_cache_miss_tokens = Some(12_000);
app.session.session_cost = 12.34;
let compact = spans_text(&footer_auxiliary_spans(&app, 48));
assert!(compact.contains("Cache: 75.0% hit"));
assert!(!compact.contains('$'));
}
#[test]
fn footer_auxiliary_spans_show_cache_unavailable_when_provider_omits_cache_fields() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(48_000);
app.session.last_completion_tokens = Some(2_000);
let roomy = spans_text(&footer_auxiliary_spans(&app, 72));
assert!(roomy.contains("Cache: unavailable"));
}
#[test]
fn footer_auxiliary_spans_show_cache_and_cost_when_roomy() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(48_000);
app.session.last_prompt_cache_hit_tokens = Some(36_000);
app.session.last_prompt_cache_miss_tokens = Some(12_000);
app.session.session_cost = 12.34;
let roomy = spans_text(&footer_auxiliary_spans(&app, 72));
assert!(roomy.contains("Cache: 75.0% hit | hit 36000 | miss 12000"));
assert!(roomy.contains("$12.34"));
assert!(
!roomy.contains("ctx"),
"context % removed from footer — shown in header only"
);
}
#[test]
fn footer_cache_low_hit_with_stable_prefix_is_not_error_colored() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(10_000);
app.session.last_prompt_cache_hit_tokens = Some(500);
app.session.last_prompt_cache_miss_tokens = Some(9_500);
app.prefix_stability_pct = Some(100);
app.prefix_change_count = 0;
let spans = footer_cache_spans(&app);
assert_eq!(spans_text(&spans), "Cache: 5.0% hit | hit 500 | miss 9500");
assert_eq!(spans[0].style.fg, Some(palette::TEXT_MUTED));
}
#[test]
fn footer_cache_low_hit_with_prefix_churn_stays_error_colored() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(10_000);
app.session.last_prompt_cache_hit_tokens = Some(500);
app.session.last_prompt_cache_miss_tokens = Some(9_500);
app.prefix_stability_pct = Some(80);
app.prefix_change_count = 2;
let spans = footer_cache_spans(&app);
assert_eq!(spans[0].style.fg, Some(palette::STATUS_ERROR));
}
#[test]
fn footer_auxiliary_spans_show_tiny_positive_cost_when_roomy() {
let mut app = create_test_app();
app.session.session_cost = 0.00005;
let roomy = spans_text(&footer_auxiliary_spans(&app, 32));
assert!(roomy.contains("<$0.0001"));
}
#[test]
fn footer_auxiliary_spans_use_configured_cost_currency() {
let mut app = create_test_app();
app.cost_currency = crate::pricing::CostCurrency::Cny;
app.session.session_cost_cny = 2.5;
let roomy = spans_text(&footer_auxiliary_spans(&app, 32));
assert!(roomy.contains("¥2.50"));
assert!(!roomy.contains('$'));
}
#[test]
fn footer_auxiliary_spans_show_reasoning_replay_chip() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(48_000);
app.session.last_reasoning_replay_tokens = Some(8_200);
let spans = footer_auxiliary_spans(&app, 64);
let text = spans_text(&spans);
assert!(
text.contains("rsn 8.2k"),
"expected replay chip, got {text:?}"
);
}
#[test]
fn footer_auxiliary_spans_hide_reasoning_replay_when_zero() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(48_000);
app.session.last_reasoning_replay_tokens = Some(0);
let spans = footer_auxiliary_spans(&app, 64);
let text = spans_text(&spans);
assert!(!text.contains("rsn"), "zero replay must not render chip");
}
#[test]
fn context_usage_snapshot_prefers_estimate_when_reported_exceeds_window() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(1_200_000);
app.api_messages = vec![Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: "hello".to_string(),
cache_control: None,
}],
}];
let (used, max, percent) =
context_usage_snapshot(&app).expect("context usage should be available");
assert_eq!(max, 1_000_000);
assert!(used > 0);
assert!(used <= i64::from(max));
assert!(percent < 100.0);
}
#[test]
fn context_usage_snapshot_prefers_estimate_when_reported_is_inflated_by_old_reasoning() {
let mut app = create_test_app();
app.session.last_prompt_tokens = Some(980_000);
app.api_messages = vec![Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: "small current context".to_string(),
cache_control: None,
}],
}];
let (used, max, percent) =
context_usage_snapshot(&app).expect("context usage should be available");
assert_eq!(max, 1_000_000);
assert!(used < 10_000);
assert!(percent < 2.0);
}
#[test]
fn context_usage_does_not_drop_when_reported_shrinks_after_multi_round_turn() {
let mut app = create_test_app();
app.api_messages = vec![Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: "context ".repeat(2_000), cache_control: None,
}],
}];
app.session.last_prompt_tokens = Some(410_000);
let (_, _, percent_after_multi_round) = context_usage_snapshot(&app).expect("usage available");
app.session.last_prompt_tokens = Some(15_000);
let (_, _, percent_after_single_round) = context_usage_snapshot(&app).expect("usage available");
let drift = (percent_after_multi_round - percent_after_single_round).abs();
assert!(
drift < 1.0,
"displayed % should not jump because reported tokens varied across rounds; \
after-multi-round={percent_after_multi_round:.2} after-single-round={percent_after_single_round:.2}"
);
}
#[test]
fn context_usage_snapshot_prefers_live_estimate_while_loading() {
let mut app = create_test_app();
app.is_loading = true;
app.session.last_prompt_tokens = Some(128);
app.api_messages = vec![Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: "context ".repeat(6_000),
cache_control: None,
}],
}];
let estimated = estimated_context_tokens(&app).expect("estimated context should be available");
let (used, max, percent) =
context_usage_snapshot(&app).expect("context usage should be available");
assert_eq!(used, estimated);
assert_eq!(max, 1_000_000);
assert!(used > i64::from(app.session.last_prompt_tokens.expect("reported tokens")));
assert!(percent > 0.0);
}
#[test]
fn should_auto_compact_before_send_uses_shared_token_threshold() {
let mut app = create_test_app();
app.api_messages = vec![Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: "context ".repeat(240_000),
cache_control: None,
}],
}];
let (used, _, _) = context_usage_snapshot(&app).expect("context snapshot");
let used = usize::try_from(used).expect("non-negative context estimate");
app.auto_compact = true;
app.compact_threshold = used;
assert!(should_auto_compact_before_send(&app));
app.compact_threshold = used.saturating_add(1);
assert!(!should_auto_compact_before_send(&app));
app.auto_compact = false;
app.compact_threshold = 0;
assert!(!should_auto_compact_before_send(&app));
}
#[test]
fn context_pressure_warning_reflects_auto_compact_threshold_state() {
let mut app = create_test_app();
app.api_messages = vec![Message {
role: "user".to_string(),
content: vec![ContentBlock::Text {
text: "context ".repeat(240_000),
cache_control: None,
}],
}];
app.auto_compact = true;
app.auto_compact_threshold_percent = 100.0;
let (used, _, percent) = context_usage_snapshot(&app).expect("context snapshot");
assert!(
percent < app.auto_compact_threshold_percent,
"fixture must remain below the raw window-relative setting"
);
app.compact_threshold = usize::try_from(used).expect("non-negative context estimate");
maybe_warn_context_pressure(&mut app);
let status = app.status_message.expect("context warning");
assert!(
status.contains("Auto-compaction will run before the next send."),
"unexpected status: {status}"
);
}
#[test]
fn test_esc_cancels_streaming_sets_is_loading_false() {
let mut app = create_test_app();
app.is_loading = true;
app.mode = AppMode::Agent;
app.is_loading = false;
app.status_message = Some("Request cancelled".to_string());
assert!(!app.is_loading);
assert_eq!(app.status_message, Some("Request cancelled".to_string()));
}
#[test]
fn test_esc_with_input_clears_input_when_not_loading() {
let mut app = create_test_app();
app.is_loading = false;
app.input = "some draft input".to_string();
app.cursor_position = app.input.chars().count();
app.clear_input();
assert!(app.input.is_empty());
assert_eq!(app.cursor_position, 0);
assert!(!app.is_loading);
}
#[test]
fn test_esc_discards_queued_draft_before_clearing_input() {
let mut app = create_test_app();
app.is_loading = false;
app.input.clear();
app.queued_draft = Some(crate::tui::app::QueuedMessage::new(
"queued draft".to_string(),
None,
));
assert_eq!(
next_escape_action(&app, false),
EscapeAction::DiscardQueuedDraft
);
}
#[test]
fn test_esc_prioritizes_queued_draft_edit_over_loading_cancel() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "editing queued follow-up".to_string();
app.queued_draft = Some(crate::tui::app::QueuedMessage::new(
"original queued follow-up".to_string(),
None,
));
assert_eq!(
next_escape_action(&app, false),
EscapeAction::DiscardQueuedDraft
);
}
#[test]
fn test_esc_is_noop_when_idle() {
let mut app = create_test_app();
app.is_loading = false;
app.input.clear();
app.cursor_position = 0;
app.mode = AppMode::Agent;
assert_eq!(next_escape_action(&app, false), EscapeAction::Noop);
assert_eq!(app.mode, AppMode::Agent);
}
#[test]
fn test_esc_closes_slash_menu_before_other_actions() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "draft".to_string();
app.queued_draft = Some(crate::tui::app::QueuedMessage::new(
"queued draft".to_string(),
None,
));
assert_eq!(next_escape_action(&app, true), EscapeAction::CloseSlashMenu);
}
#[test]
fn history_arrow_does_not_steal_open_menus() {
let mut app = create_test_app();
app.input_history.push("previous prompt".to_string());
app.input = "/".to_string();
app.cursor_position = 1;
assert!(!handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
true,
false,
));
assert_eq!(app.input, "/");
assert!(app.history_index.is_none());
}
#[test]
fn test_ctrl_c_cancels_streaming_sets_status() {
let mut app = create_test_app();
app.is_loading = true;
app.is_loading = false;
app.status_message = Some("Request cancelled".to_string());
assert!(!app.is_loading);
assert_eq!(app.status_message, Some("Request cancelled".to_string()));
}
#[test]
fn local_cancel_marks_late_stream_events_for_suppression() {
let _retry_guard = crate::retry_status::test_guard();
let mut app = create_test_app();
app.is_loading = true;
app.turn_started_at = Some(Instant::now());
app.runtime_turn_id = Some("turn_cancel_me".to_string());
app.runtime_turn_status = Some("in_progress".to_string());
app.streaming_state.start_text(0, None);
crate::retry_status::start(2, Duration::from_secs(3), "network error");
mark_active_turn_cancelled_locally(&mut app);
assert!(!app.is_loading);
assert!(app.turn_started_at.is_none());
assert!(app.runtime_turn_id.is_none());
assert!(app.runtime_turn_status.is_none());
assert!(matches!(
crate::retry_status::snapshot(),
crate::retry_status::RetryState::Idle
));
assert!(app.suppress_stream_events_until_turn_complete);
assert!(suppress_engine_event_after_local_cancel(
&EngineEvent::MessageDelta {
index: 0,
content: "late text".to_string(),
}
));
assert!(suppress_engine_event_after_local_cancel(
&EngineEvent::ThinkingDelta {
index: 0,
content: "late thinking".to_string(),
}
));
assert!(suppress_engine_event_after_local_cancel(
&EngineEvent::SessionUpdated {
session_id: "session".to_string(),
messages: Vec::new(),
system_prompt: None,
model: "deepseek-v4-flash".to_string(),
workspace: PathBuf::from("."),
}
));
assert!(ignore_stale_stream_event_while_idle(
&EngineEvent::MessageDelta {
index: 0,
content: "late text".to_string(),
}
));
assert!(!suppress_engine_event_after_local_cancel(
&EngineEvent::TurnComplete {
usage: Usage::default(),
status: crate::core::events::TurnOutcomeStatus::Interrupted,
error: None,
tool_catalog: None,
base_url: None,
}
));
assert!(!suppress_engine_event_after_local_cancel(
&EngineEvent::Status {
message: "Request cancelled".to_string(),
}
));
}
#[test]
fn turn_started_route_is_captured_before_cancel_suppression() {
let mut app = create_test_app();
app.suppress_stream_events_until_turn_complete = true;
app.ocean_completion_started_at = Some(Instant::now());
app.pending_turn_route = Some((ApiProvider::Deepseek, "pending-model".to_string(), false));
let created_at = chrono::Utc::now();
let event = EngineEvent::TurnStarted {
turn_id: "turn_cancel_race".to_string(),
created_at,
route: Some(crate::core::events::TurnRoute {
provider: ApiProvider::Openai,
provider_identity: "openai".to_string(),
model: "gpt-5.5".to_string(),
auto_model: true,
}),
};
capture_turn_started_metadata(&mut app, &event);
let active_turn = app.active_turn.as_ref().expect("captured turn");
assert_eq!(active_turn.turn_id, "turn_cancel_race");
assert_eq!(active_turn.created_at, created_at);
let route = active_turn.route.as_ref().expect("captured route");
assert_eq!(route.provider, ApiProvider::Openai);
assert_eq!(route.model, "gpt-5.5");
assert!(route.auto_model);
assert!(app.pending_turn_route.is_none());
assert!(app.ocean_completion_started_at.is_none());
let observer = turn_end_observer_metadata(Some(active_turn));
assert_eq!(observer.turn_id.as_ref(), "turn_cancel_race");
assert_eq!(observer.created_at, created_at);
assert_eq!(observer.route, Some(route));
}
#[test]
fn engine_error_health_accounting_uses_active_turn_route() {
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.model = "current-model".to_string();
let event = EngineEvent::TurnStarted {
turn_id: "routed-turn".to_string(),
created_at: chrono::Utc::now(),
route: Some(crate::core::events::TurnRoute {
provider: ApiProvider::Openai,
provider_identity: "openai".to_string(),
model: "gpt-5.5".to_string(),
auto_model: true,
}),
};
capture_turn_started_metadata(&mut app, &event);
let route = error_health_route(&app, app.api_provider);
assert_eq!(route, (ApiProvider::Openai, "gpt-5.5".to_string()));
}
#[test]
fn completion_only_hook_metadata_is_synthetic_and_non_model() {
let observed_after = chrono::Utc::now();
let first = turn_end_observer_metadata(None);
let second = turn_end_observer_metadata(None);
assert!(first.turn_id.starts_with("lifecycle_"));
assert!(second.turn_id.starts_with("lifecycle_"));
assert_ne!(first.turn_id, second.turn_id);
assert!(first.created_at >= observed_after);
assert!(first.route.is_none());
assert!(second.route.is_none());
}
#[test]
fn issue_2739_stalled_turn_snapshot_preserves_api_messages() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "hello from user"));
app.api_messages
.push(text_message("assistant", "partial reply"));
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
app.turn_started_at = Some(Instant::now());
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.messages.len(), 2);
assert_eq!(snapshot.messages[0].role, "user");
assert_eq!(snapshot.messages[1].role, "assistant");
}
#[test]
fn issue_2739_esc_cancel_preserves_session_messages_before_clear() {
let _home = SettingsHomeGuard::new();
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "esc cancel test"));
app.api_messages
.push(text_message("assistant", "interrupted by esc"));
app.is_loading = true;
app.turn_started_at = Some(Instant::now());
app.runtime_turn_id = Some("turn_esc_me".to_string());
app.runtime_turn_status = Some("in_progress".to_string());
app.streaming_state.start_text(0, None);
mark_active_turn_cancelled_locally(&mut app);
assert!(
app.current_session_id.is_some(),
"local cancel should create a resumable session snapshot"
);
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.messages.len(), 2);
assert_eq!(snapshot.messages[0].role, "user");
assert_eq!(snapshot.messages[1].role, "assistant");
assert!(!app.is_loading);
assert!(app.turn_started_at.is_none());
assert!(app.runtime_turn_status.is_none());
}
#[test]
fn issue_2739_dispatch_timeout_preserves_user_prompt() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "prompt that never dispatched"));
app.is_loading = true;
app.runtime_turn_status = None;
app.dispatch_started_at =
Some(Instant::now() - DISPATCH_WATCHDOG_TIMEOUT - Duration::from_millis(1));
app.turn_started_at = Some(Instant::now());
let recovered = reconcile_turn_liveness(&mut app, Instant::now(), false);
assert!(recovered, "dispatch-timeout branch should fire");
assert!(!app.is_loading);
assert!(app.dispatch_started_at.is_none());
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.messages.len(), 1);
assert_eq!(snapshot.messages[0].role, "user");
}
#[test]
fn test_ctrl_c_exits_when_not_loading() {
let mut app = create_test_app();
app.is_loading = false;
assert!(!app.is_loading);
}
#[test]
fn ctrl_c_disposition_idle_arms_exit_prompt() {
let app = create_test_app();
assert!(!app.is_loading);
assert!(!app.quit_is_armed());
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::ArmExit);
}
#[test]
fn ctrl_c_disposition_loading_cancels_turn() {
let mut app = create_test_app();
app.is_loading = true;
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::CancelTurn);
}
#[test]
fn ctrl_c_disposition_armed_idle_confirms_exit() {
let mut app = create_test_app();
app.arm_quit();
assert!(app.quit_is_armed());
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::ConfirmExit);
}
#[test]
fn ctrl_c_disposition_loading_beats_armed_quit() {
let mut app = create_test_app();
app.arm_quit();
app.is_loading = true;
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::CancelTurn);
}
#[test]
fn ctrl_c_disposition_no_selection_means_no_copy() {
let app = create_test_app();
assert!(!selection_has_content(&app));
assert_ne!(ctrl_c_disposition(&app), CtrlCDisposition::CopySelection);
}
#[test]
fn ctrl_c_raw_control_byte_routes_to_quit_arm_flow() {
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
let mut raw = KeyEvent::new(KeyCode::Char('\u{3}'), KeyModifiers::NONE);
normalize_raw_ctrl_c(&mut raw);
assert_eq!(raw.code, KeyCode::Char('c'));
assert!(raw.modifiers.contains(KeyModifiers::CONTROL));
let mut kitty = KeyEvent::new(KeyCode::Char('\u{3}'), KeyModifiers::CONTROL);
normalize_raw_ctrl_c(&mut kitty);
assert_eq!(kitty.code, KeyCode::Char('c'));
assert!(kitty.modifiers.contains(KeyModifiers::CONTROL));
let mut plain = KeyEvent::new(KeyCode::Char('c'), KeyModifiers::NONE);
normalize_raw_ctrl_c(&mut plain);
assert_eq!(plain.code, KeyCode::Char('c'));
assert!(!plain.modifiers.contains(KeyModifiers::CONTROL));
}
#[test]
fn ctrl_c_double_press_idle_exits_through_key_path() {
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
let mut app = create_test_app();
assert!(!app.is_loading);
assert!(!app.quit_is_armed());
let mut first = KeyEvent::new(KeyCode::Char('\u{3}'), KeyModifiers::NONE);
normalize_raw_ctrl_c(&mut first);
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::ArmExit);
app.arm_quit();
assert!(app.quit_is_armed());
let mut second = KeyEvent::new(KeyCode::Char('\u{3}'), KeyModifiers::NONE);
normalize_raw_ctrl_c(&mut second);
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::ConfirmExit);
}
#[test]
fn test_ctrl_d_exits_when_input_empty() {
let mut app = create_test_app();
app.input.clear();
assert!(app.input.is_empty());
}
#[test]
fn test_ctrl_d_does_nothing_when_input_not_empty() {
let mut app = create_test_app();
app.input = "some input".to_string();
assert!(!app.input.is_empty());
}
#[test]
fn test_esc_priority_order_matches_cancel_stack() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "draft".to_string();
app.mode = AppMode::Yolo;
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
app.input.clear();
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
app.queued_draft = Some(crate::tui::app::QueuedMessage::new(
"queued draft".to_string(),
None,
));
app.input = "editing queued draft".to_string();
assert_eq!(
next_escape_action(&app, false),
EscapeAction::DiscardQueuedDraft
);
app.queued_draft = None;
app.is_loading = false;
app.input = "draft".to_string();
assert_eq!(next_escape_action(&app, false), EscapeAction::ClearInput);
app.input.clear();
app.queued_draft = Some(crate::tui::app::QueuedMessage::new(
"queued draft".to_string(),
None,
));
assert_eq!(
next_escape_action(&app, false),
EscapeAction::DiscardQueuedDraft
);
app.queued_draft = None;
assert_eq!(next_escape_action(&app, false), EscapeAction::Noop);
}
#[test]
fn next_escape_action_pauses_then_cancels_pausable_command() {
let mut app = create_test_app();
app.is_loading = true;
app.pausable = true;
app.paused = false;
assert_eq!(next_escape_action(&app, false), EscapeAction::PauseCommand);
app.paused = true;
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
app.is_loading = false;
app.paused = false;
app.pausable = true;
app.paused_quarry = Some("Scan repos".to_string());
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
app.is_loading = true;
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
}
#[test]
fn visible_slash_menu_entries_respects_hide_flag() {
let mut app = create_test_app();
app.input = "/mo".to_string();
app.slash_menu_hidden = false;
let entries = visible_slash_menu_entries(&app, 6);
assert!(!entries.is_empty());
app.slash_menu_hidden = true;
let hidden_entries = visible_slash_menu_entries(&app, 6);
assert!(hidden_entries.is_empty());
}
#[test]
fn visible_slash_menu_entries_excludes_removed_commands() {
let mut app = create_test_app();
app.input = "/".to_string();
let entries = visible_slash_menu_entries(&app, 128);
assert!(entries.iter().any(|entry| entry.name == "/config"));
assert!(entries.iter().any(|entry| entry.name == "/links"));
assert!(!entries.iter().any(|entry| entry.name == "/set"));
assert!(!entries.iter().any(|entry| entry.name == "/codewhale"));
}
#[test]
fn visible_slash_model_completions_are_provider_scoped() {
let mut app = create_test_app();
app.api_provider = crate::config::ApiProvider::Together;
app.model = crate::config::DEFAULT_TOGETHER_MODEL.to_string();
app.provider_models.insert(
"openrouter".to_string(),
crate::config::DEFAULT_OPENROUTER_MODEL.to_string(),
);
app.input = "/model deep".to_string();
app.cursor_position = app.input.chars().count();
let entries = visible_slash_menu_entries(&app, 128);
let names = entries
.iter()
.map(|entry| entry.name.as_str())
.collect::<Vec<_>>();
assert!(names.contains(&"/model deepseek-ai/DeepSeek-V4-Pro"));
let openrouter_completion = format!("/model {}", crate::config::DEFAULT_OPENROUTER_MODEL);
assert!(
!names.contains(&openrouter_completion.as_str()),
"OpenRouter saved rows must not appear as bare Together /model completions"
);
}
#[test]
fn slash_menu_up_wraps_from_first_to_last() {
let mut app = create_test_app();
app.input = "/".to_string();
app.cursor_position = 1;
app.input_history.push("previous prompt".to_string());
let entries = visible_slash_menu_entries(&app, 128);
assert!(entries.len() > 1);
app.slash_menu_selected = 0;
select_previous_slash_menu_entry(&mut app, entries.len());
assert_eq!(app.slash_menu_selected, entries.len() - 1);
assert_eq!(app.input, "/");
}
#[test]
fn slash_menu_down_wraps_from_last_to_first() {
let mut app = create_test_app();
app.input = "/".to_string();
app.cursor_position = 1;
let entries = visible_slash_menu_entries(&app, 128);
assert!(entries.len() > 1);
app.slash_menu_selected = entries.len() - 1;
select_next_slash_menu_entry(&mut app, entries.len());
assert_eq!(app.slash_menu_selected, 0);
assert_eq!(app.input, "/");
}
#[test]
fn apply_slash_menu_selection_appends_space_for_arg_commands() {
let mut app = create_test_app();
let entries = vec![
crate::tui::widgets::SlashMenuEntry {
name: "/model".to_string(),
description: String::new(),
is_skill: false,
alias_hint: None,
},
crate::tui::widgets::SlashMenuEntry {
name: "/settings".to_string(),
description: String::new(),
is_skill: false,
alias_hint: None,
},
];
app.slash_menu_selected = 0;
assert!(apply_slash_menu_selection(&mut app, &entries, true));
assert_eq!(app.input, "/model ");
}
#[test]
fn apply_slash_menu_selection_keeps_change_executable_without_version() {
let mut app = create_test_app();
let entries = vec![crate::tui::widgets::SlashMenuEntry {
name: "/change".to_string(),
description: String::new(),
is_skill: false,
alias_hint: None,
}];
assert!(apply_slash_menu_selection(&mut app, &entries, true));
assert_eq!(app.input, "/change");
}
#[test]
fn apply_slash_menu_selection_uses_skill_command_form() {
let mut app = create_test_app();
let entries = vec![crate::tui::widgets::SlashMenuEntry {
name: "/skill search-files".to_string(),
description: "Search files".to_string(),
is_skill: true,
alias_hint: None,
}];
assert!(apply_slash_menu_selection(&mut app, &entries, true));
assert_eq!(app.input, "/skill search-files");
}
#[test]
fn inline_skill_slash_popup_lists_cached_skills_in_message() {
let mut app = create_test_app();
app.cached_skills = vec![
("search-files".to_string(), "Search files".to_string()),
("my-review".to_string(), "Review code".to_string()),
];
app.input = "please use /".to_string();
app.cursor_position = app.input.chars().count();
let entries = visible_slash_menu_entries(&app, 128);
assert!(entries.iter().any(|entry| entry.name == "/search-files"));
assert!(entries.iter().any(|entry| entry.name == "/my-review"));
assert!(entries.iter().all(|entry| entry.is_skill));
}
#[test]
fn inline_skill_slash_popup_filters_partial_without_leaking_to_command_position() {
let mut app = create_test_app();
app.cached_skills = vec![
("search-files".to_string(), "Search files".to_string()),
("my-review".to_string(), "Review code".to_string()),
];
app.input = "please use /my".to_string();
app.cursor_position = app.input.chars().count();
let entries = visible_slash_menu_entries(&app, 128);
assert_eq!(entries.len(), 1);
assert_eq!(entries[0].name, "/my-review");
app.input = "/se".to_string();
app.cursor_position = app.input.chars().count();
let command_entries = visible_slash_menu_entries(&app, 128);
assert!(
!command_entries
.iter()
.any(|entry| entry.name == "/search-files" && entry.is_skill),
"command-position slash menu should not include inline skill mentions"
);
}
#[test]
fn inline_skill_slash_popup_does_not_open_inside_command_arguments() {
let mut app = create_test_app();
app.cached_skills = vec![
(
"config-doctor".to_string(),
"Diagnose configuration".to_string(),
),
("cargo-ci-fixer".to_string(), "Fix CI failures".to_string()),
];
app.input = "/attach /".to_string();
app.cursor_position = app.input.chars().count();
let entries = visible_slash_menu_entries(&app, 128);
assert!(
entries.is_empty(),
"command argument paths should not show inline skill entries: {:?}",
entries.iter().map(|entry| &entry.name).collect::<Vec<_>>()
);
}
#[test]
fn apply_slash_menu_selection_splices_inline_skill_mention() {
let mut app = create_test_app();
app.input = "please use /se here".to_string();
app.cursor_position = "please use /se".chars().count();
let entries = vec![crate::tui::widgets::SlashMenuEntry {
name: "/search-files".to_string(),
description: "Search files".to_string(),
is_skill: true,
alias_hint: None,
}];
assert!(apply_slash_menu_selection(&mut app, &entries, true));
assert_eq!(app.input, "please use /search-files here");
assert_eq!(
app.cursor_position,
"please use /search-files".chars().count()
);
}
#[test]
fn try_autocomplete_slash_command_completes_skill_argument() {
let mut app = create_test_app();
app.cached_skills = vec![
("search-files".to_string(), "Search files".to_string()),
("my-review".to_string(), "Review code".to_string()),
];
app.input = "/skill my".to_string();
app.cursor_position = app.input.chars().count();
assert!(try_autocomplete_slash_command(&mut app));
assert_eq!(app.input, "/skill my-review");
}
#[test]
fn workspace_context_refresh_is_deferred_while_ui_is_busy() {
let repo = init_git_repo();
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
let now = Instant::now();
crate::tui::workspace_context::refresh_if_needed(&mut app, now, false);
assert!(app.workspace_context.is_none());
assert!(app.workspace_context_refreshed_at.is_none());
crate::tui::workspace_context::refresh_if_needed(&mut app, now, true);
let context = app
.workspace_context
.as_deref()
.expect("idle refresh should populate workspace context");
assert!(context.contains("clean"));
assert_eq!(app.workspace_context_refreshed_at, Some(now));
}
#[test]
fn workspace_context_refresh_respects_ttl_before_requerying_git() {
let repo = init_git_repo();
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
let start = Instant::now();
crate::tui::workspace_context::refresh_if_needed(&mut app, start, true);
let initial = app
.workspace_context
.clone()
.expect("initial refresh should populate context");
std::fs::write(repo.path().join("dirty.txt"), "dirty").expect("write dirty marker");
let before_ttl = start + Duration::from_secs(crate::tui::workspace_context::REFRESH_SECS - 1);
crate::tui::workspace_context::refresh_if_needed(&mut app, before_ttl, true);
assert_eq!(app.workspace_context.as_deref(), Some(initial.as_str()));
let after_ttl = start + Duration::from_secs(crate::tui::workspace_context::REFRESH_SECS);
crate::tui::workspace_context::refresh_if_needed(&mut app, after_ttl, true);
let refreshed = app
.workspace_context
.as_deref()
.expect("refresh after ttl should update context");
assert!(refreshed.contains("untracked"));
assert_ne!(refreshed, initial);
}
#[test]
fn completed_exec_tool_refreshes_workspace_context_before_ttl() {
let repo = init_git_repo();
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/old-branch"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
let start = Instant::now();
crate::tui::workspace_context::refresh_if_needed(&mut app, start, true);
let initial = app
.workspace_context
.clone()
.expect("initial refresh should populate context");
assert!(
initial.contains("feature/old-branch"),
"expected initial branch in {initial:?}"
);
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/new-branch"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let before_ttl = start + Duration::from_secs(crate::tui::workspace_context::REFRESH_SECS - 1);
crate::tui::workspace_context::refresh_if_needed(&mut app, before_ttl, true);
assert_eq!(
app.workspace_context.as_deref(),
Some(initial.as_str()),
"normal refresh should still respect the TTL"
);
handle_tool_call_started(
&mut app,
"shell-branch",
"exec_shell",
&serde_json::json!({"command": "git checkout -b feature/new-branch"}),
);
handle_tool_call_complete(
&mut app,
"shell-branch",
"exec_shell",
&ok_result("switched"),
);
let refreshed = app
.workspace_context
.as_deref()
.expect("shell completion should refresh context");
assert!(
refreshed.contains("feature/new-branch"),
"expected refreshed branch in {refreshed:?}"
);
}
#[test]
fn completed_task_shell_wait_refreshes_workspace_context_before_ttl() {
let repo = init_git_repo();
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/task-old"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
let start = Instant::now();
crate::tui::workspace_context::refresh_if_needed(&mut app, start, true);
let initial = app
.workspace_context
.clone()
.expect("initial refresh should populate context");
assert!(
initial.contains("feature/task-old"),
"expected initial branch in {initial:?}"
);
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/task-new"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let before_ttl = start + Duration::from_secs(crate::tui::workspace_context::REFRESH_SECS - 1);
crate::tui::workspace_context::refresh_if_needed(&mut app, before_ttl, true);
assert_eq!(
app.workspace_context.as_deref(),
Some(initial.as_str()),
"normal refresh should still respect the TTL"
);
handle_tool_call_started(
&mut app,
"task-shell-branch",
"task_shell_wait",
&serde_json::json!({"task_id": "shell_1"}),
);
handle_tool_call_complete(
&mut app,
"task-shell-branch",
"task_shell_wait",
&ok_result("completed"),
);
let refreshed = app
.workspace_context
.as_deref()
.expect("task shell completion should refresh context");
assert!(
refreshed.contains("feature/task-new"),
"expected refreshed branch in {refreshed:?}"
);
}
#[test]
fn completed_subagent_shell_tool_refreshes_workspace_context_before_ttl() {
let repo = init_git_repo();
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/subagent-old"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
let start = Instant::now();
crate::tui::workspace_context::refresh_if_needed(&mut app, start, true);
let initial = app
.workspace_context
.clone()
.expect("initial refresh should populate context");
assert!(
initial.contains("feature/subagent-old"),
"expected initial branch in {initial:?}"
);
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/subagent-new"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let before_ttl = start + Duration::from_secs(crate::tui::workspace_context::REFRESH_SECS - 1);
crate::tui::workspace_context::refresh_if_needed(&mut app, before_ttl, true);
assert_eq!(
app.workspace_context.as_deref(),
Some(initial.as_str()),
"normal refresh should still respect the TTL"
);
handle_subagent_mailbox(
&mut app,
42,
&crate::tools::subagent::MailboxMessage::ToolCallCompleted {
agent_id: "agent_branch".to_string(),
tool_name: "exec_shell".to_string(),
step: 1,
ok: true,
},
);
let refreshed = app
.workspace_context
.as_deref()
.expect("subagent shell completion should refresh context");
assert!(
refreshed.contains("feature/subagent-new"),
"expected refreshed branch in {refreshed:?}"
);
}
#[test]
fn workspace_context_drain_requests_redraw_when_context_changes() {
let mut app = create_test_app();
app.workspace_context = Some("feature/old | clean".to_string());
app.workspace_context_refreshed_at = Some(Instant::now());
app.needs_redraw = false;
{
let mut cell = app.workspace_context_cell.lock().expect("context cell");
*cell = Some("feature/new | clean".to_string());
}
crate::tui::workspace_context::refresh_if_needed(&mut app, Instant::now(), false);
assert_eq!(
app.workspace_context.as_deref(),
Some("feature/new | clean")
);
assert!(
app.needs_redraw,
"draining a changed async context should redraw the footer"
);
}
#[tokio::test]
async fn dismissed_plan_prompt_leaves_non_numeric_input_for_normal_send_path() {
let mut app = create_test_app();
app.mode = AppMode::Plan;
app.plan_prompt_pending = true;
app.offline_mode = true;
let engine = crate::core::engine::mock_engine_handle();
let config = Config::default();
let handled = handle_plan_choice(&mut app, &config, &engine.handle, "yolo")
.await
.expect("plan choice");
assert!(!handled);
assert!(!app.plan_prompt_pending);
assert_eq!(app.mode, AppMode::Plan);
let queued = build_queued_message(&mut app, "yolo".to_string());
submit_or_steer_message(&mut app, &config, &engine.handle, queued)
.await
.expect("submit normal message");
assert_eq!(app.queued_message_count(), 1);
assert_eq!(
app.queued_messages
.front()
.map(crate::tui::app::QueuedMessage::content),
Some("yolo".to_string())
);
assert_eq!(
app.status_message.as_deref(),
Some("Offline: 1 queued follow-up(s) — ↑ edit last, /queue send <n>")
);
}
#[tokio::test]
async fn dispatch_user_message_records_prompt_for_cancel_restore() {
let mut app = create_test_app();
app.show_thinking = false;
let config = Config::default();
let mut engine = crate::core::engine::mock_engine_handle();
let queued = crate::tui::app::QueuedMessage::new("fix this typo\nthen retry".to_string(), None);
dispatch_user_message(&mut app, &config, &engine.handle, queued)
.await
.expect("dispatch user message");
assert_eq!(
app.last_submitted_prompt.as_deref(),
Some("fix this typo\nthen retry")
);
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage {
content,
show_thinking,
..
} => {
assert_eq!(content, "fix this typo\nthen retry");
assert!(
!show_thinking,
"dispatch must carry the user's hidden-thinking setting into the engine"
);
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[tokio::test]
async fn startup_prompt_waits_for_onboarding_then_dispatches() {
let mut app = create_test_app();
app.input = "阅读项目 and wait".to_string();
app.cursor_position = app.input.chars().count();
app.auto_submit_initial_input = true;
app.onboarding = OnboardingState::Welcome;
let config = Config::default();
let mut engine = crate::core::engine::mock_engine_handle();
submit_initial_input_if_ready(&mut app, &config, &engine.handle)
.await
.expect("defer");
assert!(app.auto_submit_initial_input);
assert_eq!(app.input, "阅读项目 and wait");
assert_eq!(
app.status_message.as_deref(),
Some(INITIAL_PROMPT_DEFERRED_STATUS)
);
assert!(engine.rx_op.try_recv().is_err());
app.onboarding = OnboardingState::None;
submit_initial_input_if_ready(&mut app, &config, &engine.handle)
.await
.expect("submit");
assert!(!app.auto_submit_initial_input);
assert!(app.input.is_empty());
assert_eq!(
app.last_submitted_prompt.as_deref(),
Some("阅读项目 and wait")
);
match engine.rx_op.recv().await.expect("send message op") {
crate::core::ops::Op::SendMessage { content, .. } => {
assert!(content.contains("阅读项目 and wait"));
}
other => panic!("expected SendMessage, got {other:?}"),
}
}
#[tokio::test]
async fn steer_user_message_records_prompt_for_cancel_restore() {
let mut app = create_test_app();
let mut engine = crate::core::engine::mock_engine_handle();
let queued = crate::tui::app::QueuedMessage::new(
"adjust the active turn\nthen continue".to_string(),
None,
);
steer_user_message(&mut app, &engine.handle, queued)
.await
.expect("steer user message");
assert_eq!(
app.last_submitted_prompt.as_deref(),
Some("adjust the active turn\nthen continue")
);
assert_eq!(
engine.rx_steer.recv().await.as_deref(),
Some("adjust the active turn\nthen continue")
);
}
#[tokio::test]
async fn composer_send_shortcut_sends_next_queued_message_into_running_turn() {
let mut app = create_test_app();
app.is_loading = true;
app.queue_message(crate::tui::app::QueuedMessage::new(
"please attend to your sub agents".to_string(),
None,
));
let config = Config::default();
let mut engine = crate::core::engine::mock_engine_handle();
assert!(
send_shortcut_queued_message_now(&mut app, &config, &engine.handle)
.await
.expect("composer send shortcut succeeds")
);
assert_eq!(app.queued_message_count(), 0);
assert_eq!(
engine.rx_steer.recv().await.as_deref(),
Some("please attend to your sub agents")
);
}
#[tokio::test]
async fn composer_send_shortcut_sends_edited_queued_draft_into_running_turn() {
let mut app = create_test_app();
app.is_loading = true;
app.queued_draft = Some(crate::tui::app::QueuedMessage::new(
"original queued follow-up".to_string(),
Some("skill body".to_string()),
));
app.input = "edited queued follow-up".to_string();
app.cursor_position = app.input.chars().count();
let config = Config::default();
let mut engine = crate::core::engine::mock_engine_handle();
assert!(
send_shortcut_queued_message_now(&mut app, &config, &engine.handle)
.await
.expect("composer send shortcut succeeds")
);
assert!(app.queued_draft.is_none());
assert!(app.input.is_empty());
assert_eq!(app.queued_message_count(), 0);
let content = engine.rx_steer.recv().await.expect("steer content");
assert!(content.contains("edited queued follow-up"));
assert!(content.contains("skill body"));
}
#[test]
fn parse_queue_send_command_accepts_queue_alias_and_positive_index() {
assert_eq!(parse_queue_send_command("/queue send 2"), Some(Ok(1)));
assert_eq!(parse_queue_send_command("/queued now 1"), Some(Ok(0)));
assert_eq!(parse_queue_send_command("/queue SEND 1"), Some(Ok(0)));
assert!(parse_queue_send_command("/queue drop 1").is_none());
assert!(
parse_queue_send_command("/queue send 0")
.expect("send command should parse")
.is_err()
);
}
#[tokio::test]
async fn queue_send_index_sends_selected_message_into_running_turn() {
let mut app = create_test_app();
app.is_loading = true;
app.queue_message(crate::tui::app::QueuedMessage::new(
"first stays queued".to_string(),
None,
));
app.queue_message(crate::tui::app::QueuedMessage::new(
"second sends now".to_string(),
None,
));
let config = Config::default();
let mut engine = crate::core::engine::mock_engine_handle();
assert!(
send_queued_message_at_index_now(&mut app, &config, &engine.handle, 1)
.await
.expect("indexed send succeeds")
);
assert_eq!(app.queued_message_count(), 1);
assert_eq!(
app.queued_messages.front().map(|msg| msg.display.as_str()),
Some("first stays queued")
);
assert_eq!(
engine.rx_steer.recv().await.as_deref(),
Some("second sends now")
);
}
#[tokio::test]
async fn enter_while_model_waiting_steers_instead_of_queueing() {
let mut app = create_test_app();
app.is_loading = true;
app.streaming_message_index = None;
let config = Config::default();
let mut engine = crate::core::engine::mock_engine_handle();
let queued = build_queued_message(&mut app, "adjust current turn".to_string());
submit_or_steer_message(&mut app, &config, &engine.handle, queued)
.await
.expect("busy waiting submit steers");
assert_eq!(app.queued_message_count(), 0);
assert_eq!(
engine.rx_steer.recv().await.as_deref(),
Some("adjust current turn")
);
}
#[test]
fn engine_drain_budget_respects_event_and_time_limits() {
let start = Instant::now();
assert!(!engine_drain_budget_exhausted(0, start, start));
assert!(!engine_drain_budget_exhausted(1, start, start));
assert!(engine_drain_budget_exhausted(
MAX_ENGINE_EVENTS_PER_DRAIN,
start,
start
));
assert!(engine_drain_budget_exhausted(
1,
start,
start + ENGINE_DRAIN_TIME_BUDGET
));
}
#[test]
fn throttled_recovery_snapshot_persists_during_loading_turns() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "in-progress turn"));
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
let mut last_snapshot_at = None;
let t0 = Instant::now();
maybe_throttled_recovery_snapshot(&mut app, t0, &mut last_snapshot_at);
assert!(last_snapshot_at.is_some());
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.messages.len(), 1);
maybe_throttled_recovery_snapshot(
&mut app,
t0 + RECOVERY_SNAPSHOT_INTERVAL / 2,
&mut last_snapshot_at,
);
maybe_throttled_recovery_snapshot(
&mut app,
t0 + RECOVERY_SNAPSHOT_INTERVAL,
&mut last_snapshot_at,
);
}
#[tokio::test]
async fn steer_failure_queues_message_and_surfaces_toast() {
let mut app = create_test_app();
app.is_loading = true;
let engine = crate::core::engine::mock_engine_handle();
drop(engine.rx_steer);
let queued = crate::tui::app::QueuedMessage::new("follow up while busy".to_string(), None);
attempt_steer_with_queue_fallback(&mut app, &engine.handle, queued).await;
assert_eq!(app.queued_message_count(), 1);
let toast = app.status_toasts.back().expect("steer failure toast");
assert_eq!(toast.level, StatusToastLevel::Warning);
assert!(toast.text.contains("Steer failed"));
}
#[tokio::test]
async fn streaming_enter_queue_pushes_visible_toast() {
let mut app = create_test_app();
app.is_loading = true;
app.streaming_message_index = Some(0);
let config = Config::default();
let engine = crate::core::engine::mock_engine_handle();
let queued = build_queued_message(&mut app, "follow up during stream".to_string());
submit_or_steer_message(&mut app, &config, &engine.handle, queued)
.await
.expect("streaming submit queues");
assert_eq!(app.queued_message_count(), 1);
let toast = app.status_toasts.back().expect("queue toast");
assert_eq!(toast.level, StatusToastLevel::Info);
assert!(toast.text.contains("Queued follow-up"));
}
#[tokio::test]
async fn operate_streaming_enter_queues_another_parallel_task() {
let mut app = create_test_app();
app.mode = AppMode::Operate;
app.is_loading = true;
app.streaming_message_index = Some(0);
let config = Config::default();
let engine = crate::core::engine::mock_engine_handle();
let queued = build_queued_message(&mut app, "check the docs too".to_string());
submit_or_steer_message(&mut app, &config, &engine.handle, queued)
.await
.expect("Operate streaming submit queues another task");
assert_eq!(app.queued_message_count(), 1);
assert!(app.status_message.as_deref().is_some_and(
|status| status.contains("queued task(s)") && status.contains("workers continue")
));
let toast = app.status_toasts.back().expect("Operate queue toast");
assert_eq!(toast.level, StatusToastLevel::Info);
assert!(toast.text.contains("Queued task"));
assert!(toast.text.contains("dispatches next"));
}
#[tokio::test]
async fn inline_skill_request_keeps_instruction_when_busy_queueing() {
let mut app = create_test_app();
app.is_loading = true;
app.streaming_message_index = Some(0);
app.active_skill = Some("Use the test skill".to_string());
let config = Config::default();
let engine = crate::core::engine::mock_engine_handle();
let queued = build_queued_message(&mut app, "do X".to_string());
assert!(app.active_skill.is_none(), "skill must be consumed once");
submit_or_steer_message(&mut app, &config, &engine.handle, queued)
.await
.expect("streaming skill request queues");
let queued = app.queued_messages.front().expect("queued skill request");
assert_eq!(queued.display, "do X");
assert_eq!(
queued.skill_instruction.as_deref(),
Some("Use the test skill")
);
}
#[tokio::test]
async fn numeric_plan_choice_still_queues_follow_up_when_busy() {
let mut app = create_test_app();
app.mode = AppMode::Plan;
app.plan_prompt_pending = true;
app.is_loading = true;
let engine = crate::core::engine::mock_engine_handle();
let config = Config::default();
let handled = handle_plan_choice(&mut app, &config, &engine.handle, "2")
.await
.expect("plan choice");
assert!(handled);
assert!(!app.plan_prompt_pending);
assert_eq!(app.mode, AppMode::Agent);
assert_eq!(
app.approval_mode,
crate::tui::approval::ApprovalMode::Bypass
);
assert_eq!(app.queued_message_count(), 1);
assert_eq!(
app.queued_messages
.front()
.map(crate::tui::app::QueuedMessage::content),
Some("Proceed with the accepted plan.".to_string())
);
}
#[test]
fn api_key_validation_warns_without_blocking_unusual_formats() {
assert!(matches!(
crate::tui::onboarding::validate_api_key_for_onboarding(""),
crate::tui::onboarding::ApiKeyValidation::Reject(_)
));
assert!(matches!(
crate::tui::onboarding::validate_api_key_for_onboarding("sk short"),
crate::tui::onboarding::ApiKeyValidation::Reject(_)
));
assert!(matches!(
crate::tui::onboarding::validate_api_key_for_onboarding("short-key"),
crate::tui::onboarding::ApiKeyValidation::Accept { warning: Some(_) }
));
assert!(matches!(
crate::tui::onboarding::validate_api_key_for_onboarding("averylongkeywithoutdash123456"),
crate::tui::onboarding::ApiKeyValidation::Accept { warning: Some(_) }
));
assert!(matches!(
crate::tui::onboarding::validate_api_key_for_onboarding("sk-valid-format-1234567890"),
crate::tui::onboarding::ApiKeyValidation::Accept { warning: None }
));
}
#[test]
fn onboarding_after_api_key_save_does_not_repeat_language_step() {
let mut app = create_test_app();
app.onboarding = OnboardingState::ApiKey;
app.onboarding_needs_api_key = false;
app.trust_mode = true;
app.status_message = Some("saved".to_string());
crate::tui::onboarding::advance_onboarding_after_api_key(&mut app);
assert_eq!(app.onboarding, OnboardingState::Tips);
assert_eq!(app.status_message, None);
}
#[test]
fn onboarding_after_api_key_save_routes_to_trust_when_needed() {
let tmpdir = TempDir::new().expect("tempdir");
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.onboarding = OnboardingState::ApiKey;
app.onboarding_needs_api_key = false;
app.trust_mode = false;
crate::tui::onboarding::advance_onboarding_after_api_key(&mut app);
assert_eq!(app.onboarding, OnboardingState::TrustDirectory);
}
#[test]
fn api_key_escape_returns_to_provider_step() {
let mut app = create_test_app();
app.onboarding = OnboardingState::ApiKey;
app.api_key_input = "sk-test-value".to_string();
app.api_key_cursor = 4;
app.status_message = Some("editing".to_string());
back_from_api_key_onboarding(&mut app);
assert_eq!(app.onboarding, OnboardingState::Provider);
assert!(app.api_key_input.is_empty());
assert_eq!(app.api_key_cursor, 0);
assert_eq!(app.status_message, None);
}
#[test]
fn trust_directory_completion_advances_to_tips() {
let _guard = ConfigPathEnvGuard::new();
let tmpdir = TempDir::new().expect("workspace tempdir");
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.onboarding = OnboardingState::TrustDirectory;
app.onboarding_workspace_trust_gate = false;
app.trust_mode = false;
complete_trust_directory_onboarding(&mut app, &Config::default())
.expect("trust completion should succeed");
assert!(app.trust_mode);
assert_eq!(app.onboarding, OnboardingState::Tips);
assert!(app.runtime_services.hook_executor.is_some());
}
#[test]
fn prompt_override_notice_surfaces_in_transcript_and_toast() {
let _lock = crate::test_support::lock_test_env();
let _env = crate::test_support::EnvVarGuard::remove(prompts::BASE_PROMPT_OVERRIDE_OPT_IN_ENV);
let tmpdir = TempDir::new().expect("config tempdir");
let prompts_dir = tmpdir.path().join("prompts");
std::fs::create_dir_all(&prompts_dir).expect("prompts dir");
std::fs::write(prompts_dir.join("constitution.md"), "custom law\n").expect("override file");
let _ = prompts::take_prompt_override_notices();
assert!(prompts::load_config_dir_prompt_overrides(tmpdir.path()).is_empty());
let mut app = create_test_app();
surface_prompt_override_notices(&mut app);
assert!(
app.history.iter().any(|cell| matches!(
cell,
HistoryCell::System { content }
if content.contains(prompts::BASE_PROMPT_OVERRIDE_OPT_IN_ENV)
&& content.contains("bundled Constitution")
)),
"expected system warning in transcript, got {:?}",
app.history
);
let toast = app.status_toasts.back().expect("warning toast");
assert_eq!(toast.level, StatusToastLevel::Warning);
assert!(
toast
.text
.contains(prompts::BASE_PROMPT_OVERRIDE_OPT_IN_ENV)
);
}
#[test]
fn api_key_paste_shortcut_is_not_plain_text_input() {
let ctrl_v = KeyEvent::new(KeyCode::Char('v'), KeyModifiers::CONTROL);
assert!(crate::tui::key_shortcuts::is_paste_shortcut(&ctrl_v));
assert!(!crate::tui::key_shortcuts::is_text_input_key(&ctrl_v));
let legacy_ctrl_v = KeyEvent::new(KeyCode::Char('\u{16}'), KeyModifiers::NONE);
assert!(crate::tui::key_shortcuts::is_paste_shortcut(&legacy_ctrl_v));
assert!(!crate::tui::key_shortcuts::is_text_input_key(
&legacy_ctrl_v
));
let shifted = KeyEvent::new(KeyCode::Char('A'), KeyModifiers::SHIFT);
assert!(crate::tui::key_shortcuts::is_text_input_key(&shifted));
}
#[test]
fn international_layout_glyphs_remain_plain_text_input() {
for ch in ['\u{00e7}', '\u{00bf}'] {
let key = KeyEvent::new(KeyCode::Char(ch), KeyModifiers::NONE);
assert!(crate::tui::key_shortcuts::is_text_input_key(&key));
assert!(!crate::tui::shell_key_routing::is_help_shortcut(&key));
assert!(!crate::tui::shell_key_routing::is_context_inspector_shortcut(&key));
}
}
#[test]
fn jump_to_adjacent_tool_cell_finds_next_and_previous() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "hello".to_string(),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "file_search".to_string(),
status: ToolStatus::Success,
input_summary: Some("query: foo".to_string()),
output: Some("done".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
HistoryCell::Assistant {
content: "ok".to_string(),
streaming: false,
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "run_command".to_string(),
status: ToolStatus::Success,
input_summary: Some("ls".to_string()),
output: Some("...".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
];
app.mark_history_updated();
let cell_revisions = vec![app.history_version; app.history.len()];
app.viewport.transcript_cache.ensure(
&app.history,
&cell_revisions,
100,
app.transcript_render_options(),
);
app.viewport.last_transcript_top = 0;
assert!(jump_to_adjacent_tool_cell(
&mut app,
SearchDirection::Forward
));
assert!(!app.viewport.transcript_scroll.is_at_tail());
app.viewport.last_transcript_top = app
.viewport
.transcript_cache
.total_lines()
.saturating_sub(1);
assert!(jump_to_adjacent_tool_cell(
&mut app,
SearchDirection::Backward
));
}
fn first_line_for_cell(app: &App, cell_index: usize) -> usize {
app.viewport
.transcript_cache
.line_meta()
.iter()
.position(|meta| meta.cell_line().is_some_and(|(idx, _)| idx == cell_index))
.expect("cell should have rendered line")
}
fn pop_pager_body(app: &mut App) -> String {
let mut view = app.view_stack.pop().expect("pager view");
let pager = view
.as_any_mut()
.downcast_mut::<PagerView>()
.expect("top view should be pager");
pager.body_text()
}
#[test]
fn detail_target_prefers_visible_tool_card() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "hello".to_string(),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "file_search".to_string(),
status: ToolStatus::Success,
input_summary: Some("query: foo".to_string()),
output: Some("done".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
HistoryCell::Assistant {
content: "ok".to_string(),
streaming: false,
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Success,
input_summary: Some("command: ls".to_string()),
output: Some("...".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
];
app.tool_details_by_cell.insert(
1,
ToolDetailRecord {
tool_id: "search-1".to_string(),
tool_name: "file_search".to_string(),
input: serde_json::json!({"query": "foo"}),
output: Some("done".to_string()),
},
);
app.tool_details_by_cell.insert(
3,
ToolDetailRecord {
tool_id: "exec-1".to_string(),
tool_name: "exec_shell".to_string(),
input: serde_json::json!({"command": "ls"}),
output: Some("...".to_string()),
},
);
app.resync_history_revisions();
let revisions = app.history_revisions.clone();
app.viewport.transcript_cache.ensure(
&app.history,
&revisions,
100,
app.transcript_render_options(),
);
app.viewport.last_transcript_top = first_line_for_cell(&app, 1);
app.viewport.last_transcript_visible = 6;
assert_eq!(detail_target_cell_index(&app), Some(1));
let expected = format!(
"{} Turn Inspector · find · {}",
crate::tui::key_shortcuts::activity_shortcut_label(),
crate::tui::key_shortcuts::tool_details_shortcut_action_hint("raw details")
);
assert_eq!(
selected_detail_footer_label(&app).as_deref(),
Some(expected.as_str())
);
}
#[test]
fn activity_footer_hint_surfaces_visible_thinking_without_raw_tool_hint() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Thinking {
content: "visible reasoning".to_string(),
streaming: false,
duration_secs: Some(1.4),
}];
app.resync_history_revisions();
let revisions = app.history_revisions.clone();
app.viewport.transcript_cache.ensure(
&app.history,
&revisions,
100,
app.transcript_render_options(),
);
app.viewport.last_transcript_top = first_line_for_cell(&app, 0);
app.viewport.last_transcript_visible = 4;
assert_eq!(
selected_detail_footer_label(&app).as_deref(),
Some("Ctrl+O Turn Inspector · thinking")
);
}
#[test]
fn activity_footer_hint_uses_details_for_subagent_cards() {
let mut app = create_test_app();
app.history = vec![HistoryCell::SubAgent(
crate::tui::history::SubAgentCell::Delegate(
crate::tui::widgets::agent_card::DelegateCard::new("agent_123", "general"),
),
)];
app.resync_history_revisions();
let revisions = app.history_revisions.clone();
app.viewport.transcript_cache.ensure(
&app.history,
&revisions,
100,
app.transcript_render_options(),
);
app.viewport.last_transcript_top = first_line_for_cell(&app, 0);
app.viewport.last_transcript_visible = 4;
let expected = format!(
"{} Turn Inspector · sub-agent · {}",
crate::tui::key_shortcuts::activity_shortcut_label(),
crate::tui::key_shortcuts::tool_details_shortcut_action_hint("details")
);
assert_eq!(
selected_detail_footer_label(&app).as_deref(),
Some(expected.as_str())
);
}
#[test]
fn macos_option_v_glyph_is_treated_as_details_shortcut_only_on_macos() {
let option_v = KeyEvent::new(KeyCode::Char('\u{221A}'), KeyModifiers::NONE);
assert!(crate::tui::key_shortcuts::is_macos_option_v_legacy_key_for_platform(&option_v, true));
assert!(
!crate::tui::key_shortcuts::is_macos_option_v_legacy_key_for_platform(&option_v, false)
);
let modified = KeyEvent::new(KeyCode::Char('\u{221A}'), KeyModifiers::SHIFT);
assert!(!crate::tui::key_shortcuts::is_macos_option_v_legacy_key_for_platform(&modified, true));
let plain_v = KeyEvent::new(KeyCode::Char('v'), KeyModifiers::NONE);
assert!(!crate::tui::key_shortcuts::is_macos_option_v_legacy_key_for_platform(&plain_v, true));
}
#[test]
fn open_tool_details_pager_supports_active_virtual_tool_cell() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"active-1",
"exec_shell",
&serde_json::json!({"command": "echo hi"}),
);
let active_entries = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.to_vec();
app.viewport.transcript_cache.ensure_split(
&[&app.history, active_entries.as_slice()],
&[1],
100,
app.transcript_render_options(),
&app.folded_thinking,
None,
);
app.viewport.last_transcript_top = 0;
app.viewport.last_transcript_visible = 4;
assert_eq!(detail_target_cell_index(&app), Some(0));
assert!(open_tool_details_pager(&mut app));
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::Pager));
}
#[test]
fn tool_details_pager_frames_leaf_scope_and_preserves_raw_content() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Success,
input_summary: Some("command: ls".to_string()),
output: Some("total 0".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
}))];
app.tool_details_by_cell.insert(
0,
ToolDetailRecord {
tool_id: "exec-1".to_string(),
tool_name: "exec_shell".to_string(),
input: serde_json::json!({"command": "ls"}),
output: Some("total 0".to_string()),
},
);
app.resync_history_revisions();
assert!(open_details_pager_for_cell(&mut app, 0));
let mut view = app.view_stack.pop().expect("pager view");
let pager = view
.as_any_mut()
.downcast_mut::<PagerView>()
.expect("top view should be pager");
assert!(
pager.title().starts_with("Raw detail"),
"title should frame leaf scope: {}",
pager.title()
);
assert!(
pager.title().contains("exec_shell"),
"title should name the selected tool: {}",
pager.title()
);
let body = pager.body_text();
assert!(body.contains("Raw detail for the selected item"), "{body}");
assert!(body.contains("Ctrl+O"), "{body}");
assert!(body.contains("Input:"), "{body}");
assert!(body.contains("Output:"), "{body}");
assert!(body.contains("total 0"), "{body}");
assert!(body.contains("\"command\": \"ls\""), "{body}");
}
#[test]
fn tool_details_empty_state_points_to_turn_inspector() {
let mut app = create_test_app();
assert!(!open_details_pager_for_cell(&mut app, 999));
let msg = app.status_message.clone().unwrap_or_default();
assert!(
msg.contains("Ctrl+O"),
"empty state should point to Ctrl+O for the turn overview: {msg}"
);
}
#[test]
fn spillover_pager_section_returns_none_when_no_spillover() {
let mut app = create_test_app();
app.history = vec![HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Success,
input_summary: None,
output: Some("hi".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
}))];
app.resync_history_revisions();
assert!(spillover_pager_section(&app, 0).is_none());
}
#[test]
fn spillover_pager_section_loads_file_when_present() {
use std::io::Write;
let dir = tempfile::tempdir().unwrap();
let path = dir.path().join("call-test.txt");
let mut f = std::fs::File::create(&path).unwrap();
writeln!(f, "FULL_OUTPUT_BYTES_HERE").unwrap();
let mut app = create_test_app();
app.history = vec![HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Success,
input_summary: None,
output: Some("(truncated head)".to_string()),
prompts: None,
spillover_path: Some(path.clone()),
output_summary: None,
is_diff: false,
}))];
app.resync_history_revisions();
let section = spillover_pager_section(&app, 0).expect("section present");
assert!(section.contains("Full output (spillover)"));
assert!(
section.contains("FULL_OUTPUT_BYTES_HERE"),
"section missing file body: {section}"
);
assert!(section.contains(&path.display().to_string()));
}
#[test]
fn spillover_pager_section_returns_notice_when_file_missing() {
let mut app = create_test_app();
let bogus = std::path::PathBuf::from("/tmp/this/path/does/not/exist-spill.txt");
app.history = vec![HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "exec_shell".to_string(),
status: ToolStatus::Success,
input_summary: None,
output: Some("(truncated head)".to_string()),
prompts: None,
spillover_path: Some(bogus),
output_summary: None,
is_diff: false,
}))];
app.resync_history_revisions();
let section = spillover_pager_section(&app, 0).expect("still emits a notice section");
assert!(section.contains("could not read spillover file"));
}
#[test]
fn terminal_pause_has_live_owner_only_for_running_exec_cells() {
let mut app = create_test_app();
assert!(!terminal_pause_has_live_owner(&app));
let mut active = ActiveCell::new();
active.push_tool(
"tool-1",
HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "python3 -i".to_string(),
status: ToolStatus::Running,
output: None,
live_output: None,
shell_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
started_at: Some(Instant::now()),
duration_ms: None,
source: ExecSource::Assistant,
interaction: Some("interactive".to_string()),
output_summary: None,
})),
);
app.active_cell = Some(active);
assert!(terminal_pause_has_live_owner(&app));
let mut active = ActiveCell::new();
active.push_tool(
"tool-2",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "rlm".to_string(),
status: ToolStatus::Running,
input_summary: Some("file_path: Cargo.lock".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
assert!(
!terminal_pause_has_live_owner(&app),
"non-interactive RLM work must not keep the terminal in host-scrollback mode"
);
}
#[test]
fn active_rlm_task_entries_surface_foreground_rlm_work() {
let mut app = create_test_app();
app.turn_started_at = Some(Instant::now() - Duration::from_secs(3));
let mut active = ActiveCell::new();
active.push_tool(
"tool-rlm",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "rlm".to_string(),
status: ToolStatus::Running,
input_summary: Some("file_path: Cargo.lock".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
let entries = active_rlm_task_entries(&app);
assert_eq!(entries.len(), 1);
assert_eq!(entries[0].id, "rlm-1");
assert_eq!(entries[0].status, "running");
assert_eq!(entries[0].prompt_summary, "RLM: file_path: Cargo.lock");
assert_eq!(entries[0].kind, TaskPanelEntryKind::Background);
assert!(entries[0].duration_ms.unwrap_or_default() >= 3000);
}
#[test]
fn alt_nav_modifiers_require_alt_and_exclude_ctrl_super() {
assert!(!crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::NONE
));
assert!(!crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::SHIFT
));
assert!(crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::ALT
));
assert!(crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::ALT | KeyModifiers::SHIFT
));
assert!(!crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::CONTROL
));
assert!(!crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::ALT | KeyModifiers::CONTROL
));
assert!(!crate::tui::key_shortcuts::alt_nav_modifiers(
KeyModifiers::ALT | KeyModifiers::SUPER
));
}
#[test]
fn ctrl_h_is_treated_as_terminal_backspace() {
assert!(crate::tui::key_shortcuts::is_ctrl_h_backspace(
&KeyEvent::new(KeyCode::Char('h'), KeyModifiers::CONTROL)
));
assert!(!crate::tui::key_shortcuts::is_ctrl_h_backspace(
&KeyEvent::new(KeyCode::Char('h'), KeyModifiers::NONE)
));
assert!(!crate::tui::key_shortcuts::is_ctrl_h_backspace(
&KeyEvent::new(
KeyCode::Char('h'),
KeyModifiers::CONTROL | KeyModifiers::ALT
)
));
}
#[test]
fn partial_file_mention_finds_token_under_cursor() {
let input = "look at @docs/de please";
let cursor = "look at @docs/de".chars().count();
let (start, partial) = partial_file_mention_at_cursor(input, cursor)
.expect("cursor inside mention should yield a partial");
assert_eq!(start, "look at ".len(), "byte_start of @ in input");
assert_eq!(partial, "docs/de");
}
#[test]
fn partial_file_mention_returns_none_when_cursor_outside() {
let input = "look at @docs/de please";
let cursor = input.chars().count();
assert!(partial_file_mention_at_cursor(input, cursor).is_none());
let early_cursor = "look".chars().count();
assert!(partial_file_mention_at_cursor(input, early_cursor).is_none());
}
#[test]
fn partial_file_mention_handles_email_addresses() {
let input = "ping user@example.com now";
let cursor = "ping user@example.com".chars().count();
assert!(partial_file_mention_at_cursor(input, cursor).is_none());
}
#[test]
fn file_mention_completion_finds_unique_match() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(tmpdir.path().join("README.md"), "readme").unwrap();
std::fs::create_dir_all(tmpdir.path().join("docs")).unwrap();
std::fs::write(tmpdir.path().join("docs/deepseek_v4.pdf"), b"%PDF-").unwrap();
let ws = Workspace::with_cwd(tmpdir.path().to_path_buf(), None);
let matches = find_file_mention_completions(&ws, "docs/de", 16);
assert_eq!(matches, vec!["docs/deepseek_v4.pdf".to_string()]);
}
#[test]
fn file_mention_completion_ranks_prefix_before_substring() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(tmpdir.path().join("README.md"), "x").unwrap();
std::fs::create_dir_all(tmpdir.path().join("nested")).unwrap();
std::fs::write(tmpdir.path().join("nested/README.md"), "x").unwrap();
let ws = Workspace::with_cwd(tmpdir.path().to_path_buf(), None);
let matches = find_file_mention_completions(&ws, "README", 16);
assert_eq!(matches.first().map(String::as_str), Some("README.md"));
}
fn await_visible_mention_entries(app: &mut App, limit: usize) -> Vec<String> {
let partial = partial_file_mention_at_cursor(&app.input, app.cursor_position)
.expect("test input should contain a mention")
.1;
let started = Instant::now();
loop {
let entries = visible_mention_menu_entries(app, limit);
let ready = app
.composer
.mention_completion_cache
.as_ref()
.is_some_and(|cache| cache.partial == partial && cache.limit == limit);
if ready {
return entries;
}
assert!(
started.elapsed() < Duration::from_secs(2),
"timed out waiting for background mention discovery"
);
std::thread::sleep(Duration::from_millis(2));
}
}
#[test]
fn try_autocomplete_file_mention_unique_replaces_partial() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::create_dir_all(tmpdir.path().join("docs")).unwrap();
std::fs::write(tmpdir.path().join("docs/deepseek_v4.pdf"), b"%PDF-").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "summarize @docs/de".to_string();
app.cursor_position = app.input.chars().count();
let _ = await_visible_mention_entries(&mut app, 64);
assert!(try_autocomplete_file_mention(&mut app));
assert_eq!(app.input, "summarize @docs/deepseek_v4.pdf");
assert_eq!(app.cursor_position, app.input.chars().count());
}
#[test]
fn try_autocomplete_file_mention_extends_to_common_prefix() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::create_dir_all(tmpdir.path().join("crates/tui")).unwrap();
std::fs::write(tmpdir.path().join("crates/tui/lib.rs"), "//").unwrap();
std::fs::write(tmpdir.path().join("crates/tui/main.rs"), "//").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "@crates/tui/".to_string();
app.cursor_position = app.input.chars().count();
let _ = await_visible_mention_entries(&mut app, 64);
assert!(try_autocomplete_file_mention(&mut app));
assert!(app.input.starts_with("@crates/tui/"));
let preview = app
.status_message
.as_deref()
.expect("status message should describe candidates");
assert!(preview.contains("@crates/tui/lib.rs"));
assert!(preview.contains("@crates/tui/main.rs"));
}
#[test]
fn try_autocomplete_file_mention_no_match_reports_status() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(tmpdir.path().join("README.md"), "x").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "@nonexistent_xyz".to_string();
app.cursor_position = app.input.chars().count();
let _ = await_visible_mention_entries(&mut app, 64);
assert!(try_autocomplete_file_mention(&mut app));
assert_eq!(app.input, "@nonexistent_xyz");
assert_eq!(
app.status_message.as_deref(),
Some("No files match @nonexistent_xyz")
);
}
#[test]
fn try_autocomplete_file_mention_no_match_mentions_depth_cap_for_path_like_partial() {
let tmpdir = TempDir::new().expect("tempdir");
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.mention_walk_depth = 6;
app.input = "@a/b/c/d/e/f/g/target".to_string();
app.cursor_position = app.input.chars().count();
let _ = await_visible_mention_entries(&mut app, 64);
assert!(try_autocomplete_file_mention(&mut app));
assert_eq!(
app.status_message.as_deref(),
Some(
"No files match @a/b/c/d/e/f/g/target (mention_walk_depth=6; use /config set mention_walk_depth 0 to search deeper)"
)
);
}
#[test]
fn try_autocomplete_file_mention_no_match_skips_depth_hint_for_shallow_path() {
let tmpdir = TempDir::new().expect("tempdir");
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.mention_walk_depth = 6;
app.input = "@shallow_missing/main.rs".to_string();
app.cursor_position = app.input.chars().count();
let _ = await_visible_mention_entries(&mut app, 64);
assert!(try_autocomplete_file_mention(&mut app));
assert_eq!(
app.status_message.as_deref(),
Some("No files match @shallow_missing/main.rs")
);
}
#[test]
fn try_autocomplete_file_mention_no_match_skips_depth_hint_when_unlimited() {
let tmpdir = TempDir::new().expect("tempdir");
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.mention_walk_depth = 0;
app.input = "@a/b/c/d/e/f/g/target".to_string();
app.cursor_position = app.input.chars().count();
let _ = await_visible_mention_entries(&mut app, 64);
assert!(try_autocomplete_file_mention(&mut app));
assert_eq!(
app.status_message.as_deref(),
Some("No files match @a/b/c/d/e/f/g/target")
);
}
#[test]
fn try_autocomplete_file_mention_returns_false_outside_mention() {
let mut app = create_test_app();
app.input = "no mention here".to_string();
app.cursor_position = app.input.chars().count();
assert!(!try_autocomplete_file_mention(&mut app));
}
#[test]
fn mention_popup_is_empty_when_cursor_is_not_in_a_mention() {
let mut app = create_test_app();
app.input = "no mention here".to_string();
app.cursor_position = app.input.chars().count();
assert!(visible_mention_menu_entries(&mut app, 6).is_empty());
}
#[test]
fn plain_at_returns_immediately_while_discovery_is_stalled() {
let (started_tx, started_rx) = std::sync::mpsc::channel();
let (release_tx, release_rx) = std::sync::mpsc::channel();
let release_rx = std::sync::Mutex::new(release_rx);
let mut app = create_test_app();
app.input = "@".to_string();
app.cursor_position = 1;
app.composer.mention_discovery =
crate::tui::mention_completion::MentionDiscovery::with_scanner(move |_, _| {
let _ = started_tx.send(());
let _ = release_rx.lock().expect("release lock").recv();
vec!["README.md".to_string()]
});
let started = Instant::now();
let initial = visible_mention_menu_entries(&mut app, 6);
assert!(initial.is_empty());
assert!(
started.elapsed() < Duration::from_millis(50),
"plain @ waited for discovery instead of returning immediately"
);
started_rx
.recv_timeout(Duration::from_secs(1))
.expect("plain @ should start discovery in the background");
release_tx.send(()).unwrap();
assert_eq!(
await_visible_mention_entries(&mut app, 6),
vec!["README.md"]
);
}
#[test]
fn mention_popup_lists_workspace_matches_for_cursor_partial() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::create_dir_all(tmpdir.path().join("docs")).unwrap();
std::fs::write(tmpdir.path().join("docs/deepseek_v4.pdf"), b"%PDF-").unwrap();
std::fs::write(tmpdir.path().join("docs/MCP.md"), "x").unwrap();
std::fs::write(tmpdir.path().join("README.md"), "x").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "look at @docs/".to_string();
app.cursor_position = app.input.chars().count();
let entries = await_visible_mention_entries(&mut app, 6);
assert!(!entries.is_empty(), "popup should surface docs/ entries");
assert!(entries.iter().any(|e| e.starts_with("docs/")));
assert!(!entries.iter().any(|e| e == "README.md"));
}
#[test]
fn mention_popup_browser_mode_lists_immediate_directory_children() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::create_dir_all(tmpdir.path().join("src/nested")).unwrap();
std::fs::write(tmpdir.path().join("src/lib.rs"), "lib").unwrap();
std::fs::write(tmpdir.path().join("src/nested/deep.rs"), "deep").unwrap();
std::fs::write(tmpdir.path().join("README.md"), "readme").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.mention_menu_behavior = "browser".to_string();
app.input = "look at @src/".to_string();
app.cursor_position = app.input.chars().count();
let entries = await_visible_mention_entries(&mut app, 8);
assert_eq!(entries, vec!["src/lib.rs", "src/nested/"]);
}
#[test]
fn mention_popup_reuses_cache_when_cursor_moves_inside_same_token() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::create_dir_all(tmpdir.path().join("docs")).unwrap();
std::fs::write(tmpdir.path().join("docs/alpha.md"), "x").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "look at @docs/".to_string();
app.cursor_position = app.input.chars().count();
let entries = await_visible_mention_entries(&mut app, 6);
assert!(entries.iter().any(|e| e == "docs/alpha.md"));
std::fs::write(tmpdir.path().join("docs/beta.md"), "x").unwrap();
app.cursor_position = "look at @do".chars().count();
let entries_after_cursor_move = visible_mention_menu_entries(&mut app, 6);
assert_eq!(
entries_after_cursor_move, entries,
"cursor movement inside one @mention token should not re-walk the workspace",
);
app.input = "look at @docs/b".to_string();
app.cursor_position = app.input.chars().count();
app.composer.mention_discovery.invalidate();
app.composer.mention_completion_cache = None;
let entries_after_partial_change = await_visible_mention_entries(&mut app, 6);
assert!(
entries_after_partial_change
.iter()
.any(|e| e == "docs/beta.md"),
"changing the partial should invalidate the completion cache",
);
}
#[test]
fn mention_popup_respects_hidden_flag() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::write(tmpdir.path().join("README.md"), "x").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "@READ".to_string();
app.cursor_position = app.input.chars().count();
app.mention_menu_hidden = true;
assert!(
visible_mention_menu_entries(&mut app, 6).is_empty(),
"Esc-hidden popup must not surface entries until next input edit",
);
}
#[test]
fn apply_mention_menu_selection_splices_selected_entry() {
let tmpdir = TempDir::new().expect("tempdir");
std::fs::create_dir_all(tmpdir.path().join("crates/tui")).unwrap();
std::fs::write(tmpdir.path().join("crates/tui/lib.rs"), "//").unwrap();
std::fs::write(tmpdir.path().join("crates/tui/main.rs"), "//").unwrap();
let mut app = create_test_app();
app.workspace = tmpdir.path().to_path_buf();
app.input = "open @crates/tui/m".to_string();
app.cursor_position = app.input.chars().count();
let entries = await_visible_mention_entries(&mut app, 6);
assert!(!entries.is_empty(), "expected entries for @crates/tui/m");
app.mention_menu_selected = 0;
let applied = apply_mention_menu_selection(&mut app, &entries);
assert!(
applied,
"apply_mention_menu_selection should report success"
);
assert!(
app.input.starts_with("open @"),
"input should still start with `open @`, got: {input}",
input = app.input,
);
assert_eq!(app.cursor_position, app.input.chars().count());
}
#[test]
fn apply_mention_menu_selection_is_noop_outside_a_mention() {
let mut app = create_test_app();
app.input = "no @ here".to_string();
app.cursor_position = 1; let applied = apply_mention_menu_selection(&mut app, &["whatever".to_string()]);
assert!(!applied);
assert_eq!(app.input, "no @ here");
}
#[test]
fn apply_mention_menu_selection_with_no_entries_is_noop() {
let mut app = create_test_app();
app.input = "@partial".to_string();
app.cursor_position = app.input.chars().count();
let applied = apply_mention_menu_selection(&mut app, &[]);
assert!(!applied);
}
fn ok_result(
content: &str,
) -> Result<crate::tools::spec::ToolResult, crate::tools::spec::ToolError> {
Ok(crate::tools::spec::ToolResult::success(content))
}
fn hydrated_result(
content: &str,
) -> Result<crate::tools::spec::ToolResult, crate::tools::spec::ToolError> {
Ok(
crate::tools::spec::ToolResult::success(content).with_metadata(serde_json::json!({
"event": "tool.schema_hydrated",
"tool": "exec_shell",
"executed": false,
"retry_required": true,
"deferred_tool_loaded": true,
"tool_name": "exec_shell",
})),
)
}
fn rendered_text(lines: &[ratatui::text::Line<'_>]) -> String {
lines
.iter()
.map(|line| {
line.spans
.iter()
.map(|span| span.content.as_ref())
.collect::<String>()
})
.collect::<Vec<_>>()
.join("\n")
}
#[test]
fn completed_exec_tool_result_still_renders_run_done() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"shell-ok",
"exec_shell",
&serde_json::json!({"command": "echo hi"}),
);
handle_tool_call_complete(&mut app, "shell-ok", "exec_shell", &ok_result("hi"));
let exec = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.iter()
.find_map(|cell| match cell {
HistoryCell::Tool(ToolCell::Exec(exec)) => Some(exec),
_ => None,
})
.expect("exec cell");
assert_eq!(exec.status, ToolStatus::Success);
let text = rendered_text(&exec.lines_with_motion(100, true));
assert!(text.contains("run done"), "{text}");
assert!(!text.contains("tool loaded - retry required"), "{text}");
}
#[test]
fn hydrated_exec_tool_result_renders_retry_required_not_run_done() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"shell-hydrated",
"exec_shell",
&serde_json::json!({"command": "cargo test"}),
);
handle_tool_call_complete(
&mut app,
"shell-hydrated",
"exec_shell",
&hydrated_result(
"Tool exec_shell was deferred and has now been loaded.\n\
The tool was not executed. Retry with the loaded schema.",
),
);
let exec = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.iter()
.find_map(|cell| match cell {
HistoryCell::Tool(ToolCell::Exec(exec)) => Some(exec),
_ => None,
})
.expect("exec cell");
assert_eq!(exec.status, ToolStatus::Hydrated);
let text = rendered_text(&exec.lines_with_motion(120, true));
assert!(text.contains("run tool loaded - retry required"), "{text}");
assert!(!text.contains("run done"), "{text}");
}
#[test]
fn hydrated_tool_with_validation_body_still_uses_hydrated_status() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"generic-hydrated",
"deferred_tool",
&serde_json::json!({"unexpected": true}),
);
handle_tool_call_complete(
&mut app,
"generic-hydrated",
"deferred_tool",
&hydrated_result(
"Tool deferred_tool was deferred and has now been loaded.\n\n\
Missing required fields:\n command\n\n\
Unexpected fields:\n unexpected",
),
);
let generic = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.iter()
.find_map(|cell| match cell {
HistoryCell::Tool(ToolCell::Generic(generic)) => Some(generic),
_ => None,
})
.expect("generic cell");
assert_eq!(generic.status, ToolStatus::Hydrated);
let text = rendered_text(&HistoryCell::Tool(ToolCell::Generic(generic.clone())).lines(120));
assert!(text.contains("tool loaded - retry required"), "{text}");
assert!(!text.contains("tool done"), "{text}");
}
#[test]
fn failed_tool_result_with_hydration_metadata_stays_failed() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"generic-failed",
"deferred_tool",
&serde_json::json!({}),
);
let result = Ok(crate::tools::spec::ToolResult::error("boom").with_metadata(
serde_json::json!({
"event": "tool.schema_hydrated",
"executed": false,
"retry_required": true,
}),
));
handle_tool_call_complete(&mut app, "generic-failed", "deferred_tool", &result);
let generic = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.iter()
.find_map(|cell| match cell {
HistoryCell::Tool(ToolCell::Generic(generic)) => Some(generic),
_ => None,
})
.expect("generic cell");
assert_eq!(generic.status, ToolStatus::Failed);
let text = rendered_text(&HistoryCell::Tool(ToolCell::Generic(generic.clone())).lines(120));
assert!(text.contains("tool issue"), "{text}");
assert!(!text.contains("tool loaded - retry required"), "{text}");
}
#[test]
fn shell_wait_without_command_uses_task_id_until_command_metadata_arrives() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"shell-wait",
"exec_shell_wait",
&serde_json::json!({"task_id": "shell_33a08c3c"}),
);
let exec = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.iter()
.find_map(|cell| match cell {
HistoryCell::Tool(ToolCell::Exec(exec)) => Some(exec),
_ => None,
})
.expect("exec cell");
assert_eq!(exec.command, "command shell_33a08c3c");
assert!(
exec.interaction
.as_deref()
.is_some_and(|text| text.contains("shell_33a08c3c"))
);
assert!(
!exec.command.contains("<command>")
&& !exec
.interaction
.as_deref()
.unwrap_or_default()
.contains("<command>")
);
let result = Ok(crate::tools::spec::ToolResult::success(
"Background task running (no new output).",
)
.with_metadata(serde_json::json!({
"status": "Running",
"duration_ms": 178_000_u64,
"task_id": "shell_33a08c3c",
"command": "cargo test --workspace --all-features",
})));
handle_tool_call_complete(&mut app, "shell-wait", "exec_shell_wait", &result);
let exec = app
.active_cell
.as_ref()
.expect("active cell")
.entries()
.iter()
.find_map(|cell| match cell {
HistoryCell::Tool(ToolCell::Exec(exec)) => Some(exec),
_ => None,
})
.expect("exec cell");
assert_eq!(exec.command, "cargo test --workspace --all-features");
assert!(
exec.interaction
.as_deref()
.is_some_and(|text| text.contains("cargo test --workspace"))
);
}
#[test]
fn tool_child_usage_metadata_updates_live_cost_counter() {
let mut app = create_test_app();
let result = Ok(crate::tools::spec::ToolResult::success("ok").with_metadata(
serde_json::json!({
"child_model": "deepseek-v4-flash",
"child_input_tokens": 10_000,
"child_output_tokens": 1_000,
"child_prompt_cache_hit_tokens": 7_000,
"child_prompt_cache_miss_tokens": 3_000,
}),
));
handle_tool_call_complete(&mut app, "review-usage", "review", &result);
assert!(app.session.subagent_cost > 0.0);
}
#[test]
fn picker_renamed_active_title_survives_automatic_snapshot() {
let mut app = create_test_app();
let manager = SessionManager::new(tempfile::tempdir().expect("tempdir").path().to_path_buf())
.expect("session manager");
let mut metadata = crate::session_manager::create_saved_session_with_id_and_mode(
"session-active".to_string(),
&app.api_messages,
&app.model,
&app.workspace,
0,
app.system_prompt.as_ref(),
Some(app.mode.as_setting()),
)
.metadata;
metadata.title = "Before".to_string();
metadata.parent_session_id = Some("session-parent".to_string());
metadata.forked_from_message_count = Some(7);
let created_at = metadata.created_at;
app.current_session_id = Some(metadata.id.clone());
app.current_session_metadata = Some(metadata.clone());
app.session_title = Some(metadata.title.clone());
metadata.title = "After".to_string();
assert!(apply_picker_session_rename_to_active_app(
&mut app, metadata
));
let snapshot = build_session_snapshot(&mut app, &manager).expect("snapshot");
assert_eq!(snapshot.metadata.title, "After");
assert_eq!(snapshot.metadata.created_at, created_at);
assert_eq!(
snapshot.metadata.parent_session_id.as_deref(),
Some("session-parent")
);
assert_eq!(snapshot.metadata.forked_from_message_count, Some(7));
}
#[test]
fn picker_rename_of_inactive_session_does_not_touch_active_metadata() {
let mut app = create_test_app();
let mut active = crate::session_manager::create_saved_session_with_id_and_mode(
"session-active".to_string(),
&app.api_messages,
&app.model,
&app.workspace,
0,
app.system_prompt.as_ref(),
Some(app.mode.as_setting()),
)
.metadata;
active.title = "Active".to_string();
let mut inactive = active.clone();
inactive.id = "session-inactive".to_string();
inactive.title = "Renamed inactive".to_string();
app.current_session_id = Some(active.id.clone());
app.current_session_metadata = Some(active.clone());
app.session_title = Some(active.title.clone());
assert!(!apply_picker_session_rename_to_active_app(
&mut app, inactive
));
assert_eq!(app.session_title.as_deref(), Some("Active"));
assert_eq!(
app.current_session_metadata
.as_ref()
.map(|metadata| metadata.title.as_str()),
Some("Active")
);
}
#[test]
fn codex_tool_child_usage_does_not_inherit_public_api_pricing() {
let mut app = create_test_app();
app.api_provider = crate::config::ApiProvider::OpenaiCodex;
app.billing_presentation =
crate::route_billing::BillingPresentation::Subscription("Codex OAuth quota");
let result = Ok(crate::tools::spec::ToolResult::success("ok").with_metadata(
serde_json::json!({
"child_model": "gpt-5.5",
"child_input_tokens": 10_000,
"child_output_tokens": 1_000,
"child_provider": "openai-codex",
}),
));
handle_tool_call_complete(&mut app, "review-usage", "review", &result);
assert_eq!(app.session.subagent_cost, 0.0);
}
#[test]
fn spilled_tool_completion_records_session_artifact_metadata() {
let tmp = tempfile::tempdir().expect("tempdir");
let spillover_path = tmp.path().join("call-big.txt");
let raw = "checking crate ... error[E0425]: cannot find value\n".repeat(20);
std::fs::write(&spillover_path, &raw).expect("write spillover");
let result = Ok(
crate::tools::spec::ToolResult::success("checking crate ...").with_metadata(
serde_json::json!({
"spillover_path": spillover_path.display().to_string(),
"artifact_session_id": "session-123",
"artifact_relative_path": "artifacts/art_call-big.txt",
"artifact_byte_size": raw.len() as u64,
"artifact_preview": "checking crate ... error[E0425]: cannot find value",
}),
),
);
let mut app = create_test_app();
app.current_session_id = Some("session-123".to_string());
handle_tool_call_complete(&mut app, "call-big", "exec_shell", &result);
assert_eq!(app.session_artifacts.len(), 1);
let artifact = &app.session_artifacts[0];
assert_eq!(artifact.kind, crate::artifacts::ArtifactKind::ToolOutput);
assert_eq!(artifact.session_id, "session-123");
assert_eq!(artifact.tool_call_id, "call-big");
assert_eq!(artifact.tool_name, "exec_shell");
assert_eq!(artifact.byte_size, raw.len() as u64);
assert_eq!(
artifact.storage_path,
PathBuf::from("artifacts/art_call-big.txt")
);
assert!(artifact.preview.starts_with("checking crate"));
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.artifacts, app.session_artifacts);
}
#[test]
fn first_snapshot_preserves_current_session_id_for_artifact_ownership() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.current_session_id = Some("session-123".to_string());
app.api_messages.push(text_message("user", "hello"));
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.metadata.id, "session-123");
}
#[test]
fn existing_session_snapshot_updates_model_selection() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut existing = saved_session_with_messages(vec![text_message("user", "hello")]);
existing.metadata.model = "auto".to_string();
manager
.save_session(&existing)
.expect("save existing session");
let mut app = create_test_app();
app.current_session_id = Some(existing.metadata.id.clone());
app.api_messages.push(text_message("user", "hello"));
app.set_model_selection("deepseek-v4-flash".to_string());
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
assert_eq!(snapshot.metadata.id, existing.metadata.id);
assert_eq!(snapshot.metadata.model, "deepseek-v4-flash");
}
#[test]
fn automatic_session_snapshot_keeps_named_custom_identity_secret_free() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut config =
named_custom_session_config("lm-studio", "http://127.0.0.1:1234/v1", "local-code-model");
config
.providers
.as_mut()
.and_then(|providers| providers.custom.get_mut("lm-studio"))
.expect("custom provider")
.api_key = Some("super-secret-local-key".to_string());
let mut app = App::new(create_test_options(), &config);
app.api_messages.push(text_message("user", "persist me"));
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
let serialized = serde_json::to_string(&snapshot).expect("serialize session");
assert_eq!(snapshot.metadata.model_provider, "custom");
assert_eq!(
snapshot.metadata.model_provider_id.as_deref(),
Some("lm-studio")
);
assert!(serialized.contains("\"model_provider\":\"custom\""));
assert!(serialized.contains("\"model_provider_id\":\"lm-studio\""));
assert!(!serialized.contains("super-secret-local-key"));
assert!(!serialized.contains("127.0.0.1:1234"));
}
#[test]
fn automatic_session_snapshot_omits_id_for_legacy_root_custom_route() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let config = Config {
provider: Some("custom".to_string()),
base_url: Some("http://127.0.0.1:18180/v1".to_string()),
default_text_model: Some("legacy-root-model".to_string()),
..Config::default()
};
let mut app = App::new(create_test_options(), &config);
app.api_messages.push(text_message("user", "persist root"));
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
let serialized = serde_json::to_string(&snapshot).expect("serialize session");
assert_eq!(snapshot.metadata.model_provider, "custom");
assert_eq!(snapshot.metadata.model_provider_id, None);
assert!(!serialized.contains("model_provider_id"));
}
#[test]
fn session_snapshot_and_resume_round_trip_work_state() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.api_messages.push(text_message("user", "keep my work"));
app.api_messages
.push(text_message("assistant", "continuity captured"));
{
let mut todos = app.todos.try_lock().expect("todos lock");
todos.add(
"inspect".to_string(),
crate::tools::todo::TodoStatus::Completed,
);
todos.add(
"patch".to_string(),
crate::tools::todo::TodoStatus::InProgress,
);
}
{
let mut plan = app.plan_state.try_lock().expect("plan lock");
plan.update(crate::tools::plan::UpdatePlanArgs {
objective: Some("Ship continuity".to_string()),
plan: vec![crate::tools::plan::PlanItemArg {
step: "verify".to_string(),
status: crate::tools::plan::StepStatus::InProgress,
}],
..crate::tools::plan::UpdatePlanArgs::default()
});
}
let snapshot = build_session_snapshot(&mut app, &manager).expect("session snapshot");
let expected = snapshot.work_state.clone().expect("persisted Work state");
manager.save_session(&snapshot).expect("save session");
let loaded = manager
.load_session(&snapshot.metadata.id)
.expect("load session");
let mut restored = create_test_app();
let recovered = apply_loaded_session(&mut restored, &mut Config::default(), &loaded)
.expect("restore session");
assert!(!recovered);
assert_eq!(
restored.work_state_snapshot().expect("restored snapshot"),
Some(expected)
);
}
#[test]
fn session_snapshot_preserves_last_work_state_when_lock_is_busy() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.current_session_id = Some("work-lock-session".to_string());
app.todos.try_lock().expect("todos lock").add(
"durable".to_string(),
crate::tools::todo::TodoStatus::InProgress,
);
let initial = build_session_snapshot(&mut app, &manager).expect("initial snapshot");
let expected = initial.work_state.clone();
manager.save_session(&initial).expect("save initial");
let todos = app.todos.clone();
let _held = todos.try_lock().expect("hold todos lock");
let contended = build_session_snapshot(&mut app, &manager).expect("contended snapshot");
assert_eq!(contended.work_state, expected);
}
#[test]
fn session_snapshot_prefers_newer_memory_over_stale_disk_when_lock_is_busy() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.current_session_id = Some("work-cache-newer-than-disk".to_string());
app.todos.try_lock().expect("todos lock").add(
"disk state".to_string(),
crate::tools::todo::TodoStatus::Completed,
);
let disk_snapshot = build_session_snapshot(&mut app, &manager).expect("disk snapshot");
manager
.save_session(&disk_snapshot)
.expect("save disk snapshot");
app.todos.try_lock().expect("todos lock").add(
"newer memory state".to_string(),
crate::tools::todo::TodoStatus::InProgress,
);
let memory_snapshot = build_session_snapshot(&mut app, &manager).expect("memory snapshot");
assert_ne!(memory_snapshot.work_state, disk_snapshot.work_state);
let todos = app.todos.clone();
let _held = todos.try_lock().expect("hold todos lock");
let contended = build_session_snapshot(&mut app, &manager).expect("contended snapshot");
assert_eq!(contended.work_state, memory_snapshot.work_state);
}
#[test]
fn automatic_session_snapshot_never_reloads_existing_json_on_ui_thread() {
let tmp = tempfile::tempdir().expect("tempdir");
let sessions_dir = tmp.path().join("sessions");
let manager =
crate::session_manager::SessionManager::new(sessions_dir.clone()).expect("manager");
let mut app = create_test_app();
app.current_session_id = Some("nonblocking-snapshot".to_string());
app.api_messages
.push(text_message("user", "first in-memory checkpoint"));
let initial = build_session_snapshot(&mut app, &manager).expect("initial snapshot");
manager.save_session(&initial).expect("save initial");
std::fs::write(
sessions_dir.join("nonblocking-snapshot.json"),
"{ intentionally malformed and never read",
)
.expect("corrupt disk fixture");
app.api_messages
.push(text_message("assistant", "newer in-memory state"));
let snapshot = build_session_snapshot(&mut app, &manager).expect("nonblocking snapshot");
assert_eq!(snapshot.metadata.id, "nonblocking-snapshot");
assert_eq!(snapshot.messages.len(), 2);
assert_eq!(snapshot.metadata.created_at, initial.metadata.created_at);
}
#[test]
fn renamed_title_survives_next_in_memory_automatic_snapshot() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut session = saved_session_with_messages(vec![text_message("user", "original title")]);
session.metadata.title = "Original Title".to_string();
manager.save_session(&session).expect("save session");
let mut app = create_test_app();
app.current_session_id = Some(session.metadata.id.clone());
app.api_messages.clone_from(&session.messages);
let renamed = crate::commands::rename_session_with_manager(
"Renamed In Memory",
&session.metadata.id,
&manager,
&mut app,
);
assert!(!renamed.is_error, "{:?}", renamed.message);
let snapshot = build_session_snapshot(&mut app, &manager).expect("automatic snapshot");
assert_eq!(snapshot.metadata.title, "Renamed In Memory");
}
#[test]
fn session_snapshot_uses_last_known_work_before_first_file_flush() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.todos.try_lock().expect("todos lock").add(
"queued durable state".to_string(),
crate::tools::todo::TodoStatus::InProgress,
);
let initial = build_session_snapshot(&mut app, &manager).expect("initial snapshot");
let expected = initial.work_state.clone();
app.current_session_id = Some(initial.metadata.id);
app.api_messages
.push(text_message("user", "new transcript content"));
let todos = app.todos.clone();
let _held = todos.try_lock().expect("hold todos lock");
let contended = build_session_snapshot(&mut app, &manager).expect("cached snapshot");
assert_eq!(contended.work_state, expected);
assert_eq!(contended.messages.len(), 1);
}
#[test]
fn first_contended_snapshot_fails_instead_of_serializing_false_empty_work() {
let tmp = tempfile::tempdir().expect("tempdir");
let manager =
crate::session_manager::SessionManager::new(tmp.path().join("sessions")).expect("manager");
let mut app = create_test_app();
app.todos.try_lock().expect("todos lock").add(
"not empty".to_string(),
crate::tools::todo::TodoStatus::InProgress,
);
let todos = app.todos.clone();
let _held = todos.try_lock().expect("hold todos lock");
let err = build_session_snapshot(&mut app, &manager).unwrap_err();
assert!(err.contains("skipped"), "{err}");
}
#[test]
fn legacy_session_without_work_state_clears_previous_todo_on_load() {
let mut app = create_test_app();
app.todos.try_lock().expect("todos lock").add(
"old session".to_string(),
crate::tools::todo::TodoStatus::Pending,
);
let session = saved_session_with_messages(vec![text_message("user", "legacy")]);
assert!(session.work_state.is_none());
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert_eq!(app.work_state_snapshot().expect("snapshot"), None);
}
#[test]
fn contended_work_restore_leaves_current_session_wholly_unchanged() {
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "current conversation"));
app.current_session_id = Some("current-session".to_string());
let mut session = saved_session_with_messages(vec![text_message("user", "replacement")]);
session.work_state = Some(crate::session_manager::SessionWorkState {
todos: crate::tools::todo::TodoListSnapshot {
items: vec![crate::tools::todo::TodoItem {
id: 1,
content: "replacement todo".to_string(),
status: crate::tools::todo::TodoStatus::Pending,
}],
completion_pct: 0,
in_progress_id: None,
},
plan: crate::tools::plan::PlanSnapshot::default(),
});
let plan_state = app.plan_state.clone();
let _held = plan_state.try_lock().expect("hold plan lock");
let err = apply_loaded_session(&mut app, &mut Config::default(), &session).unwrap_err();
assert!(err.contains("session was not restored"), "{err}");
assert_eq!(app.api_messages.len(), 1);
assert_eq!(app.current_session_id.as_deref(), Some("current-session"));
}
#[test]
fn missing_named_custom_provider_resume_leaves_current_session_wholly_unchanged() {
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "keep the current conversation"));
app.current_session_id = Some("current-session".to_string());
app.workspace = PathBuf::from("/tmp/current-workspace");
app.set_model_selection("deepseek-v4-pro".to_string());
app.set_provider_identity(ApiProvider::Deepseek, "deepseek");
app.add_message(HistoryCell::System {
content: "existing receipt".to_string(),
});
app.status_message = Some("existing status".to_string());
let mut config = Config {
provider: Some("deepseek".to_string()),
..Config::default()
};
let mut session = saved_session_with_messages(vec![text_message("user", "replacement")]);
session.metadata.model_provider = "lm-studio".to_string();
session.metadata.model = "local-code-model".to_string();
session.metadata.workspace = PathBuf::from("/tmp/other-workspace");
let err = apply_loaded_session_config_snapshot(
&mut app,
&mut config,
&session,
Config::default(),
true,
)
.expect_err("removed custom provider must fail closed");
assert!(err.contains("[providers.lm-studio]"), "{err}");
assert!(err.contains("will not fall back"), "{err}");
assert_eq!(app.current_session_id.as_deref(), Some("current-session"));
assert_eq!(app.api_messages.len(), 1);
assert_eq!(app.workspace, PathBuf::from("/tmp/current-workspace"));
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert_eq!(app.provider_identity_for_persistence(), "deepseek");
assert_eq!(app.model_selection_for_persistence(), "deepseek-v4-pro");
assert_eq!(config.provider.as_deref(), Some("deepseek"));
assert_eq!(app.history.len(), 1);
assert!(matches!(
&app.history[0],
HistoryCell::System { content } if content == "existing receipt"
));
assert_eq!(app.status_message.as_deref(), Some("existing status"));
}
#[test]
fn custom_session_resume_requires_structural_route_not_client_construction() {
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "keep the current conversation"));
app.current_session_id = Some("current-session".to_string());
app.workspace = PathBuf::from("/tmp/current-workspace");
app.set_model_selection("deepseek-v4-pro".to_string());
app.set_provider_identity(ApiProvider::Deepseek, "deepseek");
app.add_message(HistoryCell::System {
content: "existing receipt".to_string(),
});
app.status_message = Some("existing status".to_string());
let mut config =
named_custom_session_config("lm-studio", "http://127.0.0.1:1234/v1", "local-code-model");
config
.providers
.as_mut()
.expect("providers")
.custom
.get_mut("lm-studio")
.expect("lm-studio")
.insecure_skip_tls_verify = Some(true);
let mut session = saved_session_with_messages(vec![text_message("user", "replacement")]);
session.metadata.model_provider = "lm-studio".to_string();
session.metadata.model = "local-code-model".to_string();
session.metadata.workspace = PathBuf::from("/tmp/other-workspace");
let recovered = apply_loaded_session(&mut app, &mut config, &session)
.expect("session restore should not require provider credentials or TLS client setup");
assert!(recovered);
assert_eq!(
app.current_session_id.as_deref(),
Some("resume-recovery-session")
);
assert!(app.api_messages.is_empty());
assert_eq!(app.input, "replacement");
assert_eq!(app.workspace, PathBuf::from("/tmp/other-workspace"));
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "lm-studio");
assert_eq!(app.model_selection_for_persistence(), "local-code-model");
assert!(app.history.is_empty());
assert!(
app.status_message
.as_deref()
.is_some_and(|message| message.contains("Recovered interrupted prompt"))
);
assert_eq!(config.provider.as_deref(), Some("lm-studio"));
}
#[test]
fn named_custom_provider_resume_uses_exact_live_endpoint_model_and_workspace() {
let resumed_workspace = tempfile::tempdir().expect("resume workspace");
let mut config = named_custom_session_config(
"lm-studio",
"http://127.0.0.1:1234/v1",
"configured-default",
);
let mut app = create_test_app();
let mut session = saved_session_with_messages(vec![
text_message("user", "resume local work"),
text_message("assistant", "ready"),
]);
session.metadata.model_provider = "lm-studio".to_string();
session.metadata.model = "local-code-model".to_string();
session.metadata.workspace = resumed_workspace.path().to_path_buf();
apply_loaded_session(&mut app, &mut config, &session).expect("restore exact custom route");
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "lm-studio");
assert_eq!(app.model_selection_for_persistence(), "local-code-model");
assert_eq!(app.workspace, resumed_workspace.path());
assert_eq!(config.provider.as_deref(), Some("lm-studio"));
let route = resolve_runtime_route(&config, app.api_provider, Some(&app.model))
.expect("runtime route remains exact");
assert_eq!(
route.candidate.endpoint.base_url,
"http://127.0.0.1:1234/v1"
);
assert_eq!(route.model, "local-code-model");
}
#[test]
fn same_workspace_named_custom_switch_requires_engine_respawn() {
let mut config =
named_custom_session_config("custom-a", "http://127.0.0.1:18181/v1", "model-a");
config.providers.as_mut().expect("providers").custom.insert(
"custom-b".to_string(),
crate::config::ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some("http://127.0.0.1:18182/v1".to_string()),
model: Some("model-b".to_string()),
..Default::default()
},
);
let workspace = PathBuf::from("/tmp/same-workspace-custom-switch");
let mut app = create_test_app();
app.workspace.clone_from(&workspace);
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
let previous_provider = app.api_provider;
let previous_identity = app.provider_identity_for_persistence().to_string();
let previous_workspace = app.workspace.clone();
let mut session = saved_session_with_messages(vec![text_message("user", "switch route")]);
session.metadata.model_provider = "custom-b".to_string();
session.metadata.model = "model-b".to_string();
session.metadata.workspace = workspace;
apply_loaded_session(&mut app, &mut config, &session).expect("restore custom B");
assert_eq!(previous_provider, ApiProvider::Custom);
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "custom-b");
assert!(loaded_session_requires_engine_respawn(
&app,
previous_provider,
&previous_identity,
&previous_workspace,
));
}
#[test]
fn file_load_uses_one_fresh_config_snapshot_for_custom_route_and_app_state() {
let workspace = PathBuf::from("/tmp/atomic-file-load");
let mut stale_config =
named_custom_session_config("custom-a", "http://127.0.0.1:18181/v1", "model-a");
let fresh_config =
named_custom_session_config("custom-b", "http://127.0.0.1:18182/v1", "model-b");
let mut app = create_test_app();
app.workspace.clone_from(&workspace);
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
app.api_messages
.push(text_message("user", "old conversation"));
let mut session = saved_session_with_messages(vec![
text_message("user", "new conversation"),
text_message("assistant", "loaded reply"),
]);
session.metadata.model_provider = "custom-b".to_string();
session.metadata.model = "model-b".to_string();
session.metadata.workspace = workspace;
let (recovered, respawn) = apply_loaded_session_config_snapshot(
&mut app,
&mut stale_config,
&session,
fresh_config,
true,
)
.expect("fresh disk config should atomically restore custom B");
assert!(!recovered);
assert!(respawn);
assert_eq!(app.provider_identity_for_persistence(), "custom-b");
assert_eq!(app.model_selection_for_persistence(), "model-b");
assert_eq!(app.api_messages, session.messages);
assert_eq!(stale_config.provider.as_deref(), Some("custom-b"));
assert_eq!(
stale_config.deepseek_base_url(),
"http://127.0.0.1:18182/v1"
);
}
#[test]
fn session_load_keeps_idless_custom_record_on_root_when_table_coexists() {
let mut config =
named_custom_session_config("custom", "http://127.0.0.1:18182/v1", "table-model");
config.base_url = Some("http://127.0.0.1:18181/v1".to_string());
config.default_text_model = Some("legacy-root-model".to_string());
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "current conversation"));
let mut session = saved_session_with_messages(vec![
text_message("user", "legacy custom conversation"),
text_message("assistant", "legacy reply"),
]);
session.metadata.model_provider = "custom".to_string();
session.metadata.model_provider_id = None;
session.metadata.model = "legacy-saved-model".to_string();
let recovered = apply_loaded_session(&mut app, &mut config, &session)
.expect("id-less custom record must retain root provenance");
assert!(!recovered);
assert_eq!(app.api_messages, session.messages);
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "custom");
assert_eq!(app.provider_id_for_persistence(), None);
assert_eq!(config.deepseek_base_url(), "http://127.0.0.1:18181/v1");
assert!(
config
.providers
.as_ref()
.is_none_or(|providers| !providers.custom.contains_key("custom"))
);
}
#[test]
fn session_load_rejects_exact_custom_table_record_when_only_root_remains() {
let mut config = Config {
provider: Some("custom".to_string()),
base_url: Some("http://127.0.0.1:18181/v1".to_string()),
default_text_model: Some("legacy-root-model".to_string()),
..Config::default()
};
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "current conversation"));
let previous_messages = app.api_messages.clone();
let previous_identity = app.provider_identity_for_persistence().to_string();
let mut session = saved_session_with_messages(vec![
text_message("user", "exact table conversation"),
text_message("assistant", "exact table reply"),
]);
session.metadata.model_provider = "custom".to_string();
session.metadata.model_provider_id = Some("custom".to_string());
session.metadata.model = "table-model".to_string();
let error = apply_loaded_session(&mut app, &mut config, &session)
.expect_err("exact table record must not fall back to root");
assert!(error.contains("[providers.custom]"), "{error}");
assert!(error.contains("will not fall back"), "{error}");
assert_eq!(app.api_messages, previous_messages);
assert_eq!(app.provider_identity_for_persistence(), previous_identity);
}
#[test]
fn session_load_rejects_empty_custom_id_when_root_and_table_coexist() {
let mut config =
named_custom_session_config("custom", "http://127.0.0.1:18182/v1", "table-model");
config.base_url = Some("http://127.0.0.1:18181/v1".to_string());
config.default_text_model = Some("legacy-root-model".to_string());
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "current conversation"));
app.set_provider_identity(ApiProvider::Deepseek, "deepseek");
app.set_model_selection("deepseek-v4-pro".to_string());
let previous_messages = app.api_messages.clone();
let previous_provider = app.api_provider;
let previous_identity = app.provider_identity_for_persistence().to_string();
let previous_provider_id = app.provider_id_for_persistence().map(str::to_string);
let previous_model = app.model.clone();
let previous_config_provider = config.provider.clone();
let mut session = saved_session_with_messages(vec![
text_message("user", "malformed custom conversation"),
text_message("assistant", "must never load"),
]);
session.metadata.model_provider = "custom".to_string();
session.metadata.model_provider_id = Some(" ".to_string());
session.metadata.model = "legacy-root-model".to_string();
let error = apply_loaded_session(&mut app, &mut config, &session)
.expect_err("an explicit empty id must not load either custom route");
assert!(error.contains("empty exact provider id"), "{error}");
assert_eq!(app.api_messages, previous_messages);
assert_eq!(app.api_provider, previous_provider);
assert_eq!(app.provider_identity_for_persistence(), previous_identity);
assert_eq!(
app.provider_id_for_persistence().map(str::to_string),
previous_provider_id
);
assert_eq!(app.model, previous_model);
assert_eq!(config.provider, previous_config_provider);
assert_eq!(config.deepseek_base_url(), "http://127.0.0.1:18182/v1");
}
#[test]
fn file_load_respawns_engine_when_same_custom_identity_changes_endpoint() {
let workspace = PathBuf::from("/tmp/atomic-file-load-same-custom");
let mut stale_config =
named_custom_session_config("custom-a", "http://127.0.0.1:18181/v1", "model-a");
let mut fresh_config =
named_custom_session_config("custom-a", "http://127.0.0.1:18199/v1", "model-a");
let fresh_entry = fresh_config
.providers
.as_mut()
.expect("providers")
.custom
.get_mut("custom-a")
.expect("custom A");
fresh_entry.api_key = Some("rotated-key".to_string());
fresh_entry.http_headers = Some(std::collections::HashMap::from([(
"X-Route-Version".to_string(),
"fresh".to_string(),
)]));
let mut app = create_test_app();
app.workspace.clone_from(&workspace);
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
let mut session = saved_session_with_messages(vec![text_message("user", "new endpoint")]);
session.metadata.model_provider = "custom-a".to_string();
session.metadata.model = "model-a".to_string();
session.metadata.workspace = workspace;
let (_, respawn) = apply_loaded_session_config_snapshot(
&mut app,
&mut stale_config,
&session,
fresh_config,
true,
)
.expect("same-key fresh config should load atomically");
assert!(respawn, "file loads must install the fresh engine config");
assert_eq!(app.provider_identity_for_persistence(), "custom-a");
assert_eq!(
stale_config.deepseek_base_url(),
"http://127.0.0.1:18199/v1"
);
let entry = stale_config
.provider_config_for(ApiProvider::Custom)
.expect("fresh custom route");
assert_eq!(entry.api_key.as_deref(), Some("rotated-key"));
assert_eq!(
entry
.http_headers
.as_ref()
.and_then(|headers| headers.get("X-Route-Version"))
.map(String::as_str),
Some("fresh")
);
}
#[test]
fn file_load_route_refresh_preserves_effective_permission_and_feature_overlays() {
let workspace = PathBuf::from("/tmp/atomic-file-load-policy");
let mut effective_config =
named_custom_session_config("custom-a", "http://127.0.0.1:18181/v1", "model-a");
effective_config.approval_policy = Some("never".to_string());
effective_config.sandbox_mode = Some("read-only".to_string());
effective_config.allow_shell = Some(false);
effective_config.max_subagents = Some(1);
effective_config.features = Some(crate::features::FeaturesToml {
entries: std::collections::BTreeMap::from([
("shell_tool".to_string(), false),
("subagents".to_string(), false),
]),
});
let mut raw_disk_config =
named_custom_session_config("custom-a", "http://127.0.0.1:18199/v1", "model-a");
raw_disk_config.approval_policy = Some("always".to_string());
raw_disk_config.sandbox_mode = Some("danger-full-access".to_string());
raw_disk_config.allow_shell = Some(true);
raw_disk_config.max_subagents = Some(64);
raw_disk_config.features = Some(crate::features::FeaturesToml {
entries: std::collections::BTreeMap::from([
("shell_tool".to_string(), true),
("subagents".to_string(), true),
]),
});
let mut app = create_test_app();
app.workspace.clone_from(&workspace);
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
let mut session = saved_session_with_messages(vec![text_message("user", "keep policy")]);
session.metadata.model_provider = "custom-a".to_string();
session.metadata.model = "model-a".to_string();
session.metadata.workspace = workspace;
let (_, respawn) = apply_loaded_session_config_snapshot(
&mut app,
&mut effective_config,
&session,
raw_disk_config,
true,
)
.expect("fresh route plus effective policy should load");
assert!(respawn);
assert_eq!(
effective_config.deepseek_base_url(),
"http://127.0.0.1:18199/v1"
);
assert_eq!(effective_config.approval_policy.as_deref(), Some("never"));
assert_eq!(effective_config.sandbox_mode.as_deref(), Some("read-only"));
assert_eq!(effective_config.allow_shell, Some(false));
assert_eq!(effective_config.max_subagents, Some(1));
let features = effective_config
.features
.expect("effective feature overlay");
assert_eq!(features.entries.get("shell_tool"), Some(&false));
assert_eq!(features.entries.get("subagents"), Some(&false));
}
#[test]
fn session_picker_restore_rejects_active_turn_before_mutating() {
let mut app = create_test_app();
app.api_messages
.push(text_message("user", "current active conversation"));
app.current_session_id = Some("current-session".to_string());
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
let session = saved_session_with_messages(vec![text_message("user", "replacement")]);
let err = apply_loaded_session(&mut app, &mut Config::default(), &session).unwrap_err();
assert!(err.contains("runtime work is active"), "{err}");
assert_eq!(app.api_messages.len(), 1);
assert_eq!(app.current_session_id.as_deref(), Some("current-session"));
}
#[test]
fn session_restore_rebuilds_fresh_codex_route_limits() {
let _lock = crate::test_support::lock_test_env();
let codex_home = tempfile::tempdir().expect("Codex home");
let _codex_home = crate::test_support::EnvVarGuard::set("CODEX_HOME", codex_home.path());
std::fs::write(
codex_home.path().join("models_cache.json"),
serde_json::to_vec(&serde_json::json!({
"fetched_at": chrono::Utc::now(),
"models": [{
"slug": crate::config::DEFAULT_OPENAI_CODEX_MODEL,
"priority": 1,
"context_window": 272000,
"supported_reasoning_levels": [{"effort": "high"}]
}]
}))
.expect("serialize cache"),
)
.expect("write cache");
let mut app = create_test_app();
let mut config = Config::default();
let mut session = saved_session_with_messages(vec![text_message("user", "resume Codex")]);
session.metadata.model_provider = ApiProvider::OpenaiCodex.as_str().to_string();
session.metadata.model = crate::config::DEFAULT_OPENAI_CODEX_MODEL.to_string();
apply_loaded_session(&mut app, &mut config, &session).expect("restore session");
assert_eq!(app.api_provider, ApiProvider::OpenaiCodex);
assert_eq!(app.model, crate::config::DEFAULT_OPENAI_CODEX_MODEL);
assert_eq!(
app.active_route_limits
.and_then(|limits| limits.context_tokens),
Some(272_000)
);
let engine_config = build_engine_config(&app, &config);
assert_eq!(
engine_config
.active_route_limits
.and_then(|limits| limits.context_tokens),
Some(272_000)
);
}
#[test]
fn apply_loaded_session_restores_concrete_model_mode() {
let mut app = create_test_app();
app.set_model_selection("auto".to_string());
let mut session = saved_session_with_messages(vec![
text_message("user", "hello"),
text_message("assistant", "hi"),
]);
session.metadata.model = "deepseek-v4-flash".to_string();
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(!recovered);
assert!(!app.auto_model);
assert_eq!(app.model, "deepseek-v4-flash");
assert_eq!(app.model_selection_for_persistence(), "deepseek-v4-flash");
}
#[test]
fn apply_loaded_session_restores_auto_model_mode() {
let mut app = create_test_app();
app.set_model_selection("deepseek-v4-pro".to_string());
app.reasoning_effort = ReasoningEffort::High;
app.last_effective_model = Some("deepseek-v4-flash".to_string());
app.last_effective_reasoning_effort = Some(ReasoningEffort::Low);
let mut session = saved_session_with_messages(vec![
text_message("user", "hello"),
text_message("assistant", "hi"),
]);
session.metadata.model = "auto".to_string();
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(!recovered);
assert!(app.auto_model);
assert_eq!(app.model, "auto");
assert_eq!(app.model_selection_for_persistence(), "auto");
assert_eq!(app.last_effective_model, None);
assert_eq!(app.last_effective_reasoning_effort, None);
assert_eq!(app.reasoning_effort, ReasoningEffort::Auto);
assert_eq!(app.effective_model_for_budget(), DEFAULT_TEXT_MODEL);
}
#[test]
fn apply_loaded_session_restores_saved_mode() {
let mut app = create_test_app();
app.set_mode(crate::tui::app::AppMode::Agent);
let mut session = saved_session_with_messages(vec![
text_message("user", "draft a plan"),
text_message("assistant", "plan response"),
]);
session.metadata.mode = Some("plan".to_string());
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(!recovered);
assert_eq!(app.mode, crate::tui::app::AppMode::Plan);
assert!(!app.allow_shell);
assert!(!app.trust_mode);
}
#[test]
fn app_new_restores_saved_model_and_reasoning_effort() {
let _guard = ConfigPathEnvGuard::new();
let settings = crate::settings::Settings {
default_model: Some("deepseek-v4-pro".to_string()),
reasoning_effort: Some("high".to_string()),
..Default::default()
};
settings.save().expect("save settings");
let options = TuiOptions {
model: "auto".to_string(),
workspace: PathBuf::from("."),
config_path: None,
config_profile: None,
allow_shell: false,
use_alt_screen: true,
use_mouse_capture: false,
use_bracketed_paste: true,
max_subagents: 1,
skills_dir: PathBuf::from("."),
memory_path: PathBuf::from("memory.md"),
notes_path: PathBuf::from("notes.txt"),
mcp_config_path: PathBuf::from("mcp.json"),
use_memory: false,
start_in_agent_mode: true,
skip_onboarding: false,
yolo: false,
resume_session_id: None,
initial_input: None,
};
let config = Config {
reasoning_effort: Some("max".to_string()),
..Default::default()
};
let app = App::new(options, &config);
assert!(!app.auto_model);
assert_eq!(app.model, "deepseek-v4-pro");
assert_eq!(app.reasoning_effort, ReasoningEffort::High);
}
#[tokio::test]
async fn model_picker_persists_model_and_reasoning_effort() {
let _guard = SettingsHomeGuard::new();
let mut app = create_test_app();
app.set_model_selection("auto".to_string());
app.reasoning_effort = ReasoningEffort::Auto;
let mut engine = mock_engine_handle();
let mut config = Config {
api_key: Some("test-key".to_string()),
..Default::default()
};
apply_model_picker_choice(
&mut app,
&mut engine.handle,
&mut config,
"deepseek-v4-pro".to_string(),
None,
None,
ReasoningEffort::High,
"auto".to_string(),
ReasoningEffort::Auto,
)
.await;
let settings = crate::settings::Settings::load().expect("load settings");
assert_eq!(settings.default_model.as_deref(), Some("deepseek-v4-pro"));
assert_eq!(
settings
.provider_models
.as_ref()
.and_then(|models| models.get("deepseek"))
.map(String::as_str),
Some("deepseek-v4-pro")
);
assert_eq!(settings.reasoning_effort.as_deref(), Some("high"));
assert!(!app.auto_model);
assert_eq!(app.reasoning_effort, ReasoningEffort::High);
let state = codewhale_config::SetupState::load()
.expect("load setup state")
.expect("setup state");
assert_eq!(
state.status(codewhale_config::SetupStep::ProviderModel),
codewhale_config::StepStatus::Verified
);
let provider_model_result = state
.steps
.get(&codewhale_config::SetupStep::ProviderModel)
.and_then(|entry| entry.result.as_deref())
.expect("provider/model result");
assert!(provider_model_result.contains("provider=deepseek"));
assert!(provider_model_result.contains("model=deepseek-v4-pro"));
assert!(provider_model_result.contains("auth=key saved · not checked"));
assert!(provider_model_result.contains("health=attemptable"));
assert!(!provider_model_result.contains("test-key"));
}
#[tokio::test]
async fn model_picker_switches_between_exact_named_custom_routes() {
let _guard = SettingsHomeGuard::new();
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
let previous_effort = app.reasoning_effort;
let mut custom = HashMap::new();
for (name, base_url, model) in [
("custom-a", "http://127.0.0.1:18181/v1", "model-a"),
("custom-b", "http://127.0.0.1:18182/v1", "model-b"),
] {
custom.insert(
name.to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some(base_url.to_string()),
model: Some(model.to_string()),
api_key: Some("local-test-key".to_string()),
..Default::default()
},
);
}
let mut config = Config {
provider: Some("custom-b".to_string()),
providers: Some(ProvidersConfig {
custom,
..Default::default()
}),
..Default::default()
};
let mut engine = mock_engine_handle();
apply_model_picker_choice(
&mut app,
&mut engine.handle,
&mut config,
"model-b".to_string(),
None,
Some("custom-b".to_string()),
ReasoningEffort::High,
"model-a".to_string(),
previous_effort,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "custom-b");
assert_eq!(app.model_selection_for_persistence(), "model-b");
assert_eq!(config.provider.as_deref(), Some("custom-b"));
assert_eq!(config.deepseek_base_url(), "http://127.0.0.1:18182/v1");
}
#[tokio::test]
async fn model_picker_auto_switches_exact_named_custom_route_transactionally() {
let _guard = SettingsHomeGuard::new();
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "custom-a");
app.set_model_selection("model-a".to_string());
let previous_effort = app.reasoning_effort;
let mut custom = HashMap::new();
for (name, base_url, model) in [
("custom-a", "http://127.0.0.1:18181/v1", "model-a"),
("custom-b", "http://127.0.0.1:18182/v1", "model-b"),
] {
custom.insert(
name.to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some(base_url.to_string()),
model: Some(model.to_string()),
api_key: Some("local-test-key".to_string()),
..Default::default()
},
);
}
let mut config = Config {
provider: Some("custom-b".to_string()),
providers: Some(ProvidersConfig {
custom,
..Default::default()
}),
..Default::default()
};
let mut engine = mock_engine_handle();
apply_model_picker_choice(
&mut app,
&mut engine.handle,
&mut config,
"auto".to_string(),
None,
Some("custom-b".to_string()),
ReasoningEffort::Auto,
"model-a".to_string(),
previous_effort,
)
.await;
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "custom-b");
assert!(app.auto_model);
assert_eq!(app.model_selection_for_persistence(), "auto");
assert_eq!(config.provider.as_deref(), Some("custom-b"));
assert_eq!(config.deepseek_base_url(), "http://127.0.0.1:18182/v1");
}
#[test]
fn dismissing_named_custom_model_picker_restores_app_owned_config_route() {
let mut app = create_test_app();
app.set_provider_identity(ApiProvider::Custom, "custom-a");
let mut config =
named_custom_session_config("custom-a", "http://127.0.0.1:18181/v1", "model-a");
config.providers.as_mut().expect("providers").custom.insert(
"custom-b".to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
base_url: Some("http://127.0.0.1:18182/v1".to_string()),
model: Some("model-b".to_string()),
..Default::default()
},
);
config.provider = Some("custom-b".to_string());
sync_config_provider_from_app(&mut config, &app);
assert_eq!(config.provider.as_deref(), Some("custom-a"));
}
#[tokio::test]
#[allow(clippy::await_holding_lock)]
async fn model_picker_skips_setup_receipt_when_settings_persistence_fails() {
let _lock = crate::test_support::lock_test_env();
let tmp = TempDir::new().expect("settings tempdir");
let bad_home = tmp.path().join("codewhale-home-file");
std::fs::write(&bad_home, "not a directory").expect("bad home file");
let _home = crate::test_support::EnvVarGuard::set("HOME", tmp.path().as_os_str());
let _userprofile = crate::test_support::EnvVarGuard::set("USERPROFILE", tmp.path().as_os_str());
let _codewhale_home =
crate::test_support::EnvVarGuard::set("CODEWHALE_HOME", bad_home.as_os_str());
let _deepseek_config_path = crate::test_support::EnvVarGuard::remove("DEEPSEEK_CONFIG_PATH");
let _codewhale_config_path = crate::test_support::EnvVarGuard::remove("CODEWHALE_CONFIG_PATH");
let _codewhale_provider = crate::test_support::EnvVarGuard::remove("CODEWHALE_PROVIDER");
let _deepseek_provider = crate::test_support::EnvVarGuard::remove("DEEPSEEK_PROVIDER");
let mut app = create_test_app();
app.set_model_selection("auto".to_string());
app.reasoning_effort = ReasoningEffort::Auto;
let mut engine = mock_engine_handle();
let mut config = Config {
api_key: Some("test-key".to_string()),
..Default::default()
};
apply_model_picker_choice(
&mut app,
&mut engine.handle,
&mut config,
"deepseek-v4-pro".to_string(),
None,
None,
ReasoningEffort::High,
"auto".to_string(),
ReasoningEffort::Auto,
)
.await;
assert!(
app.status_message
.as_deref()
.is_some_and(|message| message.contains("not persisted"))
);
assert!(
codewhale_config::SetupState::load()
.expect("load setup state")
.is_none(),
"failed model persistence must not create a ProviderModel setup receipt"
);
}
#[test]
fn apply_loaded_session_restores_artifact_registry() {
let mut app = create_test_app();
let mut session = saved_session_with_messages(vec![
text_message("user", "hello"),
text_message("assistant", "hi"),
]);
session.artifacts.push(crate::artifacts::ArtifactRecord {
id: "art_call_big".to_string(),
kind: crate::artifacts::ArtifactKind::ToolOutput,
session_id: "session-123".to_string(),
tool_call_id: "call-big".to_string(),
tool_name: "exec_shell".to_string(),
created_at: chrono::Utc::now(),
byte_size: 128,
preview: "hello".to_string(),
storage_path: PathBuf::from("/tmp/tool_outputs/call-big.txt"),
});
let recovered =
apply_loaded_session(&mut app, &mut Config::default(), &session).expect("restore session");
assert!(!recovered);
assert_eq!(app.session_artifacts, session.artifacts);
}
#[test]
fn parallel_exploring_tool_starts_share_one_active_entry() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"t-a",
"read_file",
&serde_json::json!({"path": "alpha.rs"}),
);
handle_tool_call_started(
&mut app,
"t-b",
"read_file",
&serde_json::json!({"path": "beta.rs"}),
);
handle_tool_call_started(
&mut app,
"t-c",
"grep_files",
&serde_json::json!({"pattern": "TODO"}),
);
assert_eq!(app.history.len(), 0, "no history cells written mid-turn");
let active = app.active_cell.as_ref().expect("active cell created");
assert_eq!(
active.entry_count(),
1,
"all exploring starts share one entry"
);
let HistoryCell::Tool(ToolCell::Exploring(explore)) = &active.entries()[0] else {
panic!("expected exploring cell")
};
assert_eq!(explore.entries.len(), 3);
for entry in &explore.entries {
assert_eq!(entry.status, ToolStatus::Running);
}
}
#[test]
fn out_of_order_completes_finalize_one_history_cell_per_turn() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"t-1",
"read_file",
&serde_json::json!({"path": "a.rs"}),
);
handle_tool_call_started(
&mut app,
"t-2",
"read_file",
&serde_json::json!({"path": "b.rs"}),
);
handle_tool_call_started(
&mut app,
"t-3",
"grep_files",
&serde_json::json!({"pattern": "x"}),
);
handle_tool_call_complete(&mut app, "t-3", "grep_files", &ok_result("two hits"));
handle_tool_call_complete(&mut app, "t-1", "read_file", &ok_result("contents A"));
handle_tool_call_complete(&mut app, "t-2", "read_file", &ok_result("contents B"));
assert_eq!(app.history.len(), 0);
let active = app.active_cell.as_ref().expect("active cell still present");
let HistoryCell::Tool(ToolCell::Exploring(explore)) = &active.entries()[0] else {
panic!("expected exploring cell")
};
assert!(
explore
.entries
.iter()
.all(|e| e.status == ToolStatus::Success),
"all exploring entries should be Success after their tools complete"
);
app.flush_active_cell();
assert!(app.active_cell.is_none(), "active cell cleared after flush");
let tool_cells = app
.history
.iter()
.filter(|c| matches!(c, HistoryCell::Tool(_)))
.count();
assert_eq!(
tool_cells, 1,
"exactly one tool history cell after parallel turn"
);
}
#[test]
fn mixed_parallel_tools_render_in_single_active_cell() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"ex-1",
"read_file",
&serde_json::json!({"path": "x.rs"}),
);
handle_tool_call_started(
&mut app,
"shell-1",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
handle_tool_call_started(
&mut app,
"gen-1",
"todo_write",
&serde_json::json!({"items": []}),
);
assert_eq!(app.history.len(), 0);
let active = app.active_cell.as_ref().expect("active cell present");
assert_eq!(active.entry_count(), 3);
handle_tool_call_complete(&mut app, "shell-1", "exec_shell", &ok_result("ok"));
handle_tool_call_complete(&mut app, "gen-1", "todo_write", &ok_result("done"));
handle_tool_call_complete(&mut app, "ex-1", "read_file", &ok_result("file body"));
assert_eq!(app.history.len(), 0);
app.flush_active_cell();
let tool_cells: Vec<_> = app
.history
.iter()
.filter(|c| matches!(c, HistoryCell::Tool(_)))
.collect();
assert_eq!(
tool_cells.len(),
3,
"three distinct tool shapes finalize as three cells in stable insertion order"
);
}
#[test]
fn orphan_tool_complete_with_unknown_id_pushes_separate_cell() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"live-1",
"read_file",
&serde_json::json!({"path": "live.rs"}),
);
handle_tool_call_complete(&mut app, "ghost-id", "mystery_tool", &ok_result("oops"));
let active = app
.active_cell
.as_ref()
.expect("active cell preserved after orphan");
assert_eq!(active.entry_count(), 1);
assert_eq!(app.history.len(), 1, "orphan added one finalized cell");
let HistoryCell::Tool(ToolCell::Generic(generic)) = &app.history[0] else {
panic!("orphan should render as a Generic tool cell")
};
assert_eq!(generic.name, "mystery_tool");
assert_eq!(generic.status, ToolStatus::Success);
}
#[test]
fn turn_complete_flushes_active_cell_into_history() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"ex-1",
"read_file",
&serde_json::json!({"path": "a.rs"}),
);
handle_tool_call_complete(&mut app, "ex-1", "read_file", &ok_result("body"));
handle_tool_call_started(
&mut app,
"shell-1",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
app.finalize_active_cell_as_interrupted();
assert!(app.active_cell.is_none(), "active cell cleared on flush");
let exec_cells: Vec<_> = app
.history
.iter()
.filter_map(|c| match c {
HistoryCell::Tool(ToolCell::Exec(exec)) => Some(exec),
_ => None,
})
.collect();
assert_eq!(exec_cells.len(), 1);
assert_eq!(
exec_cells[0].status,
ToolStatus::Failed,
"interrupted shell entry marked Failed (closest available terminal status)"
);
}
#[test]
fn orphan_during_active_keeps_subsequent_completion_routed_correctly() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"live",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
handle_tool_call_complete(&mut app, "ghost", "weird_tool", &ok_result("ghost-out"));
handle_tool_call_complete(&mut app, "live", "exec_shell", &ok_result("hello"));
let active = app.active_cell.as_ref().expect("active cell present");
let HistoryCell::Tool(ToolCell::Exec(exec)) = &active.entries()[0] else {
panic!("expected exec cell")
};
assert_eq!(exec.status, ToolStatus::Success);
assert_eq!(app.history.len(), 1);
let HistoryCell::Tool(ToolCell::Generic(generic)) = &app.history[0] else {
panic!("expected orphan generic cell")
};
assert_eq!(generic.name, "weird_tool");
app.flush_active_cell();
assert_eq!(app.history.len(), 2);
}
#[test]
fn tool_details_survive_active_cell_flush() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"tid",
"exec_shell",
&serde_json::json!({"command": "echo hi"}),
);
handle_tool_call_complete(&mut app, "tid", "exec_shell", &ok_result("hi"));
app.flush_active_cell();
assert_eq!(app.history.len(), 1);
let detail = app
.tool_details_by_cell
.get(&0)
.expect("detail record migrated to flushed cell index");
assert_eq!(detail.tool_id, "tid");
assert_eq!(detail.tool_name, "exec_shell");
}
#[test]
fn exploring_label_uses_progressive_for_read_file() {
let label = exploring_label("read_file", &serde_json::json!({"path": "src/foo.rs"}));
assert_eq!(label, "Reading src/foo.rs");
}
#[test]
fn exploring_label_uses_progressive_for_list_dir() {
let label = exploring_label("list_dir", &serde_json::json!({"path": "crates/tui/src/"}));
assert_eq!(label, "Listing crates/tui/src/");
}
#[test]
fn exploring_label_uses_progressive_for_list_dir_no_path() {
let label = exploring_label("list_dir", &serde_json::json!({}));
assert_eq!(label, "Listing directory");
}
#[test]
fn exploring_label_for_grep_quotes_pattern_with_searching_for() {
let label = exploring_label(
"grep_files",
&serde_json::json!({"pattern": "TranscriptScroll"}),
);
assert_eq!(label, "Searching for `TranscriptScroll`");
}
#[test]
fn exploring_label_for_list_files_uses_progressive() {
let label = exploring_label("list_files", &serde_json::json!({}));
assert_eq!(label, "Listing files");
}
#[test]
fn add_message_does_not_scroll_when_user_scrolled_away() {
use crate::tui::scrolling::TranscriptScroll;
let mut app = create_test_app();
app.viewport.transcript_scroll = TranscriptScroll::at_line(7);
app.user_scrolled_during_stream = true;
app.add_message(HistoryCell::User {
content: "fresh user message".to_string(),
});
assert!(
!app.viewport.transcript_scroll.is_at_tail(),
"add_message must respect user_scrolled_during_stream",
);
}
#[test]
fn add_message_pins_to_tail_when_user_was_following() {
use crate::tui::scrolling::TranscriptScroll;
let mut app = create_test_app();
app.viewport.transcript_scroll = TranscriptScroll::to_bottom();
app.user_scrolled_during_stream = false;
app.add_message(HistoryCell::User {
content: "fresh user message".to_string(),
});
assert!(
app.viewport.transcript_scroll.is_at_tail(),
"auto-pin should still work when the user hasn't opted out",
);
}
#[test]
fn tool_call_started_does_not_scroll_when_user_scrolled_away() {
use crate::tui::scrolling::TranscriptScroll;
let mut app = create_test_app();
app.viewport.transcript_scroll = TranscriptScroll::at_line(7);
app.user_scrolled_during_stream = true;
handle_tool_call_started(
&mut app,
"tid",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
assert!(
!app.viewport.transcript_scroll.is_at_tail(),
"tool-cell start must not yank scroll position to bottom",
);
}
#[test]
fn tool_call_complete_does_not_scroll_when_user_scrolled_away() {
use crate::tui::scrolling::TranscriptScroll;
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"tid",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
app.viewport.transcript_scroll = TranscriptScroll::at_line(7);
app.user_scrolled_during_stream = true;
handle_tool_call_complete(&mut app, "tid", "exec_shell", &ok_result("output"));
assert!(
!app.viewport.transcript_scroll.is_at_tail(),
"tool-cell complete must not yank scroll position to bottom",
);
}
#[test]
fn mark_history_updated_does_not_call_scroll_to_bottom() {
use crate::tui::scrolling::TranscriptScroll;
let mut app = create_test_app();
app.viewport.transcript_scroll = TranscriptScroll::at_line(3);
app.user_scrolled_during_stream = true;
app.mark_history_updated();
assert!(
!app.viewport.transcript_scroll.is_at_tail(),
"mark_history_updated must not scroll",
);
}
#[test]
fn thinking_then_tools_share_active_cell_until_text_flushes() {
let mut app = create_test_app();
let thinking_idx = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
crate::tui::streaming_thinking::append(&mut app, thinking_idx, "planning the read");
assert!(
app.history.is_empty(),
"thinking must not write into history mid-turn"
);
assert_eq!(thinking_idx, 0);
handle_tool_call_started(
&mut app,
"t-1",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
handle_tool_call_started(
&mut app,
"t-2",
"exec_shell",
&serde_json::json!({"command": "pwd"}),
);
let active = app
.active_cell
.as_ref()
.expect("active cell present mid-turn");
assert_eq!(
active.entry_count(),
3,
"thinking + two exec entries share one active cell"
);
assert!(matches!(active.entries()[0], HistoryCell::Thinking { .. }));
assert!(matches!(
active.entries()[1],
HistoryCell::Tool(ToolCell::Exec(_))
));
assert!(matches!(
active.entries()[2],
HistoryCell::Tool(ToolCell::Exec(_))
));
let finalized = crate::tui::streaming_thinking::finalize_active_entry(&mut app, Some(1.5), "");
assert!(finalized, "finalizer reports it touched the active cell");
let HistoryCell::Thinking {
streaming,
duration_secs,
content,
..
} = &app
.active_cell
.as_ref()
.expect("active cell still present after thinking complete")
.entries()[0]
else {
panic!("expected thinking entry")
};
assert!(!streaming, "thinking spinner stops after finalize");
assert_eq!(*duration_secs, Some(1.5));
assert_eq!(content, "planning the read");
assert!(
app.streaming_thinking_active_entry.is_none(),
"stream pointer cleared after finalize"
);
app.flush_active_cell();
assert!(app.active_cell.is_none(), "active cell cleared after flush");
assert_eq!(
app.history.len(),
3,
"thinking + both tool entries land in history together"
);
assert!(matches!(app.history[0], HistoryCell::Thinking { .. }));
assert!(matches!(
app.history[1],
HistoryCell::Tool(ToolCell::Exec(_))
));
assert!(matches!(
app.history[2],
HistoryCell::Tool(ToolCell::Exec(_))
));
}
#[test]
fn flush_active_cell_finalizes_unclosed_thinking_block() {
let mut app = create_test_app();
let _ = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
crate::tui::streaming_thinking::append(&mut app, 0, "incomplete");
app.flush_active_cell();
assert_eq!(app.history.len(), 1);
let HistoryCell::Thinking { streaming, .. } = &app.history[0] else {
panic!("expected thinking history cell")
};
assert!(
!*streaming,
"flush must stop the spinner even without ThinkingComplete"
);
assert!(
app.streaming_thinking_active_entry.is_none(),
"stream pointer cleared by flush"
);
}
#[test]
fn open_thinking_pager_finds_thinking_in_active_cell() {
let mut app = create_test_app();
let _ = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
crate::tui::streaming_thinking::append(&mut app, 0, "deliberating");
let finalized = crate::tui::streaming_thinking::finalize_active_entry(&mut app, Some(1.2), "");
assert!(finalized);
assert!(
app.history.is_empty(),
"thinking entry stays in active_cell until flush"
);
let active = app.active_cell.as_ref().expect("active cell present");
assert!(matches!(
active.entries().first(),
Some(HistoryCell::Thinking {
streaming: false,
..
})
));
assert!(open_thinking_pager(&mut app));
assert_eq!(
app.view_stack.top_kind(),
Some(ModalKind::Pager),
"pager must open for thinking entries still in active_cell"
);
let body = pop_pager_body(&mut app);
assert!(body.contains("Activity: reasoning timeline"), "{body}");
assert!(body.contains("Thinking chunk 1 of 1"), "{body}");
assert!(body.contains("deliberating"), "{body}");
}
#[test]
fn activity_detail_opens_reasoning_timeline_for_selected_thinking() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::Thinking {
content: "first chunk reasoning".to_string(),
streaming: false,
duration_secs: Some(0.8),
},
HistoryCell::Assistant {
content: "interlude".to_string(),
streaming: false,
},
HistoryCell::Thinking {
content: "second chunk reasoning".to_string(),
streaming: false,
duration_secs: Some(1.1),
},
];
app.resync_history_revisions();
let revisions = app.history_revisions.clone();
app.viewport.transcript_cache.ensure(
&app.history,
&revisions,
100,
app.transcript_render_options(),
);
let line = first_line_for_cell(&app, 0);
let point = TranscriptSelectionPoint {
line_index: line,
column: 0,
};
app.viewport.transcript_selection.anchor = Some(point);
app.viewport.transcript_selection.head = Some(point);
assert!(open_activity_detail_pager(&mut app));
let body = pop_pager_body(&mut app);
assert!(
body.contains("Activity: reasoning timeline"),
"activity label missing: {body}"
);
assert!(
body.contains("Selected chunk: 1 of 2"),
"chunk position missing: {body}"
);
assert!(
body.contains("Next chunk: 2 of 2 - second chunk reasoning"),
"neighboring chunk missing: {body}"
);
assert!(body.contains("Thinking chunk 1 of 2 (selected)"), "{body}");
assert!(body.contains("Thinking chunk 2 of 2"), "{body}");
assert!(body.contains("first chunk reasoning"), "body: {body}");
assert!(
body.contains("second chunk reasoning"),
"timeline should include the whole session's thinking: {body}"
);
}
#[test]
fn activity_detail_includes_tool_handle_and_neighbor_context() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::Thinking {
content: "checked approach".to_string(),
streaming: false,
duration_secs: Some(0.6),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "read_file".to_string(),
status: ToolStatus::Success,
input_summary: Some("src/main.rs".to_string()),
output: Some("bounded preview".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "grep_files".to_string(),
status: ToolStatus::Success,
input_summary: Some("TODO".to_string()),
output: Some("grep summary".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
];
app.tool_details_by_cell.insert(
1,
ToolDetailRecord {
tool_id: "call-read".to_string(),
tool_name: "read_file".to_string(),
input: serde_json::json!({"path": "src/main.rs"}),
output: Some("full output behind raw details".to_string()),
},
);
app.session_artifacts
.push(crate::artifacts::ArtifactRecord {
id: "art_call-read".to_string(),
kind: crate::artifacts::ArtifactKind::ToolOutput,
session_id: "session-activity".to_string(),
tool_call_id: "call-read".to_string(),
tool_name: "read_file".to_string(),
created_at: chrono::Utc::now(),
byte_size: 42,
preview: "bounded preview".to_string(),
storage_path: PathBuf::from("artifacts").join("art_call-read.txt"),
});
app.resync_history_revisions();
let revisions = app.history_revisions.clone();
app.viewport.transcript_cache.ensure(
&app.history,
&revisions,
100,
app.transcript_render_options(),
);
let line = first_line_for_cell(&app, 1);
let point = TranscriptSelectionPoint {
line_index: line,
column: 0,
};
app.viewport.transcript_selection.anchor = Some(point);
app.viewport.transcript_selection.head = Some(point);
assert!(open_activity_detail_pager(&mut app));
let body = pop_pager_body(&mut app);
assert!(body.contains("Activity: read"), "{body}");
assert!(body.contains("Activity chunk: 2 of 3"), "{body}");
assert!(
body.contains("Previous activity: 1 of 3 - thinking"),
"{body}"
);
assert!(body.contains("Next activity: 3 of 3 - find"), "{body}");
assert!(body.contains("Detail handle: art_call-read"), "{body}");
assert!(
body.contains("retrieve_tool_result ref=art_call-read"),
"{body}"
);
assert!(body.contains("v raw"), "{body}");
assert!(body.contains("details)"), "{body}");
}
#[test]
fn activity_detail_fallback_prefers_live_activity_context() {
let mut app = create_test_app();
let mut active = ActiveCell::new();
active.push_tool(
"active-1",
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "agent".to_string(),
status: ToolStatus::Running,
input_summary: Some("agent_id: agent_af58ba3a".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
);
app.active_cell = Some(active);
app.runtime_turn_id = Some("turn_live_123456789".to_string());
app.runtime_turn_status = Some("in_progress".to_string());
assert!(open_activity_detail_pager(&mut app));
let body = pop_pager_body(&mut app);
assert!(
body.contains("Turn turn_live_12 \u{00B7} in progress"),
"{body}"
);
assert!(!body.contains("turn_live_123456789"), "{body}");
assert!(body.contains("Activity: delegate"));
assert!(body.contains("Status: running"));
assert!(body.contains("agent_id: agent_af58ba3a"));
}
#[test]
fn activity_detail_fallback_uses_recent_meaningful_activity_without_full_tool_dump() {
let mut app = create_test_app();
let output = (0..20)
.map(|idx| format!("line {idx}"))
.collect::<Vec<_>>()
.join("\n");
app.history
.push(HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "read_file".to_string(),
status: ToolStatus::Success,
input_summary: Some("src/large.rs".to_string()),
output: Some(output),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})));
assert!(open_activity_detail_pager(&mut app));
let body = pop_pager_body(&mut app);
assert!(body.contains("Activity: read"));
assert!(body.contains("Status: done"));
assert!(body.contains("Detail handle: v details"), "{body}");
assert!(
!body.contains("Detail handle: v raw details"),
"fallback tool details should not be labeled raw: {body}"
);
assert!(
!body.contains("line 10"),
"middle of large raw output should not be dumped into Activity Detail: {body}"
);
}
#[test]
fn turn_inspector_renders_overview_sections_for_active_turn() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "Fix the flaky login test".to_string(),
},
HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test login".to_string(),
status: ToolStatus::Success,
output: Some("ok".to_string()),
live_output: None,
shell_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
started_at: None,
duration_ms: Some(2400),
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})),
HistoryCell::Tool(ToolCell::PatchSummary(
crate::tui::history::PatchSummaryCell {
path: "src/login.rs".to_string(),
summary: "guard against empty token".to_string(),
status: ToolStatus::Success,
error: None,
},
)),
HistoryCell::Assistant {
content: "Fixed the race in the login test.".to_string(),
streaming: false,
},
];
app.runtime_turn_id = Some("turn_abc123456789".to_string());
app.runtime_turn_status = Some("completed".to_string());
let body = turn_inspector_text(&app);
assert!(
body.contains("Turn turn_abc1234 \u{00B7} completed"),
"{body}"
);
assert!(!body.contains("turn_abc123456789"), "{body}");
assert!(
body.contains("press v for the selected item's raw detail"),
"{body}"
);
for header in [
"── Intent ──",
"── Strategy / To-do ──",
"── Turn timeline ──",
"── Files changed ──",
"── Diagnostics loop ──",
"── Tests / verifier ──",
"── Approvals / denials ──",
"── Model route + tokens/cost ──",
"── Final result / status ──",
] {
assert!(body.contains(header), "missing section {header}: {body}");
}
assert!(body.contains("Fix the flaky login test"), "{body}");
assert!(body.contains("test/verifier: cargo test login"), "{body}");
assert!(body.contains("2.4s"), "duration missing: {body}");
assert!(body.contains("src/login.rs"), "{body}");
assert!(
body.contains("cargo test login — done"),
"verifier section should surface the test run: {body}"
);
assert!(body.contains("Route: DeepSeek"), "route missing: {body}");
assert!(
body.contains("Result: Fixed the race in the login test."),
"{body}"
);
assert!(body.contains("Status: completed"), "{body}");
}
#[test]
fn turn_inspector_timeline_numbers_semantic_entries_and_checkpoint_actions() {
let _lock = crate::test_support::lock_test_env();
let tmp = TempDir::new().expect("tempdir");
let _home = crate::test_support::EnvVarGuard::set("HOME", tmp.path());
let _userprofile = crate::test_support::EnvVarGuard::set("USERPROFILE", tmp.path());
let workspace = tmp.path().join("workspace");
std::fs::create_dir_all(&workspace).expect("workspace");
std::fs::write(workspace.join("src.rs"), "fn main() {}\n").expect("source");
let repo = crate::snapshot::SnapshotRepo::open_or_init(&workspace).expect("snapshot repo");
repo.snapshot("pre-turn:12: Fix timeline")
.expect("pre-turn snapshot");
let mut app = create_test_app();
app.workspace = workspace;
app.turn_counter = 12;
app.runtime_turn_status = Some("completed".to_string());
app.history = vec![
HistoryCell::User {
content: "Fix timeline evidence".to_string(),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "read_file".to_string(),
status: ToolStatus::Success,
input_summary: Some("src.rs".to_string()),
output: Some("fn main() {}".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
HistoryCell::Tool(ToolCell::PatchSummary(
crate::tui::history::PatchSummaryCell {
path: "src.rs".to_string(),
summary: "add timeline evidence".to_string(),
status: ToolStatus::Success,
error: None,
},
)),
HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test timeline".to_string(),
status: ToolStatus::Success,
output: Some("ok".to_string()),
live_output: None,
shell_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
started_at: None,
duration_ms: Some(1250),
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})),
HistoryCell::Assistant {
content: "Timeline evidence is visible.".to_string(),
streaming: false,
},
];
let body = turn_inspector_text(&app);
assert!(
body.contains("1. user prompt: Fix timeline evidence"),
"{body}"
);
assert!(
body.contains("2. read/search: read · src.rs — done · actions: v raw detail"),
"{body}"
);
assert!(
body.contains(
"3. edit: src.rs — add timeline evidence — done · actions: v raw detail, d diff"
),
"{body}"
);
assert!(
body.contains(
"4. test/verifier: cargo test timeline — done · 1.2s · actions: v raw detail"
),
"{body}"
);
assert!(
body.contains("checkpoint: pre-turn:12: Fix timeline"),
"{body}"
);
assert!(
body.contains("actions: r restore via /restore (guarded), e export handoff"),
"{body}"
);
}
#[test]
fn turn_inspector_degrades_empty_sections_without_panic() {
let mut app = create_test_app();
app.history = vec![HistoryCell::User {
content: "hello".to_string(),
}];
app.runtime_turn_status = Some("in_progress".to_string());
let body = turn_inspector_text(&app);
assert!(body.contains("── Strategy / To-do ──\nnone"), "{body}");
assert!(body.contains("── Files changed ──\nnone"), "{body}");
assert!(body.contains("── Tests / verifier ──\nnone"), "{body}");
assert!(body.contains("── Approvals / denials ──\nnone"), "{body}");
assert!(body.contains("hello"), "{body}");
assert!(body.contains("Status: in progress"), "{body}");
assert!(body.contains("Result: turn still running"), "{body}");
}
#[test]
fn turn_inspector_scopes_to_latest_turn_only() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "first turn prompt".to_string(),
},
HistoryCell::Assistant {
content: "first turn answer".to_string(),
streaming: false,
},
HistoryCell::User {
content: "second turn prompt".to_string(),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "read_file".to_string(),
status: ToolStatus::Success,
input_summary: Some("src/lib.rs".to_string()),
output: Some("done".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
];
let body = turn_inspector_text(&app);
assert!(body.contains("second turn prompt"), "intent scope: {body}");
assert!(
!body.contains("first turn prompt"),
"inspector leaked prior turn intent: {body}"
);
}
#[test]
fn ctrl_o_open_turn_inspector_pager_opens_turn_overview_not_single_cell() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "do the thing".to_string(),
},
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: "grep_files".to_string(),
status: ToolStatus::Success,
input_summary: Some("TODO".to_string()),
output: Some("summary".to_string()),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
})),
];
assert!(open_turn_inspector_pager(&mut app));
assert_eq!(app.view_stack.top_kind(), Some(ModalKind::Pager));
let body = pop_pager_body(&mut app);
assert!(body.contains("── Turn timeline ──"), "{body}");
assert!(body.contains("do the thing"), "{body}");
assert!(
!body.contains("Activity chunk:"),
"Ctrl+O must open the turn overview, not the single-cell detail: {body}"
);
}
#[test]
fn turn_handoff_markdown_renders_compact_sections_for_active_turn() {
let mut app = create_test_app();
app.history = vec![
HistoryCell::User {
content: "Fix the flaky login test".to_string(),
},
HistoryCell::Tool(ToolCell::Exec(ExecCell {
command: "cargo test login".to_string(),
status: ToolStatus::Success,
output: Some("ok".to_string()),
live_output: None,
shell_task_id: None,
owner_agent_id: None,
owner_agent_name: None,
started_at: None,
duration_ms: Some(2400),
source: ExecSource::Assistant,
interaction: None,
output_summary: None,
})),
HistoryCell::Tool(ToolCell::PatchSummary(
crate::tui::history::PatchSummaryCell {
path: "src/login.rs".to_string(),
summary: "guard against empty token".to_string(),
status: ToolStatus::Success,
error: None,
},
)),
HistoryCell::Assistant {
content: "Fixed the race in the login test.".to_string(),
streaming: false,
},
];
app.runtime_turn_id = Some("turn_abc123456789".to_string());
app.runtime_turn_status = Some("completed".to_string());
let md = turn_handoff_markdown(&app);
assert!(md.contains("# Turn handoff — turn_abc1234"), "{md}");
for heading in [
"## Intent",
"## Files changed",
"## Turn timeline",
"## Tests / verifier",
"## Model route + tokens/cost",
"## Result / status",
] {
assert!(md.contains(heading), "missing heading {heading}: {md}");
}
assert!(md.contains("Fix the flaky login test"), "{md}");
assert!(md.contains("- src/login.rs"), "changed file bullet: {md}");
assert!(md.contains("cargo test login"), "{md}");
assert!(md.contains("Route: DeepSeek"), "route missing: {md}");
assert!(
md.contains("Result: Fixed the race in the login test."),
"{md}"
);
assert!(md.contains("Status: completed"), "{md}");
}
#[test]
fn turn_handoff_markdown_degrades_empty_sections_without_panic() {
let mut app = create_test_app();
app.history = vec![HistoryCell::User {
content: "hello".to_string(),
}];
app.runtime_turn_status = Some("in_progress".to_string());
let md = turn_handoff_markdown(&app);
assert!(md.contains("## Files changed\n—"), "{md}");
assert!(md.contains("## Tests / verifier\n—"), "{md}");
assert!(!md.contains("## Strategy / To-do"), "{md}");
assert!(md.contains("## Intent\nhello"), "{md}");
assert!(md.contains("Status: in progress"), "{md}");
assert!(md.contains("Result: turn still running"), "{md}");
}
#[test]
fn engine_error_finalizes_active_thinking_block() {
use crate::error_taxonomy::StreamError;
let mut app = create_test_app();
let entry_idx = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
app.thinking_started_at = Some(Instant::now());
app.streaming_state.start_thinking(0, None);
app.streaming_state.push_content(0, "partial reasoning");
apply_engine_error_to_app(
&mut app,
StreamError::Stall { timeout_secs: 60 }.into_envelope(),
);
let active = app.active_cell.as_ref().expect("active thinking remains");
let HistoryCell::Thinking {
content, streaming, ..
} = &active.entries()[entry_idx]
else {
panic!("expected active thinking cell");
};
assert!(!*streaming, "error path must stop the thinking spinner");
assert!(
content.contains("partial reasoning"),
"error path must drain pending thinking tail"
);
assert!(app.streaming_thinking_active_entry.is_none());
}
#[test]
fn message_complete_drain_preserves_thinking_when_thinking_complete_lost() {
let mut app = create_test_app();
let _ = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
app.thinking_started_at = Some(Instant::now());
app.streaming_state.start_thinking(0, None);
app.streaming_state.push_content(0, "deep reasoning text");
let _ = app.streaming_state.commit_text(0);
app.reasoning_buffer.push_str("deep reasoning text");
assert!(
app.last_reasoning.is_none(),
"precondition: ThinkingComplete has NOT fired"
);
assert!(
app.streaming_thinking_active_entry.is_some(),
"precondition: thinking entry is still active"
);
if app.streaming_thinking_active_entry.is_some() {
let _ = crate::tui::streaming_thinking::finalize_current(&mut app);
crate::tui::streaming_thinking::stash_reasoning_buffer_into_last_reasoning(&mut app);
}
assert!(
app.last_reasoning
.as_deref()
.is_some_and(|s| s.contains("deep reasoning text")),
"defensive drain must move reasoning into last_reasoning so the\
downstream `last_reasoning.take()` produces a Thinking block"
);
assert!(
app.streaming_thinking_active_entry.is_none(),
"thinking entry must be cleared after the drain"
);
}
#[test]
fn approval_prompt_uses_event_input_after_message_complete_drain() {
let mut app = create_test_app();
app.pending_tool_uses.push((
"tool-1".to_string(),
"exec_shell".to_string(),
serde_json::json!({"command": "stale value from drained list"}),
));
app.pending_tool_uses.clear();
let event_input = serde_json::json!({
"command": "cargo test -p codewhale-tui approval",
"workdir": "/repo",
});
push_approval_request_view(
&mut app,
"tool-1",
"exec_shell",
"Run cargo tests",
&event_input,
"approval-key",
None,
);
let mut view = app.view_stack.pop().expect("approval view");
let approval = view
.as_any_mut()
.downcast_mut::<ApprovalView>()
.expect("approval view");
let action = approval.handle_key(KeyEvent::new(KeyCode::Char('v'), KeyModifiers::NONE));
let ViewAction::Emit(ViewEvent::OpenTextPager { content, .. }) = action else {
panic!("expected approval params pager");
};
assert!(content.contains("cargo test -p codewhale-tui approval"));
assert!(content.contains("/repo"));
assert!(!content.contains("stale value from drained list"));
assert_ne!(content.trim(), "{}");
}
#[tokio::test]
async fn approval_decision_persists_ask_rules_to_permissions_file() {
let tmp = TempDir::new().expect("tempdir");
let config_path = tmp.path().join("config.toml");
let mut app = create_test_app();
app.config_path = Some(config_path.clone());
let mut config = Config::default();
let mut engine = mock_engine_handle();
let rule = codewhale_config::ToolAskRule::exec_shell("cargo test");
apply_approval_decision(
&mut app,
&mut engine.handle,
&mut config,
ApprovalDecisionEvent {
tool_id: "tool-1".to_string(),
tool_name: "exec_shell".to_string(),
decision: ReviewDecision::Approved,
timed_out: false,
approval_key: "approval-key".to_string(),
approval_grouping_key: "approval-group".to_string(),
persistent_ask_rules: vec![rule.clone()],
},
)
.await;
assert_eq!(
engine.recv_approval_event().await,
Some(crate::core::engine::MockApprovalEvent::Approved {
id: "tool-1".to_string()
})
);
let store = codewhale_config::ConfigStore::load(Some(config_path)).expect("load config store");
assert_eq!(store.permissions().rules, vec![rule]);
assert!(
app.status_message
.as_deref()
.is_some_and(|message| message.contains("Saved 1 ask permission rule"))
);
let decision = config
.exec_policy_engine
.check(codewhale_execpolicy::ExecPolicyContext {
command: "cargo test --workspace",
cwd: tmp.path().to_string_lossy().as_ref(),
tool: Some("exec_shell"),
path: None,
ask_for_approval: codewhale_execpolicy::AskForApproval::OnFailure,
sandbox_mode: None,
})
.expect("check persisted runtime policy");
assert!(decision.requires_approval);
}
#[test]
fn second_thinking_block_appends_new_entry_in_same_active_cell() {
let mut app = create_test_app();
let _ = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
crate::tui::streaming_thinking::append(&mut app, 0, "first plan");
let _ = crate::tui::streaming_thinking::finalize_active_entry(&mut app, Some(0.5), "");
handle_tool_call_started(
&mut app,
"t-1",
"exec_shell",
&serde_json::json!({"command": "ls"}),
);
let second_idx = crate::tui::streaming_thinking::ensure_active_entry(&mut app);
assert_eq!(
second_idx, 2,
"second thinking entry follows the tool entry"
);
crate::tui::streaming_thinking::append(&mut app, second_idx, "second plan");
let active = app.active_cell.as_ref().expect("active cell present");
assert_eq!(active.entry_count(), 3);
assert!(matches!(active.entries()[0], HistoryCell::Thinking { .. }));
assert!(matches!(
active.entries()[1],
HistoryCell::Tool(ToolCell::Exec(_))
));
assert!(matches!(active.entries()[2], HistoryCell::Thinking { .. }));
assert!(
app.history.is_empty(),
"the group still hasn't flushed — no prose yet"
);
}
#[test]
fn new_thinking_block_drains_pending_tail_from_previous_block() {
let mut app = create_test_app();
assert!(!crate::tui::streaming_thinking::start_block(&mut app));
let first_idx = app
.streaming_thinking_active_entry
.expect("first thinking entry active");
app.reasoning_buffer.push_str("first tail");
app.streaming_state.push_content(0, "first tail");
assert!(crate::tui::streaming_thinking::start_block(&mut app));
let second_idx = app
.streaming_thinking_active_entry
.expect("second thinking entry active");
let active = app.active_cell.as_ref().expect("active cell exists");
assert_ne!(first_idx, second_idx);
let HistoryCell::Thinking {
content, streaming, ..
} = &active.entries()[first_idx]
else {
panic!("expected first thinking cell");
};
assert!(!*streaming, "previous thinking block should be finalized");
assert!(
content.contains("first tail"),
"pending text must survive a new ThinkingStarted event"
);
assert!(matches!(
active.entries()[second_idx],
HistoryCell::Thinking {
streaming: true,
..
}
));
assert_eq!(app.last_reasoning.as_deref(), Some("first tail"));
}
#[test]
fn non_fanout_tool_does_not_populate_prompts() {
let mut app = create_test_app();
handle_tool_call_started(
&mut app,
"fs-1",
"file_search",
&serde_json::json!({ "query": "client.rs" }),
);
let active = app.active_cell.as_ref().expect("active cell present");
let HistoryCell::Tool(ToolCell::Generic(generic)) = &active.entries()[0] else {
panic!("expected GenericToolCell for file_search");
};
assert!(
generic.prompts.is_none(),
"non-fan-out tool must not populate prompts"
);
}
#[test]
fn noisy_subagent_progress_keeps_existing_objective_summary() {
let mut app = create_test_app();
app.agent_progress.insert(
"agent_live".to_string(),
"starting: inspect release state".to_string(),
);
let display =
friendly_subagent_progress(&app, "agent_live", "step 1/8: requesting model response");
assert_eq!(display, "starting: inspect release state");
}
#[test]
fn truncate_line_to_width_respects_display_width_for_tiny_budgets() {
use unicode_width::UnicodeWidthStr;
let trimmed = truncate_line_to_width("Agents", 3);
assert_eq!(trimmed, "Age");
assert!(UnicodeWidthStr::width(trimmed.as_str()) <= 3);
let trimmed_cjk = truncate_line_to_width("中文测试", 3);
assert!(
UnicodeWidthStr::width(trimmed_cjk.as_str()) <= 3,
"trimmed CJK width {} exceeded budget 3 (got {trimmed_cjk:?})",
UnicodeWidthStr::width(trimmed_cjk.as_str()),
);
assert_eq!(truncate_line_to_width("anything", 0), "");
assert_eq!(truncate_line_to_width("hi", 10), "hi");
let trimmed_long = truncate_line_to_width("a long sidebar label", 10);
assert!(trimmed_long.ends_with("..."));
assert!(UnicodeWidthStr::width(trimmed_long.as_str()) <= 10);
}
#[test]
fn recoverable_engine_error_does_not_enter_offline_mode() {
use crate::error_taxonomy::{ErrorEnvelope, StreamError};
let mut app = create_test_app();
assert!(!app.offline_mode);
let envelope = StreamError::Stall { timeout_secs: 60 }.into_envelope();
apply_engine_error_to_app(&mut app, envelope);
assert!(
!app.offline_mode,
"recoverable error must keep the session online so the user can retry"
);
assert!(!app.is_loading);
assert!(app.turn_error_posted, "turn_error_posted must be set");
assert!(
app.status_message.is_none(),
"recoverable error should NOT set status_message — already in transcript as HistoryCell::Error"
);
let last = app
.history
.last()
.expect("recoverable engine error should push a history cell");
assert!(
matches!(last, crate::tui::history::HistoryCell::Error { .. }),
"expected HistoryCell::Error, got {last:?}"
);
let _ = ErrorEnvelope::transient("");
}
#[test]
fn recoverable_provider_error_advances_fallback_chain() {
use crate::error_taxonomy::{ErrorCategory, ErrorEnvelope, ErrorSeverity};
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.provider_chain = Some(codewhale_config::ProviderChain::new(
codewhale_config::ProviderKind::Deepseek,
&[codewhale_config::ProviderKind::Openrouter],
));
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::new(
ErrorCategory::RateLimit,
ErrorSeverity::Warning,
true,
"rate_limit",
"provider returned 429",
),
);
assert_eq!(app.api_provider, ApiProvider::Openrouter);
assert!(app.is_fallback_active());
assert!(!app.offline_mode);
assert!(
app.status_message
.as_deref()
.unwrap_or_default()
.contains("Switched to openrouter")
);
assert!(
app.last_fallback_reason
.as_deref()
.unwrap_or_default()
.contains("provider returned 429")
);
}
#[test]
fn auth_error_does_not_trigger_provider_fallback() {
use crate::error_taxonomy::{ErrorCategory, ErrorEnvelope, ErrorSeverity};
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.api_key_env_only = false;
app.provider_chain = Some(codewhale_config::ProviderChain::new(
codewhale_config::ProviderKind::Deepseek,
&[codewhale_config::ProviderKind::Openrouter],
));
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::new(
ErrorCategory::Authentication,
ErrorSeverity::Critical,
true,
"authentication",
"provider returned 401",
),
);
assert_eq!(
app.api_provider,
ApiProvider::Deepseek,
"auth failure must not rotate providers"
);
assert!(!app.is_fallback_active());
assert_eq!(app.fallback_chain_position(), Some(0));
assert!(
app.last_fallback_reason.is_none(),
"no fallback should have been attempted on an auth error"
);
}
#[test]
fn fallback_switch_status_shows_one_based_position_and_reason() {
use crate::error_taxonomy::{ErrorCategory, ErrorEnvelope, ErrorSeverity};
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.provider_chain = Some(codewhale_config::ProviderChain::new(
codewhale_config::ProviderKind::Deepseek,
&[codewhale_config::ProviderKind::Openrouter],
));
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::new(
ErrorCategory::RateLimit,
ErrorSeverity::Warning,
true,
"rate_limit",
"provider returned 429",
),
);
assert_eq!(app.api_provider, ApiProvider::Openrouter);
assert_eq!(
app.fallback_chain_position(),
Some(1),
"first fallback sits at 1-based position 1"
);
let status = app.status_message.as_deref().unwrap_or_default();
assert!(
status.contains("Switched to openrouter") && status.contains("(fallback 1/"),
"visible status must show the destination and 1-based position: {status}"
);
}
#[tokio::test]
async fn failed_fallback_restores_exact_literal_custom_identity_without_root_crossover() {
let mut config = Config {
provider: Some("custom".to_string()),
api_key: Some("legacy-root-key".to_string()),
base_url: Some("http://127.0.0.1:18180/v1".to_string()),
default_text_model: Some("legacy-root-model".to_string()),
providers: Some(ProvidersConfig {
custom: HashMap::from([(
"custom".to_string(),
ProviderConfig {
kind: Some("openai-compatible".to_string()),
api_key: Some("literal-table-key".to_string()),
base_url: Some("http://127.0.0.1:18181/v1".to_string()),
model: Some("literal-table-model".to_string()),
..Default::default()
},
)]),
..Default::default()
}),
..Default::default()
};
let previous_identity = ProviderIdentity {
provider: ApiProvider::Custom,
key: "custom".to_string(),
exact_id: Some("custom".to_string()),
};
let mut app = create_test_app();
app.set_provider_identity_record(previous_identity.clone());
app.set_model_selection("literal-table-model".to_string());
app.api_provider = ApiProvider::Openrouter;
let mut engine = mock_engine_handle();
let previous_chain = app.provider_chain.clone();
apply_provider_fallback_switch(
&mut app,
&mut engine.handle,
&mut config,
ProviderFallbackRollback {
identity: previous_identity.clone(),
chain: previous_chain,
},
)
.await;
assert_eq!(app.api_provider, ApiProvider::Custom);
assert_eq!(app.provider_identity_for_persistence(), "custom");
assert_eq!(app.provider_id_for_persistence(), Some("custom"));
let (restored_identity, restored_config) = app_scoped_runtime_config(&app, &config);
assert_eq!(restored_identity, previous_identity);
let route = resolve_runtime_route_for_identity(
&restored_config,
&restored_identity,
Some("literal-table-model"),
)
.expect("restored identity must still resolve the exact literal table");
assert_eq!(route.identity.exact_id.as_deref(), Some("custom"));
assert_eq!(
route.candidate.endpoint.base_url,
"http://127.0.0.1:18181/v1"
);
assert_ne!(
route.candidate.endpoint.base_url,
"http://127.0.0.1:18180/v1"
);
}
#[tokio::test]
async fn provider_switch_auth_error_restores_previous_provider_and_model() {
use crate::error_taxonomy::ErrorEnvelope;
let _home = SettingsHomeGuard::new();
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.model = "deepseek-v4-pro".to_string();
app.model_ids_passthrough = false;
app.onboarding = OnboardingState::None;
app.onboarding_needs_api_key = false;
app.api_key_env_only = true;
app.active_context_window_override = Some(1_000_000);
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("deepseek".to_string()),
api_key: Some("deepseek-key".to_string()),
default_text_model: Some("deepseek-v4-pro".to_string()),
providers: Some(ProvidersConfig {
deepseek: ProviderConfig {
api_key: Some("deepseek-key".to_string()),
context_window: Some(1_000_000),
..Default::default()
},
moonshot: ProviderConfig {
api_key: Some("kimi-key".to_string()),
context_window: Some(262_144),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Moonshot,
Some("kimi-k2.6".to_string()),
)
.await;
assert_eq!(app.api_provider, ApiProvider::Moonshot);
assert_eq!(app.active_context_window_override, Some(262_144));
assert_eq!(config.provider.as_deref(), Some("moonshot"));
assert!(app.pending_provider_switch.is_some());
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::fatal_auth("Authentication failed: invalid API key"),
);
let rollback_status = rollback_provider_after_auth_failure(&mut app, &mut config)
.expect("auth failure after provider switch should roll back");
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert_eq!(app.model, "deepseek-v4-pro");
assert_eq!(app.active_context_window_override, Some(1_000_000));
assert!(!app.model_ids_passthrough);
assert!(!app.offline_mode);
assert_eq!(app.onboarding, OnboardingState::None);
assert!(!app.onboarding_needs_api_key);
assert!(app.api_key_env_only);
assert_eq!(config.provider.as_deref(), Some("deepseek"));
assert_eq!(
config.default_text_model.as_deref(),
Some("deepseek-v4-pro")
);
let settings = crate::settings::Settings::load().expect("load settings");
assert_eq!(settings.default_provider.as_deref(), Some("deepseek"));
assert_eq!(
settings
.provider_models
.as_ref()
.and_then(|models| models.get("deepseek"))
.map(String::as_str),
Some("deepseek-v4-pro")
);
assert_eq!(settings.default_model.as_deref(), Some("deepseek-v4-pro"));
let state = codewhale_config::SetupState::load()
.expect("load setup state")
.expect("setup state");
assert_eq!(
state.status(codewhale_config::SetupStep::ProviderModel),
codewhale_config::StepStatus::Verified
);
let provider_model_result = state
.steps
.get(&codewhale_config::SetupStep::ProviderModel)
.and_then(|entry| entry.result.as_deref())
.expect("provider/model result");
assert!(provider_model_result.contains("provider=deepseek"));
assert!(provider_model_result.contains("model=deepseek-v4-pro"));
assert!(provider_model_result.contains("auth=key saved · not checked"));
assert!(provider_model_result.contains("health=attemptable"));
assert!(!provider_model_result.contains("moonshot"));
assert!(!provider_model_result.contains("kimi-k2.6"));
assert!(!provider_model_result.contains("deepseek-key"));
assert!(!provider_model_result.contains("kimi-key"));
assert!(app.pending_provider_switch.is_none());
assert!(rollback_status.contains("Provider switch failed"));
assert!(
app.status_message
.as_deref()
.is_none_or(|status| !status.contains("Provider switch failed")),
"status message is set by the async event loop after engine respawn"
);
}
#[tokio::test]
async fn provider_switch_rollback_corrects_setup_receipt_when_persistence_fails() {
use crate::error_taxonomy::ErrorEnvelope;
let _home = SettingsHomeGuard::new();
let bad_config_path = TempDir::new().expect("bad config path");
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.model = "deepseek-v4-pro".to_string();
app.model_ids_passthrough = false;
app.onboarding = OnboardingState::None;
app.onboarding_needs_api_key = false;
app.api_key_env_only = true;
let mut engine = mock_engine_handle();
let mut config = Config {
provider: Some("deepseek".to_string()),
api_key: Some("deepseek-key".to_string()),
default_text_model: Some("deepseek-v4-pro".to_string()),
providers: Some(ProvidersConfig {
deepseek: ProviderConfig {
api_key: Some("deepseek-key".to_string()),
..Default::default()
},
moonshot: ProviderConfig {
api_key: Some("kimi-key".to_string()),
..Default::default()
},
..Default::default()
}),
..Default::default()
};
switch_provider(
&mut app,
&mut engine.handle,
&mut config,
ApiProvider::Moonshot,
Some("kimi-k2.6".to_string()),
)
.await;
let target_state = codewhale_config::SetupState::load()
.expect("load target setup state")
.expect("target setup state");
let target_result = target_state
.steps
.get(&codewhale_config::SetupStep::ProviderModel)
.and_then(|entry| entry.result.as_deref())
.expect("target provider/model result");
assert!(target_result.contains("provider=moonshot"));
assert!(target_result.contains("model=kimi-k2.6"));
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::fatal_auth("Authentication failed: invalid API key"),
);
app.config_path = Some(bad_config_path.path().to_path_buf());
let rollback_status = rollback_provider_after_auth_failure(&mut app, &mut config)
.expect("auth failure after provider switch should roll back");
assert_eq!(app.api_provider, ApiProvider::Deepseek);
assert_eq!(app.model, "deepseek-v4-pro");
assert!(
rollback_status.contains("not fully persisted"),
"{rollback_status}"
);
let state = codewhale_config::SetupState::load()
.expect("load setup state")
.expect("setup state");
let provider_model_result = state
.steps
.get(&codewhale_config::SetupStep::ProviderModel)
.and_then(|entry| entry.result.as_deref())
.expect("provider/model result");
assert!(provider_model_result.contains("provider=deepseek"));
assert!(provider_model_result.contains("model=deepseek-v4-pro"));
assert!(!provider_model_result.contains("moonshot"));
assert!(!provider_model_result.contains("kimi-k2.6"));
assert!(!provider_model_result.contains("deepseek-key"));
assert!(!provider_model_result.contains("kimi-key"));
}
#[test]
fn stream_error_marks_active_turn_failed_without_waiting_for_turn_complete() {
use crate::error_taxonomy::ErrorEnvelope;
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_id = Some("turn_decode_error".to_string());
app.runtime_turn_status = Some("in_progress".to_string());
handle_tool_call_started(
&mut app,
"tool-running",
"exec_shell",
&serde_json::json!({"command": "cargo test --workspace"}),
);
assert!(app.active_cell.is_some(), "precondition: live tool cell");
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::classify("chunk decode error".to_string(), true),
);
assert!(!app.is_loading);
assert_eq!(app.runtime_turn_status.as_deref(), Some("failed"));
assert!(
app.active_cell.is_none(),
"stream error should flush live cells so no row stays visually running"
);
assert!(
app.history.iter().any(|cell| {
matches!(
cell,
crate::tui::history::HistoryCell::Error { message, .. }
if message.contains("chunk decode error")
)
}),
"stream decode error should remain visible in transcript"
);
}
#[test]
fn non_recoverable_engine_error_enters_offline_mode() {
use crate::error_taxonomy::ErrorEnvelope;
let mut app = create_test_app();
assert!(!app.offline_mode);
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::fatal_auth("Authentication failed: invalid API key"),
);
assert!(
app.offline_mode,
"non-recoverable error must enter offline mode"
);
assert!(!app.is_loading);
assert!(app.turn_error_posted, "turn_error_posted must be set");
assert!(
app.status_message.is_none(),
"non-recoverable error should NOT set status_message — already in transcript as HistoryCell::Error"
);
assert!(app.pending_provider_switch.is_none());
}
#[test]
fn env_only_auth_failure_reopens_api_key_onboarding() {
use crate::error_taxonomy::ErrorEnvelope;
let mut app = create_test_app();
app.api_key_env_only = true;
app.onboarding = crate::tui::app::OnboardingState::None;
app.onboarding_needs_api_key = false;
apply_engine_error_to_app(
&mut app,
ErrorEnvelope::fatal_auth("Authentication failed: invalid API key"),
);
assert!(app.offline_mode);
assert_eq!(
app.onboarding,
crate::tui::app::OnboardingState::ApiKey,
"env-only auth failures should prompt for a saved config key"
);
assert!(app.onboarding_needs_api_key);
assert!(app.turn_error_posted, "turn_error_posted must be set");
let status = app
.status_message
.as_deref()
.expect("auth recovery should explain the env key source");
assert!(
status.contains("DEEPSEEK_API_KEY"),
"expected env-specific recovery hint, got {status:?}"
);
}
#[test]
fn next_escape_action_cancels_when_loading_with_empty_input() {
let mut app = create_test_app();
app.is_loading = true;
app.input.clear();
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
}
#[test]
fn next_escape_action_cancels_in_progress_runtime_even_if_loading_flag_was_cleared() {
let mut app = create_test_app();
app.is_loading = false;
app.runtime_turn_status = Some("in_progress".to_string());
app.input.clear();
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
}
#[test]
fn next_escape_action_cancels_when_loading_with_input() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "hold on, look at this instead".to_string();
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
}
#[test]
fn next_escape_action_treats_whitespace_only_as_empty() {
let mut app = create_test_app();
app.is_loading = true;
app.input = " \n\t".to_string();
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
}
#[test]
fn next_escape_action_idle_with_input_clears() {
let mut app = create_test_app();
app.is_loading = false;
app.input = "draft".to_string();
assert_eq!(next_escape_action(&app, false), EscapeAction::ClearInput);
}
#[test]
fn next_escape_action_idle_empty_is_noop() {
let mut app = create_test_app();
app.is_loading = false;
app.input.clear();
assert_eq!(next_escape_action(&app, false), EscapeAction::Noop);
}
#[test]
fn next_escape_action_slash_menu_takes_priority() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "anything".to_string();
assert_eq!(next_escape_action(&app, true), EscapeAction::CloseSlashMenu);
}
#[test]
fn tab_queues_running_turn_draft_for_next_turn() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "follow up next".to_string();
app.cursor_position = app.input.chars().count();
assert!(queue_current_draft_for_next_turn(&mut app));
assert!(app.input.is_empty());
assert_eq!(app.queued_message_count(), 1);
assert_eq!(
app.queued_messages.front().map(|msg| msg.display.as_str()),
Some("follow up next")
);
assert!(
app.status_message
.as_deref()
.is_some_and(|msg| msg.contains("queued follow-up(s)"))
);
}
#[test]
fn tab_queue_preserves_queued_draft_skill_instruction() {
let mut app = create_test_app();
app.is_loading = true;
app.input = "edited queued follow-up".to_string();
app.cursor_position = app.input.chars().count();
app.queued_draft = Some(QueuedMessage::new(
"original".to_string(),
Some("skill body".to_string()),
));
assert!(queue_current_draft_for_next_turn(&mut app));
let queued = app.queued_messages.front().expect("queued message");
assert_eq!(queued.display, "edited queued follow-up");
assert_eq!(queued.skill_instruction.as_deref(), Some("skill body"));
assert!(app.queued_draft.is_none());
}
#[test]
fn merge_pending_steers_returns_none_when_empty() {
let mut app = create_test_app();
assert!(merge_pending_steers(&mut app).is_none());
assert!(!app.submit_pending_steers_after_interrupt);
}
#[test]
fn merge_pending_steers_passes_through_single_message() {
let mut app = create_test_app();
app.push_pending_steer(QueuedMessage::new(
"lone steer".to_string(),
Some("skill body".to_string()),
));
let merged = merge_pending_steers(&mut app).expect("merge yields a message");
assert_eq!(merged.display, "lone steer");
assert_eq!(merged.skill_instruction.as_deref(), Some("skill body"));
assert!(app.pending_steers.is_empty());
assert!(!app.submit_pending_steers_after_interrupt);
}
#[test]
fn merge_pending_steers_concatenates_multiple_with_blank_line() {
let mut app = create_test_app();
app.push_pending_steer(QueuedMessage::new("first".to_string(), None));
app.push_pending_steer(QueuedMessage::new("second".to_string(), None));
app.push_pending_steer(QueuedMessage::new("third".to_string(), None));
let merged = merge_pending_steers(&mut app).expect("merge yields a message");
assert_eq!(merged.display, "first\n\nsecond\n\nthird");
assert!(app.pending_steers.is_empty());
}
#[test]
fn merge_pending_steers_keeps_first_skill_instruction_only() {
let mut app = create_test_app();
app.push_pending_steer(QueuedMessage::new(
"a".to_string(),
Some("first skill".to_string()),
));
app.push_pending_steer(QueuedMessage::new(
"b".to_string(),
Some("second skill".to_string()),
));
let merged = merge_pending_steers(&mut app).expect("merge yields a message");
assert_eq!(merged.skill_instruction.as_deref(), Some("first skill"));
assert_eq!(merged.display, "a\n\nb");
}
#[test]
fn build_pending_input_preview_populates_all_three_buckets() {
let mut app = create_test_app();
app.push_pending_steer(QueuedMessage::new("steer-msg".to_string(), None));
app.rejected_steers.push_back("rejected-msg".to_string());
app.queue_message(QueuedMessage::new("queued-msg".to_string(), None));
let preview = build_pending_input_preview(&app);
assert_eq!(preview.pending_steers, vec!["steer-msg".to_string()]);
assert_eq!(preview.rejected_steers, vec!["rejected-msg".to_string()]);
assert_eq!(preview.queued_messages, vec!["queued-msg".to_string()]);
}
#[test]
fn accidental_queue_edit_while_loading_is_labeled_and_recoverable() {
let mut app = create_test_app();
app.is_loading = true;
app.queue_message(QueuedMessage::new(
"original queued follow-up".to_string(),
Some("skill body".to_string()),
));
assert!(app.pop_last_queued_into_draft());
assert_eq!(app.input, "original queued follow-up");
app.input = "edited queued follow-up".to_string();
app.cursor_position = app.input.chars().count();
let preview = build_pending_input_preview(&app);
assert_eq!(
preview.editing_queued_message.as_deref(),
Some("edited queued follow-up")
);
assert!(
preview.queued_messages.is_empty(),
"the popped message should be shown as editing, not a second queued row"
);
assert_eq!(
next_escape_action(&app, false),
EscapeAction::DiscardQueuedDraft,
"Esc should cancel the queued edit before cancelling the live turn"
);
assert!(app.cancel_queued_draft_edit());
assert!(app.input.is_empty());
let restored = app.queued_messages.back().expect("follow-up restored");
assert_eq!(restored.display, "original queued follow-up");
assert_eq!(restored.skill_instruction.as_deref(), Some("skill body"));
}
#[test]
fn build_pending_input_preview_includes_current_context_chips() {
let mut app = create_test_app();
app.input = "Read @guide.md and @missing.md".to_string();
app.cursor_position = app.input.chars().count();
let preview = build_pending_input_preview(&app);
assert!(
preview
.context_items
.iter()
.any(|item| item.kind == "mention"
&& item.label == "guide.md"
&& !item.included
&& item.detail.as_deref() == Some("resolved on send")),
"neutral mention preview missing: {:?}",
preview.context_items
);
assert!(
preview
.context_items
.iter()
.any(|item| item.kind == "mention"
&& item.label == "missing.md"
&& !item.included
&& item.detail.as_deref() == Some("resolved on send")),
"unresolved mention preview missing: {:?}",
preview.context_items
);
}
#[test]
fn render_footer_from_with_default_items_leaves_header_owned_facts_out() {
let mut app = create_test_app();
app.session.session_cost = 0.00005;
let items = crate::config::StatusItem::default_footer();
let props = render_footer_from(&app, &items, None);
assert!(props.mode_label.is_empty(), "footer should not repeat mode");
assert!(props.model.is_empty(), "footer should not repeat model");
assert!(!props.cost.is_empty());
assert_eq!(spans_text(&props.cost), "<$0.0001");
}
#[test]
fn default_footer_excludes_provider_specific_diagnostic_chips() {
let items = crate::config::StatusItem::default_footer();
assert!(
!items.contains(&crate::config::StatusItem::PrefixStability),
"prefix stability is a diagnostic chip and should not crowd the default footer"
);
assert!(
!items.contains(&crate::config::StatusItem::Balance),
"balance is DeepSeek-only and should not crowd the default footer for non-DeepSeek users"
);
assert!(
items.contains(&crate::config::StatusItem::Cache),
"default footer should still include provider-reported cache hit rate"
);
assert!(
items.contains(&crate::config::StatusItem::GitBranch),
"default footer should surface the current workspace branch"
);
}
#[test]
fn render_footer_from_prefix_stability_item_renders_cache_slot_chip() {
let mut app = create_test_app();
app.prefix_stability_pct = Some(100);
app.prefix_change_count = 0;
let props = render_footer_from(&app, &[crate::config::StatusItem::PrefixStability], None);
assert_eq!(spans_text(&props.cache), "cache prefix 100%");
}
#[test]
fn render_footer_from_preserves_prefix_then_cache_order() {
let mut app = create_test_app();
app.prefix_stability_pct = Some(100);
app.prefix_change_count = 0;
app.session.last_prompt_tokens = Some(10_000);
app.session.last_prompt_cache_hit_tokens = Some(9_000);
app.session.last_prompt_cache_miss_tokens = Some(1_000);
let props = render_footer_from(
&app,
&[
crate::config::StatusItem::PrefixStability,
crate::config::StatusItem::Cache,
],
None,
);
assert!(spans_text(&props.cache).starts_with("cache prefix 100% Cache: 90.0% hit"));
}
#[test]
fn render_footer_from_with_empty_items_blanks_every_segment() {
let mut app = create_test_app();
app.session.session_cost = 1.5;
let props = render_footer_from(&app, &[], None);
assert_eq!(props.mode_label, "");
assert!(props.model.is_empty());
assert!(props.cost.is_empty());
assert!(props.agents.is_empty());
assert!(props.cache.is_empty());
}
#[test]
fn render_footer_from_surfaces_background_shell_even_without_tasks_panel() {
let mut app = create_test_app();
app.task_panel = vec![crate::tui::app::TaskPanelEntry {
id: "shell_abc".to_string(),
status: "running".to_string(),
prompt_summary: "shell: cargo test -p codewhale-tui".to_string(),
duration_ms: Some(5_000),
kind: crate::tui::app::TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
}];
let props = render_footer_from(&app, &[], None);
let shell = spans_text(&props.cache);
assert!(shell.contains("shell bg:"), "{shell}");
assert!(shell.contains("cargo test"), "{shell}");
}
#[test]
fn render_footer_from_drops_only_unselected_clusters() {
let mut app = create_test_app();
app.session.session_cost = 0.42;
let items: Vec<crate::config::StatusItem> = crate::config::StatusItem::default_footer()
.into_iter()
.filter(|item| *item != crate::config::StatusItem::Cost)
.collect();
let props = render_footer_from(&app, &items, None);
assert!(props.mode_label.is_empty());
assert!(props.model.is_empty(), "footer should not repeat model");
assert!(
props.cost.is_empty(),
"cost cluster should be empty when Cost is disabled"
);
}
#[test]
fn render_footer_from_git_branch_item_renders_workspace_branch() {
let repo = init_git_repo();
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/statusline"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
crate::tui::workspace_context::refresh_if_needed(&mut app, Instant::now(), true);
let props = render_footer_from(&app, &[crate::config::StatusItem::GitBranch], None);
let chip = spans_text(&props.cache);
assert!(
chip.starts_with("Repo: "),
"chip should name the repo: {chip:?}"
);
assert!(
chip.ends_with(" @ feature/statusline"),
"chip should show the current branch: {chip:?}"
);
}
#[test]
fn footer_balance_spans_empty_when_cell_is_none() {
let app = create_test_app();
let spans = footer_balance_spans(&app);
assert!(spans.is_empty());
}
#[test]
fn footer_balance_spans_empty_when_balance_is_zero() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "USD".into(),
total_balance: "0".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let spans = footer_balance_spans(&app);
assert!(spans.is_empty());
}
#[test]
fn footer_balance_spans_formats_cny() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "CNY".into(),
total_balance: "123.45".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let spans = footer_balance_spans(&app);
assert_eq!(spans_text(&spans), "balance ¥123.5");
}
#[test]
fn footer_balance_spans_formats_usd() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "USD".into(),
total_balance: "0.50".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let spans = footer_balance_spans(&app);
assert_eq!(spans_text(&spans), "balance $0.50");
}
#[test]
fn footer_balance_spans_rounds_large_amount() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "USD".into(),
total_balance: "1234.56".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let spans = footer_balance_spans(&app);
assert_eq!(spans_text(&spans), "balance $1235");
}
#[test]
fn footer_balance_spans_treats_unknown_currency_as_usd() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "EUR".into(),
total_balance: "10.00".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let spans = footer_balance_spans(&app);
assert_eq!(spans_text(&spans), "balance $10.0");
}
#[test]
fn render_footer_from_with_balance_item_shows_balance() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "USD".into(),
total_balance: "42.50".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let props = render_footer_from(&app, &[crate::config::StatusItem::Balance], None);
assert_eq!(spans_text(&props.balance), "balance $42.5");
}
#[test]
fn render_footer_from_without_balance_item_hides_balance() {
let app = create_test_app();
let info = crate::pricing::BalanceInfo {
currency: "USD".into(),
total_balance: "99.99".into(),
..Default::default()
};
*app.balance_cell.lock().unwrap() = Some(info);
let props = render_footer_from(&app, &[], None);
assert!(spans_text(&props.balance).is_empty());
}
#[test]
fn should_fetch_deepseek_balance_requires_balance_status_item() {
let mut app = create_test_app();
app.api_provider = ApiProvider::Deepseek;
app.status_items = crate::config::StatusItem::default_footer();
assert!(!should_fetch_deepseek_balance(&app));
app.status_items.push(crate::config::StatusItem::Balance);
assert!(should_fetch_deepseek_balance(&app));
}
#[test]
fn should_fetch_deepseek_balance_requires_deepseek_provider() {
let mut app = create_test_app();
app.status_items = vec![crate::config::StatusItem::Balance];
app.api_provider = ApiProvider::Openrouter;
assert!(!should_fetch_deepseek_balance(&app));
app.api_provider = ApiProvider::DeepseekCN;
assert!(should_fetch_deepseek_balance(&app));
}
#[test]
fn default_footer_renders_workspace_branch_when_available() {
let repo = init_git_repo();
let checkout = Command::new("git")
.args(["checkout", "-b", "feature/default-branch-chip"])
.current_dir(repo.path())
.output()
.expect("git checkout should run");
assert!(
checkout.status.success(),
"git checkout failed: {}",
String::from_utf8_lossy(&checkout.stderr)
);
let mut app = create_test_app();
app.workspace = repo.path().to_path_buf();
crate::tui::workspace_context::refresh_if_needed(&mut app, Instant::now(), true);
let props = render_footer_from(&app, &crate::config::StatusItem::default_footer(), None);
let cache = spans_text(&props.cache);
assert!(
cache.contains("feature/default-branch-chip"),
"default footer should include the current git branch: {cache:?}"
);
assert!(
cache.contains("Repo: "),
"default footer should name the repo identity (#3188): {cache:?}"
);
}
#[test]
fn displayed_session_cost_is_monotonic_under_negative_reconciliation() {
let mut app = create_test_app();
app.accrue_subagent_cost(0.50);
let after_first = app.displayed_session_cost();
assert!((after_first - 0.50).abs() < 1e-6);
app.session.subagent_cost = 0.20;
let after_recon = app.displayed_session_cost();
assert!(
after_recon >= after_first,
"displayed cost regressed: {after_recon} < {after_first}"
);
app.accrue_session_cost(0.10);
let after_add = app.displayed_session_cost();
assert!(after_add >= after_first);
}
#[test]
fn duplicate_mailbox_token_usage_does_not_regress_displayed_cost() {
let mut app = create_test_app();
let usage = crate::tools::subagent::MailboxMessage::TokenUsage {
agent_id: "agent-x".to_string(),
provider: ApiProvider::Deepseek,
model: "deepseek-v4-flash".to_string(),
usage: crate::models::Usage {
input_tokens: 10_000,
output_tokens: 1_000,
..Default::default()
},
};
handle_subagent_mailbox(&mut app, 11, &usage);
let baseline = app.displayed_session_cost();
assert!(baseline > 0.0);
handle_subagent_mailbox(&mut app, 11, &usage);
assert!(
(app.displayed_session_cost() - baseline).abs() < 1e-9,
"duplicate mailbox seq must not move displayed cost"
);
handle_subagent_mailbox(&mut app, 12, &usage);
assert!(app.displayed_session_cost() > baseline);
}
#[test]
fn checklist_write_renders_dedicated_card() {
let cell = GenericToolCell {
name: "work_update".to_string(),
status: ToolStatus::Success,
input_summary: None,
output: Some(
"Todo list updated (3 items, 33% complete)\n{\"items\":[{\"id\":1,\"content\":\"Plan it out\",\"status\":\"completed\"},{\"id\":2,\"content\":\"Wire the thing\",\"status\":\"in_progress\"},{\"id\":3,\"content\":\"Run gates\",\"status\":\"pending\"}],\"completion_pct\":33,\"in_progress_id\":2}"
.to_string(),
),
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
};
let lines = cell.lines_with_mode(80, true, crate::tui::history::RenderMode::Live);
let text: Vec<String> = lines
.iter()
.map(|line| {
line.spans
.iter()
.map(|span| span.content.as_ref())
.collect::<String>()
})
.collect();
let joined = text.join("\n");
assert!(
joined.contains("1/3"),
"header must include completed/total: {joined}"
);
assert!(
joined.contains("33%"),
"header must include percent: {joined}"
);
assert!(
joined.contains("Plan it out"),
"items must render content: {joined}"
);
assert!(
!joined.contains("\"items\""),
"raw JSON must NOT appear: {joined}"
);
}
#[test]
fn history_arrow_handles_empty_input() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input_history.push("previous prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "previous prompt");
}
#[test]
fn history_arrow_handles_whitespace_input() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input = " ".to_string();
app.cursor_position = app.input.chars().count();
app.input_history.push("previous prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "previous prompt");
}
#[test]
fn history_arrow_handles_nonempty_input() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input = "hello".to_string();
app.cursor_position = app.input.chars().count();
app.input_history.push("previous prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "previous prompt");
}
#[test]
fn composer_arrows_scroll_empty_up() {
let mut app = create_test_app();
app.composer_arrows_scroll = true;
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.viewport.pending_scroll_delta, -3);
assert!(app.input.is_empty());
}
#[test]
fn composer_arrows_scroll_empty_down() {
let mut app = create_test_app();
app.composer_arrows_scroll = true;
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Down, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.viewport.pending_scroll_delta, 3);
}
#[test]
fn composer_arrows_scroll_nonempty_also_scrolls() {
let mut app = create_test_app();
app.composer_arrows_scroll = true;
app.input = "hello".to_string();
app.cursor_position = app.input.chars().count();
app.input_history.push("previous prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.viewport.pending_scroll_delta, -3);
assert_eq!(app.input, "hello");
}
#[test]
fn composer_arrow_up_moves_within_multiline_input() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input = "line one\nline two".to_string();
app.cursor_position = app.input.chars().count();
app.input_history.push("previous prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "line one\nline two");
assert!(app.cursor_position < app.input.chars().count());
}
#[test]
fn composer_arrow_down_moves_within_multiline_input() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input = "line one\nline two".to_string();
app.cursor_position = 0;
app.input_history.push("next prompt".to_string());
app.history_index = Some(app.input_history.len() - 1);
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Down, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "line one\nline two");
assert!(app.cursor_position >= "line one\n".chars().count());
}
#[test]
fn composer_arrows_scroll_multiline_input_navigates_lines() {
let mut app = create_test_app();
app.composer_arrows_scroll = true;
app.input = "line one\nline two".to_string();
app.cursor_position = app.input.chars().count();
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "line one\nline two");
assert!(app.cursor_position < app.input.chars().count());
assert_eq!(app.viewport.pending_scroll_delta, 0);
}
#[test]
fn composer_arrow_up_at_first_line_preserves_multiline_draft() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input = "line one\nline two".to_string();
app.cursor_position = 0;
app.input_history.push("previous prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "line one\nline two");
assert_eq!(app.cursor_position, 0);
assert!(app.history_index.is_none());
}
#[test]
fn composer_arrow_down_at_last_line_preserves_multiline_draft() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input = "line one\nline two".to_string();
app.cursor_position = app.input.chars().count();
app.input_history.push("next prompt".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Down, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "line one\nline two");
assert_eq!(app.cursor_position, app.input.chars().count());
assert!(app.history_index.is_none());
}
#[test]
fn composer_arrows_scroll_defaults_true_without_mouse_capture() {
let options = TuiOptions {
use_mouse_capture: false,
..create_test_options()
};
let app = App::new(options, &Config::default());
assert!(
app.composer_arrows_scroll,
"arrows-scroll must default to true when mouse capture is off"
);
}
#[test]
fn composer_arrows_scroll_defaults_false_with_mouse_capture() {
let options = TuiOptions {
use_mouse_capture: true,
..create_test_options()
};
let app = App::new(options, &Config::default());
assert!(
!app.composer_arrows_scroll,
"arrows-scroll must default to false when mouse capture is on"
);
}
#[test]
fn composer_arrows_scroll_config_overrides_default() {
let config = Config {
tui: Some(crate::config::TuiConfig {
composer_arrows_scroll: Some(false),
..Default::default()
}),
..Config::default()
};
let options = TuiOptions {
use_mouse_capture: false,
..create_test_options()
};
let app = App::new(options, &config);
assert!(
!app.composer_arrows_scroll,
"explicit config=false must override the mouse-capture-derived default"
);
}
#[test]
fn history_arrow_down_handles_empty_input() {
let mut app = create_test_app();
app.composer_arrows_scroll = false;
app.input_history.push("older".to_string());
app.input_history.push("newer".to_string());
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Up, KeyModifiers::NONE),
false,
false,
));
assert_eq!(app.input, "newer");
assert!(handle_composer_history_arrow(
&mut app,
KeyEvent::new(KeyCode::Down, KeyModifiers::NONE),
false,
false,
));
assert!(app.input.is_empty());
assert!(app.history_index.is_none());
}
#[test]
fn home_jumps_to_line_start_multiline() {
let mut app = create_test_app();
app.input = "line one\nline two\nline three".to_string();
app.cursor_position = app.input.chars().count();
app.move_cursor_line_start();
assert_eq!(app.cursor_position, "line one\nline two\n".len());
}
#[test]
fn home_from_middle_of_line_jumps_to_line_start() {
let mut app = create_test_app();
app.input = "line one\nline two".to_string();
app.cursor_position = "line one\nli".len();
app.move_cursor_line_start();
assert_eq!(app.cursor_position, "line one\n".len());
}
#[test]
fn home_on_singleline_jumps_to_zero() {
let mut app = create_test_app();
app.input = "hello world".to_string();
app.cursor_position = 6;
app.move_cursor_line_start();
assert_eq!(app.cursor_position, 0);
}
#[test]
fn end_jumps_to_line_end_multiline() {
let mut app = create_test_app();
app.input = "line one\nline two\nline three".to_string();
app.cursor_position = 0;
app.move_cursor_line_end();
assert_eq!(app.cursor_position, "line one".len());
}
#[test]
fn end_from_middle_of_line_jumps_to_line_end() {
let mut app = create_test_app();
app.input = "line one\nline two".to_string();
app.cursor_position = "line one\nli".len();
app.move_cursor_line_end();
assert_eq!(app.cursor_position, "line one\nline two".len());
}
#[test]
fn end_on_singleline_jumps_to_absolute_end() {
let mut app = create_test_app();
app.input = "hello world".to_string();
app.cursor_position = 0;
app.move_cursor_line_end();
assert_eq!(app.cursor_position, app.input.chars().count());
}
#[test]
fn home_at_line_start_stays_put() {
let mut app = create_test_app();
app.input = "line one\nline two".to_string();
app.cursor_position = "line one\n".len();
app.move_cursor_line_start();
assert_eq!(app.cursor_position, "line one\n".len());
}
#[test]
fn end_at_newline_stays_at_line_end() {
let mut app = create_test_app();
app.input = "line one\nline two\nline three".to_string();
app.cursor_position = "line one".len();
app.move_cursor_line_end();
assert_eq!(app.cursor_position, "line one".len());
}
#[test]
fn notification_settings_tui_always_keeps_configured_method_no_threshold() {
let config = Config {
tui: Some(crate::config::TuiConfig {
notification_condition: Some(crate::config::NotificationCondition::Always),
..Default::default()
}),
notifications: Some(crate::config::NotificationsConfig {
method: crate::config::NotificationMethod::Bel,
threshold_secs: 120,
completion_sound: crate::config::CompletionSound::Beep,
sound_file: None,
include_summary: true,
subagent_completion: crate::config::SubagentCompletionNotification::default(),
}),
..Config::default()
};
let (method, threshold, include_summary) =
crate::tui::notifications::settings(&config).expect("notification should be enabled");
assert_eq!(method, crate::tui::notifications::Method::Bel);
assert_eq!(threshold, Duration::ZERO);
assert!(include_summary);
}
#[test]
fn notification_settings_tui_never_disables_notifications() {
let config = Config {
tui: Some(crate::config::TuiConfig {
notification_condition: Some(crate::config::NotificationCondition::Never),
..Default::default()
}),
..Config::default()
};
assert!(crate::tui::notifications::settings(&config).is_none());
}
#[test]
fn notification_settings_no_tui_override_uses_notifications_block() {
let config = Config {
notifications: Some(crate::config::NotificationsConfig {
method: crate::config::NotificationMethod::Osc9,
threshold_secs: 45,
completion_sound: crate::config::CompletionSound::Beep,
sound_file: None,
include_summary: false,
subagent_completion: crate::config::SubagentCompletionNotification::default(),
}),
..Config::default()
};
let (method, threshold, include_summary) =
crate::tui::notifications::settings(&config).expect("notification should be enabled");
assert_eq!(method, crate::tui::notifications::Method::Osc9);
assert_eq!(threshold, Duration::from_secs(45));
assert!(!include_summary);
}
#[test]
fn completed_turn_notification_uses_streaming_text() {
let app = create_test_app();
let msg = crate::tui::notifications::completed_turn_message(
&app,
"Hello there.\n\nWhat's next?",
false,
Duration::from_secs(12),
None,
);
assert_eq!(msg, "Turn complete\nHello there.\nWhat's next?");
}
#[test]
fn completed_turn_notification_falls_back_to_latest_assistant_message() {
let mut app = create_test_app();
app.api_messages.push(crate::models::Message {
role: "assistant".to_string(),
content: vec![crate::models::ContentBlock::Text {
text: "Earlier turn".to_string(),
cache_control: None,
}],
});
app.api_messages.push(crate::models::Message {
role: "user".to_string(),
content: vec![crate::models::ContentBlock::Text {
text: "next".to_string(),
cache_control: None,
}],
});
app.api_messages.push(crate::models::Message {
role: "assistant".to_string(),
content: vec![crate::models::ContentBlock::Text {
text: "Latest reply".to_string(),
cache_control: None,
}],
});
let msg = crate::tui::notifications::completed_turn_message(
&app,
"",
false,
Duration::from_secs(75),
None,
);
assert_eq!(msg, "Turn complete\nLatest reply");
}
#[test]
fn completed_turn_notification_falls_back_to_default_when_empty() {
let app = create_test_app();
let msg = crate::tui::notifications::completed_turn_message(
&app,
"",
false,
Duration::from_secs(5),
None,
);
assert_eq!(msg, "Turn complete");
}
#[test]
fn completed_turn_notification_truncates_long_text() {
let app = create_test_app();
let long = "a".repeat(500);
let msg = crate::tui::notifications::completed_turn_message(
&app,
&long,
false,
Duration::from_secs(5),
None,
);
assert!(msg.ends_with("..."));
let preview = msg
.strip_prefix("Turn complete\n")
.expect("notification should lead with completion status");
assert_eq!(preview.chars().count(), 363);
}
#[test]
fn completed_turn_notification_leads_with_user_locale() {
let mut app = create_test_app();
app.ui_locale = crate::localization::Locale::Ja;
let msg = crate::tui::notifications::completed_turn_message(
&app,
"完了しました。",
true,
Duration::from_secs(65),
None,
);
assert_eq!(msg, "ターン完了 (1m 5s)\n完了しました。");
}
#[test]
fn subagent_completion_notification_uses_summary_line_not_sentinel() {
let msg = crate::tui::notifications::subagent_completion_message(
crate::localization::Locale::En,
"agent_live",
"Finished the docs audit.\n<codewhale:subagent.done>{}</codewhale:subagent.done>",
false,
Duration::from_secs(42),
);
assert_eq!(
msg,
"Sub-agent complete\nagent_live: Finished the docs audit."
);
assert!(!msg.contains("codewhale:subagent.done"));
}
#[test]
fn subagent_completion_notification_can_include_elapsed_summary() {
let msg = crate::tui::notifications::subagent_completion_message(
crate::localization::Locale::En,
"agent_live",
"",
true,
Duration::from_secs(65),
);
assert_eq!(msg, "Sub-agent complete (1m 5s)\nagent_live");
}
#[test]
fn sanitize_stream_chunk_keeps_printable_and_drops_control_bytes() {
let cleaned = super::sanitize_stream_chunk("hello\tworld\n");
assert_eq!(cleaned, "hello\tworld\n", "tabs and newlines must survive");
let cleaned = super::sanitize_stream_chunk("text\u{1b}[2Jmore");
assert_eq!(cleaned, "text[2Jmore", "ESC byte must be filtered");
let cleaned = super::sanitize_stream_chunk("a\u{7}b\u{8}c\u{b}d\u{c}e");
assert_eq!(cleaned, "abcde");
let cleaned = super::sanitize_stream_chunk("line1\r\nline2");
assert_eq!(cleaned, "line1\nline2");
}
#[test]
fn sanitize_stream_chunk_preserves_unicode() {
let cjk = "\u{4f60}\u{597d}\u{ff0c}DeepSeek";
assert_eq!(super::sanitize_stream_chunk(cjk), cjk);
let emoji_and_accents = "caf\u{e9} \u{1f680} build";
assert_eq!(
super::sanitize_stream_chunk(emoji_and_accents),
emoji_and_accents,
);
}
#[test]
fn sanitize_stream_chunk_handles_empty_and_whitespace() {
assert_eq!(super::sanitize_stream_chunk(""), "");
assert_eq!(super::sanitize_stream_chunk(" "), " ");
assert_eq!(super::sanitize_stream_chunk("\u{1b}\u{7}\u{8}"), "");
}
#[test]
fn toast_stack_overlay_respects_composer_boundary() {
let _full_area = ratatui::prelude::Rect {
x: 0,
y: 0,
width: 80,
height: 16,
};
let composer_area = ratatui::prelude::Rect {
x: 0,
y: 10,
width: 80,
height: 5,
};
let footer_area = ratatui::prelude::Rect {
x: 0,
y: 15,
width: 80,
height: 1,
};
let composer_end = composer_area.y + composer_area.height;
let max_above = footer_area.y.saturating_sub(composer_end);
assert_eq!(
max_above, 0,
"with adjacent composer (rows 10-14) and footer (row 15) there is \
no gap, so the toast stack must report zero available rows"
);
let gap = footer_area.y.saturating_sub(composer_end);
assert!(
max_above <= gap,
"max_above ({max_above}) must never exceed the composer→footer gap ({gap})"
);
}
mod work_sidebar_projection_tests {
use super::*;
use crate::task_manager::{TaskStatus, TaskSummary};
use chrono::{Duration, TimeZone, Utc};
fn sample_task(
id: &str,
status: TaskStatus,
ended_at: Option<chrono::DateTime<Utc>>,
) -> TaskSummary {
TaskSummary {
id: id.to_string(),
status,
prompt_summary: format!("task {id}"),
model: "deepseek-v4-flash".to_string(),
mode: "agent".to_string(),
created_at: Utc.with_ymd_and_hms(2026, 5, 16, 12, 0, 0).unwrap(),
started_at: Some(Utc.with_ymd_and_hms(2026, 5, 16, 12, 1, 0).unwrap()),
ended_at,
duration_ms: ended_at.map(|_| 1_234),
hunt_verdict: None,
error: None,
thread_id: None,
turn_id: None,
}
}
#[test]
fn work_sidebar_hides_stale_completed_tasks_but_keeps_active_and_recent() {
let session_started_at = Utc.with_ymd_and_hms(2026, 5, 23, 10, 0, 0).unwrap();
let now = session_started_at + Duration::minutes(1);
let recent_ttl = Duration::hours(2);
let active_running = sample_task("active_run", TaskStatus::Running, None);
let active_queued = sample_task("active_q", TaskStatus::Queued, None);
let just_finished = sample_task(
"just_done",
TaskStatus::Completed,
Some(session_started_at + Duration::seconds(30)),
);
let recently_finished_before_session = sample_task(
"recent_done",
TaskStatus::Failed,
Some(session_started_at - Duration::minutes(15)),
);
let stale_completed = sample_task(
"stale_done",
TaskStatus::Completed,
Some(session_started_at - Duration::days(6)),
);
let stale_canceled = sample_task(
"stale_cancel",
TaskStatus::Canceled,
Some(session_started_at - Duration::days(7)),
);
let stale_failed = sample_task(
"stale_fail",
TaskStatus::Failed,
Some(session_started_at - Duration::days(3)),
);
let terminal_no_timestamp = sample_task("ghost", TaskStatus::Completed, None);
let tasks = vec![
active_running.clone(),
active_queued.clone(),
just_finished.clone(),
recently_finished_before_session.clone(),
stale_completed.clone(),
stale_canceled.clone(),
stale_failed.clone(),
terminal_no_timestamp.clone(),
];
let kept = select_work_sidebar_tasks(tasks, session_started_at, now, recent_ttl);
let kept_ids: Vec<&str> = kept.iter().map(|t| t.id.as_str()).collect();
assert!(
kept_ids.contains(&"active_run"),
"active running task must always show: {kept_ids:?}"
);
assert!(
kept_ids.contains(&"active_q"),
"active queued task must always show: {kept_ids:?}"
);
assert!(
kept_ids.contains(&"just_done"),
"task completed during the current session must show: {kept_ids:?}"
);
assert!(
kept_ids.contains(&"recent_done"),
"task completed within the recent TTL before session start must show: \
{kept_ids:?}"
);
assert!(
!kept_ids.contains(&"stale_done"),
"completed task from 6 days ago must be hidden (bug #1913): {kept_ids:?}"
);
assert!(
!kept_ids.contains(&"stale_cancel"),
"canceled task from 7 days ago must be hidden: {kept_ids:?}"
);
assert!(
!kept_ids.contains(&"stale_fail"),
"failed task from 3 days ago must be hidden: {kept_ids:?}"
);
assert!(
!kept_ids.contains(&"ghost"),
"terminal task missing ended_at must be hidden: {kept_ids:?}"
);
}
#[test]
fn work_sidebar_keeps_tasks_completed_at_session_boundary() {
let session_started_at = Utc.with_ymd_and_hms(2026, 5, 23, 10, 0, 0).unwrap();
let now = session_started_at + Duration::seconds(1);
let recent_ttl = Duration::hours(2);
let at_boundary = sample_task("boundary", TaskStatus::Completed, Some(session_started_at));
let kept =
select_work_sidebar_tasks(vec![at_boundary], session_started_at, now, recent_ttl);
assert_eq!(kept.len(), 1);
assert_eq!(kept[0].id, "boundary");
}
#[test]
fn receipt_summary_truncation_does_not_panic_on_multibyte_boundary() {
let prefix = "a".repeat(56); let summary = format!("{prefix}— rest of summary"); assert!(summary.len() > 60);
assert!(!summary.is_char_boundary(57));
let truncated = crate::utils::truncate_with_ellipsis(&summary, 60, "…");
assert_eq!(truncated, format!("{prefix}…"));
}
#[test]
fn shell_manager_cancel_transitions_task_to_not_running() {
let temp_dir = std::env::temp_dir().join(format!(
"codewhale-test-shell-cancel-{}",
std::process::id()
));
let _ = std::fs::create_dir_all(&temp_dir);
let mut manager = crate::tools::shell::ShellManager::new(temp_dir.clone());
let jobs = manager.list_jobs();
let running = jobs
.iter()
.filter(|j| matches!(j.status, crate::tools::shell::ShellStatus::Running))
.count();
assert_eq!(running, 0, "empty manager should have zero running jobs");
let results = manager.kill_running().unwrap();
assert!(
results.is_empty(),
"kill_running on empty should return empty"
);
let _ = std::fs::remove_dir_all(&temp_dir);
}
#[test]
fn task_panel_entry_roundtrips_status() {
let entry = crate::tui::app::TaskPanelEntry {
id: "test-id".to_string(),
status: "completed".to_string(),
prompt_summary: "echo hello".to_string(),
duration_ms: Some(100),
kind: crate::tui::app::TaskPanelEntryKind::Background,
stale: false,
elapsed_since_output_ms: None,
owner_agent_id: None,
owner_agent_name: None,
};
assert_eq!(entry.status, "completed");
assert_ne!(entry.status, "running");
}
}
#[test]
fn agent_progress_redraw_throttle_permits_first_and_spaced_events() {
let mut last_redraw = None;
let t0 = Instant::now();
assert!(
agent_progress_redraw_permitted(&mut last_redraw, t0),
"first progress event always repaints"
);
assert!(
!agent_progress_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(50)),
"events inside the 100ms window are throttled"
);
assert!(
!agent_progress_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(99)),
"throttled events must not advance the window"
);
assert!(
agent_progress_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(150)),
"events past the window repaint again"
);
}
#[test]
fn workflow_budget_redraw_throttle_matches_progress_pace() {
let mut last_redraw = None;
let t0 = Instant::now();
assert!(
workflow_budget_redraw_permitted(&mut last_redraw, t0),
"first budget tick may repaint"
);
assert!(
!workflow_budget_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(40)),
"budget churn inside 100ms is coalesced"
);
assert!(
workflow_budget_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(120)),
"budget ticks past the window repaint again"
);
}
#[test]
fn throttled_progress_event_does_not_cancel_other_events_redraw() {
let t0 = Instant::now();
let mut last_redraw = Some(t0);
let mut received_engine_event = true; let redraw_requested_before_event = received_engine_event;
received_engine_event = true; if !agent_progress_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(10)) {
received_engine_event = redraw_requested_before_event;
}
assert!(
received_engine_event,
"redraw owed to AgentSpawned must survive a throttled progress event"
);
let mut received_engine_event = false;
let redraw_requested_before_event = received_engine_event;
received_engine_event = true; if !agent_progress_redraw_permitted(&mut last_redraw, t0 + Duration::from_millis(20)) {
received_engine_event = redraw_requested_before_event;
}
assert!(
!received_engine_event,
"a lone throttled progress event must not trigger a repaint"
);
}
fn running_generic_tool_cell(name: &str) -> HistoryCell {
HistoryCell::Tool(ToolCell::Generic(GenericToolCell {
name: name.to_string(),
status: ToolStatus::Running,
input_summary: Some("action: run".to_string()),
output: None,
prompts: None,
spillover_path: None,
output_summary: None,
is_diff: false,
}))
}
#[test]
fn status_animation_ticks_for_lone_running_history_tool() {
let mut app = create_test_app();
app.history.push(running_generic_tool_cell("workflow"));
let history_motion = history_has_live_motion(&app.history);
let active_motion = active_cell_has_live_motion(&app);
assert!(
history_motion,
"running workflow tool should count as live motion"
);
assert!(
should_tick_status_animation(&app, false, history_motion, active_motion),
"a lone running tool in history must force timed redraws for the spout"
);
}
#[test]
fn status_animation_ticks_for_lone_running_active_tool() {
let mut app = create_test_app();
let mut active = ActiveCell::new();
active.push_tool("tool-1", running_generic_tool_cell("exec_shell"));
app.active_cell = Some(active);
let history_motion = history_has_live_motion(&app.history);
let active_motion = active_cell_has_live_motion(&app);
assert!(
active_motion,
"running active-cell tool should count as live motion"
);
assert!(
should_tick_status_animation(&app, false, history_motion, active_motion),
"a lone running active tool must force timed redraws for the spout"
);
}
#[test]
fn status_animation_stays_idle_without_live_motion() {
let app = create_test_app();
assert!(!history_has_live_motion(&app.history));
assert!(!active_cell_has_live_motion(&app));
assert!(
!should_tick_status_animation(&app, false, false, false),
"idle sessions should not wake the animation timer"
);
}
#[test]
fn subagent_completion_notification_modes_gate_correctly() {
use crate::config::SubagentCompletionNotification as Mode;
assert!(!should_notify_subagent_completion(Mode::Off, false, false));
assert!(!should_notify_subagent_completion(Mode::Off, true, true));
assert!(should_notify_subagent_completion(Mode::Always, true, true));
assert!(should_notify_subagent_completion(
Mode::Always,
false,
false
));
assert!(should_notify_subagent_completion(
Mode::FinalOnly,
false,
false
));
assert!(
!should_notify_subagent_completion(Mode::FinalOnly, true, false),
"final-only stays quiet while other subagents run"
);
assert!(
!should_notify_subagent_completion(Mode::FinalOnly, false, true),
"final-only stays quiet while a workflow run is active"
);
}
#[test]
fn workflow_tool_is_running_detects_running_workflow_cell() {
let mut app = create_test_app();
assert!(!workflow_tool_is_running(&app));
app.history.push(running_generic_tool_cell("read_file"));
assert!(
!workflow_tool_is_running(&app),
"a non-workflow running tool must not count as a workflow run"
);
app.history.push(running_generic_tool_cell("workflow"));
assert!(workflow_tool_is_running(&app));
}
#[test]
fn agent_progress_redraw_coalesces_once_per_agent_per_drain() {
let t0 = Instant::now();
let mut last_redraw = None;
let mut seen_agents = HashSet::new();
assert!(
agent_progress_redraw_permitted_for_drain(
&mut last_redraw,
&mut seen_agents,
"agent-a",
t0,
),
"first progress event for an agent in a drain may repaint"
);
assert!(
!agent_progress_redraw_permitted_for_drain(
&mut last_redraw,
&mut seen_agents,
"agent-a",
t0 + Duration::from_millis(150),
),
"later progress for the same agent in the same drain is coalesced"
);
let mut next_drain_seen_agents = HashSet::new();
assert!(
agent_progress_redraw_permitted_for_drain(
&mut last_redraw,
&mut next_drain_seen_agents,
"agent-a",
t0 + Duration::from_millis(150),
),
"a later drain can repaint that agent again after the throttle window"
);
}
#[test]
fn six_worker_progress_storm_keeps_input_render_and_cancel_live() {
const _: () = assert!(
MAX_ENGINE_EVENTS_PER_DRAIN >= 8 && MAX_ENGINE_EVENTS_PER_DRAIN <= 16,
"engine event drains must stay small so terminal input is polled frequently during long runs"
);
let t0 = Instant::now();
let mut last_redraw = None;
let mut seen_agents = HashSet::new();
let mut redraws = 0usize;
let mut received_engine_event = false;
for burst in 0..80 {
for worker in 0..6 {
let agent_id = format!("agent-{worker}");
let redraw_requested_before_event = received_engine_event;
received_engine_event = true;
if agent_progress_redraw_permitted_for_drain(
&mut last_redraw,
&mut seen_agents,
&agent_id,
t0 + Duration::from_millis(burst * 2 + worker),
) {
redraws += 1;
} else {
received_engine_event = redraw_requested_before_event;
}
}
}
assert_eq!(
seen_agents.len(),
6,
"storm should observe all six workers in one drain"
);
assert!(
(1..=6).contains(&redraws),
"progress storm must request a bounded redraw count, got {redraws}"
);
assert!(
received_engine_event,
"at least one bounded redraw should keep rendering live"
);
let (tx, rx) = std::sync::mpsc::channel();
tx.send(TerminalInputMessage::Event(Event::Key(KeyEvent::new(
KeyCode::Char('c'),
KeyModifiers::CONTROL,
))))
.expect("send key event");
let input = TerminalInputPump {
rx,
stop: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused_ack: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
handle: None,
last_alive_at: std::cell::Cell::new(Instant::now()),
};
let mut pending_terminal_events = VecDeque::new();
let event = next_terminal_event(
&input,
&mut pending_terminal_events,
Duration::from_millis(1),
)
.expect("terminal event read")
.expect("queued key event");
assert!(
matches!(
event,
Event::Key(key)
if key.code == KeyCode::Char('c')
&& key.modifiers.contains(KeyModifiers::CONTROL)
),
"input pump channel should deliver input despite progress noise"
);
let mut app = create_test_app();
app.is_loading = true;
app.runtime_turn_status = Some("in_progress".to_string());
assert_eq!(next_escape_action(&app, false), EscapeAction::CancelRequest);
assert_eq!(ctrl_c_disposition(&app), CtrlCDisposition::CancelTurn);
}
#[test]
fn terminal_input_child_pause_drains_codewhale_events_before_editor_handoff() {
let (tx, rx) = std::sync::mpsc::channel();
tx.send(TerminalInputMessage::Event(Event::Key(KeyEvent::new(
KeyCode::Char('x'),
KeyModifiers::NONE,
))))
.expect("send buffered key event");
tx.send(TerminalInputMessage::Heartbeat)
.expect("send buffered heartbeat");
tx.send(TerminalInputMessage::Event(Event::Key(KeyEvent::new(
KeyCode::Char('y'),
KeyModifiers::NONE,
))))
.expect("send second buffered key event");
let input = TerminalInputPump {
rx,
stop: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused_ack: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
handle: None,
last_alive_at: std::cell::Cell::new(Instant::now()),
};
let mut pending_terminal_events = VecDeque::from([Event::Key(KeyEvent::new(
KeyCode::Char('z'),
KeyModifiers::NONE,
))]);
input
.pause_for_child_terminal()
.expect("synthetic pump can pause");
drain_terminal_input_queue(&input, &mut pending_terminal_events)
.expect("queued terminal events drain before launching child editor");
assert!(
pending_terminal_events.is_empty(),
"pending CodeWhale terminal events must not leak into the editor handoff"
);
assert!(
input.try_recv().expect("drained channel").is_none(),
"input pump channel should be empty after the editor handoff drain"
);
input.resume_after_child_terminal();
assert!(!input.paused.load(std::sync::atomic::Ordering::Acquire));
assert!(!input.paused_ack.load(std::sync::atomic::Ordering::Acquire));
}
#[test]
fn input_pump_restart_detaches_wedged_thread_and_installs_fresh_parts() {
let (block_tx, block_rx) = std::sync::mpsc::channel::<()>();
let wedged = std::thread::spawn(move || {
let _ = block_rx.recv();
});
let (old_tx, old_rx) = std::sync::mpsc::channel();
let old_stop = std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false));
let mut pump = TerminalInputPump {
rx: old_rx,
stop: std::sync::Arc::clone(&old_stop),
paused: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused_ack: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
handle: Some(wedged),
last_alive_at: std::cell::Cell::new(
Instant::now()
.checked_sub(Duration::from_secs(30))
.unwrap_or_else(Instant::now),
),
};
pump.detach_current_thread();
assert!(
old_stop.load(std::sync::atomic::Ordering::Acquire),
"detach must flag the old pump thread to stop"
);
assert!(
pump.handle.is_none(),
"detach must drop the wedged handle without joining it"
);
let (new_tx, new_rx) = std::sync::mpsc::channel();
pump.install_parts(TerminalInputPumpParts {
rx: new_rx,
stop: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
paused_ack: std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false)),
handle: std::thread::spawn(|| {}),
});
assert!(pump.handle.is_some(), "fresh pump thread must be adopted");
assert!(
!pump.stop.load(std::sync::atomic::Ordering::Acquire),
"fresh pump must not start in a stopped state"
);
assert!(
pump.stalled_for(Instant::now()) < Duration::from_secs(30),
"install must reset the liveness clock"
);
new_tx
.send(TerminalInputMessage::Event(Event::Key(KeyEvent::new(
KeyCode::Char('k'),
KeyModifiers::NONE,
))))
.expect("send event on replacement channel");
let event = pump
.try_recv()
.expect("replacement channel readable")
.expect("replacement channel delivers the event");
assert!(
matches!(event, Event::Key(key) if key.code == KeyCode::Char('k')),
"restarted pump must deliver events from the replacement channel"
);
assert!(
old_tx.send(TerminalInputMessage::Heartbeat).is_err(),
"old pump channel must be disconnected after restart"
);
drop(block_tx); }
#[test]
fn raw_mode_probe_handshake_elects_exactly_one_side_sequentially() {
let enabled = std::sync::atomic::AtomicBool::new(false);
let abandoned = std::sync::atomic::AtomicBool::new(false);
let task_disables = raw_mode_probe_handshake(&enabled, &abandoned);
let caller_disables = raw_mode_probe_handshake(&abandoned, &enabled);
assert!(!task_disables, "task ran first, so it must not disable");
assert!(
caller_disables,
"timed-out caller must undo the late enable"
);
let enabled = std::sync::atomic::AtomicBool::new(false);
let abandoned = std::sync::atomic::AtomicBool::new(false);
let caller_disables = raw_mode_probe_handshake(&abandoned, &enabled);
let task_disables = raw_mode_probe_handshake(&enabled, &abandoned);
assert!(!caller_disables, "caller ran first, so it must not disable");
assert!(
task_disables,
"late-finishing task must undo its own enable"
);
}
#[test]
fn raw_mode_probe_handshake_never_leaks_under_concurrent_race() {
for _ in 0..200 {
let enabled = std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false));
let abandoned = std::sync::Arc::new(std::sync::atomic::AtomicBool::new(false));
let task_enabled = std::sync::Arc::clone(&enabled);
let task_abandoned = std::sync::Arc::clone(&abandoned);
let task =
std::thread::spawn(move || raw_mode_probe_handshake(&task_enabled, &task_abandoned));
let caller_disables = raw_mode_probe_handshake(&abandoned, &enabled);
let task_disables = task.join().expect("handshake task side");
assert!(
task_disables || caller_disables,
"at least one side must take responsibility for disabling raw mode"
);
}
}
#[test]
fn backtrack_cut_index_skips_tool_result_user_messages() {
use crate::models::{ContentBlock, Message};
let msgs = vec![
Message {
role: "user".into(),
content: vec![ContentBlock::Text {
text: "first".into(),
cache_control: None,
}],
},
Message {
role: "assistant".into(),
content: vec![ContentBlock::ToolUse {
id: "t1".into(),
name: "read_file".into(),
input: serde_json::json!({"path":"x"}),
caller: None,
}],
},
Message {
role: "user".into(),
content: vec![ContentBlock::ToolResult {
tool_use_id: "t1".into(),
content: "data".into(),
is_error: None,
content_blocks: None,
}],
},
Message {
role: "assistant".into(),
content: vec![ContentBlock::Text {
text: "answer".into(),
cache_control: None,
}],
},
Message {
role: "user".into(),
content: vec![ContentBlock::Text {
text: "second".into(),
cache_control: None,
}],
},
];
assert_eq!(super::backtrack_api_cut_index(&msgs, 0), Some(4));
assert_eq!(super::backtrack_api_cut_index(&msgs, 1), Some(0));
assert_eq!(super::backtrack_api_cut_index(&msgs, 2), None);
}