wypoon commented on a change in pull request #25673: [SPARK-28770][CORE][TEST]
Fix ReplayListenerSuite tests that sometime…
URL: https://github.com/apache/spark/pull/25673#discussion_r320850212
##########
File path:
core/src/test/scala/org/apache/spark/scheduler/ReplayListenerSuite.scala
##########
@@ -219,11 +219,25 @@ class ReplayListenerSuite extends SparkFunSuite with
BeforeAndAfter with LocalSp
// Verify the same events are replayed in the same order
assert(sc.eventLogger.isDefined)
val originalEvents = sc.eventLogger.get.loggedEvents
+ .map(JsonProtocol.sparkEventFromJson(_))
val replayedEvents = eventMonster.loggedEvents
- originalEvents.zip(replayedEvents).foreach { case (e1, e2) =>
- // Don't compare the JSON here because accumulators in StageInfo may be
out of order
- JsonProtocolSuite.assertEquals(
- JsonProtocol.sparkEventFromJson(e1),
JsonProtocol.sparkEventFromJson(e2))
+ .map(JsonProtocol.sparkEventFromJson(_))
+ // Executor metrics updates are not logged, so do not get replayed.
+ // Stage executor metrics are logged at stage completion, for any of the
executors and
+ // the driver for which we have metrics. We always have metrics for
executors, because
+ // they are sent at task end as well as in executor metrics updates. We do
not always
+ // have metrics for the driver, because they are only sent in executor
metrics updates
+ // (at heartbeat intervals), and when we do, it is only in the original
events, never
+ // in the replay, since executor metrics updates are not replayed.
+ // For this reason, exclude stage executor metrics for the driver.
+ val filteredEvents = originalEvents.filter { e =>
Review comment:
Yes, that's nicer. Thanks.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
