aws-s3 0.1.0

data/test/authentication_test.rb

@@ -0,0 +1,96 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class HeaderAuthenticationTest < Test::Unit::TestCase  
+  def test_encoded_canonical
+    signature = Authentication::Signature.new(request, key_id, secret)
+    assert_equal AmazonDocExampleData::Example1.canonical_string, signature.send(:canonical_string)
+    assert_equal AmazonDocExampleData::Example1.signature, signature.send(:encoded_canonical)
+  end
+  
+  def test_authorization_header
+    header = Authentication::Header.new(request, key_id, secret)
+    assert_equal AmazonDocExampleData::Example1.canonical_string, header.send(:canonical_string)
+    assert_equal AmazonDocExampleData::Example1.authorization_header, header
+  end
+  
+  private
+    def request; AmazonDocExampleData::Example1.request end
+    def key_id ; AmazonDocExampleData::Example1.access_key_id end
+    def secret ; AmazonDocExampleData::Example1.secret_access_key end
+end
+
+class QueryStringAuthenticationTest < Test::Unit::TestCase
+  def test_query_string
+    query_string = Authentication::QueryString.new(request, key_id, secret, :expires_in => 60)
+    assert_equal AmazonDocExampleData::Example3.canonical_string, query_string.send(:canonical_string)
+    assert_equal AmazonDocExampleData::Example3.query_string, query_string
+  end
+  
+  def test_query_string_with_explicit_expiry
+    query_string = Authentication::QueryString.new(request, key_id, secret, :expires => expires)
+    assert_equal expires, query_string.send(:canonical_string).instance_variable_get(:@options)[:expires]
+    assert_equal AmazonDocExampleData::Example3.query_string, query_string
+  end    
+  
+  private
+    def request; AmazonDocExampleData::Example3.request end
+    def key_id ; AmazonDocExampleData::Example3.access_key_id end
+    def secret ; AmazonDocExampleData::Example3.secret_access_key end
+    def expires; AmazonDocExampleData::Example3.expires end
+end
+
+class CanonicalStringTest < Test::Unit::TestCase  
+  def setup
+    @request = Net::HTTP::Post.new('/test')
+    @canonical_string = Authentication::CanonicalString.new(@request)
+  end
+  
+  def test_path_does_not_include_query_string
+    request = Net::HTTP::Get.new('/test/query/string?foo=bar&baz=quux')
+    assert_equal '/test/query/string', Authentication::CanonicalString.new(request).send(:path)
+    
+    # Make sure things still work when there is *no* query string
+    request = Net::HTTP::Get.new('/')
+    assert_equal '/', Authentication::CanonicalString.new(request).send(:path)
+    request = Net::HTTP::Get.new('/foo/bar')
+    assert_equal '/foo/bar', Authentication::CanonicalString.new(request).send(:path)
+  end
+  
+  def test_path_includes_significant_query_strings
+    significant_query_strings = [
+      ['/test/query/string?acl',             '/test/query/string?acl'],
+      ['/test/query/string?acl&foo=bar',     '/test/query/string?acl'],
+      ['/test/query/string?foo=bar&acl',     '/test/query/string?acl'],
+      ['/test/query/string?acl=foo',         '/test/query/string?acl'],
+      ['/test/query/string?torrent=foo',     '/test/query/string?torrent'],
+      ['/test/query/string?logging=foo',     '/test/query/string?logging'],
+      ['/test/query/string?bar=baz&acl=foo', '/test/query/string?acl']
+    ]
+    
+    significant_query_strings.each do |uncleaned_path, expected_cleaned_path|
+      assert_equal expected_cleaned_path, Authentication::CanonicalString.new(Net::HTTP::Get.new(uncleaned_path)).send(:path)
+    end
+  end
+  
+  def test_default_headers_set
+    Authentication::CanonicalString.default_headers.each do |header|
+      assert @canonical_string.headers.include?(header)
+    end
+  end
+  
+  def test_interesting_headers_are_copied_over
+    an_interesting_header = 'content-md5'
+    string_without_interesting_header = Authentication::CanonicalString.new(@request)
+    assert string_without_interesting_header.headers[an_interesting_header].empty?
+    
+    # Add an interesting header
+    @request[an_interesting_header] = 'foo'
+    string_with_interesting_header = Authentication::CanonicalString.new(@request)
+    assert_equal 'foo', string_with_interesting_header.headers[an_interesting_header]
+  end
+  
+  def test_canonical_string
+    request = AmazonDocExampleData::Example1.request
+    assert_equal AmazonDocExampleData::Example1.canonical_string, Authentication::CanonicalString.new(request)
+  end
+end

data/test/base_test.rb

@@ -0,0 +1,143 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class BaseTest < Test::Unit::TestCase  
+  def test_connection_established
+    assert_raises(NoConnectionEstablished) do
+      Base.connection
+    end
+    
+    Base.establish_connection!(:access_key_id => '123', :secret_access_key => 'abc')
+    assert_kind_of Connection, Base.connection
+    
+    instance = Base.new
+    assert_equal instance.send(:connection), Base.connection
+    assert_equal instance.send(:http), Base.connection.http
+  end
+  
+  def test_respond_with
+    assert_equal Base::Response, Base.send(:response_class)
+    Base.send(:respond_with, Bucket::Response) do
+      assert_equal Bucket::Response, Base.send(:response_class)
+    end
+    assert_equal Base::Response, Base.send(:response_class)
+  end
+  
+  def test_request_tries_again_when_encountering_an_internal_error
+    Bucket.in_test_mode do
+      Bucket.request_returns [
+        # First request is an internal error
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Second request is a success
+        {:body => Fixtures::Buckets.empty_bucket,  :code => 200}
+      ]
+      bucket = nil # Block scope hack
+      assert_nothing_raised do
+        bucket = Bucket.find('marcel')
+      end
+      # Don't call objects 'cause we don't want to make another request
+      assert bucket.object_cache.empty?
+    end    
+  end
+  
+  def test_request_tries_up_to_three_times
+    Bucket.in_test_mode do
+      Bucket.request_returns [
+        # First request is an internal error
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Second request is also an internal error
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Ditto third
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Fourth works
+        {:body => Fixtures::Buckets.empty_bucket,  :code => 200}      
+      ]
+      bucket = nil # Block scope hack
+      assert_nothing_raised do
+        bucket = Bucket.find('marcel')
+      end
+      # Don't call objects 'cause we don't want to make another request
+      assert bucket.object_cache.empty?
+    end
+  end
+  
+  def test_request_tries_again_three_times_and_gives_up
+    Bucket.in_test_mode do
+      Bucket.request_returns [
+        # First request is an internal error
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Second request is also an internal error
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Ditto third
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+        # Ditto fourth
+        {:body => Fixtures::Errors.internal_error, :code => 500, :error => true},
+      ]
+      assert_raises(InternalError) do
+        Bucket.find('marcel')
+      end
+    end
+  end
+end
+
+class MultiConnectionsTest < Test::Unit::TestCase
+  class ClassToTestSettingCurrentBucket < Base
+    set_current_bucket_to 'foo'
+  end
+
+  def setup
+    Base.send(:class_variable_get, :@@connections).clear
+  end
+  alias_method :teardown, :setup
+  
+  def test_default_connection_options_are_used_for_subsequent_connections
+    assert !Base.connected?
+    
+    assert_raises(MissingAccessKey) do
+      Base.establish_connection!
+    end
+    
+    assert !Base.connected?
+    
+    assert_raises(NoConnectionEstablished) do
+      Base.connection
+    end
+    
+    assert_nothing_raised do
+      Base.establish_connection!(:access_key_id => '123', :secret_access_key => 'abc')
+    end
+    
+    assert Base.connected?
+    
+    assert_nothing_raised do
+      Base.connection
+    end
+    
+    # All subclasses are currently using the default connection
+    assert Base.connection == Bucket.connection
+    
+    # No need to pass in the required options. The default connection will supply them
+    assert_nothing_raised do
+      Bucket.establish_connection!(:server => 'foo.s3.amazonaws.com')
+    end
+    
+    assert Base.connection != Bucket.connection
+    assert_equal '123', Bucket.connection.access_key_id
+    assert_equal 'foo', Bucket.connection.subdomain
+  end
+  
+  def test_current_bucket
+    Base.establish_connection!(:access_key_id => '123', :secret_access_key => 'abc') 
+    assert_raises(CurrentBucketNotSpecified) do
+      Base.current_bucket
+    end
+    
+    S3Object.establish_connection!(:server => 'foo-bucket.s3.amazonaws.com')
+    assert_nothing_raised do
+      assert_equal 'foo-bucket', S3Object.current_bucket
+    end
+  end
+  
+  def test_setting_the_current_bucket
+    assert_equal 'foo', ClassToTestSettingCurrentBucket.current_bucket
+  end
+end

data/test/bucket_test.rb

@@ -0,0 +1,48 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class BucketTest < Test::Unit::TestCase  
+  def test_bucket_name_validation
+    valid_names   = %w(123 joe step-one step_two step3 step_4 step-5)
+    invalid_names = ['12', 'jo', 'kevin spacey', 'larry@wall', '', 'a' * 256]
+    validate_name = Proc.new {|name| Bucket.send(:validate_name!, name)}
+    valid_names.each do |valid_name|
+      assert_nothing_raised { validate_name[valid_name] }
+    end
+    
+    invalid_names.each do |invalid_name|
+      assert_raises(InvalidBucketName) { validate_name[invalid_name] }
+    end
+  end
+  
+  def test_empty_bucket
+    Bucket.request_always_returns :body => Fixtures::Buckets.empty_bucket, :code => 200 do
+      bucket = Bucket.find('marcel_molina')
+      assert bucket.empty?
+    end
+  end
+  
+  def test_bucket_with_one_file
+    Bucket.request_always_returns :body => Fixtures::Buckets.bucket_with_one_key, :code => 200 do
+      bucket = Bucket.find('marcel_molina')
+      assert !bucket.empty?
+      assert_equal 1, bucket.size
+      assert_equal %w(tongue_overload.jpg), bucket.objects.map {|object| object.key}
+      assert bucket['tongue_overload.jpg']
+    end
+  end
+  
+  def test_bucket_with_more_than_one_file
+    Bucket.request_always_returns :body => Fixtures::Buckets.bucket_with_more_than_one_key, :code => 200 do
+      bucket = Bucket.find('marcel_molina')
+      assert !bucket.empty?
+      assert_equal 2, bucket.size
+      assert_equal %w(beluga_baby.jpg tongue_overload.jpg), bucket.objects.map {|object| object.key}.sort
+      assert bucket['tongue_overload.jpg']
+    end
+  end
+  
+  def test_bucket_path
+    assert_equal '/bucket_name?max-keys=2', Bucket.send(:path, 'bucket_name', :max_keys => 2)
+    assert_equal '/bucket_name', Bucket.send(:path, 'bucket_name', {})    
+  end
+end

data/test/connection_test.rb

@@ -0,0 +1,120 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class ConnectionTest < Test::Unit::TestCase
+  def setup
+    @keys = {:access_key_id => '123', :secret_access_key => 'abc'}
+  end
+  
+  def test_creating_a_connection
+    connection = Connection.new(@keys)
+    assert_kind_of Net::HTTP, connection.http
+  end
+  
+  def test_use_ssl_option_is_set_in_connection
+    connection = Connection.new(@keys.merge(:use_ssl => true))
+    assert connection.http.use_ssl?
+  end
+  
+  def test_setting_port_to_443_implies_use_ssl
+    connection = Connection.new(@keys.merge(:port => 443))
+    assert connection.http.use_ssl?
+  end
+  
+  def test_protocol
+    connection = Connection.new(@keys)
+    assert_equal 'http://', connection.protocol
+    connection = Connection.new(@keys.merge(:use_ssl => true))
+    assert_equal 'https://', connection.protocol
+  end
+  
+  def test_server_and_port_are_passed_onto_connection
+    connection = Connection.new(@keys)
+    options    = connection.instance_variable_get('@options')
+    assert_equal connection.http.address, options[:server]
+    assert_equal connection.http.port, options[:port]
+  end
+  
+  def test_not_including_required_access_keys_raises
+    assert_raises(MissingAccessKey) do
+      Connection.new
+    end
+    
+    assert_raises(MissingAccessKey) do
+      Connection.new(:access_key_id => '123')
+    end
+    
+    assert_nothing_raised do
+      Connection.new(@keys)
+    end
+  end
+  
+  def test_access_keys_extracted
+    connection = Connection.new(@keys)
+    assert_equal '123', connection.access_key_id
+    assert_equal 'abc', connection.secret_access_key
+  end
+  
+  def test_request_method_class_lookup
+    c = Connection.new(@keys)
+    expectations = {
+     :get  => Net::HTTP::Get, :post   => Net::HTTP::Post,
+     :put  => Net::HTTP::Put, :delete => Net::HTTP::Delete,
+     :head => Net::HTTP::Head
+    }
+    
+    expectations.each do |verb, klass|
+      assert_equal klass, c.send(:request_method, verb)
+    end
+  end
+end
+
+class ConnectionOptionsTest < Test::Unit::TestCase
+  
+  def setup
+    @options = generate_options(:server => 'example.org', :port => 555)
+    @default_options = generate_options
+  end
+  
+  def test_server_extracted
+    assert_key_transfered(:server, 'example.org', @options)
+  end
+  
+  def test_port_extracted
+    assert_key_transfered(:port, 555, @options)
+  end
+  
+  def test_server_defaults_to_default_host
+    assert_equal DEFAULT_HOST, @default_options[:server]
+  end
+  
+  def test_port_defaults_to_80_if_use_ssl_is_false
+    assert_equal 80, @default_options[:port]
+  end
+  
+  def test_port_is_set_to_443_if_use_ssl_is_true
+    options = generate_options(:use_ssl => true)
+    assert_equal 443, options[:port]
+  end
+  
+  def test_explicit_port_trumps_use_ssl
+    options = generate_options(:port => 555, :use_ssl => true)
+    assert_equal 555, options[:port]
+  end
+  
+  def test_invalid_options_raise
+    assert_raises(InvalidConnectionOption) do
+      generate_options(:host => 'campfire.s3.amazonaws.com')
+    end
+  end
+  
+  private
+  
+    def assert_key_transfered(key, value, options)
+      assert_equal value, options[key]
+      assert !options.instance_variable_get('@options').has_key?(key)
+    end
+      
+    def generate_options(options = {})
+      Connection::Options.new(options)
+    end
+end

data/test/error_test.rb

@@ -0,0 +1,75 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class ErrorTest < Test::Unit::TestCase
+  def setup
+    @container = AWS::S3
+    @error = Error.new(Parsing::XmlParser.new(Fixtures::Errors.access_denied))
+  end
+  
+  def teardown
+    @container.send(:remove_const, :NotImplemented) if @container.const_defined?(:NotImplemented)
+  end
+  
+  def test_error_class_is_automatically_generated
+    assert !@container.const_defined?('NotImplemented')
+    error = Error.new(Parsing::XmlParser.new(Fixtures::Errors.not_implemented))
+    assert @container.const_defined?('NotImplemented')
+  end
+  
+  def test_error_contains_attributes
+    assert_equal 'Access Denied', @error.message
+  end
+  
+  def test_error_is_raisable_as_exception
+    assert_raises(@container::AccessDenied) do
+      @error.raise
+    end
+  end
+  
+  def test_error_message_is_passed_along_to_exception    
+    @error.raise
+  rescue @container::AccessDenied => e
+    assert_equal 'Access Denied', e.message
+  end
+  
+  def test_response_is_passed_along_to_exception
+    response = Error::Response.new(FakeResponse.new(:code => 409, :body => Fixtures::Errors.access_denied))
+    response.error.raise
+  rescue @container::ResponseError => e
+    assert e.response
+    assert_kind_of Error::Response, e.response
+    assert_equal response.error, e.response.error
+  end
+  
+  def test_exception_class_clash
+    assert !@container.const_defined?(:NotImplemented)
+    # Create a class that does not inherit from exception that has the same name as the class
+    # the Error instance is about to attempt to find or create
+    @container.const_set(:NotImplemented, Class.new)
+    assert @container.const_defined?(:NotImplemented)
+    
+    assert_raises(ExceptionClassClash) do
+      Error.new(Parsing::XmlParser.new(Fixtures::Errors.not_implemented))
+    end
+  end
+  
+  def test_error_response_handles_attributes_with_no_value
+    Bucket.in_test_mode do
+      Bucket.request_returns :body => Fixtures::Errors.error_with_no_message, :code => 500
+      
+      begin
+        Bucket.create('foo', 'invalid-argument' => 'bad juju')
+      rescue ResponseError => error
+      end
+    
+      assert_nothing_raised do
+        error.response.error.message
+      end
+      assert_nil error.response.error.message
+      
+      assert_raises(NoMethodError) do
+        error.response.error.non_existant_method
+      end
+    end
+  end
+end