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

feature: add parameters back to Message models

Browse Source
This commit is contained in:
Robert C Jensen 2018-07-17 20:12:05 -04:00
parent ba1ceee95f
commit f50c6c2fb0
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
moto/ses/models.py
View File

@ -13,14 +13,21 @@ RECIPIENT_LIMIT = 50
class Message(BaseModel):
def __init__(self, message_id):
def __init__(self, message_id, source, subject, body, destinations):
self.id = message_id
self.source = source
self.subject = subject
self.body = body
self.destinations = destinations
class RawMessage(BaseModel):
def __init__(self, message_id):
def __init__(self, message_id, source, destinations, raw_data):
self.id = message_id
self.source = source
self.destinations = destinations
self.raw_data = raw_data
class SESQuota(BaseModel):
@ -79,7 +86,7 @@ class SESBackend(BaseBackend):
)
message_id = get_random_message_id()
message = Message(message_id)
message = Message(message_id, source, subject, body, destinations)
self.sent_messages.append(message)
self.sent_message_count += recipient_count
return message
@ -116,7 +123,7 @@ class SESBackend(BaseBackend):
self.sent_message_count += recipient_count
message_id = get_random_message_id()
message = RawMessage(message_id)
message = RawMessage(message_id, source, destinations, raw_data)
self.sent_messages.append(message)
return message