def test_build_fails_with_message_reader_and_serializer( dummy_message_reader: DummyMessageReader, string_message_serializer: MessageSerializer ): builder = PipelineBuilder() builder.with_message_reader(dummy_message_reader) builder.with_input_message_serializer(string_message_serializer) with pytest.raises(EsqueIOInvalidPipelineBuilderState): builder.build()
def test_build_fails_with_message_reader_and_handler( dummy_message_reader: DummyMessageReader, dummy_handler: DummyHandler ): builder = PipelineBuilder() builder.with_message_reader(dummy_message_reader) builder.with_input_handler(dummy_handler) with pytest.raises(EsqueIOInvalidPipelineBuilderState): builder.build()
def test_create_pipeline_with_message_reader( dummy_message_reader: DummyMessageReader, binary_messages: List[BinaryMessage], string_messages: List[Message] ): builder = PipelineBuilder() builder.with_message_reader(dummy_message_reader) builder.with_stream_decorator(skip_stream_events) pipeline = builder.build() dummy_message_reader.set_messages(binary_messages) assert list(pipeline.decorated_message_stream()) == string_messages
def prepared_builder( dummy_message_reader: DummyMessageReader, dummy_message_writer: DummyMessageWriter, binary_messages: List[BinaryMessage], ) -> PipelineBuilder: builder = PipelineBuilder() builder.with_message_reader(dummy_message_reader) builder.with_message_writer(dummy_message_writer) dummy_message_reader.set_messages(binary_messages) return builder
def test_build_fails_with_message_reader_and_uri( dummy_message_reader: DummyMessageReader, string_message_serializer: MessageSerializer, dummy_handler: DummyHandler, input_uri: str, ): builder = PipelineBuilder() builder.with_message_reader(dummy_message_reader) builder.with_input_from_uri(input_uri) with pytest.raises(EsqueIOInvalidPipelineBuilderState): builder.build()