Changeset View
Changeset View
Standalone View
Standalone View
swh/journal/tests/test_kafka_writer.py
| Show First 20 Lines • Show All 144 Lines • ▼ Show 20 Lines | for (object_type, (_, objects)) in OBJECT_TYPE_KEYS.items(): | ||||
| ] | ] | ||||
| method(objects_) | method(objects_) | ||||
| expected_messages += len(objects) | expected_messages += len(objects) | ||||
| elif object_type in ('origin_visit',): | elif object_type in ('origin_visit',): | ||||
| for object_ in objects: | for object_ in objects: | ||||
| object_ = object_.copy() | object_ = object_.copy() | ||||
| origin_url = object_.pop('origin') | origin_url = object_.pop('origin') | ||||
| storage.origin_add_one(Origin(url=origin_url)) | storage.origin_add_one(Origin(url=origin_url)) | ||||
| visit = method(origin=origin_url, date=object_.pop('date'), | visit = method(origin_url, date=object_.pop('date'), | ||||
| type=object_.pop('type')) | type=object_.pop('type')) | ||||
| expected_messages += 1 | expected_messages += 1 | ||||
| visit_id = visit['visit'] | storage.origin_visit_update(origin_url, visit.visit, **object_) | ||||
| storage.origin_visit_update(origin_url, visit_id, **object_) | |||||
| expected_messages += 1 | expected_messages += 1 | ||||
| else: | else: | ||||
| assert False, object_type | assert False, object_type | ||||
| assert_written(consumer, kafka_prefix, expected_messages) | assert_written(consumer, kafka_prefix, expected_messages) | ||||