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

Add test cases

Browse Source 
Signed-off-by: Frank Chen <frank.chen021@outlook.com>
This commit is contained in:
Frank Chen 2022-09-13 18:28:36 +08:00
parent d3265150c0
commit 0c37c95c22
2 changed files with 87 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/queries/0_stateless/02421_simple_queries_for_opentelemetry.reference Normal file
View File

@ -0,0 +1,4 @@
{"query":"show processlist format Null\n "}
{"query":"show databases format Null\n "}
{"query":"insert into opentelemetry_test values","read_rows":"3","read_bytes":"24","written_rows":"3","written_bytes":"24"}
{"query":"select * from opentelemetry_test format Null\n ","read_rows":"3","read_bytes":"24","written_rows":"","written_bytes":""}

83
tests/queries/0_stateless/02421_simple_queries_for_opentelemetry.sh Executable file
View File

@ -0,0 +1,83 @@
#!/usr/bin/env bash
CURDIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
. "$CURDIR"/../shell_config.sh
# This function takes 2 arguments:
# $1 - query
# $2 - trace id
function executeQuery()
{
${CLICKHOUSE_CLIENT} --database=${CLICKHOUSE_DATABASE} --opentelemetry_start_trace_probability=1 --query_id $1 -nq "
${2}
"
}
# For some quries, it's not able to know how many bytes/rows are read when tests are executed on CI,
# so we only to check the db.statement only
function check_query_span_query_only()
{
${CLICKHOUSE_CLIENT} -nq "
SYSTEM FLUSH LOGS;
SELECT attribute['db.statement'] as query
FROM system.opentelemetry_span_log
WHERE finish_date >= yesterday()
AND operation_name = 'query'
AND attribute['clickhouse.query_id'] = '${1}'
Format JSONEachRow
;"
}
function check_query_span()
{
${CLICKHOUSE_CLIENT} -nq "
SYSTEM FLUSH LOGS;
SELECT attribute['db.statement'] as query,
attribute['clickhouse.read_rows'] as read_rows,
attribute['clickhouse.read_bytes'] as read_bytes,
attribute['clickhouse.written_rows'] as written_rows,
attribute['clickhouse.written_bytes'] as written_bytes
FROM system.opentelemetry_span_log
WHERE finish_date >= yesterday()
AND operation_name = 'query'
AND attribute['clickhouse.query_id'] = '${1}'
Format JSONEachRow
;"
}
#
# Set up
#
${CLICKHOUSE_CLIENT} --database=${CLICKHOUSE_DATABASE} -nq "
DROP TABLE IF EXISTS opentelemetry_test;
CREATE TABLE opentelemetry_test (id UInt64) Engine=MergeTree Order By id;
"
# test 1, a query that has special path in the code
# Format Null is used to make sure no output is generated so that it won't pollute the reference file
query_id=$(${CLICKHOUSE_CLIENT} -q "select generateUUIDv4()");
executeQuery $query_id 'show processlist format Null'
check_query_span_query_only "$query_id"
# test 2, a normal show command
query_id=$(${CLICKHOUSE_CLIENT} -q "select generateUUIDv4()");
executeQuery $query_id 'show databases format Null'
check_query_span_query_only "$query_id"
# test 3, a normal insert query on local table
query_id=$(${CLICKHOUSE_CLIENT} -q "select generateUUIDv4()");
executeQuery $query_id 'insert into opentelemetry_test values(1)(2)(3)'
check_query_span "$query_id"
# test 4, a normal select query
query_id=$(${CLICKHOUSE_CLIENT} -q "select generateUUIDv4()");
executeQuery $query_id 'select * from opentelemetry_test format Null'
check_query_span $query_id
#
# Tear down
#
${CLICKHOUSE_CLIENT} --database=${CLICKHOUSE_DATABASE} -q "
DROP TABLE IF EXISTS opentelemetry_test;
"