From 44857cdd17335a649d43540028df2dda98c7fe98 Mon Sep 17 00:00:00 2001 From: Pere Diaz Bou Date: Tue, 11 Feb 2025 09:30:55 +0100 Subject: [PATCH] fix insert test --- core/storage/btree.rs | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/core/storage/btree.rs b/core/storage/btree.rs index ae17942ab..79f0e5166 100644 --- a/core/storage/btree.rs +++ b/core/storage/btree.rs @@ -2854,17 +2854,23 @@ mod tests { assert_eq!(buf, payload); } - fn add_record(page: &mut PageContent, record: OwnedRecord, db: &Arc) -> Vec { + fn add_record( + id: usize, + pos: usize, + page: &mut PageContent, + record: OwnedRecord, + db: &Arc, + ) -> Vec { let mut payload: Vec = Vec::new(); fill_cell_payload( page.page_type(), - Some(1), + Some(id as u64), &mut payload, &record, 4096, db.pager.clone(), ); - insert_into_cell(page, &payload, 0, 4096); + insert_into_cell(page, &payload, pos, 4096); payload } @@ -2875,7 +2881,7 @@ mod tests { let page = page.get_contents(); let header_size = 8; let record = OwnedRecord::new([OwnedValue::Integer(1)].to_vec()); - let payload = add_record(page, record, &db); + let payload = add_record(1, 0, page, record, &db); assert_eq!(page.cell_count(), 1); let free = compute_free_space(page, 4096); assert_eq!(free, 4096 - payload.len() as u16 - 2 - header_size); @@ -2894,8 +2900,8 @@ mod tests { let mut total_size = 0; let mut payloads = Vec::new(); for i in 0..10 { - let record = OwnedRecord::new([OwnedValue::Integer(1)].to_vec()); - let payload = add_record(page, record, &db); + let record = OwnedRecord::new([OwnedValue::Integer(i as i64)].to_vec()); + let payload = add_record(i, i, page, record, &db); assert_eq!(page.cell_count(), i + 1); let free = compute_free_space(page, 4096); total_size += payload.len() as u16 + 2;