test: add checker unittests (#28954)

issue: https://github.com/milvus-io/milvus/issues/28610

Signed-off-by: sunby <sunbingyi1992@gmail.com>

internal/distributed/querycoord/client/client_test.go

@@ -149,6 +149,15 @@ func Test_NewClient(t *testing.T) {
 
 		r27, err := client.DescribeResourceGroup(ctx, nil)
 		retCheck(retNotNil, r27, err)
+
+		r28, err := client.ListCheckers(ctx, nil)
+		retCheck(retNotNil, r28, err)
+
+		r29, err := client.ActivateChecker(ctx, nil)
+		retCheck(retNotNil, r29, err)
+
+		r30, err := client.DeactivateChecker(ctx, nil)
+		retCheck(retNotNil, r30, err)
 	}
 
 	client.grpcClient = &mock.GRPCClientBase[querypb.QueryCoordClient]{

internal/distributed/querycoord/service_test.go

@@ -293,6 +293,30 @@ func Test_NewServer(t *testing.T) {
 			assert.Equal(t, commonpb.ErrorCode_Success, resp.GetStatus().GetErrorCode())
 		})
 
+		t.Run("ListCheckers", func(t *testing.T) {
+			req := &querypb.ListCheckersRequest{}
+			mqc.EXPECT().ListCheckers(mock.Anything, req).Return(&querypb.ListCheckersResponse{Status: successStatus}, nil)
+			resp, err := server.ListCheckers(ctx, req)
+			assert.NoError(t, err)
+			assert.Equal(t, commonpb.ErrorCode_Success, resp.GetStatus().GetErrorCode())
+		})
+
+		t.Run("ActivateChecker", func(t *testing.T) {
+			req := &querypb.ActivateCheckerRequest{}
+			mqc.EXPECT().ActivateChecker(mock.Anything, req).Return(&commonpb.Status{ErrorCode: commonpb.ErrorCode_Success}, nil)
+			resp, err := server.ActivateChecker(ctx, req)
+			assert.NoError(t, err)
+			assert.Equal(t, commonpb.ErrorCode_Success, resp.ErrorCode)
+		})
+
+		t.Run("DeactivateChecker", func(t *testing.T) {
+			req := &querypb.DeactivateCheckerRequest{}
+			mqc.EXPECT().DeactivateChecker(mock.Anything, req).Return(&commonpb.Status{ErrorCode: commonpb.ErrorCode_Success}, nil)
+			resp, err := server.DeactivateChecker(ctx, req)
+			assert.NoError(t, err)
+			assert.Equal(t, commonpb.ErrorCode_Success, resp.ErrorCode)
+		})
+
 		err = server.Stop()
 		assert.NoError(t, err)
 	}

internal/querycoordv2/checkers/controller_base_test.go

@@ -99,6 +99,11 @@ func (s *ControllerBaseTestSuite) TestActivation() {
 	s.Equal(errTypeNotFound, err)
 }
 
+func (s *ControllerBaseTestSuite) TestListCheckers() {
+	checkers := s.controller.Checkers()
+	s.Equal(4, len(checkers))
+}
+
 func TestControllerBaseTestSuite(t *testing.T) {
 	suite.Run(t, new(ControllerBaseTestSuite))
 }