Testing drafts in integration, fixing a bug where the sent wasnt leaving the drafts mailbox

1 files changed, 4 insertions, 4 deletions

diff --git a/service/pixelated/adapter/pixelated_mail.py b/service/pixelated/adapter/pixelated_mail.py
index 89320b75..b7323af9 100644
--- a/service/pixelated/adapter/pixelated_mail.py
+++ b/service/pixelated/adapter/pixelated_mail.py
@@ -77,17 +77,17 @@ class InputMail:
             self._chash = sha256.SHA256(self._raw()).hexdigest()
         return self._chash
 
-    def _get_for_save(self, next_uid):
-        docs = [self._fdoc(next_uid), self._hdoc()]
+    def _get_for_save(self, next_uid, mailbox):
+        docs = [self._fdoc(next_uid, mailbox), self._hdoc()]
         docs.extend([m for m in self._cdocs()])
         return docs
 
-    def _fdoc(self, next_uid):
+    def _fdoc(self, next_uid, mailbox):
         if self._fd:
             return self._fd
 
         fd = {}
-        fd[fields.MBOX_KEY] = 'DRAFTS'
+        fd[fields.MBOX_KEY] = mailbox
         fd[fields.UID_KEY] = next_uid
         fd[fields.CONTENT_HASH_KEY] = self._get_chash()
         fd[fields.SIZE_KEY] = len(self._raw())