catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

SNS: do not duplicate subscriptions

Browse source
This commit is contained in:
Boris Gvozdev 2018-01-02 11:30:39 +11:00
commit 633decc6c0
2 changed files with 27 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
moto/sns/models.py
View file

@ -247,11 +247,21 @@ class SNSBackend(BaseBackend):
setattr(topic, attribute_name, attribute_value)
def subscribe(self, topic_arn, endpoint, protocol):
# AWS doesn't create duplicates
old_subscription = self._find_subscription(topic_arn, endpoint, protocol)
if old_subscription:
return old_subscription
topic = self.get_topic(topic_arn)
subscription = Subscription(topic, endpoint, protocol)
self.subscriptions[subscription.arn] = subscription
return subscription
def _find_subscription(self, topic_arn, endpoint, protocol):
for subscription in self.subscriptions.values():
if subscription.topic.arn == topic_arn and subscription.endpoint == endpoint and subscription.protocol == protocol:
return subscription
return None
def unsubscribe(self, subscription_arn):
self.subscriptions.pop(subscription_arn)