YongGang commented on code in PR #16041: URL: https://github.com/apache/druid/pull/16041#discussion_r1558839305
########## integration-tests-ex/cases/src/test/java/org/apache/druid/testsEx/msq/ITMultiStageQuery.java: ########## @@ -234,11 +236,15 @@ public void testExport() throws Exception msqHelper.pollTaskIdForSuccess(resultTaskStatus.getTaskId()); - Map<String, MSQTaskReport> statusReport = msqHelper.fetchStatusReports(resultTaskStatus.getTaskId()); - MSQTaskReport taskReport = statusReport.get(MSQTaskReport.REPORT_KEY); + Map<String, TaskReport> statusReport = msqHelper.fetchStatusReports(resultTaskStatus.getTaskId()); + + MSQTaskReport taskReport = (MSQTaskReport) statusReport.get(MSQTaskReport.REPORT_KEY); if (taskReport == null) { throw new ISE("Unable to fetch the status report for the task [%]", resultTaskStatus.getTaskId()); } + TaskContextReport taskContextReport = (TaskContextReport) statusReport.get(TaskContextReport.REPORT_KEY); + Assert.assertFalse(taskContextReport.getPayload().isEmpty()); Review Comment: This is to assure TaskContextReport is generated for the task. Since the payload is task context which is hard to know the exact value for comparison so here it just check whether the field is populated. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: commits-unsubscr...@druid.apache.org For additional commands, e-mail: commits-h...@druid.apache.org