def test_with_handler(self, mocker, pipeline, message):
# Set the handler
mock_handler = mocker.MagicMock()
pipeline.on_input_message_received = mock_handler
assert mock_handler.call_count == 0
# Create the event
input_name = "some_input"
input_message_event = pipeline_events_iothub.InputMessageEvent(input_name, message)
# Trigger the event
pipeline._pipeline.on_pipeline_event_handler(input_message_event)
assert mock_handler.call_count == 1
assert mock_handler.call_args == mocker.call(input_name, message)
def test_no_handler(self, pipeline, message):
input_name = "some_input"
input_message_event = pipeline_events_iothub.InputMessageEvent(
input_name, message)
pipeline._pipeline.on_pipeline_event_handler(input_message_event)
