Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #604 from njl/restore_sent_messages

Browse Source 
Add sent_messages back to SESBackend
This commit is contained in:
Steve Pulec 2016-05-05 23:03:26 -04:00
commit 3dd44245f7
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
moto/ses/models.py
View File

@ -33,6 +33,7 @@ class SESBackend(BaseBackend):
def __init__(self): def __init__(self):
self.addresses = [] self.addresses = []
self.domains = [] self.domains = []
self.sent_messages = []
self.sent_message_count = 0 self.sent_message_count = 0
def _is_verified_address(self, address): def _is_verified_address(self, address):
@ -67,6 +68,7 @@ class SESBackend(BaseBackend):
message_id = get_random_message_id() message_id = get_random_message_id()
message = Message(message_id) message = Message(message_id)
self.sent_messages.append(message)
self.sent_message_count += recipient_count self.sent_message_count += recipient_count
return message return message
@ -88,7 +90,9 @@ class SESBackend(BaseBackend):
self.sent_message_count += recipient_count self.sent_message_count += recipient_count
message_id = get_random_message_id() message_id = get_random_message_id()
return RawMessage(message_id) message = RawMessage(message_id)
self.sent_messages.append(message)
return message
def get_send_quota(self): def get_send_quota(self):
return SESQuota(self.sent_message_count) return SESQuota(self.sent_message_count)