fix test

apache · Oct 20, 2024 · 4f704d3 · 4f704d3
1 parent fe93f8c
commit 4f704d3
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/pinot-core/src/test/java/org/apache/pinot/core/query/executor/QueryExecutorTest.java b/pinot-core/src/test/java/org/apache/pinot/core/query/executor/QueryExecutorTest.java
@@ -202,7 +202,7 @@ public void testMaxQuery() {
     instanceRequest.setSearchSegments(_segmentNames);
     InstanceResponseBlock instanceResponse = _queryExecutor.execute(getQueryRequest(instanceRequest), QUERY_RUNNERS);
     assertTrue(instanceResponse.getResultsBlock() instanceof AggregationResultsBlock);
-    assertEquals(((AggregationResultsBlock) instanceResponse.getResultsBlock()).getResults().get(0), 2997.0);
+    assertEquals(((AggregationResultsBlock) instanceResponse.getResultsBlock()).getResults().get(0), 999.0);
   }
 
   @Test
@@ -217,19 +217,20 @@ public void testMinQuery() {
 
   @Test
   public void testTimeSeriesSumQuery() {
-    TimeBuckets timeBuckets = TimeBuckets.ofSeconds(TIME_SERIES_TEST_START_TIME, Duration.ofMinutes(1), 100);
+    TimeBuckets timeBuckets = TimeBuckets.ofSeconds(TIME_SERIES_TEST_START_TIME, Duration.ofHours(2), 1);
     ExpressionContext valueExpression = ExpressionContext.forIdentifier("orderAmount");
     TimeSeriesContext timeSeriesContext =
         new TimeSeriesContext(TIME_SERIES_LANGUAGE_NAME, TIME_SERIES_TIME_COL_NAME, TimeUnit.SECONDS, timeBuckets,
             0L /* offsetSeconds */, valueExpression, new AggInfo("SUM", null));
-    QueryContext queryContext = getQueryContextForTimeSeries(timeSeriesContext);
+    QueryContext queryContext = getQueryContextForTimeSeries(timeSeriesContext, Collections.emptyList());
     ServerQueryRequest serverQueryRequest =
         new ServerQueryRequest(queryContext, _segmentNames, new HashMap<>(), ServerMetrics.get());
     InstanceResponseBlock instanceResponse = _queryExecutor.execute(serverQueryRequest, QUERY_RUNNERS);
     assertTrue(instanceResponse.getResultsBlock() instanceof TimeSeriesResultsBlock);
     TimeSeriesResultsBlock resultsBlock = (TimeSeriesResultsBlock) instanceResponse.getResultsBlock();
     TimeSeriesBlock timeSeriesBlock = resultsBlock.getTimeSeriesBuilderBlock().build();
-    assertEquals(5, timeSeriesBlock.getSeriesMap().size());
+    assertEquals(timeSeriesBlock.getSeriesMap().size(), 1);
+    assertEquals(timeSeriesBlock.getSeriesMap().values().iterator().next().get(0).getValues()[0], 29885544.0);
   }
 
   @Test
@@ -309,12 +310,17 @@ private ServerQueryRequest getQueryRequest(InstanceRequest instanceRequest) {
   }
 
   private QueryContext getQueryContextForTimeSeries(TimeSeriesContext context) {
+    return getQueryContextForTimeSeries(context, Collections.singletonList(
+        ExpressionContext.forIdentifier("cityName")));
+  }
+
+  private QueryContext getQueryContextForTimeSeries(TimeSeriesContext context, List<ExpressionContext> groupBy) {
     QueryContext.Builder builder = new QueryContext.Builder();
     builder.setTableName(OFFLINE_TABLE_NAME);
     builder.setTimeSeriesContext(context);
     builder.setAliasList(Collections.emptyList());
     builder.setSelectExpressions(Collections.emptyList());
-    builder.setGroupByExpressions(Collections.singletonList(ExpressionContext.forIdentifier("cityName")));
+    builder.setGroupByExpressions(groupBy);
     return builder.build();
   }
 }