From c90129957ebbb1dfe23e2ea152e43a230d080b20 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Oliver=20R=C3=BCmpelein?= <oli_r@fg4f.de>
Date: Thu, 26 May 2022 12:42:20 +0200
Subject: [PATCH] Make test work.

---
 src/paperless_mail/tests/test_mail.py | 43 +++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/src/paperless_mail/tests/test_mail.py b/src/paperless_mail/tests/test_mail.py
index 24014b4dc..348876be2 100644
--- a/src/paperless_mail/tests/test_mail.py
+++ b/src/paperless_mail/tests/test_mail.py
@@ -1,5 +1,6 @@
 import dataclasses
 import email.contentmanager
+import imaplib
 import os
 import random
 import uuid
@@ -96,6 +97,11 @@ class BogusMailBox(ContextManager):
         if "UNFLAGGED" in criteria:
             msg = filter(lambda m: not m.flagged, msg)
 
+        if "UNKEYWORD" in criteria:
+            tag = criteria[criteria.index("UNKEYWORD") + 1].strip("'")
+            # For custom-flags, we need to access the `_raw_flags` field, as calls to `.flags` get cached, thus breaking the test.
+            msg = filter(lambda m: "processed" not in parse_raw_tags_omit_cache(m), msg)
+
         return list(msg)
 
     def delete(self, uid_list):
@@ -109,6 +115,8 @@ class BogusMailBox(ContextManager):
                         message.flagged = value
                     if flag == MailMessageFlags.SEEN:
                         message.seen = value
+                    if flag == "processed":
+                        message._raw_flag_data.append(f"+FLAGS (processed)".encode())
 
     def move(self, uid_list, folder):
         if folder == "spam":
@@ -130,6 +138,7 @@ def create_message(
     from_: str = "noone@mail.com",
     seen: bool = False,
     flagged: bool = False,
+    processed: bool = False,
 ) -> MailMessage:
     email_msg = email.message.EmailMessage()
     # TODO: This does NOT set the UID
@@ -175,6 +184,8 @@ def create_message(
 
     imap_msg.seen = seen
     imap_msg.flagged = flagged
+    if processed:
+        imap_msg._raw_flag_data.append(f"+FLAGS (processed)".encode())
 
     return imap_msg
 
@@ -189,6 +200,13 @@ def fake_magic_from_buffer(buffer, mime=False):
         return "Some verbose file description"
 
 
+def parse_raw_tags_omit_cache(m: MailMessage) -> List[str]:
+    raw_result: list[bytes] = []
+    for rf in m._raw_flag_data:
+        raw_result.extend(imaplib.ParseFlags(rf))
+    return [f.decode().strip() for f in raw_result]
+
+
 @mock.patch("paperless_mail.mail.magic.from_buffer", fake_magic_from_buffer)
 class TestMail(DirectoriesMixin, TestCase):
     def setUp(self):
@@ -217,6 +235,7 @@ class TestMail(DirectoriesMixin, TestCase):
                 body="cables",
                 seen=True,
                 flagged=False,
+                processed=False,
             ),
         )
         self.bogus_mailbox.messages.append(
@@ -225,6 +244,7 @@ class TestMail(DirectoriesMixin, TestCase):
                 body="from my favorite electronic store",
                 seen=False,
                 flagged=True,
+                processed=True,
             ),
         )
         self.bogus_mailbox.messages.append(
@@ -571,6 +591,29 @@ class TestMail(DirectoriesMixin, TestCase):
         self.assertEqual(len(self.bogus_mailbox.messages), 2)
         self.assertEqual(len(self.bogus_mailbox.messages_spam), 1)
 
+    def test_handle_mail_account_tag(self):
+        account = MailAccount.objects.create(
+            name="test",
+            imap_server="",
+            username="admin",
+            password="secret",
+        )
+
+        _ = MailRule.objects.create(
+            name="testrule",
+            account=account,
+            action=MailRule.MailAction.TAG,
+            action_parameter="processed",
+        )
+
+        self.assertEqual(len(self.bogus_mailbox.messages), 3)
+        self.assertEqual(self.async_task.call_count, 0)
+        self.assertEqual(len(self.bogus_mailbox.fetch("UNKEYWORD processed", False)), 2)
+        self.mail_account_handler.handle_mail_account(account)
+        self.assertEqual(self.async_task.call_count, 2)
+        self.assertEqual(len(self.bogus_mailbox.fetch("UNKEYWORD processed", False)), 0)
+        self.assertEqual(len(self.bogus_mailbox.messages), 3)
+
     def test_error_login(self):
         account = MailAccount.objects.create(
             name="test",