New privilege in test

shiyer7474 · shiyer7474 · commit 947d07796292 · 2025-01-28T02:05:34.000Z
diff --git a/src/Access/tests/gtest_access_rights_ops.cpp b/src/Access/tests/gtest_access_rights_ops.cpp
@@ -288,7 +288,7 @@ TEST(AccessRights, Union)
               "SYSTEM DROP REPLICA, SYSTEM SYNC REPLICA, SYSTEM RESTART REPLICA, "
               "SYSTEM RESTORE REPLICA, SYSTEM WAIT LOADING PARTS, SYSTEM SYNC DATABASE REPLICA, SYSTEM FLUSH DISTRIBUTED, "
               "SYSTEM LOAD PRIMARY KEY, SYSTEM UNLOAD PRIMARY KEY, dictGet ON db1.*, GRANT TABLE ENGINE ON db1, "
-              "GRANT CREATE USER, ALTER USER, DROP USER, CREATE ROLE, ALTER ROLE, DROP ROLE, SET DEFINER ON db1, "
+              "GRANT CREATE USER, ALTER USER, DROP USER, CREATE ROLE, ALTER ROLE, DROP ROLE, IMPERSONATE, SET DEFINER ON db1, "
               "GRANT NAMED COLLECTION ADMIN ON db1");
 
     lhs = {};
diff --git a/src/Interpreters/Context.cpp b/src/Interpreters/Context.cpp
@@ -1674,9 +1674,9 @@ void Context::setCurrentRolesDefault()
     setCurrentRolesImpl(user->granted_roles.findGranted(user->default_roles), /* throw_if_not_granted= */ false, /* skip_if_not_granted= */ false, user);
 }
 
-void Context::switchImpersonateUser(const RolesOrUsersSet & users)
+void Context::switchImpersonateUser(const RolesOrUsersSet & new_user)
 {
-    auto new_user_uuid = *(users.ids.begin());
+    auto new_user_uuid = *(new_user.ids.begin());
     setUser(new_user_uuid);
     setCurrentUserName(getUser()->getName());
     setInitialUserName(getUser()->getName()); /// ensure to update the client_info

Original file line number	Diff line number	Diff line change
`@@ -1674,9 +1674,9 @@ void Context::setCurrentRolesDefault()`
`1674`	`1674`	`setCurrentRolesImpl(user->granted_roles.findGranted(user->default_roles), /* throw_if_not_granted= / false, / skip_if_not_granted= */ false, user);`
`1675`	`1675`	`}`
`1676`	`1676`
`1677`		`-void Context::switchImpersonateUser(const RolesOrUsersSet & users)`
	`1677`	`+void Context::switchImpersonateUser(const RolesOrUsersSet & new_user)`
`1678`	`1678`	`{`
`1679`		`- auto new_user_uuid = *(users.ids.begin());`
	`1679`	`+ auto new_user_uuid = *(new_user.ids.begin());`
`1680`	`1680`	`setUser(new_user_uuid);`
`1681`	`1681`	`setCurrentUserName(getUser()->getName());`
`1682`	`1682`	`setInitialUserName(getUser()->getName()); /// ensure to update the client_info`