SQS 0.1.0

data/test/all_tests.rb

@@ -0,0 +1,6 @@
+#!/usr/local/bin/ruby
+dir = File.expand_path( File.dirname( __FILE__ ) )
+
+Dir[File.join( File.join( File.dirname( __FILE__ ), 'unit' ), "*_test.rb" )].each { |f| require f }
+
+

data/test/unit/sqs_grant_test.rb

@@ -0,0 +1,67 @@
+#!/usr/local/bin/ruby
+dir = File.expand_path( File.dirname( __FILE__ ) )
+
+require 'test/unit'
+require "#{File.expand_path( File.dirname( __FILE__ ) )}/test_setup"
+
+class SQS_GrantTest < Test::Unit::TestCase
+  def setup
+    @queue_prefix = 'testGRANTEE'
+    @q = SQS.create_queue( @queue_prefix ) unless SQStest.skip_live_tests?
+  end
+  def teardown
+    SQS.list_queues( @queue_prefix ).each { |q| q.force_delete } unless SQStest.skip_live_tests?
+    print SQS.counter if SQStest.print_counter?
+  end
+
+  unless SQStest.skip_live_tests?
+
+
+  end
+
+
+  def test_initialize
+    g = SQS::Grant.new( :email => 'a@b.c', :id => '123456', :display_name => 'abc', :permission => 'zyx' )
+    assert_equal 'abc', g.display_name
+    assert_equal '123456', g.id
+    assert_equal 'a@b.c', g.email
+    assert_equal 'zyx', g.permission
+    
+    g = SQS::Grant.new( 'iloveemail' )
+    assert_equal 'iloveemail', g.email
+    assert g.id.nil?
+    assert g.display_name.nil?
+    assert g.permission.nil?
+  end
+  
+  def test_permission
+    g = SQS::Grant.new
+    assert_equal nil, g.permission
+
+    g.permission = :full
+    assert_equal 'FULLCONTROL', g.permission
+    g.permission = :receive
+    assert_equal 'RECEIVEMESSAGE', g.permission
+    g.permission = :send
+    assert_equal 'SENDMESSAGE', g.permission
+
+    g.permission = 'fullcontrol'
+    assert_equal 'FULLCONTROL', g.permission
+    g.permission = 'receivemessage'
+    assert_equal 'RECEIVEMESSAGE', g.permission
+    g.permission = 'sendmessage'
+    assert_equal 'SENDMESSAGE', g.permission
+
+    g.permission = 'full'
+    assert_equal 'FULLCONTROL', g.permission
+    g.permission = 'receive'
+    assert_equal 'RECEIVEMESSAGE', g.permission
+    g.permission = 'send'
+    assert_equal 'SENDMESSAGE', g.permission
+
+    g.permission = 'somethingelse'
+    assert_equal 'SOMETHINGELSE', g.permission
+  end
+end
+
+

data/test/unit/sqs_message_test.rb

@@ -0,0 +1,81 @@
+#!/usr/local/bin/ruby
+dir = File.expand_path( File::dirname( __FILE__ ) )
+
+require 'test/unit'
+require "#{File.expand_path( File.dirname( __FILE__ ) )}/test_setup"
+
+
+class SQS_MessageTest < Test::Unit::TestCase
+  def setup
+    @queue_prefix = 'testMSG'
+    @q = SQS.create_queue( @queue_prefix ) unless SQStest.skip_live_tests?
+  end
+  def teardown
+    SQS.list_queues( @queue_prefix ).each { |q| q.force_delete } unless SQStest.skip_live_tests?
+    print SQS.counter if SQStest.print_counter?
+  end
+
+  unless SQStest.skip_live_tests?
+    def test_delete
+      m = @q.send_message( 'This is something special' )
+      assert m.is_a?( SQS::Message )
+      assert_equal @q, m.queue
+      assert m.delete
+      sleep 2
+      assert_raises RuntimeError do
+        m.peek
+      end
+    end
+
+    def test_peek
+      m = @q.send_message( 'This is something else special' )
+      assert m.peek
+      assert_equal 'This is something else special', m.body
+    end
+
+    def test_body
+      m = SQS::Message.new()
+      m.body = 'bob'
+      assert_equal 'bob', m.body
+
+      m = @q.send_message( 'This is something else really special' )
+      assert_equal 'This is something else really special', m.body
+    end
+  end
+  
+  def test_new
+    assert_nothing_raised RuntimeError do
+      SQS::Message.new
+    end
+    assert_nothing_raised RuntimeError do
+      SQS::Message.new :a => 'b'
+    end
+    assert_nothing_raised RuntimeError do
+      SQS::Message.new :a => 'b', :body => ''
+    end
+
+    assert_nothing_raised do
+      SQS::Message.new :a => 'b', :body => 'This is a test'
+    end
+
+    assert_nothing_raised do
+      SQS::Message.new 'This is a test'
+    end
+
+    m1 = SQS::Message.new :a => 'b', :body => 'This is a test'
+    assert m1.is_a?( SQS::Message )
+
+    m2 = SQS::Message.new 'This is a test'
+    assert m2.is_a?( SQS::Message )
+  end
+
+  def test_id
+    m1 = SQS::Message.new 'Message 1'
+    m2 = SQS::Message.new :id => 'Message 2'
+    m3 = SQS::Message.new :id => 'Message 3', :something => 'else'
+    
+    assert_equal 'Message 1', m1.id
+    assert_equal 'Message 2', m2.id
+    assert_equal 'Message 3', m3.id
+  end
+end

data/test/unit/sqs_queue_test.rb

@@ -0,0 +1,312 @@
+#!/usr/local/bin/ruby
+dir = File.expand_path( File.dirname( __FILE__ ) )
+
+require 'test/unit'
+require "#{File.expand_path( File.dirname( __FILE__ ) )}/test_setup"
+
+
+
+
+class SQS_QueueTest < Test::Unit::TestCase
+  def self.other_aws_account
+    SQStest.other_aws_account
+  end
+  
+  def setup
+    @queue_prefix = 'testQUEUE'
+    @q = SQStest.skip_live_tests? ? SQS::Queue.new( 'http://nowhere.com/nothing' ) : SQS.create_queue( @queue_prefix )
+  end
+  def teardown
+    SQS.list_queues( @queue_prefix ).each { |q| q.force_delete } unless SQStest.skip_live_tests?
+    print SQS.counter if SQStest.print_counter?
+  end
+
+  unless SQStest.skip_live_tests?
+    def test_delete
+      assert @q.is_a?( SQS::Queue )
+      q2 = SQS.get_queue( @queue_prefix )
+      assert q2.is_a?( SQS::Queue )
+      assert_equal @q, q2
+
+      assert @q.delete
+
+      assert_raises RuntimeError do
+        q3 = SQS.get_queue( @queue_prefix )
+      end
+
+      assert_raises RuntimeError do
+        q2.delete
+      end
+    end
+
+    def test_exist
+      assert @q.exist?
+      assert @q.exists?
+      q2 = SQS.get_queue( @queue_prefix )
+      assert q2.exists?
+      assert q2.exist?
+
+      assert @q.delete
+      sleep 2
+      
+      assert !@q.exists?
+      assert !@q.exist?
+      assert !q2.exists?
+      assert !q2.exist?
+
+      q3 = SQS::Queue.new( 'http://borker.borked/borkee' )
+      assert !q3.exists?
+      assert !q3.exist?
+    end
+
+    def test_send_message
+      m = @q.send_message( 'This is so mething special' )
+      assert m.is_a?( SQS::Message )
+      assert_equal @q, m.queue
+    end
+
+    def test_receive_message
+      @q.send_message( 'I am receiving a message from beyond' )
+      sleep 2
+      m = @q.receive_message
+
+      assert m.is_a?( SQS::Message )
+      assert_equal @q, m.queue
+      assert_equal 'I am receiving a message from beyond', m.body
+    end
+    def test_receive_message_on_empty_queue
+      m = @q.receive_message
+
+      assert m.nil?
+    end
+
+    def test_receive_messages
+      (1..10).to_a.each do |i|
+        @q.send_message( "I am receiving message #{i}" )
+      end
+
+      sleep 10
+
+      messages = @q.receive_messages :count => 2
+      assert messages.size == 1 || messages.size == 2, "Weird array size: #{messages.size}"
+
+      assert messages.is_a?( Array )
+      messages.each do |m|
+        assert m.is_a?( SQS::Message )
+        assert_equal @q, m.queue
+        assert_match /^I am receiving message \d+$/, m.body
+      end
+
+      messages = @q.receive_messages :count => 12
+      assert messages.is_a?( Array )
+      assert messages.size > 0 && messages.size < 12, "Weird array size: #{messages.size}"
+    end
+
+    def test_force_delete
+      (1..10).to_a.each do |i|
+        @q.send_message( "I am receiving message ##{i}" )
+      end
+
+      sleep 5
+
+      assert_raises RuntimeError do
+        @q.delete
+      end
+
+      assert_nothing_raised do
+        @q.force_delete
+      end
+      
+      sleep 5
+
+      assert_raises RuntimeError do
+        q = SQS.get_queue( @queue_prefix )
+      end
+    end
+
+    class DUMMY
+      def self.email
+        SQS_QueueTest::other_aws_account[:email]
+      end
+    end
+    
+    def test_add_grant
+      assert_respond_to @q, :add_grant
+      assert_equal 1, @q.list_grants.size
+      assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => :full )
+      assert_equal 2, @q.list_grants.size
+      assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => :send )
+      assert_equal 3, @q.list_grants.size
+      assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => :receive )
+      assert_equal 4, @q.list_grants.size
+      
+      assert_nothing_raised do
+        #DUMMY responds to :email (is not an address itself)
+        @q.add_grant( :email => DUMMY, :permission => :receive )
+      end
+      assert_equal 4, @q.list_grants.size
+      
+      
+      assert_raises RuntimeError do
+        assert @q.add_grant( :email => 'invalid@email.address', :permission => :receive )
+      end
+    end
+    
+    def test_remove_grant
+      assert_respond_to @q, :remove_grant
+
+
+      [:full, :send, :receive].each do |perm|
+        assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => perm )
+      end
+      @q.list_grants.each do |g|
+        unless g.display_name == SQStest.my_aws_account[:display_name]
+          assert @q.remove_grant( :email => SQStest.other_aws_account[:email], :permission => g.permission )
+        end
+      end
+
+      [:full, :send, :receive].each do |perm|
+        assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => perm )
+      end
+      @q.list_grants.each do |g|
+        unless g.display_name == SQStest.my_aws_account[:display_name]
+          #DUMMY responds to :email (is not an address itself)
+          assert @q.remove_grant( :email => DUMMY, :permission => g.permission )
+        end
+      end
+
+      [:full, :send, :receive].each do |perm|
+        assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => perm )
+      end
+      @q.list_grants.each do |g|
+        unless g.display_name == SQStest.my_aws_account[:display_name]
+          assert @q.remove_grant( :id => g.id, :permission => g.permission )
+        end
+      end
+
+      [:full, :send, :receive].each do |perm|
+        assert @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => perm )
+      end
+      @q.list_grants.each do |g|
+        unless g.display_name == SQStest.my_aws_account[:display_name]
+          # the grant responds to :id (is not an id itself)
+          assert @q.remove_grant( :id => g, :permission => g.permission )
+        end
+      end
+    end
+    
+    def test_list_grants
+      assert_respond_to @q, :list_grants
+      grant = @q.list_grants
+      assert_equal 1, grant.size
+      grant = grant.first
+      assert_equal SQS.permissions[:full], grant.permission  
+      assert_equal nil, grant.email
+      assert_equal SQStest.my_aws_account[:display_name], grant.display_name
+
+      @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => :receive)
+      grants = @q.list_grants
+      assert_equal 2, grants.size
+      grants.each do |grant|
+        case grant.display_name
+        when SQStest.my_aws_account[:display_name]
+          assert_equal SQS.permissions[:full], grant.permission
+        when SQStest.other_aws_account[:display_name]
+          assert_equal SQS.permissions[:receive], grant.permission
+        else
+          assert false, "'#{grant.display_name}' is an unknown email address"
+        end
+      end
+    end
+
+    def test_each_grant
+      assert_respond_to @q, :each_grant
+      @q.add_grant( :email => SQStest.other_aws_account[:email], :permission => :receive)
+      @q.each_grant do |g|
+        assert_kind_of SQS::Grant, g
+      end
+    end
+
+    def test_get_visiblity_timeout
+      assert_respond_to @q, :visibility_timeout
+      assert_equal SQS::Queue.default_visibility_timeout, @q.visibility_timeout
+      assert_kind_of Integer, @q.visibility_timeout
+
+      q = SQS.create_queue :name => "#{@queue_prefix}getvisi", :DefaultVisibilityTimeout => 12345
+      assert_equal 12345, q.visibility_timeout
+      assert_kind_of Integer, q.visibility_timeout
+    end
+
+    def test_set_visiblity_timeout
+      assert_respond_to @q, :visibility_timeout=
+      assert( @q.visibility_timeout = 456 )
+      assert_equal 456, @q.visibility_timeout
+
+      assert_nothing_raised RuntimeError do
+        @q.visibility_timeout = nil
+        @q.visibility_timeout = 0
+      end
+
+      assert_raises RuntimeError do
+        @q.visibility_timeout = -1
+      end
+      assert_raises RuntimeError do
+        @q.visibility_timeout = 86401
+      end
+
+      assert_raises NoMethodError do
+        @q.visibility_timeout = Object
+      end
+    end
+  end
+
+  def test_default_visibility_timeout
+    assert_respond_to SQS::Queue, :default_visibility_timeout
+    assert_equal 30, SQS::Queue.default_visibility_timeout
+  end
+  
+  
+
+  def test_new
+    assert_nothing_raised do
+      q1 = SQS::Queue.new :url => 'http://www.slappy.com/', :a => 'a', 'b' => :b
+    end
+    q1 = SQS::Queue.new :url => 'http://www.slappy.com/', :a => 'a', 'b' => :b
+    assert_equal 'http://www.slappy.com/', q1.url
+
+    q2 = SQS::Queue.new 'http://www.slappy.com/'
+    assert_equal 'http://www.slappy.com/', q2.url
+
+    assert_raises NameError do
+      q.a
+    end
+    assert_raises NameError do
+      q.b
+    end
+
+    assert_raises RuntimeError do
+      q = SQS::Queue.new :url => 'http://www.slappy.com'
+      q = SQS::Queue.new :url => 'LKJSDFLKJ@#$)()'
+    end
+  end
+
+  def test_url
+    q = SQS::Queue.new :url => 'http://queue.amazonaws.com/A3ZVJ8HH1F466/test96570'
+    assert_equal 'http://queue.amazonaws.com/A3ZVJ8HH1F466/test96570', q.url
+  end
+
+  def test_name
+    q = SQS::Queue.new :url => 'http://queue.amazonaws.com/A3ZVJ8HH1F466/test96570'
+    assert_equal 'test96570', q.name
+  end
+
+  def test_equality_check
+    q1 = SQS::Queue.new :url => 'http://queue.amazonaws.com/A3ZVJ8HH1F466/test96570'
+    q2 = SQS::Queue.new :url => 'http://queue.amazonaws.com/A3ZVJ8HH1F466/test96570'
+    q3 = SQS::Queue.new :url => 'http://queue.amazonaws.com/A3ZVJ8HH1F466/est96570'
+
+    assert q1 == q2
+    assert q1 != q3
+    assert q2 != q3
+  end
+end