I thought about using a clock abstraction here also but wasn't sure if Spark had a pattern/way to annotate time-sensitive tests like this or what the general pattern Spark uses for time-sensitive tests. I agree that we can change this to use a Clock abstraction if we see that it breaks, etc.