Adds rolling shut down test and the unit tests related to it
Make tests pass.