diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2019-09-16 22:57:02 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2019-09-16 22:57:02 +0200 |
commit | 049d1eb173f90f002e9f7251476fa0c8a71e6f7e (patch) | |
tree | 432ee8550037db4765d5fe8bad542b7038643fe3 /container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java | |
parent | 74a39bfa696568cf2d94b9c3a773f4fc0c2d0d54 (diff) |
Revert "Revert "Revert "Revert "Use a LeanHit until merging is done." MERGEOK"""
Diffstat (limited to 'container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java')
-rw-r--r-- | container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java b/container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java index 4b2f63d6b89..2fe434d6f3f 100644 --- a/container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java +++ b/container-search/src/test/java/com/yahoo/search/dispatch/MockInvoker.java @@ -1,6 +1,7 @@ // Copyright 2019 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. package com.yahoo.search.dispatch; +import com.yahoo.prelude.fastsearch.FastHit; import com.yahoo.search.Query; import com.yahoo.search.Result; import com.yahoo.search.dispatch.searchcluster.Node; @@ -37,13 +38,20 @@ class MockInvoker extends SearchInvoker { } @Override - protected Result getSearchResult(Execution execution) throws IOException { - Result ret = new Result(query); + protected InvokerResult getSearchResult(Execution execution) throws IOException { + InvokerResult ret = new InvokerResult(query, 10); if (coverage != null) { - ret.setCoverage(coverage); + ret.getResult().setCoverage(coverage); } if (hits != null) { - ret.hits().addAll(hits); + for (Hit h : hits) { + if (h instanceof FastHit) { + FastHit fh = (FastHit) h; + ret.getLeanHits().add(new LeanHit(fh.getRawGlobalId(), fh.getPartId(), fh.getDistributionKey(), fh.getRelevance().getScore())); + } else { + ret.getResult().hits().add(h); + } + } } return ret; } |