From 35f566bcb62e4ab839f7543eef011bc92aa37836 Mon Sep 17 00:00:00 2001
From: Yang Xiufeng <yangxiufeng.c@gmail.com>
Date: Wed, 14 Aug 2024 19:23:52 +0800
Subject: [PATCH] ci: add test for variable.

---
 .../src/test/java/com/databend/jdbc/TestBasicDriver.java    | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java
index ff2c3e90..cdcf153b 100644
--- a/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java
+++ b/databend-jdbc/src/test/java/com/databend/jdbc/TestBasicDriver.java
@@ -303,6 +303,12 @@ public void testUpdateSession()
             DatabendSession session = connection.unwrap(DatabendConnection.class).getSession();
             Assert.assertEquals(session.getDatabase(), "test_basic_driver_2");
             Assert.assertEquals(session.getSettings().get("max_threads"), "1");
+
+            // test variable
+            connection.createStatement().execute("set variable a=1");
+            ResultSet r = connection.createStatement().executeQuery("select $a");
+            r.next();
+            assertEquals(r.getInt(1), 1);
         }
     }