diff options
Diffstat (limited to 'py-fake-service/app/adapter/mailset.py')
-rw-r--r-- | py-fake-service/app/adapter/mailset.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/py-fake-service/app/adapter/mailset.py b/py-fake-service/app/adapter/mailset.py index 2568e33b..5a6f27a4 100644 --- a/py-fake-service/app/adapter/mailset.py +++ b/py-fake-service/app/adapter/mailset.py @@ -15,8 +15,20 @@ class MailSet: return self.mails.values() def get(self, mail_id): - return self.mails.get(int(mail_id)) + return self.mails.get(mail_id) def mark_as_read(self, mail_id): - mail = self.mails.get(int(mail_id)) + mail = self.get(mail_id) mail.status.append('read') + + def delete(self, mail_id): + """ Returns True if the email got purged, else returns False meaning the email got moved to trash """ + + mail = self.get(mail_id) + if 'trash' in mail.tags: + del self.mails[mail_id] + return True + mail.tags.append('trash') + return False + + |