aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-02-23 08:55:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

adjust fuzz test

Browse Source
This commit is contained in:
Nikita Sivukhin
2025-02-08 21:25:21 +04:00
parent 75e2f01ec4
commit 3557c8aada
1 changed files with 78 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

97
core/storage/btree.rs
View File

@@ -2459,26 +2459,47 @@ mod tests {
}
#[test]
pub fn btree_insert_fuzz() {
pub fn btree_insert_fuzz_ex() {
let _ = env_logger::init();
let (pager, root_page) = empty_btree();
let mut cursor = BTreeCursor::new(pager.clone(), root_page);
let mut keys = Vec::new();
let mut rng = ChaCha8Rng::seed_from_u64(0);
for _ in 0..16 {
let size = (rng.next_u64() % 4096) as usize;
let key = (rng.next_u64() % (1 << 30)) as i64;
keys.push(key);
log::info!("INSERT INTO t VALUES ({}, randomblob({}));", key, size);
let key = OwnedValue::Integer(key);
let value = Record::new(vec![OwnedValue::Blob(Rc::new(vec![0; size]))]);
cursor.insert(&key, &value, false).unwrap();
log::info!(
"=========== btree ===========\n{}\n\n",
format_btree(pager.clone(), root_page, 0)
);
for key in keys.iter() {
for sequence in [
&[
(293471650, 2452),
(163608869, 627),
(544576229, 464),
(705823748, 3441),
]
.as_slice(),
&[
(987283511, 2924),
(261851260, 1766),
(343847101, 1657),
(315844794, 572),
]
.as_slice(),
&[
(987283511, 2924),
(261851260, 1766),
(343847101, 1657),
(315844794, 572),
(649272840, 1632),
(723398505, 3140),
(334416967, 3874),
]
.as_slice(),
] {
let (pager, root_page) = empty_btree();
let mut cursor = BTreeCursor::new(pager.clone(), root_page);
for (key, size) in sequence.iter() {
let key = OwnedValue::Integer(*key);
let value = Record::new(vec![OwnedValue::Blob(Rc::new(vec![0; *size]))]);
log::info!("insert key:{}", key);
cursor.insert(&key, &value, false).unwrap();
log::info!(
"=========== btree ===========\n{}\n\n",
format_btree(pager.clone(), root_page, 0)
);
}
for (key, _) in sequence.iter() {
let seek_key = SeekKey::TableRowId(*key as u64);
assert!(
matches!(
@@ -2492,6 +2513,44 @@ mod tests {
}
}
#[test]
pub fn btree_insert_fuzz_run() {
let _ = env_logger::init();
let mut rng = ChaCha8Rng::seed_from_u64(0);
for _ in 0..128 {
let (pager, root_page) = empty_btree();
let mut cursor = BTreeCursor::new(pager.clone(), root_page);
let mut keys = Vec::new();
let seed = rng.next_u64();
log::info!("seed: {}", seed);
let mut rng = ChaCha8Rng::seed_from_u64(seed);
for _ in 0..16 {
let size = (rng.next_u64() % 4096) as usize;
let key = (rng.next_u64() % (1 << 30)) as i64;
keys.push(key);
log::info!("INSERT INTO t VALUES ({}, randomblob({}));", key, size);
let key = OwnedValue::Integer(key);
let value = Record::new(vec![OwnedValue::Blob(Rc::new(vec![0; size]))]);
cursor.insert(&key, &value, false).unwrap();
log::info!(
"=========== btree ===========\n{}\n\n",
format_btree(pager.clone(), root_page, 0)
);
for key in keys.iter() {
let seek_key = SeekKey::TableRowId(*key as u64);
assert!(
matches!(
cursor.seek(seek_key, SeekOp::EQ).unwrap(),
CursorResult::Ok(true)
),
"key {} is not found",
key
);
}
}
}
}
#[allow(clippy::arc_with_non_send_sync)]
fn setup_test_env(database_size: u32) -> (Rc<Pager>, Rc<RefCell<DatabaseHeader>>) {
let page_size = 512;