Add integration tests for restart command