fake_sqs 0.0.8 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 909e1da7c44403638eb5d95fe363a7d2c7282135
-  data.tar.gz: 94f87e79eef035cf8c6d89f0de61d1cd4e5f951c
+  metadata.gz: c424f4af6489d03024c8cc27bc894df0ffd8e065
+  data.tar.gz: 81d0da9ebe28dd1d47808d14f91f6a0d490d0bef
 SHA512:
-  metadata.gz: 9ace909c2bcf251ea0234b0ceb5804bf5b5780765d41ea838bff230a7a8765caa0a32b6792eace91783b8a0a73022d93bf5e5d815236e9ec1a8e8aa1f236014f
-  data.tar.gz: 2e7baaa00ba1fe6da88e20276edad298813f7477c3be0fb82bc0c988e6ddfab7cd33ff211c058677da08b53633369b80b0ef6de3382a7bed7463d78d0af85b43
+  metadata.gz: 6f8286741675fb8beb9e8206ce5102bc82183386c5ea2297917fb44e9d03ff876265c1dd3c0838717fd7b34496e4de31becb0e1a7f276c155ccbf8b3619cc2cb
+  data.tar.gz: f0f65a2c48b86beb6a935843cff6c37fe38148a4723add4176a5c24f264ca9afd1ff5b60054ee6ad97f5b48fb89d4e8e380470a40ff537d4a57ef816de75cd50

data/bin/fake_sqs

@@ -1,6 +1,7 @@
 #!/usr/bin/env ruby
 
-$LOAD_PATH.unshift(File.expand_path("../../lib", __FILE__))
+lib = File.expand_path("../../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 require 'fake_sqs'
 require 'optparse'

data/lib/fake_sqs/actions/set_queue_attributes.rb

@@ -16,7 +16,7 @@ module FakeSQS
             results[value] = params.fetch("Attribute.#{$1}.Value")
           end
         end
-        queue.attributes.merge!(results)
+        queue.add_queue_attributes(results)
         @responder.call :SetQueueAttributes
       end
 

data/lib/fake_sqs/queue.rb

@@ -6,16 +6,28 @@ module FakeSQS
 
   class Queue
 
-    attr_reader :name, :messages, :message_factory, :messages_in_flight, :attributes
+    attr_reader :name, :messages, :message_factory, :messages_in_flight, :arn
 
     def initialize(options = {})
       @name = options.fetch("QueueName")
       @message_factory = options.fetch(:message_factory)
-      @attributes = {}
-      @attributes["QueueArn"] = "arn:aws:sqs:us-east-1:#{SecureRandom.hex}:#{@name}"
+      @arn = "arn:aws:sqs:us-east-1:#{SecureRandom.hex}:#{@name}"
+      @queue_attributes = {}
       reset
     end
 
+    def add_queue_attributes(attrs)
+      @queue_attributes.merge!(attrs)
+    end
+
+    def attributes
+      @queue_attributes.merge(
+        "QueueArn" => arn,
+        "ApproximateNumberOfMessages" => messages.size,
+        "ApproximateNumberOfMessagesNotVisible" => messages_in_flight.size,
+      )
+    end
+
     def send_message(options = {})
       message = message_factory.new(options)
       messages << message

data/lib/fake_sqs/version.rb

@@ -1,3 +1,3 @@
 module FakeSQS
-  VERSION = "0.0.8"
+  VERSION = "0.0.9"
 end

data/spec/unit/queue_spec.rb

@@ -54,7 +54,7 @@ describe FakeSQS::Queue do
 
       sample_group.times do
         sent_first     = send_message
-        sent_second    = send_message
+        _    = send_message
         message = receive_message.values.first
         if message == sent_first
           indexes[:first] += 1
@@ -88,10 +88,18 @@ describe FakeSQS::Queue do
     end
 
     it "keeps track of sent messages" do
+
       send_message
+
       queue.should have(0).messages_in_flight
+      queue.attributes["ApproximateNumberOfMessagesNotVisible"].should eq 0
+      queue.attributes["ApproximateNumberOfMessages"].should eq 1
+
       receive_message
+
       queue.should have(1).messages_in_flight
+      queue.attributes["ApproximateNumberOfMessagesNotVisible"].should eq 1
+      queue.attributes["ApproximateNumberOfMessages"].should eq 0
     end
 
     it "gets multiple message" do
@@ -117,7 +125,7 @@ describe FakeSQS::Queue do
   describe "#delete_message" do
 
     it "deletes by the receipt" do
-      message = send_message
+      send_message
       receipt = receive_message.keys.first
 
       queue.should have(1).messages_in_flight
@@ -132,6 +140,20 @@ describe FakeSQS::Queue do
 
   end
 
+  describe "#add_queue_attributes" do
+
+    it "adds to it's queue attributes" do
+      queue.add_queue_attributes("foo" => "bar")
+      queue.attributes.should eq(
+        "foo"                                   => "bar",
+        "QueueArn"                              => queue.arn,
+        "ApproximateNumberOfMessages"           => 0,
+        "ApproximateNumberOfMessagesNotVisible" => 0
+      )
+    end
+
+  end
+
   def send_message(options = {})
     queue.send_message(options)
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: fake_sqs
 version: !ruby/object:Gem::Version
-  version: 0.0.8
+  version: 0.0.9
 platform: ruby
 authors:
 - iain
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-12-05 00:00:00.000000000 Z
+date: 2013-12-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sinatra