Changeset View
Changeset View
Standalone View
Standalone View
swh/journal/tests/test_replay.py
Show First 20 Lines • Show All 130 Lines • ▼ Show 20 Lines | def _test_write_replay_origin_visit(visits): | ||||
writer.send('origin', 'foo', { | writer.send('origin', 'foo', { | ||||
'url': 'http://example.com/', | 'url': 'http://example.com/', | ||||
'type': 'git', | 'type': 'git', | ||||
}) | }) | ||||
for visit in visits: | for visit in visits: | ||||
writer.send('origin_visit', 'foo', visit) | writer.send('origin_visit', 'foo', visit) | ||||
queue_size = len(queue) | queue_size = len(queue) | ||||
assert replayer.max_messages == 0 | |||||
replayer.max_messages = queue_size | |||||
storage = get_storage('memory', {}) | storage = get_storage('memory', {}) | ||||
worker_fn = functools.partial(process_replay_objects, storage=storage) | worker_fn = functools.partial(process_replay_objects, storage=storage) | ||||
nb_messages = 0 | nb_messages = 0 | ||||
while nb_messages < queue_size: | while nb_messages < queue_size: | ||||
nb_messages += replayer.process(worker_fn) | nb_messages += replayer.process(worker_fn) | ||||
actual_visits = list(storage.origin_visit_get('http://example.com/')) | actual_visits = list(storage.origin_visit_get('http://example.com/')) | ||||
▲ Show 20 Lines • Show All 87 Lines • Show Last 20 Lines |