Fix VMware scheduler test (#1094)

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: #1087 Signed-off-by: Martin Necas <[email protected]>
kubev2v · Oct 10, 2024 · 100bbea · 100bbea
1 parent a387e6b
commit 100bbea
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 @@