Added tests for VM migration