Fix reclaim test #2

Author: itsomri

pkg/scheduler/actions/reclaim/reclaim_test.go

@@ -754,6 +754,13 @@ func getTestsMetadata() []integration_tests_utils.TestTopologyMetadata {
 								NodeName: "node0",
 								State:    pod_status.Running,
 							},
+						},
+					}, {
+						Name:                "q0_job1",
+						RequiredGPUsPerTask: 1,
+						Priority:            constants.PriorityTrainNumber,
+						QueueName:           "queue0",
+						Tasks: []*tasks_fake.TestTaskBasic{
 							{
 								NodeName: "node1",
 								State:    pod_status.Running,
@@ -762,15 +769,19 @@ func getTestsMetadata() []integration_tests_utils.TestTopologyMetadata {
 								NodeName: "node2",
 								State:    pod_status.Running,
 							},
+							{
+								NodeName: "node3",
+								State:    pod_status.Running,
+							},
 						},
 					}, {
-						Name:                "q0_job1",
+						Name:                "q0_job2",
 						RequiredGPUsPerTask: 1,
 						Priority:            constants.PriorityTrainNumber,
 						QueueName:           "queue0",
 						Tasks: []*tasks_fake.TestTaskBasic{
 							{
-								NodeName: "node3",
+								NodeName: "node4",
 								State:    pod_status.Running,
 							},
 						},
@@ -805,6 +816,9 @@ func getTestsMetadata() []integration_tests_utils.TestTopologyMetadata {
 					"node3": {
 						GPUs: 1,
 					},
+					"node4": {
+						GPUs: 1,
+					},
 				},
 				Queues: []test_utils.TestQueueBasic{
 					{
@@ -820,11 +834,16 @@ func getTestsMetadata() []integration_tests_utils.TestTopologyMetadata {
 				},
 				JobExpectedResults: map[string]test_utils.TestExpectedResultBasic{
 					"q0_job0": {
+						GPUsRequired:         1,
+						Status:               pod_status.Running,
+						DontValidateGPUGroup: true,
+					},
+					"q0_job1": {
 						GPUsRequired:         3,
 						Status:               pod_status.Releasing,
 						DontValidateGPUGroup: true,
 					},
-					"q0_job1": {
+					"q0_job2": {
 						GPUsRequired:         1,
 						Status:               pod_status.Running,
 						DontValidateGPUGroup: true,