@@ -1091,41 +1091,42 @@ class TestCustomExchangeTranslator : public exec::Operator::PlanNodeTranslator {
10911091 }
10921092};
10931093
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+ // }
11291130
11301131// This test is to reproduce the race condition between task terminate and no
11311132// more split call:
0 commit comments