Skip to content

Commit b975d44

Browse files
committed
Consume sink batches and clear input
1 parent 7279c65 commit b975d44

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

cpp/src/arrow/acero/asof_join_node_test.cc

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1683,8 +1683,12 @@ TEST(AsofJoinTest, PauseProducingAsofJoinSource) {
16831683
ASSERT_FINISHES_OK(sink_gen());
16841684
BusyWait(10, [&] { return !backpressure_monitor->is_paused(); });
16851685
ASSERT_FALSE(backpressure_monitor->is_paused());
1686-
1686+
16871687
// Cleanup
1688+
for (size_t i = 0; i < cnt - 3; i++) {
1689+
ASSERT_FINISHES_OK(sink_gen());
1690+
}
1691+
16881692
batch_producer_left.producer().Push(IterationEnd<std::optional<ExecBatch>>());
16891693
batch_producer_right.producer().Push(IterationEnd<std::optional<ExecBatch>>());
16901694

0 commit comments

Comments
 (0)