From 4fcdba2c0c3e7e7da8b4d78b44433741899742b2 Mon Sep 17 00:00:00 2001 From: JiPeng Wang <103574007+WangJPLeo@users.noreply.github.com> Date: Sun, 6 Nov 2022 21:15:27 +0800 Subject: [PATCH] cherry-pick [Bug] Resource default auth function disabled false. --- .../api/service/impl/ResourcesServiceImpl.java | 8 ++++---- .../api/service/ResourcesServiceTest.java | 11 ++++++----- 2 files changed, 10 insertions(+), 9 deletions(-) diff --git a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ResourcesServiceImpl.java b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ResourcesServiceImpl.java index daa0f99b9f..377f90b54b 100644 --- a/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ResourcesServiceImpl.java +++ b/dolphinscheduler-api/src/main/java/org/apache/dolphinscheduler/api/service/impl/ResourcesServiceImpl.java @@ -1503,7 +1503,7 @@ public class ResourcesServiceImpl extends BaseServiceImpl implements ResourcesSe @Override public Map authorizeResourceTree(User loginUser, Integer userId) { Map result = new HashMap<>(); - if (!resourcePermissionCheckService.functionDisabled()) { + if (resourcePermissionCheckService.functionDisabled()) { putMsg(result, Status.FUNCTION_DISABLED); return result; } @@ -1584,7 +1584,7 @@ public class ResourcesServiceImpl extends BaseServiceImpl implements ResourcesSe @Override public Map unauthorizedUDFFunction(User loginUser, Integer userId) { Map result = new HashMap<>(); - if (!resourcePermissionCheckService.functionDisabled()) { + if (resourcePermissionCheckService.functionDisabled()) { putMsg(result, Status.FUNCTION_DISABLED); return result; } @@ -1622,7 +1622,7 @@ public class ResourcesServiceImpl extends BaseServiceImpl implements ResourcesSe @Override public Map authorizedUDFFunction(User loginUser, Integer userId) { Map result = new HashMap<>(); - if (!resourcePermissionCheckService.functionDisabled()) { + if (resourcePermissionCheckService.functionDisabled()) { putMsg(result, Status.FUNCTION_DISABLED); return result; } @@ -1642,7 +1642,7 @@ public class ResourcesServiceImpl extends BaseServiceImpl implements ResourcesSe @Override public Map authorizedFile(User loginUser, Integer userId) { Map result = new HashMap<>(); - if (!resourcePermissionCheckService.functionDisabled()) { + if (resourcePermissionCheckService.functionDisabled()) { putMsg(result, Status.FUNCTION_DISABLED); return result; } diff --git a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ResourcesServiceTest.java b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ResourcesServiceTest.java index 5443926e1a..1b951d6068 100644 --- a/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ResourcesServiceTest.java +++ b/dolphinscheduler-api/src/test/java/org/apache/dolphinscheduler/api/service/ResourcesServiceTest.java @@ -831,7 +831,7 @@ public class ResourcesServiceTest { // test admin user List resIds = new ArrayList<>(); resIds.add(1); - Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(true); + Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(false); Mockito.when(resourcesMapper.queryResourceExceptUserId(userId)).thenReturn(getResourceList()); Map result = resourcesService.authorizeResourceTree(user, userId); logger.info(result.toString()); @@ -889,7 +889,7 @@ public class ResourcesServiceTest { int userId = 3; // test admin user - Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(true); + Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(false); Mockito.when(udfFunctionMapper.queryUdfFuncExceptUserId(userId)).thenReturn(getUdfFuncList()); Mockito.when(udfFunctionMapper.queryAuthedUdfFunc(userId)).thenReturn(getSingleUdfFuncList()); Map result = resourcesService.unauthorizedUDFFunction(user, userId); @@ -916,7 +916,7 @@ public class ResourcesServiceTest { int userId = 3; // test admin user - Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(true); + Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(false); Mockito.when(udfFunctionMapper.queryAuthedUdfFunc(userId)).thenReturn(getUdfFuncList()); Map result = resourcesService.authorizedUDFFunction(user, userId); logger.info(result.toString()); @@ -945,8 +945,9 @@ public class ResourcesServiceTest { // test admin user List resIds = new ArrayList<>(); resIds.add(1); - Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(true); - Mockito.when(resourceUserMapper.queryResourcesIdListByUserIdAndPerm(Mockito.anyInt(), Mockito.anyInt())).thenReturn(resIds); + Mockito.when(resourcePermissionCheckService.functionDisabled()).thenReturn(false); + Mockito.when(resourceUserMapper.queryResourcesIdListByUserIdAndPerm(Mockito.anyInt(), Mockito.anyInt())) + .thenReturn(resIds); Mockito.when(resourcesMapper.queryResourceListById(Mockito.any())).thenReturn(getResourceList()); Map result = resourcesService.authorizedFile(user, userId); logger.info(result.toString());