|
| 1 | +//! 批量删除会话单测(批 4;AAA 模式 + :memory: 隔离库,绝不触碰真实数据)。 |
| 2 | +//! |
| 3 | +//! @ai-context: 由 commands_session_delete.rs 以 #[cfg(test)] #[path] 引入; |
| 4 | +//! 被测对象 run_batch_delete 为命令核心(注入 Db,无 Tauri 态), |
| 5 | +//! 域广播在命令宏入口、不在此测试(与批量转笔记测试同构)。 |
| 6 | +//! @ai-context: 原子性用 SQLite RAISE(ABORT) 触发器模拟中途失败——真实语句 |
| 7 | +//! 失败难以在 :memory: 造出,触发器是 schema 层最接近的失败源。 |
| 8 | +
|
| 9 | +use crate::commands_session_delete::run_batch_delete; |
| 10 | +use crate::db::Db; |
| 11 | +use crate::types::{NewNote, NewSession, NewSessionSegment}; |
| 12 | + |
| 13 | +fn mem_db() -> Db { |
| 14 | + Db::open(":memory:").expect("open in-memory db") |
| 15 | +} |
| 16 | + |
| 17 | +/// 造一个已结束(finished)会话(带转写段 = 有内容)。 |
| 18 | +fn finished_session(db: &Db, title: &str) -> i64 { |
| 19 | + let s = db |
| 20 | + .create_session(&NewSession { |
| 21 | + title: title.into(), |
| 22 | + source_window: None, |
| 23 | + profile: None, |
| 24 | + kind: None, |
| 25 | + }) |
| 26 | + .expect("create session"); |
| 27 | + db.finish_session(s.id).expect("finish session"); |
| 28 | + db.add_segment(&NewSessionSegment { |
| 29 | + session_id: s.id, |
| 30 | + start_ms: 0, |
| 31 | + end_ms: 1000, |
| 32 | + text: format!("{} 的转写", title), |
| 33 | + source: "asr".into(), |
| 34 | + confidence: Some(0.9), |
| 35 | + volume: None, |
| 36 | + speech_rate: None, |
| 37 | + pause_ms: None, |
| 38 | + speaker: None, |
| 39 | + }) |
| 40 | + .expect("add segment"); |
| 41 | + s.id |
| 42 | +} |
| 43 | + |
| 44 | +/// 直插子表行(转写段由 finished_session 已插一条;此处补 OCR 块/产物块/ |
| 45 | +/// 信号事件各一——加段共四张表四行,验证删除外键级联全清)。 |
| 46 | +fn seed_child_rows(db: &Db, session_id: i64) { |
| 47 | + db.with_conn(|conn| { |
| 48 | + conn.execute( |
| 49 | + "INSERT INTO session_ocr_blocks (session_id, timestamp_ms, text, score, region) |
| 50 | + VALUES (?1, 0, 'cascade ocr', 0.9, 'full')", |
| 51 | + rusqlite::params![session_id], |
| 52 | + )?; |
| 53 | + conn.execute( |
| 54 | + "INSERT INTO artifact_blocks (session_id, kind, payload_json, block_order) |
| 55 | + VALUES (?1, 'chapter', '{}', 0)", |
| 56 | + rusqlite::params![session_id], |
| 57 | + )?; |
| 58 | + conn.execute( |
| 59 | + "INSERT INTO session_events (session_id, kind, timestamp_ms) |
| 60 | + VALUES (?1, 'app_foreground', 0)", |
| 61 | + rusqlite::params![session_id], |
| 62 | + )?; |
| 63 | + Ok(()) |
| 64 | + }) |
| 65 | + .expect("seed child rows"); |
| 66 | +} |
| 67 | + |
| 68 | +/// 子表残留行数(>0 说明级联未生效)。 |
| 69 | +fn child_row_count(db: &Db, session_id: i64) -> i64 { |
| 70 | + db.with_conn(|conn| { |
| 71 | + let mut count = 0i64; |
| 72 | + for table in ["session_segments", "session_ocr_blocks", "artifact_blocks", "session_events"] { |
| 73 | + let sql = format!("SELECT COUNT(*) FROM {} WHERE session_id = ?1", table); |
| 74 | + count += conn.query_row(&sql, rusqlite::params![session_id], |row| { |
| 75 | + row.get::<_, i64>(0) |
| 76 | + })?; |
| 77 | + } |
| 78 | + Ok(count) |
| 79 | + }) |
| 80 | + .expect("count child rows") |
| 81 | +} |
| 82 | + |
| 83 | +#[test] |
| 84 | +fn batch_delete_removes_all_and_cascades_children() { |
| 85 | + // Arrange:三个会话(两个待删),各带四张子表行 |
| 86 | + let db = mem_db(); |
| 87 | + let a = finished_session(&db, "A课"); |
| 88 | + let b = finished_session(&db, "B课"); |
| 89 | + let keep = finished_session(&db, "保留课"); |
| 90 | + for id in [a, b, keep] { |
| 91 | + seed_child_rows(&db, id); |
| 92 | + } |
| 93 | + assert_eq!(child_row_count(&db, a), 4); |
| 94 | + // Act |
| 95 | + let result = run_batch_delete(&db, vec![a, b]).expect("batch delete"); |
| 96 | + // Assert:计数精确;会话与子表全清;未选会话原样保留 |
| 97 | + assert_eq!(result.deleted, 2); |
| 98 | + assert!(db.get_session(a).expect("db").is_none()); |
| 99 | + assert!(db.get_session(b).expect("db").is_none()); |
| 100 | + assert_eq!(child_row_count(&db, a), 0); |
| 101 | + assert_eq!(child_row_count(&db, b), 0); |
| 102 | + assert_eq!(child_row_count(&db, keep), 4); |
| 103 | + assert!(db.get_session(keep).expect("db").is_some()); |
| 104 | +} |
| 105 | + |
| 106 | +#[test] |
| 107 | +fn batch_delete_atomic_rollback_on_mid_failure() { |
| 108 | + // Arrange:A 在触发器守卫之前、B 命中守卫——任何实现若逐条提交都会半删 |
| 109 | + let db = mem_db(); |
| 110 | + let a = finished_session(&db, "A课"); |
| 111 | + let b = finished_session(&db, "B课"); |
| 112 | + db.with_conn(|conn| { |
| 113 | + // 触发器 DDL 不支持绑定变量——b 为刚由本库 create_session 生成的 i64, |
| 114 | + // 非用户输入,直接内联安全(防御注释:杜绝字符串拼接的例外场景) |
| 115 | + let sql = format!( |
| 116 | + "CREATE TRIGGER guard_delete BEFORE DELETE ON sessions |
| 117 | + WHEN OLD.id = {} BEGIN SELECT RAISE(ABORT, 'guard: session busy'); END", |
| 118 | + b, |
| 119 | + ); |
| 120 | + conn.execute_batch(&sql)?; |
| 121 | + Ok(()) |
| 122 | + }) |
| 123 | + .expect("create guard trigger"); |
| 124 | + // Act |
| 125 | + let err = run_batch_delete(&db, vec![a, b]).expect_err("must fail atomically"); |
| 126 | + // Assert:整体报错且 A 也未被删除(单事务回滚——无半删残留) |
| 127 | + assert!(err.contains("guard: session busy"), "err={err}"); |
| 128 | + assert!(db.get_session(a).expect("db").is_some(), "A 应随事务回滚保留"); |
| 129 | + assert!(db.get_session(b).expect("db").is_some()); |
| 130 | +} |
| 131 | + |
| 132 | +#[test] |
| 133 | +fn batch_delete_keeps_note_but_unbinds_session() { |
| 134 | + // Arrange:已转笔记的会话(笔记是用户资产——删会话只断关联不删笔记) |
| 135 | + let db = mem_db(); |
| 136 | + let id = finished_session(&db, "已转课"); |
| 137 | + let note = db |
| 138 | + .create_note(&NewNote { |
| 139 | + title: "转换笔记".into(), |
| 140 | + content: "x".into(), |
| 141 | + source: "classroom".into(), |
| 142 | + session_id: Some(id), |
| 143 | + rule_version: None, |
| 144 | + purify_stats: None, |
| 145 | + tags: None, |
| 146 | + properties: None, |
| 147 | + group_id: None, |
| 148 | + }) |
| 149 | + .expect("create note"); |
| 150 | + // Act |
| 151 | + run_batch_delete(&db, vec![id]).expect("batch delete"); |
| 152 | + // Assert:笔记保留且 session_id 已置空(外键 SET NULL 语义) |
| 153 | + let kept = db.get_note(note.id).expect("db").expect("note kept"); |
| 154 | + assert_eq!(kept.session_id, None); |
| 155 | + assert!(db.find_note_by_session(id).expect("db").is_none()); |
| 156 | +} |
| 157 | + |
| 158 | +#[test] |
| 159 | +fn batch_delete_coexists_with_batch_convert() { |
| 160 | + // Arrange:与批量转笔记同一编排链走一遍(先转后删——真实页面动线) |
| 161 | + let db = mem_db(); |
| 162 | + let id = finished_session(&db, "先转后删课"); |
| 163 | + let converted = crate::commands_session_note::run_batch_conversion( |
| 164 | + &db, |
| 165 | + &crate::ui_junk::UiJunkList::defaults(), |
| 166 | + &crate::note_filter::PurifyEnv::default(), |
| 167 | + std::path::Path::new("."), |
| 168 | + vec![id], |
| 169 | + ) |
| 170 | + .expect("batch convert"); |
| 171 | + assert_eq!(converted.converted.len(), 1); |
| 172 | + let note_id = converted.converted[0].note_id; |
| 173 | + // Act:删除已转会话 |
| 174 | + run_batch_delete(&db, vec![id]).expect("batch delete"); |
| 175 | + // Assert:笔记留存且解绑;会话已删 |
| 176 | + let kept = db.get_note(note_id).expect("db").expect("note kept"); |
| 177 | + assert_eq!(kept.session_id, None); |
| 178 | + assert!(db.get_session(id).expect("db").is_none()); |
| 179 | +} |
| 180 | + |
| 181 | +#[test] |
| 182 | +fn batch_delete_validation_and_counting() { |
| 183 | + let db = mem_db(); |
| 184 | + let a = finished_session(&db, "A课"); |
| 185 | + // 空 ids → 报错 |
| 186 | + let err = run_batch_delete(&db, vec![]).expect_err("empty must err"); |
| 187 | + assert!(err.contains("不能为空"), "err={err}"); |
| 188 | + // 非正 id → 报错(与单条 delete_session 同口径) |
| 189 | + let err = run_batch_delete(&db, vec![0]).expect_err("invalid must err"); |
| 190 | + assert!(err.contains("无效的会话 id"), "err={err}"); |
| 191 | + // 超上限 → 报错(201 > 200) |
| 192 | + let over: Vec<i64> = (1..=201).collect(); |
| 193 | + let err = run_batch_delete(&db, over).expect_err("over-limit must err"); |
| 194 | + assert!(err.contains("上限"), "err={err}"); |
| 195 | + // 重复 id 静默去重 + 不存在的 id 宽容(计 0,不报错——与单条一致) |
| 196 | + let result = run_batch_delete(&db, vec![a, a, 99_999]).expect("dedupe & missing ok"); |
| 197 | + assert_eq!(result.deleted, 1); |
| 198 | + assert!(db.get_session(a).expect("db").is_none()); |
| 199 | +} |
0 commit comments