Search in sources :

Example 21 with VXLong

use of org.apache.ranger.view.VXLong in project ranger by apache.

the class TestXUserREST method test87countXUserPermission.

@SuppressWarnings("unchecked")
@Test
public void test87countXUserPermission() {
    HttpServletRequest request = Mockito.mock(HttpServletRequest.class);
    SearchCriteria testSearchCriteria = createsearchCriteria();
    Mockito.when(searchUtil.extractCommonCriterias((HttpServletRequest) Mockito.any(), (List<SortField>) Mockito.any())).thenReturn(testSearchCriteria);
    vXLong.setValue(1);
    Mockito.when(xUserMgr.getXUserPermissionSearchCount(testSearchCriteria)).thenReturn(vXLong);
    VXLong testvxLong = xUserRest.countXUserPermission(request);
    Mockito.verify(xUserMgr).getXUserPermissionSearchCount(testSearchCriteria);
    Mockito.verify(searchUtil).extractCommonCriterias((HttpServletRequest) Mockito.any(), (List<SortField>) Mockito.any());
    assertNotNull(testvxLong);
    assertEquals(testvxLong.getValue(), vXLong.getValue());
    assertEquals(testvxLong.getClass(), vXLong.getClass());
}
Also used : HttpServletRequest(javax.servlet.http.HttpServletRequest) VXLong(org.apache.ranger.view.VXLong) SortField(org.apache.ranger.common.SortField) SearchCriteria(org.apache.ranger.common.SearchCriteria) Test(org.junit.Test)

Example 22 with VXLong

use of org.apache.ranger.view.VXLong in project ranger by apache.

the class AbstractBaseResourceService method getSearchCountUsingCriteria.

// -------------Criteria Usage--------------------
// -----------------------------------------------
public VXLong getSearchCountUsingCriteria(SearchCriteria searchCriteria, List<SearchField> searchFieldList) {
    EntityManager em = getDao().getEntityManager();
    CriteriaBuilder criteriaBuilder = em.getCriteriaBuilder();
    CriteriaQuery<Long> criteria = criteriaBuilder.createQuery(Long.class);
    Root<T> from = criteria.from(tEntityClass);
    Expression<Long> countExpression = criteriaBuilder.count(from.get("id"));
    criteria.select(countExpression);
    Predicate resourceConditions = buildResourceSpecificConditions(criteriaBuilder, from, searchCriteria);
    Predicate userConditions = buildUserConditions(searchCriteria.getParamList(), searchFieldList, criteriaBuilder, from);
    if (resourceConditions != null) {
        criteria.where(criteriaBuilder.and(resourceConditions, userConditions));
    } else {
        criteria.where(criteriaBuilder.and(userConditions));
    }
    TypedQuery<Long> countQuery = em.createQuery(criteria);
    long count = getDao().executeCountQueryInSecurityContext(tEntityClass, countQuery);
    VXLong vXLong = new VXLong();
    vXLong.setValue(count);
    return vXLong;
}
Also used : CriteriaBuilder(javax.persistence.criteria.CriteriaBuilder) EntityManager(javax.persistence.EntityManager) VXLong(org.apache.ranger.view.VXLong) VXLong(org.apache.ranger.view.VXLong) Predicate(javax.persistence.criteria.Predicate)

Example 23 with VXLong

use of org.apache.ranger.view.VXLong in project ranger by apache.

the class AbstractBaseResourceService method getSearchCount.

public VXLong getSearchCount(SearchCriteria searchCriteria, List<SearchField> searchFieldList) {
    long count = getCountForSearchQuery(searchCriteria, searchFieldList);
    VXLong vXLong = new VXLong();
    vXLong.setValue(count);
    return vXLong;
}
Also used : VXLong(org.apache.ranger.view.VXLong)

Aggregations

VXLong (org.apache.ranger.view.VXLong)23 Test (org.junit.Test)18 HttpServletRequest (javax.servlet.http.HttpServletRequest)12 SearchCriteria (org.apache.ranger.common.SearchCriteria)12 SortField (org.apache.ranger.common.SortField)12 ArrayList (java.util.ArrayList)5 RangerService (org.apache.ranger.plugin.model.RangerService)4 SearchFilter (org.apache.ranger.plugin.util.SearchFilter)4 RangerPolicy (org.apache.ranger.plugin.model.RangerPolicy)2 EntityManager (javax.persistence.EntityManager)1 CriteriaBuilder (javax.persistence.criteria.CriteriaBuilder)1 Predicate (javax.persistence.criteria.Predicate)1 VXAsset (org.apache.ranger.view.VXAsset)1 VXAssetList (org.apache.ranger.view.VXAssetList)1 VXAuditMap (org.apache.ranger.view.VXAuditMap)1 VXAuditMapList (org.apache.ranger.view.VXAuditMapList)1 VXPermMap (org.apache.ranger.view.VXPermMap)1 VXPermMapList (org.apache.ranger.view.VXPermMapList)1 VXPolicy (org.apache.ranger.view.VXPolicy)1 VXPolicyList (org.apache.ranger.view.VXPolicyList)1