mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-17 13:13:36 +00:00
111 lines
3.3 KiB
Python
111 lines
3.3 KiB
Python
import os.path as p
|
|
import time
|
|
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
from helpers.test_tools import TSV
|
|
|
|
import json
|
|
import subprocess
|
|
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
instance = cluster.add_instance('instance',
|
|
main_configs=['configs/kafka.xml'],
|
|
with_kafka=True)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
instance.query('CREATE DATABASE test')
|
|
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def kafka_is_available(started_cluster):
|
|
p = subprocess.Popen(('docker',
|
|
'exec',
|
|
'-i',
|
|
started_cluster.kafka_docker_id,
|
|
'/usr/bin/kafka-broker-api-versions',
|
|
'--bootstrap-server',
|
|
'PLAINTEXT://localhost:9092'),
|
|
stdout=subprocess.PIPE)
|
|
p.communicate()[0]
|
|
return p.returncode == 0
|
|
|
|
|
|
def kafka_produce(started_cluster, topic, messages):
|
|
p = subprocess.Popen(('docker',
|
|
'exec',
|
|
'-i',
|
|
started_cluster.kafka_docker_id,
|
|
'/usr/bin/kafka-console-producer',
|
|
'--broker-list',
|
|
'localhost:9092',
|
|
'--topic',
|
|
topic),
|
|
stdin=subprocess.PIPE)
|
|
p.communicate(messages)
|
|
p.stdin.close()
|
|
|
|
|
|
def kafka_check_json_numbers(instance):
|
|
retries = 0
|
|
while True:
|
|
if kafka_is_available(started_cluster):
|
|
break
|
|
else:
|
|
retries += 1
|
|
if retries > 50:
|
|
raise 'Cannot connect to kafka.'
|
|
print("Waiting for kafka to be available...")
|
|
time.sleep(1)
|
|
messages = ''
|
|
for i in xrange(50):
|
|
messages += json.dumps({'key': i, 'value': i}) + '\n'
|
|
kafka_produce(started_cluster, 'json', messages)
|
|
time.sleep(3)
|
|
result = instance.query('SELECT * FROM test.kafka;')
|
|
file = p.join(p.dirname(__file__), 'test_kafka_json.reference')
|
|
with open(file) as reference:
|
|
assert TSV(result) == TSV(reference)
|
|
|
|
|
|
def test_kafka_json(started_cluster):
|
|
instance.query('''
|
|
DROP TABLE IF EXISTS test.kafka;
|
|
CREATE TABLE test.kafka (key UInt64, value UInt64)
|
|
ENGINE = Kafka('kafka1:9092', 'json', 'json',
|
|
'JSONEachRow', '\\n');
|
|
''')
|
|
kafka_check_json_numbers(instance)
|
|
instance.query('DROP TABLE test.kafka')
|
|
|
|
|
|
def test_kafka_json_settings(started_cluster):
|
|
instance.query('''
|
|
DROP TABLE IF EXISTS test.kafka;
|
|
CREATE TABLE test.kafka (key UInt64, value UInt64)
|
|
ENGINE = Kafka
|
|
SETTINGS
|
|
kafka_broker_list = 'kafka1:9092',
|
|
kafka_topic_list = 'json'
|
|
kafka_group_name = 'json'
|
|
kafka_format = 'JSONEachRow'
|
|
kafka_row_delimiter = '\\n';
|
|
''')
|
|
kafka_check_json_numbers(instance)
|
|
instance.query('DROP TABLE test.kafka')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cluster.start()
|
|
raw_input("Cluster created, press any key to destroy...")
|
|
cluster.shutdown()
|