Changeset View
Changeset View
Standalone View
Standalone View
swh/journal/tests/utils.py
| Show First 20 Lines • Show All 63 Lines • ▼ Show 20 Lines | def subscribe(self, topics): | ||||
| if unknown_topics: | if unknown_topics: | ||||
| raise ValueError('Unknown topics %s' % ', '.join(unknown_topics)) | raise ValueError('Unknown topics %s' % ', '.join(unknown_topics)) | ||||
| class MockedJournalClient(JournalClient): | class MockedJournalClient(JournalClient): | ||||
| def __init__(self, queue, object_types=ACCEPTED_OBJECT_TYPES): | def __init__(self, queue, object_types=ACCEPTED_OBJECT_TYPES): | ||||
| self._object_types = object_types | self._object_types = object_types | ||||
| self.consumer = MockedKafkaConsumer(queue) | self.consumer = MockedKafkaConsumer(queue) | ||||
| self.process_timeout = 0 | self.process_timeout = None | ||||
| self.max_messages = 0 | self.max_messages = None | ||||
| self.value_deserializer = kafka_to_value | self.value_deserializer = kafka_to_value | ||||
| self.stop_on_eof = False | self.stop_on_eof = False | ||||