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

Merge pull request #624 from mvantellingen/store-sns-message

Browse Source 
Save the sns messages in the PlatformEndpoint
This commit is contained in:
Steve Pulec 2016-06-17 17:50:37 -04:00 committed by GitHub
commit b7a2f55afe
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/sns/models.py
View File

@ -119,6 +119,7 @@ class PlatformEndpoint(object):
self.token = token self.token = token
self.attributes = attributes self.attributes = attributes
self.id = uuid.uuid4() self.id = uuid.uuid4()
self.messages = OrderedDict()
@property @property
def arn(self): def arn(self):
@ -130,8 +131,9 @@ class PlatformEndpoint(object):
) )
def publish(self, message): def publish(self, message):
message_id = six.text_type(uuid.uuid4())
# This is where we would actually send a message # This is where we would actually send a message
message_id = six.text_type(uuid.uuid4())
self.messages[message_id] = message
return message_id return message_id