From ba1d6edfc144c0549e88392a8414a147a9f30931 Mon Sep 17 00:00:00 2001 From: Folker Bernitt 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 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'service/pixelated/adapter/model/mail.py') 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: -- cgit v1.2.3