s3tests: modify cross region copy, sync triggering

Signed-off-by: Yehuda Sadeh <yehuda@inktank.com>
This commit is contained in:
Yehuda Sadeh 2013-07-26 22:19:36 -07:00
parent b4441e3057
commit 3b733245c5
2 changed files with 46 additions and 25 deletions

View file

@ -179,6 +179,9 @@ class RegionsConn:
self.master = None self.master = None
self.secondaries = [] self.secondaries = []
def iteritems(self):
return self.m.iteritems()
def add(self, name, conn): def add(self, name, conn):
self.m[name] = conn self.m[name] = conn
if not self.default: if not self.default:

View file

@ -4560,7 +4560,13 @@ def test_region_bucket_create_secondary_access_remove_master():
conn.delete_bucket(bucket) conn.delete_bucket(bucket)
def region_sync_meta(conf): # syncs all the regions except for the one passed in
def region_sync_meta(targets, region):
for (k, r) in targets.iteritems():
if r == region:
continue
conf = r.conf
if conf.sync_agent_addr: if conf.sync_agent_addr:
ret = requests.post('http://{addr}:{port}/metadata/partial'.format(addr = conf.sync_agent_addr, port = conf.sync_agent_port)) ret = requests.post('http://{addr}:{port}/metadata/partial'.format(addr = conf.sync_agent_addr, port = conf.sync_agent_port))
eq(ret.status_code, 200) eq(ret.status_code, 200)
@ -4582,7 +4588,7 @@ def test_region_bucket_create_master_access_remove_secondary():
conn = r.connection conn = r.connection
bucket = get_new_bucket(master) bucket = get_new_bucket(master)
region_sync_meta(r.conf) region_sync_meta(targets.main, master)
e = assert_raises(boto.exception.S3ResponseError, conn.get_bucket, bucket.name) e = assert_raises(boto.exception.S3ResponseError, conn.get_bucket, bucket.name)
eq(e.status, 301) eq(e.status, 301)
@ -4590,7 +4596,6 @@ def test_region_bucket_create_master_access_remove_secondary():
e = assert_raises(boto.exception.S3ResponseError, conn.delete_bucket, bucket.name) e = assert_raises(boto.exception.S3ResponseError, conn.delete_bucket, bucket.name)
eq(e.status, 301) eq(e.status, 301)
master_conn.delete_bucket(bucket) master_conn.delete_bucket(bucket)
@attr(resource='object') @attr(resource='object')
@ -4601,15 +4606,23 @@ def test_region_bucket_create_master_access_remove_secondary():
def test_region_copy_object(): def test_region_copy_object():
assert_can_test_multiregion() assert_can_test_multiregion()
master = targets.main.master for (k, dest) in targets.main.iteritems():
master_conn = master.connection dest_conn = dest.connection
dest_bucket = get_new_bucket(dest)
print 'created new dest bucket ', dest_bucket.name
region_sync_meta(targets.main, dest)
master_bucket = get_new_bucket(master)
for file_size in (1024, 1024 * 1024, 10 * 1024 * 1024, for file_size in (1024, 1024 * 1024, 10 * 1024 * 1024,
100 * 1024 * 1024): 100 * 1024 * 1024):
for r in targets.main.secondaries: for (k2, r) in targets.main.iteritems():
if r == dest_conn:
continue
conn = r.connection conn = r.connection
bucket = get_new_bucket(r) bucket = get_new_bucket(r)
print 'created bucket', bucket.name
region_sync_meta(targets.main, r)
content = 'testcontent' content = 'testcontent'
@ -4617,10 +4630,15 @@ def test_region_copy_object():
fp_a = FakeWriteFile(file_size, 'A') fp_a = FakeWriteFile(file_size, 'A')
key.set_contents_from_file(fp_a) key.set_contents_from_file(fp_a)
dest_key = master_bucket.copy_key('testobj-dest', bucket.name, key.name) dest_key = dest_bucket.copy_key('testobj-dest', bucket.name, key.name)
# verify dest # verify dest
_verify_atomic_key_data(dest_key, file_size, 'A') _verify_atomic_key_data(dest_key, file_size, 'A')
bucket.delete_key(key.name) bucket.delete_key(key.name)
print 'removing bucket', bucket.name
conn.delete_bucket(bucket) conn.delete_bucket(bucket)
dest_bucket.delete_key(dest_key.name)
dest_conn.delete_bucket(dest_bucket)