raycluster_webhook.go initial unit tests

pkg/controllers/raycluster_webhook.go

@@ -100,27 +100,7 @@ func (w *rayClusterWebhook) Default(ctx context.Context, obj runtime.Object) err
 		}
 
 		// WorkerGroupSpec
-		for i := range rayCluster.Spec.WorkerGroupSpecs {
-			workerSpec := &rayCluster.Spec.WorkerGroupSpecs[i]
-
-			// Append the list of environment variables for the worker container
-			for _, envVar := range envVarList() {
-				workerSpec.Template.Spec.Containers[0].Env = upsert(workerSpec.Template.Spec.Containers[0].Env, envVar, withEnvVarName(envVar.Name))
-			}
-
-			// Append the CA volumes
-			for _, caVol := range caVolumes(rayCluster) {
-				workerSpec.Template.Spec.Volumes = upsert(workerSpec.Template.Spec.Volumes, caVol, withVolumeName(caVol.Name))
-			}
-
-			// Append the certificate volume mounts
-			for _, mount := range certVolumeMounts() {
-				workerSpec.Template.Spec.Containers[0].VolumeMounts = upsert(workerSpec.Template.Spec.Containers[0].VolumeMounts, mount, byVolumeMountName)
-			}
-
-			// Append the create-cert Init Container
-			workerSpec.Template.Spec.InitContainers = upsert(workerSpec.Template.Spec.InitContainers, rayWorkerInitContainer(), withContainerName(initContainerName))
-		}
+		mutateWorkerGroupSpecs(rayCluster)
 	}
 
 	return nil
@@ -392,9 +372,9 @@ func validateWorkerInitContainer(rayCluster *rayv1.RayCluster) field.ErrorList {
 	for i := range rayCluster.Spec.WorkerGroupSpecs {
 		workerSpec := &rayCluster.Spec.WorkerGroupSpecs[i]
 		if err := contains(workerSpec.Template.Spec.InitContainers, rayWorkerInitContainer(), byContainerName,
-		field.NewPath("spec", "workerGroupSpecs", strconv.Itoa(i), "template", "spec", "initContainers"),
-		"create-cert Init Container is immutable"); err != nil {
-		allErrors = append(allErrors, err)
+			field.NewPath("spec", "workerGroupSpecs", strconv.Itoa(i), "template", "spec", "initContainers"),
+			"create-cert Init Container is immutable"); err != nil {
+			allErrors = append(allErrors, err)
 		}
 	}