From 3347c8d17adcb6cedc7c6ead290ebcb78a1e93db Mon Sep 17 00:00:00 2001
From: shaoting-huang <shaoting.huang@zilliz.com>
Date: Wed, 20 Nov 2024 18:22:12 +0800
Subject: [PATCH] fix compile

Signed-off-by: shaoting-huang <shaoting.huang@zilliz.com>
---
 client/client_mock_test.go | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/client/client_mock_test.go b/client/client_mock_test.go
index 6472b46f..37f3055a 100644
--- a/client/client_mock_test.go
+++ b/client/client_mock_test.go
@@ -257,6 +257,7 @@ const (
 	MGetLoadingProgress      ServiceMethod = 110
 	MGetLoadState            ServiceMethod = 111
 	MAlterCollectionField    ServiceMethod = 112
+	MOperatePrivilegeV2      ServiceMethod = 113
 
 	MCreatePartition   ServiceMethod = 201
 	MDropPartition     ServiceMethod = 202
@@ -501,6 +502,15 @@ func (m *MockServer) AlterCollection(ctx context.Context, req *milvuspb.AlterCol
 	return SuccessStatus()
 }
 
+func (m *MockServer) OperatePrivilegeV2(ctx context.Context, req *milvuspb.OperatePrivilegeV2Request) (*commonpb.Status, error) {
+	f := m.GetInjection(MOperatePrivilegeV2)
+	if f != nil {
+		r, err := f(ctx, req)
+		return r.(*commonpb.Status), err
+	}
+	return SuccessStatus()
+}
+
 func (m *MockServer) CreatePartition(ctx context.Context, req *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) {
 	f := m.GetInjection(MCreatePartition)
 	if f != nil {