Adopt end animation callback adjustment for tests

With Animator.setPostNotifyEndListenerEnabled set to true
(for excluding potential heavy operation from last animating frame),
the end callback will be notified on next frame. So make the test
compatible with the behavior.

Bug: 300035126
Flag: com.android.window.flags.system_ui_post_animation_end
Test: atest SystemUITests:WindowMagnificationAnimationControllerTest
            SystemUITests:FoldLightRevealOverlayAnimationTest
            SystemUITests:LoadingEffectTest
            SystemUITests:GlowBoxEffectTest

Change-Id: Idb7be51f5d01e080f2375dd3d022915d48b5cc53