admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-08 01:58:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: add checker unittests (#28954)

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

Signed-off-by: sunby <sunbingyi1992@gmail.com>
This commit is contained in:
Bingyi Sun 2023-12-05 10:56:33 +08:00 committed by GitHub
parent 91e92ee8d4
commit 10bb2431d8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 38 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
internal/distributed/querycoord/client/client_test.go
View File

@ -149,6 +149,15 @@ func Test_NewClient(t *testing.T) {
r27, err := client.DescribeResourceGroup(ctx, nil) r27, err := client.DescribeResourceGroup(ctx, nil)
retCheck(retNotNil, r27, err) 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]{ client.grpcClient = &mock.GRPCClientBase[querypb.QueryCoordClient]{

24
internal/distributed/querycoord/service_test.go
View File

@ -293,6 +293,30 @@ func Test_NewServer(t *testing.T) {
assert.Equal(t, commonpb.ErrorCode_Success, resp.GetStatus().GetErrorCode()) 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() err = server.Stop()
assert.NoError(t, err) assert.NoError(t, err)
} }

5
internal/querycoordv2/checkers/controller_base_test.go
View File

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