Add tests for restarting services