Examples with TestSearchContext org.elasticsearch.test.TestSearchContext used on opensource projects

Example 6 with TestSearchContext

use of org.elasticsearch.test.TestSearchContext in project elasticsearch by elastic.

the class QueryPhaseTests method testMinScoreDisablesCountOptimization.

public void testMinScoreDisablesCountOptimization() throws Exception {
    TestSearchContext context = new TestSearchContext(null);
    context.parsedQuery(new ParsedQuery(new MatchAllDocsQuery()));
    context.setSize(0);
    context.setTask(new SearchTask(123L, "", "", "", null));
    final AtomicBoolean collected = new AtomicBoolean();
    IndexSearcher contextSearcher = new IndexSearcher(new MultiReader()) {

        protected void search(List<LeafReaderContext> leaves, Weight weight, Collector collector) throws IOException {
            collected.set(true);
            super.search(leaves, weight, collector);
        }
    };
    QueryPhase.execute(context, contextSearcher);
    assertEquals(0, context.queryResult().topDocs().totalHits);
    assertFalse(collected.get());
    context.minimumScore(1);
    QueryPhase.execute(context, contextSearcher);
    assertEquals(0, context.queryResult().topDocs().totalHits);
    assertTrue(collected.get());
}

Example 7 with TestSearchContext

use of org.elasticsearch.test.TestSearchContext in project elasticsearch by elastic.

the class SearchOperationListenerTests method testListenersAreExecuted.

// this test also tests if calls are correct if one or more listeners throw exceptions
public void testListenersAreExecuted() {
    AtomicInteger preQuery = new AtomicInteger();
    AtomicInteger failedQuery = new AtomicInteger();
    AtomicInteger onQuery = new AtomicInteger();
    AtomicInteger onFetch = new AtomicInteger();
    AtomicInteger preFetch = new AtomicInteger();
    AtomicInteger failedFetch = new AtomicInteger();
    AtomicInteger newContext = new AtomicInteger();
    AtomicInteger freeContext = new AtomicInteger();
    AtomicInteger newScrollContext = new AtomicInteger();
    AtomicInteger freeScrollContext = new AtomicInteger();
    AtomicInteger timeInNanos = new AtomicInteger(randomIntBetween(0, 10));
    SearchOperationListener listener = new SearchOperationListener() {

        @Override
        public void onPreQueryPhase(SearchContext searchContext) {
            assertNotNull(searchContext);
            preQuery.incrementAndGet();
        }

        @Override
        public void onFailedQueryPhase(SearchContext searchContext) {
            assertNotNull(searchContext);
            failedQuery.incrementAndGet();
        }

        @Override
        public void onQueryPhase(SearchContext searchContext, long tookInNanos) {
            assertEquals(timeInNanos.get(), tookInNanos);
            assertNotNull(searchContext);
            onQuery.incrementAndGet();
        }

        @Override
        public void onPreFetchPhase(SearchContext searchContext) {
            assertNotNull(searchContext);
            preFetch.incrementAndGet();
        }

        @Override
        public void onFailedFetchPhase(SearchContext searchContext) {
            assertNotNull(searchContext);
            failedFetch.incrementAndGet();
        }

        @Override
        public void onFetchPhase(SearchContext searchContext, long tookInNanos) {
            assertEquals(timeInNanos.get(), tookInNanos);
            onFetch.incrementAndGet();
        }

        @Override
        public void onNewContext(SearchContext context) {
            assertNotNull(context);
            newContext.incrementAndGet();
        }

        @Override
        public void onFreeContext(SearchContext context) {
            assertNotNull(context);
            freeContext.incrementAndGet();
        }

        @Override
        public void onNewScrollContext(SearchContext context) {
            assertNotNull(context);
            newScrollContext.incrementAndGet();
        }

        @Override
        public void onFreeScrollContext(SearchContext context) {
            assertNotNull(context);
            freeScrollContext.incrementAndGet();
        }
    };
    SearchOperationListener throwingListener = (SearchOperationListener) Proxy.newProxyInstance(SearchOperationListener.class.getClassLoader(), new Class[] { SearchOperationListener.class }, (a, b, c) -> {
        throw new RuntimeException();
    });
    final List<SearchOperationListener> indexingOperationListeners = new ArrayList<>(Arrays.asList(listener, listener));
    if (randomBoolean()) {
        indexingOperationListeners.add(throwingListener);
        if (randomBoolean()) {
            indexingOperationListeners.add(throwingListener);
        }
    }
    Collections.shuffle(indexingOperationListeners, random());
    SearchOperationListener.CompositeListener compositeListener = new SearchOperationListener.CompositeListener(indexingOperationListeners, logger);
    SearchContext ctx = new TestSearchContext(null);
    compositeListener.onQueryPhase(ctx, timeInNanos.get());
    assertEquals(0, preFetch.get());
    assertEquals(0, preQuery.get());
    assertEquals(0, failedFetch.get());
    assertEquals(0, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(0, onFetch.get());
    assertEquals(0, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onFetchPhase(ctx, timeInNanos.get());
    assertEquals(0, preFetch.get());
    assertEquals(0, preQuery.get());
    assertEquals(0, failedFetch.get());
    assertEquals(0, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(0, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onPreQueryPhase(ctx);
    assertEquals(0, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(0, failedFetch.get());
    assertEquals(0, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(0, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onPreFetchPhase(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(0, failedFetch.get());
    assertEquals(0, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(0, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onFailedFetchPhase(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(2, failedFetch.get());
    assertEquals(0, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(0, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onFailedQueryPhase(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(2, failedFetch.get());
    assertEquals(2, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(0, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onNewContext(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(2, failedFetch.get());
    assertEquals(2, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(2, newContext.get());
    assertEquals(0, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onNewScrollContext(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(2, failedFetch.get());
    assertEquals(2, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(2, newContext.get());
    assertEquals(2, newScrollContext.get());
    assertEquals(0, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onFreeContext(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(2, failedFetch.get());
    assertEquals(2, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(2, newContext.get());
    assertEquals(2, newScrollContext.get());
    assertEquals(2, freeContext.get());
    assertEquals(0, freeScrollContext.get());
    compositeListener.onFreeScrollContext(ctx);
    assertEquals(2, preFetch.get());
    assertEquals(2, preQuery.get());
    assertEquals(2, failedFetch.get());
    assertEquals(2, failedQuery.get());
    assertEquals(2, onQuery.get());
    assertEquals(2, onFetch.get());
    assertEquals(2, newContext.get());
    assertEquals(2, newScrollContext.get());
    assertEquals(2, freeContext.get());
    assertEquals(2, freeScrollContext.get());
}

Example 8 with TestSearchContext

use of org.elasticsearch.test.TestSearchContext in project elasticsearch by elastic.

the class ExceptionSerializationTests method testSearchParseException.

public void testSearchParseException() throws IOException {
    SearchContext ctx = new TestSearchContext(null);
    SearchParseException ex = serialize(new SearchParseException(ctx, "foo", new XContentLocation(66, 666)));
    assertEquals("foo", ex.getMessage());
    assertEquals(66, ex.getLineNumber());
    assertEquals(666, ex.getColumnNumber());
    assertEquals(ctx.shardTarget(), ex.shard());
}

Example 9 with TestSearchContext

use of org.elasticsearch.test.TestSearchContext in project elasticsearch by elastic.

the class ElasticsearchExceptionTests method randomExceptions.

public static Tuple<Throwable, ElasticsearchException> randomExceptions() {
    Throwable actual;
    ElasticsearchException expected;
    int type = randomIntBetween(0, 5);
    switch(type) {
        case 0:
            actual = new ClusterBlockException(singleton(DiscoverySettings.NO_MASTER_BLOCK_WRITES));
            expected = new ElasticsearchException("Elasticsearch exception [type=cluster_block_exception, " + "reason=blocked by: [SERVICE_UNAVAILABLE/2/no master];]");
            break;
        case 1:
            actual = new CircuitBreakingException("Data too large", 123, 456);
            expected = new ElasticsearchException("Elasticsearch exception [type=circuit_breaking_exception, reason=Data too large]");
            break;
        case 2:
            actual = new SearchParseException(new TestSearchContext(null), "Parse failure", new XContentLocation(12, 98));
            expected = new ElasticsearchException("Elasticsearch exception [type=search_parse_exception, reason=Parse failure]");
            break;
        case 3:
            actual = new IllegalArgumentException("Closed resource", new RuntimeException("Resource"));
            expected = new ElasticsearchException("Elasticsearch exception [type=illegal_argument_exception, reason=Closed resource]", new ElasticsearchException("Elasticsearch exception [type=runtime_exception, reason=Resource]"));
            break;
        case 4:
            actual = new SearchPhaseExecutionException("search", "all shards failed", new ShardSearchFailure[] { new ShardSearchFailure(new ParsingException(1, 2, "foobar", null), new SearchShardTarget("node_1", new Index("foo", "_na_"), 1)) });
            expected = new ElasticsearchException("Elasticsearch exception [type=search_phase_execution_exception, " + "reason=all shards failed]");
            expected.addMetadata("es.phase", "search");
            break;
        case 5:
            actual = new ElasticsearchException("Parsing failed", new ParsingException(9, 42, "Wrong state", new NullPointerException("Unexpected null value")));
            ElasticsearchException expectedCause = new ElasticsearchException("Elasticsearch exception [type=parsing_exception, " + "reason=Wrong state]", new ElasticsearchException("Elasticsearch exception [type=null_pointer_exception, " + "reason=Unexpected null value]"));
            expected = new ElasticsearchException("Elasticsearch exception [type=exception, reason=Parsing failed]", expectedCause);
            break;
        default:
            throw new UnsupportedOperationException("No randomized exceptions generated for type [" + type + "]");
    }
    if (actual instanceof ElasticsearchException) {
        ElasticsearchException actualException = (ElasticsearchException) actual;
        if (randomBoolean()) {
            int nbHeaders = randomIntBetween(1, 5);
            Map<String, List<String>> randomHeaders = new HashMap<>(nbHeaders);
            for (int i = 0; i < nbHeaders; i++) {
                List<String> values = new ArrayList<>();
                int nbValues = randomIntBetween(1, 3);
                for (int j = 0; j < nbValues; j++) {
                    values.add(frequently() ? randomAsciiOfLength(5) : "");
                }
                randomHeaders.put("header_" + i, values);
            }
            for (Map.Entry<String, List<String>> entry : randomHeaders.entrySet()) {
                actualException.addHeader(entry.getKey(), entry.getValue());
                expected.addHeader(entry.getKey(), entry.getValue());
            }
            if (rarely()) {
                // Empty or null headers are not printed out by the toXContent method
                actualException.addHeader("ignored", randomBoolean() ? emptyList() : null);
            }
        }
        if (randomBoolean()) {
            int nbMetadata = randomIntBetween(1, 5);
            Map<String, List<String>> randomMetadata = new HashMap<>(nbMetadata);
            for (int i = 0; i < nbMetadata; i++) {
                List<String> values = new ArrayList<>();
                int nbValues = randomIntBetween(1, 3);
                for (int j = 0; j < nbValues; j++) {
                    values.add(frequently() ? randomAsciiOfLength(5) : "");
                }
                randomMetadata.put("es.metadata_" + i, values);
            }
            for (Map.Entry<String, List<String>> entry : randomMetadata.entrySet()) {
                actualException.addMetadata(entry.getKey(), entry.getValue());
                expected.addMetadata(entry.getKey(), entry.getValue());
            }
            if (rarely()) {
                // Empty or null metadata are not printed out by the toXContent method
                actualException.addMetadata("es.ignored", randomBoolean() ? emptyList() : null);
            }
        }
        if (randomBoolean()) {
            int nbResources = randomIntBetween(1, 5);
            for (int i = 0; i < nbResources; i++) {
                String resourceType = "type_" + i;
                String[] resourceIds = new String[randomIntBetween(1, 3)];
                for (int j = 0; j < resourceIds.length; j++) {
                    resourceIds[j] = frequently() ? randomAsciiOfLength(5) : "";
                }
                actualException.setResources(resourceType, resourceIds);
                expected.setResources(resourceType, resourceIds);
            }
        }
    }
    return new Tuple<>(actual, expected);
}