rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-10-12 15:02:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: verify prepared query options through the real frontend

Browse Source 
Signed-off-by: discord9 <55937128+discord9@users.noreply.github.com>
This commit is contained in:
discord9
2026-10-09 18:39:39 +08:00
parent 84b7449d88
commit e08f2a941e
2 changed files with 47 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/servers/tests/postgres/mod.rs
-24
View File
@@ -551,30 +551,6 @@ async fn test_extended_query_regproc_response() -> Result<()> {
Ok(())
}
#[tokio::test]
async fn test_extended_query_show_query_option() -> Result<()> {
let server_port = start_test_server(TlsOption::default()).await?;
let client = create_connection_with_given_db(server_port, DEFAULT_SCHEMA_NAME)
.await
.unwrap();
let simple_rows = client.simple_query("SHOW query.parallelism").await.unwrap();
let simple_value = simple_rows
.iter()
.find_map(|message| match message {
SimpleQueryMessage::Row(row) => row.get(0),
_ => None,
})
.expect("SHOW query.parallelism returns a row");
let statement = client.prepare("SHOW query.parallelism").await.unwrap();
let rows = client.query(&statement, &[]).await.unwrap();
assert_eq!(rows.len(), 1);
assert_eq!(rows[0].len(), 1);
assert_eq!(rows[0].get::<usize, String>(0), simple_value);
Ok(())
}
#[tokio::test]
async fn test_extended_query() -> Result<()> {
let server_port = start_test_server(TlsOption::default()).await?;
tests-integration/tests/sql.rs
+47
View File
@@ -88,6 +88,7 @@ macro_rules! sql_tests {
test_postgres_regclass_bind_parameter,
test_postgres_uint64_parameter,
test_postgres_explain_bind_parameter,
test_postgres_show_query_options,
test_postgres_array_types,
test_mysql_prepare_stmt_insert_timestamp,
test_mysql_prepare_stmt_timezone,
@@ -1873,6 +1874,52 @@ pub async fn test_postgres_extended_query_row_returning_statements(store_type: S
guard.remove_all().await;
}
pub async fn test_postgres_show_query_options(store_type: StorageType) {
let (mut guard, fe_pg_server) =
setup_pg_server(store_type, "test_postgres_show_query_options").await;
let addr = fe_pg_server.bind_addr().unwrap().to_string();
let (client, connection) = tokio_postgres::connect(&format!("postgres://{addr}/public"), NoTls)
.await
.unwrap();
let (tx, rx) = tokio::sync::oneshot::channel();
tokio::spawn(async move {
connection.await.unwrap();
tx.send(()).unwrap();
});
for (option, value) in [
("query.parallelism", "2"),
("datafusion.optimizer.prefer_hash_join", "false"),
] {
client
.batch_execute(&format!("SET {option} TO {value}"))
.await
.unwrap();
let statement = client.prepare(&format!("SHOW {option}")).await.unwrap();
let rows = client.query(&statement, &[]).await.unwrap();
assert_eq!(rows.len(), 1);
assert_eq!(rows[0].len(), 1);
assert_eq!(rows[0].get::<usize, String>(0), value);
if option == "query.parallelism" {
client
.batch_execute("SET query.parallelism TO 7")
.await
.unwrap();
let rows = client.query(&statement, &[]).await.unwrap();
assert_eq!(rows.len(), 1);
assert_eq!(rows[0].len(), 1);
assert_eq!(rows[0].get::<usize, String>(0), "7");
}
}
drop(client);
rx.await.unwrap();
let _ = fe_pg_server.shutdown().await;
guard.remove_all().await;
}
pub async fn test_postgres_explain_bind_parameter(store_type: StorageType) {
// Regression test for #8029: EXPLAIN / EXPLAIN ANALYZE must accept bind
// parameters over the Postgres extended query protocol.