Attempts fixing the MultiWindowTests flakiness

Compute the wm state every time re-checking the focus.

Also adding log to better understand the issue if it
happens again.

Bug: 356616111
Test: MultiWindowTests
Flag: TEST_ONLY
Change-Id: Ifce43bc5c1b701ef985aaea22ea2a3e87abdc300