Fix tests that require the screen to be on.

Acquire the wake lock as the tests needs the screen to be on.
Previously the test blocked until I pressed the power button on the
device, unless the device was already on or set to keep the screen on
while plugged in.

Change-Id: I61ccbcc1b18bb954128c832b67efddd23dae5f08