Changeset View
Changeset View
Standalone View
Standalone View
swh/journal/tests/test_kafka_writer.py
Show First 20 Lines • Show All 222 Lines • ▼ Show 20 Lines | def test_write_BufferError_give_up(kafka_prefix: str, kafka_server: str, caplog): | ||||
) | ) | ||||
writer.producer.n_buffererrors = 5 | writer.producer.n_buffererrors = 5 | ||||
empty_dir = Directory(entries=()) | empty_dir = Directory(entries=()) | ||||
with pytest.raises(KafkaDeliveryError): | with pytest.raises(KafkaDeliveryError): | ||||
writer.write_addition("directory", empty_dir) | writer.write_addition("directory", empty_dir) | ||||
def test_write_addition_errors_without_unique_key(kafka_prefix: str, kafka_server: str): | |||||
writer = KafkaJournalWriter[BaseModel]( | |||||
brokers=[kafka_server], | |||||
client_id="kafka_writer", | |||||
prefix=kafka_prefix, | |||||
value_sanitizer=model_object_dict_sanitizer, | |||||
) | |||||
with pytest.raises(NotImplementedError): | |||||
writer.write_addition("BaseModel", BaseModel()) |