Wait for last orientation after expanding pip

Bug: 156871322
Test: atest PinnedStackTests
Change-Id: I14632fd91e834d5634c655870aee2720aa6f9ab8
diff --git a/tests/framework/base/windowmanager/src/android/server/wm/PinnedStackTests.java b/tests/framework/base/windowmanager/src/android/server/wm/PinnedStackTests.java
index ab3d71a..1d6ea9e 100644
--- a/tests/framework/base/windowmanager/src/android/server/wm/PinnedStackTests.java
+++ b/tests/framework/base/windowmanager/src/android/server/wm/PinnedStackTests.java
@@ -983,6 +983,7 @@
         launchActivity(PIP_ACTIVITY);
         waitForExitPipToFullscreen(PIP_ACTIVITY);
         assertPinnedStackDoesNotExist();
+        mWmState.waitForLastOrientation(ORIENTATION_LANDSCAPE);
         assertEquals(ORIENTATION_LANDSCAPE, mWmState.getLastOrientation());
     }