Merge pull request #102 from microsoft/save_bugfix

Add command to restore VM
microsoft · Mar 15, 2024 · edf3c97 · edf3c97
2 parents 248633d + 7854ffe
commit edf3c97
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/pkg/virtualization/core/virtualsystem/virtualmachine.go b/pkg/virtualization/core/virtualsystem/virtualmachine.go
@@ -227,6 +227,15 @@ func (vm *VirtualMachine) Resume() error {
 	return vm.WaitForState(Running, StateChangeTimeoutSeconds)
 }
 
+// Restore Virtual Machine
+func (vm *VirtualMachine) Restore() error {
+	err := vm.ChangeState(Running, job.ConcreteJob_JobType_Restore_Virtual_Machine, -1)
+	if err != nil {
+		return err
+	}
+	return vm.WaitForState(Running, StateChangeTimeoutSeconds)
+}
+
 // Pause Virtual Machine
 func (vm *VirtualMachine) Pause() error {
 	err := vm.ChangeState(Paused, job.ConcreteJob_JobType_Pause_Virtual_Machine, -1)

diff --git a/pkg/virtualization/core/virtualsystem/virtualmachine_test.go b/pkg/virtualization/core/virtualsystem/virtualmachine_test.go
@@ -131,6 +131,23 @@ func TestVirtualMachineResume(t *testing.T) {
 	}
 }
 
+func TestVirtualMachineRestore(t *testing.T) {
+	vm, err := GetVirtualMachineByVMName(whost, "test")
+	if err != nil {
+		t.Fatal("Failed " + err.Error())
+	}
+	defer vm.Close()
+
+	err = vm.Save()
+	if err != nil {
+		t.Fatal("Failed " + err.Error())
+	}
+	err = vm.Restore()
+	if err != nil {
+		t.Fatal("Failed " + err.Error())
+	}
+}
+
 func TestGetVirtualMachineSetting(t *testing.T) {
 	vm, err := GetVirtualMachineByVMName(whost, "test")
 	if err != nil {