1 files changed, 35 insertions, 19 deletions

diff --git a/container-search/src/test/java/com/yahoo/prelude/fastsearch/test/FastSearcherTestCase.java b/container-search/src/test/java/com/yahoo/prelude/fastsearch/test/FastSearcherTestCase.java
index 20150fc2671..3cfd8b337fc 100644
--- a/container-search/src/test/java/com/yahoo/prelude/fastsearch/test/FastSearcherTestCase.java
+++ b/container-search/src/test/java/com/yahoo/prelude/fastsearch/test/FastSearcherTestCase.java
@@ -4,6 +4,7 @@ package com.yahoo.prelude.fastsearch.test;
 import com.google.common.collect.ImmutableList;
 import com.yahoo.component.chain.Chain;
 import com.yahoo.config.subscription.ConfigGetter;
+import com.yahoo.container.handler.VipStatus;
 import com.yahoo.container.search.Fs4Config;
 import com.yahoo.fs4.mplex.*;
 import com.yahoo.fs4.test.QueryTestCase;
@@ -14,6 +15,7 @@ import com.yahoo.prelude.fastsearch.DocumentdbInfoConfig;
 import com.yahoo.container.protect.Error;
 import com.yahoo.fs4.*;
 import com.yahoo.prelude.fastsearch.test.fs4mock.MockBackend;
+import com.yahoo.prelude.fastsearch.test.fs4mock.MockFS4ResourcePool;
 import com.yahoo.prelude.fastsearch.test.fs4mock.MockFSChannel;
 import com.yahoo.processing.execution.Execution.Trace;
 import com.yahoo.search.Query;
@@ -111,30 +113,45 @@ public class FastSearcherTestCase {
                         .summaryclass(new DocumentdbInfoConfig.Documentdb.Summaryclass.Builder().name("simple").id(7))
                         .rankprofile(new DocumentdbInfoConfig.Documentdb.Rankprofile.Builder()
                                 .name("simpler").hasRankFeatures(false).hasSummaryFeatures(false))));
+
+        List<SearchCluster.Node> nodes = new ArrayList<>();
+        nodes.add(new SearchCluster.Node("host1", 5000, 0));
+        nodes.add(new SearchCluster.Node("host2", 5000, 0));
+
+        MockFS4ResourcePool mockFs4ResourcePool = new MockFS4ResourcePool();
         FastSearcher fastSearcher = new FastSearcher(new MockBackend(),
-                new FS4ResourcePool(1),
-                new MockDispatcher(Collections.emptyList()),
+                mockFs4ResourcePool,
+                new MockDispatcher(nodes, mockFs4ResourcePool, 1, new VipStatus()),
                 new SummaryParameters(null),
                 new ClusterParams("testhittype"),
-                new CacheParams(100, 1e64),
+                new CacheParams(0, 0),
                 documentdbConfigWithOneDb);
 
-        String query = "?query=sddocname:a&dispatch.summaries";
-        Result result = doSearch(fastSearcher,new Query(query), 0, 10);
-        ErrorMessage message = result.hits().getError();
-
-        assertNotNull("Got error", message);
-        assertEquals("Invalid query parameter", message.getMessage());
-        assertEquals("When using dispatch.summaries and your summary/rankprofile require the query,  you need to enable ranking.queryCache.", message.getDetailedMessage());
-        assertEquals(Error.INVALID_QUERY_PARAMETER.code, message.getCode());
+        { // No direct.summaries
+            String query = "?query=sddocname:a&summary=simple";
+            Result result = doSearch(fastSearcher, new Query(query), 0, 10);
+            doFill(fastSearcher, result);
+            ErrorMessage error = result.hits().getError();
+            assertNull("Since we don't route to the dispatcher we hit the mock backend, so no error", error);
+        }
 
-        query = "?query=sddocname:a&dispatch.summaries&ranking.queryCache";
-        result = doSearch(fastSearcher,new Query(query), 0, 10);
-        assertNull(result.hits().getError());
+        { // direct.summaries due to query cache
+            String query = "?query=sddocname:a&ranking.queryCache";
+            Result result = doSearch(fastSearcher, new Query(query), 0, 10);
+            doFill(fastSearcher, result);
+            ErrorMessage error = result.hits().getError();
+            assertEquals("Since we don't actually run summary backends we get this error when the Dispatcher is used",
+                         "Error response from rpc node connection to host1:0: Connection error", error.getDetailedMessage());
+        }
 
-        query = "?query=sddocname:a&dispatch.summaries&summary=simple&ranking=simpler";
-        result = doSearch(fastSearcher,new Query(query), 0, 10);
-        assertNull(result.hits().getError());
+        { // direct.summaries due to no summary features
+            String query = "?query=sddocname:a&dispatch.summaries&summary=simple&ranking=simpler";
+            Result result = doSearch(fastSearcher, new Query(query), 0, 10);
+            doFill(fastSearcher, result);
+            ErrorMessage error = result.hits().getError();
+            assertEquals("Since we don't actually run summary backends we get this error when the Dispatcher is used",
+                         "Error response from rpc node connection to host1:0: Connection error", error.getDetailedMessage());
+        }
     }
 
     @Test
@@ -281,7 +298,7 @@ public class FastSearcherTestCase {
 
         assertEquals(100, fastSearcher.getCacheControl().capacity()); // Default cache =100MB
 
-        Query query = new Query("?query=ignored");
+        Query query = new Query("?query=ignored&dispatch.summaries=false");
         query.getRanking().setQueryCache(true);
         Result result = doSearch(fastSearcher, query, 0, 10);
 
@@ -321,7 +338,6 @@ public class FastSearcherTestCase {
         answer.flip();
         answer.get(expected);
 
-        assertEquals(expected.length, actual.length);
         for (int i = 0; i < expected.length; ++i) {
             if (expected[i] == IGNORE) {
                 actual[i] = IGNORE;