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

Merge pull request #1724 from r1b/master

Browse Source 
feature: add parameters back to Message models
This commit is contained in:
Steve Pulec 2018-07-19 11:11:47 -04:00 committed by GitHub
commit de88ae8cec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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