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
