spec/unit/queue_spec.rb in fake_sqs-0.0.11 vs spec/unit/queue_spec.rb in fake_sqs-0.1.0

@@ -1,16 +1,12 @@
 require 'fake_sqs/queue'
+require 'fake_sqs/message'
 
 describe FakeSQS::Queue do
 
   class MessageFactory
-
-    def initialize
-      @count = 0
-    end
-
     def new(options = {})
-      @count += 1
+      FakeSQS::Message.new({'MessageBody' => 'sample-body'}.merge(options))
     end
   end
 
   let(:message_factory) { MessageFactory.new }
   subject(:queue) { FakeSQS::Queue.new(:message_factory => message_factory, "QueueName" => "test-queue") }