Update some NotificationShadeDepthController and...

Update some NotificationShadeDepthController and NotificationShadeDepthControllerTest logic around the spatial_model_pushback_in_shader flag

Need this change to flip the flag with the tests passing

- Added another check on the spatial_model_pushback_in_shader flag
- Added flag conditions in the tests

Bug: 391463550
Test: manual
Flag: com.android.systemui.spatial_model_pushback_in_shader
Change-Id: Ia52266a641873ba297e7257e9ff698d4c68f4e69