mirror of
https://github.com/aljazceru/turso.git
synced 2025-12-18 00:54:19 +01:00
936ae307b7
We currently have two value types, `Value` and `OwnedValue`. The original thinking was that `Value` is external type and `OwnedValue` is internal type. However, this just results in unnecessary transformation between the types as data crosses the Limbo library boundary. Let's just follow SQLite here and consolidate on a single value type (where `sqlite3_value` is just an alias for the internal `Mem` type). The way this will eventually work is that we can have bunch of pre-allocated `OwnedValue` objects in `ProgramState` and basically return a reference to them all the way to the application itself, which extracts the actual value.
96 lines
2.8 KiB
Rust
96 lines
2.8 KiB
Rust
use crate::common::TempDatabase;
|
|
use limbo_core::{OwnedValue, StepResult};
|
|
|
|
#[test]
|
|
fn test_statement_reset_bind() -> anyhow::Result<()> {
|
|
let _ = env_logger::try_init();
|
|
let tmp_db = TempDatabase::new_with_rusqlite("create table test (i integer);");
|
|
let conn = tmp_db.connect_limbo();
|
|
|
|
let mut stmt = conn.prepare("select ?")?;
|
|
|
|
stmt.bind_at(1.try_into()?, OwnedValue::Integer(1));
|
|
|
|
loop {
|
|
match stmt.step()? {
|
|
StepResult::Row => {
|
|
let row = stmt.row().unwrap();
|
|
assert_eq!(*row.get_value(0), limbo_core::OwnedValue::Integer(1));
|
|
}
|
|
StepResult::IO => tmp_db.io.run_once()?,
|
|
_ => break,
|
|
}
|
|
}
|
|
|
|
stmt.reset();
|
|
|
|
stmt.bind_at(1.try_into()?, OwnedValue::Integer(2));
|
|
|
|
loop {
|
|
match stmt.step()? {
|
|
StepResult::Row => {
|
|
let row = stmt.row().unwrap();
|
|
assert_eq!(*row.get_value(0), limbo_core::OwnedValue::Integer(2));
|
|
}
|
|
StepResult::IO => tmp_db.io.run_once()?,
|
|
_ => break,
|
|
}
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn test_statement_bind() -> anyhow::Result<()> {
|
|
let _ = env_logger::try_init();
|
|
let tmp_db = TempDatabase::new_with_rusqlite("create table test (i integer);");
|
|
let conn = tmp_db.connect_limbo();
|
|
|
|
let mut stmt = conn.prepare("select ?, ?1, :named, ?3, ?4")?;
|
|
|
|
stmt.bind_at(1.try_into()?, OwnedValue::build_text("hello"));
|
|
|
|
let i = stmt.parameters().index(":named").unwrap();
|
|
stmt.bind_at(i, OwnedValue::Integer(42));
|
|
|
|
stmt.bind_at(3.try_into()?, OwnedValue::from_blob(vec![0x1, 0x2, 0x3]));
|
|
|
|
stmt.bind_at(4.try_into()?, OwnedValue::Float(0.5));
|
|
|
|
assert_eq!(stmt.parameters().count(), 4);
|
|
|
|
loop {
|
|
match stmt.step()? {
|
|
StepResult::Row => {
|
|
let row = stmt.row().unwrap();
|
|
if let limbo_core::OwnedValue::Text(s) = row.get_value(0) {
|
|
assert_eq!(s.as_str(), "hello")
|
|
}
|
|
|
|
if let limbo_core::OwnedValue::Text(s) = row.get_value(1) {
|
|
assert_eq!(s.as_str(), "hello")
|
|
}
|
|
|
|
if let limbo_core::OwnedValue::Integer(i) = row.get_value(2) {
|
|
assert_eq!(*i, 42)
|
|
}
|
|
|
|
if let limbo_core::OwnedValue::Blob(v) = row.get_value(3) {
|
|
assert_eq!(v.as_ref(), &vec![0x1 as u8, 0x2, 0x3])
|
|
}
|
|
|
|
if let limbo_core::OwnedValue::Float(f) = row.get_value(4) {
|
|
assert_eq!(*f, 0.5)
|
|
}
|
|
}
|
|
StepResult::IO => {
|
|
tmp_db.io.run_once()?;
|
|
}
|
|
StepResult::Interrupt => break,
|
|
StepResult::Done => break,
|
|
StepResult::Busy => panic!("Database is busy"),
|
|
};
|
|
}
|
|
Ok(())
|
|
}
|