andreacavalli/CavalliumDBEngine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Cleanup code

Browse Source
This commit is contained in:
Andrea Cavalli 2021-10-13 13:16:56 +02:00
parent 584115c5cc
commit c8536e45da
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/test/java/it/cavallium/dbengine/TestLuceneSearches.java
View File

@ -36,7 +36,9 @@ import it.cavallium.dbengine.lucene.searcher.UnsortedUnscoredStreamingMultiSearc
import java.io.IOException; import java.io.IOException;
import java.util.List; import java.util.List;
import java.util.Objects; import java.util.Objects;
import java.util.concurrent.Callable;
import java.util.stream.Stream; import java.util.stream.Stream;
import org.apache.commons.lang3.function.FailableConsumer;
import org.jetbrains.annotations.Nullable; import org.jetbrains.annotations.Nullable;
import org.junit.jupiter.api.AfterAll; import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.Assertions;
@ -174,6 +176,14 @@ public class TestLuceneSearches {
.toStream(); .toStream();
} }
private static <E extends Throwable> void runSearchers(ExpectedQueryType expectedQueryType, FailableConsumer<LocalSearcher, E> consumer) throws E {
var searchers = run(getSearchers(expectedQueryType).collectList());
for (LocalSearcher searcher : searchers) {
log.info("Using searcher \"{}\"", searcher.getName());
consumer.accept(searcher);
}
}
@AfterAll @AfterAll
public static void afterAll() { public static void afterAll() {
TEMP_DB_GENERATOR.closeTempDb(tempDb).block(); TEMP_DB_GENERATOR.closeTempDb(tempDb).block();
@ -216,10 +226,7 @@ public class TestLuceneSearches {
@ParameterizedTest @ParameterizedTest
@MethodSource("provideQueryArgumentsScoreModeAndSort") @MethodSource("provideQueryArgumentsScoreModeAndSort")
public void testSearchNoDocs(boolean shards, LLScoreMode scoreMode, MultiSort<SearchResultKey<String>> multiSort) { public void testSearchNoDocs(boolean shards, LLScoreMode scoreMode, MultiSort<SearchResultKey<String>> multiSort) {
var searchers = run(getSearchers(new ExpectedQueryType(shards, isSorted(multiSort), isScored(scoreMode, multiSort), true, false)).collectList()); runSearchers(new ExpectedQueryType(shards, isSorted(multiSort), isScored(scoreMode, multiSort), true, false), searcher -> {
for (LocalSearcher searcher : searchers) {
log.info("Using searcher \"{}\"", searcher.getName());
var luceneIndex = getLuceneIndex(shards, searcher); var luceneIndex = getLuceneIndex(shards, searcher);
ClientQueryParamsBuilder<SearchResultKey<String>> queryBuilder = ClientQueryParams.builder(); ClientQueryParamsBuilder<SearchResultKey<String>> queryBuilder = ClientQueryParams.builder();
queryBuilder.query(new MatchNoDocsQuery()); queryBuilder.query(new MatchNoDocsQuery());
@ -236,7 +243,7 @@ public class TestLuceneSearches {
var keys = getResults(results); var keys = getResults(results);
assertEquals(List.of(), keys); assertEquals(List.of(), keys);
} }
} });
} }
private boolean supportsPreciseHitsCount(LocalSearcher searcher, private boolean supportsPreciseHitsCount(LocalSearcher searcher,
@ -257,10 +264,7 @@ public class TestLuceneSearches {
@ParameterizedTest @ParameterizedTest
@MethodSource("provideQueryArgumentsScoreModeAndSort") @MethodSource("provideQueryArgumentsScoreModeAndSort")
public void testSearchAllDocs(boolean shards, LLScoreMode scoreMode, MultiSort<SearchResultKey<String>> multiSort) { public void testSearchAllDocs(boolean shards, LLScoreMode scoreMode, MultiSort<SearchResultKey<String>> multiSort) {
var searchers = run(getSearchers(new ExpectedQueryType(shards, isSorted(multiSort), isScored(scoreMode, multiSort), true, false)).collectList()); runSearchers(new ExpectedQueryType(shards, isSorted(multiSort), isScored(scoreMode, multiSort), true, false), (LocalSearcher searcher) -> {
for (LocalSearcher searcher : searchers) {
log.info("Using searcher \"{}\"", searcher.getName());
var luceneIndex = getLuceneIndex(shards, searcher); var luceneIndex = getLuceneIndex(shards, searcher);
ClientQueryParamsBuilder<SearchResultKey<String>> queryBuilder = ClientQueryParams.builder(); ClientQueryParamsBuilder<SearchResultKey<String>> queryBuilder = ClientQueryParams.builder();
queryBuilder.query(new MatchNoDocsQuery()); queryBuilder.query(new MatchNoDocsQuery());
@ -277,7 +281,7 @@ public class TestLuceneSearches {
var keys = getResults(results); var keys = getResults(results);
assertEquals(List.of(), keys); assertEquals(List.of(), keys);
} }
} });
} }
private boolean isSorted(MultiSort<SearchResultKey<String>> multiSort) { private boolean isSorted(MultiSort<SearchResultKey<String>> multiSort) {