4 # Copyright 2011 GRNET S.A. All rights reserved.
6 # Redistribution and use in source and binary forms, with or
7 # without modification, are permitted provided that the following
10 # 1. Redistributions of source code must retain the above
11 # copyright notice, this list of conditions and the following
14 # 2. Redistributions in binary form must reproduce the above
15 # copyright notice, this list of conditions and the following
16 # disclaimer in the documentation and/or other materials
17 # provided with the distribution.
19 # THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
20 # OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22 # PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
23 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
26 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
27 # AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
29 # ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
30 # POSSIBILITY OF SUCH DAMAGE.
32 # The views and conclusions contained in the software and
33 # documentation are those of the authors and should not be
34 # interpreted as representing official policies, either expressed
35 # or implied, of GRNET S.A.
37 from lib.client import Pithos_Client, Fault
38 from lib.util import get_user, get_auth, get_server, get_api
39 from xml.dom import minidom
40 from StringIO import StringIO
52 DATE_FORMATS = ["%a %b %d %H:%M:%S %Y",
53 "%A, %d-%b-%y %H:%M:%S GMT",
54 "%a, %d %b %Y %H:%M:%S GMT"]
67 class BaseTestCase(unittest.TestCase):
68 #TODO unauthorized request
70 self.client = Pithos_Client(get_server(), get_auth(), get_user(),
72 self.invalid_client = Pithos_Client(get_server(), get_auth(), 'invalid',
75 # 'account': ('x-account-container-count',
76 # 'x-account-bytes-used',
88 # 'x-object-manifest',
90 # 'x-object-modified-by',
92 # 'x-object-version-timestamp',
94 # 'container': ('x-container-object-count',
95 # 'x-container-bytes-used',
100 # 'x-container-block-size',
101 # 'x-container-block-hash',
102 # 'x-container-policy-quota',
103 # 'x-container-policy-versioning',
105 # 'x-container-object-meta',
106 # 'x-container-policy-versioning',
109 #self.contentTypes = {'xml':'application/xml',
110 # 'json':'application/json',
118 'x_container_policy_quota',
119 'x_container_policy_versioning',),
127 self.return_codes = (400, 401, 404, 503,)
130 for c in self.client.list_containers():
132 #list objects returns at most 10000 objects
133 #so repeat until there are no more objects
134 objects = self.client.list_objects(c)
138 self.client.delete_object(c, o)
139 self.client.delete_container(c)
141 def assert_status(self, status, codes):
142 l = [elem for elem in self.return_codes]
143 if type(codes) == types.ListType:
147 self.assertTrue(status in l)
149 #def assert_headers(self, headers, type, **exp_meta):
150 # prefix = 'x-%s-meta-' %type
151 # system_headers = [h for h in headers if not h.startswith(prefix)]
152 # for k,v in headers.items():
153 # if k in system_headers:
154 # self.assertTrue(k in headers[type])
156 # k = k.split(prefix)[-1]
157 # self.assertEqual(v, exp_meta[k])
159 def assert_extended(self, data, format, type, size=10000):
161 self._assert_xml(data, type, size)
162 elif format == 'json':
163 self._assert_json(data, type, size)
165 def _assert_json(self, data, type, size):
166 convert = lambda s: s.lower()
167 info = [convert(elem) for elem in self.extended[type]]
168 self.assertTrue(len(data) <= size)
171 if 'subdir' in i.keys():
173 self.assertTrue(item in i.keys())
175 def _assert_xml(self, data, type, size):
176 convert = lambda s: s.lower()
177 info = [convert(elem) for elem in self.extended[type]]
179 info.remove('content_encoding')
183 entities = xml.getElementsByTagName(type)
184 self.assertTrue(len(entities) <= size)
187 self.assertTrue(e.getElementsByTagName(item))
189 def assert_raises_fault(self, status, callableObj, *args, **kwargs):
191 asserts that a Fault with a specific status is raised
192 when callableObj is called with the specific arguments
195 r = callableObj(*args, **kwargs)
196 self.fail('Should never reach here')
198 self.failUnless(f.status == status)
200 def assert_not_raises_fault(self, status, callableObj, *args, **kwargs):
202 asserts that a Fault with a specific status is not raised
203 when callableObj is called with the specific arguments
206 r = callableObj(*args, **kwargs)
208 self.failIfEqual(f.status, status)
210 def assert_container_exists(self, container):
212 asserts the existence of a container
215 self.client.retrieve_container_metadata(container)
217 self.failIf(f.status == 404)
219 def assert_container_not_exists(self, container):
221 asserts there is no such a container
223 self.assert_raises_fault(404, self.client.retrieve_container_metadata,
226 def assert_object_exists(self, container, object):
228 asserts the existence of an object
231 self.client.retrieve_object_metadata(container, object)
233 self.failIf(f.status == 404)
235 def assert_object_not_exists(self, container, object):
237 asserts there is no such an object
239 self.assert_raises_fault(404, self.client.retrieve_object_metadata,
242 def upload_random_data(self, container, name, length=1024, type=None,
244 data = get_random_data(length)
245 return self.upload_data(container, name, data, type, enc, **meta)
247 def upload_data(self, container, name, data, type=None, enc=None, etag=None,
253 obj['hash'] = compute_md5_hash(obj['data'])
256 args['etag'] = etag if etag else obj['hash']
259 guess = mimetypes.guess_type(name)
260 type = type if type else guess[0]
261 enc = enc if enc else guess[1]
264 args['content_type'] = type if type else 'plain/text'
265 args['content_encoding'] = enc if enc else None
269 path = '/%s/%s' % (container, name)
270 self.client.create_object(container, name, StringIO(obj['data']),
277 class AccountHead(BaseTestCase):
279 BaseTestCase.setUp(self)
280 self.account = 'test'
281 self.containers = ['apples', 'bananas', 'kiwis', 'oranges', 'pears']
282 for item in self.containers:
283 self.client.create_container(item)
285 #keep track of initial account groups
286 self.initial_groups = self.client.retrieve_account_groups()
288 #keep track of initial account meta
289 self.initial_meta = self.client.retrieve_account_metadata(restricted=True)
292 self.client.update_account_metadata(**meta)
293 self.updated_meta = self.initial_meta.update(meta)
296 #delete additionally created meta
298 for m in self.client.retrieve_account_metadata(restricted=True):
299 if m not in self.initial_meta:
301 self.client.delete_account_metadata(l)
303 #delete additionally created groups
305 for g in self.client.retrieve_account_groups():
306 if g not in self.initial_groups:
308 self.client.unset_account_groups(l)
310 BaseTestCase.tearDown(self)
312 def test_get_account_meta(self):
313 meta = self.client.retrieve_account_metadata()
315 containers = self.client.list_containers()
316 l = str(len(containers))
317 self.assertEqual(meta['x-account-container-count'], l)
320 m = self.client.retrieve_container_metadata(c)
321 size = size + int(m['x-container-bytes-used'])
322 self.assertEqual(meta['x-account-bytes-used'], str(size))
324 def test_get_account_401(self):
325 self.assert_raises_fault(401,
326 self.invalid_client.retrieve_account_metadata)
328 def test_get_account_meta_until(self):
329 t = datetime.datetime.utcnow()
330 past = t - datetime.timedelta(minutes=-15)
331 past = int(_time.mktime(past.timetuple()))
333 meta = {'premium':True}
334 self.client.update_account_metadata(**meta)
335 meta = self.client.retrieve_account_metadata(restricted=True,
337 self.assertTrue('premium' not in meta)
339 meta = self.client.retrieve_account_metadata(restricted=True)
340 self.assertTrue('premium' in meta)
342 def test_get_account_meta_until_invalid_date(self):
343 meta = {'premium':True}
344 self.client.update_account_metadata(**meta)
345 meta = self.client.retrieve_account_metadata(restricted=True,
347 self.assertTrue('premium' in meta)
349 class AccountGet(BaseTestCase):
351 BaseTestCase.setUp(self)
352 self.account = 'test'
353 #create some containers
354 self.containers = ['apples', 'bananas', 'kiwis', 'oranges', 'pears']
355 for item in self.containers:
356 self.client.create_container(item)
360 containers = self.client.list_containers()
361 self.assertEquals(self.containers, containers)
363 def test_list_401(self):
364 self.assert_raises_fault(401, self.invalid_client.list_containers)
366 def test_list_with_limit(self):
368 containers = self.client.list_containers(limit=limit)
369 self.assertEquals(len(containers), limit)
370 self.assertEquals(self.containers[:2], containers)
372 def test_list_with_marker(self):
375 containers = self.client.list_containers(limit=l, marker=m)
376 i = self.containers.index(m) + 1
377 self.assertEquals(self.containers[i:(i+l)], containers)
380 containers = self.client.list_containers(limit=l, marker=m)
381 i = self.containers.index(m) + 1
382 self.assertEquals(self.containers[i:(i+l)], containers)
384 def test_list_json_with_marker(self):
387 containers = self.client.list_containers(limit=l, marker=m, format='json')
388 self.assert_extended(containers, 'json', 'container', l)
389 self.assertEqual(containers[0]['name'], 'kiwis')
390 self.assertEqual(containers[1]['name'], 'oranges')
392 def test_list_xml_with_marker(self):
395 xml = self.client.list_containers(limit=l, marker=m, format='xml')
396 self.assert_extended(xml, 'xml', 'container', l)
397 nodes = xml.getElementsByTagName('name')
398 self.assertEqual(len(nodes), 1)
399 self.assertEqual(nodes[0].childNodes[0].data, 'pears')
401 def test_if_modified_since(self):
402 t = datetime.datetime.utcnow()
403 t2 = t - datetime.timedelta(minutes=10)
406 self.client.create_container('dummy')
408 for f in DATE_FORMATS:
409 past = t2.strftime(f)
411 c = self.client.list_containers(if_modified_since=past)
412 self.assertEqual(len(c), len(self.containers) + 1)
414 self.failIf(f.status == 304) #fail if not modified
416 def test_if_modified_since_invalid_date(self):
417 c = self.client.list_containers(if_modified_since='')
418 self.assertEqual(len(c), len(self.containers))
420 def test_if_not_modified_since(self):
421 now = datetime.datetime.utcnow()
422 since = now + datetime.timedelta(1)
424 for f in DATE_FORMATS:
425 args = {'if_modified_since':'%s' %since.strftime(f)}
428 self.assert_raises_fault(304, self.client.list_containers, **args)
430 def test_if_unmodified_since(self):
431 now = datetime.datetime.utcnow()
432 since = now + datetime.timedelta(1)
434 for f in DATE_FORMATS:
435 c = self.client.list_containers(if_unmodified_since=since.strftime(f))
438 self.assertEqual(self.containers, c)
440 def test_if_unmodified_since_precondition_failed(self):
441 t = datetime.datetime.utcnow()
442 t2 = t - datetime.timedelta(minutes=10)
445 self.client.create_container('dummy')
447 for f in DATE_FORMATS:
448 past = t2.strftime(f)
450 args = {'if_unmodified_since':'%s' %past}
452 #assert precondition failed
453 self.assert_raises_fault(412, self.client.list_containers, **args)
455 class AccountPost(BaseTestCase):
457 BaseTestCase.setUp(self)
458 self.account = 'test'
459 self.containers = ['apples', 'bananas', 'kiwis', 'oranges', 'pears']
460 for item in self.containers:
461 self.client.create_container(item)
463 #keep track of initial account groups
464 self.initial_groups = self.client.retrieve_account_groups()
466 #keep track of initial account meta
467 self.initial_meta = self.client.retrieve_account_metadata(restricted=True)
470 self.client.update_account_metadata(**meta)
471 self.updated_meta = self.initial_meta.update(meta)
474 #delete additionally created meta
476 for m in self.client.retrieve_account_metadata(restricted=True):
477 if m not in self.initial_meta:
479 self.client.delete_account_metadata(l)
481 #delete additionally created groups
483 for g in self.client.retrieve_account_groups():
484 if g not in self.initial_groups:
486 self.client.unset_account_groups(l)
488 BaseTestCase.tearDown(self)
490 def test_update_meta(self):
491 with AssertMappingInvariant(self.client.retrieve_account_groups):
492 meta = {'test':'test', 'tost':'tost'}
493 self.client.update_account_metadata(**meta)
495 meta.update(self.initial_meta)
496 self.assertEqual(meta,
497 self.client.retrieve_account_metadata(
500 def test_invalid_account_update_meta(self):
501 meta = {'test':'test', 'tost':'tost'}
502 self.assert_raises_fault(401,
503 self.invalid_client.update_account_metadata,
506 def test_reset_meta(self):
507 with AssertMappingInvariant(self.client.retrieve_account_groups):
508 meta = {'test':'test', 'tost':'tost'}
509 self.client.update_account_metadata(**meta)
511 meta = {'test':'test33'}
512 self.client.reset_account_metadata(**meta)
514 self.assertEqual(meta, self.client.retrieve_account_metadata(restricted=True))
516 def test_delete_meta(self):
517 with AssertMappingInvariant(self.client.retrieve_account_groups):
518 meta = {'test':'test', 'tost':'tost'}
519 self.client.update_account_metadata(**meta)
521 self.client.delete_account_metadata(meta.keys())
523 account_meta = self.client.retrieve_account_metadata(restricted=True)
525 self.assertTrue(m not in account_meta.keys())
527 def test_set_account_groups(self):
528 with AssertMappingInvariant(self.client.retrieve_account_metadata):
529 groups = {'pithosdev':'verigak,gtsouk,chazapis'}
530 self.client.set_account_groups(**groups)
532 self.assertEqual(groups, self.client.retrieve_account_groups())
534 more_groups = {'clientsdev':'pkanavos,mvasilak'}
535 self.client.set_account_groups(**more_groups)
537 groups.update(more_groups)
538 self.assertEqual(set(groups['pithosdev']),
539 set(self.client.retrieve_account_groups()['pithosdev']))
541 def test_reset_account_groups(self):
542 with AssertMappingInvariant(self.client.retrieve_account_metadata):
543 groups = {'pithosdev':'verigak,gtsouk,chazapis',
544 'clientsdev':'pkanavos,mvasilak'}
545 self.client.set_account_groups(**groups)
547 self.assertEqual(groups, self.client.retrieve_account_groups())
549 groups = {'pithosdev':'verigak,gtsouk,chazapis,papagian'}
550 self.client.reset_account_groups(**groups)
552 self.assertEqual(groups['pithosdev'].split(','),
553 self.client.retrieve_account_groups()['pithosdev'].split(','))
555 def test_delete_account_groups(self):
556 with AssertMappingInvariant(self.client.retrieve_account_metadata):
557 groups = {'pithosdev':'verigak,gtsouk,chazapis',
558 'clientsdev':'pkanavos,mvasilak'}
559 self.client.set_account_groups(**groups)
561 self.client.unset_account_groups(groups.keys())
563 self.assertEqual({}, self.client.retrieve_account_groups())
565 class ContainerHead(BaseTestCase):
567 BaseTestCase.setUp(self)
568 self.account = 'test'
569 self.container = 'apples'
570 self.client.create_container(self.container)
572 def test_get_meta(self):
573 meta = {'trash':'true'}
574 t1 = datetime.datetime.utcnow()
575 o = self.upload_random_data(self.container, o_names[0], **meta)
577 headers = self.client.retrieve_container_metadata(self.container)
578 self.assertEqual(headers['x-container-object-count'], '1')
579 self.assertEqual(headers['x-container-bytes-used'], str(len(o['data'])))
580 t2 = datetime.datetime.strptime(headers['last-modified'], DATE_FORMATS[2])
582 threashold = datetime.timedelta(seconds=1)
583 self.assertTrue(delta < threashold)
584 self.assertTrue(headers['x-container-object-meta'])
585 self.assertTrue('Trash' in headers['x-container-object-meta'])
587 class ContainerGet(BaseTestCase):
589 BaseTestCase.setUp(self)
590 self.account = 'test'
591 self.container = ['pears', 'apples']
592 for c in self.container:
593 self.client.create_container(c)
595 for o in o_names[:8]:
596 self.obj.append(self.upload_random_data(self.container[0], o))
597 for o in o_names[8:]:
598 self.obj.append(self.upload_random_data(self.container[1], o))
600 def test_list_objects(self):
601 objects = self.client.list_objects(self.container[0])
602 l = [elem['name'] for elem in self.obj[:8]]
604 self.assertEqual(objects, l)
606 def test_list_objects_containing_slash(self):
607 self.client.create_container('test')
608 self.upload_random_data('test', '/objectname')
610 objects = self.client.list_objects('test')
611 self.assertEqual(objects, ['/objectname'])
613 objects = self.client.list_objects('test', format='json')
614 self.assertEqual(objects[0]['name'], '/objectname')
616 objects = self.client.list_objects('test', format='xml')
617 self.assert_extended(objects, 'xml', 'object')
618 node_name = objects.getElementsByTagName('name')[0]
619 self.assertEqual(node_name.firstChild.data, '/objectname')
621 #objects = self.client.list_objects('test', prefix='/')
622 #self.assertEqual(objects, ['/objectname'])
624 #objects = self.client.list_objects('test', path='/')
625 #self.assertEqual(objects, ['/objectname'])
627 #objects = self.client.list_objects('test', prefix='/', delimiter='n')
628 #self.assertEqual(objects, ['/object'])
630 def test_list_objects_with_limit_marker(self):
631 objects = self.client.list_objects(self.container[0], limit=2)
632 l = [elem['name'] for elem in self.obj[:8]]
634 self.assertEqual(objects, l[:2])
636 markers = ['How To Win Friends And Influence People.pdf',
640 objects = self.client.list_objects(self.container[0], limit=limit,
642 l = [elem['name'] for elem in self.obj[:8]]
644 start = l.index(m) + 1
646 end = len(l) >= end and end or len(l)
647 self.assertEqual(objects, l[start:end])
650 def _test_list_limit_exceeds(self):
651 self.client.create_container('pithos')
653 for i in range(10001):
654 self.client.create_zero_length_object('pithos', i)
656 self.assertEqual(10000, len(self.client.list_objects('pithos')))
658 def test_list_empty_params(self):
659 objects = self.client.get('/%s/%s' % (get_user(), self.container[0]))[2]
661 objects = objects.strip().split('\n')
662 self.assertEqual(objects,
663 self.client.list_objects(self.container[0]))
665 def test_list_pseudo_hierarchical_folders(self):
666 objects = self.client.list_objects(self.container[1], prefix='photos',
668 self.assertEquals(['photos/animals/', 'photos/me.jpg',
669 'photos/plants/'], objects)
671 objects = self.client.list_objects(self.container[1],
672 prefix='photos/animals',
674 l = ['photos/animals/cats/', 'photos/animals/dogs/']
675 self.assertEquals(l, objects)
677 objects = self.client.list_objects(self.container[1], path='photos')
678 self.assertEquals(['photos/me.jpg'], objects)
680 def test_extended_list_json(self):
681 objects = self.client.list_objects(self.container[1], format='json',
682 limit=2, prefix='photos/animals',
684 self.assertEqual(objects[0]['subdir'], 'photos/animals/cats/')
685 self.assertEqual(objects[1]['subdir'], 'photos/animals/dogs/')
687 def test_extended_list_xml(self):
688 xml = self.client.list_objects(self.container[1], format='xml', limit=4,
689 prefix='photos', delimiter='/')
690 self.assert_extended(xml, 'xml', 'object', size=4)
691 dirs = xml.getElementsByTagName('subdir')
692 self.assertEqual(len(dirs), 2)
693 self.assertEqual(dirs[0].attributes['name'].value, 'photos/animals/')
694 self.assertEqual(dirs[1].attributes['name'].value, 'photos/plants/')
696 objects = xml.getElementsByTagName('name')
697 self.assertEqual(len(objects), 1)
698 self.assertEqual(objects[0].childNodes[0].data, 'photos/me.jpg')
700 def test_list_meta_double_matching(self):
701 meta = {'quality':'aaa', 'stock':'true'}
702 self.client.update_object_metadata(self.container[0],
703 self.obj[0]['name'], **meta)
704 obj = self.client.list_objects(self.container[0], meta='Quality,Stock')
705 self.assertEqual(len(obj), 1)
706 self.assertTrue(obj, self.obj[0]['name'])
708 def test_list_using_meta(self):
709 meta = {'quality':'aaa'}
710 for o in self.obj[:2]:
711 self.client.update_object_metadata(self.container[0], o['name'],
713 meta = {'stock':'true'}
714 for o in self.obj[3:5]:
715 self.client.update_object_metadata(self.container[0], o['name'],
718 obj = self.client.list_objects(self.container[0], meta='Quality')
719 self.assertEqual(len(obj), 2)
720 self.assertTrue(obj, [o['name'] for o in self.obj[:2]])
722 # test case insensitive
723 obj = self.client.list_objects(self.container[0], meta='quality')
724 self.assertEqual(len(obj), 2)
725 self.assertTrue(obj, [o['name'] for o in self.obj[:2]])
727 # test multiple matches
728 obj = self.client.list_objects(self.container[0], meta='Quality,Stock')
729 self.assertEqual(len(obj), 4)
730 self.assertTrue(obj, [o['name'] for o in self.obj[:4]])
732 # test non 1-1 multiple match
733 obj = self.client.list_objects(self.container[0], meta='Quality,aaaa')
734 self.assertEqual(len(obj), 2)
735 self.assertTrue(obj, [o['name'] for o in self.obj[:2]])
737 def test_if_modified_since(self):
738 t = datetime.datetime.utcnow()
739 t2 = t - datetime.timedelta(minutes=10)
742 self.upload_random_data(self.container[0], o_names[0])
744 for f in DATE_FORMATS:
745 past = t2.strftime(f)
747 o = self.client.list_objects(self.container[0],
748 if_modified_since=past)
750 self.client.list_objects(self.container[0]))
752 self.failIf(f.status == 304) #fail if not modified
754 def test_if_modified_since_invalid_date(self):
755 headers = {'if-modified-since':''}
756 o = self.client.list_objects(self.container[0], if_modified_since='')
757 self.assertEqual(o, self.client.list_objects(self.container[0]))
759 def test_if_not_modified_since(self):
760 now = datetime.datetime.utcnow()
761 since = now + datetime.timedelta(1)
763 for f in DATE_FORMATS:
764 args = {'if_modified_since':'%s' %since.strftime(f)}
767 self.assert_raises_fault(304, self.client.list_objects,
768 self.container[0], **args)
770 def test_if_unmodified_since(self):
771 now = datetime.datetime.utcnow()
772 since = now + datetime.timedelta(1)
774 for f in DATE_FORMATS:
775 obj = self.client.list_objects(self.container[0],
776 if_unmodified_since=since.strftime(f))
779 self.assertEqual(obj, self.client.list_objects(self.container[0]))
781 def test_if_unmodified_since_precondition_failed(self):
782 t = datetime.datetime.utcnow()
783 t2 = t - datetime.timedelta(minutes=10)
786 self.client.create_container('dummy')
788 for f in DATE_FORMATS:
789 past = t2.strftime(f)
791 args = {'if_unmodified_since':'%s' %past}
793 #assert precondition failed
794 self.assert_raises_fault(412, self.client.list_objects,
795 self.container[0], **args)
797 class ContainerPut(BaseTestCase):
799 BaseTestCase.setUp(self)
800 self.account = 'test'
801 self.containers = ['c1', 'c2']
803 def test_create(self):
804 self.client.create_container(self.containers[0])
805 containers = self.client.list_containers()
806 self.assertTrue(self.containers[0] in containers)
807 self.assert_container_exists(self.containers[0])
809 def test_create_twice(self):
810 self.client.create_container(self.containers[0])
811 self.assertTrue(not self.client.create_container(self.containers[0]))
813 class ContainerPost(BaseTestCase):
815 BaseTestCase.setUp(self)
816 self.account = 'test'
817 self.container = 'apples'
818 self.client.create_container(self.container)
820 def test_update_meta(self):
821 meta = {'test':'test33',
823 self.client.update_container_metadata(self.container, **meta)
824 headers = self.client.retrieve_container_metadata(self.container)
825 for k,v in meta.items():
826 k = 'x-container-meta-%s' % k
827 self.assertTrue(headers[k])
828 self.assertEqual(headers[k], v)
830 class ContainerDelete(BaseTestCase):
832 BaseTestCase.setUp(self)
833 self.account = 'test'
834 self.containers = ['c1', 'c2']
835 for c in self.containers:
836 self.client.create_container(c)
837 self.upload_random_data(self.containers[1], o_names[0])
839 def test_delete(self):
840 status = self.client.delete_container(self.containers[0])[0]
841 self.assertEqual(status, 204)
843 def test_delete_non_empty(self):
844 self.assert_raises_fault(409, self.client.delete_container,
847 def test_delete_invalid(self):
848 self.assert_raises_fault(404, self.client.delete_container, 'c3')
850 class ObjectHead(BaseTestCase):
853 class ObjectGet(BaseTestCase):
855 BaseTestCase.setUp(self)
856 self.account = 'test'
857 self.containers = ['c1', 'c2']
858 #create some containers
859 for c in self.containers:
860 self.client.create_container(c)
863 names = ('obj1', 'obj2')
866 self.objects.append(self.upload_random_data(self.containers[1], n))
870 o = self.client.retrieve_object(self.containers[1],
871 self.objects[0]['name'],
872 self.objects[0]['meta'])
873 self.assertEqual(o, self.objects[0]['data'])
875 def test_get_invalid(self):
876 self.assert_raises_fault(404, self.client.retrieve_object,
877 self.containers[0], self.objects[0]['name'])
879 def test_get_partial(self):
880 #perform get with range
881 status, headers, data = self.client.request_object(self.containers[1],
882 self.objects[0]['name'],
885 #assert successful partial content
886 self.assertEqual(status, 206)
889 self.assertEqual(headers['content-type'],
890 self.objects[0]['meta']['content_type'])
892 #assert content length
893 self.assertEqual(int(headers['content-length']), 500)
896 self.assertEqual(self.objects[0]['data'][:500], data)
898 def test_get_final_500(self):
899 #perform get with range
900 headers = {'range':'bytes=-500'}
901 status, headers, data = self.client.request_object(self.containers[1],
902 self.objects[0]['name'],
905 #assert successful partial content
906 self.assertEqual(status, 206)
909 self.assertEqual(headers['content-type'],
910 self.objects[0]['meta']['content_type'])
912 #assert content length
913 self.assertEqual(int(headers['content-length']), 500)
916 self.assertTrue(self.objects[0]['data'][-500:], data)
918 def test_get_rest(self):
919 #perform get with range
920 offset = len(self.objects[0]['data']) - 500
921 status, headers, data = self.client.request_object(self.containers[1],
922 self.objects[0]['name'],
923 range='bytes=%s-' %offset)
925 #assert successful partial content
926 self.assertEqual(status, 206)
929 self.assertEqual(headers['content-type'],
930 self.objects[0]['meta']['content_type'])
932 #assert content length
933 self.assertEqual(int(headers['content-length']), 500)
936 self.assertTrue(self.objects[0]['data'][-500:], data)
938 def test_get_range_not_satisfiable(self):
939 #perform get with range
940 offset = len(self.objects[0]['data']) + 1
942 #assert range not satisfiable
943 self.assert_raises_fault(416, self.client.retrieve_object,
944 self.containers[1], self.objects[0]['name'],
945 range='bytes=0-%s' %offset)
947 def test_multiple_range(self):
948 #perform get with multiple range
949 ranges = ['0-499', '-500', '1000-']
950 bytes = 'bytes=%s' % ','.join(ranges)
951 status, headers, data = self.client.request_object(self.containers[1],
952 self.objects[0]['name'],
955 # assert partial content
956 self.assertEqual(status, 206)
958 # assert Content-Type of the reply will be multipart/byteranges
959 self.assertTrue(headers['content-type'])
960 content_type_parts = headers['content-type'].split()
961 self.assertEqual(content_type_parts[0], ('multipart/byteranges;'))
963 boundary = '--%s' %content_type_parts[1].split('=')[-1:][0]
964 cparts = data.split(boundary)[1:-1]
966 # assert content parts are exactly 2
967 self.assertEqual(len(cparts), len(ranges))
969 # for each content part assert headers
972 content = cpart.split('\r\n')
973 headers = content[1:3]
974 content_range = headers[0].split(': ')
975 self.assertEqual(content_range[0], 'Content-Range')
977 r = ranges[i].split('-')
978 if not r[0] and not r[1]:
981 start = len(self.objects[0]['data']) - int(r[1])
982 end = len(self.objects[0]['data'])
985 end = len(self.objects[0]['data'])
989 fdata = self.objects[0]['data'][start:end]
990 sdata = '\r\n'.join(content[4:-1])
991 self.assertEqual(len(fdata), len(sdata))
992 self.assertEquals(fdata, sdata)
995 def test_multiple_range_not_satisfiable(self):
996 #perform get with multiple range
997 out_of_range = len(self.objects[0]['data']) + 1
998 ranges = ['0-499', '-500', '%d-' %out_of_range]
999 bytes = 'bytes=%s' % ','.join(ranges)
1001 # assert partial content
1002 self.assert_raises_fault(416, self.client.retrieve_object,
1004 self.objects[0]['name'], range=bytes)
1006 def test_get_with_if_match(self):
1007 #perform get with If-Match
1008 etag = self.objects[0]['hash']
1009 status, headers, data = self.client.request_object(self.containers[1],
1010 self.objects[0]['name'],
1013 self.assertEqual(status, 200)
1015 #assert content-type
1016 self.assertEqual(headers['content-type'],
1017 self.objects[0]['meta']['content_type'])
1019 #assert response content
1020 self.assertEqual(self.objects[0]['data'], data)
1022 def test_get_with_if_match_star(self):
1023 #perform get with If-Match *
1024 headers = {'if-match':'*'}
1025 status, headers, data = self.client.request_object(self.containers[1],
1026 self.objects[0]['name'],
1029 self.assertEqual(status, 200)
1031 #assert content-type
1032 self.assertEqual(headers['content-type'],
1033 self.objects[0]['meta']['content_type'])
1035 #assert response content
1036 self.assertEqual(self.objects[0]['data'], data)
1038 def test_get_with_multiple_if_match(self):
1039 #perform get with If-Match
1040 etags = [i['hash'] for i in self.objects if i]
1041 etags = ','.join('"%s"' % etag for etag in etags)
1042 status, headers, data = self.client.request_object(self.containers[1],
1043 self.objects[0]['name'],
1046 self.assertEqual(status, 200)
1048 #assert content-type
1049 self.assertEqual(headers['content-type'],
1050 self.objects[0]['meta']['content_type'])
1052 #assert content-type
1053 self.assertEqual(headers['content-type'],
1054 self.objects[0]['meta']['content_type'])
1056 #assert response content
1057 self.assertEqual(self.objects[0]['data'], data)
1059 def test_if_match_precondition_failed(self):
1060 #assert precondition failed
1061 self.assert_raises_fault(412, self.client.retrieve_object,
1063 self.objects[0]['name'], if_match='123')
1065 def test_if_none_match(self):
1066 #perform get with If-None-Match
1067 status, headers, data = self.client.request_object(self.containers[1],
1068 self.objects[0]['name'],
1069 if_none_match='123')
1072 self.assertEqual(status, 200)
1074 #assert content-type
1075 self.assertEqual(headers['content_type'],
1076 self.objects[0]['meta']['content_type'])
1078 def test_if_none_match(self):
1079 #perform get with If-None-Match * and assert not modified
1080 self.assert_raises_fault(304, self.client.retrieve_object,
1082 self.objects[0]['name'],
1085 def test_if_none_match_not_modified(self):
1086 #perform get with If-None-Match and assert not modified
1087 self.assert_raises_fault(304, self.client.retrieve_object,
1089 self.objects[0]['name'],
1090 if_none_match=self.objects[0]['hash'])
1092 meta = self.client.retrieve_object_metadata(self.containers[1],
1093 self.objects[0]['name'])
1094 self.assertEqual(meta['etag'], self.objects[0]['hash'])
1096 def test_if_modified_since(self):
1097 t = datetime.datetime.utcnow()
1098 t2 = t - datetime.timedelta(minutes=10)
1101 self.upload_data(self.containers[1],
1102 self.objects[0]['name'],
1103 self.objects[0]['data'][:200])
1105 for f in DATE_FORMATS:
1106 past = t2.strftime(f)
1108 headers = {'if-modified-since':'%s' %past}
1110 o = self.client.retrieve_object(self.containers[1],
1111 self.objects[0]['name'],
1112 if_modified_since=past)
1114 self.client.retrieve_object(self.containers[1],
1115 self.objects[0]['name']))
1117 self.failIf(f.status == 304)
1119 def test_if_modified_since_invalid_date(self):
1120 o = self.client.retrieve_object(self.containers[1],
1121 self.objects[0]['name'],
1122 if_modified_since='')
1123 self.assertEqual(o, self.client.retrieve_object(self.containers[1],
1124 self.objects[0]['name']))
1126 def test_if_not_modified_since(self):
1127 now = datetime.datetime.utcnow()
1128 since = now + datetime.timedelta(1)
1130 for f in DATE_FORMATS:
1131 #assert not modified
1132 self.assert_raises_fault(304, self.client.retrieve_object,
1133 self.containers[1], self.objects[0]['name'],
1134 if_modified_since=since.strftime(f))
1136 def test_if_unmodified_since(self):
1137 now = datetime.datetime.utcnow()
1138 since = now + datetime.timedelta(1)
1140 for f in DATE_FORMATS:
1141 t = since.strftime(f)
1142 status, headers, data = self.client.request_object(self.containers[1],
1143 self.objects[0]['name'],
1144 if_unmodified_since=t)
1146 self.assertEqual(status, 200)
1147 self.assertEqual(self.objects[0]['data'], data)
1149 #assert content-type
1150 self.assertEqual(headers['content-type'],
1151 self.objects[0]['meta']['content_type'])
1153 def test_if_unmodified_since_precondition_failed(self):
1154 t = datetime.datetime.utcnow()
1155 t2 = t - datetime.timedelta(minutes=10)
1158 self.upload_data(self.containers[1],
1159 self.objects[0]['name'],
1160 self.objects[0]['data'][:200])
1162 for f in DATE_FORMATS:
1163 past = t2.strftime(f)
1164 #assert precondition failed
1165 self.assert_raises_fault(412, self.client.retrieve_object,
1166 self.containers[1], self.objects[0]['name'],
1167 if_unmodified_since=past)
1169 def test_hashes(self):
1172 o = self.upload_random_data(self.containers[1], fname, l)
1174 body = self.client.retrieve_object(self.containers[1], fname,
1176 hashes = body['hashes']
1177 block_size = body['block_size']
1178 block_hash = body['block_hash']
1179 block_num = l/block_size == 0 and l/block_size or l/block_size + 1
1180 self.assertTrue(len(hashes), block_num)
1183 start = i * block_size
1184 end = (i + 1) * block_size
1185 hash = compute_block_hash(o['data'][start:end], block_hash)
1186 self.assertEqual(h, hash)
1189 class ObjectPut(BaseTestCase):
1191 BaseTestCase.setUp(self)
1192 self.account = 'test'
1193 self.container = 'c1'
1194 self.client.create_container(self.container)
1196 def test_upload(self):
1198 meta = {'test':'test1'}
1199 o = self.upload_random_data(self.container, name, **meta)
1201 headers = self.client.retrieve_object_metadata(self.container,
1204 self.assertTrue('test' in headers.keys())
1205 self.assertEqual(headers['test'], meta['test'])
1207 #assert uploaded content
1208 status, h, data = self.client.request_object(self.container, name)
1209 self.assertEqual(len(o['data']), int(h['content-length']))
1210 self.assertEqual(o['data'], data)
1212 #assert content-type
1213 self.assertEqual(h['content-type'], o['meta']['content_type'])
1215 def _test_maximum_upload_size_exceeds(self):
1217 meta = {'test':'test1'}
1219 length=1024*1024*100
1220 self.assert_raises_fault(400, self.upload_random_data, self.container,
1221 name, length, **meta)
1223 def test_upload_with_name_containing_slash(self):
1224 name = '/%s' % o_names[0]
1225 meta = {'test':'test1'}
1226 o = self.upload_random_data(self.container, name, **meta)
1228 self.assertEqual(o['data'],
1229 self.client.retrieve_object(self.container, name))
1231 self.assertTrue(name in self.client.list_objects(self.container))
1233 def test_create_directory_marker(self):
1234 self.client.create_directory_marker(self.container, 'foo')
1235 meta = self.client.retrieve_object_metadata(self.container, 'foo')
1236 self.assertEqual(meta['content-length'], '0')
1237 self.assertEqual(meta['content-type'], 'application/directory')
1239 def test_upload_unprocessable_entity(self):
1240 meta={'etag':'123', 'test':'test1'}
1242 #assert unprocessable entity
1243 self.assert_raises_fault(422, self.upload_random_data, self.container,
1246 def test_chunked_transfer(self):
1247 data = get_random_data()
1249 self.client.create_object_using_chunks(self.container, objname,
1252 uploaded_data = self.client.retrieve_object(self.container, objname)
1253 self.assertEqual(data, uploaded_data)
1255 def test_manifestation(self):
1256 prefix = 'myobject/'
1259 part = '%s%d' %(prefix, i)
1260 o = self.upload_random_data(self.container, part)
1263 manifest = '%s/%s' %(self.container, prefix)
1264 self.client.create_manifestation(self.container, 'large-object', manifest)
1266 self.assert_object_exists(self.container, 'large-object')
1267 self.assertEqual(data, self.client.retrieve_object(self.container,
1270 #wrong manifestation
1271 self.client.create_manifestation(self.container, 'large-object',
1272 '%s/invalid' % self.container)
1273 self.assertEqual('', self.client.retrieve_object(self.container,
1276 def test_create_zero_length_object(self):
1279 zero = self.client.create_zero_length_object(c, o)
1280 zero_meta = self.client.retrieve_object_metadata(c, o)
1281 zero_hash = self.client.retrieve_object_hashmap(c, o)
1282 zero_data = self.client.retrieve_object(c, o)
1284 self.assertEqual(int(zero_meta['content-length']), 0)
1285 self.assertEqual(zero_hash, [])
1286 self.assertEqual(zero_data, '')
1288 class ObjectCopy(BaseTestCase):
1290 BaseTestCase.setUp(self)
1291 self.account = 'test'
1292 self.containers = ['c1', 'c2']
1293 for c in self.containers:
1294 self.client.create_container(c)
1295 self.obj = self.upload_random_data(self.containers[0], o_names[0])
1300 def test_copy(self):
1301 with AssertMappingInvariant(self.client.retrieve_object_metadata,
1302 self.containers[0], self.obj['name']):
1304 meta = {'test':'testcopy'}
1305 status = self.client.copy_object(self.containers[0],
1311 #assert copy success
1312 self.assertEqual(status, 201)
1314 #assert access the new object
1315 headers = self.client.retrieve_object_metadata(self.containers[0],
1317 self.assertTrue('x-object-meta-test' in headers.keys())
1318 self.assertTrue(headers['x-object-meta-test'], 'testcopy')
1320 #assert etag is the same
1321 self.assertEqual(headers['etag'], self.obj['hash'])
1323 #assert src object still exists
1324 self.assert_object_exists(self.containers[0], self.obj['name'])
1326 def test_copy_from_different_container(self):
1327 with AssertMappingInvariant(self.client.retrieve_object_metadata,
1328 self.containers[0], self.obj['name']):
1329 meta = {'test':'testcopy'}
1330 status = self.client.copy_object(self.containers[0],
1335 self.assertEqual(status, 201)
1337 # assert updated metadata
1338 meta = self.client.retrieve_object_metadata(self.containers[1],
1341 self.assertTrue('test' in meta.keys())
1342 self.assertTrue(meta['test'], 'testcopy')
1344 #assert src object still exists
1345 self.assert_object_exists(self.containers[0], self.obj['name'])
1347 def test_copy_invalid(self):
1348 #copy from invalid object
1349 meta = {'test':'testcopy'}
1350 self.assert_raises_fault(404, self.client.copy_object, self.containers[0],
1351 'test.py', self.containers[1], 'testcopy', meta)
1353 #copy from invalid container
1354 meta = {'test':'testcopy'}
1355 self.assert_raises_fault(404, self.client.copy_object, self.containers[1],
1356 self.obj['name'], self.containers[1],
1359 class ObjectMove(BaseTestCase):
1361 BaseTestCase.setUp(self)
1362 self.account = 'test'
1363 self.containers = ['c1', 'c2']
1364 for c in self.containers:
1365 self.client.create_container(c)
1366 self.obj = self.upload_random_data(self.containers[0], o_names[0])
1368 def test_move(self):
1370 meta = {'test':'testcopy'}
1371 src_path = '/'.join(('/', self.containers[0], self.obj['name']))
1372 status = self.client.move_object(self.containers[0], self.obj['name'],
1373 self.containers[0], 'testcopy',
1376 #assert successful move
1377 self.assertEqual(status, 201)
1379 #assert updated metadata
1380 meta = self.client.retrieve_object_metadata(self.containers[0],
1383 self.assertTrue('test' in meta.keys())
1384 self.assertTrue(meta['test'], 'testcopy')
1386 #assert src object no more exists
1387 self.assert_object_not_exists(self.containers[0], self.obj['name'])
1389 class ObjectPost(BaseTestCase):
1391 BaseTestCase.setUp(self)
1392 self.account = 'test'
1393 self.containers = ['c1', 'c2']
1394 for c in self.containers:
1395 self.client.create_container(c)
1398 self.obj.append(self.upload_random_data(self.containers[0], o_names[i]))
1400 def test_update_meta(self):
1401 #perform update metadata
1402 more = {'foo':'foo', 'bar':'bar'}
1403 status = self.client.update_object_metadata(self.containers[0],
1404 self.obj[0]['name'],
1406 #assert request accepted
1407 self.assertEqual(status, 202)
1409 #assert old metadata are still there
1410 headers = self.client.retrieve_object_metadata(self.containers[0],
1411 self.obj[0]['name'],
1413 #assert new metadata have been updated
1414 for k,v in more.items():
1415 self.assertTrue(k in headers.keys())
1416 self.assertTrue(headers[k], v)
1418 def test_update_object(self,
1421 instance_length = True,
1422 content_length = 500):
1423 l = len(self.obj[0]['data'])
1424 range = 'bytes %d-%d/%s' %(first_byte_pos,
1426 l if instance_length else '*')
1427 partial = last_byte_pos - first_byte_pos + 1
1428 length = first_byte_pos + partial
1429 data = get_random_data(partial)
1430 args = {'content_type':'application/octet-stream',
1431 'content_range':'%s' %range}
1433 args['content_length'] = content_length
1435 status = self.client.update_object(self.containers[0], self.obj[0]['name'],
1436 StringIO(data), **args)[0]
1438 if partial < 0 or (instance_length and l <= last_byte_pos):
1439 self.assertEqual(status, 202)
1441 self.assertEqual(status, 204)
1442 #check modified object
1443 content = self.client.retrieve_object(self.containers[0],
1444 self.obj[0]['name'])
1445 self.assertEqual(content[:first_byte_pos], self.obj[0]['data'][:first_byte_pos])
1446 self.assertEqual(content[first_byte_pos:last_byte_pos+1], data)
1447 self.assertEqual(content[last_byte_pos+1:], self.obj[0]['data'][last_byte_pos+1:])
1449 def test_update_object_lt_blocksize(self):
1450 self.test_update_object(10, 20, content_length=None)
1452 def test_update_object_gt_blocksize(self):
1453 o = self.upload_random_data(self.containers[0], o_names[1],
1454 length=4*1024*1024+5)
1455 c = self.containers[0]
1458 first_byte_pos = 4*1024*1024+1
1459 last_byte_pos = 4*1024*1024+4
1460 l = last_byte_pos - first_byte_pos + 1
1461 data = get_random_data(l)
1462 range = 'bytes %d-%d/*' %(first_byte_pos, last_byte_pos)
1463 self.client.update_object(c, o_name, StringIO(data), content_range=range)
1464 content = self.client.retrieve_object(c, o_name)
1465 self.assertEqual(content[:first_byte_pos], o_data[:first_byte_pos])
1466 self.assertEqual(content[first_byte_pos:last_byte_pos+1], data)
1467 self.assertEqual(content[last_byte_pos+1:], o_data[last_byte_pos+1:])
1469 def test_update_object_divided_by_blocksize(self):
1470 o = self.upload_random_data(self.containers[0], o_names[1],
1471 length=4*1024*1024+5)
1472 c = self.containers[0]
1475 first_byte_pos = 4*1024*1024
1476 last_byte_pos = 5*1024*1024
1477 l = last_byte_pos - first_byte_pos + 1
1478 data = get_random_data(l)
1479 range = 'bytes %d-%d/*' %(first_byte_pos, last_byte_pos)
1480 self.client.update_object(c, o_name, StringIO(data), content_range=range)
1481 content = self.client.retrieve_object(c, o_name)
1482 self.assertEqual(content[:first_byte_pos], o_data[:first_byte_pos])
1483 self.assertEqual(content[first_byte_pos:last_byte_pos+1], data)
1484 self.assertEqual(content[last_byte_pos+1:], o_data[last_byte_pos+1:])
1486 def test_update_object_no_content_length(self):
1487 self.test_update_object(content_length = None)
1489 def test_update_object_invalid_content_length(self):
1490 with AssertContentInvariant(self.client.retrieve_object,
1491 self.containers[0], self.obj[0]['name']):
1492 self.assert_raises_fault(400, self.test_update_object,
1493 content_length = 1000)
1495 def test_update_object_invalid_range(self):
1496 with AssertContentInvariant(self.client.retrieve_object,
1497 self.containers[0], self.obj[0]['name']):
1498 self.assert_raises_fault(416, self.test_update_object, 499, 0, True)
1500 def test_update_object_invalid_range_and_length(self):
1501 with AssertContentInvariant(self.client.retrieve_object,
1502 self.containers[0], self.obj[0]['name']):
1503 self.assert_raises_fault(416, self.test_update_object, 499, 0, True,
1506 def test_update_object_invalid_range_with_no_content_length(self):
1507 with AssertContentInvariant(self.client.retrieve_object,
1508 self.containers[0], self.obj[0]['name']):
1509 self.assert_raises_fault(416, self.test_update_object, 499, 0, True,
1510 content_length = None)
1512 def test_update_object_out_of_limits(self):
1513 with AssertContentInvariant(self.client.retrieve_object,
1514 self.containers[0], self.obj[0]['name']):
1515 l = len(self.obj[0]['data'])
1516 self.assert_raises_fault(416, self.test_update_object, 0, l+1, True)
1518 def test_append(self):
1519 data = get_random_data(500)
1521 self.client.update_object(self.containers[0], self.obj[0]['name'],
1522 StringIO(data), content_length=500,
1523 content_type='application/octet-stream')
1525 content = self.client.retrieve_object(self.containers[0],
1526 self.obj[0]['name'])
1527 self.assertEqual(len(content), len(self.obj[0]['data']) + 500)
1528 self.assertEqual(content[:-500], self.obj[0]['data'])
1530 def test_update_with_chunked_transfer(self):
1531 data = get_random_data(500)
1533 fl = len(self.obj[0]['data'])
1535 self.client.update_object_using_chunks(self.containers[0],
1536 self.obj[0]['name'],
1539 content_type='application/octet-stream')
1541 #check modified object
1542 content = self.client.retrieve_object(self.containers[0],
1543 self.obj[0]['name'])
1544 self.assertEqual(content[0:dl], data)
1545 self.assertEqual(content[dl:fl], self.obj[0]['data'][dl:fl])
1547 def test_update_from_other_object(self):
1548 c = self.containers[0]
1552 source_data = self.client.retrieve_object(c, src)
1553 source_meta = self.client.retrieve_object_metadata(c, src)
1554 source_hash = self.client.retrieve_object_hashmap(c, src)
1556 #update zero length object
1557 self.client.create_zero_length_object(c, dest)
1558 source_object = '/%s/%s' % (c, src)
1559 self.client.update_from_other_source(c, dest, source_object)
1560 dest_data = self.client.retrieve_object(c, src)
1561 dest_meta = self.client.retrieve_object_metadata(c, dest)
1562 dest_hash = self.client.retrieve_object_hashmap(c, src)
1563 self.assertEqual(source_data, dest_data)
1564 self.assertEqual(source_hash, dest_hash)
1567 self.client.update_from_other_source(c, dest, source_object)
1568 content = self.client.retrieve_object(c, dest)
1569 self.assertEqual(source_data * 2, content)
1571 def test_update_range_from_other_object(self):
1572 c = self.containers[0]
1576 src = self.obj[1]['name']
1577 src_data = self.client.retrieve_object(c, src)
1579 #update zero length object
1580 prev_data = self.upload_random_data(c, dest, length=4*1024*1024+10)['data']
1581 source_object = '/%s/%s' % (c, src)
1582 first_byte_pos = 4*1024*1024+1
1583 last_byte_pos = 4*1024*1024+4
1584 range = 'bytes %d-%d/*' %(first_byte_pos, last_byte_pos)
1585 self.client.update_from_other_source(c, dest, source_object,
1586 content_range=range)
1587 content = self.client.retrieve_object(c, dest)
1588 self.assertEqual(content[:first_byte_pos], prev_data[:first_byte_pos])
1589 self.assertEqual(content[first_byte_pos:last_byte_pos+1], src_data[:last_byte_pos - first_byte_pos + 1])
1590 self.assertEqual(content[last_byte_pos+1:], prev_data[last_byte_pos+1:])
1592 def test_update_hashes_from_other_object(self):
1593 c = self.containers[0]
1597 src_data = self.upload_random_data(c, o_names[0], length=1024*1024+10)['data']
1599 #update zero length object
1600 prev_data = self.upload_random_data(c, dest, length=5*1024*1024+10)['data']
1601 source_object = '/%s/%s' % (c, o_names[0])
1602 first_byte_pos = 4*1024*1024
1603 last_byte_pos = 5*1024*1024
1604 range = 'bytes %d-%d/*' %(first_byte_pos, last_byte_pos)
1605 self.client.update_from_other_source(c, dest, source_object,
1606 content_range=range)
1607 content = self.client.retrieve_object(c, dest)
1608 self.assertEqual(content[:first_byte_pos], prev_data[:first_byte_pos])
1609 self.assertEqual(content[first_byte_pos:last_byte_pos+1], src_data[:last_byte_pos - first_byte_pos + 1])
1610 self.assertEqual(content[last_byte_pos+1:], prev_data[last_byte_pos+1:])
1613 def test_update_zero_length_object(self):
1614 c = self.containers[0]
1617 zero = self.client.create_zero_length_object(c, o)
1619 data = get_random_data()
1620 self.client.update_object(c, o, StringIO(data))
1621 self.client.create_object(c, other, StringIO(data))
1623 self.assertEqual(self.client.retrieve_object(c, o),
1624 self.client.retrieve_object(c, other))
1626 self.assertEqual(self.client.retrieve_object_hashmap(c, o),
1627 self.client.retrieve_object_hashmap(c, other))
1629 class ObjectDelete(BaseTestCase):
1631 BaseTestCase.setUp(self)
1632 self.account = 'test'
1633 self.containers = ['c1', 'c2']
1634 for c in self.containers:
1635 self.client.create_container(c)
1636 self.obj = self.upload_random_data(self.containers[0], o_names[0])
1638 def test_delete(self):
1639 #perform delete object
1640 self.client.delete_object(self.containers[0], self.obj['name'])[0]
1642 def test_delete_invalid(self):
1643 #assert item not found
1644 self.assert_raises_fault(404, self.client.delete_object, self.containers[1],
1647 class ListSharing(BaseTestCase):
1649 BaseTestCase.setUp(self)
1650 self.client.create_container('c')
1652 self.upload_random_data('c', 'o%s' %i)
1653 accounts = OTHER_ACCOUNTS.copy()
1654 self.o1_sharing_with = accounts.popitem()
1655 self.o1_sharing = [self.o1_sharing_with[1]]
1656 self.client.share_object('c', 'o1', self.o1_sharing, read=True)
1660 l.append(accounts.popitem())
1661 #self.client.set_account_groups({'pithos-dev':'chazapis,verigak,papagian'})
1662 #self.o2_sharing = 'write=%s' %
1663 #self.client.share_object('c', 'o2', self.o2_sharing)
1665 def test_listing(self):
1666 self.other = Pithos_Client(get_server(),
1667 self.o1_sharing_with[0],
1668 self.o1_sharing_with[1],
1670 self.assertTrue('test' in self.other.list_shared_by_others())
1672 class TestGreek(BaseTestCase):
1674 BaseTestCase.setUp(self)
1675 #keep track of initial account groups
1676 self.initial_groups = self.client.retrieve_account_groups()
1678 #keep track of initial account meta
1679 self.initial_meta = self.client.retrieve_account_metadata(restricted=True)
1682 #delete additionally created meta
1684 for m in self.client.retrieve_account_metadata(restricted=True):
1685 if m not in self.initial_meta:
1687 self.client.delete_account_metadata(l)
1689 #delete additionally created groups
1691 for g in self.client.retrieve_account_groups():
1692 if g not in self.initial_groups:
1694 self.client.unset_account_groups(l)
1696 BaseTestCase.tearDown(self)
1698 def test_create_container(self):
1699 self.client.create_container('φάκελος')
1700 self.assert_container_exists('φάκελος')
1702 self.assertTrue('φάκελος' in self.client.list_containers())
1704 def test_create_object(self):
1705 self.client.create_container('φάκελος')
1706 self.upload_random_data('φάκελος', 'αντικείμενο')
1708 self.assert_object_exists('φάκελος', 'αντικείμενο')
1709 self.assertTrue('αντικείμενο' in self.client.list_objects('φάκελος'))
1711 def test_copy_object(self):
1712 src_container = 'φάκελος'
1713 src_object = 'αντικείμενο'
1714 dest_container = 'αντίγραφα'
1715 dest_object = 'ασφαλές-αντίγραφο'
1717 self.client.create_container(src_container)
1718 self.upload_random_data(src_container, src_object)
1720 self.client.create_container(dest_container)
1721 self.client.copy_object(src_container, src_object, dest_container,
1724 self.assert_object_exists(src_container, src_object)
1725 self.assert_object_exists(dest_container, dest_object)
1726 self.assertTrue(dest_object in self.client.list_objects(dest_container))
1728 def test_move_object(self):
1729 src_container = 'φάκελος'
1730 src_object = 'αντικείμενο'
1731 dest_container = 'αντίγραφα'
1732 dest_object = 'ασφαλές-αντίγραφο'
1734 self.client.create_container(src_container)
1735 self.upload_random_data(src_container, src_object)
1737 self.client.create_container(dest_container)
1738 self.client.move_object(src_container, src_object, dest_container,
1741 self.assert_object_not_exists(src_container, src_object)
1742 self.assert_object_exists(dest_container, dest_object)
1743 self.assertTrue(dest_object in self.client.list_objects(dest_container))
1745 def test_delete_object(self):
1746 self.client.create_container('φάκελος')
1747 self.upload_random_data('φάκελος', 'αντικείμενο')
1748 self.assert_object_exists('φάκελος', 'αντικείμενο')
1750 self.client.delete_object('φάκελος', 'αντικείμενο')
1751 self.assert_object_not_exists('φάκελος', 'αντικείμενο')
1752 self.assertTrue('αντικείμενο' not in self.client.list_objects('φάκελος'))
1754 def test_delete_container(self):
1755 self.client.create_container('φάκελος')
1756 self.assert_container_exists('φάκελος')
1758 self.client.delete_container('φάκελος')
1759 self.assert_container_not_exists('φάκελος')
1760 self.assertTrue('φάκελος' not in self.client.list_containers())
1762 def test_account_meta(self):
1763 meta = {'ποιότητα':'ΑΑΑ'}
1764 self.client.update_account_metadata(**meta)
1765 meta = self.client.retrieve_account_metadata(restricted=True)
1766 self.assertTrue('ποιότητα' in meta.keys())
1767 self.assertEqual(meta['ποιότητα'], 'ΑΑΑ')
1769 def test_container_meta(self):
1770 meta = {'ποιότητα':'ΑΑΑ'}
1771 self.client.create_container('φάκελος', **meta)
1773 meta = self.client.retrieve_container_metadata('φάκελος', restricted=True)
1774 self.assertTrue('ποιότητα' in meta.keys())
1775 self.assertEqual(meta['ποιότητα'], 'ΑΑΑ')
1777 def test_object_meta(self):
1778 self.client.create_container('φάκελος')
1779 meta = {'ποιότητα':'ΑΑΑ'}
1780 self.upload_random_data('φάκελος', 'αντικείμενο', **meta)
1782 meta = self.client.retrieve_object_metadata('φάκελος', 'αντικείμενο',
1784 self.assertTrue('ποιότητα' in meta.keys())
1785 self.assertEqual(meta['ποιότητα'], 'ΑΑΑ')
1787 def test_list_meta_filtering(self):
1788 self.client.create_container('φάκελος')
1789 meta = {'ποιότητα':'ΑΑΑ'}
1790 self.upload_random_data('φάκελος', 'ο1', **meta)
1791 self.upload_random_data('φάκελος', 'ο2')
1792 self.upload_random_data('φάκελος', 'ο3')
1794 meta = {'ποσότητα':'μεγάλη'}
1795 self.client.update_object_metadata('φάκελος', 'ο2', **meta)
1796 objects = self.client.list_objects('φάκελος', meta='ποιότητα, ποσότητα')
1797 self.assertTrue('ο1' in objects)
1798 self.assertTrue('ο2' in objects)
1799 self.assertTrue('ο3' not in objects)
1801 def test_groups(self):
1803 groups = {'σεφς':'chazapis,διογένης'}
1804 self.client.set_account_groups(**groups)
1805 groups.update(self.initial_groups)
1806 self.assertEqual(groups['σεφς'],
1807 self.client.retrieve_account_groups()['σεφς'])
1810 self.client.create_container('φάκελος')
1811 o = self.upload_random_data('φάκελος', 'ο1')
1812 self.client.share_object('φάκελος', 'ο1', ['test:σεφς'])
1813 chef = Pithos_Client(get_server(),
1817 self.assert_not_raises_fault(401, chef.retrieve_object_metadata,
1818 'φάκελος', 'ο1', account=get_user())
1821 self.client.share_object('φάκελος', 'ο1', ['διογένης'], read=False)
1822 new_data = get_random_data()
1823 self.assert_not_raises_fault(401, chef.update_object,
1824 'φάκελος', 'ο1', StringIO(new_data),
1827 server_data = self.client.retrieve_object('φάκελος', 'ο1')
1828 self.assertEqual(server_data[:len(o['data'])], o['data'])
1829 self.assertEqual(server_data[len(o['data']):], new_data)
1831 def test_manifestation(self):
1832 self.client.create_container('κουβάς')
1836 part = '%s%d' %(prefix, i)
1837 o = self.upload_random_data('κουβάς', part)
1840 self.client.create_container('φάκελος')
1841 manifest = '%s/%s' %('κουβάς', prefix)
1842 self.client.create_manifestation('φάκελος', 'άπαντα', manifest)
1844 self.assert_object_exists('φάκελος', 'άπαντα')
1845 self.assertEqual(data, self.client.retrieve_object('φάκελος',
1848 #wrong manifestation
1849 self.client.create_manifestation('φάκελος', 'άπαντα', 'κουβάς/άκυρο')
1850 self.assertEqual('', self.client.retrieve_object('φάκελος', 'άπαντα'))
1852 def test_update_from_another_object(self):
1853 self.client.create_container('κουβάς')
1854 src_data = self.upload_random_data('κουβάς', 'πηγή')['data']
1855 initial_data = self.upload_random_data('κουβάς', 'νέο')['data']
1856 source_object = '/%s/%s' % ('κουβάς', 'πηγή')
1857 self.client.update_from_other_source('κουβάς', 'νέο', source_object)
1860 self.client.retrieve_object('κουβάς', 'νέο'),
1861 '%s%s' % (initial_data, self.client.retrieve_object('κουβάς', 'πηγή')))
1863 class TestPermissions(BaseTestCase):
1865 BaseTestCase.setUp(self)
1866 #keep track of initial account groups
1867 self.initial_groups = self.client.retrieve_account_groups()
1868 #keep track of initial account meta
1869 self.initial_meta = self.client.retrieve_account_metadata(restricted=True)
1872 self.authorized = ['chazapis', 'verigak', 'gtsouk', 'papagian']
1873 groups = {'pithosdev':','.join(self.authorized)}
1874 self.client.set_account_groups(**groups)
1877 #delete additionally created meta
1879 for m in self.client.retrieve_account_metadata(restricted=True):
1880 if m not in self.initial_meta:
1882 self.client.delete_account_metadata(l)
1884 #delete additionally created groups
1886 for g in self.client.retrieve_account_groups():
1887 if g not in self.initial_groups:
1889 self.client.unset_account_groups(l)
1891 BaseTestCase.tearDown(self)
1893 def test_read_access(self):
1894 self.client.create_container('c')
1895 o = self.upload_random_data('c', 'o')
1896 self.client.share_object('c', 'o', ['test:pithosdev'])
1897 for token, account in OTHER_ACCOUNTS.items():
1898 cl = Pithos_Client(get_server(), token, account, get_api())
1899 if account in self.authorized:
1900 self.assert_not_raises_fault(401, cl.retrieve_object_metadata,
1901 'c', 'o', account=get_user())
1903 self.assert_raises_fault(401, cl.retrieve_object_metadata,
1904 'c', 'o', account=get_user())
1907 def test_write_access(self):
1908 self.client.create_container('c')
1909 o = self.upload_random_data('c', 'o')
1910 self.client.share_object('c', 'o', ['chazapis'], read=False)
1911 for token, account in OTHER_ACCOUNTS.items():
1912 cl = Pithos_Client(get_server(), token, account, get_api())
1913 new_data = get_random_data()
1914 if account == 'chazapis':
1915 self.assert_not_raises_fault(401, cl.update_object,
1916 'c', 'o', StringIO(new_data),
1918 server_data = self.client.retrieve_object('c', 'o')
1919 self.assertEqual(o['data'], server_data[:len(o['data'])])
1920 self.assertEqual(new_data, server_data[len(o['data']):])
1922 self.assert_raises_fault(401, cl.update_object,
1923 'c', 'o', StringIO(new_data),
1926 class AssertMappingInvariant(object):
1927 def __init__(self, callable, *args, **kwargs):
1928 self.callable = callable
1930 self.kwargs = kwargs
1932 def __enter__(self):
1933 self.map = self.callable(*self.args, **self.kwargs)
1936 def __exit__(self, type, value, tb):
1937 map = self.callable(*self.args, **self.kwargs)
1938 for k in self.map.keys():
1939 if is_date(self.map[k]):
1941 assert map[k] == self.map[k]
1943 class AssertContentInvariant(object):
1944 def __init__(self, callable, *args, **kwargs):
1945 self.callable = callable
1947 self.kwargs = kwargs
1949 def __enter__(self):
1950 self.content = self.callable(*self.args, **self.kwargs)[2]
1953 def __exit__(self, type, value, tb):
1954 content = self.callable(*self.args, **self.kwargs)[2]
1955 assert self.content == content
1957 def get_content_splitted(response):
1959 return response.content.split('\n')
1961 def compute_md5_hash(data):
1965 return md5.hexdigest().lower()
1967 def compute_block_hash(data, algorithm):
1968 h = hashlib.new(algorithm)
1969 h.update(data.rstrip('\x00'))
1970 return h.hexdigest()
1972 def get_random_data(length=500):
1973 char_set = string.ascii_uppercase + string.digits
1974 return ''.join(random.choice(char_set) for x in range(length))
1977 MONTHS = 'jan feb mar apr may jun jul aug sep oct nov dec'.split()
1978 __D = r'(?P<day>\d{2})'
1979 __D2 = r'(?P<day>[ \d]\d)'
1980 __M = r'(?P<mon>\w{3})'
1981 __Y = r'(?P<year>\d{4})'
1982 __Y2 = r'(?P<year>\d{2})'
1983 __T = r'(?P<hour>\d{2}):(?P<min>\d{2}):(?P<sec>\d{2})'
1984 RFC1123_DATE = re.compile(r'^\w{3}, %s %s %s %s GMT$' % (__D, __M, __Y, __T))
1985 RFC850_DATE = re.compile(r'^\w{6,9}, %s-%s-%s %s GMT$' % (__D, __M, __Y2, __T))
1986 ASCTIME_DATE = re.compile(r'^\w{3} %s %s %s %s$' % (__M, __D2, __T, __Y))
1987 for regex in RFC1123_DATE, RFC850_DATE, ASCTIME_DATE:
1988 m = regex.match(date)
1993 o_names = ['kate.jpg',
1994 'kate_beckinsale.jpg',
1995 'How To Win Friends And Influence People.pdf',
1996 'moms_birthday.jpg',
1998 'Disturbed - Down With The Sickness.mp3',
1999 'army_of_darkness.avi',
2001 'photos/animals/dogs/poodle.jpg',
2002 'photos/animals/dogs/terrier.jpg',
2003 'photos/animals/cats/persian.jpg',
2004 'photos/animals/cats/siamese.jpg',
2005 'photos/plants/fern.jpg',
2006 'photos/plants/rose.jpg',
2009 if __name__ == "__main__":