def test_publish_with_headers_non_pika_event_broker():
event_broker = Mock(SQLEventBroker)
exporter = MockExporter(event_broker=event_broker)
headers = {"some": "header"}
event = {"some": "event"}
# noinspection PyProtectedMember
exporter._publish_with_message_headers(event, headers)
# the `SQLEventBroker`'s `publish()` method was called with only the `event`
# argument
event_broker.publish.assert_called_with(event)
def test_publish_with_headers_pika_event_broker():
event_broker = Mock(spec=PikaEventBroker)
exporter = MockExporter(event_broker=event_broker)
headers = {"some": "header"}
event = {"some": "event"}
# noinspection PyProtectedMember
exporter._publish_with_message_headers(event, headers)
# the `PikaEventBroker`'s `publish()` method was called with both
# the `event` and `headers` arguments
event_broker.publish.assert_called_with(event=event, headers=headers)
