thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-20 08:40:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #22380 from kssenii/fix-rabbit-test

Browse Source 
Try fix flaky rabbitmq test
This commit is contained in:
tavplubix 2021-04-01 15:44:45 +03:00 committed by GitHub
commit bbae136a1e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/integration/test_storage_rabbitmq/test.py
View File

@ -253,12 +253,20 @@ def test_rabbitmq_csv_with_delimiter(rabbitmq_cluster):
@pytest.mark.timeout(240) @pytest.mark.timeout(240)
def test_rabbitmq_tsv_with_delimiter(rabbitmq_cluster): def test_rabbitmq_tsv_with_delimiter(rabbitmq_cluster):
instance.query(''' instance.query('''
DROP TABLE IF EXISTS test.view;
DROP TABLE IF EXISTS test.consumer;
CREATE TABLE test.rabbitmq (key UInt64, value UInt64) CREATE TABLE test.rabbitmq (key UInt64, value UInt64)
ENGINE = RabbitMQ ENGINE = RabbitMQ
SETTINGS rabbitmq_host_port = 'rabbitmq1:5672', SETTINGS rabbitmq_host_port = 'rabbitmq1:5672',
rabbitmq_exchange_name = 'tsv', rabbitmq_exchange_name = 'tsv',
rabbitmq_format = 'TSV', rabbitmq_format = 'TSV',
rabbitmq_queue_base = 'tsv',
rabbitmq_row_delimiter = '\\n'; rabbitmq_row_delimiter = '\\n';
CREATE TABLE test.view (key UInt64, value UInt64)
ENGINE = MergeTree()
ORDER BY key;
CREATE MATERIALIZED VIEW test.consumer TO test.view AS
SELECT * FROM test.rabbitmq;
''') ''')
credentials = pika.PlainCredentials('root', 'clickhouse') credentials = pika.PlainCredentials('root', 'clickhouse')
@ -272,13 +280,11 @@ def test_rabbitmq_tsv_with_delimiter(rabbitmq_cluster):
for message in messages: for message in messages:
channel.basic_publish(exchange='tsv', routing_key='', body=message) channel.basic_publish(exchange='tsv', routing_key='', body=message)
connection.close() connection.close()
time.sleep(1)
result = '' result = ''
while True: while True:
result += instance.query('SELECT * FROM test.rabbitmq ORDER BY key', ignore_error=True) result = instance.query('SELECT * FROM test.view ORDER BY key')
if rabbitmq_check_result(result): if rabbitmq_check_result(result):
break break