Fix VMware scheduler test

Now we allow migrating with more disks than the MaxInFlight in case there is no VM being migrated at that time. This is done for the user's convenience. Related to: kubev2v#1087 Signed-off-by: Martin Necas <[email protected]>
mnecas · Oct 9, 2024 · 55a7807 · 55a7807
1 parent 8a8c742
commit 55a7807
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/pkg/controller/plan/scheduler/vsphere/scheduler.go b/pkg/controller/plan/scheduler/vsphere/scheduler.go
@@ -226,9 +226,9 @@ func (r *Scheduler) cost(vm *model.VM, vmStatus *plan.VMStatus) int {
 func (r *Scheduler) finishedDisks(vmStatus *plan.VMStatus) int {
 	var resp = 0
 	for _, step := range vmStatus.Pipeline {
-		if step.Name == "DiskTransfer" {
+		if step.Name == DiskTransfer {
 			for _, task := range step.Tasks {
-				if task.Phase == "Completed" {
+				if task.Phase == Completed {
 					resp += 1
 				}
 			}

diff --git a/pkg/controller/plan/scheduler/vsphere/scheduler_test.go b/pkg/controller/plan/scheduler/vsphere/scheduler_test.go
@@ -83,6 +83,9 @@ func TestScheduler(t *testing.T) {
 			},
 		},
 		hostC: {
+			{
+				cost: 11,
+			},
 			{
 				cost: 1,
 			},
-Original file line number
+Diff line change
@@ Expand Up / @@ -83,6 +83,9 @@ func TestScheduler(t *testing.T) { @@
     			},
     		},
     		hostC: {
+    			{
+    				cost: 11,
+    			},
     			{
     				cost: 1,
     			},
@@ Expand Down @@