def test_missing_mapped_destination_tasks_src_not_complete():
""" Tests expected behaviours on mapped tasks that are missing in
the destination.
"""
src = random_task()
src.completed = False
src_tasks = [src]
src_svc = MockTaskService(src_tasks)
dst = random_task()
dst_tasks = []
dst_svc = MockTaskService(dst_tasks)
map = TaskMap()
# create the pre-existing mapping
map.map(src, dst)
# preconditions
assert len(map.get_all_src_keys()) == 1
assert map.get_dst_id(src.id) == dst.id
assert map.get_src_id(dst.id) == src.id
TaskSync(src_svc, dst_svc, map).synchronise()
assert dst.id != dst_svc.tasks[0].id
assert len(map.get_all_src_keys()) == 1, "should still be just one mapping"
assert not map.try_get_src_id(dst.id), "old dst should be unmapped"
assert map.get_dst_id(src.id) != dst.id, "src should be mapped to something else"
assert dst_svc.tasks[0].status == SyncStatus.new, "should be flagged as a new task"
assert len(dst_svc.tasks) == 1
def test_missing_mapped_destination_tasks_src_complete():
""" Tests expected behaviours on mapped tasks that are missing in
the destination.
"""
src = random_task()
src.completed = True
src_tasks = [src]
src_svc = MockTaskService(src_tasks)
dst = random_task()
dst_tasks = []
dst_svc = MockTaskService(dst_tasks)
map = TaskMap()
# create the pre-existing mapping
map.map(src, dst)
# preconditions
assert len(map.get_all_src_keys()) == 1
assert map.get_dst_id(src.id) == dst.id
assert map.get_src_id(dst.id) == src.id
TaskSync(src_svc, dst_svc, map).synchronise()
assert len(dst_svc.tasks) == 0
def test_persist_task_mapping(self):
expected = TaskMap()
tasks = [MockTask(_id=i) for i in range(4)]
expected.map(tasks[0], tasks[1])
expected.map(tasks[2], tasks[3])
filename = NamedTemporaryFile(suffix='.pickle')
expected.persist(filename.name)
actual = TaskMap(filename.name)
assert actual.get_dst_id(tasks[0].id) == tasks[1].id
assert actual.get_dst_id(tasks[2].id) == tasks[3].id
assert actual.get_src_id(tasks[1].id) == tasks[0].id
assert actual.get_src_id(tasks[3].id) == tasks[2].id
def test_new_tasks_are_mapped():
src_tasks = [random_task()]
dst_tasks = []
src = MockTaskService(src_tasks)
dst = MockTaskService(dst_tasks)
map = TaskMap()
sync = TaskSync(src, dst, map)
# preconditions
assert len(map.get_all_src_keys()) == 0
sync.synchronise()
assert len(map.get_all_src_keys()) == 1
assert src_tasks[0].id in map.get_all_src_keys()
assert map.get_dst_id(src_tasks[0].id) == dst_tasks[0].id
def test_persist_task_mapping(self):
expected = TaskMap()
tasks = [
MockTask(_id='aaa'),
MockTask(_id='bbb'),
MockTask(_id='ccc'),
MockTask(_id='ddd')]
expected.map(tasks[0], tasks[1])
expected.map(tasks[2], tasks[3])
filename = NamedTemporaryFile(suffix='.tm')
expected.persist(filename.name)
actual = TaskMap(filename.name)
assert actual.get_dst_id(tasks[0].id) == tasks[1].id
assert actual.get_dst_id(tasks[2].id) == tasks[3].id
assert actual.get_src_id(tasks[1].id) == tasks[0].id
assert actual.get_src_id(tasks[3].id) == tasks[2].id
def test_persist_task_mapping(self):
expected = TaskMap()
tasks = [
MockTask(_id='aaa'),
MockTask(_id='bbb'),
MockTask(_id='ccc'),
MockTask(_id='ddd')
]
expected.map(tasks[0], tasks[1])
expected.map(tasks[2], tasks[3])
filename = NamedTemporaryFile(suffix='.tm')
expected.persist(filename.name)
actual = TaskMap(filename.name)
assert actual.get_dst_id(tasks[0].id) == tasks[1].id
assert actual.get_dst_id(tasks[2].id) == tasks[3].id
assert actual.get_src_id(tasks[1].id) == tasks[0].id
assert actual.get_src_id(tasks[3].id) == tasks[2].id
def test_remove_orphan_mappings():
src_tasks = [random_task()]
dst_tasks = []
ss = MockTaskService(src_tasks)
ds = MockTaskService(dst_tasks)
map = TaskMap()
# add a few task mappings that won't exist in either source or destination
map.map(random_task(), random_task())
map.map(random_task(), random_task())
map.map(random_task(), random_task())
TaskSync(ss, ds, map).synchronise(clean_orphans=True)
# We now expect just one mapping for the new src task
all_mappings = map.get_all_src_keys()
assert len(all_mappings) == 1
assert map.get_dst_id(src_tasks[0].id)
class TestTaskMap(object):
def setup(self):
self.tm = TaskMap()
self.src = MockTask(_id='1')
self.dst = MockTask(_id='a')
self.missing = MockTask(_id='blah')
def test_create_when_file_doesnt_exist(self):
tmpfile = NamedTemporaryFile(suffix='.pickle')
name = tmpfile.name
tmpfile.close()
tm = TaskMap(name)
assert tm
def test_persist_task_mapping(self):
expected = TaskMap()
tasks = [
MockTask(_id='aaa'),
MockTask(_id='bbb'),
MockTask(_id='ccc'),
MockTask(_id='ddd')
]
expected.map(tasks[0], tasks[1])
expected.map(tasks[2], tasks[3])
filename = NamedTemporaryFile(suffix='.tm')
expected.persist(filename.name)
actual = TaskMap(filename.name)
assert actual.get_dst_id(tasks[0].id) == tasks[1].id
assert actual.get_dst_id(tasks[2].id) == tasks[3].id
assert actual.get_src_id(tasks[1].id) == tasks[0].id
assert actual.get_src_id(tasks[3].id) == tasks[2].id
def test_duplicate_src(self):
s = MockTask('1')
d = MockTask('a')
dd = MockTask('aa')
self.tm.map(s, d)
with pytest.raises(KeyDuplicationError):
self.tm.map(s, dd)
def test_duplicate_dst(self):
src2 = MockTask(_id='9')
self.tm.map(self.src, self.dst)
with pytest.raises(ValueDuplicationError):
self.tm.map(src2, self.dst)
def test_duplicate_src_dst(self):
tasks = [MockTask(_id=i) for i in range(4)]
self.tm.map(tasks[0], tasks[1])
self.tm.map(tasks[2], tasks[3])
with pytest.raises(KeyAndValueDuplicationError):
# both src and dst are already mapped to something
self.tm.map(tasks[0], tasks[3])
def test_valid_forward(self):
self.tm.map(self.src, self.dst)
assert self.tm.get_dst_id(self.src.id) == self.dst.id
def test_invalid_forward(self):
with pytest.raises(KeyError):
self.tm.get_dst_id(self.missing)
def test_valid_reverse(self):
self.tm.map(self.src, self.dst)
assert self.tm.get_src_id(self.dst.id) == self.src.id
def test_invalid_reverse(self):
with pytest.raises(KeyError):
self.tm.get_src_id(self.missing.id)
def test_valid_forward_try_get(self):
self.tm.map(self.src, self.dst)
assert self.tm.try_get_dst_id(self.src.id) == self.dst.id
def test_invalid_forward_try_get(self):
assert not self.tm.try_get_dst_id(self.missing.id)
def test_valid_reverse_try_get(self):
self.tm.map(self.src, self.dst)
assert self.tm.try_get_src_id(self.dst.id) == self.src.id
def test_invalid_reverse_try_get(self):
assert not self.tm.try_get_src_id(self.missing.id)
def test_get_all_src_keys(self):
src_keys = (1, 2, 3, 4)
src_tasks = [MockTask(_id=i) for i in src_keys]
dst_tasks = [MockTask(_id=i + 20) for i in range(4)]
map = TaskMap()
for s, d in zip(src_tasks, dst_tasks):
map.map(s, d)
for actual in map.get_all_src_keys():
assert actual in src_keys
def test_get_all_dst_keys(self):
src_tasks = [MockTask(_id=i + 20) for i in range(4)]
dst_keys = (1, 2, 3, 4)
dst_tasks = [MockTask(_id=i) for i in dst_keys]
map = TaskMap()
for s, d in zip(src_tasks, dst_tasks):
map.map(s, d)
for actual in map.get_all_dst_keys():
assert actual in dst_keys
def test_delete_mapping(self):
self.tm.map(self.src, self.dst)
self.tm.unmap(self.src.id)
assert not self.tm.try_get_dst_id(self.src.id)
assert not self.tm.try_get_src_id(self.dst.id)
class TestTaskMap(object):
def setup(self):
self.tm = TaskMap()
self.src = MockTask(_id='1')
self.dst = MockTask(_id='a')
self.missing = MockTask(_id='blah')
def test_create_when_file_doesnt_exist(self):
tmpfile = NamedTemporaryFile(suffix='.pickle')
name = tmpfile.name
tmpfile.close()
tm = TaskMap(name)
assert tm
def test_persist_task_mapping(self):
expected = TaskMap()
tasks = [
MockTask(_id='aaa'),
MockTask(_id='bbb'),
MockTask(_id='ccc'),
MockTask(_id='ddd')]
expected.map(tasks[0], tasks[1])
expected.map(tasks[2], tasks[3])
filename = NamedTemporaryFile(suffix='.tm')
expected.persist(filename.name)
actual = TaskMap(filename.name)
assert actual.get_dst_id(tasks[0].id) == tasks[1].id
assert actual.get_dst_id(tasks[2].id) == tasks[3].id
assert actual.get_src_id(tasks[1].id) == tasks[0].id
assert actual.get_src_id(tasks[3].id) == tasks[2].id
def test_duplicate_src(self):
s = MockTask('1')
d = MockTask('a')
dd = MockTask('aa')
self.tm.map(s, d)
with pytest.raises(KeyDuplicationError):
self.tm.map(s, dd)
def test_duplicate_dst(self):
src2 = MockTask(_id='9')
self.tm.map(self.src, self.dst)
with pytest.raises(ValueDuplicationError):
self.tm.map(src2, self.dst)
def test_duplicate_src_dst(self):
tasks = [MockTask(_id=i) for i in range(4)]
self.tm.map(tasks[0], tasks[1])
self.tm.map(tasks[2], tasks[3])
with pytest.raises(KeyAndValueDuplicationError):
# both src and dst are already mapped to something
self.tm.map(tasks[0], tasks[3])
def test_valid_forward(self):
self.tm.map(self.src, self.dst)
assert self.tm.get_dst_id(self.src.id) == self.dst.id
def test_invalid_forward(self):
with pytest.raises(KeyError):
self.tm.get_dst_id(self.missing)
def test_valid_reverse(self):
self.tm.map(self.src, self.dst)
assert self.tm.get_src_id(self.dst.id) == self.src.id
def test_invalid_reverse(self):
with pytest.raises(KeyError):
self.tm.get_src_id(self.missing.id)
def test_valid_forward_try_get(self):
self.tm.map(self.src, self.dst)
assert self.tm.try_get_dst_id(self.src.id) == self.dst.id
def test_invalid_forward_try_get(self):
assert not self.tm.try_get_dst_id(self.missing.id)
def test_valid_reverse_try_get(self):
self.tm.map(self.src, self.dst)
assert self.tm.try_get_src_id(self.dst.id) == self.src.id
def test_invalid_reverse_try_get(self):
assert not self.tm.try_get_src_id(self.missing.id)
def test_get_all_src_keys(self):
src_keys = (1,2,3,4)
src_tasks = [MockTask(_id=i) for i in src_keys]
dst_tasks = [MockTask(_id=i+20) for i in range(4)]
map = TaskMap()
for s,d in zip(src_tasks, dst_tasks):
map.map(s,d)
for actual in map.get_all_src_keys():
assert actual in src_keys
def test_get_all_dst_keys(self):
src_tasks = [MockTask(_id=i+20) for i in range(4)]
dst_keys = (1,2,3,4)
dst_tasks = [MockTask(_id=i) for i in dst_keys]
map = TaskMap()
for s,d in zip(src_tasks, dst_tasks):
map.map(s,d)
for actual in map.get_all_dst_keys():
assert actual in dst_keys
def test_delete_mapping(self):
self.tm.map(self.src, self.dst)
self.tm.unmap(self.src.id)
assert not self.tm.try_get_dst_id(self.src.id)
assert not self.tm.try_get_src_id(self.dst.id)
