Lock device on short power button press.

If a device is configured to dream on short power press, also make sure
it locks as well.

Also fixes an issue where the screen does not turn off when another
user (not user zero) presses the power button.

Bug: 260135090
Test: manually by configuring device to dream on short power press and
then making sure it is also locked after press.

Change-Id: I16ca4eab7a0e8a3f2ef42c090147bd7e62445c69