aboisvert_aws 3.0.0

data/test/sdb/test_batch_put_attributes.rb

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestSdb < Test::Unit::TestCase
+
+  def setup
+    STDOUT.sync  = true
+    @domain = 'right_sdb_awesome_test_domain'
+    @attributes = {
+      'a' => { 'foo' => '123' },
+      'b' => { 'bar' => '456' }
+    }
+    # Interface instance
+    @sdb = Rightscale::SdbInterface.new
+    @sdb.delete_domain(@domain)
+    wait(SDB_DELAY, "after removing domain")
+    @sdb.create_domain(@domain)
+    wait(SDB_DELAY, "after recreating domain")
+  end
+
+  SDB_DELAY = 2
+  
+  def wait(delay, msg='')
+    print "waiting #{delay} seconds #{msg}"
+    while delay>0 do
+      delay -= 1
+      print '.'
+      sleep 1
+    end
+    puts
+  end
+
+  def test_batch_put_attributes
+    @sdb.batch_put_attributes(@domain, @attributes)
+    wait(SDB_DELAY, "after putting attributes")
+    a = @sdb.get_attributes(@domain, 'a')[:attributes]
+    b = @sdb.get_attributes(@domain, 'b')[:attributes]
+    assert_equal( {'foo' => ['123']}, a)
+    assert_equal( {'bar' => ['456']}, b)
+    
+    # Replace = false
+    @sdb.batch_put_attributes(@domain, { 'a' => {'foo' => ['789']}})
+    wait(SDB_DELAY, "after putting attributes") 
+    a = @sdb.get_attributes(@domain, 'a')[:attributes]
+    assert_equal ['123', '789'], a['foo'].sort
+
+    # Replace = true
+    @sdb.batch_put_attributes(@domain, {'b' => {'bar' => ['789']}}, true)
+    wait(SDB_DELAY, "after putting attributes") 
+    b = @sdb.get_attributes(@domain, 'b')[:attributes]
+    assert_equal ['789'], b['bar'].sort
+
+  end
+end

data/test/sdb/test_helper.rb

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

data/test/sdb/test_right_sdb.rb

@@ -0,0 +1,253 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestSdb < Test::Unit::TestCase
+
+  def setup
+    STDOUT.sync  = true
+    @domain = 'right_sdb_awesome_test_domain'
+    @item   = 'toys'
+    @attr   = { 'Jon' => %w{beer car} }
+    # Interface instance
+    @sdb    = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+  end
+
+  SDB_DELAY = 7
+  
+  def wait(delay, msg='')
+    print "waiting #{delay} seconds #{msg}"
+    while delay>0 do
+      delay -= 1
+      print '.'
+      sleep 1
+    end
+    puts
+  end
+
+  #---------------------------
+  # Rightscale::SdbInterface
+  #---------------------------
+
+  def test_00_delete_domain
+    # delete the domain to reset all the things
+    assert @sdb.delete_domain(@domain), 'delete_domain fail'
+    wait SDB_DELAY, 'after domain deletion'
+  end
+  
+  def test_01_create_domain
+    # check that domain does not exist
+    assert !@sdb.list_domains[:domains].include?(@domain)
+    # create domain
+    assert @sdb.create_domain(@domain), 'create_domain fail'
+    wait SDB_DELAY, 'after domain creation'
+    # check that we have received new domain from Amazin
+    assert @sdb.list_domains[:domains].include?(@domain)
+  end
+
+  def test_02_put_attributes
+    # put attributes
+    assert @sdb.put_attributes(@domain, @item, @attr)
+    wait SDB_DELAY, 'after putting attributes'
+  end
+  
+  def test_03_get_attributes
+    # get attributes
+    values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Jon']).sort
+    # compare to original list
+    assert_equal values, @attr['Jon'].sort
+  end
+
+  def test_04_add_attributes
+    # add new attribute
+    new_value = 'girls'
+    @sdb.put_attributes @domain, @item, {'Jon' => new_value}
+    wait SDB_DELAY, 'after putting attributes'
+    # get attributes ('girls' must be added to already existent attributes)
+    values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Jon']).sort
+    assert_equal values, (@attr['Jon'] << new_value).sort
+  end
+  
+  def test_05_replace_attributes
+    # replace attributes
+    @sdb.put_attributes @domain, @item, {'Jon' => 'pub'}, :replace
+    wait SDB_DELAY, 'after replacing attributes'
+    # get attributes (all must be removed except of 'pub')
+    values = @sdb.get_attributes(@domain, @item)[:attributes]['Jon']
+    assert_equal values, ['pub']
+  end
+  
+  def test_06_delete_attribute
+    # add value 'girls' and 'vodka' to 'Jon'
+    @sdb.put_attributes @domain, @item, {'Jon' => ['girls','vodka']}
+    wait SDB_DELAY, 'after adding attributes'
+    # get attributes ('girls' and 'vodka' must be added 'pub')
+    values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Jon']).sort
+    assert_equal values, ['girls', 'pub', 'vodka']
+    # delete a single value 'girls' from attribute 'Jon'
+    @sdb.delete_attributes @domain, @item, 'Jon' => ['girls']
+    wait SDB_DELAY, 'after the deletion of attribute'
+    # get attributes ('girls' must be removed)
+    values = @sdb.get_attributes(@domain, @item)[:attributes]['Jon']
+    assert_equal values, ['pub', 'vodka']
+    # delete all values from attribute 'Jon'
+    @sdb.delete_attributes @domain, @item, ['Jon']
+    wait SDB_DELAY, 'after the deletion of attributes'
+    # get attributes (values must be empty)
+    values = @sdb.get_attributes(@domain, @item)[:attributes]['Jon']
+    assert_equal values, nil
+  end
+
+  def test_07_delete_item
+    @sdb.put_attributes @domain, @item, {'Volodya' => ['girls','vodka']}
+    wait SDB_DELAY, 'after adding attributes'
+    # get attributes ('girls' and 'vodka' must be there)
+    values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Volodya']).sort
+    assert_equal values, ['girls', 'vodka']
+    # delete an item
+    @sdb.delete_attributes @domain, @item
+    wait SDB_DELAY, 'after deleting attributes'
+    # get attributes (values must be empty)
+    values = @sdb.get_attributes(@domain, @item)[:attributes]['Volodya']
+    assert_equal values, nil
+  end  
+  
+  def test_08_query
+    # add some values for query
+    @sdb.put_attributes @domain, @item, {'Jon' => ['girls','vodka']}
+    wait SDB_DELAY, 'after adding attributes'
+    items = @sdb.query(@domain, ['[?=?]', 'Jon','vodka'])[:items]
+    assert_equal items.size, 1
+    assert_equal items.first, @item
+  end
+  
+  def test_09_signature_version_0 
+    sdb    = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '0') 
+    item   = 'toys' 
+    # TODO: need to change the below test.  I think Juergen's intention was to include some umlauts in the values
+    # put attributes 
+    # mhhh... Not sure how to translate this: hölzchehn klötzchen grÃŒnspan buße... Lets assume this is:
+    attributes = { 'Jurgen' => %w{kitten puppy chickabiddy piglet} } 
+    assert sdb.put_attributes(@domain, item, attributes) 
+    wait SDB_DELAY, 'after putting attributes' 
+    # get attributes 
+    values = Array(sdb.get_attributes(@domain, item)[:attributes]['Jurgen']).sort
+    # compare to original list 
+    assert_equal values, attributes['Jurgen'].sort 
+    # check that the request has correct signature version
+    assert sdb.last_request.path.include?('SignatureVersion=0')
+  end 
+
+  def test_10_signature_version_1
+    sdb = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '1')
+    domains = nil
+    assert_nothing_thrown "Failed to use signature V1" do
+      domains = sdb.list_domains
+    end
+    assert domains
+  end
+
+  def test_11_signature_version_1
+    sdb = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '2')
+    domains = nil
+    assert_nothing_thrown "Failed to use signature V2" do
+      domains = sdb.list_domains
+    end
+    assert domains
+  end
+
+  def test_12_array_of_attrs
+    item = 'multiples'
+    assert_nothing_thrown "Failed to put multiple attrs" do
+      @sdb.put_attributes(@domain, item, {:one=>1, :two=>2, :three=>3})
+    end
+  end
+  
+  def test_13_zero_len_attrs
+    item = 'zeroes'
+    assert_nothing_thrown "Failed to put zero-length attributes" do
+      @sdb.put_attributes(@domain, item, {:one=>"", :two=>"", :three=>""})
+    end
+  end
+  
+  def test_14_nil_attrs
+    item = 'nils'
+    res = nil
+    assert_nothing_thrown do
+      @sdb.put_attributes(@domain, item, {:one=>nil, :two=>nil, :three=>'chunder'})
+    end
+    wait SDB_DELAY, 'after putting attributes'
+    assert_nothing_thrown do
+      res = @sdb.get_attributes(@domain, item)
+    end
+    assert_nil(res[:attributes]['one'][0])
+    assert_nil(res[:attributes]['two'][0])
+    assert_not_nil(res[:attributes]['three'][0])
+  end
+  
+  def test_15_url_escape
+    item = 'urlescapes'
+    content = {:a=>"one & two & three",
+               :b=>"one ? two / three"}
+    @sdb.put_attributes(@domain, item, content)
+    wait SDB_DELAY, 'after putting attributes'
+
+    res = @sdb.get_attributes(@domain, item)
+    assert_equal(content[:a], res[:attributes]['a'][0])
+    assert_equal(content[:b], res[:attributes]['b'][0])
+  end
+  
+  def test_16_put_attrs_by_post
+    item = 'reqgirth'
+    i = 0
+    sa = ""
+    while(i < 64) do
+      sa += "aaaaaaaa"
+      i += 1
+    end
+    @sdb.put_attributes(@domain, item, {:a => sa, :b => sa, :c => sa, :d => sa, :e => sa})
+    wait SDB_DELAY, 'after putting attributes'
+  end
+
+  def test_20_query_with_atributes
+    response = @sdb.query_with_attributes(@domain)
+    # convers response to a hash representation
+    items = {};
+    response[:items].each{ |item| items.merge!(item) }
+    # check we have receied all 5 items each full of attributes
+    assert_equal 6, items.keys.size
+    assert items['toys'].size > 0
+    assert items['nils'].size > 0
+    assert items['urlescapes'].size > 0
+    assert items['multiples'].size > 0
+    assert items['reqgirth'].size > 0
+    assert items['zeroes'].size > 0
+    # fetch only Jon's attributes from all items
+    response = @sdb.query_with_attributes(@domain,['Jon'])
+    items = {};
+    response[:items].each{ |item| items.merge!(item) }
+    # check we have receied all 5 items
+    # check we have receied all 5 items, but only 'toys' has attributes
+    puts items.inspect
+    assert_equal 2, items['toys']['Jon'].size
+    assert_equal 0, items['nils'].size
+    assert_equal 0, items['urlescapes'].size
+    assert_equal 0, items['multiples'].size
+    assert_equal 0, items['reqgirth'].size
+    assert_equal 0, items['zeroes'].size
+    # kust Jurgen's attriburs
+    response = @sdb.query_with_attributes(@domain,['Jurgen'], "['Jurgen'='piglet']")
+    items = {};
+    response[:items].each{ |item| items.merge!(item) }
+    # check we have receied an only item
+    assert_equal 1, items.keys.size
+    assert_equal ["chickabiddy", "kitten", "piglet", "puppy"], items['toys']['Jurgen'].sort
+  end
+
+  # Keep this test last, because it deletes the domain...
+  def test_21_delete_domain
+    assert @sdb.delete_domain(@domain), 'delete_domain fail'
+    wait SDB_DELAY, 'after domain deletion'
+    # check that domain does not exist
+    assert !@sdb.list_domains[:domains].include?(@domain)
+  end
+  
+end

data/test/sns/test_helper.rb

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

data/test/sns/test_right_sns.rb

@@ -0,0 +1,153 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class TestSns < Test::Unit::TestCase
+
+  # You can change these things to whatever you like
+  @@subscriber_email    = 'foo@bar.baz'
+  @@topic_name          = 'RightTestTopic'
+  @@topic_display_name  = 'right_aws test notification topic'
+  @@queue_name          = "sns_subscribe_queue_#{Time.now.utc.to_i}"
+
+  # These are placeholders for values that get set, and consumed during the course of testing.
+  @@topic_arn           = ''
+  @@subscription_arn    = ''
+  @@queue_url           = ''
+  @@queue_arn           = ''
+
+  @@policy_template     = <<-EOF
+{
+  "Id": "Policy1300753700208",
+  "Statement": [
+    {
+      "Sid": "Stmt1300753696680",
+      "Action": [
+        "SNS:Publish",
+        "SNS:RemovePermission",
+        "SNS:SetTopicAttributes",
+        "SNS:DeleteTopic",
+        "SNS:ListSubscriptionsByTopic",
+        "SNS:GetTopicAttributes",
+        "SNS:Receive",
+        "SNS:AddPermission",
+        "SNS:Subscribe"
+      ],
+      "Effect": "Allow",
+      "Resource": "@@topic_arn@@",
+      "Principal": {
+        "AWS": [
+          "*"
+        ]
+      }
+    }
+  ]
+}
+    EOF
+
+  def policy_text
+    @@policy_template.gsub('@@topic_arn@@', @@topic_arn)
+  end
+
+  def setup
+    @sns = Rightscale::SnsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+    @sqs = Rightscale::SqsGen2Interface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
+  end
+
+  def test_01_create_topic
+    response = @sns.create_topic(@@topic_name)
+    assert_not_nil(response)
+    @@topic_arn = response
+  end
+
+  def test_02_set_topic_attributes
+    response = @sns.set_topic_attribute(@@topic_arn, 'Policy', policy_text())
+    assert_not_nil(response)
+
+    response = @sns.set_topic_attribute(@@topic_arn, 'DisplayName', @@topic_display_name)
+    assert_not_nil(response)
+
+    assert_raise ArgumentError do
+      @sns.set_topic_attribute(@@topic_arn, 'Foo', 'val')
+    end
+  end
+
+  def test_03_get_topic_attributes
+    response = @sns.get_topic_attributes(@@topic_arn)
+    assert_not_nil(response)
+    assert(response['DisplayName'] == @@topic_display_name)
+    assert(response['Policy'] =~ /Policy1300753700208/)
+  end
+
+  def test_04_list_topics
+    sleep(1)
+    assert(@sns.list_topics.collect{|topic| topic[:arn] }.include?(@@topic_arn))
+  end
+
+  def test_05_subscribe_email
+    response = @sns.subscribe(@@topic_arn, 'email', @@subscriber_email)
+    assert_not_nil(response)
+    @@subscription_arn = response
+  end
+
+  def test_06_list_subscriptions
+    sleep(1)
+
+    response = @sns.list_subscriptions()
+    assert_not_nil(response)
+    assert(response.count == 1)
+    assert(response[0][:endpoint] == @@subscriber_email)
+    assert(response[0][:protocol] == 'email')
+    assert(response[0][:subscription_arn] == 'PendingConfirmation')
+    assert(response[0][:topic_arn] == @@topic_arn)
+  end
+
+  def test_07_list_subscriptions_by_topic
+    response = @sns.list_subscriptions(@@topic_arn)
+    assert_not_nil(response)
+    assert(response.count == 1)
+    assert(response[0][:endpoint] == @@subscriber_email)
+    assert(response[0][:protocol] == 'email')
+    assert(response[0][:subscription_arn] == 'PendingConfirmation')
+    assert(response[0][:topic_arn] == @@topic_arn)
+  end
+
+  def test_08_unsubscribe
+    @@queue_url = @sqs.create_queue(@@queue_name)
+    @@queue_arn = "arn:aws:sqs:us-east-1:#{TestCredentials.account_number.gsub('-','')}:#{@@queue_name}"
+    sub_response = @sns.subscribe(@@topic_arn, 'sqs', @@queue_arn)
+    assert_not_nil(sub_response)
+    unsub_response = @sns.unsubscribe(sub_response)
+    @sqs.delete_queue(@@queue_url)
+  end
+
+  def test_09_publish
+    response = @sns.publish(@@topic_arn, 'Message to publish', 'Message Subject')
+    assert_not_nil(response)
+  end
+
+  def test_10_add_and_remove_permission
+    acct_num = TestCredentials.account_number.gsub('-','')
+
+    add_response = @sns.add_permission(@@topic_arn, 'PermissionLbl', [
+      {:aws_account_id => acct_num, :action => "GetTopicAttributes"},
+      {:aws_account_id => acct_num, :action => "Publish"}
+    ])
+    assert_not_nil(add_response)
+
+    remove_response = @sns.remove_permission(@@topic_arn, 'PermissionLbl')
+    assert_not_nil(remove_response)
+  end
+
+# TODO: Cannot easily test confirming subscription because it's only valid for http(s) and email subscriptions.
+# Since we don't want to setup an email box or HTTP server to recive the token, we can't really simulate this
+#  def test_10_confirm_subscription
+#    response = @sns.confirm_subscription(@@topic_arn, 'SomeToken')
+#    assert_not_null(response)
+#  end
+
+  def test_30_delete_topic
+    response = @sns.delete_topic(@@topic_arn)
+    assert_not_nil(response)
+    sleep(1)
+    assert(!@sns.list_topics.collect{|topic| topic[:arn] }.include?(@@topic_arn))
+  end
+end