def test_get_message_headers_pika_event_broker(): event_broker = Mock(spec=PikaEventBroker) exporter = MockExporter(event_broker=event_broker) # noinspection PyProtectedMember headers = exporter._get_message_headers() assert headers.get(RASA_EXPORT_PROCESS_ID_HEADER_NAME)
def test_get_message_headers_non_pika_broker(): event_broker = Mock() exporter = MockExporter(event_broker=event_broker) # noinspection PyProtectedMember assert exporter._get_message_headers() is None