Fix tests

This commit is contained in:
Markus Unterwaditzer 2014-02-16 15:38:01 +01:00
parent 3af84ccf90
commit 2d30d1b7f2
2 changed files with 33 additions and 17 deletions

View file

@ -20,7 +20,7 @@ class StorageTests(object):
def _get_storage(self, **kwargs):
raise NotImplementedError()
def test_generic_upload(self):
def test_generic(self):
items = [
'UID:1',
'UID:2',
@ -36,9 +36,14 @@ class StorageTests(object):
s = self._get_storage(fileext=fileext)
for item in items:
s.upload(Item(item))
a = set(uid for uid, etag in s.list_items())
a = set(uid for uid, etag in s.list())
b = set(str(y) for y in range(1, 10))
assert a == b
for i in b:
assert s.has(i)
item, uid, etag = s.get(i)
assert uid == i
assert item.raw == 'UID:{}'.format(i)
class FilesystemStorageTests(TestCase, StorageTests):

View file

@ -13,40 +13,51 @@ from vdirsyncer.storage.memory import MemoryStorage
from vdirsyncer.sync import sync
import vdirsyncer.exceptions as exceptions
def only(x):
x = list(x)
assert len(x) == 1, x
return x[0]
def empty_storage(x):
return list(x.list()) == []
class SyncTests(TestCase):
def test_basic(self):
a = MemoryStorage()
b = MemoryStorage()
status = {}
sync(a, b, status)
assert len(status) == 0
assert list(a.list_items()) == []
assert list(b.list_items()) == []
assert not status
assert empty_storage(a)
assert empty_storage(b)
# creation
item = Item('UID:1')
a.upload(item)
sync(a, b, status)
assert list(status) == ['1']
obj, uid, etag = next(b.get_items(['1']))
assert obj.raw == item.raw
obj_a, uid_a, etag_a = a.get('1')
obj_b, uid_b, etag_b = b.get('1')
assert obj_a.raw == obj_b.raw == item.raw
assert uid_a == uid_b == only(status) == '1'
# creation and deletion
item2 = Item('UID:2')
b.upload(item2)
b.delete('1')
b.delete('1', etag_b)
sync(a, b, status)
assert list(status) == ['2']
assert next(a.list_items())[0] == '2'
assert next(b.list_items())[0] == '2'
obj, uid, etag = next(a.get_items(['2']))
assert obj.raw == item2.raw
assert next(a.list())[0] == '2'
assert next(b.list())[0] == '2'
obj2_a, uid2_a, etag2_a = a.get('2')
assert obj2_a.raw == item2.raw
new_item2 = Item('UID:2\nHUEHUEHUE:PRECISELY')
old_status = status.copy()
a.update(new_item2, next(a.list_items())[1])
a.update(new_item2, next(a.list())[1])
sync(a, b, status)
assert status != old_status
assert list(status) == list(old_status)
assert next(a.list_items())[0] == '2'
assert next(b.list_items())[0] == '2'
obj, uid, etag = next(b.get_items(['2']))
assert next(a.list())[0] == '2'
assert next(b.list())[0] == '2'
obj, uid, etag = b.get('2')
assert obj.raw == new_item2.raw