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)