Merge pull request #19969 from chengpohi/master

fix CompletionSuggestion test failed caused by shard is 1
This commit is contained in:
Areek Zillur 2016-08-12 14:20:21 -04:00 committed by GitHub
commit a4f4525955
1 changed files with 5 additions and 3 deletions

View File

@ -32,7 +32,6 @@ import static org.hamcrest.Matchers.lessThanOrEqualTo;
public class CompletionSuggestionTests extends ESTestCase {
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/19896")
public void testToReduce() throws Exception {
List<Suggest.Suggestion<CompletionSuggestion.Entry>> shardSuggestions = new ArrayList<>();
int nShards = randomIntBetween(1, 10);
@ -47,9 +46,12 @@ public class CompletionSuggestionTests extends ESTestCase {
float maxScore = randomIntBetween(totalResults, totalResults*2);
for (int i = 0; i < totalResults; i++) {
Suggest.Suggestion<CompletionSuggestion.Entry> suggestion = randomFrom(shardSuggestions);
suggestion.getEntries().get(0).addOption(new CompletionSuggestion.Entry.Option(i, new Text(""),
CompletionSuggestion.Entry entry = suggestion.getEntries().get(0);
if (entry.getOptions().size() < size) {
entry.addOption(new CompletionSuggestion.Entry.Option(i, new Text(""),
maxScore - i, Collections.emptyMap()));
}
}
CompletionSuggestion reducedSuggestion = CompletionSuggestion.reduceTo(shardSuggestions);
assertNotNull(reducedSuggestion);
assertThat(reducedSuggestion.getOptions().size(), lessThanOrEqualTo(size));