Skip to content

Commit ad21576

Browse files
comments unstable customPlanNodeWithExchangeClient (#248)
1 parent 95edde0 commit ad21576

File tree

1 file changed

+36
-35
lines changed

1 file changed

+36
-35
lines changed

velox/exec/tests/MultiFragmentTest.cpp

Lines changed: 36 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)