add integration test (#183)

Co-authored-by: Srikar Mannepalli <[email protected]>

document-store/src/integrationTest/java/org/hypertrace/core/documentstore/DocStoreQueryV1Test.java

@@ -83,6 +83,7 @@
 import org.hypertrace.core.documentstore.expression.impl.LogicalExpression;
 import org.hypertrace.core.documentstore.expression.impl.RelationalExpression;
 import org.hypertrace.core.documentstore.expression.impl.UnnestExpression;
+import org.hypertrace.core.documentstore.expression.type.FilterTypeExpression;
 import org.hypertrace.core.documentstore.model.options.UpdateOptions;
 import org.hypertrace.core.documentstore.model.subdoc.SubDocumentUpdate;
 import org.hypertrace.core.documentstore.model.subdoc.SubDocumentValue;
@@ -822,6 +823,53 @@ public void testAggregateWithFunctionalLeftHandSideFilter(final String dataStore
     testCountApi(dataStoreName, query, "query/test_functional_lhs_in_filter_response.json");
   }
 
+  @ParameterizedTest
+  @ArgumentsSource(AllProvider.class)
+  void testAggregateWithNestedArraysAndUnnestFilters(final String dataStoreName)
+      throws IOException {
+    final Datastore datastore = datastoreMap.get(dataStoreName);
+    final Collection collection = datastore.getCollection(COLLECTION_NAME);
+    final FilterTypeExpression filter =
+        LogicalExpression.builder()
+            .operator(AND)
+            .operand(
+                RelationalExpression.of(
+                    IdentifierExpression.of("sales.medium.type"),
+                    EQ,
+                    ConstantExpression.of("distributionChannel")))
+            .operand(
+                RelationalExpression.of(
+                    IdentifierExpression.of("price"), LT, ConstantExpression.of(20)))
+            .build();
+    final Query query =
+        Query.builder()
+            .setSelections(
+                List.of(
+                    SelectionSpec.of(IdentifierExpression.of("item")),
+                    SelectionSpec.of(IdentifierExpression.of("sales.city")),
+                    SelectionSpec.of(IdentifierExpression.of("price")),
+                    SelectionSpec.of(IdentifierExpression.of("sales.medium.type")),
+                    SelectionSpec.of(IdentifierExpression.of("sales.medium.volume"))))
+            .setFilter(filter)
+            .addFromClauses(
+                List.of(
+                    UnnestExpression.builder()
+                        .preserveNullAndEmptyArrays(false)
+                        .identifierExpression(IdentifierExpression.of("sales"))
+                        .filterTypeExpression(filter)
+                        .build(),
+                    UnnestExpression.builder()
+                        .preserveNullAndEmptyArrays(false)
+                        .identifierExpression(IdentifierExpression.of("sales.medium"))
+                        .filterTypeExpression(filter)
+                        .build()))
+            .build();
+
+    final Iterator<Document> resultDocs = collection.aggregate(query);
+    assertDocsAndSizeEqual(
+        dataStoreName, resultDocs, "query/test_aggr_nested_arrays_and_unnest_filters.json", 4);
+  }
+
   @ParameterizedTest
   @ArgumentsSource(AllProvider.class)
   public void testFindWithFunctionalLeftHandSideFilter(final String dataStoreName)

document-store/src/integrationTest/resources/query/test_aggr_nested_arrays_and_unnest_filters.json

@@ -0,0 +1,46 @@
+[
+  {
+    "item": "Soap",
+    "price": 10,
+    "sales": {
+      "city": "delhi",
+      "medium": {
+        "type": "distributionChannel",
+        "volume": 1000
+      }
+    }
+  },
+  {
+    "item": "Soap",
+    "price": 10,
+    "sales": {
+      "city": "pune",
+      "medium": {
+        "type": "distributionChannel",
+        "volume": 300
+      }
+    }
+  },
+  {
+    "item": "Shampoo",
+    "price": 5,
+    "sales": {
+      "city": "delhi",
+      "medium": {
+        "type": "distributionChannel",
+        "volume": 3000
+      }
+    }
+  },
+  {
+    "item": "Shampoo",
+    "price": 5,
+    "sales": {
+      "city": "mumbai",
+      "medium": {
+        "type": "distributionChannel",
+        "volume": 700
+      }
+    }
+  }
+]