mirror of
https://github.com/aljazceru/turso.git
synced 2025-12-25 20:14:21 +01:00
They're both using the same database file which is wrong (tests run in parallel). But more importantly, they test almost nothing, and we have a better checkpoint test already.
1322 lines
49 KiB
Rust
1322 lines
49 KiB
Rust
#![allow(non_camel_case_types)]
|
|
#![allow(dead_code)]
|
|
use std::ptr;
|
|
|
|
#[repr(C)]
|
|
struct sqlite3 {
|
|
_private: [u8; 0],
|
|
}
|
|
|
|
#[repr(C)]
|
|
struct sqlite3_stmt {
|
|
_private: [u8; 0],
|
|
}
|
|
|
|
#[cfg_attr(not(feature = "sqlite3"), link(name = "turso_sqlite3"))]
|
|
#[cfg_attr(feature = "sqlite3", link(name = "sqlite3"))]
|
|
extern "C" {
|
|
fn sqlite3_libversion() -> *const libc::c_char;
|
|
fn sqlite3_libversion_number() -> i32;
|
|
fn sqlite3_close(db: *mut sqlite3) -> i32;
|
|
fn sqlite3_open(filename: *const libc::c_char, db: *mut *mut sqlite3) -> i32;
|
|
fn sqlite3_db_filename(db: *mut sqlite3, db_name: *const libc::c_char) -> *const libc::c_char;
|
|
fn sqlite3_prepare_v2(
|
|
db: *mut sqlite3,
|
|
sql: *const libc::c_char,
|
|
n_bytes: i32,
|
|
stmt: *mut *mut sqlite3_stmt,
|
|
tail: *mut *const libc::c_char,
|
|
) -> i32;
|
|
fn sqlite3_step(stmt: *mut sqlite3_stmt) -> i32;
|
|
fn sqlite3_reset(stmt: *mut sqlite3_stmt) -> i32;
|
|
fn sqlite3_finalize(stmt: *mut sqlite3_stmt) -> i32;
|
|
fn sqlite3_wal_checkpoint(db: *mut sqlite3, db_name: *const libc::c_char) -> i32;
|
|
fn sqlite3_wal_checkpoint_v2(
|
|
db: *mut sqlite3,
|
|
db_name: *const libc::c_char,
|
|
mode: i32,
|
|
log_size: *mut i32,
|
|
checkpoint_count: *mut i32,
|
|
) -> i32;
|
|
fn sqlite3_column_int64(stmt: *mut sqlite3_stmt, idx: i32) -> i64;
|
|
fn libsql_wal_frame_count(db: *mut sqlite3, p_frame_count: *mut u32) -> i32;
|
|
fn libsql_wal_get_frame(
|
|
db: *mut sqlite3,
|
|
frame_no: u32,
|
|
p_frame: *mut u8,
|
|
frame_len: u32,
|
|
) -> i32;
|
|
fn libsql_wal_disable_checkpoint(db: *mut sqlite3) -> i32;
|
|
fn sqlite3_column_int(stmt: *mut sqlite3_stmt, idx: i32) -> i64;
|
|
fn sqlite3_next_stmt(db: *mut sqlite3, stmt: *mut sqlite3_stmt) -> *mut sqlite3_stmt;
|
|
fn sqlite3_bind_int(stmt: *mut sqlite3_stmt, idx: i32, val: i64) -> i32;
|
|
fn sqlite3_bind_parameter_count(stmt: *mut sqlite3_stmt) -> i32;
|
|
fn sqlite3_bind_parameter_name(stmt: *mut sqlite3_stmt, idx: i32) -> *const libc::c_char;
|
|
fn sqlite3_bind_parameter_index(stmt: *mut sqlite3_stmt, name: *const libc::c_char) -> i32;
|
|
fn sqlite3_clear_bindings(stmt: *mut sqlite3_stmt) -> i32;
|
|
fn sqlite3_column_name(stmt: *mut sqlite3_stmt, idx: i32) -> *const libc::c_char;
|
|
fn sqlite3_last_insert_rowid(db: *mut sqlite3) -> i32;
|
|
fn sqlite3_column_count(stmt: *mut sqlite3_stmt) -> i32;
|
|
fn sqlite3_bind_text(
|
|
stmt: *mut sqlite3_stmt,
|
|
idx: i32,
|
|
text: *const libc::c_char,
|
|
len: i32,
|
|
destructor: Option<unsafe extern "C" fn(*mut libc::c_void)>,
|
|
) -> i32;
|
|
fn sqlite3_bind_blob(
|
|
stmt: *mut sqlite3_stmt,
|
|
idx: i32,
|
|
blob: *const libc::c_void,
|
|
len: i32,
|
|
destructor: Option<unsafe extern "C" fn(*mut libc::c_void)>,
|
|
) -> i32;
|
|
fn sqlite3_column_text(stmt: *mut sqlite3_stmt, idx: i32) -> *const libc::c_char;
|
|
fn sqlite3_column_bytes(stmt: *mut sqlite3_stmt, idx: i32) -> i64;
|
|
fn sqlite3_column_blob(stmt: *mut sqlite3_stmt, idx: i32) -> *const libc::c_void;
|
|
fn sqlite3_column_type(stmt: *mut sqlite3_stmt, idx: i32) -> i32;
|
|
fn sqlite3_column_decltype(stmt: *mut sqlite3_stmt, idx: i32) -> *const libc::c_char;
|
|
fn sqlite3_get_autocommit(db: *mut sqlite3) -> i32;
|
|
}
|
|
|
|
const SQLITE_OK: i32 = 0;
|
|
const SQLITE_CANTOPEN: i32 = 14;
|
|
const SQLITE_ROW: i32 = 100;
|
|
const SQLITE_DONE: i32 = 101;
|
|
|
|
const SQLITE_CHECKPOINT_PASSIVE: i32 = 0;
|
|
const SQLITE_CHECKPOINT_FULL: i32 = 1;
|
|
const SQLITE_CHECKPOINT_RESTART: i32 = 2;
|
|
const SQLITE_CHECKPOINT_TRUNCATE: i32 = 3;
|
|
const SQLITE_INTEGER: i32 = 1;
|
|
const SQLITE_FLOAT: i32 = 2;
|
|
const SQLITE_TEXT: i32 = 3;
|
|
const SQLITE3_TEXT: i32 = 3;
|
|
const SQLITE_BLOB: i32 = 4;
|
|
const SQLITE_NULL: i32 = 5;
|
|
|
|
#[cfg(not(target_os = "windows"))]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_libversion() {
|
|
unsafe {
|
|
let version = sqlite3_libversion();
|
|
assert!(!version.is_null());
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_libversion_number() {
|
|
unsafe {
|
|
let version_num = sqlite3_libversion_number();
|
|
assert!(version_num >= 3042000);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_open_not_found() {
|
|
unsafe {
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_open(c"not-found/local.db".as_ptr(), &mut db),
|
|
SQLITE_CANTOPEN
|
|
);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_open_existing() {
|
|
unsafe {
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_open(c"../testing/testing_clone.db".as_ptr(), &mut db),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_close() {
|
|
unsafe {
|
|
assert_eq!(sqlite3_close(ptr::null_mut()), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_prepare_misuse() {
|
|
unsafe {
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(c":memory:".as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"SELECT 1".as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_bind_int() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_bind (id INTEGER PRIMARY KEY, value INTEGER)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_bind (value) VALUES (?)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_bind_int(stmt, 1, 42), SQLITE_OK);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT value FROM test_bind LIMIT 1".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
assert_eq!(sqlite3_column_int(stmt, 0), 42);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
#[test]
|
|
fn test_sqlite3_bind_parameter_name_and_count() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_params (id INTEGER PRIMARY KEY, value TEXT)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_params (id, value) VALUES (?1, ?2)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
let param_count = sqlite3_bind_parameter_count(stmt);
|
|
assert_eq!(param_count, 2);
|
|
|
|
println!("parameter count {param_count}");
|
|
let name1 = sqlite3_bind_parameter_name(stmt, 1);
|
|
assert!(!name1.is_null());
|
|
let name1_str = std::ffi::CStr::from_ptr(name1).to_str().unwrap();
|
|
assert_eq!(name1_str, "?1");
|
|
|
|
let name2 = sqlite3_bind_parameter_name(stmt, 2);
|
|
assert!(!name2.is_null());
|
|
let name2_str = std::ffi::CStr::from_ptr(name2).to_str().unwrap();
|
|
assert_eq!(name2_str, "?2");
|
|
|
|
let invalid_name = sqlite3_bind_parameter_name(stmt, 99);
|
|
assert!(invalid_name.is_null());
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_last_insert_rowid() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = std::ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_rowid (value INTEGER)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_rowid (value) VALUES (6)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let last_rowid = sqlite3_last_insert_rowid(db);
|
|
assert!(last_rowid > 0);
|
|
println!("last insert rowid: {last_rowid}");
|
|
let query = format!("SELECT value FROM test_rowid WHERE rowid = {last_rowid}");
|
|
let query_cstring = std::ffi::CString::new(query).unwrap();
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
query_cstring.as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
let value_int = sqlite3_column_int(stmt, 0);
|
|
assert_eq!(value_int, 6);
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
#[test]
|
|
fn test_sqlite3_column_name() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = std::ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_cols (id INTEGER PRIMARY KEY, value TEXT)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT id, value FROM test_cols".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
let col_count = sqlite3_column_count(stmt);
|
|
assert_eq!(col_count, 2);
|
|
|
|
let name1 = sqlite3_column_name(stmt, 0);
|
|
assert!(!name1.is_null());
|
|
let name1_str = std::ffi::CStr::from_ptr(name1).to_str().unwrap();
|
|
assert_eq!(name1_str, "id");
|
|
|
|
let name2 = sqlite3_column_name(stmt, 1);
|
|
assert!(!name2.is_null());
|
|
let name2_str = std::ffi::CStr::from_ptr(name2).to_str().unwrap();
|
|
assert_eq!(name2_str, "value");
|
|
|
|
// will lead to panic
|
|
//let invalid = sqlite3_column_name(stmt, 5);
|
|
//assert!(invalid.is_null());
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_bind_text() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_bind_text_rs (id INTEGER PRIMARY KEY, value TEXT)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
let destructor = std::mem::transmute::<
|
|
isize,
|
|
Option<unsafe extern "C" fn(*mut std::ffi::c_void)>,
|
|
>(-1isize);
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_bind_text_rs (value) VALUES (?)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
let val = std::ffi::CString::new("hello world").unwrap();
|
|
assert_eq!(
|
|
sqlite3_bind_text(stmt, 1, val.as_ptr(), -1, destructor),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_bind_text_rs (value) VALUES (?)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
let val2 = std::ffi::CString::new("abcdef").unwrap();
|
|
assert_eq!(
|
|
sqlite3_bind_text(stmt, 1, val2.as_ptr(), 3, destructor),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT value FROM test_bind_text_rs ORDER BY id".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
let col1_ptr = sqlite3_column_text(stmt, 0);
|
|
assert!(!col1_ptr.is_null());
|
|
let col1_str = std::ffi::CStr::from_ptr(col1_ptr).to_str().unwrap();
|
|
assert_eq!(col1_str, "hello world");
|
|
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
|
|
let col2_ptr = sqlite3_column_text(stmt, 0);
|
|
let col2_len = sqlite3_column_bytes(stmt, 0);
|
|
assert!(!col2_ptr.is_null());
|
|
|
|
let col2_slice = std::slice::from_raw_parts(col2_ptr as *const u8, col2_len as usize);
|
|
let col2_str = std::str::from_utf8(col2_slice).unwrap().to_owned();
|
|
|
|
assert_eq!(col2_str, "abc");
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_bind_blob() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_bind_blob_rs (id INTEGER PRIMARY KEY, data BLOB)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_bind_blob_rs (data) VALUES (?)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
let data1 = b"\x01\x02\x03\x04\x05";
|
|
assert_eq!(
|
|
sqlite3_bind_blob(
|
|
stmt,
|
|
1,
|
|
data1.as_ptr() as *const _,
|
|
data1.len() as i32,
|
|
None
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_bind_blob_rs (data) VALUES (?)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
let data2 = b"\xAA\xBB\xCC\xDD";
|
|
assert_eq!(
|
|
sqlite3_bind_blob(stmt, 1, data2.as_ptr() as *const _, 2, None),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT data FROM test_bind_blob_rs ORDER BY id".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
let col1_ptr = sqlite3_column_blob(stmt, 0);
|
|
let col1_len = sqlite3_column_bytes(stmt, 0);
|
|
let col1_slice = std::slice::from_raw_parts(col1_ptr as *const u8, col1_len as usize);
|
|
assert_eq!(col1_slice, data1);
|
|
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
let col2_ptr = sqlite3_column_blob(stmt, 0);
|
|
let col2_len = sqlite3_column_bytes(stmt, 0);
|
|
let col2_slice = std::slice::from_raw_parts(col2_ptr as *const u8, col2_len as usize);
|
|
assert_eq!(col2_slice, &data2[..2]);
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_column_type() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = std::ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_types (col_int INTEGER, col_float REAL, col_text TEXT, col_blob BLOB, col_null text)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test_types VALUES (123, 45.67, 'hello', x'010203', null)"
|
|
.as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT col_int, col_float, col_text, col_blob, col_null FROM test_types"
|
|
.as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
|
|
assert_eq!(sqlite3_column_type(stmt, 0), SQLITE_INTEGER);
|
|
assert_eq!(sqlite3_column_type(stmt, 1), SQLITE_FLOAT);
|
|
assert_eq!(sqlite3_column_type(stmt, 2), SQLITE_TEXT);
|
|
assert_eq!(sqlite3_column_type(stmt, 3), SQLITE_BLOB);
|
|
assert_eq!(sqlite3_column_type(stmt, 4), SQLITE_NULL);
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_column_decltype() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = std::ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test_decltype (col_int INTEGER, col_float REAL, col_text TEXT, col_blob BLOB, col_null NULL)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut stmt = std::ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT col_int, col_float, col_text, col_blob, col_null FROM test_decltype"
|
|
.as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
std::ptr::null_mut(),
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
let expected = [
|
|
Some("INTEGER"),
|
|
Some("REAL"),
|
|
Some("TEXT"),
|
|
Some("BLOB"),
|
|
None,
|
|
];
|
|
|
|
for i in 0..sqlite3_column_count(stmt) {
|
|
let decl = sqlite3_column_decltype(stmt, i);
|
|
|
|
if decl.is_null() {
|
|
assert!(expected[i as usize].is_none());
|
|
} else {
|
|
let s = std::ffi::CStr::from_ptr(decl)
|
|
.to_string_lossy()
|
|
.into_owned();
|
|
assert_eq!(Some(s.as_str()), expected[i as usize]);
|
|
}
|
|
}
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[cfg(not(feature = "sqlite3"))]
|
|
mod libsql_ext {
|
|
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn test_wal_frame_count() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Ensure that WAL is initially empty.
|
|
let mut frame_count = 0;
|
|
assert_eq!(libsql_wal_frame_count(db, &mut frame_count), SQLITE_OK);
|
|
assert_eq!(frame_count, 0);
|
|
// Create a table and insert a row.
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test (id INTEGER PRIMARY KEY)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test (id) VALUES (1)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
// Check that WAL has three frames.
|
|
assert_eq!(libsql_wal_frame_count(db, &mut frame_count), SQLITE_OK);
|
|
assert_eq!(frame_count, 3);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_read_frame() {
|
|
unsafe {
|
|
let mut db = ptr::null_mut();
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Create a table and insert a row.
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test (id INTEGER PRIMARY KEY)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test (id) VALUES (1)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
// Check that WAL has three frames.
|
|
let mut frame_count = 0;
|
|
assert_eq!(libsql_wal_frame_count(db, &mut frame_count), SQLITE_OK);
|
|
assert_eq!(frame_count, 3);
|
|
for i in 1..frame_count + 1 {
|
|
let frame_len = 4096 + 24;
|
|
let mut frame = vec![0; frame_len];
|
|
assert_eq!(
|
|
libsql_wal_get_frame(db, i, frame.as_mut_ptr(), frame_len as u32),
|
|
SQLITE_OK
|
|
);
|
|
}
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_disable_wal_checkpoint() {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
unsafe {
|
|
let mut db = ptr::null_mut();
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Create a table and insert a row.
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test (id INTEGER PRIMARY KEY)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test (id) VALUES (0)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut log_size = 0;
|
|
let mut checkpoint_count = 0;
|
|
|
|
assert_eq!(
|
|
sqlite3_wal_checkpoint_v2(
|
|
db,
|
|
ptr::null(),
|
|
SQLITE_CHECKPOINT_PASSIVE,
|
|
&mut log_size,
|
|
&mut checkpoint_count
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
let mut wal_path = temp_file.path().to_path_buf();
|
|
assert!(wal_path.set_extension("db-wal"));
|
|
std::fs::remove_file(wal_path.clone()).unwrap();
|
|
|
|
{
|
|
let mut db = ptr::null_mut();
|
|
unsafe {
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
assert_eq!(libsql_wal_disable_checkpoint(db), SQLITE_OK);
|
|
// Insert at least 1000 rows to go over checkpoint threshold.
|
|
let mut stmt = ptr::null_mut();
|
|
for i in 1..2000 {
|
|
let sql =
|
|
std::ffi::CString::new(format!("INSERT INTO test (id) VALUES ({i})"))
|
|
.unwrap();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, sql.as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
}
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
// Delete WAL to ensure that we don't load anything from it
|
|
std::fs::remove_file(wal_path).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
unsafe {
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Insert at least 1000 rows to go over checkpoint threshold.
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT count() FROM test".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
let count = sqlite3_column_int64(stmt, 0);
|
|
assert_eq!(count, 1);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_get_autocommit() {
|
|
unsafe {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
// Should be in autocommit mode by default
|
|
assert_eq!(sqlite3_get_autocommit(db), 1);
|
|
|
|
// Begin a transaction
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"BEGIN".as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
// Should NOT be in autocommit mode during transaction
|
|
assert_eq!(sqlite3_get_autocommit(db), 0);
|
|
|
|
// Create a table within the transaction
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test (id INTEGER PRIMARY KEY)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
// Still not in autocommit mode
|
|
assert_eq!(sqlite3_get_autocommit(db), 0);
|
|
|
|
// Commit the transaction
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"COMMIT".as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
// Should be back in autocommit mode after commit
|
|
assert_eq!(sqlite3_get_autocommit(db), 1);
|
|
|
|
// Test with ROLLBACK
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"BEGIN".as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
assert_eq!(sqlite3_get_autocommit(db), 0);
|
|
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"ROLLBACK".as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
// Should be back in autocommit mode after rollback
|
|
assert_eq!(sqlite3_get_autocommit(db), 1);
|
|
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_wal_checkpoint() {
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
unsafe {
|
|
let mut db = ptr::null_mut();
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Create a table and insert a row.
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE test (id INTEGER PRIMARY KEY)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO test (id) VALUES (0)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
let mut log_size = 0;
|
|
let mut checkpoint_count = 0;
|
|
|
|
assert_eq!(
|
|
sqlite3_wal_checkpoint_v2(
|
|
db,
|
|
ptr::null(),
|
|
SQLITE_CHECKPOINT_PASSIVE,
|
|
&mut log_size,
|
|
&mut checkpoint_count
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
let mut wal_path = temp_file.path().to_path_buf();
|
|
assert!(wal_path.set_extension("db-wal"));
|
|
std::fs::remove_file(wal_path.clone()).unwrap();
|
|
|
|
{
|
|
let mut db = ptr::null_mut();
|
|
unsafe {
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Insert at least 1000 rows to go over checkpoint threshold.
|
|
let mut stmt = ptr::null_mut();
|
|
for i in 1..2000 {
|
|
let sql =
|
|
std::ffi::CString::new(format!("INSERT INTO test (id) VALUES ({i})"))
|
|
.unwrap();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, sql.as_ptr(), -1, &mut stmt, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
}
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
// Delete WAL to ensure that we don't load anything from it
|
|
std::fs::remove_file(wal_path).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
unsafe {
|
|
let path = temp_file.path();
|
|
let c_path = std::ffi::CString::new(path.to_str().unwrap()).unwrap();
|
|
assert_eq!(sqlite3_open(c_path.as_ptr(), &mut db), SQLITE_OK);
|
|
// Insert at least 1000 rows to go over checkpoint threshold.
|
|
let mut stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT count() FROM test".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
let count = sqlite3_column_int64(stmt, 0);
|
|
// with a sane `should_checkpoint` method we have no garuantee that all 2000 rows are present, as the checkpoint was
|
|
// triggered by cacheflush on insertions. the pattern will trigger a checkpoint when the wal has > 1000 frames,
|
|
// so it will be triggered but will no longer be triggered on each consecutive
|
|
// write. here we can assert that we have > 1500 rows.
|
|
assert!(count > 1500);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_clear_bindings() {
|
|
unsafe {
|
|
let mut db: *mut sqlite3 = ptr::null_mut();
|
|
let mut stmt: *mut sqlite3_stmt = ptr::null_mut();
|
|
|
|
assert_eq!(sqlite3_open(c":memory:".as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"CREATE TABLE person (id INTEGER, name TEXT, age INTEGER)".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"INSERT INTO person (id, name, age) VALUES (1, 'John', 25), (2, 'Jane', 30)"
|
|
.as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT * FROM person WHERE id = ? AND age > ?".as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
// Bind parameters - should find John (id=1, age=25 > 20)
|
|
assert_eq!(sqlite3_bind_int(stmt, 1, 1), SQLITE_OK);
|
|
assert_eq!(sqlite3_bind_int(stmt, 2, 20), SQLITE_OK);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_ROW);
|
|
assert_eq!(sqlite3_column_int(stmt, 0), 1);
|
|
assert_eq!(sqlite3_column_int(stmt, 2), 25);
|
|
|
|
// Reset and clear bindings, query should return no rows
|
|
assert_eq!(sqlite3_reset(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_clear_bindings(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_step(stmt), SQLITE_DONE);
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_bind_parameter_index() {
|
|
const SQLITE_OK: i32 = 0;
|
|
|
|
unsafe {
|
|
let mut db: *mut sqlite3 = ptr::null_mut();
|
|
let mut stmt: *mut sqlite3_stmt = ptr::null_mut();
|
|
|
|
assert_eq!(sqlite3_open(c":memory:".as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(
|
|
db,
|
|
c"SELECT * FROM sqlite_master WHERE name = :table_name AND type = :object_type"
|
|
.as_ptr(),
|
|
-1,
|
|
&mut stmt,
|
|
ptr::null_mut()
|
|
),
|
|
SQLITE_OK
|
|
);
|
|
|
|
let index1 = sqlite3_bind_parameter_index(stmt, c":table_name".as_ptr());
|
|
assert_eq!(index1, 1);
|
|
|
|
let index2 = sqlite3_bind_parameter_index(stmt, c":object_type".as_ptr());
|
|
assert_eq!(index2, 2);
|
|
|
|
let index3 = sqlite3_bind_parameter_index(stmt, c":nonexistent".as_ptr());
|
|
assert_eq!(index3, 0);
|
|
|
|
let index4 = sqlite3_bind_parameter_index(stmt, ptr::null());
|
|
assert_eq!(index4, 0);
|
|
|
|
assert_eq!(sqlite3_finalize(stmt), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_db_filename() {
|
|
const SQLITE_OK: i32 = 0;
|
|
|
|
unsafe {
|
|
// Test with in-memory database
|
|
let mut db: *mut sqlite3 = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(c":memory:".as_ptr(), &mut db), SQLITE_OK);
|
|
let filename = sqlite3_db_filename(db, c"main".as_ptr());
|
|
assert!(!filename.is_null());
|
|
let filename_str = std::ffi::CStr::from_ptr(filename).to_str().unwrap();
|
|
assert_eq!(filename_str, "");
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
|
|
// Open a file-backed database
|
|
let temp_file = tempfile::NamedTempFile::with_suffix(".db").unwrap();
|
|
let path = std::ffi::CString::new(temp_file.path().to_str().unwrap()).unwrap();
|
|
let mut db = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(path.as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
// Test with "main" database name
|
|
let filename = sqlite3_db_filename(db, c"main".as_ptr());
|
|
assert!(!filename.is_null());
|
|
let filename_str = std::ffi::CStr::from_ptr(filename).to_str().unwrap();
|
|
assert_eq!(filename_str, temp_file.path().to_str().unwrap());
|
|
|
|
// Test with NULL database name (defaults to main)
|
|
let filename_default = sqlite3_db_filename(db, ptr::null());
|
|
assert!(!filename_default.is_null());
|
|
assert_eq!(filename, filename_default);
|
|
|
|
// Test with non-existent database name
|
|
let filename = sqlite3_db_filename(db, c"temp".as_ptr());
|
|
assert!(filename.is_null());
|
|
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_sqlite3_next_stmt() {
|
|
const SQLITE_OK: i32 = 0;
|
|
|
|
unsafe {
|
|
let mut db: *mut sqlite3 = ptr::null_mut();
|
|
assert_eq!(sqlite3_open(c":memory:".as_ptr(), &mut db), SQLITE_OK);
|
|
|
|
// Initially, there should be no prepared statements
|
|
let iter = sqlite3_next_stmt(db, ptr::null_mut());
|
|
assert!(iter.is_null());
|
|
|
|
// Prepare first statement
|
|
let mut stmt1: *mut sqlite3_stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"SELECT 1;".as_ptr(), -1, &mut stmt1, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert!(!stmt1.is_null());
|
|
|
|
// Now there should be one statement
|
|
let iter = sqlite3_next_stmt(db, ptr::null_mut());
|
|
assert_eq!(iter, stmt1);
|
|
|
|
// And no more after that
|
|
let iter = sqlite3_next_stmt(db, stmt1);
|
|
assert!(iter.is_null());
|
|
|
|
// Prepare second statement
|
|
let mut stmt2: *mut sqlite3_stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"SELECT 2;".as_ptr(), -1, &mut stmt2, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert!(!stmt2.is_null());
|
|
|
|
// Prepare third statement
|
|
let mut stmt3: *mut sqlite3_stmt = ptr::null_mut();
|
|
assert_eq!(
|
|
sqlite3_prepare_v2(db, c"SELECT 3;".as_ptr(), -1, &mut stmt3, ptr::null_mut()),
|
|
SQLITE_OK
|
|
);
|
|
assert!(!stmt3.is_null());
|
|
|
|
// Count all statements
|
|
let mut count = 0;
|
|
let mut iter = sqlite3_next_stmt(db, ptr::null_mut());
|
|
while !iter.is_null() {
|
|
count += 1;
|
|
iter = sqlite3_next_stmt(db, iter);
|
|
}
|
|
assert_eq!(count, 3);
|
|
|
|
// Finalize the middle statement
|
|
assert_eq!(sqlite3_finalize(stmt2), SQLITE_OK);
|
|
|
|
// Count should now be 2
|
|
count = 0;
|
|
iter = sqlite3_next_stmt(db, ptr::null_mut());
|
|
while !iter.is_null() {
|
|
count += 1;
|
|
iter = sqlite3_next_stmt(db, iter);
|
|
}
|
|
assert_eq!(count, 2);
|
|
|
|
// Finalize remaining statements
|
|
assert_eq!(sqlite3_finalize(stmt1), SQLITE_OK);
|
|
assert_eq!(sqlite3_finalize(stmt3), SQLITE_OK);
|
|
|
|
// Should be no statements left
|
|
let iter = sqlite3_next_stmt(db, ptr::null_mut());
|
|
assert!(iter.is_null());
|
|
|
|
assert_eq!(sqlite3_close(db), SQLITE_OK);
|
|
}
|
|
}
|
|
}
|