Add unit tests for FluxGroupFor

api/v1/common_types_test.go

@@ -7,6 +7,73 @@ import (
 	"testing"
 )
 
+func TestFluxGroupFor(t *testing.T) {
+	testCases := []struct {
+		kind          string
+		expectedGroup string
+	}{
+		// Flux Operator kinds
+		{FluxInstanceKind, GroupVersion.Group},
+		{FluxReportKind, GroupVersion.Group},
+		{ResourceSetKind, GroupVersion.Group},
+		{ResourceSetInputProviderKind, GroupVersion.Group},
+		// Source kinds
+		{FluxGitRepositoryKind, FluxSourceGroup},
+		{FluxOCIRepositoryKind, FluxSourceGroup},
+		{FluxBucketKind, FluxSourceGroup},
+		{FluxHelmChartKind, FluxSourceGroup},
+		{FluxHelmRepositoryKind, FluxSourceGroup},
+		{FluxExternalArtifactKind, FluxSourceGroup},
+		// Notification kinds
+		{FluxAlertKind, FluxNotificationGroup},
+		{FluxAlertProviderKind, FluxNotificationGroup},
+		{FluxReceiverKind, FluxNotificationGroup},
+		// Image kinds
+		{FluxImageRepositoryKind, FluxImageGroup},
+		{FluxImagePolicyKind, FluxImageGroup},
+		{FluxImageUpdateAutomationKind, FluxImageGroup},
+		// Kustomize kind
+		{FluxKustomizationKind, FluxKustomizeGroup},
+		// Helm kind
+		{FluxHelmReleaseKind, FluxHelmGroup},
+		// Source extensions kind
+		{FluxArtifactGeneratorKind, FluxSourceExtensionsGroup},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.kind, func(t *testing.T) {
+			gk, err := FluxGroupFor(tc.kind)
+			if err != nil {
+				t.Fatalf("unexpected error for kind %s: %v", tc.kind, err)
+			}
+			if gk.Group != tc.expectedGroup {
+				t.Errorf("expected group %s, got %s", tc.expectedGroup, gk.Group)
+			}
+			if gk.Kind != tc.kind {
+				t.Errorf("expected kind %s, got %s", tc.kind, gk.Kind)
+			}
+		})
+	}
+}
+
+func TestFluxGroupFor_UnknownKind(t *testing.T) {
+	unknownKinds := []string{
+		"Deployment",
+		"Service",
+		"UnknownKind",
+		"",
+	}
+
+	for _, kind := range unknownKinds {
+		t.Run(kind, func(t *testing.T) {
+			_, err := FluxGroupFor(kind)
+			if err == nil {
+				t.Error("expected error for unknown kind, got nil")
+			}
+		})
+	}
+}
+
 func TestFindFluxKindInfo_ExactMatch(t *testing.T) {
 	testCases := []struct {
 		input    string