Repository: incubator-ranger Updated Branches: refs/heads/master d673e5940 -> 376d32497
http://git-wip-us.apache.org/repos/asf/incubator-ranger/blob/376d3249/security-admin/src/test/java/org/apache/ranger/service/TestRangerServiceServiceBase.java ---------------------------------------------------------------------- diff --git a/security-admin/src/test/java/org/apache/ranger/service/TestRangerServiceServiceBase.java b/security-admin/src/test/java/org/apache/ranger/service/TestRangerServiceServiceBase.java new file mode 100644 index 0000000..f359118 --- /dev/null +++ b/security-admin/src/test/java/org/apache/ranger/service/TestRangerServiceServiceBase.java @@ -0,0 +1,221 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.ranger.service; + +import java.util.ArrayList; +import java.util.Date; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +import javax.persistence.EntityManager; + +import org.apache.ranger.biz.RangerBizUtil; +import org.apache.ranger.common.ContextUtil; +import org.apache.ranger.common.RangerSearchUtil; +import org.apache.ranger.common.SearchField; +import org.apache.ranger.common.UserSessionBase; +import org.apache.ranger.common.db.BaseDao; +import org.apache.ranger.db.RangerDaoManager; +import org.apache.ranger.db.XXServiceDefDao; +import org.apache.ranger.entity.XXService; +import org.apache.ranger.entity.XXServiceDef; +import org.apache.ranger.plugin.model.RangerService; +import org.apache.ranger.plugin.util.SearchFilter; +import org.apache.ranger.security.context.RangerContextHolder; +import org.apache.ranger.security.context.RangerSecurityContext; +import org.apache.ranger.view.RangerServiceList; +import org.junit.Assert; +import org.junit.FixMethodOrder; +import org.junit.Rule; +import org.junit.Test; +import org.junit.rules.ExpectedException; +import org.junit.runner.RunWith; +import org.junit.runners.MethodSorters; +import org.mockito.InjectMocks; +import org.mockito.Mock; +import org.mockito.Mockito; +import org.mockito.runners.MockitoJUnitRunner; + +@RunWith(MockitoJUnitRunner.class) +@FixMethodOrder(MethodSorters.NAME_ASCENDING) +public class TestRangerServiceServiceBase { + + private static Long Id = 8L; + + @InjectMocks + RangerServiceService rangerServiceService = new RangerServiceService(); + + @Mock + RangerDaoManager daoManager; + + @Mock + RangerSearchUtil searchUtil; + + @Mock + RangerBizUtil bizUtil; + + @Rule + public ExpectedException thrown = ExpectedException.none(); + + public void setup() { + RangerSecurityContext context = new RangerSecurityContext(); + context.setUserSession(new UserSessionBase()); + RangerContextHolder.setSecurityContext(context); + UserSessionBase currentUserSession = ContextUtil + .getCurrentUserSession(); + currentUserSession.setUserAdmin(true); + } + + private RangerService rangerService() { + Map<String, String> configs = new HashMap<String, String>(); + configs.put("username", "servicemgr"); + configs.put("password", "servicemgr"); + configs.put("namenode", "servicemgr"); + configs.put("hadoop.security.authorization", "No"); + configs.put("hadoop.security.authentication", "Simple"); + configs.put("hadoop.security.auth_to_local", ""); + configs.put("dfs.datanode.kerberos.principal", ""); + configs.put("dfs.namenode.kerberos.principal", ""); + configs.put("dfs.secondary.namenode.kerberos.principal", ""); + configs.put("hadoop.rpc.protection", "Privacy"); + configs.put("commonNameForCertificate", ""); + + RangerService rangerService = new RangerService(); + rangerService.setId(Id); + rangerService.setConfigs(configs); + rangerService.setCreateTime(new Date()); + rangerService.setDescription("service"); + rangerService.setGuid("serviceguid"); + rangerService.setIsEnabled(true); + rangerService.setName("Hdfs service"); + rangerService.setPolicyUpdateTime(new Date()); + rangerService.setPolicyVersion(1L); + rangerService.setType(null); + rangerService.setUpdatedBy("Admin"); + rangerService.setUpdateTime(new Date()); + rangerService.setVersion(Id); + + return rangerService; + } + + private XXService service() { + XXService xService = new XXService(); + xService.setAddedByUserId(Id); + xService.setCreateTime(new Date()); + xService.setDescription("Hdfs service"); + xService.setGuid("serviceguid"); + xService.setId(Id); + xService.setIsEnabled(true); + xService.setName("Hdfs"); + xService.setPolicyUpdateTime(new Date()); + xService.setPolicyVersion(1L); + xService.setType(1L); + xService.setUpdatedByUserId(Id); + xService.setUpdateTime(new Date()); + xService.setVersion(1L); + + return xService; + } + + @Test + public void test1MapViewToEntityBean() { + XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class); + XXServiceDef xServiceDef = Mockito.mock(XXServiceDef.class); + RangerService rangerService = rangerService(); + XXService service = service(); + int OPERATION_CONTEXT = 1; + + Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao); + Mockito.when(xServiceDefDao.findByName(rangerService.getType())) + .thenReturn(xServiceDef); + + XXService dbService = rangerServiceService.mapViewToEntityBean( + rangerService, service, OPERATION_CONTEXT); + Assert.assertNotNull(dbService); + Assert.assertEquals(dbService, service); + Assert.assertEquals(dbService.getDescription(), + service.getDescription()); + Assert.assertEquals(dbService.getGuid(), service.getGuid()); + Assert.assertEquals(dbService.getName(), service.getName()); + Assert.assertEquals(dbService.getAddedByUserId(), + service.getAddedByUserId()); + Assert.assertEquals(dbService.getId(), service.getId()); + Assert.assertEquals(dbService.getVersion(), service.getVersion()); + Assert.assertEquals(dbService.getType(), service.getType()); + Assert.assertEquals(dbService.getUpdatedByUserId(), + service.getUpdatedByUserId()); + + Mockito.verify(daoManager).getXXServiceDef(); + } + + @Test + public void test2mapEntityToViewBean() { + XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class); + XXServiceDef xServiceDef = Mockito.mock(XXServiceDef.class); + RangerService rangerService = rangerService(); + XXService service = service(); + + Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao); + Mockito.when(xServiceDefDao.getById(service.getType())).thenReturn( + xServiceDef); + + RangerService dbRangerService = rangerServiceService + .mapEntityToViewBean(rangerService, service); + Assert.assertNotNull(dbRangerService); + Assert.assertEquals(dbRangerService, rangerService); + Assert.assertEquals(dbRangerService.getDescription(), + rangerService.getDescription()); + Assert.assertEquals(dbRangerService.getGuid(), rangerService.getGuid()); + Assert.assertEquals(dbRangerService.getName(), rangerService.getName()); + Assert.assertEquals(dbRangerService.getId(), rangerService.getId()); + Assert.assertEquals(dbRangerService.getVersion(), + rangerService.getVersion()); + Assert.assertEquals(dbRangerService.getType(), rangerService.getType()); + + Mockito.verify(daoManager).getXXServiceDef(); + } + + @Test + public void test3searchRangerServices() { + EntityManager entityManager = Mockito.mock(EntityManager.class); + SearchFilter searchFilter = new SearchFilter(); + searchFilter.setParam(SearchFilter.POLICY_NAME, "policyName"); + searchFilter.setParam(SearchFilter.SERVICE_NAME, "serviceName"); + + String searchString = "policyName"; + String sortString = "asc"; + List<SearchField> searchFieldList = new ArrayList<SearchField>(); + boolean isCountQuery = false; + + BaseDao baseDao = Mockito.mock(BaseDao.class); + Mockito.when(daoManager.getDaoForClassName(Mockito.anyString())) + .thenReturn(baseDao); + Mockito.when(daoManager.getEntityManager()).thenReturn(entityManager); + Mockito.when( + searchUtil + .createSearchQuery(entityManager, searchString, + sortString, searchFilter, searchFieldList, + isCountQuery)).thenReturn(null); + Mockito.when(bizUtil.getClassType(getClass())).thenReturn(1); + RangerServiceList dbRangerServiceList = rangerServiceService + .searchRangerServices(searchFilter); + Assert.assertNotNull(dbRangerServiceList); + Mockito.verify(daoManager).getDaoForClassName(Mockito.anyString()); + + } +}
