fix: cean up created users

4paradigm · May 31, 2024 · a2770ec · a2770ec
1 parent cb7661a
commit a2770ec
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/cmd/sql_cmd_test.cc b/src/cmd/sql_cmd_test.cc
@@ -338,6 +338,10 @@ TEST_P(DBSDKTest, TestGrantCreateUser) {
  router->ExecuteSQL(absl::StrCat("DROP USER user2"), &status);
  ASSERT_FALSE(status.IsOK());
  }
+ sr->ExecuteSQL(absl::StrCat("DROP USER IF EXISTS user1"), &status);
+ ASSERT_TRUE(status.IsOK());
+ sr->ExecuteSQL(absl::StrCat("DROP USER IF EXISTS user2"), &status);
+ ASSERT_TRUE(status.IsOK());
 }
 
 TEST_P(DBSDKTest, TestGrantCreateUserGrantOption) {
@@ -384,6 +388,10 @@ TEST_P(DBSDKTest, TestGrantCreateUserGrantOption) {
  router->ExecuteSQL(absl::StrCat("GRANT CREATE USER ON *.* TO 'user2@%'"), &status);
  ASSERT_TRUE(status.IsOK());
  }
+ sr->ExecuteSQL(absl::StrCat("DROP USER IF EXISTS user1"), &status);
+ ASSERT_TRUE(status.IsOK());
+ sr->ExecuteSQL(absl::StrCat("DROP USER IF EXISTS user2"), &status);
+ ASSERT_TRUE(status.IsOK());
 }
 
 TEST_P(DBSDKTest, CreateDatabase) {