/** * Verify that a query returns the expected list of keys. Ordering is ignored. */ public static <K> void verifyQueryKeys(LuceneQuery<K, Object> query, K... expectedKeys) throws LuceneQueryException { Set<K> expectedKeySet = new HashSet<>(Arrays.asList(expectedKeys)); Set<K> actualKeySet = new HashSet<>(query.findKeys()); assertEquals(expectedKeySet, actualKeySet); }
queryInfo.getRegionPath(), queryInfo.getQueryString(), queryInfo.getDefaultField()); if (queryInfo.getKeysOnly()) { query.findKeys().forEach(key -> result.add(new LuceneSearchResults(key.toString()))); } else { PageableLuceneQueryResults pageableLuceneQueryResults = query.findPages();
private static LuceneQuery mockLuceneQuery(int limit, int pageSize) { LuceneQuery mockLuceneQuery = mock(LuceneQuery.class); when(mockLuceneQuery.getLimit()).thenReturn(limit); when(mockLuceneQuery.getPageSize()).thenReturn(pageSize); ObjectUtils.doOperationSafely(() -> when(mockLuceneQuery.findKeys()) .thenReturn(Collections.emptySet()), Collections.emptySet()); rethrowAsRuntimeException(() -> when(mockLuceneQuery.findPages()) .thenThrow(newUnsupportedOperationException("Operation Not Supported!"))); ObjectUtils.doOperationSafely(() -> when(mockLuceneQuery.findResults()) .thenReturn(Collections.emptyList()), Collections.emptyList()); ObjectUtils.doOperationSafely(() -> when(mockLuceneQuery.findValues()) .thenReturn(Collections.emptyList()), Collections.emptyList()); return mockLuceneQuery; }