Fix some shell unit tests with shell transitions

Bug: 262783148
Test: atest WMShellUnitTests
Change-Id: I2147c2aa76991ced19e80dc44b68262907caca87
2 files changed