Skip to content

Commit a80d23e

Browse files
fix(test): fix metadata-io test (#12503)
1 parent d091fff commit a80d23e

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

metadata-io/src/test/java/com/linkedin/metadata/search/query/request/SearchRequestHandlerTest.java

+10-8
Original file line numberDiff line numberDiff line change
@@ -195,13 +195,14 @@ public void testSearchRequestHandlerHighlightingTurnedOff() {
195195
// Filters
196196
Collection<AggregationBuilder> aggBuilders =
197197
sourceBuilder.aggregations().getAggregatorFactories();
198-
// Expect 3 aggregations: textFieldOverride, missing␝textFieldOverride, and _index
199-
assertEquals(aggBuilders.size(), 3);
198+
// Expect 3 aggregations: textFieldOverride, missing␝textFieldOverride, and _entityType,
199+
// _entityType␝typeNames
200+
assertEquals(aggBuilders.size(), 4);
200201
for (AggregationBuilder aggBuilder : aggBuilders) {
201-
if (aggBuilder.getName().equals("textFieldOverride")) {
202+
if (aggBuilder.getName().startsWith("textFieldOverride")) {
202203
TermsAggregationBuilder filterPanelBuilder = (TermsAggregationBuilder) aggBuilder;
203204
assertEquals(filterPanelBuilder.field(), "textFieldOverride.keyword");
204-
} else if (!aggBuilder.getName().equals("_entityType")
205+
} else if (!aggBuilder.getName().startsWith("_entityType")
205206
&& !aggBuilder
206207
.getName()
207208
.equals(
@@ -241,13 +242,14 @@ public void testSearchRequestHandler() {
241242
// Filters
242243
Collection<AggregationBuilder> aggBuilders =
243244
sourceBuilder.aggregations().getAggregatorFactories();
244-
// Expect 2 aggregations: textFieldOverride and _index
245-
assertEquals(aggBuilders.size(), 3);
245+
// Expect 4 aggregations: textFieldOverride, missing:textFieldOverride, _entityType and
246+
// _entityType:typeNames
247+
assertEquals(aggBuilders.size(), 4);
246248
for (AggregationBuilder aggBuilder : aggBuilders) {
247-
if (aggBuilder.getName().equals("textFieldOverride")) {
249+
if (aggBuilder.getName().startsWith("textFieldOverride")) {
248250
TermsAggregationBuilder filterPanelBuilder = (TermsAggregationBuilder) aggBuilder;
249251
assertEquals(filterPanelBuilder.field(), "textFieldOverride.keyword");
250-
} else if (!aggBuilder.getName().equals("_entityType")
252+
} else if (!aggBuilder.getName().startsWith("_entityType")
251253
&& !aggBuilder
252254
.getName()
253255
.equals(

0 commit comments

Comments
 (0)