fake_sns 0.0.1

data/lib/fake_sns/topic_collection.rb

@@ -0,0 +1,41 @@
+module FakeSNS
+  class TopicCollection
+
+    include Enumerable
+
+    def initialize(store)
+      @store = store
+      @store["topics"] ||= []
+    end
+
+    def collection
+      @store["topics"]
+    end
+
+    def reset
+      @store["topics"] = []
+    end
+
+    def each(*args, &block)
+      p @store.to_yaml
+      collection.map { |item| Topic.new(item) }.each(*args, &block)
+    end
+
+    def fetch(arn, &default)
+      default ||= -> { raise InvalidParameterValue, "Unknown topic #{arn}" }
+      found = collection.find do |topic|
+        topic["arn"] == arn
+      end
+      found || default.call
+    end
+
+    def create(attributes)
+      collection << attributes
+    end
+
+    def delete(arn)
+      collection.delete(fetch(arn))
+    end
+
+  end
+end

data/lib/fake_sns/version.rb

@@ -0,0 +1,3 @@
+module FakeSNS
+  VERSION = "0.0.1"
+end

data/lib/fake_sns/views/create_topic.xml.erb

@@ -0,0 +1,8 @@
+<CreateTopicResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <CreateTopicResult>
+    <TopicArn><%= arn %></TopicArn>
+  </CreateTopicResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</CreateTopicResponse>

data/lib/fake_sns/views/delete_topic.xml.erb

@@ -0,0 +1,5 @@
+<DeleteTopicResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</DeleteTopicResponse>

data/lib/fake_sns/views/error.xml.erb

@@ -0,0 +1,8 @@
+<ErrorResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <Error>
+    <Type><%= type %></Type>
+    <Code><%= code %></Code>
+    <Message><%= message %></Message>
+  </Error>
+  <RequestId><%= request_id %></RequestId>
+</ErrorResponse>

data/lib/fake_sns/views/get_topic_attributes.xml.erb

@@ -0,0 +1,15 @@
+<GetTopicAttributesResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <GetTopicAttributesResult>
+    <Attributes>
+      <% each_attribute do |key, value| %>
+      <entry>
+        <key><%= key %></key>
+        <value><%= value %></value>
+      </entry>
+      <% end %>
+    </Attributes>
+  </GetTopicAttributesResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</GetTopicAttributesResponse>

data/lib/fake_sns/views/list_subscriptions.xml.erb

@@ -0,0 +1,18 @@
+<ListSubscriptionsResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <ListSubscriptionsResult>
+    <Subscriptions>
+      <% each_subscription do |subscription| %>
+        <member>
+          <TopicArn><%= subscription.topic_arn %></TopicArn>
+          <Protocol><%= subscription.protocol %></Protocol>
+          <SubscriptionArn><%= subscription.arn %></SubscriptionArn>
+          <Owner><%= subscription.owner %></Owner>
+          <Endpoint><%= subscription.endpoint %></Endpoint>
+        </member>
+      <% end %>
+    </Subscriptions>
+  </ListSubscriptionsResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</ListSubscriptionsResponse>

data/lib/fake_sns/views/list_subscriptions_by_topic.xml.erb

@@ -0,0 +1,18 @@
+<ListSubscriptionsByTopicResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <ListSubscriptionsByTopicResult>
+    <Subscriptions>
+      <% each_subscription do |subscription| %>
+        <member>
+          <TopicArn><%= subscription.topic_arn %></TopicArn>
+          <Protocol><%= subscription.protocol %></Protocol>
+          <SubscriptionArn><%= subscription.arn %></SubscriptionArn>
+          <Owner><%= subscription.owner %></Owner>
+          <Endpoint><%= subscription.endpoint %></Endpoint>
+        </member>
+      <% end %>
+    </Subscriptions>
+  </ListSubscriptionsByTopicResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</ListSubscriptionsByTopicResponse>

data/lib/fake_sns/views/list_topics.xml.erb

@@ -0,0 +1,14 @@
+<ListTopicsResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <ListTopicsResult>
+    <Topics>
+      <% each_topic do |topic| %>
+        <member>
+          <TopicArn><%= topic.arn %></TopicArn>
+        </member>
+      <% end %>
+    </Topics>
+  </ListTopicsResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</ListTopicsResponse>

data/lib/fake_sns/views/publish.xml.erb

@@ -0,0 +1,8 @@
+<PublishResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <PublishResult>
+    <MessageId><%= message_id %></MessageId>
+  </PublishResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</PublishResponse>

data/lib/fake_sns/views/set_topic_attributes.xml.erb

@@ -0,0 +1,5 @@
+<SetTopicAttributesResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</SetTopicAttributesResponse>

data/lib/fake_sns/views/subscribe.xml.erb

@@ -0,0 +1,8 @@
+<SubscribeResponse xmlns="http://sns.amazonaws.com/doc/2010-03-31/">
+  <SubscribeResult>
+    <SubscriptionArn><%= subscription_arn %></SubscriptionArn>
+  </SubscribeResult>
+  <ResponseMetadata>
+    <RequestId><%= request_id %></RequestId>
+  </ResponseMetadata>
+</SubscribeResponse>

data/lib/fake_sns.rb

@@ -0,0 +1,51 @@
+Encoding.default_external = Encoding::UTF_8
+Encoding.default_internal = Encoding::UTF_8
+
+require "virtus"
+require "verbose_hash_fetch"
+
+require "fake_sns/error"
+require "fake_sns/error_response"
+
+require "fake_sns/database"
+require "fake_sns/storage"
+
+require "fake_sns/topic_collection"
+require "fake_sns/topic"
+require "fake_sns/subscription_collection"
+require "fake_sns/subscription"
+
+require "fake_sns/message_collection"
+require "fake_sns/message"
+require "fake_sns/deliver_message"
+
+require "fake_sns/response"
+require "fake_sns/action"
+require "fake_sns/server"
+
+# load all the actions
+action_files = File.expand_path("../fake_sns/actions/*.rb", __FILE__)
+Dir.glob(action_files).each do |file|
+  require file
+end
+
+module FakeSNS
+
+  def self.server(options)
+    app = Server
+    if log = options[:log]
+      $stdout.reopen(log, "w:utf-8")
+      $stderr.reopen(log, "w:utf-8")
+      app.enable :logging
+    end
+    if options[:verbose]
+      require "fake_sns/show_output"
+      app.use FakeSNS::ShowOutput
+    end
+    options.each do |key, value|
+      app.set key, value
+    end
+    app
+  end
+
+end

data/spec/fake_sns/drain_spec.rb

@@ -0,0 +1,91 @@
+require "spec_helper"
+require "sinatra/base"
+require 'json_expressions/rspec'
+
+describe "Drain messages", :sqs do
+
+  it "works for SQS" do
+    topic = sns.topics.create("my-topic")
+    queue = sqs.queues.create("my-queue")
+    topic.subscribe(queue)
+
+    topic.publish("X")
+
+    $fake_sns.drain
+    expect(queue.visible_messages).to eq 1
+  end
+
+  it "works for SQS with a single message" do
+    topic = sns.topics.create("my-topic")
+    queue = sqs.queues.create("my-queue")
+    topic.subscribe(queue)
+
+    message_id = topic.publish("X")
+    topic.publish("Y")
+
+    $fake_sns.drain(message_id)
+    expect(queue.visible_messages).to eq 1
+  end
+
+  it "works for HTTP" do
+    requests = []
+    _headers = []
+    target_app = Class.new(Sinatra::Base) do
+      get("/") { 200 } # check if server started
+      post("/endpoint") do
+        requests << request.body.read
+        _headers << request.env
+        200
+      end
+    end
+
+    app_runner = Thread.new do
+      target_app.set :port, 5051
+      target_app.run!
+    end
+
+    topic = sns.topics.create("my-topic")
+    subscription = topic.subscribe("http://localhost:5051/endpoint")
+
+    message_id = topic.publish("X")
+
+    wait_for { Faraday.new("http://localhost:5051").get("/").success? rescue false }
+
+    $fake_sns.drain
+
+    app_runner.kill
+
+    expect(requests.size).to eq 1
+    expect(requests.first).to match_json_expression(
+      "Type"             => "Notification",
+      "Message"          => "X",
+      "MessageId"        => message_id,
+      "Signature"        => "Fake",
+      "SignatureVersion" => "1",
+      "SigningCertURL"   => "https://sns.us-east-1.amazonaws.com/SimpleNotificationService-f3ecfb7224c7233fe7bb5f59f96de52f.pem",
+      "Subject"          => nil,
+      "Timestamp"        => anything,
+      "TopicArn"         => topic.arn,
+      "Type"             => "Notification",
+      "UnsubscribeURL"   => "",
+    )
+
+    expect(_headers.size).to eq 1
+    expect(_headers.first).to match_json_expression({
+      "HTTP_X_AMZ_SNS_MESSAGE_TYPE"     => "Notification",
+      "HTTP_X_AMZ_SNS_MESSAGE_ID"       => message_id,
+      "HTTP_X_AMZ_SNS_TOPIC_ARN"        => topic.arn,
+      "HTTP_X_AMZ_SNS_SUBSCRIPTION_ARN" => subscription.arn,
+    }.ignore_extra_keys!)
+
+  end
+
+  def wait_for(&condition)
+    Timeout.timeout 1 do
+      until condition.call
+        sleep 0.01
+      end
+    end
+  end
+
+end

data/spec/fake_sns/publish_spec.rb

@@ -0,0 +1,28 @@
+require "spec_helper"
+
+describe "Publishing" do
+
+  let(:existing_topic) { sns.topics.create("my-topic") }
+
+  it "remembers published messages" do
+    message_id = existing_topic.publish("hallo")
+    messages = $fake_sns.data.fetch("messages")
+    expect(messages.size).to eq 1
+    message = messages.first
+    expect(message.fetch(:id)).to eq message_id
+  end
+
+  it "needs an existing topic" do
+    topic = sns.topics["arn:aws:sns:us-east-1:5068edfd0f7ee3ea9ccc1e73cbb17569:not-exist"]
+    expect {
+      topic.publish("hallo")
+    }.to raise_error AWS::SNS::Errors::InvalidParameterValue
+  end
+
+  it "doesn't allow messages that are too big" do
+    expect {
+      existing_topic.publish("A" * 262145)
+    }.to raise_error AWS::SNS::Errors::InvalidParameterValue
+  end
+
+end

data/spec/fake_sns/replace_spec.rb

@@ -0,0 +1,14 @@
+require "spec_helper"
+
+describe "Replacing the data" do
+
+  it "works by submitting new YAML" do
+    topic = sns.topics.create("my-topic")
+    previous_data = $fake_sns.data.to_yaml
+    apply_change = lambda { |data| data.gsub("my-topic", "your-topic") }
+    new_data = apply_change.call(previous_data)
+    $fake_sns.connection.put("/", new_data)
+    expect(sns.topics.map(&:arn)).to eq [apply_change.call(topic.arn)]
+  end
+
+end

data/spec/fake_sns/subscribing_spec.rb

@@ -0,0 +1,42 @@
+require "spec_helper"
+
+describe "Subscribing", :sqs do
+
+  it "lists subscriptions globally" do
+    topic = sns.topics.create("my-topic")
+    subscription = topic.subscribe("http://example.com")
+    expect(sns.subscriptions.map(&:topic_arn)).to eq [topic.arn]
+    expect(sns.subscriptions.map(&:arn)).to eq [subscription.arn]
+  end
+
+  it "filters by topic" do
+    topic = sns.topics.create("my-topic")
+    other_topic = sns.topics.create("my-topic-2")
+    subscription = topic.subscribe("http://example.com")
+    expect(topic.subscriptions.map(&:topic_arn)).to eq [topic.arn]
+    expect(topic.subscriptions.map(&:arn)).to eq [subscription.arn]
+    expect(other_topic.subscriptions.map(&:arn)).to eq []
+  end
+
+  it "needs an existing topic" do
+    topic = sns.topics["arn:aws:sns:us-east-1:5068edfd0f7ee3ea9ccc1e73cbb17569:not-exist"]
+    expect {
+      topic.subscribe("http://example.com")
+    }.to raise_error AWS::SNS::Errors::InvalidParameterValue
+  end
+
+  it "can subscribe to a SQS queue" do
+    queue = AWS::SQS.new.queues.create("my-queue")
+    topic = sns.topics.create("my-topic")
+    topic.subscribe(queue)
+  end
+
+  it "won't subscribe twice to the same endpoint" do
+    queue = AWS::SQS.new.queues.create("my-queue")
+    topic = sns.topics.create("my-topic")
+    topic.subscribe(queue)
+    topic.subscribe(queue)
+    expect(sns.subscriptions.to_a.size).to eq 1
+  end
+
+end

data/spec/fake_sns/topics_spec.rb

@@ -0,0 +1,44 @@
+require "spec_helper"
+
+describe "Topics" do
+
+  it "rejects invalid characters in topic names" do
+    expect {
+      sns.topics.create("dot.dot")
+    }.to raise_error(AWS::SNS::Errors::InvalidParameterValue)
+  end
+
+  it "creates a new topic" do
+    topic = sns.topics.create("my-topic")
+    expect(topic.arn).to match(/arn:aws:sns:us-east-1:(\w+):my-topic/)
+
+    new_topic = sns.topics.create("other-topic")
+    expect(new_topic.arn).not_to eq topic.arn
+
+    existing_topic = sns.topics.create("my-topic")
+    expect(existing_topic.arn).to eq topic.arn
+  end
+
+  it "lists topics" do
+    topic1 = sns.topics.create("my-topic-1")
+    topic2 = sns.topics.create("my-topic-2")
+
+    expect(sns.topics.map(&:arn)).to match_array [topic1.arn, topic2.arn]
+  end
+
+  it "deletes topics" do
+    topic = sns.topics.create("my-topic")
+    expect(sns.topics.map(&:arn)).to eq [topic.arn]
+    sns.topics[topic.arn].delete
+    expect(sns.topics.map(&:arn)).to eq []
+  end
+
+  it "can set and read attributes" do
+    topic = sns.topics.create("my-topic")
+    expect(topic.display_name).to eq "my-topic"
+    topic.display_name = "the display name"
+    expect(topic.display_name).to eq "the display name"
+    expect(topic.name).to eq "my-topic"
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,54 @@
+ENV["RACK_ENV"] ||= 'test'
+
+require "bundler/setup"
+Bundler.setup
+
+require "verbose_hash_fetch"
+require "aws-sdk"
+require "pry"
+require "fake_sns/test_integration"
+require "fake_sqs/test_integration"
+
+
+AWS.config(
+  use_ssl:            false,
+  sqs_endpoint:       "localhost",
+  sqs_port:           4568,
+  sns_endpoint:       "localhost",
+  sns_port:           9293,
+  access_key_id:      "fake access key",
+  secret_access_key:  "fake secret key",
+)
+
+db = ENV["SNS_DATABASE"]
+db = ":memory:" if ENV["SNS_DATABASE"].to_s == ""
+
+puts "Running tests with database stored in #{db}"
+
+$fake_sns = FakeSNS::TestIntegration.new(database: db)
+$fake_sqs = FakeSQS::TestIntegration.new(database: ":memory:")
+
+module SpecHelper
+  def sns
+    AWS::SNS.new
+  end
+  def sqs
+    AWS::SQS.new
+  end
+end
+
+RSpec.configure do |config|
+
+  config.treat_symbols_as_metadata_keys_with_true_values = true
+  config.expect_with :rspec do |rspec|
+    rspec.syntax = :expect
+  end
+
+  config.before(:each) { $fake_sns.start }
+  config.after(:suite) { $fake_sns.stop }
+  config.include SpecHelper
+
+  config.before(:each, :sqs) { $fake_sqs.start }
+  config.after(:suite) { $fake_sqs.stop }
+
+end