@Test public void shouldCreateQueryWithCorrectAttributes() { Cache cache = mock(Cache.class); Region region = mock(Region.class); when(cache.getRegion(any())).thenReturn(region); LuceneQueryFactoryImpl f = new LuceneQueryFactoryImpl(cache); f.setPageSize(5); f.setLimit(25); LuceneQuery<Object, Object> query = f.create("index", "region", new StringQueryProvider("test", DEFAULT_FIELD)); assertEquals(25, query.getLimit()); assertEquals(5, query.getPageSize()); Mockito.verify(cache).getRegion(Mockito.eq("region")); }
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; }