Update UMO host attachment before state change

Sets full shade transition amount to maximum before changing statusbar
state to shade_locked. This makes sure that UMO is attached to QS and is
not hanging in the IN_OVERLAY location.

Flag: NONE
Bug: 295269178
Test: Checked UMO when transitioning between LS and QS.
Change-Id: Ic74bcd9e5a2cdbb48712fb2c861c540425b8bd7e