From d7b63b9a831b4514598ee9a989aa69152e656388 Mon Sep 17 00:00:00 2001
From: kailixu <klxu@taosdata.com>
Date: Tue, 4 Jun 2024 16:50:49 +0800
Subject: [PATCH] test: adjust test case for createdb

---
 tests/script/tsim/user/privilege_create_db.sim                  | 2 ++
 tests/system-test/0-others/taosShell.py                         | 1 +
 tests/system-test/0-others/view/non_marterial_view/test_view.py | 1 +
 3 files changed, 4 insertions(+)

diff --git a/tests/script/tsim/user/privilege_create_db.sim b/tests/script/tsim/user/privilege_create_db.sim
index f199e2ee9c..42a527ef8b 100644
--- a/tests/script/tsim/user/privilege_create_db.sim
+++ b/tests/script/tsim/user/privilege_create_db.sim
@@ -6,6 +6,8 @@ sql connect
 print ========================root user create user
 sql create user u1 pass "taosdata"
 sql create user u2 pass "taosdata"
+sql alter user u1 createdb 1
+sql alter user u2 createdb 1
 sql create database test
 sql select * from information_schema.ins_user_privileges where user_name == "root"
 if $rows != 1 then
diff --git a/tests/system-test/0-others/taosShell.py b/tests/system-test/0-others/taosShell.py
index 1227378799..cc6298da2c 100644
--- a/tests/system-test/0-others/taosShell.py
+++ b/tests/system-test/0-others/taosShell.py
@@ -133,6 +133,7 @@ class TDTestCase:
         tdSql.prepare()
         # time.sleep(2)
         tdSql.query("create user testpy pass 'testpy'")
+        tdSql.query("alter user testpy createdb 1")
 
         #hostname = socket.gethostname()
         #tdLog.info ("hostname: %s" % hostname)
diff --git a/tests/system-test/0-others/view/non_marterial_view/test_view.py b/tests/system-test/0-others/view/non_marterial_view/test_view.py
index 4b829b4049..3b6f774788 100644
--- a/tests/system-test/0-others/view/non_marterial_view/test_view.py
+++ b/tests/system-test/0-others/view/non_marterial_view/test_view.py
@@ -80,6 +80,7 @@ class TDTestCase:
         
     def create_user(self, username, password):
         tdSql.execute(f"create user {username} pass '{password}';")
+        tdSql.execute(f"alter user {username} createdb 1;")
         tdLog.debug("Create user {} with password {} successfully".format(username, password))
         
     def check_permissions(self, username, db_name, permission_dict, view_name=None):