ISAR/rusqlite
3
0
Fork 0
mirror of https://github.com/isar/rusqlite.git synced 2025-04-05 23:07:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1480 from gwenn/test_busy_handler

Browse Source 
Fix test_busy_handler
This commit is contained in:
gwenn 2024-03-23 18:01:04 +01:00 committed by GitHub
commit e66397b410
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/busy.rs
View File

@ -132,38 +132,27 @@ mod test {
} }
#[test] #[test]
#[ignore] // FIXME: unstable fn test_busy_handler() -> Result<()> {
fn test_busy_handler() {
static CALLED: AtomicBool = AtomicBool::new(false); static CALLED: AtomicBool = AtomicBool::new(false);
fn busy_handler(_: i32) -> bool { fn busy_handler(n: i32) -> bool {
CALLED.store(true, Ordering::Relaxed); if n > 2 {
thread::sleep(Duration::from_millis(100)); false
true } else {
CALLED.swap(true, Ordering::Relaxed)
}
} }
let temp_dir = tempfile::tempdir().unwrap(); let temp_dir = tempfile::tempdir().unwrap();
let path = temp_dir.path().join("test.db3"); let path = temp_dir.path().join("busy-handler.db3");
let db2 = Connection::open(&path).unwrap(); let db1 = Connection::open(&path)?;
db2.busy_handler(Some(busy_handler)).unwrap(); db1.execute_batch("CREATE TABLE IF NOT EXISTS t(a)")?;
let db2 = Connection::open(&path)?;
let (rx, tx) = sync_channel(0); db2.busy_handler(Some(busy_handler))?;
let child = thread::spawn(move || { db1.execute_batch("BEGIN EXCLUSIVE")?;
let mut db1 = Connection::open(&path).unwrap(); let err = db2.prepare("SELECT * FROM t").unwrap_err();
let tx1 = db1 assert_eq!(err.sqlite_error_code(), Some(ErrorCode::DatabaseBusy));
.transaction_with_behavior(TransactionBehavior::Exclusive)
.unwrap();
rx.send(1).unwrap();
thread::sleep(Duration::from_millis(100));
tx1.rollback().unwrap();
});
assert_eq!(tx.recv().unwrap(), 1);
let _ = db2
.query_row("PRAGMA schema_version", [], |row| row.get::<_, i32>(0))
.expect("unexpected error");
assert!(CALLED.load(Ordering::Relaxed)); assert!(CALLED.load(Ordering::Relaxed));
Ok(())
child.join().unwrap();
} }
} }