Update session_test.go

auxten · web-flow · commit 54e2bdd43a67 · 2025-02-17T13:07:55.000+08:00
diff --git a/chdb/session_test.go b/chdb/session_test.go
@@ -53,7 +53,9 @@ func TestSessionCleanup(t *testing.T) {
 }
 
 func TestQuery(t *testing.T) {
-	session, _ := NewSession()
+	path := filepath.Join(os.TempDir(), "chdb_test")
+	defer os.RemoveAll(path)
+	session, _ := NewSession(path)
 	defer session.Cleanup()
 
 	session.Query("CREATE TABLE IF NOT EXISTS testtable (id UInt32) ENGINE = MergeTree() ORDER BY id;")

Original file line number	Diff line number	Diff line change
`@@ -53,7 +53,9 @@ func TestSessionCleanup(t *testing.T) {`
`53`	`53`	`}`
`54`	`54`
`55`	`55`	`func TestQuery(t *testing.T) {`
`56`		`- session, _ := NewSession()`
	`56`	`+ path := filepath.Join(os.TempDir(), "chdb_test")`
	`57`	`+ defer os.RemoveAll(path)`
	`58`	`+ session, _ := NewSession(path)`
`57`	`59`	`defer session.Cleanup()`
`58`	`60`
`59`	`61`	`session.Query("CREATE TABLE IF NOT EXISTS testtable (id UInt32) ENGINE = MergeTree() ORDER BY id;")`