More tests from Wesley Spikes.

This commit is contained in:
Tommi Virtanen 2011-06-10 09:43:18 -07:00
parent 818523f48c
commit 53596480f2
2 changed files with 118 additions and 1 deletions

8
AnonymousAuth.py Normal file
View file

@ -0,0 +1,8 @@
from boto.auth_handler import AuthHandler
class AnonymousAuthHandler(AuthHandler):
def __init__(self, host, config, provider):
AuthHandler.__init__(self, host, config, provider)
def add_auth(self, http_request, **kwargs):
return # Nothing to do for anonymous access!

View file

@ -14,6 +14,7 @@ from nose.tools import eq_ as eq
from nose.plugins.attrib import attr
from utils import assert_raises
import AnonymousAuth
NONEXISTENT_EMAIL = 'doesnotexist@dreamhost.com.invalid'
@ -235,12 +236,22 @@ def test_object_read_notexist():
eq(e.error_code, 'NoSuchKey')
def test_object_write_then_read():
# This should test the basic lifecycle of the key
def test_object_write_read_update_read_delete():
bucket = get_new_bucket()
# Write
key = bucket.new_key('foo')
key.set_contents_from_string('bar')
# Read
got = key.get_contents_as_string()
eq(got, 'bar')
# Update
key.set_contents_from_string('soup')
# Read
got = key.get_contents_as_string()
eq(got, 'soup')
# Delete
key.delete()
def test_object_set_get_metadata():
@ -296,6 +307,8 @@ def test_bucket_create_naming_bad_short_one():
def test_bucket_create_naming_bad_short_two():
check_bad_bucket_name('aa')
def test_bucket_create_naming_good_short_3():
check_good_bucket_name('aaa')
def test_bucket_create_naming_bad_long():
check_bad_bucket_name(256*'a')
@ -527,6 +540,10 @@ def test_bucket_acl_grant_userid():
key.set_contents_from_string('bar')
# This test will fail on DH Objects. DHO allows multiple users with one account, which
# would violate the uniqueness requirement of a user's email. As such, DHO users are
# created without an email.
@attr('fails_on_dho')
def test_bucket_acl_grant_email():
bucket = get_new_bucket()
# add alt user
@ -767,3 +784,95 @@ def test_object_giveaway():
raise RuntimeError("S3 implementation allowed us to give away an object!")
except boto.exception.S3ResponseError, e:
pass
def test_buckets_create_then_list():
create_buckets = [get_new_bucket() for i in xrange(5)]
list_buckets = s3.main.get_all_buckets()
for bucket in create_buckets:
if not len([f for f in list_buckets if f.name == bucket.name]):
raise RuntimeError("S3 implementation's GET on Service did not return bucket we created")
# Common code to create a connection object, which'll use bad authorization information
def _create_connection_bad_auth():
# We're going to need to manually build a connection using bad authorization info.
# But to save the day, lets just hijack the settings from s3.main. :)
main = s3.main
conn = boto.s3.connection.S3Connection(
aws_access_key_id='badauth',
aws_secret_access_key='roflmao',
is_secure=main.is_secure,
port=main.port,
host=main.host,
calling_format=main.calling_format,
)
return conn
def test_list_buckets_anonymous():
# Get a connection with bad authorization, then change it to be our new Anonymous auth mechanism,
# emulating standard HTTP access.
#
# While it may have been possible to use httplib directly, doing it this way takes care of also
# allowing us to vary the calling format in testing.
conn = _create_connection_bad_auth()
conn._auth_handler = AnonymousAuth.AnonymousAuthHandler(None, None, None) # Doesn't need this
buckets = conn.get_all_buckets()
eq(len(buckets), 0)
def test_list_buckets_bad_auth():
conn = _create_connection_bad_auth()
e = assert_raises(boto.exception.S3ResponseError, conn.get_all_buckets)
eq(e.status, 403)
eq(e.reason, 'Forbidden')
eq(e.error_code, 'AccessDenied')
def test_bucket_create_good_starts_alpha():
check_good_bucket_name('a'*10)
def test_bucket_create_good_starts_number():
check_good_bucket_name('1'*10)
def test_bucket_create_naming_good_contains_period():
check_good_bucket_name('aaa.111')
def test_bucket_create_naming_good_contains_hyphen():
check_good_bucket_name('aaa-111')
def test_bucket_delete_notexist():
name = '{prefix}{num}'.format(
prefix=prefix,
num=next(bucket_counter),
)
try:
bucket = s3.main.delete_bucket(name)
raise RuntimeError("S3 implementation allowed us to delete a bucket that shouldn't exist")
except boto.exception.S3ResponseError, e:
pass
def test_object_write_to_nonexist_bucket():
name = '{prefix}{num}'.format(
prefix=prefix,
num=next(bucket_counter),
)
bucket = s3.main.get_bucket(name, validate=False)
key = bucket.new_key('foo123bar')
try:
key.set_contents_from_string('foo')
raise RuntimeError("S3 implementation allowed us to write a key to a bucket that shouldn't exist")
except boto.exception.S3ResponseError, e:
pass
def test_object_copy_same_bucket():
bucket = get_new_bucket()
key = bucket.new_key('foo123bar')
key.set_contents_from_string('foo')
key.copy(bucket, 'bar321foo')
key2 = bucket.get_key('bar321foo')
eq(key2.get_contents_as_string(), 'foo')
def test_object_copy_diff_bucket():
buckets = [get_new_bucket(), get_new_bucket()]
key = buckets[0].new_key('foo123bar')
key.set_contents_from_string('foo')
key.copy(buckets[1], 'bar321foo')
key2 = buckets[1].get_key('bar321foo')
eq(key2.get_contents_as_string(), 'foo')