org.elasticsearch.search.aggregations

Class AggregatorTestCase

java.lang.Object

org.junit.Assert

org.apache.lucene.util.LuceneTestCase

org.elasticsearch.test.ESTestCase

org.elasticsearch.search.aggregations.AggregatorTestCase

public abstract class AggregatorTestCase
extends ESTestCase

Base class for testing Aggregator implementations. Provides helpers for constructing and searching an Aggregator implementation based on a provided AggregationBuilder instance.

Nested Class Summary

Nested classes/interfaces inherited from class org.elasticsearch.test.ESTestCase

ESTestCase.GeohashGenerator, ESTestCase.TestAnalysis

Nested classes/interfaces inherited from class org.apache.lucene.util.LuceneTestCase

org.apache.lucene.util.LuceneTestCase.AwaitsFix, org.apache.lucene.util.LuceneTestCase.BadApple, org.apache.lucene.util.LuceneTestCase.Monster, org.apache.lucene.util.LuceneTestCase.Nightly, org.apache.lucene.util.LuceneTestCase.Slow, org.apache.lucene.util.LuceneTestCase.SuppressCodecs, org.apache.lucene.util.LuceneTestCase.SuppressFileSystems, org.apache.lucene.util.LuceneTestCase.SuppressFsync, org.apache.lucene.util.LuceneTestCase.SuppressReproduceLine, org.apache.lucene.util.LuceneTestCase.SuppressSysoutChecks, org.apache.lucene.util.LuceneTestCase.SuppressTempFileChecks, org.apache.lucene.util.LuceneTestCase.ThrowingRunnable, org.apache.lucene.util.LuceneTestCase.Weekly

Field Summary

Fields inherited from class org.elasticsearch.test.ESTestCase

checkIndexFailed, deprecationLogger, failureAndSuccessEvents, logger

Fields inherited from class org.apache.lucene.util.LuceneTestCase

assertsAreEnabled, classRules, DEFAULT_LINE_DOCS_FILE, INFOSTREAM, JENKINS_LARGE_LINE_DOCS_FILE, LEAVE_TEMPORARY, MAYBE_CACHE_POLICY, RANDOM_MULTIPLIER, ruleChain, suiteFailureMarker, SYSPROP_AWAITSFIX, SYSPROP_BADAPPLES, SYSPROP_FAILFAST, SYSPROP_MAXFAILURES, SYSPROP_MONSTER, SYSPROP_NIGHTLY, SYSPROP_SLOW, SYSPROP_WEEKLY, TEST_ASSERTS_ENABLED, TEST_AWAITSFIX, TEST_CODEC, TEST_DIRECTORY, TEST_DOCVALUESFORMAT, TEST_LINE_DOCS_FILE, TEST_NIGHTLY, TEST_POSTINGSFORMAT, TEST_SLOW, TEST_THROTTLING, TEST_WEEKLY, VERBOSE

Constructor Summary

Constructors

Constructor and Description
AggregatorTestCase()

Method Summary

Modifier and Type	Method and Description
protected <A extends org.elasticsearch.search.aggregations.Aggregator> A	createAggregator(org.elasticsearch.search.aggregations.AggregationBuilder aggregationBuilder, org.apache.lucene.search.IndexSearcher indexSearcher, org.elasticsearch.index.mapper.MappedFieldType... fieldTypes)
protected org.elasticsearch.search.aggregations.AggregatorFactory<?>	createAggregatorFactory(org.elasticsearch.search.aggregations.AggregationBuilder aggregationBuilder, org.apache.lucene.search.IndexSearcher indexSearcher, org.elasticsearch.index.mapper.MappedFieldType... fieldTypes) Create a factory for the given aggregation builder.
protected org.elasticsearch.index.IndexSettings	createIndexSettings()
protected org.elasticsearch.search.internal.SearchContext	createSearchContext(org.apache.lucene.search.IndexSearcher indexSearcher, org.elasticsearch.index.IndexSettings indexSettings)
protected org.elasticsearch.index.mapper.MapperService	mapperServiceMock() sub-tests that need a more complex mock can overwrite this
protected static org.apache.lucene.index.IndexReader	maybeWrapReaderEs(org.apache.lucene.index.DirectoryReader reader) Added to randomly run with more assertions on the index reader level, like LuceneTestCase.wrapReader(IndexReader), which can't be used because it also wraps in the IndexReader with other implementations that we can't handle.
protected static org.apache.lucene.search.IndexSearcher	newIndexSearcher(org.apache.lucene.index.IndexReader indexReader) Added to randomly run with more assertions on the index searcher level, like LuceneTestCase.newSearcher(IndexReader), which can't be used because it also wraps in the IndexSearcher's IndexReader with other implementations that we can't handle.
protected org.elasticsearch.index.query.QueryShardContext	queryShardContextMock(org.elasticsearch.index.mapper.MapperService mapperService, org.elasticsearch.index.mapper.MappedFieldType[] fieldTypes, org.elasticsearch.indices.breaker.CircuitBreakerService circuitBreakerService) sub-tests that need a more complex mock can overwrite this
protected <A extends org.elasticsearch.search.aggregations.InternalAggregation,C extends org.elasticsearch.search.aggregations.Aggregator> A	search(org.apache.lucene.search.IndexSearcher searcher, org.apache.lucene.search.Query query, org.elasticsearch.search.aggregations.AggregationBuilder builder, org.elasticsearch.index.mapper.MappedFieldType... fieldTypes)
protected <A extends org.elasticsearch.search.aggregations.InternalAggregation,C extends org.elasticsearch.search.aggregations.Aggregator> A	searchAndReduce(org.apache.lucene.search.IndexSearcher searcher, org.apache.lucene.search.Query query, org.elasticsearch.search.aggregations.AggregationBuilder builder, org.elasticsearch.index.mapper.MappedFieldType... fieldTypes) Divides the provided IndexSearcher in sub-searcher, one for each segment, builds an aggregator for each sub-searcher filtered by the provided Query and returns the reduced InternalAggregation.
protected static org.apache.lucene.index.DirectoryReader	wrap(org.apache.lucene.index.DirectoryReader directoryReader)

Methods inherited from class org.elasticsearch.test.ESTestCase

after, afterIfFailed, afterIfSuccessful, assertAllIndicesRemovedAndDeletionCompleted, assertArrayEquals, assertBusy, assertBusy, assertEquals, assertEqualsWithErrorMessageFromXContent, assertPathHasBeenCleared, assertSettingDeprecationsAndWarnings, assertWarnings, awaitBusy, awaitBusy, before, between, buildNewFakeTransportAddress, checkStaticState, copyWriteable, createParser, createParser, createParser, createParser, createParser, createTestAnalysis, createTestAnalysis, createTestAnalysis, enableWarningsCheck, ensureAllSearchContextsReleased, ensureCheckIndexPassed, frequently, generateRandomStringArray, generateRandomStringArray, getBwcIndicesPath, getDataPath, getSuiteFailureMarker, iterations, maybeSet, mockScript, newNodeEnvironment, newNodeEnvironment, newTestIndicesModule, newTestScriptModule, randomAlphaOfLength, randomAlphaOfLengthBetween, randomBoolean, randomByte, randomDateTimeZone, randomDouble, randomDoubleBetween, randomFloat, randomFrom, randomFrom, randomFrom, randomFrom, randomFrom, randomGeohash, randomInt, randomInt, randomIntBetween, randomLong, randomNonNegativeLong, randomPositiveTimeValue, randomRealisticUnicodeOfCodepointLength, randomRealisticUnicodeOfCodepointLengthBetween, randomRealisticUnicodeOfLength, randomRealisticUnicodeOfLengthBetween, randomShort, randomSubsetOf, randomSubsetOf, randomSubsetOf, randomTimeValue, randomTimeValue, randomTimeValue, randomUnicodeOfCodepointLength, randomUnicodeOfCodepointLengthBetween, randomUnicodeOfLength, randomUnicodeOfLengthBetween, randomUnique, randomValueOtherThan, randomValueOtherThanMany, resetCheckIndexStatus, resetPortCounter, restoreContentType, restoreFileSystem, scaledRandomIntBetween, setContentType, setFileSystem, settings, shuffleMap, shuffleXContent, shuffleXContent, spinForAtLeastNMilliseconds, spinForAtLeastOneMillisecond, terminate, terminate, tmpPaths, toShuffledXContent, xContentRegistry

Methods inherited from class org.apache.lucene.util.LuceneTestCase

addVirusChecker, assertDeletedDocsEquals, assertDocsAndPositionsEnumEquals, assertDocsEnumEquals, assertDocsSkippingEquals, assertDocValuesEquals, assertDocValuesEquals, assertFieldInfosEquals, assertFieldsEquals, assertFieldStatisticsEquals, assertNormsEquals, assertPointsEquals, assertPositionsSkippingEquals, assertReaderEquals, assertReaderStatisticsEquals, assertStoredFieldEquals, assertStoredFieldsEquals, assertTermsEnumEquals, assertTermsEquals, assertTermsStatisticsEquals, assertTermStatsEquals, assertTermVectorsEquals, asSet, assumeFalse, assumeNoException, assumeTrue, assumeWorkingMMapOnWindows, atLeast, atLeast, closeAfterSuite, closeAfterTest, collate, createTempDir, createTempDir, createTempFile, createTempFile, dumpArray, dumpIterator, expectThrows, expectThrows, getBaseTempDirForTestClass, getDataInputStream, getOnlyLeafReader, getTestClass, getTestName, hasWorkingMMapOnWindows, isTestThread, localeForLanguageTag, maybeChangeLiveIndexWriterConfig, maybeWrapReader, newAlcoholicMergePolicy, newAlcoholicMergePolicy, newDirectory, newDirectory, newDirectory, newDirectory, newDirectory, newField, newField, newFSDirectory, newFSDirectory, newIndexWriterConfig, newIndexWriterConfig, newIndexWriterConfig, newIOContext, newIOContext, newLogMergePolicy, newLogMergePolicy, newLogMergePolicy, newLogMergePolicy, newLogMergePolicy, newMaybeVirusCheckingDirectory, newMaybeVirusCheckingFSDirectory, newMergePolicy, newMergePolicy, newMockDirectory, newMockDirectory, newMockDirectory, newMockFSDirectory, newMockFSDirectory, newSearcher, newSearcher, newSearcher, newStringField, newStringField, newStringField, newStringField, newTextField, newTextField, newTieredMergePolicy, newTieredMergePolicy, overrideDefaultQueryCache, overrideTestDefaultQueryCache, random, randomLocale, randomTimeZone, rarely, rarely, replaceMaxFailureRule, resetDefaultQueryCache, restoreCPUCoreCount, restoreIndexWriterMaxDocs, restoreSpins, runWithRestrictedPermissions, setIndexWriterMaxDocs, setUp, setupCPUCoreCount, setupSpins, slowFileExists, tearDown, usually, usually, wrapReader

Methods inherited from class org.junit.Assert

assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertArrayEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertEquals, assertFalse, assertFalse, assertNotEquals, assertNotEquals, assertNotEquals, assertNotEquals, assertNotEquals, assertNotEquals, assertNotEquals, assertNotEquals, assertNotNull, assertNotNull, assertNotSame, assertNotSame, assertNull, assertNull, assertSame, assertSame, assertThat, assertThat, assertTrue, assertTrue, fail, fail

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

AggregatorTestCase

public AggregatorTestCase()

Method Detail

createAggregatorFactory

protected org.elasticsearch.search.aggregations.AggregatorFactory<?> createAggregatorFactory(org.elasticsearch.search.aggregations.AggregationBuilder aggregationBuilder,
                                                                                             org.apache.lucene.search.IndexSearcher indexSearcher,
                                                                                             org.elasticsearch.index.mapper.MappedFieldType... fieldTypes)
                                                                                      throws java.io.IOException

Create a factory for the given aggregation builder.

Throws:

java.io.IOException

createAggregator

protected <A extends org.elasticsearch.search.aggregations.Aggregator> A createAggregator(org.elasticsearch.search.aggregations.AggregationBuilder aggregationBuilder,
                                                                                          org.apache.lucene.search.IndexSearcher indexSearcher,
                                                                                          org.elasticsearch.index.mapper.MappedFieldType... fieldTypes)
                                                                                   throws java.io.IOException

Throws:

java.io.IOException

createSearchContext

protected org.elasticsearch.search.internal.SearchContext createSearchContext(org.apache.lucene.search.IndexSearcher indexSearcher,
                                                                              org.elasticsearch.index.IndexSettings indexSettings)

createIndexSettings

protected org.elasticsearch.index.IndexSettings createIndexSettings()

mapperServiceMock

protected org.elasticsearch.index.mapper.MapperService mapperServiceMock()

sub-tests that need a more complex mock can overwrite this

queryShardContextMock

protected org.elasticsearch.index.query.QueryShardContext queryShardContextMock(org.elasticsearch.index.mapper.MapperService mapperService,
                                                                                org.elasticsearch.index.mapper.MappedFieldType[] fieldTypes,
                                                                                org.elasticsearch.indices.breaker.CircuitBreakerService circuitBreakerService)

sub-tests that need a more complex mock can overwrite this

search

protected <A extends org.elasticsearch.search.aggregations.InternalAggregation,C extends org.elasticsearch.search.aggregations.Aggregator> A search(org.apache.lucene.search.IndexSearcher searcher,
                                                                                                                                                    org.apache.lucene.search.Query query,
                                                                                                                                                    org.elasticsearch.search.aggregations.AggregationBuilder builder,
                                                                                                                                                    org.elasticsearch.index.mapper.MappedFieldType... fieldTypes)
                                                                                                                                             throws java.io.IOException

Throws:

java.io.IOException

searchAndReduce

protected <A extends org.elasticsearch.search.aggregations.InternalAggregation,C extends org.elasticsearch.search.aggregations.Aggregator> A searchAndReduce(org.apache.lucene.search.IndexSearcher searcher,
                                                                                                                                                             org.apache.lucene.search.Query query,
                                                                                                                                                             org.elasticsearch.search.aggregations.AggregationBuilder builder,
                                                                                                                                                             org.elasticsearch.index.mapper.MappedFieldType... fieldTypes)
                                                                                                                                                      throws java.io.IOException

Divides the provided IndexSearcher in sub-searcher, one for each segment, builds an aggregator for each sub-searcher filtered by the provided Query and returns the reduced InternalAggregation.

Throws:

java.io.IOException

wrap

protected static org.apache.lucene.index.DirectoryReader wrap(org.apache.lucene.index.DirectoryReader directoryReader)
                                                       throws java.io.IOException

Throws:

java.io.IOException

newIndexSearcher

protected static org.apache.lucene.search.IndexSearcher newIndexSearcher(org.apache.lucene.index.IndexReader indexReader)

Added to randomly run with more assertions on the index searcher level, like LuceneTestCase.newSearcher(IndexReader), which can't be used because it also wraps in the IndexSearcher's IndexReader with other implementations that we can't handle. (e.g. ParallelCompositeReader)

maybeWrapReaderEs

protected static org.apache.lucene.index.IndexReader maybeWrapReaderEs(org.apache.lucene.index.DirectoryReader reader)
                                                                throws java.io.IOException

Added to randomly run with more assertions on the index reader level, like LuceneTestCase.wrapReader(IndexReader), which can't be used because it also wraps in the IndexReader with other implementations that we can't handle. (e.g. ParallelCompositeReader)

Throws:

java.io.IOException