Merge pull request iain#1 from behance/feature-message-attributes

Feature message attributes

fake_sqs.gemspec

@@ -20,8 +20,9 @@ Gem::Specification.new do |gem|
 
   gem.add_dependency "sinatra"
   gem.add_dependency "builder"
+  gem.add_dependency "deep_merge"
 
-  gem.add_development_dependency "rspec"
+  gem.add_development_dependency "rspec", "< 3.0"
   gem.add_development_dependency "rake"
   gem.add_development_dependency "aws-sdk"
   gem.add_development_dependency "faraday"

lib/fake_sqs.rb

@@ -11,6 +11,7 @@
 require 'fake_sqs/version'
 require 'fake_sqs/memory_database'
 require 'fake_sqs/file_database'
+require 'fake_sqs/extended_hash'
 
 module FakeSQS
 

lib/fake_sqs/actions/receive_message.rb

@@ -10,14 +10,27 @@ def initialize(options = {})
 
       def call(name, params)
         queue = @queues.get(name)
-        messages = queue.receive_message(params)
+        messages = queue.receive_message(params.dotted_to_nested_hash)
         @responder.call :ReceiveMessage do |xml|
           messages.each do |receipt, message|
             xml.Message do
               xml.MessageId message.id
               xml.ReceiptHandle receipt
               xml.MD5OfBody message.md5
               xml.Body message.body
+
+              # TODO: should only return the requested attribtues
+              message.message_attributes.each do |index, attribute|
+                xml.MessageAttribute do
+                  xml.Name attribute['Name']
+                  xml.Value do |value|
+                    xml.DataType attribute['Value']['DataType']
+
+                    # TODO: other types exist
+                    xml.StringValue attribute['Value']['StringValue']
+                  end
+                end
+              end
             end
           end
         end

lib/fake_sqs/actions/send_message.rb

@@ -10,7 +10,7 @@ def initialize(options = {})
 
       def call(name, params)
         queue = @queues.get(name)
-        message = queue.send_message(params)
+        message = queue.send_message(params.dotted_to_nested_hash)
         @responder.call :SendMessage do |xml|
           xml.MD5OfMessageBody message.md5
           xml.MessageId message.id

lib/fake_sqs/extended_hash.rb

@@ -0,0 +1,17 @@
+require 'deep_merge'
+
+module FakeSQS
+  module ExtendedHash
+    def dotted_to_nested_hash
+      self.map do |dotted_key, value|
+        dotted_key.split('.').reverse.inject(value) do |v, k|
+          {k => v}
+        end
+      end.inject(&:deep_merge)
+    end
+  end
+end
+
+class Hash
+  include FakeSQS::ExtendedHash
+end

lib/fake_sqs/message.rb

@@ -3,13 +3,14 @@
 module FakeSQS
   class Message
 
-    attr_reader :body, :id, :md5
+    attr_reader :body, :id, :md5, :message_attributes
     attr_accessor :visibility_timeout
 
     def initialize(options = {})
       @body = options.fetch("MessageBody")
       @id = options.fetch("Id") { SecureRandom.uuid }
       @md5 = options.fetch("MD5") { Digest::MD5.hexdigest(@body) }
+      @message_attributes = options.fetch("MessageAttribute", [])
     end
 
     def expire!
@@ -29,6 +30,7 @@ def attributes
         "MessageBody" => body,
         "Id" => id,
         "MD5" => md5,
+        "MessageAttribute" => message_attributes
       }
     end
 

lib/fake_sqs/show_output.rb

@@ -1,4 +1,6 @@
+require 'rack'
 require 'rack/request'
+require 'yaml'
 
 module FakeSQS
   class ShowOutput