test fixes

apache · Jan 12, 2024 · ed13089 · ed13089
1 parent c3f722a
commit ed13089
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQSelectTest.java b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/exec/MSQSelectTest.java
@@ -1418,14 +1418,14 @@ public void testExternSelectWithMultipleWorkers() throws IOException
         .setExpectedResultRows(ImmutableList.of(
             new Object[]{1466985600000L, "Lsjbot"},
             new Object[]{1466985600000L, "Lsjbot"},
-            new Object[]{1466985600000L, "Beau.bot"},
-            new Object[]{1466985600000L, "Beau.bot"},
             new Object[]{1466985600000L, "Lsjbot"},
             new Object[]{1466985600000L, "Lsjbot"},
-            new Object[]{1466985600000L, "TaxonBot"},
-            new Object[]{1466985600000L, "TaxonBot"},
             new Object[]{1466985600000L, "GiftBot"},
-            new Object[]{1466985600000L, "GiftBot"}
+            new Object[]{1466985600000L, "GiftBot"},
+            new Object[]{1466985600000L, "Beau.bot"},
+            new Object[]{1466985600000L, "Beau.bot"},
+            new Object[]{1466985600000L, "TaxonBot"},
+            new Object[]{1466985600000L, "TaxonBot"}
         ))
         .setExpectedMSQSpec(
             MSQSpec

diff --git a/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/MSQTestBase.java b/extensions-core/multi-stage-query/src/test/java/org/apache/druid/msq/test/MSQTestBase.java
@@ -1404,8 +1404,8 @@ public Pair<MSQSpec, Pair<List<MSQResultsReport.ColumnAndType>, List<Object[]>>>
               );
               rows.addAll(new FrameChannelSequence(inputChannelFactory.openChannel(
                   finalStage.getId(),
-                  pageInformation.getWorker(),
-                  pageInformation.getPartition()
+                  pageInformation.getWorker() == null ? 0 : pageInformation.getWorker(),
+                  pageInformation.getPartition() == null ? 0 : pageInformation.getPartition()
               )).flatMap(frame -> SqlStatementResourceHelper.getResultSequence(
                   msqControllerTask,
                   finalStage,