Exemplos de MockExporter._get_message_headers em Python

Exemplo n.º 1

Arquivo: test_exporter.py Projeto: zuiwanting/rasa

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)

Exemplo n.º 2

Arquivo: test_exporter.py Projeto: zuiwanting/rasa

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