fix build and test

Author: cecemei

multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQCompactionTaskRunTest.java

@@ -556,11 +556,12 @@ public void testIncrementalCompaction() throws Exception
     Assume.assumeTrue("Incremental compaction depends on concurrent lock", useConcurrentLocks);
     verifyTaskSuccessRowsAndSchemaMatch(runIndexTask(), TOTAL_TEST_ROWS);
 
-    final CompactionTask compactionTask1 = compactionTaskBuilder(segmentGranularity).interval(inputInterval).build();
+    final CompactionTask compactionTask1 =
+        compactionTaskBuilder(segmentGranularity).interval(inputInterval, true).build();
 
     final Pair<TaskStatus, DataSegmentsWithSchemas> resultPair1 = runTask(compactionTask1);
     verifyTaskSuccessRowsAndSchemaMatch(resultPair1, TOTAL_TEST_ROWS);
-    verifyCompactedSegment(List.copyOf(resultPair1.rhs.getSegments()), segmentGranularity, DEFAULT_QUERY_GRAN);
+    verifyCompactedSegment(List.copyOf(resultPair1.rhs.getSegments()), segmentGranularity, DEFAULT_QUERY_GRAN, false);
     Assert.assertEquals(1, resultPair1.rhs.getSegments().size());
     final DataSegment compactedSegment1 = Iterables.getOnlyElement(resultPair1.rhs.getSegments());
 
@@ -573,7 +574,7 @@ public void testIncrementalCompaction() throws Exception
                                                         .collect(Collectors.toList());
     final CompactionTask compactionTask2 =
         compactionTaskBuilder(segmentGranularity)
-            .inputSpec(new CompactionIntervalSpec(inputInterval, uncompacted, null))
+            .inputSpec(new CompactionIntervalSpec(inputInterval, uncompacted, null), true)
             .build();
     final Pair<TaskStatus, DataSegmentsWithSchemas> resultPair2 = runTask(compactionTask2);
     verifyTaskSuccessRowsAndSchemaMatch(resultPair2, TOTAL_TEST_ROWS);
@@ -618,12 +619,17 @@ public void testIncrementalCompactionOverlappingInterval() throws Exception
         indexTaskResult.rhs.getSegments().stream().map(DataSegment::toDescriptor).collect(Collectors.toList());
     final CompactionTask compactionTask1 =
         compactionTaskBuilder(segmentGranularity)
-            .inputSpec(new CompactionIntervalSpec(inputInterval, uncompactedFromIndexTask, null))
+            .inputSpec(new CompactionIntervalSpec(inputInterval, uncompactedFromIndexTask, null), true)
             .build();
     final Pair<TaskStatus, DataSegmentsWithSchemas> compactionResult1 = runTask(compactionTask1);
     Assert.assertEquals(1, compactionResult1.rhs.getSegments().size());
     verifyTaskSuccessRowsAndSchemaMatch(compactionResult1, TOTAL_TEST_ROWS);
-    verifyCompactedSegment(List.copyOf(compactionResult1.rhs.getSegments()), segmentGranularity, DEFAULT_QUERY_GRAN);
+    verifyCompactedSegment(
+        List.copyOf(compactionResult1.rhs.getSegments()),
+        segmentGranularity,
+        DEFAULT_QUERY_GRAN,
+        false
+    );
     // Now we have 4 segments from the appendTask, and 1 segment from compactionTask.
     Assert.assertEquals(
         5,
@@ -641,7 +647,7 @@ public void testIncrementalCompactionOverlappingInterval() throws Exception
     Interval compactionInterval = Intervals.of("2014-01-01T00:00:00Z/2014-01-01T06:00:00Z");
     final CompactionTask compactionTask2 =
         compactionTaskBuilder(segmentGranularity)
-            .inputSpec(new CompactionIntervalSpec(compactionInterval, uncompactedSegmentsInHour3, null))
+            .inputSpec(new CompactionIntervalSpec(compactionInterval, uncompactedSegmentsInHour3, null), true)
             .build();
     final Pair<TaskStatus, DataSegmentsWithSchemas> resultPair2 = runTask(compactionTask2);
     verifyTaskSuccessRowsAndSchemaMatch(resultPair2, 3);
@@ -683,7 +689,8 @@ public void testIncrementalCompactionOverlappingInterval() throws Exception
     Assert.assertTrue(usedLoadPath.contains(compactedSegment2Path));
 
     // Final compaction task, we should get 10 (from index task) + 6 (from append task) total rows.
-    CompactionTask finalTask = compactionTaskBuilder(segmentGranularity).interval(Intervals.of("2014-01-01/2014-01-02")).build();
+    CompactionTask finalTask =
+        compactionTaskBuilder(segmentGranularity).interval(Intervals.of("2014-01-01/2014-01-02"), true).build();
     Pair<TaskStatus, DataSegmentsWithSchemas> finalResult = runTask(finalTask);
     verifyTaskSuccessRowsAndSchemaMatch(finalResult, 16);
   }