fix case

2022-05-21 17:50:10 +08:00 · 2022-05-21 17:50:10 +08:00 · 0288faac7f
parent 1a8917cb5f
commit 0288faac7f
1 changed files with 2 additions and 1 deletions
--- a/tests/system-test/0-others/user_control.py
+++ b/tests/system-test/0-others/user_control.py
@ -311,7 +311,8 @@ class TDTestCase:
            tdSql.error(sql)
        tdSql.query(sql)
        self.__change_user_priv(user=user, pre_priv=priv, invoke=True)
-        user.db_set.remove(dbname)
+        if user.name != "root":
+            user.db_set.remove(dbname) if dbname else user.db_set.clear()
        time.sleep(1)

    def test_priv_change_current(self):