diff options
Diffstat (limited to 'py-fake-service/app/adapter/mailset.py')
-rw-r--r-- | py-fake-service/app/adapter/mailset.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/py-fake-service/app/adapter/mailset.py b/py-fake-service/app/adapter/mailset.py index d81600e0..2b1627ae 100644 --- a/py-fake-service/app/adapter/mailset.py +++ b/py-fake-service/app/adapter/mailset.py @@ -8,7 +8,13 @@ class MailSet: self.mails = {} def add(self, mbox_mail): - self.mails[self.ident] = Mail(mbox_mail) + self.mails[self.ident] = Mail(mbox_mail, self.ident) self.ident += 1 + def values(self): + return self.mails.values() + + def get(self, mail_id): + return self.mails.get(int(mail_id)) + |