From e08f2a941e28449c681c99a29484c26a7d072966 Mon Sep 17 00:00:00 2001
From: discord9 <55937128+discord9@users.noreply.github.com>
Date: Fri, 9 Oct 2026 18:39:39 +0800
Subject: [PATCH] test: verify prepared query options through the real frontend

Signed-off-by: discord9 <55937128+discord9@users.noreply.github.com>
---
 src/servers/tests/postgres/mod.rs | 24 ----------------
 tests-integration/tests/sql.rs    | 47 +++++++++++++++++++++++++++++++
 2 files changed, 47 insertions(+), 24 deletions(-)

diff --git a/src/servers/tests/postgres/mod.rs b/src/servers/tests/postgres/mod.rs
index d71c9912944..2e031c0d0db 100644
--- a/src/servers/tests/postgres/mod.rs
+++ b/src/servers/tests/postgres/mod.rs
@@ -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?;
diff --git a/tests-integration/tests/sql.rs b/tests-integration/tests/sql.rs
index 6c58ab0246c..d055c80bdba 100644
--- a/tests-integration/tests/sql.rs
+++ b/tests-integration/tests/sql.rs
@@ -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.