fix NodePodProbe nil labels panic

Signed-off-by: Jayant <212013719+Jayant-kernel@users.noreply.github.com>

Author: Jayant-kernel

pkg/controller/nodepodprobe/node_pod_probe_controller.go

@@ -337,6 +337,9 @@ func (r *ReconcileNodePodProbe) updatePodProbeStatus(pod *corev1.Pod, status app
 			util.SetPodConditionIfMsgChanged(podClone, condition)
 		}
 		oldMetadata := podClone.ObjectMeta.DeepCopy()
+		if podClone.Labels == nil {
+			podClone.Labels = map[string]string{}
+		}
 		if podClone.Annotations == nil {
 			podClone.Annotations = map[string]string{}
 		}

pkg/controller/nodepodprobe/node_pod_probe_controller_test.go

@@ -797,6 +797,62 @@ func TestSyncPodFromNodePodProbe(t *testing.T) {
 	}
 }
 
+func TestUpdatePodProbeStatusInitializesNilLabels(t *testing.T) {
+	pod := &corev1.Pod{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "pod-1",
+			Namespace: "default",
+		},
+	}
+	ppm := &appsv1alpha1.PodProbeMarker{
+		ObjectMeta: metav1.ObjectMeta{
+			Name:      "ppm-1",
+			Namespace: "default",
+		},
+		Spec: appsv1alpha1.PodProbeMarkerSpec{
+			Probes: []appsv1alpha1.PodContainerProbe{
+				{
+					Name: "healthy",
+					MarkerPolicy: []appsv1alpha1.ProbeMarkerPolicy{
+						{
+							State: appsv1alpha1.ProbeSucceeded,
+							Labels: map[string]string{
+								"server-healthy": "true",
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+	status := appsv1alpha1.PodProbeStatus{
+		ProbeStates: []appsv1alpha1.ContainerProbeState{
+			{
+				Name:  "ppm-1#healthy",
+				State: appsv1alpha1.ProbeSucceeded,
+			},
+		},
+	}
+
+	fakeClient := fake.NewClientBuilder().
+		WithScheme(scheme).
+		WithObjects(pod, ppm).
+		Build()
+	recon := ReconcileNodePodProbe{Client: fakeClient}
+
+	if err := recon.updatePodProbeStatus(pod, status); err != nil {
+		t.Fatalf("updatePodProbeStatus failed: %s", err.Error())
+	}
+
+	updatedPod := &corev1.Pod{}
+	if err := fakeClient.Get(context.TODO(), types.NamespacedName{Namespace: pod.Namespace, Name: pod.Name}, updatedPod); err != nil {
+		t.Fatalf("get Pod failed: %s", err.Error())
+	}
+	if updatedPod.Labels["server-healthy"] != "true" {
+		t.Fatalf("expected label server-healthy=true, got %v", updatedPod.Labels)
+	}
+}
+
 func checkNodePodProbeEqual(c client.WithWatch, t *testing.T, expect []*appsv1alpha1.NodePodProbe) bool {
 	for i := range expect {
 		obj := expect[i]