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);
     }
 }