mirror of
https://github.com/samsonjs/vdirsyncer.git
synced 2026-03-25 08:55:50 +00:00
119 lines
3.9 KiB
Python
119 lines
3.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
|
|
import pytest
|
|
|
|
|
|
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_args(self, collection=None):
|
|
raise NotImplementedError()
|
|
|
|
def _get_storage(self):
|
|
s = self.storage_class(**self.get_storage_args())
|
|
assert not list(s.list())
|
|
return s
|
|
|
|
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)
|
|
with pytest.raises(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)
|
|
with pytest.raises(exceptions.PreconditionFailed):
|
|
s.update(s._get_href('1'), item, '"123"')
|
|
with pytest.raises(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)
|
|
with pytest.raises(exceptions.PreconditionFailed):
|
|
s.update(href, obj, '"lolnope"')
|
|
with pytest.raises(exceptions.PreconditionFailed):
|
|
s.delete(href, '"lolnope"')
|
|
|
|
def test_delete_nonexisting(self):
|
|
s = self._get_storage()
|
|
with pytest.raises(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())
|
|
|
|
def test_discover(self):
|
|
items = []
|
|
for i in range(4):
|
|
i += 1
|
|
s = self.storage_class(**self.get_storage_args(collection='test' + str(i)))
|
|
items.append(self._create_bogus_item(str(i)))
|
|
s.upload(items[-1])
|
|
|
|
d = self.storage_class.discover(
|
|
**self.get_storage_args(collection=None))
|
|
for s in d:
|
|
((href, etag),) = s.list()
|
|
item, etag = s.get(href)
|
|
assert item.raw in set(x.raw for x in items)
|
|
|
|
def test_collection_arg(self):
|
|
s = self.storage_class(**self.get_storage_args(collection='test2'))
|
|
# Can't do stronger assertion because of radicale, which needs a
|
|
# fileextension to guess the collection type.
|
|
assert 'test2' in s.collection
|