Refactor the buildTransitions into overridable functions

Cleanup setup states in some of the tests.

Test: atest WMShellFlickerTests
Bug: 285226733

Change-Id: I1f59794ec331bc0f59febd316ad29b20ed06805d
20 files changed