org.elasticsearch.action.search.searchresponse

Here are the examples of the java api class org.elasticsearch.action.search.searchresponse taken from open source projects.

1. OpenNlpPluginIntegrationTest#testThatOwnAnalyzersCanBeDefinedPerNlpMappedField()

Project: elasticsearch-opennlp-plugin
File: OpenNlpPluginIntegrationTest.java
@Test
public void testThatOwnAnalyzersCanBeDefinedPerNlpMappedField() throws IOException {
    putMapping("/test-mapping-analyzers.json");
    String sampleText = copyToStringFromClasspath("/sample-text.txt");
    IndexResponse indexResponse = indexElement(sampleText);
    SearchResponse nonAnalyzedFieldSearchResponse = query(QueryBuilders.termQuery("someField.name", "jack"));
    assertThat(nonAnalyzedFieldSearchResponse.getHits().totalHits(), is(1L));
    // analyzed, therefore not resulting anything like the above query
    SearchResponse analyzedFieldSearchResponse = query(QueryBuilders.termQuery("someFieldAnalyzed.name", "jack"));
    assertThat(analyzedFieldSearchResponse.getHits().totalHits(), is(0L));
    SearchResponse searchResponse = query(QueryBuilders.prefixQuery("someFieldAnalyzed.name", "Jack"));
    assertThat(searchResponse.getHits().totalHits(), is(1L));
    searchResponse = query(QueryBuilders.matchQuery("someFieldAnalyzed.name", "Jack Nicholson"));
    assertThat(searchResponse.getHits().totalHits(), is(1L));
    assertThat(searchResponse.getHits().getAt(0).id(), is(indexResponse.getId()));
}

2. SuggestSearchIT#searchSuggest()

Project: elasticsearch
File: SuggestSearchIT.java
protected Suggest searchSuggest(String suggestText, int expectShardsFailed, Map<String, SuggestionBuilder<?>> suggestions) {
    SearchRequestBuilder builder = client().prepareSearch().setSize(0);
    SuggestBuilder suggestBuilder = new SuggestBuilder();
    if (suggestText != null) {
        suggestBuilder.setGlobalText(suggestText);
    }
    for (Entry<String, SuggestionBuilder<?>> suggestion : suggestions.entrySet()) {
        suggestBuilder.addSuggestion(suggestion.getKey(), suggestion.getValue());
    }
    builder.suggest(suggestBuilder);
    SearchResponse actionGet = builder.execute().actionGet();
    assertThat(Arrays.toString(actionGet.getShardFailures()), actionGet.getFailedShards(), equalTo(expectShardsFailed));
    return actionGet.getSuggest();
}

3. MatchedQueriesIT#testSpanFirstQuerySupportsName()

Project: elasticsearch
File: MatchedQueriesIT.java
public void testSpanFirstQuerySupportsName() {
    createIndex("test1");
    ensureGreen();
    client().prepareIndex("test1", "type1", "1").setSource("title", "title1 title2").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch().setQuery(QueryBuilders.spanFirstQuery(QueryBuilders.spanTermQuery("title", "title1"), 10).queryName("span")).get();
    assertHitCount(searchResponse, 1L);
    for (SearchHit hit : searchResponse.getHits()) {
        if (hit.id().equals("1")) {
            assertThat(hit.matchedQueries().length, equalTo(1));
            assertThat(hit.matchedQueries(), hasItemInArray("span"));
        } else {
            fail("Unexpected document returned with id " + hit.id());
        }
    }
}

4. MatchedQueriesIT#testWildcardQuerySupportsName()

Project: elasticsearch
File: MatchedQueriesIT.java
public void testWildcardQuerySupportsName() {
    createIndex("test1");
    ensureGreen();
    client().prepareIndex("test1", "type1", "1").setSource("title", "title1").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch().setQuery(QueryBuilders.wildcardQuery("title", "titl*").queryName("wildcard")).get();
    assertHitCount(searchResponse, 1L);
    for (SearchHit hit : searchResponse.getHits()) {
        if (hit.id().equals("1")) {
            assertThat(hit.matchedQueries().length, equalTo(1));
            assertThat(hit.matchedQueries(), hasItemInArray("wildcard"));
        } else {
            fail("Unexpected document returned with id " + hit.id());
        }
    }
}

5. MatchedQueriesIT#testFuzzyQuerySupportsName()

Project: elasticsearch
File: MatchedQueriesIT.java
public void testFuzzyQuerySupportsName() {
    createIndex("test1");
    ensureGreen();
    client().prepareIndex("test1", "type1", "1").setSource("title", "title1").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch().setQuery(QueryBuilders.fuzzyQuery("title", "titel1").queryName("fuzzy")).get();
    assertHitCount(searchResponse, 1L);
    for (SearchHit hit : searchResponse.getHits()) {
        if (hit.id().equals("1")) {
            assertThat(hit.matchedQueries().length, equalTo(1));
            assertThat(hit.matchedQueries(), hasItemInArray("fuzzy"));
        } else {
            fail("Unexpected document returned with id " + hit.id());
        }
    }
}

6. MatchedQueriesIT#testPrefixQuerySupportsName()

Project: elasticsearch
File: MatchedQueriesIT.java
public void testPrefixQuerySupportsName() {
    createIndex("test1");
    ensureGreen();
    client().prepareIndex("test1", "type1", "1").setSource("title", "title1").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch().setQuery(QueryBuilders.prefixQuery("title", "title").queryName("prefix")).get();
    assertHitCount(searchResponse, 1L);
    for (SearchHit hit : searchResponse.getHits()) {
        if (hit.id().equals("1")) {
            assertThat(hit.matchedQueries().length, equalTo(1));
            assertThat(hit.matchedQueries(), hasItemInArray("prefix"));
        } else {
            fail("Unexpected document returned with id " + hit.id());
        }
    }
}

7. MatchedQueriesIT#testRegExpQuerySupportsName()

Project: elasticsearch
File: MatchedQueriesIT.java
public void testRegExpQuerySupportsName() {
    createIndex("test1");
    ensureGreen();
    client().prepareIndex("test1", "type1", "1").setSource("title", "title1").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch().setQuery(QueryBuilders.regexpQuery("title", "title1").queryName("regex")).get();
    assertHitCount(searchResponse, 1L);
    for (SearchHit hit : searchResponse.getHits()) {
        if (hit.id().equals("1")) {
            assertThat(hit.matchedQueries().length, equalTo(1));
            assertThat(hit.matchedQueries(), hasItemInArray("regex"));
        } else {
            fail("Unexpected document returned with id " + hit.id());
        }
    }
}

8. ElasticSearchState#search()

Project: trident-tutorial
File: ElasticSearchState.java
/**
     * It basically searches and returns a set of tweet ids for a given keyword.
     *
     * @param keyword
     * @return
     */
public Set<String> search(String keyword) {
    SearchResponse response;
    try {
        response = client.prepareSearch().setIndices("hackaton").setTypes("tweets").addFields("id", "text").setQuery(QueryBuilders.fieldQuery("text", keyword)).execute().actionGet();
    } catch (Throwable e) {
        return new HashSet<String>();
    }
    Set<String> result = new HashSet<String>();
    for (SearchHit hit : response.getHits()) {
        Long id = hit.field("id").<Long>getValue();
        result.add(String.valueOf(id));
    }
    return result;
}

9. ESIndexer#search()

Project: Kundera
File: ESIndexer.java
/*
     * (non-Javadoc)
     * 
     * @see com.impetus.kundera.index.Indexer#search(java.lang.Class,
     * com.impetus.kundera.metadata.model.EntityMetadata, java.lang.String, int,
     * int)
     */
@Override
public Map<String, Object> search(Class<?> clazz, EntityMetadata m, String luceneQuery, int start, int count) {
    if (log.isInfoEnabled()) {
        log.info("Executing lucene query " + luceneQuery);
    }
    ListenableActionFuture<SearchResponse> listenableActionFuture = client.prepareSearch(m.getSchema().toLowerCase()).setQuery(QueryBuilders.queryStringQuery(luceneQuery)).setSize(40000).execute();
    SearchResponse response = listenableActionFuture.actionGet();
    Map<String, Object> results = new HashMap<String, Object>();
    for (SearchHit hit : response.getHits()) {
        Object id = PropertyAccessorHelper.fromSourceToTargetClass(((AbstractAttribute) m.getIdAttribute()).getBindableJavaType(), String.class, hit.getId());
        results.put(hit.getId(), id);
    }
    return results;
}

10. FsCrawlerImplAllParametersTest#test_defaults()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_defaults() throws Exception {
    startCrawler();
    // We expect to have one file
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1);
    // The default configuration should not add file attributes
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.ATTRIBUTES + "." + FsCrawlerUtil.Doc.Attributes.OWNER), nullValue());
    }
}

11. FsCrawlerImplAllParametersTest#test_do_not_store_source()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_do_not_store_source() throws Exception {
    startCrawler();
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "_source", "*");
    for (SearchHit hit : searchResponse.getHits()) {
        // We check that the field has not been stored
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.ATTACHMENT), nullValue());
        // We check that the field is not part of _source
        assertThat(hit.getSource().get(FsCrawlerUtil.Doc.ATTACHMENT), nullValue());
    }
}

12. FsCrawlerImplAllParametersTest#test_store_source()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_store_source() throws Exception {
    Fs fs = startCrawlerDefinition().setStoreSource(true).build();
    startCrawler(getCrawlerName(), fs, endCrawlerDefinition(getCrawlerName()), null);
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "_source", "*");
    for (SearchHit hit : searchResponse.getHits()) {
        // We check that the field has been stored
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.ATTACHMENT), notNullValue());
        // We check that the field is not part of _source
        assertThat(hit.getSource().get(FsCrawlerUtil.Doc.ATTACHMENT), nullValue());
    }
}

13. FsCrawlerImplAllParametersTest#test_default_metadata()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_default_metadata() throws Exception {
    startCrawler();
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "*");
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.ATTACHMENT), nullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.FILENAME), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.CONTENT_TYPE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.URL), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.FILESIZE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXING_DATE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS), nullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.LAST_MODIFIED), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.META + "." + FsCrawlerUtil.Doc.Meta.TITLE), notNullValue());
    }
}

14. FsCrawlerImplAllParametersTest#test_metadata()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_metadata() throws Exception {
    Fs fs = startCrawlerDefinition().setIndexedChars(new Percentage(1)).build();
    startCrawler(getCrawlerName(), fs, endCrawlerDefinition(getCrawlerName()), null);
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "*");
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.FILENAME), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.CONTENT_TYPE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.URL), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.FILESIZE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXING_DATE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.LAST_MODIFIED), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.META + "." + FsCrawlerUtil.Doc.Meta.TITLE), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.META + "." + FsCrawlerUtil.Doc.Meta.KEYWORDS), notNullValue());
    }
}

15. FsCrawlerImplAllParametersTest#test_filesize_disabled()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_filesize_disabled() throws Exception {
    Fs fs = startCrawlerDefinition().setAddFilesize(false).build();
    startCrawler(getCrawlerName(), fs, endCrawlerDefinition(getCrawlerName()), null);
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1);
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get("filesize"), nullValue());
    }
}

16. FsCrawlerImplAllParametersTest#test_filesize_nolimit()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_filesize_nolimit() throws Exception {
    Fs fs = startCrawlerDefinition().setIndexedChars(new Percentage(-1)).build();
    startCrawler(getCrawlerName(), fs, endCrawlerDefinition(getCrawlerName()), null);
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "*");
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.CONTENT), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS), nullValue());
        // Our original text: "Bonjour David\n\n\n" should not be truncated
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.CONTENT).getValue(), is("Bonjour David\n\n\n"));
    }
}

17. FsCrawlerImplAllParametersTest#test_filesize_limit_percentage()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_filesize_limit_percentage() throws Exception {
    Fs fs = startCrawlerDefinition().setIndexedChars(Percentage.parse("0.1%")).build();
    startCrawler(getCrawlerName(), fs, endCrawlerDefinition(getCrawlerName()), null);
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "*");
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.CONTENT), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS), notNullValue());
        // Our original text: "Bonjour David..." should be truncated
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.CONTENT).getValue(), is("Bonjour "));
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS).getValue(), is(8L));
    }
}

18. FsCrawlerImplAllParametersTest#test_filesize_limit()

Project: fscrawler
File: FsCrawlerImplAllParametersTest.java
@Test
public void test_filesize_limit() throws Exception {
    Fs fs = startCrawlerDefinition().setIndexedChars(new Percentage(7)).build();
    startCrawler(getCrawlerName(), fs, endCrawlerDefinition(getCrawlerName()), null);
    SearchResponse searchResponse = countTestHelper(getCrawlerName(), null, 1, null, null, "*");
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.CONTENT), notNullValue());
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS), notNullValue());
        // Our original text: "Bonjour David..." should be truncated
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.CONTENT).getValue(), is("Bonjour"));
        assertThat(hit.getFields().get(FsCrawlerUtil.Doc.FILE + "." + FsCrawlerUtil.Doc.File.INDEXED_CHARS).getValue(), is(7L));
    }
}

19. GeoPolygonIT#testSimpleUnclosedPolygon()

Project: elasticsearch
File: GeoPolygonIT.java
public void testSimpleUnclosedPolygon() throws Exception {
    List<GeoPoint> points = new ArrayList<>();
    points.add(new GeoPoint(40.7, -74.0));
    points.add(new GeoPoint(40.7, -74.1));
    points.add(new GeoPoint(40.8, -74.1));
    points.add(new GeoPoint(40.8, -74.0));
    SearchResponse searchResponse = // from NY
    client().prepareSearch("test").setQuery(boolQuery().must(geoPolygonQuery("location", points))).execute().actionGet();
    assertHitCount(searchResponse, 4);
    assertThat(searchResponse.getHits().hits().length, equalTo(4));
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.id(), anyOf(equalTo("1"), equalTo("3"), equalTo("4"), equalTo("5")));
    }
}

20. GeoPolygonIT#testSimplePolygon()

Project: elasticsearch
File: GeoPolygonIT.java
public void testSimplePolygon() throws Exception {
    List<GeoPoint> points = new ArrayList<>();
    points.add(new GeoPoint(40.7, -74.0));
    points.add(new GeoPoint(40.7, -74.1));
    points.add(new GeoPoint(40.8, -74.1));
    points.add(new GeoPoint(40.8, -74.0));
    points.add(new GeoPoint(40.7, -74.0));
    SearchResponse searchResponse = // from NY
    client().prepareSearch("test").setQuery(boolQuery().must(geoPolygonQuery("location", points))).execute().actionGet();
    assertHitCount(searchResponse, 4);
    assertThat(searchResponse.getHits().hits().length, equalTo(4));
    for (SearchHit hit : searchResponse.getHits()) {
        assertThat(hit.id(), anyOf(equalTo("1"), equalTo("3"), equalTo("4"), equalTo("5")));
    }
}

21. FieldSortIT#testIssue2986()

Project: elasticsearch
File: FieldSortIT.java
public void testIssue2986() {
    assertAcked(client().admin().indices().prepareCreate("test").addMapping("type", "field1", "type=keyword").get());
    client().prepareIndex("test", "post", "1").setSource("{\"field1\":\"value1\"}").execute().actionGet();
    client().prepareIndex("test", "post", "2").setSource("{\"field1\":\"value2\"}").execute().actionGet();
    client().prepareIndex("test", "post", "3").setSource("{\"field1\":\"value3\"}").execute().actionGet();
    refresh();
    SearchResponse result = client().prepareSearch("test").setQuery(matchAllQuery()).setTrackScores(true).addSort("field1", SortOrder.ASC).execute().actionGet();
    for (SearchHit hit : result.getHits()) {
        assertFalse(Float.isNaN(hit.getScore()));
    }
}

22. AggregationTest#testFromSizeWithAggregations()

Project: elasticsearch-sql
File: AggregationTest.java
@Test
public void testFromSizeWithAggregations() throws Exception {
    final String query1 = String.format("SELECT /*! DOCS_WITH_AGGREGATION(0,1) */" + " account_number FROM %s/account GROUP BY gender", TEST_INDEX);
    SearchResponse response1 = (SearchResponse) getSearchRequestBuilder(query1).get();
    Assert.assertEquals(1, response1.getHits().getHits().length);
    Terms gender1 = response1.getAggregations().get("gender");
    Assert.assertEquals(2, gender1.getBuckets().size());
    Object account1 = response1.getHits().getHits()[0].getSource().get("account_number");
    final String query2 = String.format("SELECT /*! DOCS_WITH_AGGREGATION(1,1) */" + " account_number FROM %s/account GROUP BY gender", TEST_INDEX);
    SearchResponse response2 = (SearchResponse) getSearchRequestBuilder(query2).get();
    Assert.assertEquals(1, response2.getHits().getHits().length);
    Terms gender2 = response2.getAggregations().get("gender");
    Assert.assertEquals(2, gender2.getBuckets().size());
    Object account2 = response2.getHits().getHits()[0].getSource().get("account_number");
    Assert.assertEquals(response1.getHits().getTotalHits(), response2.getHits().getTotalHits());
    Assert.assertNotEquals(account1, account2);
}

23. SuggestSearchTests#searchSuggest()

Project: elasticsearch
File: SuggestSearchTests.java
protected Suggest searchSuggest(String suggestText, int expectShardsFailed, Map<String, SuggestionBuilder<?>> suggestions) {
    SearchRequestBuilder builder = client().prepareSearch().setSize(0);
    SuggestBuilder suggestBuilder = new SuggestBuilder();
    if (suggestText != null) {
        suggestBuilder.setGlobalText(suggestText);
    }
    for (Entry<String, SuggestionBuilder<?>> suggestion : suggestions.entrySet()) {
        suggestBuilder.addSuggestion(suggestion.getKey(), suggestion.getValue());
    }
    builder.suggest(suggestBuilder);
    SearchResponse actionGet = builder.execute().actionGet();
    assertThat(Arrays.toString(actionGet.getShardFailures()), actionGet.getFailedShards(), equalTo(expectShardsFailed));
    return actionGet.getSuggest();
}

24. ElasticsearchInterpreter#processCount()

Project: zeppelin
File: ElasticsearchInterpreter.java
/**
   * Processes a "count" request.
   *
   * @param urlItems Items of the URL
   * @param data May contains the JSON of the request
   * @return Result of the count request, it contains the total hits
   */
private InterpreterResult processCount(String[] urlItems, String data) {
    if (urlItems.length > 2) {
        return new InterpreterResult(InterpreterResult.Code.ERROR, "Bad URL (it should be /index1,index2,.../type1,type2,...)");
    }
    final SearchResponse response = searchData(urlItems, data, 0);
    return new InterpreterResult(InterpreterResult.Code.SUCCESS, InterpreterResult.Type.TEXT, "" + response.getHits().getTotalHits());
}

25. ElasticSearchStreamingInputFormat#findNumHits()

Project: wonderdog
File: ElasticSearchStreamingInputFormat.java
private void findNumHits() {
    SearchRequestBuilder request = client.prepareSearch(indexName);
    if (mappingName != null && mappingName.length() > 0) {
        request.setTypes(mappingName);
    }
    request.setSearchType(SearchType.COUNT);
    if (queryJSON != null && queryJSON.length() > 0) {
        request.setQuery(queryJSON);
    }
    SearchResponse response = request.execute().actionGet();
    this.numHits = response.getHits().totalHits();
    LOG.info("Ran query: " + String.valueOf(numHits) + " hits");
}

26. ElasticSearchInputFormat#initiate_search()

Project: wonderdog
File: ElasticSearchInputFormat.java
private void initiate_search() {
    SearchResponse response = client.prepareSearch(indexName).setTypes(objType).setSearchType(SearchType.COUNT).setQuery(QueryBuilders.queryString(queryString)).setSize(requestSize).execute().actionGet();
    this.numHits = response.getHits().totalHits();
    // This could be bad
    if (numSplits > numHits)
        numSplits = numHits;
    this.numSplitRecords = (numHits / numSplits);
}

27. ElasticSearchIndex#query()

Project: titan
File: ElasticSearchIndex.java
@Override
public Iterable<RawQuery.Result<String>> query(RawQuery query, KeyInformation.IndexRetriever informations, BaseTransaction tx) throws BackendException {
    SearchRequestBuilder srb = client.prepareSearch(indexName);
    srb.setTypes(query.getStore());
    srb.setQuery(QueryBuilders.queryStringQuery(query.getQuery()));
    srb.setFrom(query.getOffset());
    if (query.hasLimit())
        srb.setSize(query.getLimit());
    else
        srb.setSize(maxResultsSize);
    srb.setNoFields();
    //srb.setExplain(true);
    SearchResponse response = srb.execute().actionGet();
    log.debug("Executed query [{}] in {} ms", query.getQuery(), response.getTookInMillis());
    SearchHits hits = response.getHits();
    if (!query.hasLimit() && hits.totalHits() >= maxResultsSize)
        log.warn("Query result set truncated to first [{}] elements for query: {}", maxResultsSize, query);
    List<RawQuery.Result<String>> result = new ArrayList<RawQuery.Result<String>>(hits.hits().length);
    for (SearchHit hit : hits) {
        result.add(new RawQuery.Result<String>(hit.id(), hit.getScore()));
    }
    return result;
}

28. AsyncClientTest#test_node_client()

Project: spring-elasticsearch
File: AsyncClientTest.java
@Test
public void test_node_client() throws Exception {
    Node node = ctx.getBean(Node.class);
    try {
        Proxy.getInvocationHandler(node);
        throw new Exception("Must not be proxyfied");
    } catch (IllegalArgumentException e) {
    }
    Client client = checkClient(true);
    client.admin().cluster().prepareState().execute().get();
    // #92: prepareSearch() errors with async created TransportClient
    client.prepareIndex("index", "type").setSource("foo", "bar").setRefresh(true).get();
    SearchResponse response = client.prepareSearch("index").get();
    assertThat(response.getHits().getTotalHits(), is(1L));
}

29. ProxySearchScrollRequestBuilderTest#no_trace_logs()

Project: sonarqube
File: ProxySearchScrollRequestBuilderTest.java
@Test
public void no_trace_logs() {
    logTester.setLevel(LoggerLevel.DEBUG);
    SearchResponse response = esTester.client().prepareSearch(FakeIndexDefinition.INDEX).setSearchType(SearchType.SCAN).setScroll(TimeValue.timeValueMinutes(1)).get();
    logTester.clear();
    esTester.client().prepareSearchScroll(response.getScrollId()).get();
    assertThat(logTester.logs()).isEmpty();
}

30. ProxySearchScrollRequestBuilderTest#trace_logs()

Project: sonarqube
File: ProxySearchScrollRequestBuilderTest.java
@Test
public void trace_logs() {
    logTester.setLevel(LoggerLevel.TRACE);
    SearchResponse response = esTester.client().prepareSearch(FakeIndexDefinition.INDEX).setSearchType(SearchType.SCAN).setScroll(TimeValue.timeValueMinutes(1)).get();
    logTester.clear();
    esTester.client().prepareSearchScroll(response.getScrollId()).get();
    assertThat(logTester.logs()).hasSize(1);
}

31. EsTester#getDocuments()

Project: sonarqube
File: EsTester.java
/**
   * Get all the indexed documents (no paginated results). Results are not sorted.
   */
public List<SearchHit> getDocuments(String indexName, String typeName) {
    SearchRequestBuilder req = client.nativeClient().prepareSearch(indexName).setTypes(typeName).setQuery(QueryBuilders.matchAllQuery());
    req.setSearchType(SearchType.SCAN).setScroll(new TimeValue(60000)).setSize(100);
    SearchResponse response = req.get();
    List<SearchHit> result = newArrayList();
    while (true) {
        Iterables.addAll(result, response.getHits());
        response = client.nativeClient().prepareSearchScroll(response.getScrollId()).setScroll(new TimeValue(600000)).execute().actionGet();
        // Break condition: No hits are returned
        if (response.getHits().getHits().length == 0) {
            break;
        }
    }
    return result;
}

32. UserIndex#selectUsersForBatch()

Project: sonarqube
File: UserIndex.java
public Iterator<UserDoc> selectUsersForBatch(List<String> logins) {
    BoolQueryBuilder filter = boolQuery().must(termsQuery(UserIndexDefinition.FIELD_LOGIN, logins));
    SearchRequestBuilder requestBuilder = esClient.prepareSearch(UserIndexDefinition.INDEX).setTypes(UserIndexDefinition.TYPE_USER).setSearchType(SearchType.SCAN).addSort(SortBuilders.fieldSort(UserIndexDefinition.FIELD_LOGIN).order(SortOrder.ASC)).setScroll(TimeValue.timeValueMinutes(EsUtils.SCROLL_TIME_IN_MINUTES)).setSize(10_000).setFetchSource(new String[] { UserIndexDefinition.FIELD_LOGIN, UserIndexDefinition.FIELD_NAME }, null).setQuery(QueryBuilders.filteredQuery(matchAllQuery(), filter));
    SearchResponse response = requestBuilder.get();
    return EsUtils.scroll(esClient, response.getScrollId(), DOC_CONVERTER);
}

33. RuleIndex#searchAll()

Project: sonarqube
File: RuleIndex.java
/**
   * Return all keys matching the search query, without pagination nor facets
   */
public Iterator<RuleKey> searchAll(RuleQuery query) {
    SearchRequestBuilder esSearch = getClient().prepareSearch(INDEX).setTypes(TYPE_RULE).setSearchType(SearchType.SCAN).setScroll(TimeValue.timeValueMinutes(SCROLL_TIME_IN_MINUTES));
    QueryBuilder qb = buildQuery(query);
    Map<String, QueryBuilder> filters = buildFilters(query);
    setSorting(query, esSearch);
    BoolQueryBuilder fb = boolQuery();
    for (QueryBuilder filterBuilder : filters.values()) {
        fb.must(filterBuilder);
    }
    esSearch.setQuery(boolQuery().must(qb).filter(fb));
    SearchResponse response = esSearch.get();
    return scrollIds(getClient(), response.getScrollId(), ToRuleKey.INSTANCE);
}

34. ActiveRuleIndex#getStatsByProfileKeys()

Project: sonarqube
File: ActiveRuleIndex.java
public Map<String, Multimap<String, FacetValue>> getStatsByProfileKeys(List<String> keys) {
    SearchRequestBuilder request = getClient().prepareSearch(INDEX).setQuery(QueryBuilders.boolQuery().must(QueryBuilders.termsQuery(FIELD_ACTIVE_RULE_PROFILE_KEY, keys)).filter(QueryBuilders.boolQuery().mustNot(QueryBuilders.hasParentQuery(TYPE_RULE, QueryBuilders.termQuery(FIELD_RULE_STATUS, RuleStatus.REMOVED.name()))))).addAggregation(AggregationBuilders.terms(FIELD_ACTIVE_RULE_PROFILE_KEY).field(RuleIndexDefinition.FIELD_ACTIVE_RULE_PROFILE_KEY).size(0).subAggregation(AggregationBuilders.terms(FIELD_ACTIVE_RULE_INHERITANCE).field(RuleIndexDefinition.FIELD_ACTIVE_RULE_INHERITANCE)).subAggregation(AggregationBuilders.terms(FIELD_ACTIVE_RULE_SEVERITY).field(RuleIndexDefinition.FIELD_ACTIVE_RULE_SEVERITY)).subAggregation(AggregationBuilders.count(COUNT_ACTIVE_RULES))).setSize(0).setTypes(TYPE_ACTIVE_RULE);
    SearchResponse response = request.get();
    Map<String, Multimap<String, FacetValue>> stats = new HashMap<>();
    Aggregation aggregation = response.getAggregations().get(FIELD_ACTIVE_RULE_PROFILE_KEY);
    for (Terms.Bucket value : ((Terms) aggregation).getBuckets()) {
        stats.put(value.getKeyAsString(), processAggregations(value.getAggregations()));
    }
    return stats;
}

35. ActiveRuleIndex#countByField()

Project: sonarqube
File: ActiveRuleIndex.java
private Map<String, Long> countByField(String indexField, QueryBuilder filter) {
    Map<String, Long> counts = new HashMap<>();
    SearchRequestBuilder request = getClient().prepareSearch(INDEX).setTypes(TYPE_ACTIVE_RULE).setQuery(QueryBuilders.filteredQuery(QueryBuilders.matchAllQuery(), filter)).setSize(0).addAggregation(AggregationBuilders.terms(indexField).field(indexField).order(Terms.Order.count(false)).size(Integer.MAX_VALUE).minDocCount(0));
    SearchResponse response = request.get();
    Terms values = response.getAggregations().get(indexField);
    for (Terms.Bucket value : values.getBuckets()) {
        counts.put(value.getKeyAsString(), value.getDocCount());
    }
    return counts;
}

36. IssueIndex#listTermsMatching()

Project: sonarqube
File: IssueIndex.java
private Terms listTermsMatching(String fieldName, IssueQuery query, @Nullable String textQuery, Terms.Order termsOrder, int maxNumberOfTags) {
    SearchRequestBuilder requestBuilder = getClient().prepareSearch(IssueIndexDefinition.INDEX).setSize(// Avoids returning search hits
    0).setTypes(IssueIndexDefinition.TYPE_ISSUE);
    requestBuilder.setQuery(boolQuery().must(QueryBuilders.matchAllQuery()).filter(createBoolFilter(query)));
    TermsBuilder aggreg = AggregationBuilders.terms("_ref").field(fieldName).size(maxNumberOfTags).order(termsOrder).minDocCount(1L);
    if (textQuery != null) {
        aggreg.include(String.format(SUBSTRING_MATCH_REGEXP, textQuery));
    }
    SearchResponse searchResponse = requestBuilder.addAggregation(aggreg).get();
    return searchResponse.getAggregations().get("_ref");
}

37. ElasticSearchIndexBuilder#getResourceNames()

Project: sakai
File: ElasticSearchIndexBuilder.java
/**
     * Get all indexed resources for a site
     *
     * @param siteId Site containing indexed resources
     * @return a collection of resource references or an empty collection if no resource was found
     */
protected Collection<String> getResourceNames(String siteId) {
    log.debug("Obtaining indexed elements for site: '" + siteId + "'");
    SearchResponse response = client.prepareSearch(indexName).setSearchType(SearchType.QUERY_THEN_FETCH).setQuery(termQuery(SearchService.FIELD_SITEID, siteId)).setTypes(ElasticSearchService.SAKAI_DOC_TYPE).setSize(Integer.MAX_VALUE).addFields(SearchService.FIELD_REFERENCE).execute().actionGet();
    Collection<String> resourceNames = new ArrayList<String>();
    for (SearchHit hit : response.getHits().hits()) {
        resourceNames.add(getFieldFromSearchHit(SearchService.FIELD_REFERENCE, hit));
    }
    return resourceNames;
}

38. IndexQuery#executeSearchRequest()

Project: play2-elasticsearch
File: IndexQuery.java
public IndexResults<T> executeSearchRequest(SearchRequestBuilder request) {
    SearchResponse searchResponse = request.execute().actionGet();
    if (IndexClient.config.showRequest) {
        Logger.debug("ElasticSearch : Response -> " + searchResponse.toString());
    }
    IndexResults<T> searchResults = toSearchResults(searchResponse);
    return searchResults;
}

39. ESIndexer#search()

Project: Kundera
File: ESIndexer.java
/*
     * (non-Javadoc)
     * 
     * @see
     * com.impetus.kundera.index.Indexer#search(com.impetus.kundera.persistence
     * .EntityManagerFactoryImpl.KunderaMetadata,
     * com.impetus.kundera.query.KunderaQuery,
     * com.impetus.kundera.persistence.PersistenceDelegator,
     * com.impetus.kundera.metadata.model.EntityMetadata)
     */
@Override
public Map<String, Object> search(KunderaMetadata kunderaMetadata, KunderaQuery kunderaQuery, PersistenceDelegator persistenceDelegator, EntityMetadata m, int maxResults) {
    ESQuery query = new ESQuery<>(kunderaQuery, persistenceDelegator, kunderaMetadata);
    MetamodelImpl metaModel = (MetamodelImpl) kunderaMetadata.getApplicationMetadata().getMetamodel(m.getPersistenceUnit());
    Expression whereExpression = KunderaQueryUtils.getWhereClause(kunderaQuery.getJpqlExpression());
    QueryBuilder filter = whereExpression != null ? query.getEsFilterBuilder().populateFilterBuilder(((WhereClause) whereExpression).getConditionalExpression(), m) : null;
    FilteredQueryBuilder queryBuilder = QueryBuilders.filteredQuery(null, filter);
    SearchResponse response = getSearchResponse(kunderaQuery, queryBuilder, filter, query, m, maxResults, kunderaMetadata);
    return buildResultMap(response, kunderaQuery, m, metaModel);
}

40. ESClient#findIdsByColumn()

Project: Kundera
File: ESClient.java
/*
     * (non-Javadoc)
     * 
     * @see com.impetus.kundera.client.Client#findIdsByColumn(java.lang.String,
     * java.lang.String, java.lang.String, java.lang.String, java.lang.Object,
     * java.lang.Class)
     */
@Override
public Object[] findIdsByColumn(String schemaName, String tableName, String pKeyName, String columnName, Object columnValue, Class entityClazz) {
    TermQueryBuilder filter = QueryBuilders.termQuery(columnName, columnValue);
    SearchResponse response = txClient.prepareSearch(schemaName.toLowerCase()).setTypes(tableName).addField(pKeyName).setPostFilter(filter).execute().actionGet();
    SearchHits hits = response.getHits();
    Long length = hits.getTotalHits();
    int absoluteLength = length.intValue();
    Object[] ids = new Object[absoluteLength];
    int counter = 0;
    for (SearchHit hit : hits.getHits()) {
        Map<String, SearchHitField> fields = hit.getFields();
        ids[counter++] = fields.get(pKeyName).getValue();
    }
    return ids;
}

41. ESClient#getColumnsById()

Project: Kundera
File: ESClient.java
/*
     * (non-Javadoc)
     * 
     * @see com.impetus.kundera.client.Client#getColumnsById(java.lang.String,
     * java.lang.String, java.lang.String, java.lang.String, java.lang.Object,
     * java.lang.Class)
     */
@Override
public <E> List<E> getColumnsById(String schemaName, String tableName, String pKeyColumnName, String columnName, Object pKeyColumnValue, Class columnJavaType) {
    // fetch list ADDRESS_ID for given PERSON_ID
    QueryBuilder filterBuilder = new TermQueryBuilder(pKeyColumnName, pKeyColumnValue);
    SearchResponse response = txClient.prepareSearch(schemaName.toLowerCase()).setTypes(tableName).setPostFilter(filterBuilder).addField(columnName).execute().actionGet();
    SearchHits hits = response.getHits();
    List columns = new ArrayList();
    for (SearchHit hit : hits.getHits()) {
        Map<String, SearchHitField> fields = hit.getFields();
        columns.add(fields.get(columnName).getValue());
    }
    return columns;
}

42. Searches#search()

Project: graylog2-server
File: Searches.java
public SearchResult search(SearchesConfig config) {
    Set<IndexRange> indices = IndexHelper.determineAffectedIndicesWithRanges(indexRangeService, deflector, config.range());
    Set<String> indexNames = Sets.newHashSet();
    for (IndexRange index : indices) {
        indexNames.add(index.indexName());
    }
    SearchRequest request = searchRequest(config, indexNames).request();
    SearchResponse r = c.search(request).actionGet();
    recordEsMetrics(r, config.range());
    return new SearchResult(r.getHits(), indices, config.query(), request.source(), r.getTook());
}

43. Searches#count()

Project: graylog2-server
File: Searches.java
public CountResult count(String query, TimeRange range, String filter) {
    Set<String> indices = IndexHelper.determineAffectedIndices(indexRangeService, deflector, range);
    final SearchRequestBuilder srb;
    if (filter == null) {
        srb = standardSearchRequest(query, indices, range);
    } else {
        srb = filteredSearchRequest(query, filter, indices, range);
    }
    srb.setSize(0);
    final SearchResponse r = c.search(srb.request()).actionGet();
    recordEsMetrics(r, range);
    return CountResult.create(r.getHits().getTotalHits(), r.getTookInMillis());
}

44. ScrollResult#nextChunk()

Project: graylog2-server
File: ScrollResult.java
public ScrollChunk nextChunk() {
    final SearchResponse search;
    // make sure to return the initial hits, see https://github.com/Graylog2/graylog2-server/issues/2126
    if (firstResponse == null) {
        search = client.prepareSearchScroll(scrollId).setScroll(TimeValue.timeValueMinutes(1)).execute().actionGet();
    } else {
        search = firstResponse;
        firstResponse = null;
    }
    final SearchHits hits = search.getHits();
    if (hits.getHits().length == 0) {
        // scroll exhausted
        LOG.debug("[{}] Reached end of scroll results.", queryHash, getOriginalQuery());
        return null;
    }
    LOG.debug("[{}][{}] New scroll id {}, number of hits in chunk: {}", queryHash, chunkId, search.getScrollId(), hits.getHits().length);
    // save the id for the next request.
    scrollId = search.getScrollId();
    return new ScrollChunk(hits, fields, chunkId++);
}

45. ElasticsearchDataModel#loadValueStats()

Project: elasticsearch-taste
File: ElasticsearchDataModel.java
protected synchronized void loadValueStats() {
    if (stats != null) {
        return;
    }
    // TODO join userQueryBuilder and itemQueryBuilder
    final SearchResponse response = client.prepareSearch(preferenceIndex).setTypes(preferenceType).setQuery(getLastAccessedFilterQuery()).setSize(0).addAggregation(AggregationBuilders.stats(valueField).field(valueField)).execute().actionGet();
    final Aggregations aggregations = response.getAggregations();
    stats = aggregations.get(valueField);
}

46. HighlighterSearchIT#testGlobalHighlightingSettingsOverriddenAtFieldLevel()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testGlobalHighlightingSettingsOverriddenAtFieldLevel() {
    createIndex("test");
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", new String[] { "this is a test", "this is the second test" }, "field2", new String[] { "this is another test", "yet another test" }).get();
    refresh();
    logger.info("--> highlighting and searching on field1 and field2 produces different tags");
    SearchSourceBuilder source = searchSource().query(termQuery("field1", "test")).highlighter(highlight().order("score").preTags("<global>").postTags("</global>").fragmentSize(1).numOfFragments(1).field(new HighlightBuilder.Field("field1").numOfFragments(2)).field(new HighlightBuilder.Field("field2").preTags("<field2>").postTags("</field2>").fragmentSize(50).requireFieldMatch(false)));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field1", 0, 2, equalTo(" <global>test</global>"));
    assertHighlight(searchResponse, 0, "field1", 1, 2, equalTo(" <global>test</global>"));
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("this is another <field2>test</field2>"));
}

47. HighlighterSearchIT#testHighTermFrequencyDoc()

Project: elasticsearch
File: HighlighterSearchIT.java
// see #3486
public void testHighTermFrequencyDoc() throws IOException {
    assertAcked(prepareCreate("test").addMapping("test", "name", "type=text,term_vector=with_positions_offsets,store=" + randomBoolean()));
    ensureYellow();
    StringBuilder builder = new StringBuilder();
    for (int i = 0; i < 6000; i++) {
        builder.append("abc").append(" ");
    }
    client().prepareIndex("test", "test", "1").setSource("name", builder.toString()).get();
    refresh();
    SearchResponse search = client().prepareSearch().setQuery(constantScoreQuery(matchQuery("name", "abc"))).highlighter(new HighlightBuilder().field("name")).get();
    assertHighlight(search, 0, "name", 0, startsWith("<em>abc</em> <em>abc</em> <em>abc</em> <em>abc</em>"));
}

48. CustomHighlighterSearchIT#testThatCustomHighlighterReceivesFieldsInOrder()

Project: elasticsearch
File: CustomHighlighterSearchIT.java
public void testThatCustomHighlighterReceivesFieldsInOrder() throws Exception {
    SearchResponse searchResponse = client().prepareSearch("test").setTypes("test").setQuery(QueryBuilders.boolQuery().must(QueryBuilders.matchAllQuery()).should(QueryBuilders.termQuery("name", "arbitrary"))).highlighter(new HighlightBuilder().highlighterType("test-custom").field("name").field("other_name").field("other_other_name").useExplicitFieldOrder(true)).get();
    assertHighlight(searchResponse, 0, "name", 0, equalTo("standard response for name at position 1"));
    assertHighlight(searchResponse, 0, "other_name", 0, equalTo("standard response for other_name at position 2"));
    assertHighlight(searchResponse, 0, "other_other_name", 0, equalTo("standard response for other_other_name at position 3"));
    assertHighlight(searchResponse, 1, "name", 0, equalTo("standard response for name at position 1"));
    assertHighlight(searchResponse, 1, "other_name", 0, equalTo("standard response for other_name at position 2"));
    assertHighlight(searchResponse, 1, "other_other_name", 0, equalTo("standard response for other_other_name at position 3"));
}

49. CustomHighlighterSearchIT#testThatCustomHighlighterCanBeConfiguredPerField()

Project: elasticsearch
File: CustomHighlighterSearchIT.java
public void testThatCustomHighlighterCanBeConfiguredPerField() throws Exception {
    HighlightBuilder.Field highlightConfig = new HighlightBuilder.Field("name");
    highlightConfig.highlighterType("test-custom");
    Map<String, Object> options = new HashMap<>();
    options.put("myFieldOption", "someValue");
    highlightConfig.options(options);
    SearchResponse searchResponse = client().prepareSearch("test").setTypes("test").setQuery(QueryBuilders.matchAllQuery()).highlighter(new HighlightBuilder().field(highlightConfig)).execute().actionGet();
    assertHighlight(searchResponse, 0, "name", 0, equalTo("standard response for name at position 1"));
    assertHighlight(searchResponse, 0, "name", 1, equalTo("field:myFieldOption:someValue"));
}

50. GeoShapeQueryTests#testContainsShapeQuery()

Project: elasticsearch
File: GeoShapeQueryTests.java
public void testContainsShapeQuery() throws Exception {
    // Create a random geometry collection.
    Rectangle mbr = xRandomRectangle(random(), xRandomPoint(random()), true);
    GeometryCollectionBuilder gcb = createGeometryCollectionWithin(random(), mbr);
    client().admin().indices().prepareCreate("test").addMapping("type", "location", "type=geo_shape,tree=quadtree").execute().actionGet();
    XContentBuilder docSource = gcb.toXContent(jsonBuilder().startObject().field("location"), null).endObject();
    client().prepareIndex("test", "type", "1").setSource(docSource).setRefreshPolicy(IMMEDIATE).get();
    // index the mbr of the collection
    EnvelopeBuilder env = new EnvelopeBuilder(new Coordinate(mbr.getMinX(), mbr.getMaxY()), new Coordinate(mbr.getMaxX(), mbr.getMinY()));
    docSource = env.toXContent(jsonBuilder().startObject().field("location"), null).endObject();
    client().prepareIndex("test", "type", "2").setSource(docSource).setRefreshPolicy(IMMEDIATE).get();
    ShapeBuilder filterShape = (gcb.getShapeAt(randomIntBetween(0, gcb.numShapes() - 1)));
    GeoShapeQueryBuilder filter = QueryBuilders.geoShapeQuery("location", filterShape).relation(ShapeRelation.CONTAINS);
    SearchResponse response = client().prepareSearch("test").setTypes("type").setQuery(QueryBuilders.matchAllQuery()).setPostFilter(filter).get();
    assertSearchResponse(response);
    assertThat(response.getHits().totalHits(), greaterThan(0L));
}

51. GeoShapeQueryTests#testShapeFilterWithRandomGeoCollection()

Project: elasticsearch
File: GeoShapeQueryTests.java
public void testShapeFilterWithRandomGeoCollection() throws Exception {
    // Create a random geometry collection.
    GeometryCollectionBuilder gcb = RandomShapeGenerator.createGeometryCollection(random());
    logger.info("Created Random GeometryCollection containing {} shapes", gcb.numShapes());
    client().admin().indices().prepareCreate("test").addMapping("type", "location", "type=geo_shape,tree=quadtree").execute().actionGet();
    XContentBuilder docSource = gcb.toXContent(jsonBuilder().startObject().field("location"), null).endObject();
    client().prepareIndex("test", "type", "1").setSource(docSource).setRefreshPolicy(IMMEDIATE).get();
    ShapeBuilder filterShape = (gcb.getShapeAt(randomIntBetween(0, gcb.numShapes() - 1)));
    GeoShapeQueryBuilder filter = QueryBuilders.geoShapeQuery("location", filterShape);
    filter.relation(ShapeRelation.INTERSECTS);
    SearchResponse result = client().prepareSearch("test").setTypes("type").setQuery(QueryBuilders.matchAllQuery()).setPostFilter(filter).get();
    assertSearchResponse(result);
    assertHitCount(result, 1);
}

52. FunctionScoreBackwardCompatibilityIT#checkFunctionScoreStillWorks()

Project: elasticsearch
File: FunctionScoreBackwardCompatibilityIT.java
private void checkFunctionScoreStillWorks(String... ids) throws ExecutionException, InterruptedException, IOException {
    SearchResponse response = client().search(searchRequest().source(searchSource().query(functionScoreQuery(termQuery("text", "value"), new FilterFunctionBuilder[] { new FilterFunctionBuilder(gaussDecayFunction("loc", new GeoPoint(10, 20), "1000km")), new FilterFunctionBuilder(scriptFunction(new Script("_index['text']['value'].tf()"))), new FilterFunctionBuilder(termQuery("text", "boosted"), weightFactorFunction(5)) })))).actionGet();
    assertSearchResponse(response);
    assertOrderedSearchHits(response, ids);
}

53. ChildQuerySearchIT#testParentFieldInMultiMatchField()

Project: elasticsearch
File: ChildQuerySearchIT.java
// Issue #6256
public void testParentFieldInMultiMatchField() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1").addMapping("type2", "_parent", "type=type1"));
    ensureGreen();
    client().prepareIndex("test", "type2", "1").setParent("1").setSource("field", "value").get();
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(multiMatchQuery("1", "_parent#type1")).get();
    assertThat(response.getHits().totalHits(), equalTo(1L));
    assertThat(response.getHits().getAt(0).id(), equalTo("1"));
}

54. ChildQuerySearchIT#testQueryBeforeChildType()

Project: elasticsearch
File: ChildQuerySearchIT.java
// Issue #5783
public void testQueryBeforeChildType() throws Exception {
    assertAcked(prepareCreate("test").addMapping("features").addMapping("posts", "_parent", "type=features").addMapping("specials"));
    ensureGreen();
    client().prepareIndex("test", "features", "1").setSource("field", "foo").get();
    client().prepareIndex("test", "posts", "1").setParent("1").setSource("field", "bar").get();
    refresh();
    SearchResponse resp;
    resp = client().prepareSearch("test").setSource(new SearchSourceBuilder().query(QueryBuilders.hasChildQuery("posts", QueryBuilders.matchQuery("field", "bar"), ScoreMode.None))).get();
    assertHitCount(resp, 1L);
}

55. ChildQuerySearchIT#testParentIdQuery()

Project: elasticsearch
File: ChildQuerySearchIT.java
public void testParentIdQuery() throws Exception {
    assertAcked(prepareCreate("test").setSettings(Settings.builder().put(indexSettings()).put("index.refresh_interval", -1)).addMapping("parent").addMapping("child", "_parent", "type=parent"));
    ensureGreen();
    client().prepareIndex("test", "child", "c1").setSource("{}").setParent("p1").get();
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(parentId("child", "p1")).get();
    assertHitCount(response, 1L);
    client().prepareIndex("test", "child", "c2").setSource("{}").setParent("p2").get();
    refresh();
    response = client().prepareSearch("test").setQuery(boolQuery().should(parentId("child", "p1")).should(parentId("child", "p2"))).get();
    assertHitCount(response, 2L);
}

56. ChildQuerySearchIT#testHasChildQueryWithMinimumScore()

Project: elasticsearch
File: ChildQuerySearchIT.java
// Issue #3203
public void testHasChildQueryWithMinimumScore() throws Exception {
    assertAcked(prepareCreate("test").addMapping("parent").addMapping("child", "_parent", "type=parent"));
    ensureGreen();
    // index simple data
    client().prepareIndex("test", "parent", "p1").setSource("p_field", "p_value1").get();
    client().prepareIndex("test", "child", "c1").setSource("c_field", "x").setParent("p1").get();
    client().prepareIndex("test", "parent", "p2").setSource("p_field", "p_value2").get();
    client().prepareIndex("test", "child", "c3").setSource("c_field", "x").setParent("p2").get();
    client().prepareIndex("test", "child", "c4").setSource("c_field", "x").setParent("p2").get();
    client().prepareIndex("test", "child", "c5").setSource("c_field", "x").setParent("p2").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch("test").setQuery(hasChildQuery("child", matchAllQuery(), ScoreMode.Total)).setMinScore(// Score needs to be 3 or above!
    3).get();
    assertNoFailures(searchResponse);
    assertThat(searchResponse.getHits().totalHits(), equalTo(1L));
    assertThat(searchResponse.getHits().getAt(0).id(), equalTo("p2"));
    assertThat(searchResponse.getHits().getAt(0).score(), equalTo(3.0f));
}

57. ChildQuerySearchIT#testHasChildAndHasParentFailWhenSomeSegmentsDontContainAnyParentOrChildDocs()

Project: elasticsearch
File: ChildQuerySearchIT.java
public void testHasChildAndHasParentFailWhenSomeSegmentsDontContainAnyParentOrChildDocs() throws Exception {
    assertAcked(prepareCreate("test").addMapping("parent").addMapping("child", "_parent", "type=parent"));
    ensureGreen();
    client().prepareIndex("test", "parent", "1").setSource("p_field", 1).get();
    client().prepareIndex("test", "child", "1").setParent("1").setSource("c_field", 1).get();
    client().admin().indices().prepareFlush("test").get();
    client().prepareIndex("test", "type1", "1").setSource("p_field", 1).get();
    client().admin().indices().prepareFlush("test").get();
    SearchResponse searchResponse = client().prepareSearch("test").setQuery(boolQuery().must(matchAllQuery()).filter(hasChildQuery("child", matchAllQuery(), ScoreMode.None))).get();
    assertNoFailures(searchResponse);
    assertThat(searchResponse.getHits().totalHits(), equalTo(1L));
    searchResponse = client().prepareSearch("test").setQuery(boolQuery().must(matchAllQuery()).filter(hasParentQuery("parent", matchAllQuery(), false))).get();
    assertNoFailures(searchResponse);
    assertThat(searchResponse.getHits().totalHits(), equalTo(1L));
}

58. ChildQuerySearchIT#test2744()

Project: elasticsearch
File: ChildQuerySearchIT.java
// see #2744
public void test2744() throws IOException {
    assertAcked(prepareCreate("test").addMapping("foo").addMapping("test", "_parent", "type=foo"));
    ensureGreen();
    // index simple data
    client().prepareIndex("test", "foo", "1").setSource("foo", 1).get();
    client().prepareIndex("test", "test").setSource("foo", 1).setParent("1").get();
    refresh();
    SearchResponse searchResponse = client().prepareSearch("test").setQuery(hasChildQuery("test", matchQuery("foo", 1), ScoreMode.None)).get();
    assertNoFailures(searchResponse);
    assertThat(searchResponse.getHits().totalHits(), equalTo(1L));
    assertThat(searchResponse.getHits().getAt(0).id(), equalTo("1"));
}

59. TransportTwoNodesSearchIT#testSimpleFacets()

Project: elasticsearch
File: TransportTwoNodesSearchIT.java
public void testSimpleFacets() throws Exception {
    prepareData();
    SearchSourceBuilder sourceBuilder = searchSource().query(termQuery("multi", "test")).from(0).size(20).explain(true).aggregation(AggregationBuilders.global("global").subAggregation(AggregationBuilders.filter("all", termQuery("multi", "test")))).aggregation(AggregationBuilders.filter("test1", termQuery("name", "test1")));
    SearchResponse searchResponse = client().search(searchRequest("test").source(sourceBuilder)).actionGet();
    assertNoFailures(searchResponse);
    assertThat(searchResponse.getHits().totalHits(), equalTo(100L));
    Global global = searchResponse.getAggregations().get("global");
    Filter all = global.getAggregations().get("all");
    Filter test1 = searchResponse.getAggregations().get("test1");
    assertThat(test1.getDocCount(), equalTo(1L));
    assertThat(all.getDocCount(), equalTo(100L));
}

60. TransportTwoNodesSearchIT#testQueryThenFetchWithSort()

Project: elasticsearch
File: TransportTwoNodesSearchIT.java
public void testQueryThenFetchWithSort() throws Exception {
    prepareData();
    int total = 0;
    SearchResponse searchResponse = client().prepareSearch("test").setQuery(termQuery("multi", "test")).setSize(60).setExplain(true).addSort("age", SortOrder.ASC).setScroll(TimeValue.timeValueSeconds(30)).get();
    while (true) {
        assertNoFailures(searchResponse);
        assertThat(searchResponse.getHits().totalHits(), equalTo(100L));
        SearchHit[] hits = searchResponse.getHits().hits();
        if (hits.length == 0) {
            // finished
            break;
        }
        for (int i = 0; i < hits.length; ++i) {
            SearchHit hit = hits[i];
            assertThat(hit.explanation(), notNullValue());
            assertThat("id[" + hit.id() + "]", hit.id(), equalTo(Integer.toString(total + i)));
        }
        total += hits.length;
        searchResponse = client().prepareSearchScroll(searchResponse.getScrollId()).setScroll(TimeValue.timeValueSeconds(30)).get();
    }
    clearScroll(searchResponse.getScrollId());
    assertEquals(100, total);
}

61. TransportTwoNodesSearchIT#testQueryThenFetch()

Project: elasticsearch
File: TransportTwoNodesSearchIT.java
public void testQueryThenFetch() throws Exception {
    prepareData();
    int total = 0;
    SearchResponse searchResponse = client().prepareSearch("test").setSearchType(QUERY_THEN_FETCH).setQuery(termQuery("multi", "test")).setSize(60).setExplain(true).addSort("nid", SortOrder.DESC).setScroll(TimeValue.timeValueSeconds(30)).get();
    while (true) {
        assertNoFailures(searchResponse);
        assertThat(searchResponse.getHits().totalHits(), equalTo(100L));
        SearchHit[] hits = searchResponse.getHits().hits();
        if (hits.length == 0) {
            // finished
            break;
        }
        for (int i = 0; i < hits.length; ++i) {
            SearchHit hit = hits[i];
            assertThat(hit.explanation(), notNullValue());
            assertThat("id[" + hit.id() + "]", hit.id(), equalTo(Integer.toString(100 - total - i - 1)));
        }
        total += hits.length;
        searchResponse = client().prepareSearchScroll(searchResponse.getScrollId()).setScroll(TimeValue.timeValueSeconds(30)).get();
    }
    clearScroll(searchResponse.getScrollId());
    assertEquals(100, total);
}

62. TransportTwoNodesSearchIT#testDfsQueryThenFetchWithSort()

Project: elasticsearch
File: TransportTwoNodesSearchIT.java
public void testDfsQueryThenFetchWithSort() throws Exception {
    prepareData();
    int total = 0;
    SearchResponse searchResponse = client().prepareSearch("test").setSearchType(DFS_QUERY_THEN_FETCH).setQuery(termQuery("multi", "test")).setSize(60).setExplain(true).addSort("age", SortOrder.ASC).setScroll(TimeValue.timeValueSeconds(30)).get();
    while (true) {
        assertNoFailures(searchResponse);
        assertThat(searchResponse.getHits().totalHits(), equalTo(100L));
        SearchHit[] hits = searchResponse.getHits().hits();
        if (hits.length == 0) {
            // finished
            break;
        }
        for (int i = 0; i < hits.length; ++i) {
            SearchHit hit = hits[i];
            assertThat(hit.explanation(), notNullValue());
            assertThat("id[" + hit.id() + "]", hit.id(), equalTo(Integer.toString(total + i)));
        }
        total += hits.length;
        searchResponse = client().prepareSearchScroll(searchResponse.getScrollId()).setScroll(TimeValue.timeValueSeconds(30)).get();
    }
    clearScroll(searchResponse.getScrollId());
    assertEquals(100, total);
}

63. SumBucketIT#testNoBuckets()

Project: elasticsearch
File: SumBucketIT.java
public void testNoBuckets() throws Exception {
    SearchResponse response = client().prepareSearch("idx").addAggregation(terms("terms").field("tag").includeExclude(new IncludeExclude(null, "tag.*")).subAggregation(sum("sum").field(SINGLE_VALUED_FIELD_NAME))).addAggregation(sumBucket("sum_bucket", "terms>sum")).execute().actionGet();
    assertSearchResponse(response);
    Terms terms = response.getAggregations().get("terms");
    assertThat(terms, notNullValue());
    assertThat(terms.getName(), equalTo("terms"));
    List<Terms.Bucket> buckets = terms.getBuckets();
    assertThat(buckets.size(), equalTo(0));
    InternalSimpleValue sumBucketValue = response.getAggregations().get("sum_bucket");
    assertThat(sumBucketValue, notNullValue());
    assertThat(sumBucketValue.getName(), equalTo("sum_bucket"));
    assertThat(sumBucketValue.value(), equalTo(0.0));
}

64. StatsBucketIT#testNoBuckets()

Project: elasticsearch
File: StatsBucketIT.java
public void testNoBuckets() throws Exception {
    SearchResponse response = client().prepareSearch("idx").addAggregation(terms("terms").field("tag").includeExclude(new IncludeExclude(null, "tag.*")).subAggregation(sum("sum").field(SINGLE_VALUED_FIELD_NAME))).addAggregation(statsBucket("stats_bucket", "terms>sum")).execute().actionGet();
    assertSearchResponse(response);
    Terms terms = response.getAggregations().get("terms");
    assertThat(terms, notNullValue());
    assertThat(terms.getName(), equalTo("terms"));
    List<Terms.Bucket> buckets = terms.getBuckets();
    assertThat(buckets.size(), equalTo(0));
    StatsBucket statsBucketValue = response.getAggregations().get("stats_bucket");
    assertThat(statsBucketValue, notNullValue());
    assertThat(statsBucketValue.getName(), equalTo("stats_bucket"));
    assertThat(statsBucketValue.getAvg(), equalTo(Double.NaN));
}

65. PercentilesBucketIT#testNoBuckets()

Project: elasticsearch
File: PercentilesBucketIT.java
public void testNoBuckets() throws Exception {
    SearchResponse response = client().prepareSearch("idx").addAggregation(terms("terms").field("tag").includeExclude(new IncludeExclude(null, "tag.*")).subAggregation(sum("sum").field(SINGLE_VALUED_FIELD_NAME))).addAggregation(percentilesBucket("percentiles_bucket", "terms>sum").percents(PERCENTS)).execute().actionGet();
    assertSearchResponse(response);
    Terms terms = response.getAggregations().get("terms");
    assertThat(terms, notNullValue());
    assertThat(terms.getName(), equalTo("terms"));
    List<Terms.Bucket> buckets = terms.getBuckets();
    assertThat(buckets.size(), equalTo(0));
    PercentilesBucket percentilesBucketValue = response.getAggregations().get("percentiles_bucket");
    assertThat(percentilesBucketValue, notNullValue());
    assertThat(percentilesBucketValue.getName(), equalTo("percentiles_bucket"));
    for (Double p : PERCENTS) {
        assertThat(percentilesBucketValue.percentile(p), equalTo(Double.NaN));
    }
}

66. MovAvgIT#testNoBucketsInHistogramWithPredict()

Project: elasticsearch
File: MovAvgIT.java
public void testNoBucketsInHistogramWithPredict() {
    int numPredictions = randomIntBetween(1, 10);
    SearchResponse response = client().prepareSearch("idx").setTypes("type").addAggregation(histogram("histo").field("test").interval(interval).subAggregation(randomMetric("the_metric", VALUE_FIELD)).subAggregation(movingAvg("movavg_counts", "the_metric").window(windowSize).modelBuilder(new SimpleModel.SimpleModelBuilder()).gapPolicy(gapPolicy).predict(numPredictions))).execute().actionGet();
    assertSearchResponse(response);
    InternalHistogram<Bucket> histo = response.getAggregations().get("histo");
    assertThat(histo, notNullValue());
    assertThat(histo.getName(), equalTo("histo"));
    List<? extends Bucket> buckets = histo.getBuckets();
    assertThat(buckets.size(), equalTo(0));
}

67. MovAvgIT#testNoBucketsInHistogram()

Project: elasticsearch
File: MovAvgIT.java
public void testNoBucketsInHistogram() {
    SearchResponse response = client().prepareSearch("idx").setTypes("type").addAggregation(histogram("histo").field("test").interval(interval).subAggregation(randomMetric("the_metric", VALUE_FIELD)).subAggregation(movingAvg("movavg_counts", "the_metric").window(windowSize).modelBuilder(new SimpleModel.SimpleModelBuilder()).gapPolicy(gapPolicy))).execute().actionGet();
    assertSearchResponse(response);
    InternalHistogram<Bucket> histo = response.getAggregations().get("histo");
    assertThat(histo, notNullValue());
    assertThat(histo.getName(), equalTo("histo"));
    List<? extends Bucket> buckets = histo.getBuckets();
    assertThat(buckets.size(), equalTo(0));
}

68. QueryProfilerIT#testNoProfile()

Project: elasticsearch
File: QueryProfilerIT.java
/**
     * This test makes sure no profile results are returned when profiling is disabled
     */
public void testNoProfile() throws Exception {
    createIndex("test");
    ensureGreen();
    int numDocs = randomIntBetween(100, 150);
    IndexRequestBuilder[] docs = new IndexRequestBuilder[numDocs];
    for (int i = 0; i < numDocs; i++) {
        docs[i] = client().prepareIndex("test", "type1", String.valueOf(i)).setSource("field1", English.intToEnglish(i), "field2", i);
    }
    indexRandom(true, docs);
    refresh();
    QueryBuilder q = QueryBuilders.rangeQuery("field2").from(0).to(5);
    logger.info("Query: {}", q);
    SearchResponse resp = client().prepareSearch().setQuery(q).setProfile(false).execute().actionGet();
    assertThat("Profile response element should be an empty map", resp.getProfileResults().size(), equalTo(0));
}

69. AggregationProfilerIT#testNoProfile()

Project: elasticsearch
File: AggregationProfilerIT.java
public void testNoProfile() {
    SearchResponse response = client().prepareSearch("idx").setProfile(false).addAggregation(histogram("histo").field(NUMBER_FIELD).interval(1L).subAggregation(terms("tags").field(TAG_FIELD).subAggregation(avg("avg").field(NUMBER_FIELD)).subAggregation(max("max").field(NUMBER_FIELD))).subAggregation(terms("strings").field(STRING_FIELD).subAggregation(avg("avg").field(NUMBER_FIELD)).subAggregation(max("max").field(NUMBER_FIELD)).subAggregation(terms("tags").field(TAG_FIELD).subAggregation(avg("avg").field(NUMBER_FIELD)).subAggregation(max("max").field(NUMBER_FIELD))))).get();
    assertSearchResponse(response);
    Map<String, ProfileShardResult> profileResults = response.getProfileResults();
    assertThat(profileResults, notNullValue());
    assertThat(profileResults.size(), equalTo(0));
}

70. SearchPreferenceIT#testReplicaPreference()

Project: elasticsearch
File: SearchPreferenceIT.java
public void testReplicaPreference() throws Exception {
    client().admin().indices().prepareCreate("test").setSettings("number_of_replicas=0").get();
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "value1").execute().actionGet();
    refresh();
    try {
        client().prepareSearch().setQuery(matchAllQuery()).setPreference("_replica").execute().actionGet();
        fail("should have failed because there are no replicas");
    } catch (Exception e) {
    }
    SearchResponse resp = client().prepareSearch().setQuery(matchAllQuery()).setPreference("_replica_first").execute().actionGet();
    assertThat(resp.getHits().totalHits(), equalTo(1L));
    client().admin().indices().prepareUpdateSettings("test").setSettings("number_of_replicas=1").get();
    ensureGreen("test");
    resp = client().prepareSearch().setQuery(matchAllQuery()).setPreference("_replica").execute().actionGet();
    assertThat(resp.getHits().totalHits(), equalTo(1L));
}

71. SearchPreferenceIT#testNoPreferenceRandom()

Project: elasticsearch
File: SearchPreferenceIT.java
public void testNoPreferenceRandom() throws Exception {
    assertAcked(prepareCreate("test").setSettings(//this test needs at least a replica to make sure two consecutive searches go to two different copies of the same data
    Settings.builder().put(indexSettings()).put(SETTING_NUMBER_OF_REPLICAS, between(1, maximumNumberOfReplicas()))));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "value1").execute().actionGet();
    refresh();
    final Client client = internalCluster().smartClient();
    SearchResponse searchResponse = client.prepareSearch("test").setQuery(matchAllQuery()).execute().actionGet();
    String firstNodeId = searchResponse.getHits().getAt(0).shard().nodeId();
    searchResponse = client.prepareSearch("test").setQuery(matchAllQuery()).execute().actionGet();
    String secondNodeId = searchResponse.getHits().getAt(0).shard().nodeId();
    assertThat(firstNodeId, not(equalTo(secondNodeId)));
}

72. MoreLikeThisIT#testMoreLikeThisArtificialDocs()

Project: elasticsearch
File: MoreLikeThisIT.java
public void testMoreLikeThisArtificialDocs() throws Exception {
    int numFields = randomIntBetween(5, 10);
    createIndex("test");
    ensureGreen();
    logger.info("Indexing a single document ...");
    XContentBuilder doc = jsonBuilder().startObject();
    for (int i = 0; i < numFields; i++) {
        // make sure they are not all empty
        doc.field("field" + i, generateRandomStringArray(5, 10, false) + "a");
    }
    doc.endObject();
    indexRandom(true, client().prepareIndex("test", "type1", "0").setSource(doc));
    logger.info("Checking the document matches ...");
    MoreLikeThisQueryBuilder mltQuery = // routing to ensure we hit the shard with the doc
    moreLikeThisQuery(new Item[] { new Item("test", "type1", doc).routing("0") }).minTermFreq(0).minDocFreq(0).maxQueryTerms(100).minimumShouldMatch(// strict all terms must match!
    "100%");
    SearchResponse response = client().prepareSearch("test").setTypes("type1").setQuery(mltQuery).get();
    assertSearchResponse(response);
    assertHitCount(response, 1);
}

73. MoreLikeThisIT#testSimpleMoreLikeThisIds()

Project: elasticsearch
File: MoreLikeThisIT.java
public void testSimpleMoreLikeThisIds() throws Exception {
    logger.info("Creating index test");
    assertAcked(prepareCreate("test").addMapping("type1", jsonBuilder().startObject().startObject("type1").startObject("properties").startObject("text").field("type", "text").endObject().endObject().endObject().endObject()));
    logger.info("Running Cluster Health");
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));
    logger.info("Indexing...");
    List<IndexRequestBuilder> builders = new ArrayList<>();
    builders.add(client().prepareIndex("test", "type1").setSource("text", "lucene").setId("1"));
    builders.add(client().prepareIndex("test", "type1").setSource("text", "lucene release").setId("2"));
    builders.add(client().prepareIndex("test", "type1").setSource("text", "apache lucene").setId("3"));
    indexRandom(true, builders);
    logger.info("Running MoreLikeThis");
    MoreLikeThisQueryBuilder queryBuilder = QueryBuilders.moreLikeThisQuery(new String[] { "text" }, null, ids("1")).include(true).minTermFreq(1).minDocFreq(1);
    SearchResponse mltResponse = client().prepareSearch().setTypes("type1").setQuery(queryBuilder).execute().actionGet();
    assertHitCount(mltResponse, 3L);
}

74. MoreLikeThisIT#testMoreLikeThisIssueRoutingNotSerialized()

Project: elasticsearch
File: MoreLikeThisIT.java
// Issue #3039
public void testMoreLikeThisIssueRoutingNotSerialized() throws Exception {
    String mapping = XContentFactory.jsonBuilder().startObject().startObject("bar").startObject("properties").endObject().endObject().endObject().string();
    assertAcked(prepareCreate("foo", 2, Settings.builder().put(SETTING_NUMBER_OF_SHARDS, 2).put(SETTING_NUMBER_OF_REPLICAS, 0)).addMapping("bar", mapping));
    ensureGreen();
    client().prepareIndex("foo", "bar", "1").setSource(jsonBuilder().startObject().startObject("foo").field("bar", "boz").endObject().endObject()).setRouting("4000").execute().actionGet();
    client().admin().indices().prepareRefresh("foo").execute().actionGet();
    SearchResponse response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item("foo", "bar", "1").routing("4000") })).get();
    assertNoFailures(response);
    assertThat(response, notNullValue());
}

75. MoreLikeThisIT#testMoreLikeWithCustomRouting()

Project: elasticsearch
File: MoreLikeThisIT.java
// Issue #2489
public void testMoreLikeWithCustomRouting() throws Exception {
    String mapping = XContentFactory.jsonBuilder().startObject().startObject("bar").startObject("properties").endObject().endObject().endObject().string();
    client().admin().indices().prepareCreate("foo").addMapping("bar", mapping).execute().actionGet();
    ensureGreen();
    client().prepareIndex("foo", "bar", "1").setSource(jsonBuilder().startObject().startObject("foo").field("bar", "boz").endObject().endObject()).setRouting("2").execute().actionGet();
    client().admin().indices().prepareRefresh("foo").execute().actionGet();
    SearchResponse response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item("foo", "bar", "1").routing("2") })).get();
    assertNoFailures(response);
    assertThat(response, notNullValue());
}

76. MoreLikeThisIT#testMoreLikeThisIssue2197()

Project: elasticsearch
File: MoreLikeThisIT.java
public void testMoreLikeThisIssue2197() throws Exception {
    Client client = client();
    String mapping = XContentFactory.jsonBuilder().startObject().startObject("bar").startObject("properties").endObject().endObject().endObject().string();
    client().admin().indices().prepareCreate("foo").addMapping("bar", mapping).execute().actionGet();
    client().prepareIndex("foo", "bar", "1").setSource(jsonBuilder().startObject().startObject("foo").field("bar", "boz").endObject().endObject()).execute().actionGet();
    client().admin().indices().prepareRefresh("foo").execute().actionGet();
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));
    SearchResponse response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item("foo", "bar", "1") })).get();
    assertNoFailures(response);
    assertThat(response, notNullValue());
    response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item("foo", "bar", "1") })).get();
    assertNoFailures(response);
    assertThat(response, notNullValue());
}

77. MoreLikeThisIT#testMoreLikeThisWithAliasesInLikeDocuments()

Project: elasticsearch
File: MoreLikeThisIT.java
// Issue #14944
public void testMoreLikeThisWithAliasesInLikeDocuments() throws Exception {
    String indexName = "foo";
    String aliasName = "foo_name";
    String typeName = "bar";
    String mapping = XContentFactory.jsonBuilder().startObject().startObject("bar").startObject("properties").endObject().endObject().endObject().string();
    client().admin().indices().prepareCreate(indexName).addMapping(typeName, mapping).execute().actionGet();
    client().admin().indices().aliases(indexAliasesRequest().addAlias(aliasName, indexName)).actionGet();
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));
    client().index(indexRequest(indexName).type(typeName).id("1").source(jsonBuilder().startObject().field("text", "elasticsearch index").endObject())).actionGet();
    client().index(indexRequest(indexName).type(typeName).id("2").source(jsonBuilder().startObject().field("text", "lucene index").endObject())).actionGet();
    client().index(indexRequest(indexName).type(typeName).id("3").source(jsonBuilder().startObject().field("text", "elasticsearch index").endObject())).actionGet();
    refresh(indexName);
    SearchResponse response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item(aliasName, typeName, "1") }).minTermFreq(1).minDocFreq(1)).get();
    assertHitCount(response, 2L);
    assertThat(response.getHits().getAt(0).id(), equalTo("3"));
}

78. MoreLikeThisIT#testSimpleMoreLikeOnLongField()

Project: elasticsearch
File: MoreLikeThisIT.java
public void testSimpleMoreLikeOnLongField() throws Exception {
    logger.info("Creating index test");
    assertAcked(prepareCreate("test").addMapping("type1", "some_long", "type=long"));
    logger.info("Running Cluster Health");
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));
    logger.info("Indexing...");
    client().index(indexRequest("test").type("type1").id("1").source(jsonBuilder().startObject().field("some_long", 1367484649580L).endObject())).actionGet();
    client().index(indexRequest("test").type("type2").id("2").source(jsonBuilder().startObject().field("some_long", 0).endObject())).actionGet();
    client().index(indexRequest("test").type("type1").id("3").source(jsonBuilder().startObject().field("some_long", -666).endObject())).actionGet();
    client().admin().indices().refresh(refreshRequest()).actionGet();
    logger.info("Running moreLikeThis");
    SearchResponse response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item("test", "type1", "1") }).minTermFreq(1).minDocFreq(1)).get();
    assertHitCount(response, 0L);
}

79. MoreLikeThisIT#testSimpleMoreLikeThis()

Project: elasticsearch
File: MoreLikeThisIT.java
public void testSimpleMoreLikeThis() throws Exception {
    logger.info("Creating index test");
    assertAcked(prepareCreate("test").addMapping("type1", jsonBuilder().startObject().startObject("type1").startObject("properties").startObject("text").field("type", "text").endObject().endObject().endObject().endObject()));
    logger.info("Running Cluster Health");
    assertThat(ensureGreen(), equalTo(ClusterHealthStatus.GREEN));
    logger.info("Indexing...");
    client().index(indexRequest("test").type("type1").id("1").source(jsonBuilder().startObject().field("text", "lucene").endObject())).actionGet();
    client().index(indexRequest("test").type("type1").id("2").source(jsonBuilder().startObject().field("text", "lucene release").endObject())).actionGet();
    client().admin().indices().refresh(refreshRequest()).actionGet();
    logger.info("Running moreLikeThis");
    SearchResponse response = client().prepareSearch().setQuery(new MoreLikeThisQueryBuilder(null, new Item[] { new Item("test", "type1", "1") }).minTermFreq(1).minDocFreq(1)).get();
    assertHitCount(response, 1L);
}

80. InnerHitsIT#testNestedDefinedAsObject()

Project: elasticsearch
File: InnerHitsIT.java
// Issue #9723
public void testNestedDefinedAsObject() throws Exception {
    assertAcked(prepareCreate("articles").addMapping("article", "comments", "type=nested", "title", "type=text"));
    List<IndexRequestBuilder> requests = new ArrayList<>();
    requests.add(client().prepareIndex("articles", "article", "1").setSource(jsonBuilder().startObject().field("title", "quick brown fox").startObject("comments").field("message", "fox eat quick").endObject().endObject()));
    indexRandom(true, requests);
    SearchResponse response = client().prepareSearch("articles").setQuery(nestedQuery("comments", matchQuery("comments.message", "fox"), ScoreMode.Avg).innerHit(new InnerHitBuilder())).get();
    assertNoFailures(response);
    assertHitCount(response, 1);
    assertThat(response.getHits().getAt(0).id(), equalTo("1"));
    assertThat(response.getHits().getAt(0).getInnerHits().get("comments").getTotalHits(), equalTo(1L));
    assertThat(response.getHits().getAt(0).getInnerHits().get("comments").getAt(0).id(), equalTo("1"));
    assertThat(response.getHits().getAt(0).getInnerHits().get("comments").getAt(0).getNestedIdentity().getField().string(), equalTo("comments"));
    assertThat(response.getHits().getAt(0).getInnerHits().get("comments").getAt(0).getNestedIdentity().getOffset(), equalTo(0));
    assertThat(response.getHits().getAt(0).getInnerHits().get("comments").getAt(0).getNestedIdentity().getChild(), nullValue());
}

81. HighlighterSearchIT#phraseBoostTestCaseForClauses()

Project: elasticsearch
File: HighlighterSearchIT.java
private <P extends AbstractQueryBuilder<P>> void phraseBoostTestCaseForClauses(String highlighterType, float boost, QueryBuilder terms, P phrase) {
    Matcher<String> highlightedMatcher = Matchers.either(containsString("<em>highlight words together</em>")).or(containsString("<em>highlight</em> <em>words</em> <em>together</em>"));
    SearchRequestBuilder search = client().prepareSearch("test").highlighter(new HighlightBuilder().field("field1", 100, 1).order("score").highlighterType(highlighterType).requireFieldMatch(true));
    // Try with a bool query
    phrase.boost(boost);
    SearchResponse response = search.setQuery(boolQuery().must(terms).should(phrase)).get();
    assertHighlight(response, 0, "field1", 0, 1, highlightedMatcher);
    phrase.boost(1);
    // Try with a boosting query
    response = search.setQuery(boostingQuery(phrase, terms).boost(boost).negativeBoost(1)).get();
    assertHighlight(response, 0, "field1", 0, 1, highlightedMatcher);
    // Try with a boosting query using a negative boost
    response = search.setQuery(boostingQuery(phrase, terms).boost(1).negativeBoost(1 / boost)).get();
    assertHighlight(response, 0, "field1", 0, 1, highlightedMatcher);
}

82. HighlighterSearchIT#testPostingsHighlighterQueryStringWithinFilteredQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterQueryStringWithinFilteredQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "The photography word will get highlighted").get();
    refresh();
    logger.info("--> highlighting and searching on field1");
    SearchSourceBuilder source = searchSource().query(boolQuery().must(queryStringQuery("field1:photo*")).mustNot(existsQuery("field_null"))).highlighter(highlight().field("field1"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field1", 0, 1, equalTo("The <em>photography</em> word will get highlighted"));
}

83. HighlighterSearchIT#testPostingsHighlighterPrefixQueryWithinBooleanQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterPrefixQueryWithinBooleanQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "The photography word will get highlighted").get();
    refresh();
    logger.info("--> highlighting and searching on field1");
    SearchSourceBuilder source = searchSource().query(boolQuery().must(prefixQuery("field1", "photo")).should(matchQuery("field1", "test").minimumShouldMatch("0"))).highlighter(highlight().field("field1"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field1", 0, 1, equalTo("The <em>photography</em> word will get highlighted"));
}

84. HighlighterSearchIT#testPostingsHighlighterMultiTermQueryMultipleLevels()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterMultiTermQueryMultipleLevels() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "The photography word will get highlighted").get();
    refresh();
    logger.info("--> highlighting and searching on field1");
    SearchSourceBuilder source = searchSource().query(boolQuery().should(boolQuery().mustNot(QueryBuilders.existsQuery("field1"))).should(matchQuery("field1", "test")).should(constantScoreQuery(queryStringQuery("field1:photo*")))).highlighter(highlight().field("field1"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field1", 0, 1, equalTo("The <em>photography</em> word will get highlighted"));
}

85. HighlighterSearchIT#testPostingsHighlighterRegexpQueryWithinConstantScoreQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterRegexpQueryWithinConstantScoreQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "The photography word will get highlighted").get();
    refresh();
    logger.info("--> highlighting and searching on field1");
    SearchSourceBuilder source = searchSource().query(constantScoreQuery(regexpQuery("field1", "pho[a-z]+"))).highlighter(highlight().field("field1"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field1", 0, 1, equalTo("The <em>photography</em> word will get highlighted"));
}

86. HighlighterSearchIT#testPostingsHighlighterQueryString()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterQueryString() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field2");
    SearchSourceBuilder source = searchSource().query(queryStringQuery("qui*").defaultField("field2")).highlighter(highlight().field("field2"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <em>quick</em> brown fox jumps over the lazy dog!"));
}

87. HighlighterSearchIT#testPostingsHighlighterTermRangeQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterTermRangeQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "aaab").get();
    refresh();
    logger.info("--> highlighting and searching on field2");
    SearchSourceBuilder source = searchSource().query(rangeQuery("field2").gte("aaaa").lt("zzzz")).highlighter(highlight().field("field2"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("<em>aaab</em>"));
}

88. HighlighterSearchIT#testPostingsHighlighterWildcardQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterWildcardQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field2");
    SearchSourceBuilder source = searchSource().query(wildcardQuery("field2", "qui*")).highlighter(highlight().field("field2"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <em>quick</em> brown fox jumps over the lazy dog!"));
    source = searchSource().query(wildcardQuery("field2", "qu*k")).highlighter(highlight().field("field2"));
    searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHitCount(searchResponse, 1L);
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <em>quick</em> brown fox jumps over the lazy dog!"));
}

89. HighlighterSearchIT#testPostingsHighlighterRegexpQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterRegexpQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field2");
    SearchSourceBuilder source = searchSource().query(regexpQuery("field2", "qu[a-l]+k")).highlighter(highlight().field("field2"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <em>quick</em> brown fox jumps over the lazy dog!"));
}

90. HighlighterSearchIT#testPostingsHighlighterFuzzyQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterFuzzyQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field2");
    SearchSourceBuilder source = searchSource().query(fuzzyQuery("field2", "quck")).highlighter(highlight().field("field2"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <em>quick</em> brown fox jumps over the lazy dog!"));
}

91. HighlighterSearchIT#testPostingsHighlighterPrefixQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterPrefixQuery() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field2");
    SearchSourceBuilder source = searchSource().query(prefixQuery("field2", "qui")).highlighter(highlight().field("field2"));
    SearchResponse searchResponse = client().prepareSearch("test").setSource(source).get();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <em>quick</em> brown fox jumps over the lazy dog!"));
}

92. HighlighterSearchIT#testPostingsHighlighterCommonTermsQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterCommonTermsQuery() throws IOException {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field1");
    SearchSourceBuilder source = searchSource().query(commonTermsQuery("field2", "quick brown").cutoffFrequency(100)).highlighter(highlight().field("field2").preTags("<x>").postTags("</x>"));
    SearchResponse searchResponse = client().search(searchRequest("test").source(source)).actionGet();
    assertHitCount(searchResponse, 1L);
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The <x>quick</x> <x>brown</x> fox jumps over the lazy dog!"));
}

93. HighlighterSearchIT#testPostingsHighlighterBoostingQuery()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterBoostingQuery() throws IOException {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()));
    ensureGreen();
    client().prepareIndex("test", "type1").setSource("field1", "this is a test", "field2", "The quick brown fox jumps over the lazy dog! Second sentence.").get();
    refresh();
    logger.info("--> highlighting and searching on field1");
    SearchSourceBuilder source = searchSource().query(boostingQuery(termQuery("field2", "brown"), termQuery("field2", "foobar")).negativeBoost(0.5f)).highlighter(highlight().field("field2").preTags("<x>").postTags("</x>"));
    SearchResponse searchResponse = client().search(searchRequest("test").source(source)).actionGet();
    assertHighlight(searchResponse, 0, "field2", 0, 1, equalTo("The quick <x>brown</x> fox jumps over the lazy dog!"));
}

94. HighlighterSearchIT#testPostingsHighlighterEscapeHtml()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterEscapeHtml() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", "title", "type=text," + randomStoreField() + "index_options=offsets"));
    ensureYellow();
    IndexRequestBuilder[] indexRequestBuilders = new IndexRequestBuilder[5];
    for (int i = 0; i < 5; i++) {
        indexRequestBuilders[i] = client().prepareIndex("test", "type1", Integer.toString(i)).setSource("title", "This is a html escaping highlighting test for *&? elasticsearch");
    }
    indexRandom(true, indexRequestBuilders);
    SearchResponse searchResponse = client().prepareSearch().setQuery(matchQuery("title", "test")).highlighter(new HighlightBuilder().field("title").encoder("html")).get();
    for (int i = 0; i < indexRequestBuilders.length; i++) {
        assertHighlight(searchResponse, i, "title", 0, 1, equalTo("This is a html escaping highlighting <em>test</em> for *&?"));
    }
}

95. HighlighterSearchIT#testPostingsHighlighterMultipleFields()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPostingsHighlighterMultipleFields() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", type1PostingsffsetsMapping()).get());
    ensureGreen();
    index("test", "type1", "1", "field1", "The <b>quick<b> brown fox. Second sentence.", "field2", "The <b>slow<b> brown fox. Second sentence.");
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(QueryBuilders.matchQuery("field1", "fox")).highlighter(new HighlightBuilder().field(new HighlightBuilder.Field("field1").preTags("<1>").postTags("</1>").requireFieldMatch(true))).get();
    assertHighlight(response, 0, "field1", 0, 1, equalTo("The <b>quick<b> brown <1>fox</1>."));
}

96. HighlighterSearchIT#testResetTwice()

Project: elasticsearch
File: HighlighterSearchIT.java
// Issue #3200
public void testResetTwice() throws Exception {
    assertAcked(prepareCreate("test").setSettings(Settings.builder().put(indexSettings()).put("analysis.analyzer.my_analyzer.type", "pattern").put("analysis.analyzer.my_analyzer.pattern", "\\s+").build()).addMapping("type", "text", "type=text,analyzer=my_analyzer"));
    ensureGreen();
    client().prepareIndex("test", "type", "1").setSource("text", "elasticsearch test").get();
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(QueryBuilders.matchQuery("text", "test").type(MatchQuery.Type.BOOLEAN)).highlighter(new HighlightBuilder().field("text")).execute().actionGet();
    // PatternAnalyzer will throw an exception if it is resetted twice
    assertHitCount(response, 1L);
}

97. HighlighterSearchIT#testNumericHighlighting()

Project: elasticsearch
File: HighlighterSearchIT.java
// Issue #3211
public void testNumericHighlighting() throws Exception {
    assertAcked(prepareCreate("test").addMapping("test", "text", "type=text", "byte", "type=byte", "short", "type=short", "int", "type=integer", "long", "type=long", "float", "type=float", "double", "type=double"));
    ensureGreen();
    client().prepareIndex("test", "test", "1").setSource("text", "elasticsearch test", "byte", 25, "short", 42, "int", 100, "long", -1, "float", 3.2f, "double", 42.42).get();
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(QueryBuilders.matchQuery("text", "test").type(MatchQuery.Type.BOOLEAN)).highlighter(new HighlightBuilder().field("text").field("byte").field("short").field("int").field("long").field("float").field("double")).get();
    // Highlighting of numeric fields is not supported, but it should not raise errors
    // (this behavior is consistent with version 0.20)
    assertHitCount(response, 1L);
}

98. HighlighterSearchIT#testMissingStoredField()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testMissingStoredField() throws Exception {
    assertAcked(prepareCreate("test").addMapping("type1", "highlight_field", "type=text,store=true"));
    ensureGreen();
    client().prepareIndex("test", "type1", "1").setSource(jsonBuilder().startObject().field("field", "highlight").endObject()).get();
    refresh();
    // This query used to fail when the field to highlight was absent
    SearchResponse response = client().prepareSearch("test").setQuery(QueryBuilders.matchQuery("field", "highlight").type(MatchQuery.Type.BOOLEAN)).highlighter(new HighlightBuilder().field(new HighlightBuilder.Field("highlight_field").fragmentSize(-1).numOfFragments(1).fragmenter("simple"))).get();
    assertThat(response.getHits().hits()[0].highlightFields().isEmpty(), equalTo(true));
}

99. HighlighterSearchIT#testFastVectorHighlighterMultipleFields()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testFastVectorHighlighterMultipleFields() {
    assertAcked(prepareCreate("test").addMapping("type1", "field1", "type=text,term_vector=with_positions_offsets", "field2", "type=text,term_vector=with_positions_offsets"));
    ensureGreen();
    index("test", "type1", "1", "field1", "The <b>quick<b> brown fox", "field2", "The <b>slow<b> brown fox");
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(QueryBuilders.matchQuery("field1", "fox")).highlighter(new HighlightBuilder().field(new HighlightBuilder.Field("field1").preTags("<1>").postTags("</1>").requireFieldMatch(true)).field(new HighlightBuilder.Field("field2").preTags("<2>").postTags("</2>").requireFieldMatch(false))).get();
    assertHighlight(response, 0, "field1", 0, 1, equalTo("The <b>quick<b> brown <1>fox</1>"));
    assertHighlight(response, 0, "field2", 0, 1, equalTo("The <b>slow<b> brown <2>fox</2>"));
}

100. HighlighterSearchIT#testPlainHighlighterMultipleFields()

Project: elasticsearch
File: HighlighterSearchIT.java
public void testPlainHighlighterMultipleFields() {
    createIndex("test");
    ensureGreen();
    index("test", "type1", "1", "field1", "The <b>quick<b> brown fox", "field2", "The <b>slow<b> brown fox");
    refresh();
    SearchResponse response = client().prepareSearch("test").setQuery(QueryBuilders.matchQuery("field1", "fox")).highlighter(new HighlightBuilder().field(new HighlightBuilder.Field("field1").preTags("<1>").postTags("</1>").requireFieldMatch(true)).field(new HighlightBuilder.Field("field2").preTags("<2>").postTags("</2>").requireFieldMatch(false))).get();
    assertHighlight(response, 0, "field1", 0, 1, equalTo("The <b>quick<b> brown <1>fox</1>"));
    assertHighlight(response, 0, "field2", 0, 1, equalTo("The <b>slow<b> brown <2>fox</2>"));
}