vdirsyncer/tests/storage/__init__.py
Markus Unterwaditzer 1018ed6da6 More tests
2014-03-05 21:42:37 +01:00

90 lines
2.9 KiB
Python

# -*- coding: utf-8 -*-
'''
vdirsyncer.tests.storage
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
:copyright: (c) 2014 Markus Unterwaditzer
:license: MIT, see LICENSE for more details.
'''
from vdirsyncer.storage.base import Item
import vdirsyncer.exceptions as exceptions
from .. import assert_item_equals
import random
class StorageTests(object):
item_template = u'UID:{uid}\nX-SOMETHING:{r}'
def _create_bogus_item(self, uid, item_template=None):
r = random.random()
item_template = item_template or self.item_template
return Item(item_template.format(uid=uid, r=r))
def _get_storage(self, **kwargs):
raise NotImplementedError()
def test_generic(self):
items = map(self._create_bogus_item, range(1, 10))
for i, item in enumerate(items):
assert item.uid == unicode(i + 1), item.raw
s = self._get_storage()
hrefs = []
for item in items:
hrefs.append(s.upload(item))
hrefs.sort()
assert hrefs == sorted(s.list())
for href, etag in hrefs:
assert s.has(href)
obj, etag2 = s.get(href)
assert etag == etag2
assert 'UID:{}'.format(obj.uid) in obj.raw
def test_upload_already_existing(self):
s = self._get_storage()
item = self._create_bogus_item(1)
s.upload(item)
self.assertRaises(exceptions.PreconditionFailed, s.upload, item)
def test_upload(self):
s = self._get_storage()
item = self._create_bogus_item(1)
href, etag = s.upload(item)
assert_item_equals(s.get(href)[0], item)
def test_update(self):
s = self._get_storage()
item = self._create_bogus_item(1)
href, etag = s.upload(item)
assert_item_equals(s.get(href)[0], item)
new_item = self._create_bogus_item(1)
s.update(href, new_item, etag)
assert_item_equals(s.get(href)[0], new_item)
def test_update_nonexisting(self):
s = self._get_storage()
item = self._create_bogus_item(1)
self.assertRaises(exceptions.PreconditionFailed,
s.update, s._get_href('1'), item, 123)
self.assertRaises(exceptions.PreconditionFailed,
s.update, 'huehue', item, 123)
def test_wrong_etag(self):
s = self._get_storage()
obj = self._create_bogus_item(1)
href, etag = s.upload(obj)
self.assertRaises(
exceptions.PreconditionFailed, s.update, href, obj, 'lolnope')
self.assertRaises(
exceptions.PreconditionFailed, s.delete, href, 'lolnope')
def test_delete_nonexisting(self):
s = self._get_storage()
self.assertRaises(exceptions.PreconditionFailed, s.delete, '1', 123)
def test_list(self):
s = self._get_storage()
assert not list(s.list())
s.upload(self._create_bogus_item('1'))
assert list(s.list())