honeymoose/OpenSearch
1
0
Fork 0
mirror of https://github.com/honeymoose/OpenSearch.git synced 2025-02-26 06:46:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[TEST] Randomize Preference in search requests

Browse Source 
Closes #6863
This commit is contained in:
Simon Willnauer 2014-07-15 16:10:42 +02:00
parent 6b519633ab
commit a302f3a9dd
1 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/test/java/org/elasticsearch/test/client/RandomizingClient.java
View File

@ -20,18 +20,23 @@
package org.elasticsearch.test.client; package org.elasticsearch.test.client;
import com.carrotsearch.randomizedtesting.generators.RandomPicks; import com.carrotsearch.randomizedtesting.generators.RandomPicks;
import org.apache.lucene.util.TestUtil;
import org.elasticsearch.action.search.SearchRequestBuilder; import org.elasticsearch.action.search.SearchRequestBuilder;
import org.elasticsearch.action.search.SearchType; import org.elasticsearch.action.search.SearchType;
import org.elasticsearch.client.Client; import org.elasticsearch.client.Client;
import org.elasticsearch.client.FilterClient; import org.elasticsearch.client.FilterClient;
import org.elasticsearch.cluster.routing.operation.plain.Preference;
import java.util.Arrays; import java.util.Arrays;
import java.util.EnumSet;
import java.util.Random; import java.util.Random;
/** A {@link Client} that randomizes request parameters. */ /** A {@link Client} that randomizes request parameters. */
public class RandomizingClient extends FilterClient { public class RandomizingClient extends FilterClient {
private final SearchType defaultSearchType; private final SearchType defaultSearchType;
private final String defaultPreference;
public RandomizingClient(Client client, Random random) { public RandomizingClient(Client client, Random random) {
super(client); super(client);
@ -40,11 +45,20 @@ public class RandomizingClient extends FilterClient {
defaultSearchType = RandomPicks.randomFrom(random, Arrays.asList( defaultSearchType = RandomPicks.randomFrom(random, Arrays.asList(
SearchType.DFS_QUERY_THEN_FETCH, SearchType.DFS_QUERY_THEN_FETCH,
SearchType.QUERY_THEN_FETCH)); SearchType.QUERY_THEN_FETCH));
if (random.nextInt(10) == 0) {
defaultPreference = RandomPicks.randomFrom(random, EnumSet.of(Preference.PRIMARY_FIRST, Preference.PRIMARY, Preference.LOCAL)).type();
} else if (random.nextInt(10) == 0) {
String s = TestUtil.randomRealisticUnicodeString(random, 1, 10);
defaultPreference = s.startsWith("_") ? null : s; // '_' is a reserved character
} else {
defaultPreference = null;
}
} }
@Override @Override
public SearchRequestBuilder prepareSearch(String... indices) { public SearchRequestBuilder prepareSearch(String... indices) {
return in.prepareSearch(indices).setSearchType(defaultSearchType); return in.prepareSearch(indices).setSearchType(defaultSearchType).setPreference(defaultPreference);
} }
@Override @Override