Fix tests for keyguard_wm_state_refactor flag.

We need to set up more consistent state for the refactor or it won't actually start the transitions.

Test: atest SystemUITests
Flag: EXEMPT tests
Bug: 278086361
Change-Id: I6ce67018b30a07630432d7d657a826d6062b7023
2 files changed