fog-maestrodev 1.19.0.20140212012611 → 1.20.0.20140305101305

data/lib/fog/rackspace/requests/queues/create_claim.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation claims a set of messages (up to the value of the limit parameter) from oldest to newest and skips any messages that are already claimed.
@@ -33,7 +34,43 @@ module Fog
             :query => query
           )
         end
+
+      end
+
+      class Mock
+        def create_claim(queue_name, ttl, grace, options = {})
+          queue = mock_queue!(queue_name)
+
+          limit = options[:limit] || 10
+
+          claim = queue.add_claim(ttl, grace)
+
+          claimed = queue.messages.select do |message|
+            ! message.claimed?
+          end.first(limit)
+
+          if claimed.empty?
+            response = Excon::Response.new
+            response.status = 204
+            return response
+          end
+
+          claimed.each do |message|
+            message.claim = claim
+
+            # Extend the message's lifetime to include the lifetime of the claim, plus the claim's
+            # grace period.
+            message.extend_life
+          end
+
+          response = Excon::Response.new
+          response.status = 201
+          response.body = claimed.map { |msg| msg.to_h }
+          response.headers['Location'] = "#{PATH_BASE}/#{queue_name}/claims/#{claim.id}"
+          response
+        end
       end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/create_message.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation posts the specified message or messages.
@@ -31,6 +32,28 @@ module Fog
           )
         end
       end
+
+      class Mock
+        def create_message(client_id, queue_name, body, ttl)
+          queue = mock_queue!(queue_name)
+
+          raise BadRequest.new if body.nil? || body.empty?
+
+          # Ensure that any Symbol keys within +body+ are converted to Strings, just as being
+          # round-tripped through the API will.
+          converted = MockData.stringify(body)
+          message = queue.add_message(client_id, converted, ttl)
+
+          response = Excon::Response.new
+          response.status = 201
+          response.body = {
+            "partial" => false,
+            "resources" => ["#{PATH_BASE}/#{queue_name}/messages/#{message.id}"]
+          }
+          response
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/create_queue.rb

@@ -1,8 +1,8 @@
 module Fog
   module Rackspace
     class Queues
-      class Real
 
+      class Real
         # This operation creates a new queue.
         # The body of the request is empty.
         #
@@ -22,6 +22,20 @@ module Fog
           )
         end
       end
+
+      class Mock
+        def create_queue(queue_name)
+          raise MethodNotAllowed.new if queue_name.nil? || queue_name.empty?
+
+          existed = ! mock_queue(queue_name).nil?
+          add_queue(queue_name) unless existed
+
+          response = Excon::Response.new
+          response.status = existed ? 201 : 204
+          response
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/delete_claim.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation immediately releases a claim, making any remaining, undeleted) messages that are associated with the claim available to other workers.
@@ -22,6 +23,23 @@ module Fog
           )
         end
       end
+
+      class Mock
+        def delete_claim(queue_name, claim_id)
+          queue = mock_queue!(queue_name)
+          claim = queue.claim!(claim_id)
+
+          claim.messages.each do |message|
+            message.claim = nil
+          end
+          queue.claims.delete(claim_id)
+
+          response = Excon::Response.new
+          response.status = 204
+          response
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/delete_message.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation immediately deletes the specified message.
@@ -26,7 +27,40 @@ module Fog
             :query => query
           )
         end
+
+      end
+
+      class Mock
+        def delete_message(queue_name, message_id, options = {})
+          queue = mock_queue!(queue_name)
+
+          claim_id = options[:claim_id]
+
+          message = queue.messages.detect { |m| m.id == message_id }
+
+          perform_delete = true
+          if message && message.claimed?
+            unless message.claim.id == claim_id
+              # Currently returns a 204 without deleting!
+              perform_delete = false
+            end
+          else
+            if claim_id
+              # Currently returns a 204 without deleting!
+              perform_delete = false
+            end
+          end
+
+          if perform_delete
+            queue.messages.reject! { |m| m.id == message_id }
+          end
+
+          response = Excon::Response.new
+          response.status = 204
+          response
+        end
       end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/delete_queue.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation immediately deletes a queue and all of its existing messages.
@@ -19,7 +20,18 @@ module Fog
             :path => "queues/#{queue_name}"
           )
         end
+
       end
+
+      class Mock
+        def delete_queue(queue_name)
+          data.delete(queue_name)
+          response = Excon::Response.new
+          response.status = 204
+          response
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/get_claim.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation queries the specified claim for the specified queue. Claims with malformed IDs or claims that are not found by ID are ignored.
@@ -20,7 +21,21 @@ module Fog
             :path => "queues/#{queue_name}/claims/#{claim_id}"
           )
         end
+
+      end
+
+      class Mock
+        def get_claim(queue_name, claim_id)
+          queue = mock_queue!(queue_name)
+          claim = queue.claim!(claim_id)
+
+          response = Excon::Response.new
+          response.status = 200
+          response.body = claim.to_h
+          response
+        end
       end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/get_message.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation gets the specified message from the specified queue.
@@ -22,7 +23,23 @@ module Fog
             :headers => { 'Client-ID' => client_id }
           )
         end
+
       end
+
+      class Mock
+        def get_message(client_id, queue_name, message_id)
+          queue = mock_queue!(queue_name)
+
+          message = queue.messages.find { |msg| msg.id == message_id }
+          raise NotFound.new unless message
+
+          response = Excon::Response.new
+          response.status = 200
+          response.body = message.to_h
+          response
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/get_queue.rb

@@ -1,8 +1,8 @@
 module Fog
   module Rackspace
     class Queues
-      class Real
 
+      class Real
         # This operation verifies whether the specified queue exists.
         #
         # @param [String] queue_name Specifies the name of the queue.
@@ -20,6 +20,19 @@ module Fog
           )
         end
       end
+
+      class Mock
+        def get_queue(queue_name)
+          if mock_queue(queue_name).nil?
+            raise NotFound.new
+          else
+            response = Excon::Response.new
+            response.status = 204
+            response
+          end
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/get_queue_stats.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation returns queue statistics, including how many messages are in the queue, categorized by status.
@@ -19,7 +20,42 @@ module Fog
             :path => "queues/#{queue_name}/stats"
           )
         end
+
+      end
+
+      class Mock
+        def get_queue_stats(queue_name)
+          queue = mock_queue!(queue_name)
+
+          payload = {
+            "claimed" => queue.claimed,
+            "total" => queue.total,
+            "free" => queue.free
+          }
+
+          report_message(payload, "oldest", queue.oldest)
+          report_message(payload, "newest", queue.newest)
+
+          response = Excon::Response.new
+          response.status = 200
+          response.body = { "messages" => payload }
+          response
+        end
+
+        private
+
+        def report_message(payload, description, element)
+          return unless element
+
+          ctime = Time.at(element.created).utc
+          payload[description] = {
+            "age" => element.age,
+            "href" => element.href,
+            "created" => ctime.strftime("%Y-%m-%dT%I:%M:%SZ")
+          }
+        end
       end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/list_messages.rb

@@ -1,6 +1,7 @@
 module Fog
   module Rackspace
     class Queues
+
       class Real
 
         # This operation gets the message or messages in the specified queue.
@@ -38,6 +39,38 @@ module Fog
           )
         end
       end
+
+      class Mock
+        def list_messages(client_id, queue_name, options = {})
+          queue = mock_queue!(queue_name)
+
+          marker = (options[:marker] || "0").to_i
+          limit = options[:limit] || 10
+          echo = options[:echo] || false
+          include_claimed = options[:include_claimed] || false
+
+          next_marker = marker + limit + 1
+          messages = queue.messages[marker...next_marker]
+          messages.reject! { |m| m.producer_id == client_id } unless echo
+          messages.reject! { |m| m.claimed? } unless include_claimed
+
+          response = Excon::Response.new
+          if queue.messages.empty?
+            response.status = 204
+          else
+            response.status = 200
+            response.body = {
+              "messages" => messages.map { |m| m.to_h },
+              "links" => [{
+                "href" => "#{PATH_BASE}/#{queue_name}/messages?marker=#{next_marker}",
+                "rel" => "next"
+              }]
+            }
+          end
+          response
+        end
+      end
+
     end
   end
 end

data/lib/fog/rackspace/requests/queues/list_queues.rb

@@ -1,8 +1,8 @@
 module Fog
   module Rackspace
     class Queues
-      class Real
 
+      class Real
         # This operation lists queues for the project. The queues are sorted alphabetically by name.
         # @note A request to list queues when you have no queues in your account returns 204, instead of 200, because there was no information to send back.
         #
@@ -25,6 +25,40 @@ module Fog
           )
         end
       end
+
+      class Mock
+        def list_queues(options={})
+          limit = options[:limit] || 10
+          marker = options[:marker]
+          detailed = options[:detailed] || false
+
+          queue_names = data.keys.sort
+          start_index = marker.nil? ? 0 : queue_names.count { |name| name <= marker }
+          stop_index = start_index + limit
+
+          queue_names = queue_names[start_index..stop_index]
+          queue_data = queue_names.map do |qname|
+            { "href" => "#{PATH_BASE}/#{qname}", "name" => qname }
+          end
+
+          if detailed
+            queue_data.each { |d| d["metadata"] = data[d["name"]].metadata }
+          end
+
+          response = Excon::Response.new
+          if data.empty?
+            response.status = 204
+          else
+            response.status = 200
+            response.body = {
+              "queues" => queue_data,
+              "links" => [{ "href" => "#{PATH_BASE}?marker=#{queue_names.last}", "rel" => "next" }]
+            }
+          end
+          response
+        end
+      end
+
     end
   end
 end