diff options
| -rw-r--r-- | src/db.rs | 2 | ||||
| -rw-r--r-- | src/posts.rs | 7 |
2 files changed, 6 insertions, 3 deletions
diff --git a/src/db.rs b/src/db.rs index b79a789..d09ce21 100644 --- a/src/db.rs +++ b/src/db.rs @@ -18,7 +18,7 @@ pub struct Conn { } impl Conn { - fn init(path: &str) -> rusqlite::Connection { + pub fn init(path: &str) -> rusqlite::Connection { let start = time::Instant::now(); info!("Connecting to database"); let conn = rusqlite::Connection::open_with_flags( diff --git a/src/posts.rs b/src/posts.rs index 88368a6..698f8a6 100644 --- a/src/posts.rs +++ b/src/posts.rs @@ -42,7 +42,8 @@ mod tests { #[test] fn post_new() { - let db = db::Conn::new(); + let db = db::Conn::init("/tmp/clinte.db"); + let db = db::Conn { conn: db }; let mut stmt = db .conn .prepare("INSERT INTO posts (title, author, body) VALUES (:title, :author, :body)") @@ -51,16 +52,18 @@ mod tests { let title = String::from("TEST TITLE"); new(&mut stmt, &title, "TEST BODY").unwrap(); + update("NEW TITLE", "TEST BODY", 1, &db).unwrap(); let mut stmt = db .conn .prepare("SELECT * FROM posts WHERE title = :title") .unwrap(); + let title = String::from("NEW TITLE"); let out: String = stmt .query_row_named(&[(":title", &title)], |row| row.get::<usize, String>(1)) .unwrap(); - assert_eq!("TEST TITLE", &out); + assert_eq!("NEW TITLE", &out); } } |