right_aws 1.1.0

data/test/awsbase/test_helper.rb

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

data/test/awsbase/test_right_awsbase.rb

@@ -0,0 +1,12 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+require 'pp'
+
+class TestAwsbase < Test::Unit::TestCase
+
+  def setup
+  end
+  
+  def test_01_create_describe_key_pairs
+  end
+  
+end

data/test/ec2/test_helper.rb

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

data/test/ec2/test_right_ec2.rb

@@ -0,0 +1,67 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+require 'pp'
+
+class TestEc2 < Test::Unit::TestCase
+
+    # Some of RightEc2 instance methods concerning instance launching and image registration
+    # are not tested here due to their potentially risk.
+  
+  def setup
+    @ec2   = Rightscale::Ec2.new(TestCredentials.aws_access_key_id,
+                                 TestCredentials.aws_secret_access_key)
+    @key   = 'right_ec2_awesome_test_key'
+    @group = 'right_ec2_awesome_test_security_group'
+  end
+  
+  def test_01_create_describe_key_pairs
+    new_key = @ec2.create_key_pair(@key)
+    assert new_key[:aws_material][/BEGIN RSA PRIVATE KEY/], "New key material is absent"
+    keys = @ec2.describe_key_pairs
+    assert keys.map{|key| key[:aws_key_name] }.include?(@key), "#{@key} must exist"
+  end
+  
+  def test_02_create_security_group
+    assert @ec2.create_security_group(@group,'My awesone test group'), 'Create_security_group fail'
+    group = @ec2.describe_security_groups([@group])[0]
+    assert_equal @group, group[:aws_group_name], 'Group must be created but does not exist'
+  end
+  
+  def test_03_perms_add
+    assert @ec2.authorize_security_group_named_ingress(@group, TestCredentials.account_number, 'default')
+    assert @ec2.authorize_security_group_IP_ingress(@group, 80,80,'udp','192.168.1.0/8')
+  end
+  
+  def test_04_check_new_perms_exist
+    assert_equal 2, @ec2.describe_security_groups([@group])[0][:aws_perms].size
+  end
+
+  def test_05_perms_remove
+    assert @ec2.revoke_security_group_IP_ingress(@group, 80,80,'udp','192.168.1.0/8')
+    assert @ec2.revoke_security_group_named_ingress(@group,
+                                                    TestCredentials.account_number, 'default')
+  end
+
+  def test_06_describe_images
+    images = @ec2.describe_images
+    assert images.size>0, 'Amazon must have at least some public images'
+      # unknown image
+    assert_raise(Rightscale::AwsError){ @ec2.describe_images(['ami-ABCDEFGH'])}
+  end
+
+  def test_07_describe_instanses
+    assert @ec2.describe_instances
+      # unknown image
+    assert_raise(Rightscale::AwsError){ @ec2.describe_instances(['i-ABCDEFGH'])}
+  end
+
+  def test_08_delete_security_group
+    assert @ec2.delete_security_group(@group), 'Delete_security_group fail'
+  end
+  
+  def test_09_delete_key_pair
+    assert @ec2.delete_key_pair(@key), 'Delete_key_pair fail'
+      # key must be deleted already
+    assert_raise(Rightscale::AwsError) { @ec2.delete_key_pair(@key) }
+  end
+  
+end

data/test/s3/test_helper.rb

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

data/test/s3/test_right_s3.rb

@@ -0,0 +1,217 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestS3 < Test::Unit::TestCase
+
+  RIGHT_OBJECT_TEXT     = 'Right test message'
+  
+  def setup
+    @s3     = Rightscale::S3Interface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+    @bucket = 'right_s3_awesome_test_bucket'
+    @key1   = 'test/woohoo1'
+    @key2   = 'test1/key/woohoo2'
+    @s      = Rightscale::S3.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+  end
+
+  #---------------------------
+  # Rightscale::S3Interface
+  #---------------------------
+  
+  def test_01_create_bucket
+    assert @s3.create_bucket(@bucket), 'Create_bucket fail'
+  end
+
+  def test_02_list_all_my_buckets
+    assert @s3.list_all_my_buckets.map{|bucket| bucket[:name]}.include?(@bucket), "#{@bucket} must exist in bucket list"
+  end
+
+  def test_03_list_empty_bucket
+    assert_equal 0, @s3.list_bucket(@bucket).size, "#{@queue_name} must exist!"
+  end
+  
+  def test_04_put
+    assert @s3.put(@bucket, @key1, RIGHT_OBJECT_TEXT, 'x-amz-meta-family'=>'Woohoo1!'), 'Put bucket fail'
+    assert @s3.put(@bucket, @key2, RIGHT_OBJECT_TEXT, 'x-amz-meta-family'=>'Woohoo2!'), 'Put bucket fail'
+  end
+  
+  def test_05_get_and_get_object
+    assert_raise(Rightscale::AwsError) { @s3.get(@bucket, 'undefined/key') }
+    data1 = @s3.get(@bucket, @key1)
+    assert_equal RIGHT_OBJECT_TEXT, data1[:object], "Object text must be equal to '#{RIGHT_OBJECT_TEXT}'"
+    assert_equal RIGHT_OBJECT_TEXT, @s3.get_object(@bucket, @key1), "Get_object text must return '#{RIGHT_OBJECT_TEXT}'"
+    assert_equal 'Woohoo1!', data1[:headers]['x-amz-meta-family'], "x-amz-meta-family header must be equal to 'Woohoo1!'"
+  end
+  
+  def test_06_head
+    assert_equal 'Woohoo1!', @s3.head(@bucket,@key1)['x-amz-meta-family'], "x-amz-meta-family header must be equal to 'Woohoo1!'"
+  end
+  
+  def test_07_delete_folder
+    assert_equal 1, @s3.delete_folder(@bucket, 'test').size, "Only one key(#{@key1}) must be deleted!"
+  end
+
+  def test_08_delete_bucket
+    assert_raise(Rightscale::AwsError) { @s3.delete_bucket(@bucket) }
+    assert @s3.clear_bucket(@bucket), 'Clear_bucket fail'
+    assert_equal 0, @s3.list_bucket(@bucket).size, 'Bucket must be empty'
+    assert @s3.delete_bucket(@bucket)
+    assert !@s3.list_all_my_buckets.map{|bucket| bucket[:name]}.include?(@bucket), "#{@bucket} must not exist"
+  end
+
+  #---------------------------
+  # Rightscale::S3 classes
+  #---------------------------
+
+  def test_20_s3
+      # create bucket
+    bucket = @s.bucket(@bucket, true)
+    assert bucket
+      # check that the bucket exists
+    assert @s.buckets.map{|b| b.name}.include?(@bucket)
+      # delete bucket
+    assert bucket.clear
+    assert bucket.delete
+  end
+
+  def test_21_bucket_create_put_get_key
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, true)
+      # check that the bucket exists
+    assert @s.buckets.map{|b| b.name}.include?(@bucket)
+    assert bucket.keys.empty?
+      # put data
+    assert bucket.put(@key1, RIGHT_OBJECT_TEXT, {'family'=>'123456'})
+      # get data and compare
+    assert_equal RIGHT_OBJECT_TEXT, bucket.get(@key1)
+      # get key object
+    key = bucket.key(@key1, true)
+    assert_equal Rightscale::S3::Key, key.class
+    assert       key.exists?
+    assert_equal '123456', key.meta_headers['family']
+  end
+
+  def test_22_keys
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, false)
+      # create first key
+    key1 = Rightscale::S3::Key.create(bucket, @key1)
+    key1.refresh
+    assert key1.exists?
+    assert_equal '123456', key1.meta_headers['family']
+      # create second key
+    key2 = Rightscale::S3::Key.create(bucket, @key2)
+    assert !key2.refresh
+    assert !key2.exists?
+    assert_raise(Rightscale::AwsError) { key2.head }
+      # store key
+    key2.meta_headers = {'family'=>'111222333'}
+    assert key2.put(RIGHT_OBJECT_TEXT)
+      # make sure that the key exists
+    assert key2.refresh
+    assert key2.exists?
+    assert key2.head
+      # get its data
+    assert_equal RIGHT_OBJECT_TEXT, key2.get
+      # drop key
+    assert key2.delete
+    assert !key2.exists?
+  end
+  
+  def test_23_clear_delete
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, false)
+      # add another key
+    bucket.put(@key2, RIGHT_OBJECT_TEXT)
+      # delete 'folder'
+    assert_equal 1, bucket.delete_folder(@key1).size
+      # delete
+    assert_raise(Rightscale::AwsError) { bucket.delete }
+    bucket.delete(true)
+  end
+
+    # Grantees
+    
+  def test_30_create_bucket
+    bucket = @s.bucket(@bucket, true, 'public-read')
+    assert bucket
+  end
+  
+  def test_31_list_grantees
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, false)
+      # get grantees list
+    grantees = bucket.grantees
+      # check that the grantees count equal to 2 (root, AllUsers)
+    assert_equal 2, grantees.size
+  end
+  
+  def test_32_grant_revoke_drop
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, false)
+    grantees = bucket.grantees
+      # Take one of grantees
+    grantee = grantees[0]
+      # Add grant as String
+    assert grantee.grant('WRITE')
+      # Add grants as Array
+    assert grantee.grant(['READ_ACP', 'WRITE_ACP'])
+      # Check perms count
+    assert_equal 4, grantee.perms.size
+      # revoke 'WRITE_ACP'
+    assert grantee.revoke('WRITE_ACP')
+      # Check manual perm removal method
+    grantee.perms -= ['READ_ACP']
+    grantee.apply
+    assert_equal 2, grantee.perms.size
+      # check 'Drop' method
+    assert grantee.drop
+    assert_equal 1, bucket.grantees.size
+      # Delete bucket
+    bucket.delete(true)
+  end
+  
+  def test_33_key_grantees
+      # Create bucket
+    bucket = @s.bucket(@bucket, true)
+      # Create key
+    key = bucket.key(@key1)
+    assert key.put(RIGHT_OBJECT_TEXT, 'public-read')
+      # Get grantees list (must be == 2)
+    grantees = key.grantees
+    assert grantees
+    assert_equal 2, grantees.size
+      # Take one of grantees and give him 'Write' perms
+    grantee = grantees[0]
+    assert grantee.grant('WRITE')
+      # Drop grantee
+    assert grantee.drop
+      # Drop bucket
+    bucket.delete(true)
+  end
+
+  def test_34_bucket_create_put_with_perms
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, true)
+      # check that the bucket exists
+    assert @s.buckets.map{|b| b.name}.include?(@bucket)
+    assert bucket.keys.empty?
+      # put data (with canned ACL)
+    assert bucket.put(@key1, RIGHT_OBJECT_TEXT, {'family'=>'123456'}, "public-read")
+      # get data and compare
+    assert_equal RIGHT_OBJECT_TEXT, bucket.get(@key1)
+      # get key object
+    key = bucket.key(@key1, true)
+    assert_equal Rightscale::S3::Key, key.class
+    assert       key.exists?
+    assert_equal '123456', key.meta_headers['family']
+  end
+
+  def test_35_key_put_with_perms
+    bucket = Rightscale::S3::Bucket.create(@s, @bucket, false)
+      # create first key
+    key1 = Rightscale::S3::Key.create(bucket, @key1)
+    key1.refresh
+    assert key1.exists?
+    assert key1.put(RIGHT_OBJECT_TEXT, "public-read")
+      # get its data
+    assert_equal RIGHT_OBJECT_TEXT, key1.get
+      # drop key
+    assert key1.delete
+    assert !key1.exists?
+  end
+  
+
+end

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,226 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestSqs < Test::Unit::TestCase
+
+  GRANTEE_EMAIL_ADDRESS = 'fester@example.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
+  
+  #---------------------------
+  # 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
+    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 5 # 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'
+    assert_equal 222, @sqs.get_visibility_timeout(queue_url), 'Get_visibility_timeout must return to 222'
+  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) }
+    assert @sqs.force_clear_queue(queue_url), 'Force_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 = @s.queues
+      # create new queue
+    queue  = @s.queue("#{@queue_name}_20", true)
+      # check that it is created
+    assert queue.is_a?(Rightscale::Sqs::Queue)
+      # 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 = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_21", true)
+      # check that it is created
+    assert queue.is_a?(Rightscale::Sqs::Queue)
+    # Don't test this - just for cleanup purposes
+    queue.delete
+  end
+  
+  def test_22_queue_attributes
+    queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_22", 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 5
+      # 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
+    # Don't test this - just for cleanup purposes
+    queue.delete
+  end
+  
+  def test_23_grantees
+    queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_23", false)
+      # get a list of grantees
+    grantees = queue.grantees
+      # 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 = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_24", false)
+      # 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
+    # Don't test this - just for cleanup purposes
+    queue.clear
+    queue.delete
+  end
+  
+  def test_25_message_receive_pop_peek_delete
+    queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_25", 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
+    # Don't test this - just for cleanup purposes
+    queue.delete
+  end
+  
+  def test_26
+    queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_26", false)
+      # lock message 
+    queue.receive(100)
+      # clear queue
+    assert queue.clear 
+      # queue size is greater than zero
+    assert queue.size>0
+      # force clear queue
+    assert queue.clear(true)
+      # queue size must be equal to zero
+    assert queue.size==0
+      # push new message
+    queue.push('123456')
+    assert_raise(Rightscale::AwsError) { queue.delete }
+    assert queue.delete(true)
+  end
+
+end