ci: add test for variable. (#137)

datafuselabs · Aug 28, 2024 · bf97a20 · bf97a20
1 parent 67f935d
commit bf97a20
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/session_test.go b/tests/session_test.go
@@ -83,3 +83,15 @@ func (s *DatabendTestSuite) TestSessionConfig() {
 	r.Nil(err)
 	r.Equal(int64(0), result)
 }
+
+func (s *DatabendTestSuite) TestSessionVariable() {
+	r := require.New(s.T())
+
+	var result int64
+
+	_, err := s.db.Exec("set variable a = 100")
+	r.Nil(err)
+	err = s.db.QueryRow("select $a").Scan(&result)
+	r.Nil(err)
+	r.Equal(int64(100), result)
+}