diff options
-rw-r--r-- | test/test_consumer_group.py | 2 | ||||
-rw-r--r-- | test/test_producer.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/test/test_consumer_group.py b/test/test_consumer_group.py index 885ae83..8f25e9f 100644 --- a/test/test_consumer_group.py +++ b/test/test_consumer_group.py @@ -16,7 +16,7 @@ from test.testutil import random_string def get_connect_str(kafka_broker): - return 'localhost:' + str(kafka_broker.port) + return kafka_broker.host + ':' + str(kafka_broker.port) @pytest.fixture diff --git a/test/test_producer.py b/test/test_producer.py index 54b9db2..1f6608a 100644 --- a/test/test_producer.py +++ b/test/test_producer.py @@ -35,7 +35,7 @@ def test_end_to_end(kafka_broker, compression): elif platform.python_implementation() == 'PyPy': return - connect_str = 'localhost:' + str(kafka_broker.port) + connect_str = ':'.join([kafka_broker.host, str(kafka_broker.port)]) producer = KafkaProducer(bootstrap_servers=connect_str, retries=5, max_block_ms=10000, |