From 82397c05542a794c8629fe611cbca58a3d831780 Mon Sep 17 00:00:00 2001
From: Simon Willnauer <simonw@apache.org>
Date: Wed, 28 Aug 2013 20:57:39 +0200
Subject: [PATCH] Treat empty prefrence as a `not set` in Plain Operation
 Routing

An empty preference was causing a AIOOB exception in
PlainOperationRouting. We now check for `null` or `empty` instead of
just `null`

Closes #3591
---
 .../plain/PlainOperationRouting.java          |  2 +-
 .../search/simple/SimpleSearchTests.java      | 24 ++++++++++++++++---
 2 files changed, 22 insertions(+), 4 deletions(-)

diff --git a/src/main/java/org/elasticsearch/cluster/routing/operation/plain/PlainOperationRouting.java b/src/main/java/org/elasticsearch/cluster/routing/operation/plain/PlainOperationRouting.java
index aaa8b29cc6d..3b5376ae799 100644
--- a/src/main/java/org/elasticsearch/cluster/routing/operation/plain/PlainOperationRouting.java
+++ b/src/main/java/org/elasticsearch/cluster/routing/operation/plain/PlainOperationRouting.java
@@ -161,7 +161,7 @@ public class PlainOperationRouting extends AbstractComponent implements Operatio
     }
 
     private ShardIterator preferenceActiveShardIterator(IndexShardRoutingTable indexShard, String localNodeId, DiscoveryNodes nodes, @Nullable String preference) {
-        if (preference == null) {
+        if (preference == null || preference.isEmpty()) {
             String[] awarenessAttributes = awarenessAllocationDecider.awarenessAttributes();
             if (awarenessAttributes.length == 0) {
                 return indexShard.activeInitializingShardsRandomIt();
diff --git a/src/test/java/org/elasticsearch/test/integration/search/simple/SimpleSearchTests.java b/src/test/java/org/elasticsearch/test/integration/search/simple/SimpleSearchTests.java
index 4a4497fbe27..82b8fd69bf6 100644
--- a/src/test/java/org/elasticsearch/test/integration/search/simple/SimpleSearchTests.java
+++ b/src/test/java/org/elasticsearch/test/integration/search/simple/SimpleSearchTests.java
@@ -27,6 +27,8 @@ import org.elasticsearch.index.query.QueryBuilders;
 import org.elasticsearch.test.integration.AbstractSharedClusterTest;
 import org.junit.Test;
 
+import java.util.concurrent.ExecutionException;
+
 import static org.elasticsearch.common.xcontent.XContentFactory.jsonBuilder;
 import static org.elasticsearch.index.query.QueryBuilders.boolQuery;
 import static org.elasticsearch.index.query.QueryBuilders.rangeQuery;
@@ -51,11 +53,28 @@ public class SimpleSearchTests extends AbstractSharedClusterTest {
 
         }
     }
+    
+    @Test
+    public void testSearchRandomPreference() throws InterruptedException, ExecutionException {
+        client().admin().indices().prepareCreate("test").setSettings(ImmutableSettings.settingsBuilder().put("index.number_of_shards", between(1,3))).get();
+        indexRandom("test", true, 
+        client().prepareIndex("test", "type", "1").setSource("field", "value"),
+        client().prepareIndex("test", "type", "2").setSource("field", "value"),
+        client().prepareIndex("test", "type", "3").setSource("field", "value"),
+        client().prepareIndex("test", "type", "4").setSource("field", "value"),
+        client().prepareIndex("test", "type", "5").setSource("field", "value"),
+        client().prepareIndex("test", "type", "6").setSource("field", "value"));
+        
+        int iters = atLeast(10);
+        for (int i = 0; i < iters; i++) {
+            // id is not indexed, but lets see that we automatically convert to
+            SearchResponse searchResponse = client().prepareSearch().setQuery(QueryBuilders.matchAllQuery()).setPreference(randomUnicodeOfLengthBetween(0, 4)).get();
+            assertThat(searchResponse.getHits().totalHits(), equalTo(6l));
+        }
+    }
 
     @Test
     public void simpleIpTests() throws Exception {
-        client().admin().indices().prepareDelete().execute().actionGet();
-
         client().admin().indices().prepareCreate("test").setSettings(ImmutableSettings.settingsBuilder().put("index.number_of_shards", 1)).execute().actionGet();
 
         client().admin().indices().preparePutMapping("test").setType("type1")
@@ -76,7 +95,6 @@ public class SimpleSearchTests extends AbstractSharedClusterTest {
 
     @Test
     public void simpleIdTests() {
-        client().admin().indices().prepareDelete().execute().actionGet();
         client().admin().indices().prepareCreate("test").setSettings(ImmutableSettings.settingsBuilder().put("index.number_of_shards", 1)).execute().actionGet();
 
         client().prepareIndex("test", "type", "XXX1").setSource("field", "value").setRefresh(true).execute().actionGet();