62 lines
1.7 KiB
Rust
Raw Normal View History

2015-05-28 19:19:08 -04:00
extern crate sqlite;
extern crate temporary;
use std::path::PathBuf;
use temporary::Directory;
macro_rules! ok(
($result:expr) => ($result.unwrap());
);
#[test]
2015-05-29 11:24:01 -04:00
fn workflow() {
use sqlite::Binding::*;
2015-06-08 17:43:31 -04:00
use sqlite::State;
2015-05-29 11:24:01 -04:00
2015-05-28 22:49:00 -04:00
macro_rules! pair(
($one:expr, $two:expr) => (($one, Some($two)));
2015-05-28 22:49:00 -04:00
);
2015-05-28 19:19:08 -04:00
let (path, _directory) = setup();
let database = ok!(sqlite::open(&path));
2015-05-28 21:30:02 -04:00
let sql = r#"CREATE TABLE `users` (id INTEGER, name VARCHAR(255), age REAL);"#;
ok!(database.execute(sql));
2015-05-28 21:30:02 -04:00
2015-05-29 13:08:02 -04:00
{
let sql = r#"INSERT INTO `users` (id, name, age) VALUES (?, ?, ?);"#;
let mut statement = ok!(database.prepare(sql));
2015-05-29 13:08:02 -04:00
ok!(statement.bind(&[Integer(1, 1), Text(2, "Alice"), Float(3, 20.99)]));
2015-06-08 17:43:31 -04:00
assert!(ok!(statement.step()) == State::Done);
2015-05-29 13:08:02 -04:00
}
{
let mut done = false;
let sql = r#"SELECT * FROM `users`;"#;
ok!(database.process(sql, |pairs| {
2015-05-29 13:08:02 -04:00
assert!(pairs.len() == 3);
assert!(pairs[0] == pair!("id", "1"));
assert!(pairs[1] == pair!("name", "Alice"));
assert!(pairs[2] == pair!("age", "20.99"));
done = true;
true
}));
2015-05-29 13:08:02 -04:00
assert!(done);
}
{
let sql = r#"SELECT * FROM `users`;"#;
let mut statement = ok!(database.prepare(sql));
2015-06-08 17:43:31 -04:00
assert!(ok!(statement.step()) == State::Row);
2015-05-29 13:08:02 -04:00
assert!(ok!(statement.column::<i64>(0)) == 1);
assert!(ok!(statement.column::<String>(1)) == String::from("Alice"));
assert!(ok!(statement.column::<f64>(2)) == 20.99);
2015-06-08 17:43:31 -04:00
assert!(ok!(statement.step()) == State::Done);
2015-05-29 13:08:02 -04:00
}
2015-05-28 19:19:08 -04:00
}
fn setup() -> (PathBuf, Directory) {
let directory = ok!(Directory::new("sqlite"));
(directory.path().join("database.sqlite3"), directory)
}