root / pithos / api / tests.py @ 23028d31
History | View | Annotate | Download (59.8 kB)
1 |
# Copyright 2011 GRNET S.A. All rights reserved.
|
---|---|
2 |
#
|
3 |
# Redistribution and use in source and binary forms, with or
|
4 |
# without modification, are permitted provided that the following
|
5 |
# conditions are met:
|
6 |
#
|
7 |
# 1. Redistributions of source code must retain the above
|
8 |
# copyright notice, this list of conditions and the following
|
9 |
# disclaimer.
|
10 |
#
|
11 |
# 2. Redistributions in binary form must reproduce the above
|
12 |
# copyright notice, this list of conditions and the following
|
13 |
# disclaimer in the documentation and/or other materials
|
14 |
# provided with the distribution.
|
15 |
#
|
16 |
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
|
17 |
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
18 |
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
19 |
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
|
20 |
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
21 |
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
22 |
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
|
23 |
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
|
24 |
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
25 |
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
|
26 |
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
27 |
# POSSIBILITY OF SUCH DAMAGE.
|
28 |
#
|
29 |
# The views and conclusions contained in the software and
|
30 |
# documentation are those of the authors and should not be
|
31 |
# interpreted as representing official policies, either expressed
|
32 |
# or implied, of GRNET S.A.
|
33 |
|
34 |
from pithos.lib.client import Client, Fault |
35 |
import unittest |
36 |
from django.utils import simplejson as json |
37 |
from xml.dom import minidom |
38 |
import types |
39 |
import hashlib |
40 |
import os |
41 |
import mimetypes |
42 |
import random |
43 |
import datetime |
44 |
import string |
45 |
import re |
46 |
|
47 |
#from pithos.backends import backend
|
48 |
|
49 |
DATE_FORMATS = ["%a %b %d %H:%M:%S %Y",
|
50 |
"%A, %d-%b-%y %H:%M:%S GMT",
|
51 |
"%a, %d %b %Y %H:%M:%S GMT"]
|
52 |
|
53 |
DEFAULT_HOST = 'pithos.dev.grnet.gr'
|
54 |
#DEFAULT_HOST = '127.0.0.1:8000'
|
55 |
DEFAULT_API = 'v1'
|
56 |
DEFAULT_USER = 'papagian'
|
57 |
DEFAULT_AUTH = '0004'
|
58 |
|
59 |
class BaseTestCase(unittest.TestCase): |
60 |
#TODO unauthorized request
|
61 |
def setUp(self): |
62 |
self.client = Client(DEFAULT_HOST, DEFAULT_AUTH, DEFAULT_USER, DEFAULT_API)
|
63 |
self.invalid_client = Client(DEFAULT_HOST, DEFAULT_AUTH, 'non-existing', DEFAULT_API) |
64 |
self.unauthorised_client = Client(DEFAULT_HOST, '', DEFAULT_USER, DEFAULT_API) |
65 |
self.headers = {
|
66 |
'account': ('x-account-container-count', |
67 |
'x-account-bytes-used',
|
68 |
'last-modified',
|
69 |
'content-length',
|
70 |
'date',
|
71 |
'content-type',
|
72 |
'server',),
|
73 |
'object': ('etag', |
74 |
'content-length',
|
75 |
'content-type',
|
76 |
'content-encoding',
|
77 |
'last-modified',
|
78 |
'date',
|
79 |
'x-object-manifest',
|
80 |
'content-range',
|
81 |
'x-object-modified-by',
|
82 |
'x-object-version',
|
83 |
'x-object-version-timestamp',
|
84 |
'server',),
|
85 |
'container': ('x-container-object-count', |
86 |
'x-container-bytes-used',
|
87 |
'content-type',
|
88 |
'last-modified',
|
89 |
'content-length',
|
90 |
'date',
|
91 |
'x-container-block-size',
|
92 |
'x-container-block-hash',
|
93 |
'x-container-policy-quota',
|
94 |
'x-container-policy-versioning',
|
95 |
'server',
|
96 |
'x-container-object-meta',
|
97 |
'x-container-policy-versioning',
|
98 |
'server',)}
|
99 |
|
100 |
self.contentTypes = {'xml':'application/xml', |
101 |
'json':'application/json', |
102 |
'':'text/plain'} |
103 |
self.extended = {
|
104 |
'container':(
|
105 |
'name',
|
106 |
'count',
|
107 |
'bytes',
|
108 |
'last_modified'),
|
109 |
'object':(
|
110 |
'name',
|
111 |
'hash',
|
112 |
'bytes',
|
113 |
'content_type',
|
114 |
'content_encoding',
|
115 |
'last_modified',)}
|
116 |
self.return_codes = (400, 401, 404, 503,) |
117 |
|
118 |
def assert_status(self, status, codes): |
119 |
l = [elem for elem in self.return_codes] |
120 |
if type(codes) == types.ListType: |
121 |
l.extend(codes) |
122 |
else:
|
123 |
l.append(codes) |
124 |
self.assertTrue(status in l) |
125 |
|
126 |
def assert_list(self, path, entity, limit=10000, format='text', params=None, **headers): |
127 |
status, headers, data = self.client.get(path, format=format,
|
128 |
headers=headers, params=params) |
129 |
|
130 |
self.assert_status(status, [200, 204, 304, 412]) |
131 |
if format == 'text': |
132 |
data = data.strip().split('\n') if data else [] |
133 |
self.assertTrue(len(data) <= limit) |
134 |
else:
|
135 |
exp_content_type = self.contentTypes[format]
|
136 |
self.assertEqual(headers['content-type'].find(exp_content_type), 0) |
137 |
#self.assert_extended(data, format, entity, limit)
|
138 |
if format == 'json': |
139 |
data = json.loads(data) if data else [] |
140 |
elif format == 'xml': |
141 |
data = minidom.parseString(data) |
142 |
return status, headers, data
|
143 |
|
144 |
def list_containers(self, limit=10000, marker='', format='text', **headers): |
145 |
params = locals()
|
146 |
params.pop('self')
|
147 |
return self.assert_list('', 'account', limit, format, params, **headers) |
148 |
|
149 |
def list_objects(self, container, limit=10000, marker='', |
150 |
prefix='', format='', path='', delimiter='', meta='', |
151 |
**headers): |
152 |
params = locals()
|
153 |
params.pop('self')
|
154 |
params.pop('container')
|
155 |
path = '/' + container
|
156 |
format = 'text' if format == '' else format |
157 |
return self.assert_list(path, 'container', limit, format, params, **headers) |
158 |
|
159 |
def _assert_get_meta(self, path, entity, params=None, **exp_meta): |
160 |
status, headers, data = self.client.head(path, params)
|
161 |
self.assert_status(status, 204) |
162 |
#self.assert_headers(headers, entity, **exp_meta)
|
163 |
return status, headers, data
|
164 |
|
165 |
def get_account_meta(self, params=None, **exp_meta): |
166 |
return self._assert_get_meta('', 'account', params, **exp_meta) |
167 |
|
168 |
def get_container_meta(self, container, params=None, **exp_meta): |
169 |
path = '/%s' % container
|
170 |
return self._assert_get_meta(path, 'container', params, **exp_meta) |
171 |
|
172 |
def create_container(self, name, **meta): |
173 |
headers = {} |
174 |
for k,v in meta.items(): |
175 |
headers['x-container-meta-%s' %k.strip().upper()] = v.strip()
|
176 |
status, header, data = self.client.put('/' + name, headers=headers) |
177 |
self.assert_status(status, [201, 202]) |
178 |
return status, header, data
|
179 |
|
180 |
def get_object(self, container, name, format='', version=None, **headers): |
181 |
path = '/%s/%s' % (container, name)
|
182 |
params = {'version':version} if version else None |
183 |
status, headers, data = self.client.get(path, format, headers, params)
|
184 |
self.assert_status(status, [200, 206, 304, 412, 416]) |
185 |
#if status in [200, 206]:
|
186 |
# self.assert_headers(headers, 'object')
|
187 |
return status, headers, data
|
188 |
|
189 |
def update_object(self, container, name, data='', content_type='', **headers): |
190 |
if content_type != '': |
191 |
headers['content-type'] = content_type
|
192 |
status, headers, data = self.client.update_object_data(container,
|
193 |
name, |
194 |
data, |
195 |
headers) |
196 |
self.assert_status(status, [202, 204, 416]) |
197 |
return status, headers, data
|
198 |
|
199 |
def assert_headers(self, headers, type, **exp_meta): |
200 |
prefix = 'x-%s-meta-' %type |
201 |
system_headers = [h for h in headers if not h.startswith(prefix)] |
202 |
for k,v in headers.items(): |
203 |
if k in system_headers: |
204 |
self.assertTrue(k in headers[type]) |
205 |
elif exp_meta:
|
206 |
k = k.split(prefix)[-1]
|
207 |
self.assertEqual(v, exp_meta[k])
|
208 |
|
209 |
#def assert_extended(self, data, format, type, size):
|
210 |
# if format == 'xml':
|
211 |
# self._assert_xml(data, type, size)
|
212 |
# elif format == 'json':
|
213 |
# self._assert_json(data, type, size)
|
214 |
#
|
215 |
#def _assert_json(self, data, type, size):
|
216 |
# print '#', data
|
217 |
# convert = lambda s: s.lower()
|
218 |
# info = [convert(elem) for elem in self.extended[type]]
|
219 |
# data = json.loads(data)
|
220 |
# self.assertTrue(len(data) <= size)
|
221 |
# for item in info:
|
222 |
# for i in data:
|
223 |
# if 'subdir' in i.keys():
|
224 |
# continue
|
225 |
# self.assertTrue(item in i.keys())
|
226 |
#
|
227 |
#def _assert_xml(self, data, type, size):
|
228 |
# print '#', data
|
229 |
# convert = lambda s: s.lower()
|
230 |
# info = [convert(elem) for elem in self.extended[type]]
|
231 |
# try:
|
232 |
# info.remove('content_encoding')
|
233 |
# except ValueError:
|
234 |
# pass
|
235 |
# xml = minidom.parseString(data)
|
236 |
# entities = xml.getElementsByTagName(type)
|
237 |
# self.assertTrue(len(entities) <= size)
|
238 |
# for e in entities:
|
239 |
# for item in info:
|
240 |
# self.assertTrue(e.hasAttribute(item))
|
241 |
|
242 |
def assert_raises_fault(self, status, callableObj, *args, **kwargs): |
243 |
"""
|
244 |
asserts that a Fault with a specific status is raised
|
245 |
when callableObj is called with the specific arguments
|
246 |
"""
|
247 |
try:
|
248 |
callableObj(*args, **kwargs) |
249 |
self.fail('Should never reach here') |
250 |
except Fault, f:
|
251 |
self.failUnless(f.status == status)
|
252 |
|
253 |
def assert_object_exists(self, container, object): |
254 |
"""
|
255 |
asserts the existence of an object
|
256 |
"""
|
257 |
try:
|
258 |
self.client.retrieve_object_metadata(container, object) |
259 |
except Fault, f:
|
260 |
self.failIf(f.status == 404) |
261 |
|
262 |
def assert_object_not_exists(self, container, object): |
263 |
"""
|
264 |
asserts there is no such an object
|
265 |
"""
|
266 |
self.assert_raises_fault(404, self.client.retrieve_object_metadata, |
267 |
container, object)
|
268 |
|
269 |
def upload_random_data(self, container, name, length=1024, type=None, |
270 |
enc=None, **meta):
|
271 |
data = get_random_data(length) |
272 |
return self.upload_data(container, name, data, type, enc, **meta) |
273 |
|
274 |
def upload_data(self, container, name, data, type=None, enc=None, etag=None, |
275 |
**meta): |
276 |
obj = {} |
277 |
obj['name'] = name
|
278 |
try:
|
279 |
obj['data'] = data
|
280 |
obj['hash'] = compute_md5_hash(obj['data']) |
281 |
|
282 |
headers = {} |
283 |
for k,v in meta.items(): |
284 |
key = 'x-object-meta-%s' % k
|
285 |
headers[key] = v |
286 |
headers['etag'] = etag if etag else obj['hash'] |
287 |
guess = mimetypes.guess_type(name) |
288 |
type = type if type else guess[0] |
289 |
enc = enc if enc else guess[1] |
290 |
headers['content-type'] = type if type else 'plain/text' |
291 |
headers['content-encoding'] = enc if enc else None |
292 |
obj['meta'] = headers
|
293 |
|
294 |
path = '/%s/%s' % (container, name)
|
295 |
status, headers, data = self.client.put(path, obj['data'], |
296 |
headers=headers) |
297 |
if status == 201: |
298 |
self.assertTrue('etag' in headers) |
299 |
self.assertEqual(obj['hash'], headers['etag']) |
300 |
return obj
|
301 |
except IOError: |
302 |
return
|
303 |
|
304 |
class AccountHead(BaseTestCase): |
305 |
def setUp(self): |
306 |
BaseTestCase.setUp(self)
|
307 |
self.account = 'test' |
308 |
self.containers = ['apples', 'bananas', 'kiwis', 'oranges', 'pears'] |
309 |
for item in self.containers: |
310 |
self.create_container(item)
|
311 |
|
312 |
def tearDown(self): |
313 |
for c in self.list_containers()[2]: |
314 |
self.client.delete_container(c)
|
315 |
|
316 |
def test_get_account_meta(self): |
317 |
headers = self.get_account_meta()[1] |
318 |
|
319 |
containers = self.list_containers()[2] |
320 |
l = str(len(containers)) |
321 |
self.assertEqual(headers['x-account-container-count'], l) |
322 |
size = 0
|
323 |
for c in containers: |
324 |
h = self.get_container_meta(c)[1] |
325 |
size = size + int(h['x-container-bytes-used']) |
326 |
self.assertEqual(headers['x-account-bytes-used'], str(size)) |
327 |
|
328 |
#def test_get_account_401(self):
|
329 |
# response = self.get_account_meta('non-existing-account')
|
330 |
# print response
|
331 |
# self.assertEqual(response.status_code, 401)
|
332 |
|
333 |
class AccountGet(BaseTestCase): |
334 |
def setUp(self): |
335 |
BaseTestCase.setUp(self)
|
336 |
self.account = 'test' |
337 |
#create some containers
|
338 |
self.containers = ['apples', 'bananas', 'kiwis', 'oranges', 'pears'] |
339 |
for item in self.containers: |
340 |
self.create_container(item)
|
341 |
|
342 |
def tearDown(self): |
343 |
for c in self.list_containers()[2]: |
344 |
for o in self.list_objects(c)[2]: |
345 |
self.client.delete_object(c, o)
|
346 |
self.client.delete_container(c)
|
347 |
|
348 |
def test_list(self): |
349 |
#list containers
|
350 |
containers = self.list_containers()[2] |
351 |
self.assertEquals(self.containers, containers) |
352 |
|
353 |
#def test_list_204(self):
|
354 |
# response = self.list_containers('non-existing-account')
|
355 |
# self.assertEqual(response.status_code, 204)
|
356 |
|
357 |
def test_list_with_limit(self): |
358 |
limit = 2
|
359 |
containers = self.list_containers(limit=limit)[2] |
360 |
self.assertEquals(len(containers), limit) |
361 |
self.assertEquals(self.containers[:2], containers) |
362 |
|
363 |
def test_list_with_marker(self): |
364 |
l = 2
|
365 |
m = 'bananas'
|
366 |
containers = self.list_containers(limit=l, marker=m)[2] |
367 |
i = self.containers.index(m) + 1 |
368 |
self.assertEquals(self.containers[i:(i+l)], containers) |
369 |
|
370 |
m = 'oranges'
|
371 |
containers = self.list_containers(limit=l, marker=m)[2] |
372 |
i = self.containers.index(m) + 1 |
373 |
self.assertEquals(self.containers[i:(i+l)], containers) |
374 |
|
375 |
#def test_extended_list(self):
|
376 |
# self.list_containers(self.account, limit=3, format='xml')
|
377 |
# self.list_containers(self.account, limit=3, format='json')
|
378 |
|
379 |
def test_list_json_with_marker(self): |
380 |
l = 2
|
381 |
m = 'bananas'
|
382 |
status, headers, containers = self.list_containers(limit=l, marker=m,
|
383 |
format='json')
|
384 |
self.assertEqual(containers[0]['name'], 'kiwis') |
385 |
self.assertEqual(containers[1]['name'], 'oranges') |
386 |
|
387 |
def test_list_xml_with_marker(self): |
388 |
l = 2
|
389 |
m = 'oranges'
|
390 |
status, headers, xml = self.list_containers(limit=l, marker=m,
|
391 |
format='xml')
|
392 |
nodes = xml.getElementsByTagName('name')
|
393 |
self.assertEqual(len(nodes), 1) |
394 |
self.assertEqual(nodes[0].childNodes[0].data, 'pears') |
395 |
|
396 |
def test_if_modified_since(self): |
397 |
t = datetime.datetime.utcnow() |
398 |
t2 = t - datetime.timedelta(minutes=10)
|
399 |
|
400 |
#add a new container
|
401 |
self.create_container('dummy') |
402 |
|
403 |
for f in DATE_FORMATS: |
404 |
past = t2.strftime(f) |
405 |
|
406 |
headers = {'if-modified-since':'%s' %past} |
407 |
status, headers, data = self.list_containers(**headers)
|
408 |
|
409 |
#assert get success
|
410 |
self.assertEqual(status, 200) |
411 |
|
412 |
def test_if_modified_since_invalid_date(self): |
413 |
headers = {'if-modified-since':''} |
414 |
status, headers, data = self.list_containers(**headers)
|
415 |
|
416 |
#assert get success
|
417 |
self.assertEqual(status, 200) |
418 |
|
419 |
def test_if_not_modified_since(self): |
420 |
now = datetime.datetime.utcnow() |
421 |
since = now + datetime.timedelta(1)
|
422 |
|
423 |
for f in DATE_FORMATS: |
424 |
headers = {'if-modified-since':'%s' %since.strftime(f)} |
425 |
#assert not modified
|
426 |
self.assert_raises_fault(304, self.list_containers, **headers) |
427 |
|
428 |
def test_if_unmodified_since(self): |
429 |
now = datetime.datetime.utcnow() |
430 |
since = now + datetime.timedelta(1)
|
431 |
|
432 |
for f in DATE_FORMATS: |
433 |
headers = {'if-unmodified-since':'%s' %since.strftime(f)} |
434 |
status, headers, data = self.list_containers(**headers)
|
435 |
|
436 |
#assert success
|
437 |
self.assertEqual(status, 200) |
438 |
self.assertEqual(self.containers, data) |
439 |
|
440 |
def test_if_unmodified_since_precondition_failed(self): |
441 |
t = datetime.datetime.utcnow() |
442 |
t2 = t - datetime.timedelta(minutes=10)
|
443 |
|
444 |
#add a new container
|
445 |
self.create_container('dummy') |
446 |
|
447 |
for f in DATE_FORMATS: |
448 |
past = t2.strftime(f) |
449 |
|
450 |
headers = {'if-unmodified-since':'%s' %past} |
451 |
#assert precondition failed
|
452 |
self.assert_raises_fault(412, self.list_containers, **headers) |
453 |
|
454 |
class AccountPost(BaseTestCase): |
455 |
def setUp(self): |
456 |
BaseTestCase.setUp(self)
|
457 |
self.account = 'test' |
458 |
self.containers = ['apples', 'bananas', 'kiwis', 'oranges', 'pears'] |
459 |
for item in self.containers: |
460 |
self.create_container(item)
|
461 |
|
462 |
def tearDown(self): |
463 |
containers = self.list_containers()[2] |
464 |
for c in containers: |
465 |
self.client.delete_container(c)
|
466 |
|
467 |
def test_update_meta(self): |
468 |
meta = {'test':'test', 'tost':'tost'} |
469 |
status, headers, data = self.get_account_meta(**meta)
|
470 |
|
471 |
#def test_invalid_account_update_meta(self):
|
472 |
# with AssertMappingInvariant(self.get_account_meta, self.account):
|
473 |
# meta = {'HTTP_X_ACCOUNT_META_TEST':'test',
|
474 |
# 'HTTP_X_ACCOUNT_META_TOST':'tost'}
|
475 |
# response = self.update_account_meta('non-existing-account', **meta)
|
476 |
|
477 |
class ContainerHead(BaseTestCase): |
478 |
def setUp(self): |
479 |
BaseTestCase.setUp(self)
|
480 |
self.account = 'test' |
481 |
self.container = 'apples' |
482 |
status = self.create_container(self.container)[0] |
483 |
|
484 |
def tearDown(self): |
485 |
for o in self.list_objects(self.container)[2]: |
486 |
self.client.delete_object(self.container, o) |
487 |
self.client.delete_container(self.container) |
488 |
|
489 |
def test_get_meta(self): |
490 |
meta = {'trash':'true'} |
491 |
t1 = datetime.datetime.utcnow() |
492 |
o = self.upload_random_data(self.container, o_names[0], **meta) |
493 |
if o:
|
494 |
status, headers, data = self.get_container_meta(self.container) |
495 |
self.assertEqual(headers['x-container-object-count'], '1') |
496 |
self.assertEqual(headers['x-container-bytes-used'], str(len(o['data']))) |
497 |
t2 = datetime.datetime.strptime(headers['last-modified'], DATE_FORMATS[2]) |
498 |
delta = (t2 - t1) |
499 |
threashold = datetime.timedelta(seconds=1)
|
500 |
self.assertTrue(delta < threashold)
|
501 |
self.assertTrue(headers['x-container-object-meta']) |
502 |
self.assertTrue('Trash' in headers['x-container-object-meta']) |
503 |
|
504 |
class ContainerGet(BaseTestCase): |
505 |
def setUp(self): |
506 |
BaseTestCase.setUp(self)
|
507 |
self.account = 'test' |
508 |
self.container = ['pears', 'apples'] |
509 |
for c in self.container: |
510 |
self.create_container(c)
|
511 |
self.obj = []
|
512 |
for o in o_names[:8]: |
513 |
self.obj.append(self.upload_random_data(self.container[0], o)) |
514 |
for o in o_names[8:]: |
515 |
self.obj.append(self.upload_random_data(self.container[1], o)) |
516 |
|
517 |
def tearDown(self): |
518 |
for c in self.container: |
519 |
for obj in self.list_objects(c)[2]: |
520 |
self.client.delete_object(c, obj)
|
521 |
self.client.delete_container(c)
|
522 |
|
523 |
def test_list_objects(self): |
524 |
objects = self.list_objects(self.container[0])[2] |
525 |
l = [elem['name'] for elem in self.obj[:8]] |
526 |
l.sort() |
527 |
self.assertEqual(objects, l)
|
528 |
|
529 |
def test_list_objects_with_limit_marker(self): |
530 |
objects = self.list_objects(self.container[0], limit=2)[2] |
531 |
l = [elem['name'] for elem in self.obj[:8]] |
532 |
l.sort() |
533 |
self.assertEqual(objects, l[:2]) |
534 |
|
535 |
markers = ['How To Win Friends And Influence People.pdf',
|
536 |
'moms_birthday.jpg']
|
537 |
limit = 4
|
538 |
for m in markers: |
539 |
objects = self.list_objects(self.container[0], limit=limit, |
540 |
marker=m)[2]
|
541 |
l = [elem['name'] for elem in self.obj[:8]] |
542 |
l.sort() |
543 |
start = l.index(m) + 1
|
544 |
end = start + limit |
545 |
end = len(l) >= end and end or len(l) |
546 |
self.assertEqual(objects, l[start:end])
|
547 |
|
548 |
def test_list_pseudo_hierarchical_folders(self): |
549 |
objects = self.list_objects(self.container[1], prefix='photos', |
550 |
delimiter='/')[2] |
551 |
self.assertEquals(['photos/animals/', 'photos/me.jpg', |
552 |
'photos/plants/'], objects)
|
553 |
|
554 |
objects = self.list_objects(self.container[1], prefix='photos/animals', |
555 |
delimiter='/')[2] |
556 |
l = ['photos/animals/cats/', 'photos/animals/dogs/'] |
557 |
self.assertEquals(l, objects)
|
558 |
|
559 |
objects = self.list_objects(self.container[1], path='photos')[2] |
560 |
self.assertEquals(['photos/me.jpg'], objects) |
561 |
|
562 |
def test_extended_list_json(self): |
563 |
objects = self.list_objects(self.container[1], |
564 |
format='json', limit=2, |
565 |
prefix='photos/animals',
|
566 |
delimiter='/')[2] |
567 |
self.assertEqual(objects[0]['subdir'], 'photos/animals/cats/') |
568 |
self.assertEqual(objects[1]['subdir'], 'photos/animals/dogs/') |
569 |
|
570 |
def test_extended_list_xml(self): |
571 |
xml = self.list_objects(self.container[1], format='xml', limit=4, |
572 |
prefix='photos', delimiter='/')[2] |
573 |
dirs = xml.getElementsByTagName('subdir')
|
574 |
self.assertEqual(len(dirs), 2) |
575 |
self.assertEqual(dirs[0].attributes['name'].value, 'photos/animals/') |
576 |
self.assertEqual(dirs[1].attributes['name'].value, 'photos/plants/') |
577 |
|
578 |
objects = xml.getElementsByTagName('name')
|
579 |
self.assertEqual(len(objects), 1) |
580 |
self.assertEqual(objects[0].childNodes[0].data, 'photos/me.jpg') |
581 |
|
582 |
def test_list_meta_double_matching(self): |
583 |
meta = {'quality':'aaa', 'stock':'true'} |
584 |
self.client.update_object_metadata(self.container[0], |
585 |
self.obj[0]['name'], **meta) |
586 |
obj = self.list_objects(self.container[0], meta='Quality,Stock')[2] |
587 |
self.assertEqual(len(obj), 1) |
588 |
self.assertTrue(obj, self.obj[0]['name']) |
589 |
|
590 |
def test_list_using_meta(self): |
591 |
meta = {'quality':'aaa'} |
592 |
for o in self.obj[:2]: |
593 |
self.client.update_object_metadata(self.container[0], o['name'], |
594 |
**meta) |
595 |
meta = {'stock':'true'} |
596 |
for o in self.obj[3:5]: |
597 |
self.client.update_object_metadata(self.container[0], o['name'], |
598 |
**meta) |
599 |
|
600 |
status, headers, data = self.list_objects(self.container[0], |
601 |
meta='Quality')
|
602 |
self.assertEqual(status, 200) |
603 |
self.assertEqual(len(data), 2) |
604 |
self.assertTrue(data, [o['name'] for o in self.obj[:2]]) |
605 |
|
606 |
# test case insensitive
|
607 |
status, headers, obj = self.list_objects(self.container[0], |
608 |
meta='quality')
|
609 |
self.assertEqual(status, 200) |
610 |
self.assertEqual(len(obj), 2) |
611 |
self.assertTrue(obj, [o['name'] for o in self.obj[:2]]) |
612 |
|
613 |
# test multiple matches
|
614 |
status, headers, obj = self.list_objects(self.container[0], |
615 |
meta='Quality,Stock')
|
616 |
self.assertEqual(status, 200) |
617 |
self.assertEqual(len(obj), 4) |
618 |
self.assertTrue(obj, [o['name'] for o in self.obj[:4]]) |
619 |
|
620 |
# test non 1-1 multiple match
|
621 |
status, headers, obj = self.list_objects(self.container[0], |
622 |
meta='Quality,aaaa')
|
623 |
self.assertEqual(status, 200) |
624 |
self.assertEqual(len(obj), 2) |
625 |
self.assertTrue(obj, [o['name'] for o in self.obj[:2]]) |
626 |
|
627 |
def test_if_modified_since(self): |
628 |
t = datetime.datetime.utcnow() |
629 |
t2 = t - datetime.timedelta(minutes=10)
|
630 |
|
631 |
#add a new object
|
632 |
self.upload_random_data(self.container[0], o_names[0]) |
633 |
|
634 |
for f in DATE_FORMATS: |
635 |
past = t2.strftime(f) |
636 |
|
637 |
headers = {'if-modified-since':'%s' %past} |
638 |
status, headers, data = self.list_objects(self.container[0], |
639 |
**headers) |
640 |
|
641 |
#assert get success
|
642 |
self.assertEqual(status, 200) |
643 |
|
644 |
def test_if_modified_since_invalid_date(self): |
645 |
headers = {'if-modified-since':''} |
646 |
status, headers, data = self.list_objects(self.container[0], **headers) |
647 |
|
648 |
#assert get success
|
649 |
self.assertEqual(status, 200) |
650 |
|
651 |
def test_if_not_modified_since(self): |
652 |
now = datetime.datetime.utcnow() |
653 |
since = now + datetime.timedelta(1)
|
654 |
|
655 |
for f in DATE_FORMATS: |
656 |
headers = {'if-modified-since':'%s' %since.strftime(f)} |
657 |
#assert not modified
|
658 |
self.assert_raises_fault(304, self.list_objects, self.container[0], |
659 |
**headers) |
660 |
|
661 |
def test_if_unmodified_since(self): |
662 |
now = datetime.datetime.utcnow() |
663 |
since = now + datetime.timedelta(1)
|
664 |
|
665 |
for f in DATE_FORMATS: |
666 |
headers = {'if-unmodified-since':'%s' %since.strftime(f)} |
667 |
status, headers, data = self.list_objects(self.container[0], **headers) |
668 |
|
669 |
#assert success
|
670 |
self.assertEqual(status, 200) |
671 |
objlist = self.list_objects(self.container[0])[2] |
672 |
self.assertEqual(data, objlist)
|
673 |
|
674 |
def test_if_unmodified_since_precondition_failed(self): |
675 |
t = datetime.datetime.utcnow() |
676 |
t2 = t - datetime.timedelta(minutes=10)
|
677 |
|
678 |
#add a new container
|
679 |
self.create_container('dummy') |
680 |
|
681 |
for f in DATE_FORMATS: |
682 |
past = t2.strftime(f) |
683 |
|
684 |
headers = {'if-unmodified-since':'%s' %past} |
685 |
#assert precondition failed
|
686 |
self.assert_raises_fault(412, self.list_objects, self.container[0], |
687 |
**headers) |
688 |
|
689 |
class ContainerPut(BaseTestCase): |
690 |
def setUp(self): |
691 |
BaseTestCase.setUp(self)
|
692 |
self.account = 'test' |
693 |
self.containers = ['c1', 'c2'] |
694 |
|
695 |
def tearDown(self): |
696 |
for c in self.list_containers()[2]: |
697 |
r = self.client.delete_container(c)
|
698 |
|
699 |
def test_create(self): |
700 |
status = self.create_container(self.containers[0])[0] |
701 |
self.assertEqual(status, 201) |
702 |
|
703 |
containers = self.list_containers()[2] |
704 |
self.assertTrue(self.containers[0] in containers) |
705 |
status = self.get_container_meta(self.containers[0])[0] |
706 |
self.assertEqual(status, 204) |
707 |
|
708 |
def test_create_twice(self): |
709 |
status, header, data = self.create_container(self.containers[0]) |
710 |
if status == 201: |
711 |
status, header, data = self.create_container(self.containers[0]) |
712 |
self.assertTrue(status, 202) |
713 |
|
714 |
class ContainerPost(BaseTestCase): |
715 |
def setUp(self): |
716 |
BaseTestCase.setUp(self)
|
717 |
self.account = 'test' |
718 |
self.container = 'apples' |
719 |
self.create_container(self.container) |
720 |
|
721 |
def tearDown(self): |
722 |
for o in self.list_objects(self.container)[2]: |
723 |
self.client.delete_object(self.account, self.container, o) |
724 |
self.client.delete_container(self.container) |
725 |
|
726 |
def test_update_meta(self): |
727 |
meta = {'test':'test33', |
728 |
'tost':'tost22'} |
729 |
self.client.update_container_metadata(self.container, **meta) |
730 |
headers = self.get_container_meta(self.container)[1] |
731 |
for k,v in meta.items(): |
732 |
k = 'x-container-meta-%s' % k
|
733 |
self.assertTrue(headers[k])
|
734 |
self.assertEqual(headers[k], v)
|
735 |
|
736 |
class ContainerDelete(BaseTestCase): |
737 |
def setUp(self): |
738 |
BaseTestCase.setUp(self)
|
739 |
self.account = 'test' |
740 |
self.containers = ['c1', 'c2'] |
741 |
for c in self.containers: |
742 |
self.create_container(c)
|
743 |
self.upload_random_data(self.containers[1], o_names[0]) |
744 |
|
745 |
def tearDown(self): |
746 |
for c in self.list_containers()[2]: |
747 |
for o in self.list_objects(c)[2]: |
748 |
self.client.delete_object(c, o)
|
749 |
self.client.delete_container(c)
|
750 |
|
751 |
def test_delete(self): |
752 |
status = self.client.delete_container(self.containers[0])[0] |
753 |
self.assertEqual(status, 204) |
754 |
|
755 |
def test_delete_non_empty(self): |
756 |
self.assert_raises_fault(409, self.client.delete_container, |
757 |
self.containers[1]) |
758 |
|
759 |
def test_delete_invalid(self): |
760 |
self.assert_raises_fault(404, self.client.delete_container, 'c3') |
761 |
|
762 |
class ObjectHead(BaseTestCase): |
763 |
pass
|
764 |
|
765 |
class ObjectGet(BaseTestCase): |
766 |
def setUp(self): |
767 |
BaseTestCase.setUp(self)
|
768 |
self.account = 'test' |
769 |
self.containers = ['c1', 'c2'] |
770 |
#create some containers
|
771 |
for c in self.containers: |
772 |
self.create_container(c)
|
773 |
|
774 |
#upload a file
|
775 |
names = ('obj1', 'obj2') |
776 |
self.objects = []
|
777 |
for n in names: |
778 |
self.objects.append(self.upload_random_data(self.containers[1], n)) |
779 |
|
780 |
def tearDown(self): |
781 |
for c in self.containers: |
782 |
for o in self.list_objects(c)[2]: |
783 |
self.client.delete_object(c, o)
|
784 |
self.client.delete_container(c)
|
785 |
|
786 |
def test_get(self): |
787 |
#perform get
|
788 |
status, headers, data = self.get_object(self.containers[1], |
789 |
self.objects[0]['name'], |
790 |
self.objects[0]['meta']) |
791 |
#assert success
|
792 |
self.assertEqual(status, 200) |
793 |
|
794 |
#assert content-type
|
795 |
self.assertEqual(headers['content-type'], |
796 |
self.objects[0]['meta']['content-type']) |
797 |
|
798 |
def test_get_invalid(self): |
799 |
self.assert_raises_fault(404, self.get_object, self.containers[0], |
800 |
self.objects[0]['name']) |
801 |
|
802 |
def test_get_partial(self): |
803 |
#perform get with range
|
804 |
headers = {'range':'bytes=0-499'} |
805 |
status, headers, data = self.get_object(self.containers[1], |
806 |
self.objects[0]['name'], |
807 |
**headers) |
808 |
|
809 |
#assert successful partial content
|
810 |
self.assertEqual(status, 206) |
811 |
|
812 |
#assert content-type
|
813 |
self.assertEqual(headers['content-type'], |
814 |
self.objects[0]['meta']['content-type']) |
815 |
|
816 |
#assert content length
|
817 |
self.assertEqual(int(headers['content-length']), 500) |
818 |
|
819 |
#assert content
|
820 |
self.assertEqual(self.objects[0]['data'][:500], data) |
821 |
|
822 |
def test_get_final_500(self): |
823 |
#perform get with range
|
824 |
headers = {'range':'bytes=-500'} |
825 |
status, headers, data = self.get_object(self.containers[1], |
826 |
self.objects[0]['name'], |
827 |
**headers) |
828 |
|
829 |
#assert successful partial content
|
830 |
self.assertEqual(status, 206) |
831 |
|
832 |
#assert content-type
|
833 |
self.assertEqual(headers['content-type'], |
834 |
self.objects[0]['meta']['content-type']) |
835 |
|
836 |
#assert content length
|
837 |
self.assertEqual(int(headers['content-length']), 500) |
838 |
|
839 |
#assert content
|
840 |
self.assertTrue(self.objects[0]['data'][-500:], data) |
841 |
|
842 |
def test_get_rest(self): |
843 |
#perform get with range
|
844 |
offset = len(self.objects[0]['data']) - 500 |
845 |
headers = {'range':'bytes=%s-' %offset} |
846 |
status, headers, data = self.get_object(self.containers[1], |
847 |
self.objects[0]['name'], |
848 |
**headers) |
849 |
|
850 |
#assert successful partial content
|
851 |
self.assertEqual(status, 206) |
852 |
|
853 |
#assert content-type
|
854 |
self.assertEqual(headers['content-type'], |
855 |
self.objects[0]['meta']['content-type']) |
856 |
|
857 |
#assert content length
|
858 |
self.assertEqual(int(headers['content-length']), 500) |
859 |
|
860 |
#assert content
|
861 |
self.assertTrue(self.objects[0]['data'][-500:], data) |
862 |
|
863 |
def test_get_range_not_satisfiable(self): |
864 |
#perform get with range
|
865 |
offset = len(self.objects[0]['data']) + 1 |
866 |
headers = {'range':'bytes=0-%s' %offset} |
867 |
|
868 |
#assert range not satisfiable
|
869 |
self.assert_raises_fault(416, self.get_object, self.containers[1], |
870 |
self.objects[0]['name'], **headers) |
871 |
|
872 |
def test_multiple_range(self): |
873 |
#perform get with multiple range
|
874 |
ranges = ['0-499', '-500', '1000-'] |
875 |
headers = {'range' : 'bytes=%s' % ','.join(ranges)} |
876 |
status, headers, data = self.get_object(self.containers[1], |
877 |
self.objects[0]['name'], |
878 |
**headers) |
879 |
|
880 |
# assert partial content
|
881 |
self.assertEqual(status, 206) |
882 |
|
883 |
# assert Content-Type of the reply will be multipart/byteranges
|
884 |
self.assertTrue(headers['content-type']) |
885 |
content_type_parts = headers['content-type'].split()
|
886 |
self.assertEqual(content_type_parts[0], ('multipart/byteranges;')) |
887 |
|
888 |
boundary = '--%s' %content_type_parts[1].split('=')[-1:][0] |
889 |
cparts = data.split(boundary)[1:-1] |
890 |
|
891 |
# assert content parts are exactly 2
|
892 |
self.assertEqual(len(cparts), len(ranges)) |
893 |
|
894 |
# for each content part assert headers
|
895 |
i = 0
|
896 |
for cpart in cparts: |
897 |
content = cpart.split('\r\n')
|
898 |
headers = content[1:3] |
899 |
content_range = headers[0].split(': ') |
900 |
self.assertEqual(content_range[0], 'Content-Range') |
901 |
|
902 |
r = ranges[i].split('-')
|
903 |
if not r[0] and not r[1]: |
904 |
pass
|
905 |
elif not r[0]: |
906 |
start = len(self.objects[0]['data']) - int(r[1]) |
907 |
end = len(self.objects[0]['data']) |
908 |
elif not r[1]: |
909 |
start = int(r[0]) |
910 |
end = len(self.objects[0]['data']) |
911 |
else:
|
912 |
start = int(r[0]) |
913 |
end = int(r[1]) + 1 |
914 |
fdata = self.objects[0]['data'][start:end] |
915 |
sdata = '\r\n'.join(content[4:-1]) |
916 |
self.assertEqual(len(fdata), len(sdata)) |
917 |
self.assertEquals(fdata, sdata)
|
918 |
i+=1
|
919 |
|
920 |
def test_multiple_range_not_satisfiable(self): |
921 |
#perform get with multiple range
|
922 |
out_of_range = len(self.objects[0]['data']) + 1 |
923 |
ranges = ['0-499', '-500', '%d-' %out_of_range] |
924 |
headers = {'range' : 'bytes=%s' % ','.join(ranges)} |
925 |
|
926 |
# assert partial content
|
927 |
self.assert_raises_fault(416, self.get_object, self.containers[1], |
928 |
self.objects[0]['name'], **headers) |
929 |
|
930 |
|
931 |
def test_get_with_if_match(self): |
932 |
#perform get with If-Match
|
933 |
headers = {'if-match':self.objects[0]['hash']} |
934 |
status, headers, data = self.get_object(self.containers[1], |
935 |
self.objects[0]['name'], |
936 |
**headers) |
937 |
#assert get success
|
938 |
self.assertEqual(status, 200) |
939 |
|
940 |
#assert content-type
|
941 |
self.assertEqual(headers['content-type'], |
942 |
self.objects[0]['meta']['content-type']) |
943 |
|
944 |
#assert response content
|
945 |
self.assertEqual(self.objects[0]['data'], data) |
946 |
|
947 |
def test_get_with_if_match_star(self): |
948 |
#perform get with If-Match *
|
949 |
headers = {'if-match':'*'} |
950 |
status, headers, data = self.get_object(self.containers[1], |
951 |
self.objects[0]['name'], |
952 |
**headers) |
953 |
#assert get success
|
954 |
self.assertEqual(status, 200) |
955 |
|
956 |
#assert content-type
|
957 |
self.assertEqual(headers['content-type'], |
958 |
self.objects[0]['meta']['content-type']) |
959 |
|
960 |
#assert response content
|
961 |
self.assertEqual(self.objects[0]['data'], data) |
962 |
|
963 |
def test_get_with_multiple_if_match(self): |
964 |
#perform get with If-Match
|
965 |
etags = [i['hash'] for i in self.objects if i] |
966 |
etags = ','.join('"%s"' % etag for etag in etags) |
967 |
headers = {'if-match':etags}
|
968 |
status, headers, data = self.get_object(self.containers[1], |
969 |
self.objects[0]['name'], |
970 |
**headers) |
971 |
#assert get success
|
972 |
self.assertEqual(status, 200) |
973 |
|
974 |
#assert content-type
|
975 |
self.assertEqual(headers['content-type'], |
976 |
self.objects[0]['meta']['content-type']) |
977 |
|
978 |
#assert content-type
|
979 |
self.assertEqual(headers['content-type'], |
980 |
self.objects[0]['meta']['content-type']) |
981 |
|
982 |
#assert response content
|
983 |
self.assertEqual(self.objects[0]['data'], data) |
984 |
|
985 |
def test_if_match_precondition_failed(self): |
986 |
#perform get with If-Match
|
987 |
headers = {'if-match':'123'} |
988 |
|
989 |
#assert precondition failed
|
990 |
self.assert_raises_fault(412, self.get_object, self.containers[1], |
991 |
self.objects[0]['name'], **headers) |
992 |
|
993 |
|
994 |
def test_if_none_match(self): |
995 |
#perform get with If-None-Match
|
996 |
headers = {'if-none-match':'123'} |
997 |
status, headers, data = self.get_object(self.containers[1], |
998 |
self.objects[0]['name'], |
999 |
**headers) |
1000 |
|
1001 |
#assert get success
|
1002 |
self.assertEqual(status, 200) |
1003 |
|
1004 |
#assert content-type
|
1005 |
self.assertEqual(headers['content-type'], |
1006 |
self.objects[0]['meta']['content-type']) |
1007 |
|
1008 |
def test_if_none_match(self): |
1009 |
#perform get with If-None-Match *
|
1010 |
headers = {'if-none-match':'*'} |
1011 |
|
1012 |
#assert not modified
|
1013 |
self.assert_raises_fault(304, self.get_object, self.containers[1], |
1014 |
self.objects[0]['name'], |
1015 |
**headers) |
1016 |
|
1017 |
def test_if_none_match_not_modified(self): |
1018 |
#perform get with If-None-Match
|
1019 |
headers = {'if-none-match':'%s' %self.objects[0]['hash']} |
1020 |
|
1021 |
#assert not modified
|
1022 |
self.assert_raises_fault(304, self.get_object, self.containers[1], |
1023 |
self.objects[0]['name'], |
1024 |
**headers) |
1025 |
|
1026 |
headers = self.get_object(self.containers[1], |
1027 |
self.objects[0]['name'])[1] |
1028 |
self.assertEqual(headers['etag'], self.objects[0]['hash']) |
1029 |
|
1030 |
def test_if_modified_since(self): |
1031 |
t = datetime.datetime.utcnow() |
1032 |
t2 = t - datetime.timedelta(minutes=10)
|
1033 |
|
1034 |
#modify the object
|
1035 |
self.upload_data(self.containers[1], |
1036 |
self.objects[0]['name'], |
1037 |
self.objects[0]['data'][:200]) |
1038 |
|
1039 |
for f in DATE_FORMATS: |
1040 |
past = t2.strftime(f) |
1041 |
|
1042 |
headers = {'if-modified-since':'%s' %past} |
1043 |
status, headers, data = self.get_object(self.containers[1], |
1044 |
self.objects[0]['name'], |
1045 |
**headers) |
1046 |
|
1047 |
#assert get success
|
1048 |
self.assertEqual(status, 200) |
1049 |
|
1050 |
#assert content-type
|
1051 |
self.assertEqual(headers['content-type'], |
1052 |
self.objects[0]['meta']['content-type']) |
1053 |
|
1054 |
def test_if_modified_since_invalid_date(self): |
1055 |
headers = {'if-modified-since':''} |
1056 |
status, headers, data = self.get_object(self.containers[1], |
1057 |
self.objects[0]['name'], |
1058 |
**headers) |
1059 |
|
1060 |
#assert get success
|
1061 |
self.assertEqual(status, 200) |
1062 |
|
1063 |
#assert content-type
|
1064 |
self.assertEqual(headers['content-type'], |
1065 |
self.objects[0]['meta']['content-type']) |
1066 |
|
1067 |
def test_if_not_modified_since(self): |
1068 |
now = datetime.datetime.utcnow() |
1069 |
since = now + datetime.timedelta(1)
|
1070 |
|
1071 |
for f in DATE_FORMATS: |
1072 |
headers = {'if-modified-since':'%s' %since.strftime(f)} |
1073 |
|
1074 |
#assert not modified
|
1075 |
self.assert_raises_fault(304, self.get_object, self.containers[1], |
1076 |
self.objects[0]['name'], **headers) |
1077 |
|
1078 |
|
1079 |
def test_if_unmodified_since(self): |
1080 |
now = datetime.datetime.utcnow() |
1081 |
since = now + datetime.timedelta(1)
|
1082 |
|
1083 |
for f in DATE_FORMATS: |
1084 |
headers = {'if-unmodified-since':'%s' %since.strftime(f)} |
1085 |
status, headers, data = self.get_object(self.containers[1], |
1086 |
self.objects[0]['name'], |
1087 |
**headers) |
1088 |
#assert success
|
1089 |
self.assertEqual(status, 200) |
1090 |
self.assertEqual(self.objects[0]['data'], data) |
1091 |
|
1092 |
#assert content-type
|
1093 |
self.assertEqual(headers['content-type'], |
1094 |
self.objects[0]['meta']['content-type']) |
1095 |
|
1096 |
def test_if_unmodified_since_precondition_failed(self): |
1097 |
t = datetime.datetime.utcnow() |
1098 |
t2 = t - datetime.timedelta(minutes=10)
|
1099 |
|
1100 |
#modify the object
|
1101 |
self.upload_data(self.containers[1], |
1102 |
self.objects[0]['name'], |
1103 |
self.objects[0]['data'][:200]) |
1104 |
|
1105 |
for f in DATE_FORMATS: |
1106 |
past = t2.strftime(f) |
1107 |
|
1108 |
headers = {'if-unmodified-since':'%s' %past} |
1109 |
|
1110 |
#assert precondition failed
|
1111 |
self.assert_raises_fault(412, self.get_object, self.containers[1], |
1112 |
self.objects[0]['name'], **headers) |
1113 |
|
1114 |
|
1115 |
def test_hashes(self): |
1116 |
l = 8388609
|
1117 |
fname = 'largefile'
|
1118 |
o = self.upload_random_data(self.containers[1], fname, l) |
1119 |
if o:
|
1120 |
data = self.get_object(self.containers[1], |
1121 |
fname, |
1122 |
'json')[2] |
1123 |
body = json.loads(data) |
1124 |
hashes = body['hashes']
|
1125 |
block_size = body['block_size']
|
1126 |
block_hash = body['block_hash']
|
1127 |
block_num = l/block_size == 0 and l/block_size or l/block_size + 1 |
1128 |
self.assertTrue(len(hashes), block_num) |
1129 |
i = 0
|
1130 |
for h in hashes: |
1131 |
start = i * block_size |
1132 |
end = (i + 1) * block_size
|
1133 |
hash = compute_block_hash(o['data'][start:end], block_hash)
|
1134 |
self.assertEqual(h, hash) |
1135 |
i += 1
|
1136 |
|
1137 |
class ObjectPut(BaseTestCase): |
1138 |
def setUp(self): |
1139 |
BaseTestCase.setUp(self)
|
1140 |
self.account = 'test' |
1141 |
self.container = 'c1' |
1142 |
self.create_container(self.container) |
1143 |
|
1144 |
def tearDown(self): |
1145 |
objects = self.list_objects(self.container)[2] |
1146 |
for o in objects: |
1147 |
self.client.delete_object(self.container, o) |
1148 |
self.client.delete_container(self.container) |
1149 |
|
1150 |
def test_upload(self): |
1151 |
name = o_names[0]
|
1152 |
meta = {'test':'test1'} |
1153 |
o = self.upload_random_data(self.container, name, **meta) |
1154 |
|
1155 |
headers = self.client.retrieve_object_metadata(self.container, |
1156 |
name, |
1157 |
restricted=True)
|
1158 |
self.assertTrue('test' in headers.keys()) |
1159 |
self.assertEqual(headers['test'], meta['test']) |
1160 |
|
1161 |
#assert uploaded content
|
1162 |
status, headers, content = self.get_object(self.container, name) |
1163 |
self.assertEqual(len(o['data']), int(headers['content-length'])) |
1164 |
self.assertEqual(o['data'], content) |
1165 |
|
1166 |
def test_upload_unprocessable_entity(self): |
1167 |
meta={'etag':'123', 'test':'test1'} |
1168 |
|
1169 |
#assert unprocessable entity
|
1170 |
self.assert_raises_fault(422, self.upload_random_data,self.container, |
1171 |
o_names[0], **meta)
|
1172 |
|
1173 |
def test_chucked_transfer(self): |
1174 |
fname = './api/tests.py'
|
1175 |
objname = os.path.split(fname)[-1:][0] |
1176 |
f = open(fname, 'r') |
1177 |
status = self.client.create_object(self.container, |
1178 |
objname, |
1179 |
f, |
1180 |
chunked=True)[0] |
1181 |
self.assertEqual(status, 201) |
1182 |
|
1183 |
uploaded_data = self.get_object(self.container, |
1184 |
objname)[2]
|
1185 |
f = open(fname, 'r') |
1186 |
actual_data = f.read() |
1187 |
self.assertEqual(actual_data, uploaded_data)
|
1188 |
|
1189 |
class ObjectCopy(BaseTestCase): |
1190 |
def setUp(self): |
1191 |
BaseTestCase.setUp(self)
|
1192 |
self.account = 'test' |
1193 |
self.containers = ['c1', 'c2'] |
1194 |
for c in self.containers: |
1195 |
self.create_container(c)
|
1196 |
self.obj = self.upload_random_data(self.containers[0], o_names[0]) |
1197 |
|
1198 |
def tearDown(self): |
1199 |
for c in self.containers: |
1200 |
for o in self.list_objects(c)[2]: |
1201 |
self.client.delete_object(c, o)
|
1202 |
self.client.delete_container(c)
|
1203 |
|
1204 |
def test_copy(self): |
1205 |
with AssertMappingInvariant(self.client.retrieve_object_metadata, |
1206 |
self.containers[0], self.obj['name']): |
1207 |
#perform copy
|
1208 |
meta = {'test':'testcopy'} |
1209 |
status = self.client.copy_object(self.containers[0], |
1210 |
self.obj['name'], |
1211 |
self.containers[0], |
1212 |
'testcopy',
|
1213 |
**meta)[0]
|
1214 |
|
1215 |
#assert copy success
|
1216 |
self.assertEqual(status, 201) |
1217 |
|
1218 |
#assert access the new object
|
1219 |
headers = self.client.retrieve_object_metadata(self.containers[0], |
1220 |
'testcopy')
|
1221 |
self.assertTrue('x-object-meta-test' in headers.keys()) |
1222 |
self.assertTrue(headers['x-object-meta-test'], 'testcopy') |
1223 |
|
1224 |
#assert etag is the same
|
1225 |
self.assertEqual(headers['etag'], self.obj['hash']) |
1226 |
|
1227 |
#assert src object still exists
|
1228 |
self.assert_object_exists(self.containers[0], self.obj['name']) |
1229 |
|
1230 |
def test_copy_from_different_container(self): |
1231 |
with AssertMappingInvariant(self.client.retrieve_object_metadata, |
1232 |
self.containers[0], self.obj['name']): |
1233 |
meta = {'test':'testcopy'} |
1234 |
status = self.client.copy_object(self.containers[0], |
1235 |
self.obj['name'], |
1236 |
self.containers[1], |
1237 |
'testcopy',
|
1238 |
**meta)[0]
|
1239 |
self.assertEqual(status, 201) |
1240 |
|
1241 |
# assert updated metadata
|
1242 |
meta = self.client.retrieve_object_metadata(self.containers[1], |
1243 |
'testcopy',
|
1244 |
restricted=True)
|
1245 |
self.assertTrue('test' in meta.keys()) |
1246 |
self.assertTrue(meta['test'], 'testcopy') |
1247 |
|
1248 |
#assert src object still exists
|
1249 |
self.assert_object_exists(self.containers[0], self.obj['name']) |
1250 |
|
1251 |
def test_copy_invalid(self): |
1252 |
#copy from invalid object
|
1253 |
meta = {'test':'testcopy'} |
1254 |
self.assert_raises_fault(404, self.client.copy_object, self.containers[0], |
1255 |
'test.py', self.containers[1], 'testcopy', |
1256 |
**meta) |
1257 |
|
1258 |
#copy from invalid container
|
1259 |
meta = {'test':'testcopy'} |
1260 |
self.assert_raises_fault(404, self.client.copy_object, self.containers[1], |
1261 |
self.obj['name'], self.containers[1], |
1262 |
'testcopy', **meta)
|
1263 |
|
1264 |
|
1265 |
class ObjectMove(ObjectCopy): |
1266 |
def test_move(self): |
1267 |
#perform move
|
1268 |
meta = {'test':'testcopy'} |
1269 |
src_path = os.path.join('/', self.containers[0], self.obj['name']) |
1270 |
status = self.client.move_object(self.containers[0], self.obj['name'], |
1271 |
self.containers[0], 'testcopy', |
1272 |
**meta)[0]
|
1273 |
|
1274 |
#assert successful move
|
1275 |
self.assertEqual(status, 201) |
1276 |
|
1277 |
#assert updated metadata
|
1278 |
meta = self.client.retrieve_object_metadata(self.containers[0], |
1279 |
'testcopy',
|
1280 |
restricted=True)
|
1281 |
self.assertTrue('test' in meta.keys()) |
1282 |
self.assertTrue(meta['test'], 'testcopy') |
1283 |
|
1284 |
#assert src object no more exists
|
1285 |
self.assert_object_not_exists(self.containers[0], self.obj['name']) |
1286 |
|
1287 |
class ObjectPost(BaseTestCase): |
1288 |
def setUp(self): |
1289 |
BaseTestCase.setUp(self)
|
1290 |
self.account = 'test' |
1291 |
self.containers = ['c1', 'c2'] |
1292 |
for c in self.containers: |
1293 |
self.create_container(c)
|
1294 |
self.obj = self.upload_random_data(self.containers[0], o_names[0]) |
1295 |
|
1296 |
def tearDown(self): |
1297 |
for c in self.containers: |
1298 |
for o in self.list_objects(c)[2]: |
1299 |
self.client.delete_object(c, o)
|
1300 |
self.client.delete_container(c)
|
1301 |
|
1302 |
def test_update_meta(self): |
1303 |
#perform update metadata
|
1304 |
more = {'foo':'foo', 'bar':'bar'} |
1305 |
status = self.client.update_object_metadata(self.containers[0], |
1306 |
self.obj['name'], |
1307 |
**more)[0]
|
1308 |
#assert request accepted
|
1309 |
self.assertEqual(status, 202) |
1310 |
|
1311 |
#assert old metadata are still there
|
1312 |
headers = self.client.retrieve_object_metadata(self.containers[0], |
1313 |
self.obj['name'], |
1314 |
restricted=True)
|
1315 |
#assert new metadata have been updated
|
1316 |
for k,v in more.items(): |
1317 |
self.assertTrue(k in headers.keys()) |
1318 |
self.assertTrue(headers[k], v)
|
1319 |
|
1320 |
def test_update_object(self, |
1321 |
first_byte_pos=0,
|
1322 |
last_byte_pos=499,
|
1323 |
instance_length = True,
|
1324 |
content_length = 500):
|
1325 |
l = len(self.obj['data']) |
1326 |
length = l if instance_length else '*' |
1327 |
range = 'bytes %d-%d/%s' %(first_byte_pos,
|
1328 |
last_byte_pos, |
1329 |
length) |
1330 |
partial = last_byte_pos - first_byte_pos + 1
|
1331 |
data = get_random_data(partial) |
1332 |
headers = {'content-range':range, |
1333 |
'content-type':'application/octet-stream'} |
1334 |
if content_length:
|
1335 |
headers.update({'content-length':'%s' % content_length}) |
1336 |
|
1337 |
status = self.client.update_object_data(self.containers[0], |
1338 |
self.obj['name'], |
1339 |
data, |
1340 |
headers)[0]
|
1341 |
|
1342 |
|
1343 |
if partial < 0 or (instance_length and l <= last_byte_pos): |
1344 |
self.assertEqual(status, 202) |
1345 |
else:
|
1346 |
self.assertEqual(status, 204) |
1347 |
|
1348 |
#check modified object
|
1349 |
content = self.get_object(self.containers[0], self.obj['name'])[2] |
1350 |
self.assertEqual(content[0:partial], data) |
1351 |
self.assertEqual(content[partial:l], self.obj['data'][partial:l]) |
1352 |
|
1353 |
def test_update_object_no_content_length(self): |
1354 |
self.test_update_object(content_length = None) |
1355 |
|
1356 |
|
1357 |
#fails if the server resets the content-legth
|
1358 |
#def test_update_object_invalid_content_length(self):
|
1359 |
# with AssertContentInvariant(self.get_object, self.containers[0],
|
1360 |
# self.obj['name']):
|
1361 |
# self.test_update_object(content_length = 1000)
|
1362 |
|
1363 |
def test_update_object_with_unknown_instance_length(self): |
1364 |
self.test_update_object(instance_length = False) |
1365 |
|
1366 |
def test_update_object_invalid_range(self): |
1367 |
with AssertContentInvariant(self.get_object, self.containers[0], |
1368 |
self.obj['name']): |
1369 |
self.test_update_object(499, 0, True) |
1370 |
|
1371 |
#no use if the server resets the content-legth
|
1372 |
def test_update_object_invalid_range_and_length(self): |
1373 |
with AssertContentInvariant(self.get_object, self.containers[0], |
1374 |
self.obj['name']): |
1375 |
self.test_update_object(499, 0, True, -1) |
1376 |
|
1377 |
#no use if the server resets the content-legth
|
1378 |
def test_update_object_invalid_range_with_no_content_length(self): |
1379 |
with AssertContentInvariant(self.get_object, self.containers[0], |
1380 |
self.obj['name']): |
1381 |
self.test_update_object(499, 0, True, content_length = None) |
1382 |
|
1383 |
def test_update_object_out_of_limits(self): |
1384 |
with AssertContentInvariant(self.get_object, self.containers[0], |
1385 |
self.obj['name']): |
1386 |
l = len(self.obj['data']) |
1387 |
self.assert_raises_fault(416, self.test_update_object, 0, l+1, True) |
1388 |
|
1389 |
def test_append(self): |
1390 |
data = get_random_data(500)
|
1391 |
headers = {'content-type':'application/octet-stream', |
1392 |
'content-length':'500'} |
1393 |
status = self.client.update_object_data(self.containers[0], |
1394 |
self.obj['name'], |
1395 |
data, headers)[0]
|
1396 |
|
1397 |
self.assertEqual(status, 204) |
1398 |
|
1399 |
content = self.get_object(self.containers[0], self.obj['name'])[2] |
1400 |
self.assertEqual(len(content), len(self.obj['data']) + 500) |
1401 |
self.assertEqual(content[:-500], self.obj['data']) |
1402 |
|
1403 |
def test_update_with_chunked_transfer(self): |
1404 |
data, pure = create_random_chunked_data() |
1405 |
dl = len(pure)
|
1406 |
fl = len(self.obj['data']) |
1407 |
meta = {'transfer-encoding':'chunked', |
1408 |
'content-range':'bytes 0-/%d' %fl} |
1409 |
self.update_object(self.containers[0], self.obj['name'], data, |
1410 |
'application/octet-stream', **meta)
|
1411 |
|
1412 |
#check modified object
|
1413 |
content = self.get_object(self.containers[0], self.obj['name'])[2] |
1414 |
self.assertEqual(content[0:dl], pure) |
1415 |
self.assertEqual(content[dl:fl], self.obj['data'][dl:fl]) |
1416 |
|
1417 |
def test_update_with_chunked_transfer_strict_range(self): |
1418 |
data, pure = create_random_chunked_data() |
1419 |
dl = len(pure) - 1 |
1420 |
fl = len(self.obj['data']) |
1421 |
meta = {'transfer-encoding':'chunked', |
1422 |
'content-range':'bytes 0-%d/%d' %(dl, fl)} |
1423 |
self.update_object(self.containers[0], self.obj['name'], data, |
1424 |
'application/octet-stream', **meta)
|
1425 |
|
1426 |
#check modified object
|
1427 |
content = self.get_object(self.containers[0], self.obj['name'])[2] |
1428 |
self.assertEqual(content[0:dl+1], pure) |
1429 |
self.assertEqual(content[dl+1:fl], self.obj['data'][dl+1:fl]) |
1430 |
|
1431 |
class ObjectDelete(BaseTestCase): |
1432 |
def setUp(self): |
1433 |
BaseTestCase.setUp(self)
|
1434 |
self.account = 'test' |
1435 |
self.containers = ['c1', 'c2'] |
1436 |
for c in self.containers: |
1437 |
self.create_container(c)
|
1438 |
self.obj = self.upload_random_data(self.containers[0], o_names[0]) |
1439 |
|
1440 |
def tearDown(self): |
1441 |
for c in self.containers: |
1442 |
for o in self.list_objects(c)[2]: |
1443 |
self.client.delete_object(c, o)
|
1444 |
self.client.delete_container(c)
|
1445 |
|
1446 |
def test_delete(self): |
1447 |
#perform delete object
|
1448 |
self.client.delete_object(self.containers[0], self.obj['name'])[0] |
1449 |
|
1450 |
def test_delete_invalid(self): |
1451 |
#assert item not found
|
1452 |
self.assert_raises_fault(404, self.client.delete_object, self.containers[1], |
1453 |
self.obj['name']) |
1454 |
|
1455 |
class AssertMappingInvariant(object): |
1456 |
def __init__(self, callable, *args, **kwargs): |
1457 |
self.callable = callable |
1458 |
self.args = args
|
1459 |
self.kwargs = kwargs
|
1460 |
|
1461 |
def __enter__(self): |
1462 |
self.map = self.callable(*self.args, **self.kwargs) |
1463 |
return self.map |
1464 |
|
1465 |
def __exit__(self, type, value, tb): |
1466 |
map = self.callable(*self.args, **self.kwargs) |
1467 |
for k in self.map.keys(): |
1468 |
if is_date(map[k]): |
1469 |
continue
|
1470 |
assert map[k] == self.map[k] |
1471 |
|
1472 |
class AssertContentInvariant(object): |
1473 |
def __init__(self, callable, *args, **kwargs): |
1474 |
self.callable = callable |
1475 |
self.args = args
|
1476 |
self.kwargs = kwargs
|
1477 |
|
1478 |
def __enter__(self): |
1479 |
self.content = self.callable(*self.args, **self.kwargs)[2] |
1480 |
return self.content |
1481 |
|
1482 |
def __exit__(self, type, value, tb): |
1483 |
content = self.callable(*self.args, **self.kwargs)[2] |
1484 |
assert self.content == content |
1485 |
|
1486 |
def get_content_splitted(response): |
1487 |
if response:
|
1488 |
return response.content.split('\n') |
1489 |
|
1490 |
def compute_md5_hash(data): |
1491 |
md5 = hashlib.md5() |
1492 |
offset = 0
|
1493 |
md5.update(data) |
1494 |
return md5.hexdigest().lower()
|
1495 |
|
1496 |
def compute_block_hash(data, algorithm): |
1497 |
h = hashlib.new(algorithm) |
1498 |
h.update(data.rstrip('\x00'))
|
1499 |
return h.hexdigest()
|
1500 |
|
1501 |
def create_chunked_update_test_file(src, dest): |
1502 |
fr = open(src, 'r') |
1503 |
fw = open(dest, 'w') |
1504 |
data = fr.readline() |
1505 |
while data:
|
1506 |
fw.write(hex(len(data))) |
1507 |
fw.write('\r\n')
|
1508 |
fw.write(data) |
1509 |
data = fr.readline() |
1510 |
fw.write(hex(0)) |
1511 |
fw.write('\r\n')
|
1512 |
|
1513 |
def create_random_chunked_data(rows=5): |
1514 |
i = 0
|
1515 |
out = [] |
1516 |
pure= [] |
1517 |
while i < rows:
|
1518 |
data = get_random_data(random.randint(1, 100)) |
1519 |
out.append(hex(len(data))) |
1520 |
out.append(data) |
1521 |
pure.append(data) |
1522 |
i+=1
|
1523 |
out.append(hex(0)) |
1524 |
out.append('\r\n')
|
1525 |
return '\r\n'.join(out), ''.join(pure) |
1526 |
|
1527 |
def get_random_data(length=500): |
1528 |
char_set = string.ascii_uppercase + string.digits |
1529 |
return ''.join(random.choice(char_set) for x in range(length)) |
1530 |
|
1531 |
def is_date(date): |
1532 |
MONTHS = 'jan feb mar apr may jun jul aug sep oct nov dec'.split()
|
1533 |
__D = r'(?P<day>\d{2})'
|
1534 |
__D2 = r'(?P<day>[ \d]\d)'
|
1535 |
__M = r'(?P<mon>\w{3})'
|
1536 |
__Y = r'(?P<year>\d{4})'
|
1537 |
__Y2 = r'(?P<year>\d{2})'
|
1538 |
__T = r'(?P<hour>\d{2}):(?P<min>\d{2}):(?P<sec>\d{2})'
|
1539 |
RFC1123_DATE = re.compile(r'^\w{3}, %s %s %s %s GMT$' % (__D, __M, __Y, __T))
|
1540 |
RFC850_DATE = re.compile(r'^\w{6,9}, %s-%s-%s %s GMT$' % (__D, __M, __Y2, __T))
|
1541 |
ASCTIME_DATE = re.compile(r'^\w{3} %s %s %s %s$' % (__M, __D2, __T, __Y))
|
1542 |
for regex in RFC1123_DATE, RFC850_DATE, ASCTIME_DATE: |
1543 |
m = regex.match(date) |
1544 |
if m is not None: |
1545 |
return True |
1546 |
return False |
1547 |
|
1548 |
o_names = ['kate.jpg',
|
1549 |
'kate_beckinsale.jpg',
|
1550 |
'How To Win Friends And Influence People.pdf',
|
1551 |
'moms_birthday.jpg',
|
1552 |
'poodle_strut.mov',
|
1553 |
'Disturbed - Down With The Sickness.mp3',
|
1554 |
'army_of_darkness.avi',
|
1555 |
'the_mad.avi',
|
1556 |
'photos/animals/dogs/poodle.jpg',
|
1557 |
'photos/animals/dogs/terrier.jpg',
|
1558 |
'photos/animals/cats/persian.jpg',
|
1559 |
'photos/animals/cats/siamese.jpg',
|
1560 |
'photos/plants/fern.jpg',
|
1561 |
'photos/plants/rose.jpg',
|
1562 |
'photos/me.jpg']
|
1563 |
|
1564 |
if __name__ == "__main__": |
1565 |
unittest.main() |