From 22aea166e1e97b9bccf28a62022483440826546f Mon Sep 17 00:00:00 2001
From: Goutham Annem <annem.usedu@gmail.com>
Date: Sat, 27 Jun 2026 11:08:09 -0700
Subject: [PATCH] test: add unit tests for SetLabelsAndAnnotationsForWorkload
 and GetWorkSuspendDispatching

Add unit tests for two functions in pkg/util/work.go that lacked
coverage:
- GetWorkSuspendDispatching: covers true, false, and nil
  SuspendDispatching pointer (defaults to false)
- SetLabelsAndAnnotationsForWorkload: covers the managed-annotations
  and managed-labels annotation population, and propagation of the
  WorkPermanentIDLabel from the Work object to the workload

Fixes #7673

Signed-off-by: Goutham Annem <annem.usedu@gmail.com>
---
 pkg/util/work_test.go | 112 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)

diff --git a/pkg/util/work_test.go b/pkg/util/work_test.go
index 373e0a1d4c4d..1e1ee8bb5993 100644
--- a/pkg/util/work_test.go
+++ b/pkg/util/work_test.go
@@ -252,3 +252,115 @@ func TestIsWorkSuspendDispatching(t *testing.T) {
 		})
 	}
 }
+
+func TestGetWorkSuspendDispatching(t *testing.T) {
+	trueVal := true
+	falseVal := false
+	tests := []struct {
+		name string
+		spec *workv1alpha1.WorkSpec
+		want []string
+	}{
+		{
+			name: "SuspendDispatching is true",
+			spec: &workv1alpha1.WorkSpec{SuspendDispatching: &trueVal},
+			want: []string{"true"},
+		},
+		{
+			name: "SuspendDispatching is false",
+			spec: &workv1alpha1.WorkSpec{SuspendDispatching: &falseVal},
+			want: []string{"false"},
+		},
+		{
+			name: "SuspendDispatching is nil, defaults to false",
+			spec: &workv1alpha1.WorkSpec{},
+			want: []string{"false"},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got := GetWorkSuspendDispatching(tt.spec)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestSetLabelsAndAnnotationsForWorkload(t *testing.T) {
+	tests := []struct {
+		name              string
+		workload          *unstructured.Unstructured
+		work              *workv1alpha1.Work
+		wantLabelKey      string
+		wantLabelValue    string
+		wantAnnotationKey string
+	}{
+		{
+			name: "work has permanent ID label — propagated to workload",
+			workload: &unstructured.Unstructured{
+				Object: map[string]any{
+					"apiVersion": "apps/v1",
+					"kind":       "Deployment",
+					"metadata": map[string]any{
+						"name":      "demo",
+						"namespace": "default",
+					},
+				},
+			},
+			work: &workv1alpha1.Work{
+				Spec: workv1alpha1.WorkSpec{},
+			},
+		},
+		{
+			name: "work has permanent ID label set",
+			workload: &unstructured.Unstructured{
+				Object: map[string]any{
+					"apiVersion": "apps/v1",
+					"kind":       "Deployment",
+					"metadata": map[string]any{
+						"name":      "demo",
+						"namespace": "default",
+					},
+				},
+			},
+			work: &workv1alpha1.Work{
+				Spec: workv1alpha1.WorkSpec{},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			SetLabelsAndAnnotationsForWorkload(tt.workload, tt.work)
+			// RecordManagedAnnotations should always set the managed-annotations annotation.
+			annotations := tt.workload.GetAnnotations()
+			assert.NotNil(t, annotations)
+			assert.Contains(t, annotations, workv1alpha2.ManagedAnnotation)
+			// RecordManagedLabels should always set the managed-labels annotation.
+			assert.Contains(t, annotations, workv1alpha2.ManagedLabels)
+		})
+	}
+}
+
+func TestSetLabelsAndAnnotationsForWorkload_WithPermanentID(t *testing.T) {
+	const permanentID = "test-permanent-id-123"
+	workload := &unstructured.Unstructured{
+		Object: map[string]any{
+			"apiVersion": "apps/v1",
+			"kind":       "Deployment",
+			"metadata": map[string]any{
+				"name":      "demo",
+				"namespace": "default",
+			},
+		},
+	}
+	work := &workv1alpha1.Work{}
+	work.Labels = map[string]string{
+		workv1alpha2.WorkPermanentIDLabel: permanentID,
+	}
+
+	SetLabelsAndAnnotationsForWorkload(workload, work)
+
+	labels := workload.GetLabels()
+	assert.Equal(t, permanentID, labels[workv1alpha2.WorkPermanentIDLabel])
+}