Add basic SNS support.

tests/test_sns/test_publishing.py

@@ -0,0 +1,62 @@
+from urlparse import parse_qs
+
+import boto
+from freezegun import freeze_time
+import httpretty
+import sure  # noqa
+
+from moto import mock_sns, mock_sqs
+
+
+@mock_sqs
+@mock_sns
+def test_publish_to_sqs():
+    conn = boto.connect_sns()
+    conn.create_topic("some-topic")
+    topics_json = conn.get_all_topics()
+    topic_arn = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"][0]['TopicArn']
+
+    sqs_conn = boto.connect_sqs()
+    sqs_conn.create_queue("test-queue")
+
+    conn.subscribe(topic_arn, "sqs", "arn:aws:sqs:us-east-1:123456789012:test-queue")
+
+    conn.publish(topic=topic_arn, message="my message")
+
+    queue = sqs_conn.get_queue("test-queue")
+    message = queue.read(1)
+    message.get_body().should.equal('my message')
+
+
+@freeze_time("2013-01-01")
+@mock_sns
+def test_publish_to_http():
+    httpretty.HTTPretty.register_uri(
+        method="POST",
+        uri="http://example.com/foobar",
+    )
+
+    conn = boto.connect_sns()
+    conn.create_topic("some-topic")
+    topics_json = conn.get_all_topics()
+    topic_arn = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"][0]['TopicArn']
+
+    conn.subscribe(topic_arn, "http", "http://example.com/foobar")
+
+    response = conn.publish(topic=topic_arn, message="my message", subject="my subject")
+    message_id = response['PublishResponse']['PublishResult']['MessageId']
+
+    last_request = httpretty.last_request()
+    last_request.method.should.equal("POST")
+    parse_qs(last_request.body).should.equal({
+        "Type": ["Notification"],
+        "MessageId": [message_id],
+        "TopicArn": ["arn:aws:sns:us-east-1:123456789012:some-topic"],
+        "Subject": ["my subject"],
+        "Message": ["my message"],
+        "Timestamp": ["2013-01-01T00:00:00Z"],
+        "SignatureVersion": ["1"],
+        "Signature": ["EXAMPLElDMXvB8r9R83tGoNn0ecwd5UjllzsvSvbItzfaMpN2nk5HVSw7XnOn/49IkxDKz8YrlH2qJXj2iZB0Zo2O71c4qQk1fMUDi3LGpij7RCW7AW9vYYsSqIKRnFS94ilu7NFhUzLiieYr4BKHpdTmdD6c0esKEYBpabxDSc="],
+        "SigningCertURL": ["https://sns.us-east-1.amazonaws.com/SimpleNotificationService-f3ecfb7224c7233fe7bb5f59f96de52f.pem"],
+        "UnsubscribeURL": ["https://sns.us-east-1.amazonaws.com/?Action=Unsubscribe&SubscriptionArn=arn:aws:sns:us-east-1:123456789012:some-topic:2bcfbf39-05c3-41de-beaa-fcfcc21c8f55"],
+    })

tests/test_sns/test_subscriptions.py

@@ -0,0 +1,49 @@
+import boto
+
+import sure  # noqa
+
+from moto import mock_sns
+
+
+@mock_sns
+def test_creating_subscription():
+    conn = boto.connect_sns()
+    conn.create_topic("some-topic")
+    topics_json = conn.get_all_topics()
+    topic_arn = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"][0]['TopicArn']
+
+    conn.subscribe(topic_arn, "http", "http://example.com/")
+
+    subscriptions = conn.get_all_subscriptions()["ListSubscriptionsResponse"]["ListSubscriptionsResult"]["Subscriptions"]
+    subscriptions.should.have.length_of(1)
+    subscription = subscriptions[0]
+    subscription["TopicArn"].should.equal(topic_arn)
+    subscription["Protocol"].should.equal("http")
+    subscription["SubscriptionArn"].should.contain(topic_arn)
+    subscription["Endpoint"].should.equal("http://example.com/")
+
+    # Now unsubscribe the subscription
+    conn.unsubscribe(subscription["SubscriptionArn"])
+
+    # And there should be zero subscriptions left
+    subscriptions = conn.get_all_subscriptions()["ListSubscriptionsResponse"]["ListSubscriptionsResult"]["Subscriptions"]
+    subscriptions.should.have.length_of(0)
+
+
+@mock_sns
+def test_getting_subscriptions_by_topic():
+    conn = boto.connect_sns()
+    conn.create_topic("topic1")
+    conn.create_topic("topic2")
+
+    topics_json = conn.get_all_topics()
+    topics = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"]
+    topic1_arn = topics[0]['TopicArn']
+    topic2_arn = topics[1]['TopicArn']
+
+    conn.subscribe(topic1_arn, "http", "http://example1.com/")
+    conn.subscribe(topic2_arn, "http", "http://example2.com/")
+
+    topic1_subscriptions = conn.get_all_subscriptions_by_topic(topic1_arn)["ListSubscriptionsByTopicResponse"]["ListSubscriptionsByTopicResult"]["Subscriptions"]
+    topic1_subscriptions.should.have.length_of(1)
+    topic1_subscriptions[0]['Endpoint'].should.equal("http://example1.com/")

tests/test_sns/test_topics.py

@@ -0,0 +1,54 @@
+import boto
+
+import sure  # noqa
+
+from moto import mock_sns
+from moto.sns.models import DEFAULT_TOPIC_POLICY, DEFAULT_EFFECTIVE_DELIVERY_POLICY
+
+
+@mock_sns
+def test_create_and_delete_topic():
+    conn = boto.connect_sns()
+    conn.create_topic("some-topic")
+
+    topics_json = conn.get_all_topics()
+    topics = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"]
+    topics.should.have.length_of(1)
+    topics[0]['TopicArn'].should.equal("arn:aws:sns:us-east-1:123456789012:some-topic")
+
+    # Delete the topic
+    conn.delete_topic(topics[0]['TopicArn'])
+
+    # And there should now be 0 topics
+    topics_json = conn.get_all_topics()
+    topics = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"]
+    topics.should.have.length_of(0)
+
+
+@mock_sns
+def test_topic_attributes():
+    conn = boto.connect_sns()
+    conn.create_topic("some-topic")
+
+    topics_json = conn.get_all_topics()
+    topic_arn = topics_json["ListTopicsResponse"]["ListTopicsResult"]["Topics"][0]['TopicArn']
+
+    attributes = conn.get_topic_attributes(topic_arn)['GetTopicAttributesResponse']['GetTopicAttributesResult']['Attributes']
+    attributes["TopicArn"].should.equal("arn:aws:sns:us-east-1:123456789012:some-topic")
+    attributes["Owner"].should.equal(123456789012)
+    attributes["Policy"].should.equal(DEFAULT_TOPIC_POLICY)
+    attributes["DisplayName"].should.equal("")
+    attributes["SubscriptionsPending"].should.equal(0)
+    attributes["SubscriptionsConfirmed"].should.equal(0)
+    attributes["SubscriptionsDeleted"].should.equal(0)
+    attributes["DeliveryPolicy"].should.equal("")
+    attributes["EffectiveDeliveryPolicy"].should.equal(DEFAULT_EFFECTIVE_DELIVERY_POLICY)
+
+    conn.set_topic_attributes(topic_arn, "Policy", {"foo": "bar"})
+    conn.set_topic_attributes(topic_arn, "DisplayName", "My display name")
+    conn.set_topic_attributes(topic_arn, "DeliveryPolicy", {"http": {"defaultHealthyRetryPolicy": {"numRetries": 5}}})
+
+    attributes = conn.get_topic_attributes(topic_arn)['GetTopicAttributesResponse']['GetTopicAttributesResult']['Attributes']
+    attributes["Policy"].should.equal("{'foo': 'bar'}")
+    attributes["DisplayName"].should.equal("My display name")
+    attributes["DeliveryPolicy"].should.equal("{'http': {'defaultHealthyRetryPolicy': {'numRetries': 5}}}")