From 8a101ebcb7c798a440153a3243f4992d87bad5db Mon Sep 17 00:00:00 2001
From: changzhen <changzhen5@huawei.com>
Date: Wed, 28 Feb 2024 22:49:53 +0800
Subject: [PATCH] set enum value for operator in Remedy API

Signed-off-by: changzhen <changzhen5@huawei.com>
---
 .../karmada/_crds/bases/remedy/remedy.karmada.io_remedies.yaml | 3 +++
 pkg/apis/remedy/v1alpha1/remedy_types.go                       | 2 ++
 pkg/scheduler/framework/testing/mock_interface.go              | 3 +--
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/charts/karmada/_crds/bases/remedy/remedy.karmada.io_remedies.yaml b/charts/karmada/_crds/bases/remedy/remedy.karmada.io_remedies.yaml
index 2cd82265d..1da1e8d8a 100644
--- a/charts/karmada/_crds/bases/remedy/remedy.karmada.io_remedies.yaml
+++ b/charts/karmada/_crds/bases/remedy/remedy.karmada.io_remedies.yaml
@@ -79,6 +79,9 @@ spec:
                         operator:
                           description: Operator represents a conditionType's relationship
                             to a conditionStatus. Valid operators are Equal, NotEqual.
+                          enum:
+                          - Equal
+                          - NotEqual
                           type: string
                       required:
                       - conditionStatus
diff --git a/pkg/apis/remedy/v1alpha1/remedy_types.go b/pkg/apis/remedy/v1alpha1/remedy_types.go
index 870d0f4c1..1700744b1 100644
--- a/pkg/apis/remedy/v1alpha1/remedy_types.go
+++ b/pkg/apis/remedy/v1alpha1/remedy_types.go
@@ -70,6 +70,8 @@ type ClusterConditionRequirement struct {
 	ConditionType ConditionType `json:"conditionType"`
 	// Operator represents a conditionType's relationship to a conditionStatus.
 	// Valid operators are Equal, NotEqual.
+	//
+	// +kubebuilder:validation:Enum=Equal;NotEqual
 	// +required
 	Operator ClusterConditionOperator `json:"operator"`
 	// ConditionStatus specifies the ClusterStatue condition status.
diff --git a/pkg/scheduler/framework/testing/mock_interface.go b/pkg/scheduler/framework/testing/mock_interface.go
index f72952cf0..def6cb8cd 100644
--- a/pkg/scheduler/framework/testing/mock_interface.go
+++ b/pkg/scheduler/framework/testing/mock_interface.go
@@ -13,11 +13,10 @@ import (
 	context "context"
 	reflect "reflect"
 
-	gomock "go.uber.org/mock/gomock"
-
 	v1alpha1 "github.com/karmada-io/karmada/pkg/apis/cluster/v1alpha1"
 	v1alpha2 "github.com/karmada-io/karmada/pkg/apis/work/v1alpha2"
 	framework "github.com/karmada-io/karmada/pkg/scheduler/framework"
+	gomock "go.uber.org/mock/gomock"
 )
 
 // MockFramework is a mock of Framework interface.