test_kafka.py

# coding: utf-8

"""Test message logging end to end."""

from __future__ import (
    absolute_import, division, print_function, unicode_literals
)


from kafka import KafkaConsumer, KafkaProducer

def test_kafka_pass_through():
    consumer = KafkaConsumer(bootstrap_servers=['127.0.0.1:9092'])
    consumer.subscribe('test_topic')

    producer = KafkaProducer(bootstrap_servers=['127.0.0.1:9092'])
    future = producer.send('test_topic', b'some_message_bytes')
    future.get(timeout=10)
    producer.flush()
    print('produced')
    # blocks
    msg = next(consumer)
    print('consumed', msg)