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());
}
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;
}
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;
}
Aggregations