revans_right_aws 2.0.1

data/test/sqs/test_helper.rb

@@ -0,0 +1,2 @@
+require 'test/unit'
+require File.dirname(__FILE__) + '/../../lib/right_aws'

data/test/sqs/test_right_sqs.rb

@@ -0,0 +1,291 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestSqs < Test::Unit::TestCase
+
+  GRANTEE_EMAIL_ADDRESS = 'madhur@amazon.com'
+  RIGHT_MESSAGE_TEXT    = 'Right test message'
+
+  
+  def setup
+    @sqs = Rightscale::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+    @queue_name = 'right_sqs_test_awesome_queue'
+      # for classes
+    @s = Rightscale::Sqs.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+  end
+  
+  # Wait for the queue to appears 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
+
+  
+
+  
+  def assert_eventually_equal(value, timeout=30, failmsg="", &block)
+    start_time = Time.now.to_i
+    tries = 0
+    while(yield != value) do
+      tries += 1
+      print '-' 
+      STDOUT.flush
+      s = Time.now.to_i - start_time
+      flunk("Timeout: #{failmsg}: did not equal \"#{value}\" after #{tries} tries in #{s}s.") if s > timeout
+      sleep(1)
+      setup if (tries % 10) == 0 
+    end
+  end
+  
+  
+  #---------------------------
+  # Rightscale::SqsInterface
+  #---------------------------
+
+  def test_01_create_queue
+    queue_url = @sqs.create_queue @queue_name
+    assert queue_url[/http.*#{@queue_name}/], 'New queue creation fail'
+  end
+
+  def test_02_list_queues
+    wait_for_queue_url(@queue_name)
+    queues = @sqs.list_queues('right_')
+    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_04_set_and_get_visibility_timeout
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert @sqs.set_visibility_timeout(queue_url, 222), 'Set_visibility_timeout fail'
+    sleep 20 # Amazon needs some time to change attribute
+    #assert_equal 222, @sqs.get_visibility_timeout(queue_url), 'Get_visibility_timeout must return to 222'
+    assert_eventually_equal(222, 60, 'Get_visibility_timeout must return to 222') do
+      @sqs.get_visibility_timeout(queue_url)
+    end
+  end
+  
+  def test_05_add_test_remove_grant
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert @sqs.add_grant(queue_url, GRANTEE_EMAIL_ADDRESS, 'FULLCONTROL'), 'Add grant fail'
+    grants_list = @sqs.list_grants(queue_url, GRANTEE_EMAIL_ADDRESS)
+    assert grants_list.size>0, 'List_grants must return at least 1 record for user #{GRANTEE_EMAIL_ADDRESS}'
+    assert @sqs.remove_grant(queue_url, GRANTEE_EMAIL_ADDRESS, 'FULLCONTROL'), 'Remove_grant fail'
+  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, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+  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)
+    assert_equal RIGHT_MESSAGE_TEXT, r_message[:body], 'Receive message get wron message text'
+    p_message = @sqs.peek_message(queue_url, r_message[:id])
+    assert_equal r_message[:body], p_message[:body], 'Received and Peeked messages must be equal'
+    assert @sqs.change_message_visibility(queue_url, r_message[:id], 0), 'Change_message_visibility fail'
+  end
+  
+  def test_09_delete_message
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    message = @sqs.receive_message(queue_url)
+    assert @sqs.delete_message(queue_url, message[:id]), '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_raise(Rightscale::AwsError) { @sqs.delete_queue(queue_url) }
+## oops, force_clear_queue does not work any more - amazon expects for 60 secs timeout between 
+## queue deletion and recreation...
+##    assert @sqs.force_clear_queue(queue_url), 'Force_clear_queue fail'
+    assert @sqs.clear_queue(queue_url), 'Clear_queue fail'
+    assert @sqs.delete_queue(queue_url), 'Delete_queue fail'
+  end
+  
+  #---------------------------
+  # Rightscale::Sqs classes
+  #---------------------------
+
+  def test_20_sqs_create_delete_queue
+    assert @s, 'Rightscale::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?(Rightscale::Sqs::Queue)
+    wait_for_queue_url(@queue_name)
+      # check that amount of queues has increased
+    assert_eventually_equal(queues_size + 1, 60, "The number of queues did not increase by one") do
+      @s.queues.size
+    end
+      # delete queue
+    assert queue.delete
+  end
+  
+  def test_21_queue_create
+      # create new queue
+    queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_21", true)
+      # check that it is created
+    assert queue.is_a?(Rightscale::Sqs::Queue)
+    wait_for_queue_url(@queue_name)
+  end
+  
+  def test_22_queue_attributes
+    queue = Rightscale::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 v.to_i, queue.visibility
+      # change it 
+    queue.visibility += 10
+      # make sure that it is changed
+    assert v.to_i + 10, queue.visibility
+  end
+  
+  def test_23_grantees
+    queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_21", false)
+      # get a list of grantees
+    grantees = queue.grantees
+      # well, queue must exist at least some seconds before we could add grantees to it....
+      # otherwise we get "Queue does not exists" message. Hence we use the queue
+      # has been created at previous step.
+      #
+      # create new grantee
+    grantee = Rightscale::Sqs::Grantee.new(queue, GRANTEE_EMAIL_ADDRESS)
+    assert grantee.perms.empty?
+      # grant perms
+    assert grantee.grant('FULLCONTROL')
+    assert grantee.grant('RECEIVEMESSAGE')
+    assert_equal 2, grantee.perms.size
+      # make sure that amount of grantees has increased
+    assert grantees.size < queue.grantees.size
+      # revoke perms
+    assert grantee.revoke('RECEIVEMESSAGE')
+    assert_equal 1, grantee.perms.size
+      # remove grantee
+    assert grantee.drop
+    # Don't test this - just for cleanup purposes
+    queue.delete
+  end
+  
+  def test_24_send_size
+    queue_url = @sqs.queue_url_by_name("#{@queue_name}_24")
+    @sqs.delete_queue(queue_url)
+    queue = Rightscale::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_peek_delete
+    queue = Rightscale::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?(Rightscale::Sqs::Message)
+      # pop second message
+    m2 = queue.pop
+    assert m2.is_a?(Rightscale::Sqs::Message)
+      # make sure that queue size has decreased
+    assert_equal size-1, queue.size
+      # peek message 1
+    m1p = queue.peek(m1.id)
+    assert m1p.is_a?(Rightscale::Sqs::Message)
+    assert_equal m1.id,   m1p.id
+    assert_equal m1.body, m1p.body
+      # change message visibility
+    assert m1.visibility = 30
+      # delete messsage
+    assert m1.delete
+      # make sure that queue size has decreased again
+    assert_equal size-2, queue.size
+  end
+  
+  def test_26
+    queue = Rightscale::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
+    queue.push('123456')
+    assert_raise(Rightscale::AwsError) { queue.delete }
+    assert queue.delete(true)
+  end
+
+  def test_27_set_amazon_problems
+    original_problems = Rightscale::SqsInterface.amazon_problems
+    assert(original_problems.length > 0)
+    Rightscale::SqsInterface.amazon_problems= original_problems << "A New Problem"
+    new_problems = Rightscale::SqsInterface.amazon_problems
+    assert_equal(new_problems, original_problems)
+
+    Rightscale::SqsInterface.amazon_problems= nil
+    assert_nil(Rightscale::SqsInterface.amazon_problems)
+  end
+
+  def test_28_check_threading_model
+    assert(!@sqs.multi_thread)
+    newsqs = Rightscale::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, {:multi_thread => true})
+    assert(newsqs.multi_thread)
+  end
+
+  def test_29_signature_version_0
+    sqs = Rightscale::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '0')
+    assert_nothing_raised do
+      sqs.list_queues
+    end
+    # check that the request has correct signature version
+    assert sqs.last_request.path.include?('SignatureVersion=0')
+  end
+  
+end

data/test/sqs/test_right_sqs_gen2.rb

@@ -0,0 +1,264 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestSqsGen2 < Test::Unit::TestCase
+
+  GRANTEE_EMAIL_ADDRESS = 'fester@example.com'
+  RIGHT_MESSAGE_TEXT    = 'Right test message'
+
+  
+  def setup
+    $stdout.sync = true
+    @grantee_aws_id = '100000000001'
+    @sqs = Rightscale::SqsGen2Interface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+    @queue_name  = 'right_sqs_test_gen2_queue'
+    @queue2_name = @queue_name + '_2'
+    @queue3_name = @queue_name + '_3'
+    @queue4_name = @queue_name + '_4'
+      # for classes
+    @s = Rightscale::SqsGen2.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
+    do_sleep(180) do
+      queue_url = @sqs.queue_url_by_name(queue_name)
+    end
+    sleep 30
+    queue_url
+  end
+
+  def do_sleep(delay, &block)
+    puts "sleeping #{block ? 'up to ' : ''}#{delay} seconds:"
+    wake_up_at = Time.now+delay
+    while Time.now < wake_up_at do
+      sleep 1
+      print '.'
+      break if block && block.call
+    end
+    puts
+  end
+    
+  #---------------------------
+  # Rightscale::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('right_')
+    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[/https.*#{@queue_name}/], "#{@queue_name} must exist!"
+    assert @sqs.set_queue_attributes(queue_url, 'VisibilityTimeout', 111), 'Set_queue_attributes fail'
+    do_sleep 60 # 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_04_get_queue_attributes_forms
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    all = nil
+    assert_nothing_raised do
+      all = @sqs.get_queue_attributes(queue_url, 'All')
+    end
+    assert_nothing_raised do
+      assert all, @sqs.get_queue_attributes(queue_url)
+    end
+    assert_nothing_raised do
+      attributes = @sqs.get_queue_attributes(queue_url, 'ApproximateNumberOfMessages', 'VisibilityTimeout')
+      assert_equal 2, attributes.size
+    end
+    assert_nothing_raised do
+      attributes = @sqs.get_queue_attributes(queue_url, ['ApproximateNumberOfMessages', 'VisibilityTimeout'])
+      assert_equal 2, attributes.size
+    end
+  end
+
+  def test_05_add_permissions
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert @sqs.add_permissions(queue_url, 'test01', @grantee_aws_id, 'SendMessage')
+    assert @sqs.add_permissions(queue_url, 'test02', @grantee_aws_id, ['DeleteMessage','ReceiveMessage'])
+    do_sleep 60
+  end
+
+  def test_06_test_permissions
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    permissions = @sqs.get_queue_attributes(queue_url, 'Policy')
+    assert !permissions.blank?
+  end
+
+  def test_07_revoke_permissions
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert @sqs.remove_permissions(queue_url, 'test01')
+    assert @sqs.remove_permissions(queue_url, 'test02')
+  end
+
+  def test_14_send_message
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+      # send 5 messages for the tests below
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
+    do_sleep 60
+  end
+  
+  def test_15_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_16_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 RIGHT_MESSAGE_TEXT, r_message['Body'], 'Receive message got wrong message text'
+  end
+  
+  def test_17_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_18_clear_and_delete_queue
+    queue_url = @sqs.queue_url_by_name(@queue_name)
+    assert @sqs.delete_queue(queue_url) 
+  end
+  
+  #---------------------------
+  # Rightscale::Sqs classes
+  #---------------------------
+
+  def test_20_sqs_create_queue
+    assert @s, 'Rightscale::SqsGen2 must exist'
+      # get queues list
+    queues_size = @s.queues.size
+      # create new queue
+    queue = @s.queue(@queue2_name, true)
+      # check that it is created
+    assert queue.is_a?(Rightscale::SqsGen2::Queue)
+    wait_for_queue_url(queue.name)
+      # check that amount of queues has increased
+    assert_equal queues_size + 1, @s.queues.size
+    do_sleep 10
+  end
+
+  def test_21_sqs_delete_queue
+    queue = @s.queue(@queue2_name, false)
+    assert queue.delete
+  end
+  
+  def test_22_queue_create
+      # create new queue
+    queue = Rightscale::SqsGen2::Queue.create(@s, @queue3_name, true)
+      # check that it is created
+    assert queue.is_a?(Rightscale::SqsGen2::Queue)
+    wait_for_queue_url("#{@queue_name}_21")
+    do_sleep 10
+  end
+  
+  def test_23_queue_attributes
+    queue = Rightscale::SqsGen2::Queue.create(@s, @queue3_name, 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
+    do_sleep 60
+      # 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
+    queue = Rightscale::SqsGen2::Queue.create(@s, @queue3_name, false)
+    assert queue.delete
+  end
+  
+  def test_25_send_size
+    queue = Rightscale::SqsGen2::Queue.create(@s, @queue4_name, true)
+      # send 5 messages
+    assert queue.push('a1')
+    assert queue.push('a2')
+    assert queue.push('a3')
+    assert queue.push('a4')
+    assert queue.push('a5')
+      #
+    do_sleep(300){ queue.size == 5 }
+      # check queue size
+    assert_equal 5, queue.size
+      # send one more
+    assert queue.push('a6')
+      #
+    do_sleep(300){ queue.size == 6 }
+      # check queue size again
+    assert_equal 6, queue.size
+  end
+  
+  def test_26_message_receive_pop_delete
+    queue = Rightscale::SqsGen2::Queue.create(@s,  @queue4_name, false)
+      # get queue size
+    size = queue.size
+      # get first message
+    m1 = queue.receive(10)
+    assert m1.is_a?(Rightscale::SqsGen2::Message)
+      # pop second message
+    m2 = queue.pop
+    assert m2.is_a?(Rightscale::SqsGen2::Message)
+      #
+    do_sleep 30
+      # make sure that queue size has decreased
+    assert_equal size-1, queue.size
+      # delete messsage
+    assert m1.delete
+      #
+    do_sleep 15
+      # make sure that queue size has decreased again
+    assert_equal size-2, queue.size
+  end
+ 
+  def test_27
+    queue = Rightscale::SqsGen2::Queue.create(@s, @queue4_name, false)
+      # lock message 
+    queue.receive(100)
+      # clear queue
+    assert queue.clear 
+      # queue size is greater than zero
+    assert queue.size>0
+      # delete queue
+    assert queue.delete
+  end
+
+  def test_28_set_amazon_problems
+    original_problems = Rightscale::SqsGen2Interface.amazon_problems
+    assert(original_problems.length > 0)
+    Rightscale::SqsGen2Interface.amazon_problems= original_problems << "A New Problem"
+    new_problems = Rightscale::SqsGen2Interface.amazon_problems
+    assert_equal(new_problems, original_problems)
+
+    Rightscale::SqsGen2Interface.amazon_problems= nil
+    assert_nil(Rightscale::SqsGen2Interface.amazon_problems)
+  end
+  
+end