add Unit Test for modified (#15771)

apache · Mar 27, 2024 · 915843a · 915843a
1 parent b40460e
commit 915843a
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/...scheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/UsersServiceTest.java b/...scheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/UsersServiceTest.java
@@ -392,6 +392,12 @@ public void testGrantProject() {
         logger.info(result.toString());
         Assertions.assertEquals(Status.USER_NOT_EXIST, result.get(Constants.STATUS));
 
+        // ERROR: NO_CURRENT_OPERATING_PERMISSION
+        loginUser.setUserType(UserType.GENERAL_USER);
+        result = this.usersService.grantProject(loginUser, userId, projectIds);
+        logger.info(result.toString());
+        Assertions.assertEquals(Status.NO_CURRENT_OPERATING_PERMISSION, result.get(Constants.STATUS));
+
         // SUCCESS
         when(userMapper.selectById(userId)).thenReturn(getUser());
         result = usersService.grantProject(loginUser, userId, projectIds);
@@ -413,6 +419,12 @@ public void testGrantProjectWithReadPerm() {
         logger.info(result.toString());
         Assertions.assertEquals(Status.USER_NOT_EXIST, result.get(Constants.STATUS));
 
+        // ERROR: NO_CURRENT_OPERATING_PERMISSION
+        loginUser.setUserType(UserType.GENERAL_USER);
+        result = this.usersService.grantProjectWithReadPerm(loginUser, userId, projectIds);
+        logger.info(result.toString());
+        Assertions.assertEquals(Status.NO_CURRENT_OPERATING_PERMISSION, result.get(Constants.STATUS));
+
         // SUCCESS
         when(userMapper.selectById(userId)).thenReturn(getUser());
         result = usersService.grantProjectWithReadPerm(loginUser, userId, projectIds);
@@ -436,6 +448,12 @@ public void testGrantProjectByCode() {
         logger.info(result.toString());
         Assertions.assertEquals(Status.USER_NOT_EXIST, result.get(Constants.STATUS));
 
+        // ERROR: NO_CURRENT_OPERATING_PERMISSION
+        loginUser.setUserType(UserType.GENERAL_USER);
+        result = this.usersService.grantProjectByCode(loginUser, 999, projectCode);
+        logger.info(result.toString());
+        Assertions.assertEquals(Status.NO_CURRENT_OPERATING_PERMISSION, result.get(Constants.STATUS));
+
         // ERROR: PROJECT_NOT_FOUNT
         result = this.usersService.grantProjectByCode(loginUser, authorizer, 999);
         logger.info(result.toString());
@@ -566,6 +584,12 @@ public void testGrantDataSource() {
         logger.info(result.toString());
         Assertions.assertEquals(Status.USER_NOT_EXIST, result.get(Constants.STATUS));
 
+        // ERROR: NO_CURRENT_OPERATING_PERMISSION
+        loginUser.setUserType(UserType.GENERAL_USER);
+        result = this.usersService.grantDataSource(loginUser, userId, datasourceIds);
+        logger.info(result.toString());
+        Assertions.assertEquals(Status.NO_CURRENT_OPERATING_PERMISSION, result.get(Constants.STATUS));
+
         // test admin user
         when(userMapper.selectById(userId)).thenReturn(getUser());
         when(datasourceUserMapper.deleteByUserId(Mockito.anyInt())).thenReturn(1);