From 3e0e2004550afd2c7b73100a014625e2ae45b61e Mon Sep 17 00:00:00 2001 From: Yang Xiufeng Date: Wed, 14 Aug 2024 19:31:11 +0800 Subject: [PATCH] ci: add test for variable. --- tests/session_test.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/session_test.go b/tests/session_test.go index 9b9658e..c6537dd 100644 --- 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) +}