Skip to content
Snippets Groups Projects
Commit 754f853b authored by robbins's avatar robbins Committed by Andrew Or
Browse files

[SPARK-9869] [STREAMING] Wait for all event notifications before asserting results

Author: robbins <robbins@uk.ibm.com>

Closes #8589 from robbinspg/InputStreamSuite-fix.
parent d911c682
No related branches found
No related tags found
No related merge requests found
......@@ -76,6 +76,9 @@ class InputStreamsSuite extends TestSuiteBase with BeforeAndAfter {
fail("Timeout: cannot finish all batches in 30 seconds")
}
// Ensure progress listener has been notified of all events
ssc.scheduler.listenerBus.waitUntilEmpty(500)
// Verify all "InputInfo"s have been reported
assert(ssc.progressListener.numTotalReceivedRecords === input.size)
assert(ssc.progressListener.numTotalProcessedRecords === input.size)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment