Fix waitingForDrawn callback not called when adjusting for IME

am: 3188386ff8

Change-Id: I6c844d1b86836673329a198b0f1d97bd07ff22f3

services/core/java/com/android/server/wm/DockedStackDividerController.java

@@ -658,6 +658,14 @@ public class DockedStackDividerController implements DimLayerUser {
                 mDelayedImeWin = imeWin;
                 imeWin.mWinAnimator.startDelayingAnimationStart();
             }
+
+            // If we are already waiting for something to be drawn, clear out the old one so it
+            // still gets executed.
+            // TODO: Have a real system where we can wait on different windows to be drawn with
+            // different callbacks.
+            if (mService.mWaitingForDrawnCallback != null) {
+                mService.mWaitingForDrawnCallback.run();
+            }
             mService.mWaitingForDrawnCallback = () -> {
                 mAnimationStartDelayed = false;
                 if (mDelayedImeWin != null) {