Use WMStateHelper in testImeVisibleAfterRotation

This adds WindowManagerStateHelper to waitForActivityOrientation,
instead of waiting for idle sync, in testImeVisibleAfterRotation.

Flag: EXEMPT testfix
Bug: 417553159
Test: atest InputMethodServiceTest#testImeVisibleAfterRotation
Change-Id: I8c0dca9d076b4037f31dc6c063c15178fc5cec6e
1 file changed