Search in sources :

Example 21 with XXServiceDefDao

use of org.apache.ranger.db.XXServiceDefDao in project ranger by apache.

the class TestServiceREST method test4getServiceDefById.

@Test
public void test4getServiceDefById() throws Exception {
    RangerServiceDef rangerServiceDef = rangerServiceDef();
    XXServiceDef xServiceDef = serviceDef();
    XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class);
    Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao);
    Mockito.when(xServiceDefDao.getById(Id)).thenReturn(xServiceDef);
    Mockito.when(!bizUtil.hasAccess(xServiceDef, null)).thenReturn(true);
    Mockito.when(svcStore.getServiceDef(rangerServiceDef.getId())).thenReturn(rangerServiceDef);
    RangerServiceDef dbRangerServiceDef = serviceREST.getServiceDef(rangerServiceDef.getId());
    Assert.assertNotNull(dbRangerServiceDef);
    Assert.assertEquals(dbRangerServiceDef.getId(), rangerServiceDef.getId());
    Mockito.verify(svcStore).getServiceDef(rangerServiceDef.getId());
    Mockito.verify(daoManager).getXXServiceDef();
    Mockito.verify(bizUtil).hasAccess(xServiceDef, null);
}
Also used : XXServiceDef(org.apache.ranger.entity.XXServiceDef) XXServiceDefDao(org.apache.ranger.db.XXServiceDefDao) RangerServiceDef(org.apache.ranger.plugin.model.RangerServiceDef) Test(org.junit.Test)

Example 22 with XXServiceDefDao

use of org.apache.ranger.db.XXServiceDefDao in project ranger by apache.

the class TestServiceREST method test16createPolicyFalse.

@Test
public void test16createPolicyFalse() throws Exception {
    RangerPolicy rangerPolicy = rangerPolicy();
    RangerServiceDef rangerServiceDef = rangerServiceDef();
    List<RangerPolicy> policies = new ArrayList<RangerPolicy>();
    RangerPolicy rangPolicy = new RangerPolicy();
    policies.add(rangPolicy);
    String userName = "admin";
    Set<String> userGroupsList = new HashSet<String>();
    userGroupsList.add("group1");
    userGroupsList.add("group2");
    ServicePolicies servicePolicies = new ServicePolicies();
    servicePolicies.setServiceId(Id);
    servicePolicies.setServiceName("Hdfs_1");
    servicePolicies.setPolicyVersion(1L);
    servicePolicies.setPolicyUpdateTime(new Date());
    servicePolicies.setServiceDef(rangerServiceDef);
    servicePolicies.setPolicies(policies);
    List<RangerAccessTypeDef> rangerAccessTypeDefList = new ArrayList<RangerServiceDef.RangerAccessTypeDef>();
    RangerAccessTypeDef rangerAccessTypeDefObj = new RangerAccessTypeDef();
    rangerAccessTypeDefObj.setLabel("Read");
    rangerAccessTypeDefObj.setName("read");
    rangerAccessTypeDefObj.setRbKeyLabel(null);
    rangerAccessTypeDefList.add(rangerAccessTypeDefObj);
    XXServiceDef xServiceDef = serviceDef();
    XXService xService = xService();
    XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class);
    XXServiceDao xServiceDao = Mockito.mock(XXServiceDao.class);
    Mockito.when(validatorFactory.getPolicyValidator(svcStore)).thenReturn(policyValidator);
    Mockito.when(bizUtil.isAdmin()).thenReturn(true);
    Mockito.when(bizUtil.getCurrentUserLoginId()).thenReturn(userName);
    Mockito.when(daoManager.getXXService()).thenReturn(xServiceDao);
    Mockito.when(xServiceDao.findByName(Mockito.anyString())).thenReturn(xService);
    Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao);
    Mockito.when(xServiceDefDao.getById(xService.getType())).thenReturn(xServiceDef);
    Mockito.when(svcStore.createPolicy((RangerPolicy) Mockito.any())).thenReturn(rangPolicy);
    RangerPolicy dbRangerPolicy = serviceREST.createPolicy(rangerPolicy, null);
    Assert.assertNotNull(dbRangerPolicy);
    Mockito.verify(bizUtil, Mockito.times(2)).isAdmin();
    Mockito.verify(validatorFactory).getPolicyValidator(svcStore);
    Mockito.verify(daoManager).getXXService();
    Mockito.verify(daoManager).getXXServiceDef();
}
Also used : XXServiceDef(org.apache.ranger.entity.XXServiceDef) ServicePolicies(org.apache.ranger.plugin.util.ServicePolicies) ArrayList(java.util.ArrayList) VXString(org.apache.ranger.view.VXString) XXServiceDao(org.apache.ranger.db.XXServiceDao) Date(java.util.Date) RangerAccessTypeDef(org.apache.ranger.plugin.model.RangerServiceDef.RangerAccessTypeDef) RangerPolicy(org.apache.ranger.plugin.model.RangerPolicy) XXServiceDefDao(org.apache.ranger.db.XXServiceDefDao) RangerServiceDef(org.apache.ranger.plugin.model.RangerServiceDef) XXService(org.apache.ranger.entity.XXService) HashSet(java.util.HashSet) Test(org.junit.Test)

Example 23 with XXServiceDefDao

use of org.apache.ranger.db.XXServiceDefDao in project ranger by apache.

the class TestServiceREST method test18deletePolicyFalse.

@Test
public void test18deletePolicyFalse() throws Exception {
    RangerPolicy rangerPolicy = rangerPolicy();
    Mockito.when(validatorFactory.getPolicyValidator(svcStore)).thenReturn(policyValidator);
    String userName = "admin";
    Set<String> userGroupsList = new HashSet<String>();
    userGroupsList.add("group1");
    userGroupsList.add("group2");
    List<RangerAccessTypeDef> rangerAccessTypeDefList = new ArrayList<RangerServiceDef.RangerAccessTypeDef>();
    RangerAccessTypeDef rangerAccessTypeDefObj = new RangerAccessTypeDef();
    rangerAccessTypeDefObj.setLabel("Read");
    rangerAccessTypeDefObj.setName("read");
    rangerAccessTypeDefObj.setRbKeyLabel(null);
    rangerAccessTypeDefList.add(rangerAccessTypeDefObj);
    XXServiceDef xServiceDef = serviceDef();
    XXService xService = xService();
    XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class);
    XXServiceDao xServiceDao = Mockito.mock(XXServiceDao.class);
    Mockito.when(validatorFactory.getPolicyValidator(svcStore)).thenReturn(policyValidator);
    Mockito.when(bizUtil.isAdmin()).thenReturn(true);
    Mockito.when(bizUtil.getCurrentUserLoginId()).thenReturn(userName);
    Mockito.when(svcStore.getPolicy(Id)).thenReturn(rangerPolicy);
    Mockito.when(daoManager.getXXService()).thenReturn(xServiceDao);
    Mockito.when(xServiceDao.findByName(Mockito.anyString())).thenReturn(xService);
    Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao);
    Mockito.when(xServiceDefDao.getById(xService.getType())).thenReturn(xServiceDef);
    serviceREST.deletePolicy(rangerPolicy.getId());
    Mockito.verify(validatorFactory).getPolicyValidator(svcStore);
}
Also used : RangerAccessTypeDef(org.apache.ranger.plugin.model.RangerServiceDef.RangerAccessTypeDef) XXServiceDef(org.apache.ranger.entity.XXServiceDef) RangerPolicy(org.apache.ranger.plugin.model.RangerPolicy) XXServiceDefDao(org.apache.ranger.db.XXServiceDefDao) RangerServiceDef(org.apache.ranger.plugin.model.RangerServiceDef) ArrayList(java.util.ArrayList) VXString(org.apache.ranger.view.VXString) XXService(org.apache.ranger.entity.XXService) XXServiceDao(org.apache.ranger.db.XXServiceDao) HashSet(java.util.HashSet) Test(org.junit.Test)

Example 24 with XXServiceDefDao

use of org.apache.ranger.db.XXServiceDefDao in project ranger by apache.

the class TestServiceREST method test9deleteService.

@Test
public void test9deleteService() throws Exception {
    RangerService rangerService = rangerService();
    XXServiceDef xServiceDef = serviceDef();
    XXService xService = xService();
    XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class);
    XXServiceDao xServiceDao = Mockito.mock(XXServiceDao.class);
    Mockito.when(validatorFactory.getServiceValidator(svcStore)).thenReturn(serviceValidator);
    Mockito.when(daoManager.getXXService()).thenReturn(xServiceDao);
    Mockito.when(xServiceDao.getById(Id)).thenReturn(xService);
    Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao);
    Mockito.when(xServiceDefDao.getById(xService.getType())).thenReturn(xServiceDef);
    serviceREST.deleteService(rangerService.getId());
    Mockito.verify(validatorFactory).getServiceValidator(svcStore);
    Mockito.verify(daoManager).getXXService();
    Mockito.verify(daoManager).getXXServiceDef();
}
Also used : XXServiceDef(org.apache.ranger.entity.XXServiceDef) XXServiceDefDao(org.apache.ranger.db.XXServiceDefDao) RangerService(org.apache.ranger.plugin.model.RangerService) XXService(org.apache.ranger.entity.XXService) XXServiceDao(org.apache.ranger.db.XXServiceDao) Test(org.junit.Test)

Example 25 with XXServiceDefDao

use of org.apache.ranger.db.XXServiceDefDao in project ranger by apache.

the class TestServiceREST method test3deleteServiceDef.

@Test
public void test3deleteServiceDef() throws Exception {
    HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
    RangerServiceDef rangerServiceDef = rangerServiceDef();
    XXServiceDef xServiceDef = serviceDef();
    XXServiceDefDao xServiceDefDao = Mockito.mock(XXServiceDefDao.class);
    Mockito.when(validatorFactory.getServiceDefValidator(svcStore)).thenReturn(serviceDefValidator);
    Mockito.when(daoManager.getXXServiceDef()).thenReturn(xServiceDefDao);
    Mockito.when(xServiceDefDao.getById(Id)).thenReturn(xServiceDef);
    serviceREST.deleteServiceDef(rangerServiceDef.getId(), request);
    Mockito.verify(validatorFactory).getServiceDefValidator(svcStore);
    Mockito.verify(daoManager).getXXServiceDef();
}
Also used : HttpServletRequest(javax.servlet.http.HttpServletRequest) XXServiceDef(org.apache.ranger.entity.XXServiceDef) XXServiceDefDao(org.apache.ranger.db.XXServiceDefDao) RangerServiceDef(org.apache.ranger.plugin.model.RangerServiceDef) Test(org.junit.Test)

Aggregations

XXServiceDefDao (org.apache.ranger.db.XXServiceDefDao)25 XXServiceDef (org.apache.ranger.entity.XXServiceDef)25 Test (org.junit.Test)24 XXService (org.apache.ranger.entity.XXService)17 RangerService (org.apache.ranger.plugin.model.RangerService)16 XXServiceDao (org.apache.ranger.db.XXServiceDao)13 RangerServiceDef (org.apache.ranger.plugin.model.RangerServiceDef)8 ArrayList (java.util.ArrayList)7 WebApplicationException (javax.ws.rs.WebApplicationException)6 ServiceTags (org.apache.ranger.plugin.util.ServiceTags)6 ExpectedException (org.junit.rules.ExpectedException)6 VXString (org.apache.ranger.view.VXString)5 Date (java.util.Date)4 HashSet (java.util.HashSet)4 XXServiceVersionInfoDao (org.apache.ranger.db.XXServiceVersionInfoDao)4 XXServiceVersionInfo (org.apache.ranger.entity.XXServiceVersionInfo)4 RangerPolicy (org.apache.ranger.plugin.model.RangerPolicy)4 RangerAccessTypeDef (org.apache.ranger.plugin.model.RangerServiceDef.RangerAccessTypeDef)4 HttpServletRequest (javax.servlet.http.HttpServletRequest)3 XXPortalUserDao (org.apache.ranger.db.XXPortalUserDao)3