updating tags on py-fake-service

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
+
+