1 files changed, 147 insertions, 0 deletions
diff --git a/container-search/src/test/java/com/yahoo/prelude/fastsearch/PartialFillTestCase.java b/container-search/src/test/java/com/yahoo/prelude/fastsearch/PartialFillTestCase.java
new file mode 100644
index 00000000000..7e3509cbef9
--- /dev/null
+++ b/container-search/src/test/java/com/yahoo/prelude/fastsearch/PartialFillTestCase.java
@@ -0,0 +1,147 @@
+// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.prelude.fastsearch;
+
+import com.yahoo.search.Query;
+import com.yahoo.search.Result;
+import com.yahoo.search.result.ErrorHit;
+import com.yahoo.search.result.ErrorMessage;
+import org.junit.jupiter.api.Test;
+
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+/**
+ * @author havardpe
+ */
+public class PartialFillTestCase {
+    private static final ClusterParams CLUSTER_PARAMS = new ClusterParams("container.0");
+
+    public static class FS4 extends VespaBackend {
+        public List<Result> history = new ArrayList<>();
+        FS4() {
+            super(CLUSTER_PARAMS);
+        }
+        protected Result doSearch2(String schema, Query query) {
+            return new Result(query);
+        }
+        protected void doPartialFill(Result result, String summaryClass) {
+            history.add(result);
+        }
+    }
+
+    public static class BadFS4 extends VespaBackend {
+        BadFS4() {
+            super(CLUSTER_PARAMS);
+        }
+        protected Result doSearch2(String schema, Query query) {
+            return new Result(query);
+        }
+        protected void doPartialFill(Result result, String summaryClass) {
+            if (result.hits().getErrorHit() == null) {
+                result.hits().addError(ErrorMessage.createUnspecifiedError("error"));
+            }
+        }
+    }
+
+    @Test
+    void testPartitioning() {
+        FS4 fs4 = new FS4();
+        Query a = new Query("/?query=foo");
+        Query b = new Query("/?query=bar");
+        Query c = new Query("/?query=foo"); // equal to a
+        Result r = new Result(new Query("/?query=ignorethis"));
+        for (int i = 0; i < 7; i++) {
+            FastHit h = new FastHit();
+            h.setQuery(a);
+            h.setFillable();
+            r.hits().add(h);
+        }
+        for (int i = 0; i < 5; i++) {
+            FastHit h = new FastHit();
+            h.setQuery(b);
+            h.setFillable();
+            r.hits().add(h);
+        }
+        for (int i = 0; i < 3; i++) {
+            FastHit h = new FastHit();
+            h.setQuery(c);
+            h.setFillable();
+            r.hits().add(h);
+        }
+        for (int i = 0; i < 2; i++) {
+            FastHit h = new FastHit();
+            // no query assigned
+            h.setFillable();
+            r.hits().add(h);
+        }
+        for (int i = 0; i < 5; i++) {
+            FastHit h = new FastHit();
+            // not fillable
+            h.setQuery(a);
+            r.hits().add(h);
+        }
+        for (int i = 0; i < 5; i++) {
+            FastHit h = new FastHit();
+            // already filled
+            h.setQuery(a);
+            h.setFilled("default");
+            r.hits().add(h);
+        }
+        doFill(fs4, r, "default");
+        assertNull(r.hits().getErrorHit());
+        assertEquals(4, fs4.history.size());
+        assertEquals(a, fs4.history.get(0).getQuery());
+        assertEquals(7, fs4.history.get(0).getHitCount());
+        assertEquals(b, fs4.history.get(1).getQuery());
+        assertEquals(5, fs4.history.get(1).getHitCount());
+        assertEquals(c, fs4.history.get(2).getQuery());
+        assertEquals(3, fs4.history.get(2).getHitCount());
+        assertEquals(r.getQuery(), fs4.history.get(3).getQuery());
+        assertEquals(2, fs4.history.get(3).getHitCount());
+    }
+
+    @Test
+    void testMergeErrors() {
+        BadFS4 fs4 = new BadFS4();
+        Query a = new Query("/?query=foo");
+        Query b = new Query("/?query=bar");
+        Result r = new Result(new Query("/?query=ignorethis"));
+        {
+            FastHit h = new FastHit();
+            h.setQuery(a);
+            h.setFillable();
+            r.hits().add(h);
+        }
+        {
+            FastHit h = new FastHit();
+            h.setQuery(b);
+            h.setFillable();
+            r.hits().add(h);
+        }
+        doFill(fs4, r, "default");
+        ErrorHit eh = r.hits().getErrorHit();
+        assertNotNull(eh);
+        ErrorMessage exp_sub = ErrorMessage.createUnspecifiedError("error");
+        int n = 0;
+        for (Iterator<? extends com.yahoo.search.result.ErrorMessage> i = eh.errorIterator(); i.hasNext(); ) {
+            com.yahoo.search.result.ErrorMessage error = i.next();
+            switch (n) {
+                case 0:
+                case 1:
+                    assertEquals(exp_sub, error);
+                    break;
+                default:
+                    fail();
+            }
+            n++;
+        }
+    }
+
+    private void doFill(VespaBackend searcher, Result result, String summaryClass) {
+        searcher.fill(result, summaryClass);
+    }
+
+}