use sqlite::{Connection, State};
macro_rules! ok(($result:expr) => ($result.unwrap()));
pub fn create() -> Connection {
let connection = ok!(Connection::open(":memory:"));
let query = "CREATE TABLE data (a INTEGER, b REAL, c REAL, d REAL)";
ok!(connection.execute(query));
connection
}
pub fn populate(connection: &Connection, count: usize) {
let query = "INSERT INTO data (a, b, c, d) VALUES (?, ?, ?, ?)";
let mut statement = ok!(connection.prepare(query));
for i in 1..(count + 1) {
ok!(statement.reset());
ok!(statement.bind((1, i as i64)));
ok!(statement.bind((2, i as f64)));
ok!(statement.bind((3, i as f64)));
ok!(statement.bind((4, i as f64)));
assert_eq!(ok!(statement.next()), State::Done);
}
}