kafka_prefix = 'fgnsukippl', kafka_server = '127.0.0.1:36889' consumer = <cimpl.Consumer object at 0x7f6a5f98a8c8> def test_storage_direct_writer(kafka_prefix: str, kafka_server, consumer: Consumer): writer_config = { "cls": "kafka", "brokers": [kafka_server], "client_id": "kafka_writer", "prefix": kafka_prefix, "anonymize": False, } storage_config = { "cls": "pipeline", "steps": [{"cls": "memory", "journal_writer": writer_config},], } storage = get_storage(**storage_config) expected_messages = 0 for obj_type, objs in TEST_OBJECTS.items(): method = getattr(storage, obj_type + "_add") if obj_type in ( "content", "skipped_content", "directory", "revision", "release", "snapshot", "origin", "origin_visit_status", ): method(objs) expected_messages += len(objs) elif obj_type in ("origin_visit",): for obj in objs: assert isinstance(obj, OriginVisit) storage.origin_add_one(Origin(url=obj.origin)) > visit = method(obj.origin, date=obj.date, type=obj.type) E TypeError: origin_visit_add() got an unexpected keyword argument 'date' .tox/py3/lib/python3.7/site-packages/swh/storage/tests/test_kafka_writer.py:56: TypeError