From 6eb79ea8eec27d05bbafbdad30b7d5b19da3ddbd Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Mon, 15 Jan 2024 13:56:33 +0800
Subject: [PATCH] fix: add user db

---
 tests/pytest/util/autogen.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/pytest/util/autogen.py b/tests/pytest/util/autogen.py
index 4a09395f51..72e64dc448 100644
--- a/tests/pytest/util/autogen.py
+++ b/tests/pytest/util/autogen.py
@@ -114,6 +114,7 @@ class AutoGen:
     def create_db(self, dbname, vgroups = 2, replica = 1):
         self.dbname  = dbname
         tdSql.execute(f'create database {dbname} vgroups {vgroups} replica {replica}')
+        tdSql.execute(f"use {dbname}")
         
     # create table or stable
     def create_stable(self, stbname, tag_cnt, column_cnt, binary_len, nchar_len):