qoobaa-aws-sqs 0.1.1

data/test/sqs_test.rb

@@ -0,0 +1,197 @@
+require "test_helper"
+
+class TestSqs < Test::Unit::TestCase
+  GRANTEE_EMAIL_ADDRESS = 'fester@example.com'
+  MESSAGE_TEXT    = 'Test message'
+
+  def setup
+    @sqs = Aws::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+    @queue_name = 'sqs_test_queue'
+      # for classes
+    @s = Aws::Sqs.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+  end
+
+  # Wait for the queue to appear in the queues list.
+  # Amazon needs some time to after the queue creation to place
+  # it to the accessible queues list. If we dont want to get
+  # the additional faults then wait a bit...
+  def wait_for_queue_url(queue_name)
+    queue_url = nil
+    until queue_url
+      queue_url = @sqs.queue_url_by_name(queue_name)
+      unless queue_url
+        print '-'
+        STDOUT.flush
+        sleep 1
+      end
+    end
+    queue_url
+  end
+
+  #---------------------------
+  # Aws::SqsInterface
+  #---------------------------
+
+  def test_01_create_queue
+    queue_url = @sqs.create_queue @queue_name
+    assert queue_url[/http.*#{@queue_name}/], 'New queue creation failed'
+  end
+
+  def test_02_list_queues
+    wait_for_queue_url(@queue_name)
+    queues = @sqs.list_queues('sqs_test_')
+    assert queues.size>0, 'Must more that 0 queues in list'
+  end
+
+  def test_03_set_and_get_queue_attributes
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert queue_url[/http.*#{@queue_name}/], "#{@queue_name} must exist!"
+    assert @sqs.set_queue_attributes(queue_url, 'VisibilityTimeout', 111), 'Set_queue_attributes fail'
+    sleep 20 # Amazon needs some time to change attribute
+    assert_equal '111', @sqs.get_queue_attributes(queue_url)['VisibilityTimeout'], 'New VisibilityTimeout must be equal to 111'
+  end
+
+  def test_06_send_message
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+      # send 5 messages for the tests below
+    assert @sqs.send_message(queue_url, MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, MESSAGE_TEXT)
+    sleep 5
+  end
+
+  def test_07_get_queue_length
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert_equal 5, @sqs.get_queue_length(queue_url), 'Queue must have 5 messages'
+  end
+
+  def test_08_receive_message
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    r_message = @sqs.receive_message(queue_url, 1)[0]
+    assert r_message, "Receive returned no message(s), but this is not necessarily incorrect"
+    assert_equal MESSAGE_TEXT, r_message['Body'], 'Receive message got wrong message text'
+  end
+
+  def test_09_delete_message
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    message = @sqs.receive_message(queue_url)[0]
+    assert @sqs.delete_message(queue_url, message['ReceiptHandle']), 'Delete_message fail'
+    assert @sqs.pop_message(queue_url), 'Pop_message fail'
+  end
+
+  def test_10_clear_and_delete_queue
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert @sqs.delete_queue(queue_url)
+  end
+
+  #---------------------------
+  # Aws::Sqs classes
+  #---------------------------
+
+  def test_20_sqs_create_delete_queue
+    assert @s, 'Aws::Sqs must exist'
+      # get queues list
+    queues_size = @s.queues.size
+      # create new queue
+    queue  = @s.queue("#{@queue_name}_20", true)
+      # check that it is created
+    assert queue.is_a?(Aws::Sqs::Queue)
+    wait_for_queue_url(@queue_name)
+      # check that amount of queues has increased
+    assert_equal queues_size + 1, @s.queues.size
+      # delete queue
+    assert queue.delete
+  end
+
+  def test_21_queue_create
+      # create new queue
+    queue = Aws::Sqs::Queue.create(@s, "#{@queue_name}_21", true)
+      # check that it is created
+    assert queue.is_a?(Aws::Sqs::Queue)
+    wait_for_queue_url(@queue_name)
+  end
+
+  def test_22_queue_attributes
+    queue = Aws::Sqs::Queue.create(@s, "#{@queue_name}_21", false)
+      # get a list of attrinutes
+    attributes = queue.get_attribute
+    assert attributes.is_a?(Hash) && attributes.size>0
+      # get attribute value and increase it by 10
+    v = (queue.get_attribute('VisibilityTimeout').to_i + 10).to_s
+      # set attribute
+    assert queue.set_attribute('VisibilityTimeout', v)
+      # wait a bit
+    sleep 20
+      # check that attribute has changed
+    assert_equal v, queue.get_attribute('VisibilityTimeout')
+      # get queue visibility timeout
+    assert_equal v, queue.visibility
+      # change it
+    queue.visibility = queue.visibility.to_i + 10
+      # make sure that it is changed
+    assert v.to_i + 10, queue.visibility
+  end
+
+  def test_24_send_size
+    queue = Aws::Sqs::Queue.create(@s, "#{@queue_name}_24", true)
+      # send 5 messages
+    assert queue.push('a1')
+    assert queue.push('a2')
+    assert queue.push('a3')
+    assert queue.push('a4')
+    assert queue.push('a5')
+      # check queue size
+    assert_equal 5, queue.size
+      # send one more
+    assert queue.push('a6')
+      # check queue size again
+    assert_equal 6, queue.size
+  end
+
+  def test_25_message_receive_pop_delete
+    queue = Aws::Sqs::Queue.create(@s, "#{@queue_name}_24", false)
+      # get queue size
+    size = queue.size
+      # get first message
+    m1 = queue.receive(10)
+    assert m1.is_a?(Aws::Sqs::Message)
+      # pop second message
+    m2 = queue.pop
+    assert m2.is_a?(Aws::Sqs::Message)
+      # make sure that queue size has decreased
+    assert_equal size-1, queue.size
+      # delete messsage
+    assert m1.delete
+      # make sure that queue size has decreased again
+    assert_equal size-2, queue.size
+  end
+
+  def test_26
+    queue = Aws::Sqs::Queue.create(@s, "#{@queue_name}_24", false)
+      # lock message
+    queue.receive(100)
+      # clear queue
+    assert queue.clear
+      # queue size is greater than zero
+    assert queue.size>0
+  end
+
+  def test_27_set_amazon_problems
+    original_problems = Aws::SqsInterface.amazon_problems
+    assert(original_problems.length > 0)
+    Aws::SqsInterface.amazon_problems= original_problems << "A New Problem"
+    new_problems = Aws::SqsInterface.amazon_problems
+    assert_equal(new_problems, original_problems)
+
+    Aws::SqsInterface.amazon_problems= nil
+    assert_nil(Aws::SqsInterface.amazon_problems)
+  end
+
+  def test_28_check_threading_model
+    assert(!@sqs.multi_thread)
+    newsqs = Aws::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, {:multi_thread => true})
+    assert(newsqs.multi_thread)
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,101 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+require 'aws'
+
+module Net
+  class HTTPResponse
+  alias_method :real_body, :body
+    def setmsg(msg)
+      @mymsg = msg
+    end
+
+    def body
+      # defined?() helps us to get rid of a bunch of 'warnings'
+      (defined?(@mymsg) && @mymsg) ? @mymsg : real_body
+    end
+  end
+end
+
+module Aws
+  class HttpConnection
+    @@response_stack = []
+
+    alias_method :real_request, :request
+
+    def request(request_params, &block)
+      if(@@response_stack.length == 0)
+        return real_request(request_params, &block)
+      end
+
+      if(block)
+        # Do something special
+      else
+        next_response = HttpConnection::pop()
+        classname = Net::HTTPResponse::CODE_TO_OBJ["#{next_response[:code]}"]
+        response = classname.new("1.1", next_response[:code], next_response[:msg])
+        if(next_response[:msg])
+          response.setmsg(next_response[:msg])
+        end
+        response
+      end
+    end
+
+    def self.reset
+      @@response_stack = []
+    end
+
+    def self.push(code, msg=nil)
+      response = {:code => code, :msg => msg}
+      @@response_stack << response
+    end
+
+    def self.pop
+      @@response_stack.pop
+    end
+
+    def self.length
+      @@response_stack.length
+    end
+  end
+end
+
+class TestCredentials
+  @@aws_access_key_id = ""
+  @@aws_secret_access_key = ""
+  @@account_number = nil
+
+  def self.aws_access_key_id
+    @@aws_access_key_id
+  end
+  def self.aws_access_key_id=(newval)
+    @@aws_access_key_id = newval
+  end
+  def self.account_number
+    @@account_number
+  end
+  def self.account_number=(newval)
+    @@account_number = newval
+  end
+  def self.aws_secret_access_key
+    @@aws_secret_access_key
+  end
+  def self.aws_secret_access_key=(newval)
+    @@aws_secret_access_key = newval
+  end
+
+  def self.get_credentials
+    begin
+      require 'testcredentials'
+    rescue Exception => e
+      puts "Credentials not found in file test/testcredentials.rb"
+    end
+  end
+end
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification 
+name: qoobaa-aws-sqs
+version: !ruby/object:Gem::Version 
+  version: 0.1.1
+platform: ruby
+authors: 
+- "Jakub Ku\xC5\xBAma"
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-06-02 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: qoobaa@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- aws-sqs.gemspec
+- lib/awsbase/awsbase.rb
+- lib/awsbase/benchmark_fix.rb
+- lib/awsbase/http_connection.rb
+- lib/awsbase/support.rb
+- lib/sqs/sqs.rb
+- lib/sqs/sqs_interface.rb
+- test/sqs_test.rb
+- test/test_helper.rb
+has_rdoc: false
+homepage: http://github.com/qoobaa/aws-sqs
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 3
+summary: AWS SQS
+test_files: 
+- test/sqs_test.rb
+- test/test_helper.rb