Ensure lock screen setting is enabled before running the tests

Also, wait for the screen is on/off even after the screen on/off value
is settled.

Bug: 338203824
Bug: 337259815
Bug: 338685116
Bug: 338687045

Test: atest AccessibilityMenuServiceTests --iterations 10
Flag: TEST_ONLY
Change-Id: I16d531a6e6dfcca7c01429b7f856758cd124241f