Ensure the screen is locked after closing the screen

There are tests assume the screen is on but locked by calling close
screen and wake up screen. In order to make sure the precondition of the
test is correct, we need to ensure that the entire screen is locked when
close the screen.

Bug: 337259815
Test: atest AccessibilityMenuTests --iterations 50
Flag: TEST_ONLY
Change-Id: I70c395881b4951cfc4a66eec98238cdb1262fd3b