One more test

kubeflow · Oct 31, 2024 · 083f308 · 083f308
1 parent c0a34b9
commit 083f308
Showing 1 changed file with 33 additions and 0 deletions.
diff --git a/pkg/controller/podgroup_test.go b/pkg/controller/podgroup_test.go
@@ -519,6 +519,39 @@ func TestCalculatePGMinResources(t *testing.T) {
 				corev1.ResourceMemory: resource.MustParse("65Gi"),
 			},
 		},
+		"without worker without priorityClass": {
+			minMember: 3,
+			job: &kubeflow.MPIJob{
+				ObjectMeta: metav1.ObjectMeta{
+					Name: "test",
+				},
+				Spec: kubeflow.MPIJobSpec{
+					MPIReplicaSpecs: map[kubeflow.MPIReplicaType]*kubeflow.ReplicaSpec{
+						kubeflow.MPIReplicaTypeLauncher: {
+							Replicas: ptr.To[int32](1),
+							Template: corev1.PodTemplateSpec{
+								Spec: corev1.PodSpec{
+									Containers: []corev1.Container{
+										{
+											Resources: corev1.ResourceRequirements{
+												Requests: corev1.ResourceList{
+													corev1.ResourceCPU:    resource.MustParse("2"),
+													corev1.ResourceMemory: resource.MustParse("1Gi"),
+												},
+											},
+										},
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			want: &corev1.ResourceList{
+				corev1.ResourceCPU:    resource.MustParse("2"),
+				corev1.ResourceMemory: resource.MustParse("1Gi"),
+			},
+		},
 	}
 	for name, tc := range volcanoTests {
 		t.Run(name, func(t *testing.T) {