@@ -1091,41 +1091,42 @@ class TestCustomExchangeTranslator : public exec::Operator::PlanNodeTranslator {
1091
1091
}
1092
1092
};
1093
1093
1094
- TEST_F (MultiFragmentTest, customPlanNodeWithExchangeClient) {
1095
- setupSources (5 , 100 );
1096
- Operator::registerOperator (std::make_unique<TestCustomExchangeTranslator>());
1097
- auto leafTaskId = makeTaskId (" leaf" , 0 );
1098
- auto leafPlan =
1099
- PlanBuilder ().values (vectors_).partitionedOutput ({}, 1 ).planNode ();
1100
- auto leafTask = makeTask (leafTaskId, leafPlan, 0 );
1101
- Task::start (leafTask, 1 );
1102
-
1103
- CursorParameters params;
1104
- core::PlanNodeId testNodeId;
1105
- params.maxDrivers = 1 ;
1106
- params.planNode =
1107
- PlanBuilder ()
1108
- .addNode ([&leafPlan](std::string id, core::PlanNodePtr /* input */ ) {
1109
- return std::make_shared<TestCustomExchangeNode>(
1110
- id, leafPlan->outputType ());
1111
- })
1112
- .capturePlanNodeId (testNodeId)
1113
- .planNode ();
1114
-
1115
- auto cursor = std::make_unique<TaskCursor>(params);
1116
- auto task = cursor->task ();
1117
- addRemoteSplits (task, {leafTaskId});
1118
- while (cursor->moveNext ()) {
1119
- }
1120
- EXPECT_NE (
1121
- toPlanStats (task->taskStats ())
1122
- .at (testNodeId)
1123
- .customStats .count (" testCustomExchangeStat" ),
1124
- 0 );
1125
- ASSERT_TRUE (waitForTaskCompletion (leafTask.get (), 3'000'000 ))
1126
- << leafTask->taskId ();
1127
- ASSERT_TRUE (waitForTaskCompletion (task.get (), 3'000'000 )) << task->taskId ();
1128
- }
1094
+ // TEST_F(MultiFragmentTest, customPlanNodeWithExchangeClient) {
1095
+ // setupSources(5, 100);
1096
+ // Operator::registerOperator(std::make_unique<TestCustomExchangeTranslator>());
1097
+ // auto leafTaskId = makeTaskId("leaf", 0);
1098
+ // auto leafPlan =
1099
+ // PlanBuilder().values(vectors_).partitionedOutput({}, 1).planNode();
1100
+ // auto leafTask = makeTask(leafTaskId, leafPlan, 0);
1101
+ // Task::start(leafTask, 1);
1102
+
1103
+ // CursorParameters params;
1104
+ // core::PlanNodeId testNodeId;
1105
+ // params.maxDrivers = 1;
1106
+ // params.planNode =
1107
+ // PlanBuilder()
1108
+ // .addNode([&leafPlan](std::string id, core::PlanNodePtr /* input */)
1109
+ // {
1110
+ // return std::make_shared<TestCustomExchangeNode>(
1111
+ // id, leafPlan->outputType());
1112
+ // })
1113
+ // .capturePlanNodeId(testNodeId)
1114
+ // .planNode();
1115
+
1116
+ // auto cursor = std::make_unique<TaskCursor>(params);
1117
+ // auto task = cursor->task();
1118
+ // addRemoteSplits(task, {leafTaskId});
1119
+ // while (cursor->moveNext()) {
1120
+ // }
1121
+ // EXPECT_NE(
1122
+ // toPlanStats(task->taskStats())
1123
+ // .at(testNodeId)
1124
+ // .customStats.count("testCustomExchangeStat"),
1125
+ // 0);
1126
+ // ASSERT_TRUE(waitForTaskCompletion(leafTask.get(), 3'000'000))
1127
+ // << leafTask->taskId();
1128
+ // ASSERT_TRUE(waitForTaskCompletion(task.get(), 3'000'000)) << task->taskId();
1129
+ // }
1129
1130
1130
1131
// This test is to reproduce the race condition between task terminate and no
1131
1132
// more split call:
0 commit comments