From a302f3a9dd4476150fd8b1e13b4d4b70808e8b31 Mon Sep 17 00:00:00 2001
From: Simon Willnauer <simonw@apache.org>
Date: Tue, 15 Jul 2014 16:10:42 +0200
Subject: [PATCH] [TEST] Randomize Preference in search requests

Closes #6863
---
 .../test/client/RandomizingClient.java           | 16 +++++++++++++++-
 1 file changed, 15 insertions(+), 1 deletion(-)

diff --git a/src/test/java/org/elasticsearch/test/client/RandomizingClient.java b/src/test/java/org/elasticsearch/test/client/RandomizingClient.java
index 6c7a9377e5f..6c9283986fa 100644
--- a/src/test/java/org/elasticsearch/test/client/RandomizingClient.java
+++ b/src/test/java/org/elasticsearch/test/client/RandomizingClient.java
@@ -20,18 +20,23 @@
 package org.elasticsearch.test.client;
 
 import com.carrotsearch.randomizedtesting.generators.RandomPicks;
+import org.apache.lucene.util.TestUtil;
 import org.elasticsearch.action.search.SearchRequestBuilder;
 import org.elasticsearch.action.search.SearchType;
 import org.elasticsearch.client.Client;
 import org.elasticsearch.client.FilterClient;
+import org.elasticsearch.cluster.routing.operation.plain.Preference;
 
 import java.util.Arrays;
+import java.util.EnumSet;
 import java.util.Random;
 
 /** A {@link Client} that randomizes request parameters. */
 public class RandomizingClient extends FilterClient {
 
     private final SearchType defaultSearchType;
+    private final String defaultPreference;
+
 
     public RandomizingClient(Client client, Random random) {
         super(client);
@@ -40,11 +45,20 @@ public class RandomizingClient extends FilterClient {
         defaultSearchType = RandomPicks.randomFrom(random, Arrays.asList(
                 SearchType.DFS_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
     public SearchRequestBuilder prepareSearch(String... indices) {
-        return in.prepareSearch(indices).setSearchType(defaultSearchType);
+        return in.prepareSearch(indices).setSearchType(defaultSearchType).setPreference(defaultPreference);
     }
 
     @Override