chore(driver): test format_null_as_str with set (#478)

databendlabs · Aug 28, 2024 · 5c922c3 · 5c922c3
1 parent 0e13616
commit 5c922c3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/driver/tests/driver/select_simple.rs b/driver/tests/driver/select_simple.rs
@@ -48,8 +48,9 @@ async fn select_null() {
         let dsn = option_env!("TEST_DATABEND_DSN").unwrap_or(DEFAULT_DSN);
         // ignore null to str test for flightsql
         if !dsn.starts_with("databend+flight://") {
-            let client = Client::new(format!("{dsn}&format_null_as_str=1"));
+            let client = Client::new(dsn.to_string());
             let conn = client.get_conn().await.unwrap();
+            conn.exec("SET format_null_as_str=1").await.unwrap();
             let row = conn.query_row("select * from select_null").await.unwrap();
             assert!(row.is_some());
             let row = row.unwrap();
@@ -62,8 +63,9 @@ async fn select_null() {
     }
     {
         let dsn = option_env!("TEST_DATABEND_DSN").unwrap_or(DEFAULT_DSN);
-        let client = Client::new(format!("{dsn}&format_null_as_str=0"));
+        let client = Client::new(dsn.to_string());
         let conn = client.get_conn().await.unwrap();
+        conn.exec("SET format_null_as_str=0").await.unwrap();
         let row = conn.query_row("select * from select_null").await.unwrap();
         assert!(row.is_some());
         let row = row.unwrap();

diff --git a/tests/docker-compose.yaml b/tests/docker-compose.yaml
@@ -6,7 +6,7 @@ services:
     volumes:
       - ./data:/data
   databend:
-    image: docker.io/datafuselabs/databend:nightly
+    image: docker.io/datafuselabs/databend
     environment:
       - QUERY_STORAGE_TYPE=s3
       - QUERY_DATABEND_ENTERPRISE_LICENSE