HSEARCH-5133 Test all types with metric aggregations

integrationtest/backend/tck/src/main/java/org/hibernate/search/integrationtest/backend/tck/search/aggregation/MetricFieldAggregationsIT.java

@@ -0,0 +1,88 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.search.integrationtest.backend.tck.search.aggregation;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+import java.util.ArrayList;
+import java.util.LinkedHashSet;
+import java.util.List;
+import java.util.Set;
+import java.util.function.Function;
+
+import org.hibernate.search.engine.backend.document.model.dsl.IndexSchemaElement;
+import org.hibernate.search.engine.backend.types.Aggregable;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.model.singlefield.SingleFieldIndexBinding;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.operations.MetricAggregationsTestCase;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.FieldTypeDescriptor;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.StandardFieldTypeDescriptor;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.util.extension.SearchSetupHelper;
+import org.hibernate.search.util.impl.integrationtest.mapper.stub.BulkIndexer;
+import org.hibernate.search.util.impl.integrationtest.mapper.stub.SimpleMappedIndex;
+import org.hibernate.search.util.impl.integrationtest.mapper.stub.StubMappingScope;
+
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.extension.RegisterExtension;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.Arguments;
+import org.junit.jupiter.params.provider.MethodSource;
+
+public class MetricFieldAggregationsIT {
+
+	private static final Set<StandardFieldTypeDescriptor<?>> supportedFieldTypes = new LinkedHashSet<>();
+	private static final List<MetricAggregationsTestCase<?>> testCases = new ArrayList<>();
+	private static final List<Arguments> parameters = new ArrayList<>();
+
+	static {
+		for ( StandardFieldTypeDescriptor<?> typeDescriptor : FieldTypeDescriptor.getAllStandard() ) {
+			MetricAggregationsTestCase<?> scenario = new MetricAggregationsTestCase<>( typeDescriptor );
+			if ( !scenario.supported() ) {
+				continue;
+			}
+			testCases.add( scenario );
+			supportedFieldTypes.add( typeDescriptor );
+			parameters.add( Arguments.of( scenario ) );
+		}
+	}
+
+	public static List<? extends Arguments> params() {
+		return parameters;
+	}
+
+	@RegisterExtension
+	public static final SearchSetupHelper setupHelper = SearchSetupHelper.create();
+
+	private static final Function<IndexSchemaElement, SingleFieldIndexBinding> bindingFactory =
+			root -> SingleFieldIndexBinding.create( root, supportedFieldTypes, c -> c.aggregable( Aggregable.YES ) );
+	private static final SimpleMappedIndex<SingleFieldIndexBinding> mainIndex =
+			SimpleMappedIndex.of( bindingFactory ).name( "main" );
+
+	@BeforeAll
+	static void setup() {
+		int expectedDocuments = 0;
+
+		setupHelper.start().withIndexes( mainIndex ).setup();
+		BulkIndexer indexer = mainIndex.bulkIndexer();
+		for ( MetricAggregationsTestCase<?> scenario : testCases ) {
+			expectedDocuments += scenario.contribute( indexer, mainIndex.binding() );
+		}
+		indexer.join();
+
+		long createdDocuments = mainIndex.createScope().query().where( f -> f.matchAll() )
+				.totalHitCountThreshold( expectedDocuments )
+				.toQuery().fetch( 0 ).total().hitCountLowerBound();
+		assertThat( createdDocuments ).isEqualTo( expectedDocuments );
+	}
+
+	@ParameterizedTest(name = "{0}")
+	@MethodSource("params")
+	public void test(MetricAggregationsTestCase<?> testCase) {
+		StubMappingScope scope = mainIndex.createScope();
+		MetricAggregationsTestCase.Result<?> result = testCase.testMetricsAggregation( scope, mainIndex.binding() );
+		if ( result.expectedSum != null ) {
+			assertThat( result.sumResult ).isEqualTo( result.expectedSum );
+		}
+	}
+}

integrationtest/backend/tck/src/main/java/org/hibernate/search/integrationtest/backend/tck/testsupport/operations/MetricAggregationsTestCase.java

@@ -0,0 +1,122 @@
+/*
+ * SPDX-License-Identifier: Apache-2.0
+ * Copyright Red Hat Inc. and Hibernate Authors
+ */
+package org.hibernate.search.integrationtest.backend.tck.testsupport.operations;
+
+import static org.hibernate.search.util.impl.integrationtest.mapper.stub.StubMapperUtils.documentProvider;
+
+import java.util.List;
+import java.util.Locale;
+import java.util.StringJoiner;
+
+import org.hibernate.search.engine.backend.common.DocumentReference;
+import org.hibernate.search.engine.search.aggregation.AggregationKey;
+import org.hibernate.search.engine.search.predicate.dsl.SearchPredicateFactory;
+import org.hibernate.search.engine.search.query.SearchQuery;
+import org.hibernate.search.engine.search.query.SearchResult;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.model.singlefield.SingleFieldIndexBinding;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.FieldTypeDescriptor;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.types.values.MetricAggregationsValues;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.util.IndexFieldLocation;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.util.IndexFieldValueCardinality;
+import org.hibernate.search.integrationtest.backend.tck.testsupport.util.TestedFieldStructure;
+import org.hibernate.search.util.impl.integrationtest.mapper.stub.BulkIndexer;
+import org.hibernate.search.util.impl.integrationtest.mapper.stub.StubMappingScope;
+
+/**
+ * Denotes a metric aggregations test case for a particular {@link FieldTypeDescriptor}.
+ */
+public class MetricAggregationsTestCase<F> {
+
+	private final FieldTypeDescriptor<F, ?> typeDescriptor;
+	private final boolean supported;
+	private final MetricAggregationsValues<F> metricAggregationsValues;
+
+	public MetricAggregationsTestCase(FieldTypeDescriptor<F, ?> typeDescriptor) {
+		this.typeDescriptor = typeDescriptor;
+		metricAggregationsValues = typeDescriptor.metricAggregationsValues();
+		this.supported = metricAggregationsValues != null;
+	}
+
+	public boolean supported() {
+		return supported;
+	}
+
+	public int contribute(BulkIndexer indexer, SingleFieldIndexBinding binding) {
+		int i = 0;
+		for ( F value : metricAggregationsValues.values() ) {
+			String uniqueName = typeDescriptor.getUniqueName();
+			String keyA = String.format( Locale.ROOT, "%03d_ROOT_%s", ++i, uniqueName );
+			String keyB = String.format( Locale.ROOT, "%03d_NEST_%s", i, uniqueName );
+			String keyC = String.format( Locale.ROOT, "%03d_FLAT_%s", i, uniqueName );
+			indexer.add( documentProvider( keyA, document -> binding.initSingleValued( typeDescriptor,
+					IndexFieldLocation.ROOT, document, value ) ) );
+			indexer.add( documentProvider( keyB, document -> binding.initSingleValued(
+					typeDescriptor, IndexFieldLocation.IN_NESTED, document, value ) ) );
+			indexer.add( documentProvider( keyC, document -> binding.initSingleValued(
+					typeDescriptor, IndexFieldLocation.IN_FLATTENED, document, value ) ) );
+		}
+		return metricAggregationsValues.values().size() * 3;
+	}
+
+	public Result<F> testMetricsAggregation(StubMappingScope scope, SingleFieldIndexBinding binding) {
+		InternalResult<F> result = new InternalResult<>();
+		String fieldPath = binding.getFieldPath( TestedFieldStructure.of(
+				IndexFieldLocation.ROOT, IndexFieldValueCardinality.SINGLE_VALUED ), typeDescriptor );
+
+		SearchQuery<DocumentReference> query = scope.query().where( SearchPredicateFactory::matchAll )
+				.aggregation( result.sumKey(), f -> f.sum().field( fieldPath, typeDescriptor.getJavaType() ) )
+				.toQuery();
+		result.apply( query );
+		return new Result<>( typeDescriptor.getJavaType(), metricAggregationsValues, result );
+	}
+
+	@Override
+	public String toString() {
+		return "Case{" + typeDescriptor + '}';
+	}
+
+	public static class Result<F> {
+		public final Class<F> javaType;
+		public final List<F> values;
+		public final F expectedSum;
+		public final F sumResult;
+
+		public Result(Class<F> javaType, MetricAggregationsValues<F> metricAggregationsValues,
+				InternalResult<F> result) {
+			this.javaType = javaType;
+			values = metricAggregationsValues.values();
+			expectedSum = metricAggregationsValues.expectedSum();
+			sumResult = result.sumResult;
+		}
+
+		@Override
+		public String toString() {
+			return new StringJoiner( ", ", Result.class.getSimpleName() + "[", "]" )
+					.add( "javaType=" + javaType )
+					.add( "values=" + values )
+					.add( "expectedSum=" + expectedSum )
+					.add( "sumResult=" + sumResult )
+					.toString();
+		}
+	}
+
+	private static class InternalResult<F> {
+		AggregationKey<F> sumKey = AggregationKey.of( "sum" );
+		F sumResult = null;
+
+		AggregationKey<F> sumKey() {
+			return sumKey;
+		}
+
+		F sum() {
+			return sumResult;
+		}
+
+		void apply(SearchQuery<DocumentReference> query) {
+			SearchResult<DocumentReference> result = query.fetch( 0 );
+			sumResult = result.aggregation( sumKey );
+		}
+	}
+}

integrationtest/backend/tck/src/main/java/org/hibernate/search/integrationtest/backend/tck/testsupport/util/TestedFieldStructure.java

@@ -29,7 +29,12 @@ public static List<TestedFieldStructure> all() {
 		return ALL;
 	}
 
+	public static TestedFieldStructure of(IndexFieldLocation location, IndexFieldValueCardinality cardinality) {
+		return new TestedFieldStructure( location, cardinality );
+	}
+
 	private static final List<TestedFieldStructure> ALL;
+
 	static {
 		List<TestedFieldStructure> values = new ArrayList<>();
 		for ( IndexFieldLocation location : IndexFieldLocation.values() ) {