more tests for tags

3 files changed, 173 insertions, 3 deletions

diff --git a/tests/fixtures.rs b/tests/fixtures.rs
index c9e04c8..a8166be 100644
--- a/tests/fixtures.rs
+++ b/tests/fixtures.rs
@@ -147,16 +147,41 @@ impl MailBox {
             if seen {
                 flags += "S";
             }
-            self.maildir.store_cur_with_flags(&msg.message_to_string().unwrap().as_bytes(), flags.as_str()).unwrap()
-        };
+            println!("flags: {:?}", flags);
+            let mid = self.maildir.store_cur_with_flags(&msg.message_to_string().unwrap().as_bytes(), flags.as_str()).unwrap();
 
+            // I have no idea what the reasoning for the :2 here is, but ok.
+            format!("{}:2,{}", mid, flags)
+        };
+        
+        // let mut flags = String::from("");
+        // if flagged {
+        //     flags += "F";
+        // }
+        // if replied {
+        //     flags += "R";
+        // }
+        // if seen {
+        //     flags += "S";
+        // }
+        // println!("flags: {:?}", flags);
+        // let id = self.maildir.store_cur_with_flags(&msg.message_to_string().unwrap().as_bytes(), flags.as_str()).unwrap();
+
+        // if is_new {
+        //     let msgpath = format!("{}{}", id, flags);
+        //     std::fs::rename(msgpath, newpath)?;
+
+        //     self.maildir.path()
+        // }
+
+        
         let mut msgpath = self.path();
         msgpath = if is_new {
             msgpath.join("new")
         } else {
             msgpath.join("cur")
         };
-
+        
         msgpath = msgpath.join(&id);
 
         Ok((msg_id, msgpath))
diff --git a/tests/lib.rs b/tests/lib.rs
index bd3138d..e7ee0bc 100644
--- a/tests/lib.rs
+++ b/tests/lib.rs
@@ -10,4 +10,5 @@ mod fixtures;
 mod test_database;
 mod test_thread;
 mod test_message;
+mod test_tags;
 
diff --git a/tests/test_tags.rs b/tests/test_tags.rs
new file mode 100644
index 0000000..5ddabeb
--- /dev/null
+++ b/tests/test_tags.rs
@@ -0,0 +1,144 @@
+use std::sync::Arc;
+use std::path::PathBuf;
+use fixtures::{MailBox, NotmuchCommand};
+
+struct TagSetFixture {
+    // An non-empty immutable tagset.
+    // This will have the default new mail tags: inbox, unread.
+    pub mailbox: MailBox,
+    pub cmd: NotmuchCommand,
+    pub database: Arc<notmuch::Database>,
+    pub message: notmuch::Message<'static, notmuch::Database>
+}
+
+impl TagSetFixture {
+    pub fn new(mutable: bool, flagged: bool) -> Self{
+        let mailbox = MailBox::new();
+        let (_msg, filename) = mailbox.deliver(None, None, None, None, vec![], !flagged, None, false, false, flagged).unwrap();
+      
+        let cmd = NotmuchCommand::new(&mailbox.path());
+        cmd.run(vec!["new"]).unwrap();
+
+        let database = Arc::new(notmuch::Database::open(&mailbox.path(), if !mutable {notmuch::DatabaseMode::ReadOnly} else { notmuch::DatabaseMode::ReadWrite }).unwrap());
+        let message = <notmuch::Database as notmuch::DatabaseExt>::find_message_by_filename(database.clone(), &filename).unwrap().unwrap();    
+
+        Self {
+            mailbox,
+            database,
+            cmd, 
+            message
+        }
+    }
+}
+
+mod immutable {
+
+    use super::*;
+
+    #[test]
+    fn test_neg(){
+        let tagset = TagSetFixture::new(false, false);
+
+        let tags: Vec<String> = tagset.database.all_tags().unwrap().collect();
+        tagset.cmd.run(vec!["tag", "+foo", "*"]).unwrap();
+
+        let database = notmuch::Database::open(&tagset.mailbox.path(), notmuch::DatabaseMode::ReadOnly).unwrap();
+        let ntags: Vec<String> = database.all_tags().unwrap().collect();
+
+        assert_ne!(tags, ntags);
+    }
+
+    #[test]
+    fn test_contains(){
+        let tagset = TagSetFixture::new(false, false);
+        let tags: Vec<String> = tagset.database.all_tags().unwrap().collect();
+
+        assert!(tags.iter().any(|x| x == "unread"));
+        assert!(!tags.iter().any(|x| x == "foo"));
+    }
+
+
+    #[test]
+    fn test_len(){
+        let tagset = TagSetFixture::new(false, false);
+        assert_eq!(tagset.database.all_tags().unwrap().count(), 2);
+    }
+
+}
+
+mod mutable {
+
+    use super::*;
+
+    #[test]
+    fn test_add(){
+        let tagset = TagSetFixture::new(true, false);
+        assert!(!tagset.message.tags().any(|x| x == "foo"));
+
+        tagset.message.add_tag("foo").unwrap();
+        assert!(tagset.message.tags().any(|x| x == "foo"));
+    }
+
+    #[test]
+    fn test_discard(){
+        let tagset = TagSetFixture::new(true, false);
+        assert!(tagset.message.tags().any(|x| x == "inbox"));
+
+        tagset.message.remove_tag("inbox").unwrap();
+        assert!(!tagset.message.tags().any(|x| x == "unbox"));
+    }
+
+    #[test]
+    fn test_discard_not_present(){
+        let tagset = TagSetFixture::new(true, false);
+        assert!(!tagset.message.tags().any(|x| x == "foo"));
+
+        tagset.message.remove_tag("foo").unwrap();
+    }
+
+    #[test]
+    fn test_clear(){
+        let tagset = TagSetFixture::new(true, false);
+        assert!(tagset.message.tags().count() > 0);
+        tagset.message.remove_all_tags().unwrap();
+
+        assert!(tagset.message.tags().count() == 0);
+    }
+
+    #[test]
+    fn test_from_maildir_flags(){
+        let tagset = TagSetFixture::new(true, true);
+
+        let msgid = tagset.message.id();
+        tagset.message.remove_tag(&"flagged").unwrap();
+        tagset.message.maildir_flags_to_tags().unwrap();
+
+        assert!(tagset.message.tags().any(|x| x == "flagged"));
+    }
+
+
+    #[test]
+    fn test_to_maildir_flags(){
+
+        let tagset = TagSetFixture::new(true, true);
+
+        let filename = tagset.message.filename();
+        let filestr = filename.to_string_lossy();
+
+        let file_parts: Vec<&str> = filestr.split(",").collect();
+        let flags = file_parts.last().unwrap();
+        println!("Flags {:?}", flags);
+
+        assert!(flags.contains("F"));
+        tagset.message.remove_tag(&"flagged").unwrap();
+        tagset.message.tags_to_maildir_flags().unwrap();
+
+        let filename = tagset.message.filename();
+        let filestr = filename.to_string_lossy();
+
+        let file_parts: Vec<&str> = filestr.split(",").collect();
+        let flags = file_parts.last().unwrap();
+        assert!(!flags.contains("F"));
+    }
+
+}
\ No newline at end of file