fix unit test for travis

2 files changed, 17 insertions, 9 deletions

diff --git a/README.md b/README.md
index 28ea256..3f5886f 100644
--- a/README.md
+++ b/README.md
@@ -4,6 +4,7 @@ notmuch-rs
 This is not much more than a wrapper for the [notmuch](https://notmuchmail.org/) C api.
 
 [![Build Status](https://travis-ci.org/vhdirk/notmuch-rs.svg?branch=master)](https://travis-ci.org/vhdirk/notmuch-rs)
+[![Crates.io](https://img.shields.io/crates/v/notmuch.svg)](https://crates.io/crates/notmuch)
 
 ## Building
 **notmuch-rs** expects libnotmuch development files to be installed on your system.
diff --git a/tests/main.rs b/tests/main.rs
index a6dd022..b92bcf7 100644
--- a/tests/main.rs
+++ b/tests/main.rs
@@ -5,16 +5,23 @@ fn main() {
     let mut mail_path = std::env::home_dir().unwrap();
     mail_path.push(".mail");
 
-    let db = notmuch::Database::open(&mail_path.to_str().unwrap().to_string(), notmuch::DatabaseMode::ReadOnly).unwrap();
-    let query = db.create_query(&"".to_string()).unwrap();
-    let mut threads = query.search_threads().unwrap();
+    match notmuch::Database::open(&mail_path.to_str().unwrap().to_string(), notmuch::DatabaseMode::ReadOnly){
+        Ok(db) => {
+            let query = db.create_query(&"".to_string()).unwrap();
+            let mut threads = query.search_threads().unwrap();
 
-    loop {
-        match threads.next() {
-            Some(thread) => {
-                println!("thread {:?} {:?}", thread.subject(), thread.authors());
-            },
-            None => { break }
+            loop {
+                match threads.next() {
+                    Some(thread) => {
+                        println!("thread {:?} {:?}", thread.subject(), thread.authors());
+                    },
+                    None => { break }
+                }
+            }
+
+        },
+        Err(err) =>{
+            println!("Got error while trying to open db: {:?}", err);
         }
     }
 }