fix

Author: Caideyipi

iotdb-core/confignode/src/test/java/org/apache/iotdb/confignode/manager/pipe/source/PipeConfigTreePrivilegeParseVisitorTest.java

@@ -25,6 +25,7 @@
 import org.apache.iotdb.commons.exception.MetadataException;
 import org.apache.iotdb.commons.utils.StatusUtils;
 import org.apache.iotdb.confignode.consensus.request.ConfigPhysicalPlanType;
+import org.apache.iotdb.confignode.consensus.request.write.auth.AuthorTreePlan;
 import org.apache.iotdb.confignode.consensus.request.write.database.DatabaseSchemaPlan;
 import org.apache.iotdb.confignode.manager.ConfigManager;
 import org.apache.iotdb.confignode.manager.PermissionManager;
@@ -96,6 +97,49 @@ ConfigPhysicalPlanType.CreateDatabase, new TDatabaseSchema("root.db1")),
     Assert.assertTrue(skipVisitor.canReadSysSchema("root.db", null, false));
   }
 
+  @Test
+  public void testAuthPrivilege() {
+    permissionManager.setUserPrivilege(
+        (userName, privilegeUnion) ->
+            privilegeUnion.getPrivilegeType() == PrivilegeType.MANAGE_USER);
+    Assert.assertTrue(
+        skipVisitor
+            .visitGrantUser(new AuthorTreePlan(ConfigPhysicalPlanType.GrantUser), null)
+            .isPresent());
+    Assert.assertTrue(
+        skipVisitor
+            .visitRevokeUser(new AuthorTreePlan(ConfigPhysicalPlanType.RevokeUser), null)
+            .isPresent());
+    Assert.assertFalse(
+        skipVisitor
+            .visitGrantRole(new AuthorTreePlan(ConfigPhysicalPlanType.GrantRole), null)
+            .isPresent());
+    Assert.assertFalse(
+        skipVisitor
+            .visitRevokeRole(new AuthorTreePlan(ConfigPhysicalPlanType.RevokeRole), null)
+            .isPresent());
+
+    permissionManager.setUserPrivilege(
+        (userName, privilegeUnion) ->
+            privilegeUnion.getPrivilegeType() == PrivilegeType.MANAGE_ROLE);
+    Assert.assertFalse(
+        skipVisitor
+            .visitGrantUser(new AuthorTreePlan(ConfigPhysicalPlanType.GrantUser), null)
+            .isPresent());
+    Assert.assertFalse(
+        skipVisitor
+            .visitRevokeUser(new AuthorTreePlan(ConfigPhysicalPlanType.RevokeUser), null)
+            .isPresent());
+    Assert.assertTrue(
+        skipVisitor
+            .visitGrantRole(new AuthorTreePlan(ConfigPhysicalPlanType.GrantRole), null)
+            .isPresent());
+    Assert.assertTrue(
+        skipVisitor
+            .visitRevokeRole(new AuthorTreePlan(ConfigPhysicalPlanType.RevokeRole), null)
+            .isPresent());
+  }
+
   private static class TestPermissionManager extends PermissionManager {
 
     private BiFunction<String, PrivilegeUnion, Boolean> checkUserPrivileges =