fix compile

Signed-off-by: shaoting-huang <shaoting.huang@zilliz.com>
shaoting-huang · Nov 20, 2024 · 3347c8d · 3347c8d
1 parent 3f6510a
commit 3347c8d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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 {