From 5053cfbb3ec2037e0cc59684e24d31499759f1c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Michael=20B=C3=B8cker-Larsen?= <m19n@pm.me>
Date: Sun, 20 Dec 2020 13:04:31 +0800
Subject: [PATCH] feat: add 'flags' method to add multiple flags at once

---
 src/client.rs             | 6 ++++++
 tests/imap_integration.rs | 3 +--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/client.rs b/src/client.rs
index 4fc4c25..74127ef 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -100,6 +100,12 @@ impl<'a, T: Read + Write> AppendCmd<'a, T> {
         self
     }
 
+    /// Append an array of flags
+    pub fn flags(&mut self, flags: &'a [Flag<'a>]) -> &mut Self {
+        self.flags.append(&mut flags.to_vec());
+        self
+    }
+
     /// Set the internal date
     pub fn internal_date(&mut self, date: DateTime<FixedOffset>) -> &mut Self {
         self.date = Some(date);
diff --git a/tests/imap_integration.rs b/tests/imap_integration.rs
index b32a0c0..536efd2 100644
--- a/tests/imap_integration.rs
+++ b/tests/imap_integration.rs
@@ -304,8 +304,7 @@ fn append_with_flags() {
     //append
     let flags: &[Flag] = &[Flag::Seen, Flag::Flagged];
     c.append(mbox, e.message_to_string().unwrap().as_bytes())
-        .flag(Flag::Seen)
-        .flag(Flag::Flagged)
+        .flags(flags)
         .run()
         .unwrap();