From ba1d6edfc144c0549e88392a8414a147a9f30931 Mon Sep 17 00:00:00 2001
From: Folker Bernitt <fbernitt@thoughtworks.com>
Date: Tue, 21 Jul 2015 14:48:52 +0000
Subject: Fixed MarkAsReadUnreadTest#test_mark_single_as_read and underlying
 implementation.

---
 service/pixelated/adapter/model/mail.py            | 15 +++++++++------
 service/pixelated/adapter/services/mail_service.py |  8 +++++---
 2 files changed, 14 insertions(+), 9 deletions(-)

(limited to 'service/pixelated/adapter')

diff --git a/service/pixelated/adapter/model/mail.py b/service/pixelated/adapter/model/mail.py
index dd32d665..5002974f 100644
--- a/service/pixelated/adapter/model/mail.py
+++ b/service/pixelated/adapter/model/mail.py
@@ -426,18 +426,21 @@ class PixelatedMail(Mail):
         yield self._persist_mail_tags(tags)
         defer.returnValue(self.tags)
 
+    @defer.inlineCallbacks
     def mark_as_read(self):
         if Status.SEEN in self.flags:
-            return self
-        self.flags.append(Status.SEEN)
-        self.save()
-        return self
+            defer.returnValue(self)
+        else:
+            self.flags.append(Status.SEEN)
+            yield self.save()
+            defer.returnValue(self)
 
+    @defer.inlineCallbacks
     def mark_as_unread(self):
         if Status.SEEN in self.flags:
             self.flags.remove(Status.SEEN)
-            self.save()
-        return self
+            yield self.save()
+        defer.returnValue(self)
 
     def mark_as_not_recent(self):
         if Status.RECENT in self.flags:
diff --git a/service/pixelated/adapter/services/mail_service.py b/service/pixelated/adapter/services/mail_service.py
index a22ca394..91eeb28f 100644
--- a/service/pixelated/adapter/services/mail_service.py
+++ b/service/pixelated/adapter/services/mail_service.py
@@ -88,13 +88,15 @@ class MailService(object):
             yield (yield self.mailboxes.drafts).remove(last_draft_ident)
         defer.returnValue((yield (yield self.mailboxes.sent).add(mail)))
 
+    @defer.inlineCallbacks
     def mark_as_read(self, mail_id):
-        mail = self.mail(mail_id)
-        mail.mark_as_read()
+        mail = yield self.mail(mail_id)
+        yield mail.mark_as_read()
         self.search_engine.index_mail(mail)
 
+    @defer.inlineCallbacks
     def mark_as_unread(self, mail_id):
-        mail = self.mail(mail_id)
+        mail = yield self.mail(mail_id)
         mail.mark_as_unread()
         self.search_engine.index_mail(mail)
 
-- 
cgit v1.2.3