use arrow::datatypes::{DataType, Field, Schema};
use dbx_core::Database;
#[test]
fn test_query_builder_simple() -> dbx_core::DbxResult<()> {
let db = Database::open_in_memory()?;
let schema = Schema::new(vec![
Field::new("id", DataType::Int64, false),
Field::new("name", DataType::Utf8, true),
]);
db.create_table("test_users", schema)?;
db.execute_sql("INSERT INTO test_users (id, name) VALUES (1, 'Alice')")?;
db.execute_sql("INSERT INTO test_users (id, name) VALUES (2, 'Bob')")?;
let results = db
.query_builder()
.select(&["id", "name"])
.from("test_users")
.execute()?;
let total_rows: usize = results.iter().map(|b| b.num_rows()).sum();
assert_eq!(total_rows, 2);
Ok(())
}