fix remaining metadata-io test i forgot about

datahub-project · Dec 11, 2024 · b51804f · b51804f
1 parent 5064172
commit b51804f
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/...ata-io/src/test/java/com/linkedin/metadata/search/fixtures/SampleDataFixtureTestBase.java b/...ata-io/src/test/java/com/linkedin/metadata/search/fixtures/SampleDataFixtureTestBase.java
@@ -1124,9 +1124,10 @@ public void testFacets() {
   @Test
   public void testNestedAggregation() {
     Set<String> expectedFacets = Set.of("platform");
+    OperationContext context =
+        getOperationContext().withSearchFlags(flags -> flags.setIncludeDefaultFacets(false));
     SearchResult testResult =
-        facetAcrossEntities(
-            getOperationContext(), getSearchService(), "cypress", List.copyOf(expectedFacets));
+        facetAcrossEntities(context, getSearchService(), "cypress", List.copyOf(expectedFacets));
     assertEquals(testResult.getMetadata().getAggregations().size(), 1);
     expectedFacets.forEach(
         facet -> {
@@ -1143,8 +1144,7 @@ public void testNestedAggregation() {
 
     expectedFacets = Set.of("platform", "typeNames", "_entityType", "entity");
     SearchResult testResult2 =
-        facetAcrossEntities(
-            getOperationContext(), getSearchService(), "cypress", List.copyOf(expectedFacets));
+        facetAcrossEntities(context, getSearchService(), "cypress", List.copyOf(expectedFacets));
     assertEquals(testResult2.getMetadata().getAggregations().size(), 4);
     expectedFacets.forEach(
         facet -> {
@@ -1191,8 +1191,7 @@ public void testNestedAggregation() {
 
     expectedFacets = Set.of("platform", "typeNames", "entity");
     SearchResult testResult3 =
-        facetAcrossEntities(
-            getOperationContext(), getSearchService(), "cypress", List.copyOf(expectedFacets));
+        facetAcrossEntities(context, getSearchService(), "cypress", List.copyOf(expectedFacets));
     assertEquals(testResult3.getMetadata().getAggregations().size(), 4);
     expectedFacets.forEach(
         facet -> {
@@ -1222,8 +1221,7 @@ public void testNestedAggregation() {
     String singleNestedFacet = String.format("_entityType%sowners", AGGREGATION_SEPARATOR_CHAR);
     expectedFacets = Set.of(singleNestedFacet);
     SearchResult testResultSingleNested =
-        facetAcrossEntities(
-            getOperationContext(), getSearchService(), "cypress", List.copyOf(expectedFacets));
+        facetAcrossEntities(context, getSearchService(), "cypress", List.copyOf(expectedFacets));
     assertEquals(testResultSingleNested.getMetadata().getAggregations().size(), 1);
     Map<String, Long> expectedNestedFacetCounts = new HashMap<>();
     expectedNestedFacetCounts.put("datajob␞urn:li:corpuser:datahub", 2L);
@@ -1245,8 +1243,7 @@ public void testNestedAggregation() {
 
     expectedFacets = Set.of("platform", singleNestedFacet, "typeNames", "origin");
     SearchResult testResultNested =
-        facetAcrossEntities(
-            getOperationContext(), getSearchService(), "cypress", List.copyOf(expectedFacets));
+        facetAcrossEntities(context, getSearchService(), "cypress", List.copyOf(expectedFacets));
     assertEquals(testResultNested.getMetadata().getAggregations().size(), 4);
     expectedFacets.forEach(
         facet -> {