Fix Mockito flakiness when paired with Parameterized runner

Mockito throwing stubbing exceptions when paired with parameterized
junit runner in ShortcutLoggingTests. Make PWM spy() lenient and
add wait statement to end all runnables before starting more
stubbing in ShortcutLoggingTests.

Test: atest ShortcutLoggingTests
Bug: 293273386
Change-Id: I9f96e3a53423d14259c751b49970277709e2738f