r = callableObj(*args, **kwargs)
self.fail('Should never reach here')
except Fault, f:
- self.failUnless(f.status == status)
+ if type(status) == types.ListType:
+ self.failUnless(f.status in status)
+ else:
+ self.failUnless(f.status == status)
def assert_not_raises_fault(self, status, callableObj, *args, **kwargs):
"""
self.assert_raises_fault(404, self.client.retrieve_object_metadata,
container, object)
+ def assert_versionlist_structure(self, versionlist):
+ self.assertTrue(type(versionlist) == types.ListType)
+ for elem in versionlist:
+ self.assertTrue(type(elem) == types.ListType)
+ self.assertEqual(len(elem), 2)
+
def upload_random_data(self, container, name, length=1024, type=None,
enc=None, **meta):
data = get_random_data(length)
obj['meta'] = args
path = '/%s/%s' % (container, name)
- self.client.create_object(container, name, StringIO(obj['data']),
- meta, **args)
+ self.client.create_object(container, name, f=StringIO(obj['data']),
+ meta=meta, **args)
return obj
except IOError:
for n in names:
self.objects.append(self.upload_random_data(self.containers[1], n))
+ def test_versions(self):
+ c = self.containers[1]
+ o = self.objects[0]
+ b = self.client.retrieve_object_versionlist(c, o['name'])['versions']
+ self.assert_versionlist_structure(b)
+
+ #update meta
+ meta = {'quality':'AAA', 'stock':True}
+ self.client.update_object_metadata(c, o['name'], **meta)
+
+ a = self.client.retrieve_object_versionlist(c, o['name'])['versions']
+ self.assert_versionlist_structure(a)
+ self.assertEqual(len(b)+1, len(a))
+ self.assertEqual(b, a[:-1])
+
+ #get exact previous version metadata
+ v = a[-2][0]
+ v_meta = self.client.retrieve_object_metadata(c, o['name'],
+ restricted=True,
+ version=v)
+ for k in meta.keys():
+ self.assertTrue(k not in v_meta)
+
+ #update obejct
+ data = get_random_data()
+ self.client.update_object(c, o['name'], StringIO(data))
+
+ aa = self.client.retrieve_object_versionlist(c, o['name'])['versions']
+ self.assert_versionlist_structure(aa)
+ self.assertEqual(len(a)+1, len(aa))
+ self.assertEqual(a, aa[:-1])
+
+ #get exact previous version
+ v = aa[-3][0]
+ v_data = self.client.retrieve_object_version(c, o['name'], version=v)
+ self.assertEqual(o['data'], v_data)
+ self.assertEqual(self.client.retrieve_object(c, o['name']),
+ '%s%s' %(v_data, data))
+
def test_get(self):
#perform get
o = self.client.retrieve_object(self.containers[1],
self.assertEqual(int(zero_meta['content-length']), 0)
self.assertEqual(zero_hash, [])
self.assertEqual(zero_data, '')
+
+ def test_create_object_by_hashmap(self):
+ c = self.container
+ o = 'object'
+ self.upload_random_data(c, o)
+ hashmap = self.client.retrieve_object(c, o, format='json')
+ o2 = 'object-copy'
+ self.client.create_object_by_hashmap(c, o2, hashmap)
+ self.assertEqual(self.client.retrieve_object(c, o),
+ self.client.retrieve_object(c, o))
class ObjectCopy(BaseTestCase):
def setUp(self):
for c in self.containers:
self.client.create_container(c)
self.obj = self.upload_random_data(self.containers[0], o_names[0])
-
- def tearDown(self):
- pass
-
+
def test_copy(self):
with AssertMappingInvariant(self.client.retrieve_object_metadata,
self.containers[0], self.obj['name']):
def test_update_object_invalid_range_and_length(self):
with AssertContentInvariant(self.client.retrieve_object,
self.containers[0], self.obj[0]['name']):
- self.assert_raises_fault(416, self.test_update_object, 499, 0, True,
+ self.assert_raises_fault([400, 416], self.test_update_object, 499, 0, True,
-1)
def test_update_object_invalid_range_with_no_content_length(self):
class ListSharing(BaseTestCase):
def setUp(self):
BaseTestCase.setUp(self)
+ for i in range(2):
+ self.client.create_container('c%s' %i)
self.client.create_container('c')
for i in range(2):
- self.upload_random_data('c', 'o%s' %i)
+ self.upload_random_data('c1', 'o%s' %i)
accounts = OTHER_ACCOUNTS.copy()
self.o1_sharing_with = accounts.popitem()
self.o1_sharing = [self.o1_sharing_with[1]]
- self.client.share_object('c', 'o1', self.o1_sharing, read=True)
+ self.client.share_object('c1', 'o1', self.o1_sharing, read=True)
l = []
for i in range(2):
l.append(accounts.popitem())
- #self.client.set_account_groups({'pithos-dev':'chazapis,verigak,papagian'})
- #self.o2_sharing = 'write=%s' %
- #self.client.share_object('c', 'o2', self.o2_sharing)
- def test_listing(self):
+ def test_list_other_shared(self):
self.other = Pithos_Client(get_server(),
self.o1_sharing_with[0],
self.o1_sharing_with[1],
get_api())
self.assertTrue(get_user() in self.other.list_shared_by_others())
-
+
+ def test_list_my_shared(self):
+ my_shared_containers = self.client.list_containers(shared=True)
+ self.assertTrue('c1' in my_shared_containers)
+ self.assertTrue('c2' not in my_shared_containers)
+
+ my_shared_objects = self.client.list_objects('c1', shared=True)
+ self.assertTrue('o1' in my_shared_objects)
+ self.assertTrue('o2' not in my_shared_objects)
+
class TestGreek(BaseTestCase):
def setUp(self):
BaseTestCase.setUp(self)
self.initial_meta = self.client.retrieve_account_metadata(restricted=True)
#create a group
- self.authorized = ['chazapis', 'verigak', 'gtsouk', 'papagian']
+ self.authorized = ['chazapis', 'verigak', 'gtsouk']
groups = {'pithosdev':','.join(self.authorized)}
self.client.set_account_groups(**groups)
BaseTestCase.tearDown(self)
- def test_read_access(self):
- self.client.create_container('c')
- o = self.upload_random_data('c', 'o')
- self.client.share_object('c', 'o', ['%s:pithosdev' % get_user()])
+ def assert_read(self, authorized=[], any=False):
for token, account in OTHER_ACCOUNTS.items():
cl = Pithos_Client(get_server(), token, account, get_api())
- if account in self.authorized:
+ if account in authorized or any:
self.assert_not_raises_fault(401, cl.retrieve_object_metadata,
'c', 'o', account=get_user())
else:
self.assert_raises_fault(401, cl.retrieve_object_metadata,
'c', 'o', account=get_user())
+ #check inheritance
+ o = self.upload_random_data('c', 'o/also-shared')
+ for token, account in OTHER_ACCOUNTS.items():
+ cl = Pithos_Client(get_server(), token, account, get_api())
+ if account in authorized or any:
+ self.assert_not_raises_fault(401, cl.retrieve_object_metadata,
+ 'c', 'o/also-shared', account=get_user())
+ else:
+ self.assert_raises_fault(401, cl.retrieve_object_metadata,
+ 'c', 'o/also-shared', account=get_user())
- def test_write_access(self):
- self.client.create_container('c')
- o = self.upload_random_data('c', 'o')
- self.client.share_object('c', 'o', ['chazapis'], read=False)
+ def assert_write(self, o_data, authorized=[], any=False):
for token, account in OTHER_ACCOUNTS.items():
cl = Pithos_Client(get_server(), token, account, get_api())
new_data = get_random_data()
- if account == 'chazapis':
+ if account in authorized or any:
+ # test write access
self.assert_not_raises_fault(401, cl.update_object,
'c', 'o', StringIO(new_data),
account=get_user())
- server_data = self.client.retrieve_object('c', 'o')
- self.assertEqual(o['data'], server_data[:len(o['data'])])
- self.assertEqual(new_data, server_data[len(o['data']):])
+ try:
+ # test read access
+ server_data = cl.retrieve_object('c', 'o', account=get_user())
+ self.assertEqual(o_data, server_data[:len(o_data)])
+ self.assertEqual(new_data, server_data[len(o_data):])
+ o_data = server_data
+ except Fault, f:
+ self.failIf(f.status == 401)
else:
self.assert_raises_fault(401, cl.update_object,
'c', 'o', StringIO(new_data),
account=get_user())
+
+ #check inheritance
+ o = self.upload_random_data('c', 'o/also-shared')
+ o_data = o['data']
+ for token, account in OTHER_ACCOUNTS.items():
+ cl = Pithos_Client(get_server(), token, account, get_api())
+ new_data = get_random_data()
+ if account in authorized or any:
+ # test write access
+ self.assert_not_raises_fault(401, cl.update_object,
+ 'c', o['name'],
+ StringIO(new_data),
+ account=get_user())
+ try:
+ server_data = cl.retrieve_object('c', o['name'], account=get_user())
+ self.assertEqual(o_data, server_data[:len(o_data)])
+ self.assertEqual(new_data, server_data[len(o_data):])
+ o_data = server_data
+ except Fault, f:
+ self.failIf(f.status == 401)
+ else:
+ self.assert_raises_fault(401, cl.update_object,
+ 'c', o['name'],
+ StringIO(new_data),
+ account=get_user())
+
+ def test_group_read(self):
+ self.client.create_container('c')
+ o = self.upload_random_data('c', 'o')
+ self.client.share_object('c', 'o', ['%s:pithosdev' % get_user()])
+ self.assert_read(authorized=self.authorized)
+
+ def test_read_many(self):
+ #test read access
+ self.client.create_container('c')
+ o = self.upload_random_data('c', 'o')
+ self.client.share_object('c', 'o', self.authorized)
+ self.assert_read(authorized=self.authorized)
+
+ def test_read_by_everyone(self):
+ self.client.create_container('c')
+ o = self.upload_random_data('c', 'o')
+ self.client.share_object('c', 'o', ['*'])
+ self.assert_read(any=True)
+
+ def test_group_write(self):
+ self.client.create_container('c')
+ o = self.upload_random_data('c', 'o')
+ self.client.share_object('c', 'o', ['%s:pithosdev' % get_user()], read=False)
+ self.assert_write(o['data'], authorized=self.authorized)
+
+ def test_write_many(self):
+ self.client.create_container('c')
+ o = self.upload_random_data('c', 'o')
+ self.client.share_object('c', 'o', self.authorized, read=False)
+ self.assert_write(o['data'], authorized=self.authorized)
+
+ def test_write_by_everyone(self):
+ self.client.create_container('c')
+ o = self.upload_random_data('c', 'o')
+ self.client.share_object('c', 'o', ['*'], read=False)
+ o_data = o['data']
+ self.assert_write(o['data'], any=True)
+
+class TestPublish(BaseTestCase):
+ def test_publish(self):
+ self.client.create_container('c')
+ o_data = self.upload_random_data('c', 'o')['data']
+ self.client.publish_object('c', 'o')
+ meta = self.client.retrieve_object_metadata('c', 'o')
+ self.assertTrue('x-object-public' in meta)
+ url = '/public/%s/c/o' % get_user()
+ self.assertEqual(meta['x-object-public'], url)
+ public_client = Pithos_Client(get_server(), get_auth(), get_user(), api='')
+ data = public_client.get(url)[2]
+ self.assertEqual(o_data, data)
class AssertMappingInvariant(object):
def __init__(self, callable, *args, **kwargs):
'photos/me.jpg']
if __name__ == "__main__":
- unittest.main()
+ if get_user() == 'test':
+ unittest.main()
+ else:
+ print 'Will not run tests as any other user except \'test\' (current user: %s).' % get_user()