From 6851738fd1522dd227ca1b76f3dfcf8e6ae497f9 Mon Sep 17 00:00:00 2001 From: Bingyi Sun Date: Thu, 17 Oct 2024 12:11:31 +0800 Subject: [PATCH] fix: fix `make generate-mockery` panic with go1.22 (#36830) https://github.com/milvus-io/milvus/issues/36831 Fix `make generate-mockery` panic. Signed-off-by: sunby --- Makefile | 2 +- .../datacoord/allocator/mock_allocator.go | 14 +- .../broker/mock_coordinator_broker.go | 22 +- internal/datacoord/mock_channel_store.go | 34 +- internal/datacoord/mock_channelmanager.go | 46 +- internal/datacoord/mock_cluster.go | 54 ++- internal/datacoord/mock_compaction_meta.go | 62 ++- .../datacoord/mock_compaction_plan_context.go | 18 +- internal/datacoord/mock_handler.go | 22 +- .../mock_index_engine_version_manager.go | 10 +- internal/datacoord/mock_job_manager.go | 14 +- internal/datacoord/mock_segment_manager.go | 22 +- internal/datacoord/mock_subcluster.go | 10 +- internal/datacoord/mock_trigger_manager.go | 6 +- .../session/mock_datanode_manager.go | 74 ++- .../datacoord/session/mock_worker_manager.go | 22 +- internal/kv/mocks/meta_kv.go | 62 ++- internal/kv/mocks/snapshot_kv.go | 26 +- internal/kv/mocks/txn_kv.go | 50 +- internal/kv/mocks/watch_kv.go | 74 ++- .../mock_streaming/mock_WALAccesser.go | 22 +- .../grpc/mock_balancer/mock_SubConn.go | 6 +- .../grpc/mock_resolver/mock_ClientConn.go | 10 +- .../mock_grpc/mock_ClientStream.go | 26 +- internal/mocks/mock_datacoord.go | 214 ++++++++- internal/mocks/mock_datacoord_client.go | 202 +++++++- internal/mocks/mock_datanode.go | 118 ++++- internal/mocks/mock_datanode_client.go | 86 +++- internal/mocks/mock_indexnode.go | 66 ++- internal/mocks/mock_indexnode_client.go | 50 +- .../mock_StreamingCoordCataLog.go | 10 +- .../mock_StreamingNodeCataLog.go | 10 +- internal/mocks/mock_proxy.go | 430 +++++++++++++++++- internal/mocks/mock_proxy_client.go | 62 ++- internal/mocks/mock_querycoord.go | 178 +++++++- internal/mocks/mock_querycoord_client.go | 158 ++++++- internal/mocks/mock_querynode.go | 122 ++++- internal/mocks/mock_querynode_client.go | 102 ++++- internal/mocks/mock_rootcoord.go | 226 ++++++++- internal/mocks/mock_rootcoord_client.go | 206 ++++++++- .../streamingcoord/mock_client/mock_Client.go | 6 +- .../server/mock_balancer/mock_Balancer.go | 14 +- .../handler/mock_assignment/mock_Watcher.go | 10 +- .../handler/mock_consumer/mock_Consumer.go | 10 +- .../handler/mock_producer/mock_Producer.go | 18 +- .../client/mock_handler/mock_HandlerClient.go | 10 +- .../client/mock_manager/mock_ManagerClient.go | 18 +- .../server/mock_flusher/mock_Flusher.go | 6 +- .../server/mock_wal/mock_Opener.go | 6 +- .../server/mock_wal/mock_OpenerBuilder.go | 10 +- .../server/mock_wal/mock_Scanner.go | 22 +- .../streamingnode/server/mock_wal/mock_WAL.go | 26 +- .../server/mock_walmanager/mock_Manager.go | 18 +- .../mock_inspector/mock_SealOperator.go | 10 +- .../mock_TimeTickSyncOperator.go | 10 +- .../wal/mock_interceptors/mock_Interceptor.go | 6 +- .../mock_InterceptorBuilder.go | 6 +- .../mock_InterceptorWithReady.go | 10 +- .../mock_discoverer/mock_Discoverer.go | 10 +- .../service/mock_lazygrpc/mock_Service.go | 10 +- .../service/mock_resolver/mock_Builder.go | 14 +- .../service/mock_resolver/mock_Resolver.go | 10 +- internal/proxy/mock_cache.go | 70 ++- internal/proxy/mock_channels_manager.go | 14 +- internal/proxy/mock_lb_balancer.go | 6 +- internal/proxy/mock_lb_policy.go | 10 +- internal/proxy/mock_shardclient_manager.go | 10 +- internal/proxy/mock_tso_test.go | 6 +- .../querycoordv2/balance/mock_balancer.go | 14 +- internal/querycoordv2/dist/mock_controller.go | 2 +- internal/querycoordv2/meta/mock_broker.go | 38 +- .../querycoordv2/meta/mock_target_manager.go | 74 ++- internal/querycoordv2/mocks/mock_querynode.go | 98 +++- internal/querycoordv2/session/mock_cluster.go | 42 +- internal/querycoordv2/task/mock_scheduler.go | 26 +- internal/querynodev2/cluster/mock_manager.go | 6 +- internal/querynodev2/cluster/mock_worker.go | 34 +- .../querynodev2/delegator/mock_delegator.go | 62 ++- .../querynodev2/optimizers/mock_query_hook.go | 18 +- .../segments/mock_collection_manager.go | 18 +- internal/querynodev2/segments/mock_loader.go | 26 +- internal/querynodev2/segments/mock_segment.go | 142 +++++- .../segments/mock_segment_manager.go | 50 +- internal/rootcoord/mocks/garbage_collector.go | 10 +- internal/rootcoord/mocks/meta_table.go | 186 +++++++- pkg/eventlog/mock_logger.go | 6 +- pkg/kv/predicates/mock_predicate.go | 22 +- pkg/mocks/mock_kv/mock_MetaKv.go | 62 ++- .../mock_walimpls/mock_OpenerBuilderImpls.go | 10 +- .../mock_walimpls/mock_OpenerImpls.go | 6 +- .../mock_walimpls/mock_ScannerImpls.go | 22 +- .../streaming/mock_walimpls/mock_WALImpls.go | 18 +- ...k_StreamingCoordAssignmentServiceClient.go | 6 +- ...ignmentService_AssignmentDiscoverClient.go | 34 +- ...ignmentService_AssignmentDiscoverServer.go | 30 +- .../mock_StreamingNodeHandlerServiceClient.go | 10 +- ...reamingNodeHandlerService_ConsumeClient.go | 34 +- ...reamingNodeHandlerService_ConsumeServer.go | 30 +- ...reamingNodeHandlerService_ProduceClient.go | 34 +- ...reamingNodeHandlerService_ProduceServer.go | 30 +- .../mock_StreamingNodeManagerServiceClient.go | 14 +- .../mock_message/mock_ImmutableMessage.go | 50 +- .../mock_message/mock_ImmutableTxnMessage.go | 66 ++- .../util/mock_message/mock_MessageID.go | 26 +- .../util/mock_message/mock_MutableMessage.go | 66 ++- .../util/mock_message/mock_RProperties.go | 14 +- .../mock_AssignmentDiscoverWatcher.go | 10 +- .../mock_AssignmentRebalanceTrigger.go | 6 +- pkg/mq/common/mock_id.go | 18 +- pkg/mq/msgdispatcher/mock_client.go | 6 +- pkg/mq/msgstream/mock_msgstream.go | 34 +- pkg/mq/msgstream/mock_msgstream_factory.go | 14 +- 112 files changed, 4736 insertions(+), 112 deletions(-) diff --git a/Makefile b/Makefile index c8a87670a9..cb5a7001f3 100644 --- a/Makefile +++ b/Makefile @@ -49,7 +49,7 @@ GOLANGCI_LINT_VERSION := 1.55.2 GOLANGCI_LINT_OUTPUT := $(shell $(INSTALL_PATH)/golangci-lint --version 2>/dev/null) INSTALL_GOLANGCI_LINT := $(findstring $(GOLANGCI_LINT_VERSION), $(GOLANGCI_LINT_OUTPUT)) # mockery -MOCKERY_VERSION := 2.32.4 +MOCKERY_VERSION := 2.46.0 MOCKERY_OUTPUT := $(shell $(INSTALL_PATH)/mockery --version 2>/dev/null) INSTALL_MOCKERY := $(findstring $(MOCKERY_VERSION),$(MOCKERY_OUTPUT)) # gci diff --git a/internal/datacoord/allocator/mock_allocator.go b/internal/datacoord/allocator/mock_allocator.go index 0f5fd4bf61..ab76ebc76e 100644 --- a/internal/datacoord/allocator/mock_allocator.go +++ b/internal/datacoord/allocator/mock_allocator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package allocator @@ -25,6 +25,10 @@ func (_m *MockAllocator) EXPECT() *MockAllocator_Expecter { func (_m *MockAllocator) AllocID(_a0 context.Context) (int64, error) { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for AllocID") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context) (int64, error)); ok { @@ -77,6 +81,10 @@ func (_c *MockAllocator_AllocID_Call) RunAndReturn(run func(context.Context) (in func (_m *MockAllocator) AllocN(n int64) (int64, int64, error) { ret := _m.Called(n) + if len(ret) == 0 { + panic("no return value specified for AllocN") + } + var r0 int64 var r1 int64 var r2 error @@ -136,6 +144,10 @@ func (_c *MockAllocator_AllocN_Call) RunAndReturn(run func(int64) (int64, int64, func (_m *MockAllocator) AllocTimestamp(_a0 context.Context) (uint64, error) { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for AllocTimestamp") + } + var r0 uint64 var r1 error if rf, ok := ret.Get(0).(func(context.Context) (uint64, error)); ok { diff --git a/internal/datacoord/broker/mock_coordinator_broker.go b/internal/datacoord/broker/mock_coordinator_broker.go index c952eba15b..bdec77b28c 100644 --- a/internal/datacoord/broker/mock_coordinator_broker.go +++ b/internal/datacoord/broker/mock_coordinator_broker.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package broker @@ -26,6 +26,10 @@ func (_m *MockBroker) EXPECT() *MockBroker_Expecter { func (_m *MockBroker) DescribeCollectionInternal(ctx context.Context, collectionID int64) (*milvuspb.DescribeCollectionResponse, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for DescribeCollectionInternal") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -81,6 +85,10 @@ func (_c *MockBroker_DescribeCollectionInternal_Call) RunAndReturn(run func(cont func (_m *MockBroker) HasCollection(ctx context.Context, collectionID int64) (bool, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for HasCollection") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (bool, error)); ok { @@ -134,6 +142,10 @@ func (_c *MockBroker_HasCollection_Call) RunAndReturn(run func(context.Context, func (_m *MockBroker) ListDatabases(ctx context.Context) (*milvuspb.ListDatabasesResponse, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for ListDatabases") + } + var r0 *milvuspb.ListDatabasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context) (*milvuspb.ListDatabasesResponse, error)); ok { @@ -188,6 +200,10 @@ func (_c *MockBroker_ListDatabases_Call) RunAndReturn(run func(context.Context) func (_m *MockBroker) ShowCollections(ctx context.Context, dbName string) (*milvuspb.ShowCollectionsResponse, error) { ret := _m.Called(ctx, dbName) + if len(ret) == 0 { + panic("no return value specified for ShowCollections") + } + var r0 *milvuspb.ShowCollectionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, string) (*milvuspb.ShowCollectionsResponse, error)); ok { @@ -243,6 +259,10 @@ func (_c *MockBroker_ShowCollections_Call) RunAndReturn(run func(context.Context func (_m *MockBroker) ShowPartitionsInternal(ctx context.Context, collectionID int64) ([]int64, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for ShowPartitionsInternal") + } + var r0 []int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) ([]int64, error)); ok { diff --git a/internal/datacoord/mock_channel_store.go b/internal/datacoord/mock_channel_store.go index 01464ae7d2..f14691c8ab 100644 --- a/internal/datacoord/mock_channel_store.go +++ b/internal/datacoord/mock_channel_store.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -54,6 +54,10 @@ func (_c *MockRWChannelStore_AddNode_Call) RunAndReturn(run func(int64)) *MockRW func (_m *MockRWChannelStore) GetBufferChannelInfo() *NodeChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetBufferChannelInfo") + } + var r0 *NodeChannelInfo if rf, ok := ret.Get(0).(func() *NodeChannelInfo); ok { r0 = rf() @@ -97,6 +101,10 @@ func (_c *MockRWChannelStore_GetBufferChannelInfo_Call) RunAndReturn(run func() func (_m *MockRWChannelStore) GetNode(nodeID int64) *NodeChannelInfo { ret := _m.Called(nodeID) + if len(ret) == 0 { + panic("no return value specified for GetNode") + } + var r0 *NodeChannelInfo if rf, ok := ret.Get(0).(func(int64) *NodeChannelInfo); ok { r0 = rf(nodeID) @@ -148,6 +156,10 @@ func (_m *MockRWChannelStore) GetNodeChannelsBy(nodeSelector NodeSelector, chann _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetNodeChannelsBy") + } + var r0 []*NodeChannelInfo if rf, ok := ret.Get(0).(func(NodeSelector, ...ChannelSelector) []*NodeChannelInfo); ok { r0 = rf(nodeSelector, channelSelectors...) @@ -200,6 +212,10 @@ func (_c *MockRWChannelStore_GetNodeChannelsBy_Call) RunAndReturn(run func(NodeS func (_m *MockRWChannelStore) GetNodeChannelsByCollectionID(collectionID int64) map[int64][]string { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for GetNodeChannelsByCollectionID") + } + var r0 map[int64][]string if rf, ok := ret.Get(0).(func(int64) map[int64][]string); ok { r0 = rf(collectionID) @@ -244,6 +260,10 @@ func (_c *MockRWChannelStore_GetNodeChannelsByCollectionID_Call) RunAndReturn(ru func (_m *MockRWChannelStore) GetNodes() []int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetNodes") + } + var r0 []int64 if rf, ok := ret.Get(0).(func() []int64); ok { r0 = rf() @@ -287,6 +307,10 @@ func (_c *MockRWChannelStore_GetNodes_Call) RunAndReturn(run func() []int64) *Mo func (_m *MockRWChannelStore) GetNodesChannels() []*NodeChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetNodesChannels") + } + var r0 []*NodeChannelInfo if rf, ok := ret.Get(0).(func() []*NodeChannelInfo); ok { r0 = rf() @@ -330,6 +354,10 @@ func (_c *MockRWChannelStore_GetNodesChannels_Call) RunAndReturn(run func() []*N func (_m *MockRWChannelStore) Reload() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Reload") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -450,6 +478,10 @@ func (_c *MockRWChannelStore_SetLegacyChannelByNode_Call) RunAndReturn(run func( func (_m *MockRWChannelStore) Update(op *ChannelOpSet) error { ret := _m.Called(op) + if len(ret) == 0 { + panic("no return value specified for Update") + } + var r0 error if rf, ok := ret.Get(0).(func(*ChannelOpSet) error); ok { r0 = rf(op) diff --git a/internal/datacoord/mock_channelmanager.go b/internal/datacoord/mock_channelmanager.go index 5239ab6e91..9a61852db8 100644 --- a/internal/datacoord/mock_channelmanager.go +++ b/internal/datacoord/mock_channelmanager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -25,6 +25,10 @@ func (_m *MockChannelManager) EXPECT() *MockChannelManager_Expecter { func (_m *MockChannelManager) AddNode(nodeID int64) error { ret := _m.Called(nodeID) + if len(ret) == 0 { + panic("no return value specified for AddNode") + } + var r0 error if rf, ok := ret.Get(0).(func(int64) error); ok { r0 = rf(nodeID) @@ -99,6 +103,10 @@ func (_c *MockChannelManager_Close_Call) RunAndReturn(run func()) *MockChannelMa func (_m *MockChannelManager) DeleteNode(nodeID int64) error { ret := _m.Called(nodeID) + if len(ret) == 0 { + panic("no return value specified for DeleteNode") + } + var r0 error if rf, ok := ret.Get(0).(func(int64) error); ok { r0 = rf(nodeID) @@ -141,6 +149,10 @@ func (_c *MockChannelManager_DeleteNode_Call) RunAndReturn(run func(int64) error func (_m *MockChannelManager) FindWatcher(channel string) (int64, error) { ret := _m.Called(channel) + if len(ret) == 0 { + panic("no return value specified for FindWatcher") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(string) (int64, error)); ok { @@ -193,6 +205,10 @@ func (_c *MockChannelManager_FindWatcher_Call) RunAndReturn(run func(string) (in func (_m *MockChannelManager) GetChannel(nodeID int64, channel string) (RWChannel, bool) { ret := _m.Called(nodeID, channel) + if len(ret) == 0 { + panic("no return value specified for GetChannel") + } + var r0 RWChannel var r1 bool if rf, ok := ret.Get(0).(func(int64, string) (RWChannel, bool)); ok { @@ -248,6 +264,10 @@ func (_c *MockChannelManager_GetChannel_Call) RunAndReturn(run func(int64, strin func (_m *MockChannelManager) GetChannelNamesByCollectionID(collectionID int64) []string { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for GetChannelNamesByCollectionID") + } + var r0 []string if rf, ok := ret.Get(0).(func(int64) []string); ok { r0 = rf(collectionID) @@ -292,6 +312,10 @@ func (_c *MockChannelManager_GetChannelNamesByCollectionID_Call) RunAndReturn(ru func (_m *MockChannelManager) GetChannelsByCollectionID(collectionID int64) []RWChannel { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for GetChannelsByCollectionID") + } + var r0 []RWChannel if rf, ok := ret.Get(0).(func(int64) []RWChannel); ok { r0 = rf(collectionID) @@ -336,6 +360,10 @@ func (_c *MockChannelManager_GetChannelsByCollectionID_Call) RunAndReturn(run fu func (_m *MockChannelManager) GetNodeChannelsByCollectionID(collectionID int64) map[int64][]string { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for GetNodeChannelsByCollectionID") + } + var r0 map[int64][]string if rf, ok := ret.Get(0).(func(int64) map[int64][]string); ok { r0 = rf(collectionID) @@ -380,6 +408,10 @@ func (_c *MockChannelManager_GetNodeChannelsByCollectionID_Call) RunAndReturn(ru func (_m *MockChannelManager) Match(nodeID int64, channel string) bool { ret := _m.Called(nodeID, channel) + if len(ret) == 0 { + panic("no return value specified for Match") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, string) bool); ok { r0 = rf(nodeID, channel) @@ -423,6 +455,10 @@ func (_c *MockChannelManager_Match_Call) RunAndReturn(run func(int64, string) bo func (_m *MockChannelManager) Release(nodeID int64, channelName string) error { ret := _m.Called(nodeID, channelName) + if len(ret) == 0 { + panic("no return value specified for Release") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, string) error); ok { r0 = rf(nodeID, channelName) @@ -466,6 +502,10 @@ func (_c *MockChannelManager_Release_Call) RunAndReturn(run func(int64, string) func (_m *MockChannelManager) Startup(ctx context.Context, legacyNodes []int64, allNodes []int64) error { ret := _m.Called(ctx, legacyNodes, allNodes) + if len(ret) == 0 { + panic("no return value specified for Startup") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []int64, []int64) error); ok { r0 = rf(ctx, legacyNodes, allNodes) @@ -510,6 +550,10 @@ func (_c *MockChannelManager_Startup_Call) RunAndReturn(run func(context.Context func (_m *MockChannelManager) Watch(ctx context.Context, ch RWChannel) error { ret := _m.Called(ctx, ch) + if len(ret) == 0 { + panic("no return value specified for Watch") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, RWChannel) error); ok { r0 = rf(ctx, ch) diff --git a/internal/datacoord/mock_cluster.go b/internal/datacoord/mock_cluster.go index 982ee03d76..62c60249c5 100644 --- a/internal/datacoord/mock_cluster.go +++ b/internal/datacoord/mock_cluster.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -60,6 +60,10 @@ func (_c *MockCluster_Close_Call) RunAndReturn(run func()) *MockCluster_Close_Ca func (_m *MockCluster) DropImport(nodeID int64, in *datapb.DropImportRequest) error { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for DropImport") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.DropImportRequest) error); ok { r0 = rf(nodeID, in) @@ -103,6 +107,10 @@ func (_c *MockCluster_DropImport_Call) RunAndReturn(run func(int64, *datapb.Drop func (_m *MockCluster) Flush(ctx context.Context, nodeID int64, channel string, segments []*datapb.SegmentInfo) error { ret := _m.Called(ctx, nodeID, channel, segments) + if len(ret) == 0 { + panic("no return value specified for Flush") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, string, []*datapb.SegmentInfo) error); ok { r0 = rf(ctx, nodeID, channel, segments) @@ -148,6 +156,10 @@ func (_c *MockCluster_Flush_Call) RunAndReturn(run func(context.Context, int64, func (_m *MockCluster) FlushChannels(ctx context.Context, nodeID int64, flushTs uint64, channels []string) error { ret := _m.Called(ctx, nodeID, flushTs, channels) + if len(ret) == 0 { + panic("no return value specified for FlushChannels") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, uint64, []string) error); ok { r0 = rf(ctx, nodeID, flushTs, channels) @@ -193,6 +205,10 @@ func (_c *MockCluster_FlushChannels_Call) RunAndReturn(run func(context.Context, func (_m *MockCluster) GetSessions() []*session.Session { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetSessions") + } + var r0 []*session.Session if rf, ok := ret.Get(0).(func() []*session.Session); ok { r0 = rf() @@ -236,6 +252,10 @@ func (_c *MockCluster_GetSessions_Call) RunAndReturn(run func() []*session.Sessi func (_m *MockCluster) ImportV2(nodeID int64, in *datapb.ImportRequest) error { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.ImportRequest) error); ok { r0 = rf(nodeID, in) @@ -279,6 +299,10 @@ func (_c *MockCluster_ImportV2_Call) RunAndReturn(run func(int64, *datapb.Import func (_m *MockCluster) PreImport(nodeID int64, in *datapb.PreImportRequest) error { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for PreImport") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.PreImportRequest) error); ok { r0 = rf(nodeID, in) @@ -322,6 +346,10 @@ func (_c *MockCluster_PreImport_Call) RunAndReturn(run func(int64, *datapb.PreIm func (_m *MockCluster) QueryImport(nodeID int64, in *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error) { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for QueryImport") + } + var r0 *datapb.QueryImportResponse var r1 error if rf, ok := ret.Get(0).(func(int64, *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error)); ok { @@ -377,6 +405,10 @@ func (_c *MockCluster_QueryImport_Call) RunAndReturn(run func(int64, *datapb.Que func (_m *MockCluster) QueryPreImport(nodeID int64, in *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error) { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for QueryPreImport") + } + var r0 *datapb.QueryPreImportResponse var r1 error if rf, ok := ret.Get(0).(func(int64, *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error)); ok { @@ -432,6 +464,10 @@ func (_c *MockCluster_QueryPreImport_Call) RunAndReturn(run func(int64, *datapb. func (_m *MockCluster) QuerySlots() map[int64]int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for QuerySlots") + } + var r0 map[int64]int64 if rf, ok := ret.Get(0).(func() map[int64]int64); ok { r0 = rf() @@ -475,6 +511,10 @@ func (_c *MockCluster_QuerySlots_Call) RunAndReturn(run func() map[int64]int64) func (_m *MockCluster) Register(node *session.NodeInfo) error { ret := _m.Called(node) + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func(*session.NodeInfo) error); ok { r0 = rf(node) @@ -517,6 +557,10 @@ func (_c *MockCluster_Register_Call) RunAndReturn(run func(*session.NodeInfo) er func (_m *MockCluster) Startup(ctx context.Context, nodes []*session.NodeInfo) error { ret := _m.Called(ctx, nodes) + if len(ret) == 0 { + panic("no return value specified for Startup") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []*session.NodeInfo) error); ok { r0 = rf(ctx, nodes) @@ -560,6 +604,10 @@ func (_c *MockCluster_Startup_Call) RunAndReturn(run func(context.Context, []*se func (_m *MockCluster) UnRegister(node *session.NodeInfo) error { ret := _m.Called(node) + if len(ret) == 0 { + panic("no return value specified for UnRegister") + } + var r0 error if rf, ok := ret.Get(0).(func(*session.NodeInfo) error); ok { r0 = rf(node) @@ -602,6 +650,10 @@ func (_c *MockCluster_UnRegister_Call) RunAndReturn(run func(*session.NodeInfo) func (_m *MockCluster) Watch(ctx context.Context, ch RWChannel) error { ret := _m.Called(ctx, ch) + if len(ret) == 0 { + panic("no return value specified for Watch") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, RWChannel) error); ok { r0 = rf(ctx, ch) diff --git a/internal/datacoord/mock_compaction_meta.go b/internal/datacoord/mock_compaction_meta.go index ec90d4b216..80a24bf191 100644 --- a/internal/datacoord/mock_compaction_meta.go +++ b/internal/datacoord/mock_compaction_meta.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -24,6 +24,10 @@ func (_m *MockCompactionMeta) EXPECT() *MockCompactionMeta_Expecter { func (_m *MockCompactionMeta) CheckAndSetSegmentsCompacting(segmentIDs []int64) (bool, bool) { ret := _m.Called(segmentIDs) + if len(ret) == 0 { + panic("no return value specified for CheckAndSetSegmentsCompacting") + } + var r0 bool var r1 bool if rf, ok := ret.Get(0).(func([]int64) (bool, bool)); ok { @@ -76,6 +80,10 @@ func (_c *MockCompactionMeta_CheckAndSetSegmentsCompacting_Call) RunAndReturn(ru func (_m *MockCompactionMeta) CleanPartitionStatsInfo(info *datapb.PartitionStatsInfo) error { ret := _m.Called(info) + if len(ret) == 0 { + panic("no return value specified for CleanPartitionStatsInfo") + } + var r0 error if rf, ok := ret.Get(0).(func(*datapb.PartitionStatsInfo) error); ok { r0 = rf(info) @@ -118,6 +126,10 @@ func (_c *MockCompactionMeta_CleanPartitionStatsInfo_Call) RunAndReturn(run func func (_m *MockCompactionMeta) CompleteCompactionMutation(t *datapb.CompactionTask, result *datapb.CompactionPlanResult) ([]*SegmentInfo, *segMetricMutation, error) { ret := _m.Called(t, result) + if len(ret) == 0 { + panic("no return value specified for CompleteCompactionMutation") + } + var r0 []*SegmentInfo var r1 *segMetricMutation var r2 error @@ -182,6 +194,10 @@ func (_c *MockCompactionMeta_CompleteCompactionMutation_Call) RunAndReturn(run f func (_m *MockCompactionMeta) DropCompactionTask(task *datapb.CompactionTask) error { ret := _m.Called(task) + if len(ret) == 0 { + panic("no return value specified for DropCompactionTask") + } + var r0 error if rf, ok := ret.Get(0).(func(*datapb.CompactionTask) error); ok { r0 = rf(task) @@ -224,6 +240,10 @@ func (_c *MockCompactionMeta_DropCompactionTask_Call) RunAndReturn(run func(*dat func (_m *MockCompactionMeta) GetAnalyzeMeta() *analyzeMeta { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAnalyzeMeta") + } + var r0 *analyzeMeta if rf, ok := ret.Get(0).(func() *analyzeMeta); ok { r0 = rf() @@ -267,6 +287,10 @@ func (_c *MockCompactionMeta_GetAnalyzeMeta_Call) RunAndReturn(run func() *analy func (_m *MockCompactionMeta) GetCompactionTaskMeta() *compactionTaskMeta { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetCompactionTaskMeta") + } + var r0 *compactionTaskMeta if rf, ok := ret.Get(0).(func() *compactionTaskMeta); ok { r0 = rf() @@ -310,6 +334,10 @@ func (_c *MockCompactionMeta_GetCompactionTaskMeta_Call) RunAndReturn(run func() func (_m *MockCompactionMeta) GetCompactionTasks() map[int64][]*datapb.CompactionTask { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetCompactionTasks") + } + var r0 map[int64][]*datapb.CompactionTask if rf, ok := ret.Get(0).(func() map[int64][]*datapb.CompactionTask); ok { r0 = rf() @@ -353,6 +381,10 @@ func (_c *MockCompactionMeta_GetCompactionTasks_Call) RunAndReturn(run func() ma func (_m *MockCompactionMeta) GetCompactionTasksByTriggerID(triggerID int64) []*datapb.CompactionTask { ret := _m.Called(triggerID) + if len(ret) == 0 { + panic("no return value specified for GetCompactionTasksByTriggerID") + } + var r0 []*datapb.CompactionTask if rf, ok := ret.Get(0).(func(int64) []*datapb.CompactionTask); ok { r0 = rf(triggerID) @@ -397,6 +429,10 @@ func (_c *MockCompactionMeta_GetCompactionTasksByTriggerID_Call) RunAndReturn(ru func (_m *MockCompactionMeta) GetHealthySegment(segID int64) *SegmentInfo { ret := _m.Called(segID) + if len(ret) == 0 { + panic("no return value specified for GetHealthySegment") + } + var r0 *SegmentInfo if rf, ok := ret.Get(0).(func(int64) *SegmentInfo); ok { r0 = rf(segID) @@ -441,6 +477,10 @@ func (_c *MockCompactionMeta_GetHealthySegment_Call) RunAndReturn(run func(int64 func (_m *MockCompactionMeta) GetIndexMeta() *indexMeta { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetIndexMeta") + } + var r0 *indexMeta if rf, ok := ret.Get(0).(func() *indexMeta); ok { r0 = rf() @@ -484,6 +524,10 @@ func (_c *MockCompactionMeta_GetIndexMeta_Call) RunAndReturn(run func() *indexMe func (_m *MockCompactionMeta) GetPartitionStatsMeta() *partitionStatsMeta { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetPartitionStatsMeta") + } + var r0 *partitionStatsMeta if rf, ok := ret.Get(0).(func() *partitionStatsMeta); ok { r0 = rf() @@ -527,6 +571,10 @@ func (_c *MockCompactionMeta_GetPartitionStatsMeta_Call) RunAndReturn(run func() func (_m *MockCompactionMeta) GetSegment(segID int64) *SegmentInfo { ret := _m.Called(segID) + if len(ret) == 0 { + panic("no return value specified for GetSegment") + } + var r0 *SegmentInfo if rf, ok := ret.Get(0).(func(int64) *SegmentInfo); ok { r0 = rf(segID) @@ -571,6 +619,10 @@ func (_c *MockCompactionMeta_GetSegment_Call) RunAndReturn(run func(int64) *Segm func (_m *MockCompactionMeta) SaveCompactionTask(task *datapb.CompactionTask) error { ret := _m.Called(task) + if len(ret) == 0 { + panic("no return value specified for SaveCompactionTask") + } + var r0 error if rf, ok := ret.Get(0).(func(*datapb.CompactionTask) error); ok { r0 = rf(task) @@ -619,6 +671,10 @@ func (_m *MockCompactionMeta) SelectSegments(filters ...SegmentFilter) []*Segmen _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SelectSegments") + } + var r0 []*SegmentInfo if rf, ok := ret.Get(0).(func(...SegmentFilter) []*SegmentInfo); ok { r0 = rf(filters...) @@ -710,6 +766,10 @@ func (_m *MockCompactionMeta) UpdateSegmentsInfo(operators ...UpdateOperator) er _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateSegmentsInfo") + } + var r0 error if rf, ok := ret.Get(0).(func(...UpdateOperator) error); ok { r0 = rf(operators...) diff --git a/internal/datacoord/mock_compaction_plan_context.go b/internal/datacoord/mock_compaction_plan_context.go index 2eed7e7664..845c62795d 100644 --- a/internal/datacoord/mock_compaction_plan_context.go +++ b/internal/datacoord/mock_compaction_plan_context.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -24,6 +24,10 @@ func (_m *MockCompactionPlanContext) EXPECT() *MockCompactionPlanContext_Expecte func (_m *MockCompactionPlanContext) enqueueCompaction(task *datapb.CompactionTask) error { ret := _m.Called(task) + if len(ret) == 0 { + panic("no return value specified for enqueueCompaction") + } + var r0 error if rf, ok := ret.Get(0).(func(*datapb.CompactionTask) error); ok { r0 = rf(task) @@ -66,6 +70,10 @@ func (_c *MockCompactionPlanContext_enqueueCompaction_Call) RunAndReturn(run fun func (_m *MockCompactionPlanContext) getCompactionInfo(signalID int64) *compactionInfo { ret := _m.Called(signalID) + if len(ret) == 0 { + panic("no return value specified for getCompactionInfo") + } + var r0 *compactionInfo if rf, ok := ret.Get(0).(func(int64) *compactionInfo); ok { r0 = rf(signalID) @@ -110,6 +118,10 @@ func (_c *MockCompactionPlanContext_getCompactionInfo_Call) RunAndReturn(run fun func (_m *MockCompactionPlanContext) getCompactionTasksNumBySignalID(signalID int64) int { ret := _m.Called(signalID) + if len(ret) == 0 { + panic("no return value specified for getCompactionTasksNumBySignalID") + } + var r0 int if rf, ok := ret.Get(0).(func(int64) int); ok { r0 = rf(signalID) @@ -152,6 +164,10 @@ func (_c *MockCompactionPlanContext_getCompactionTasksNumBySignalID_Call) RunAnd func (_m *MockCompactionPlanContext) isFull() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for isFull") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() diff --git a/internal/datacoord/mock_handler.go b/internal/datacoord/mock_handler.go index 3e02c25c8d..20f84a8f53 100644 --- a/internal/datacoord/mock_handler.go +++ b/internal/datacoord/mock_handler.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -26,6 +26,10 @@ func (_m *NMockHandler) EXPECT() *NMockHandler_Expecter { func (_m *NMockHandler) CheckShouldDropChannel(ch string) bool { ret := _m.Called(ch) + if len(ret) == 0 { + panic("no return value specified for CheckShouldDropChannel") + } + var r0 bool if rf, ok := ret.Get(0).(func(string) bool); ok { r0 = rf(ch) @@ -68,6 +72,10 @@ func (_c *NMockHandler_CheckShouldDropChannel_Call) RunAndReturn(run func(string func (_m *NMockHandler) FinishDropChannel(ch string, collectionID int64) error { ret := _m.Called(ch, collectionID) + if len(ret) == 0 { + panic("no return value specified for FinishDropChannel") + } + var r0 error if rf, ok := ret.Get(0).(func(string, int64) error); ok { r0 = rf(ch, collectionID) @@ -111,6 +119,10 @@ func (_c *NMockHandler_FinishDropChannel_Call) RunAndReturn(run func(string, int func (_m *NMockHandler) GetCollection(ctx context.Context, collectionID int64) (*collectionInfo, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetCollection") + } + var r0 *collectionInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (*collectionInfo, error)); ok { @@ -166,6 +178,10 @@ func (_c *NMockHandler_GetCollection_Call) RunAndReturn(run func(context.Context func (_m *NMockHandler) GetDataVChanPositions(ch RWChannel, partitionID int64) *datapb.VchannelInfo { ret := _m.Called(ch, partitionID) + if len(ret) == 0 { + panic("no return value specified for GetDataVChanPositions") + } + var r0 *datapb.VchannelInfo if rf, ok := ret.Get(0).(func(RWChannel, int64) *datapb.VchannelInfo); ok { r0 = rf(ch, partitionID) @@ -218,6 +234,10 @@ func (_m *NMockHandler) GetQueryVChanPositions(ch RWChannel, partitionIDs ...int _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetQueryVChanPositions") + } + var r0 *datapb.VchannelInfo if rf, ok := ret.Get(0).(func(RWChannel, ...int64) *datapb.VchannelInfo); ok { r0 = rf(ch, partitionIDs...) diff --git a/internal/datacoord/mock_index_engine_version_manager.go b/internal/datacoord/mock_index_engine_version_manager.go index c1da3ed7c4..7b686276cb 100644 --- a/internal/datacoord/mock_index_engine_version_manager.go +++ b/internal/datacoord/mock_index_engine_version_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -57,6 +57,10 @@ func (_c *MockVersionManager_AddNode_Call) RunAndReturn(run func(*sessionutil.Se func (_m *MockVersionManager) GetCurrentIndexEngineVersion() int32 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetCurrentIndexEngineVersion") + } + var r0 int32 if rf, ok := ret.Get(0).(func() int32); ok { r0 = rf() @@ -98,6 +102,10 @@ func (_c *MockVersionManager_GetCurrentIndexEngineVersion_Call) RunAndReturn(run func (_m *MockVersionManager) GetMinimalIndexEngineVersion() int32 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetMinimalIndexEngineVersion") + } + var r0 int32 if rf, ok := ret.Get(0).(func() int32); ok { r0 = rf() diff --git a/internal/datacoord/mock_job_manager.go b/internal/datacoord/mock_job_manager.go index de47574859..c0361da48c 100644 --- a/internal/datacoord/mock_job_manager.go +++ b/internal/datacoord/mock_job_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -24,6 +24,10 @@ func (_m *MockStatsJobManager) EXPECT() *MockStatsJobManager_Expecter { func (_m *MockStatsJobManager) DropStatsTask(originSegmentID int64, subJobType indexpb.StatsSubJob) error { ret := _m.Called(originSegmentID, subJobType) + if len(ret) == 0 { + panic("no return value specified for DropStatsTask") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, indexpb.StatsSubJob) error); ok { r0 = rf(originSegmentID, subJobType) @@ -67,6 +71,10 @@ func (_c *MockStatsJobManager_DropStatsTask_Call) RunAndReturn(run func(int64, i func (_m *MockStatsJobManager) GetStatsTaskState(originSegmentID int64, subJobType indexpb.StatsSubJob) indexpb.JobState { ret := _m.Called(originSegmentID, subJobType) + if len(ret) == 0 { + panic("no return value specified for GetStatsTaskState") + } + var r0 indexpb.JobState if rf, ok := ret.Get(0).(func(int64, indexpb.StatsSubJob) indexpb.JobState); ok { r0 = rf(originSegmentID, subJobType) @@ -174,6 +182,10 @@ func (_c *MockStatsJobManager_Stop_Call) RunAndReturn(run func()) *MockStatsJobM func (_m *MockStatsJobManager) SubmitStatsTask(originSegmentID int64, targetSegmentID int64, subJobType indexpb.StatsSubJob, canRecycle bool) error { ret := _m.Called(originSegmentID, targetSegmentID, subJobType, canRecycle) + if len(ret) == 0 { + panic("no return value specified for SubmitStatsTask") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, int64, indexpb.StatsSubJob, bool) error); ok { r0 = rf(originSegmentID, targetSegmentID, subJobType, canRecycle) diff --git a/internal/datacoord/mock_segment_manager.go b/internal/datacoord/mock_segment_manager.go index c2df11ab17..f3c2f6db6d 100644 --- a/internal/datacoord/mock_segment_manager.go +++ b/internal/datacoord/mock_segment_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -25,6 +25,10 @@ func (_m *MockManager) EXPECT() *MockManager_Expecter { func (_m *MockManager) AllocNewGrowingSegment(ctx context.Context, collectionID int64, partitionID int64, segmentID int64, channelName string) (*SegmentInfo, error) { ret := _m.Called(ctx, collectionID, partitionID, segmentID, channelName) + if len(ret) == 0 { + panic("no return value specified for AllocNewGrowingSegment") + } + var r0 *SegmentInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, int64, int64, string) (*SegmentInfo, error)); ok { @@ -83,6 +87,10 @@ func (_c *MockManager_AllocNewGrowingSegment_Call) RunAndReturn(run func(context func (_m *MockManager) AllocSegment(ctx context.Context, collectionID int64, partitionID int64, channelName string, requestRows int64) ([]*Allocation, error) { ret := _m.Called(ctx, collectionID, partitionID, channelName, requestRows) + if len(ret) == 0 { + panic("no return value specified for AllocSegment") + } + var r0 []*Allocation var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, int64, string, int64) ([]*Allocation, error)); ok { @@ -209,6 +217,10 @@ func (_c *MockManager_DropSegmentsOfChannel_Call) RunAndReturn(run func(context. func (_m *MockManager) ExpireAllocations(channel string, ts uint64) error { ret := _m.Called(channel, ts) + if len(ret) == 0 { + panic("no return value specified for ExpireAllocations") + } + var r0 error if rf, ok := ret.Get(0).(func(string, uint64) error); ok { r0 = rf(channel, ts) @@ -252,6 +264,10 @@ func (_c *MockManager_ExpireAllocations_Call) RunAndReturn(run func(string, uint func (_m *MockManager) GetFlushableSegments(ctx context.Context, channel string, ts uint64) ([]int64, error) { ret := _m.Called(ctx, channel, ts) + if len(ret) == 0 { + panic("no return value specified for GetFlushableSegments") + } + var r0 []int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, uint64) ([]int64, error)); ok { @@ -308,6 +324,10 @@ func (_c *MockManager_GetFlushableSegments_Call) RunAndReturn(run func(context.C func (_m *MockManager) SealAllSegments(ctx context.Context, collectionID int64, segIDs []int64) ([]int64, error) { ret := _m.Called(ctx, collectionID, segIDs) + if len(ret) == 0 { + panic("no return value specified for SealAllSegments") + } + var r0 []int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, []int64) ([]int64, error)); ok { diff --git a/internal/datacoord/mock_subcluster.go b/internal/datacoord/mock_subcluster.go index 465eb2ac73..378d8ec6e8 100644 --- a/internal/datacoord/mock_subcluster.go +++ b/internal/datacoord/mock_subcluster.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -26,6 +26,10 @@ func (_m *MockSubCluster) EXPECT() *MockSubCluster_Expecter { func (_m *MockSubCluster) CheckChannelOperationProgress(ctx context.Context, nodeID int64, info *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error) { ret := _m.Called(ctx, nodeID, info) + if len(ret) == 0 { + panic("no return value specified for CheckChannelOperationProgress") + } + var r0 *datapb.ChannelOperationProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error)); ok { @@ -82,6 +86,10 @@ func (_c *MockSubCluster_CheckChannelOperationProgress_Call) RunAndReturn(run fu func (_m *MockSubCluster) NotifyChannelOperation(ctx context.Context, nodeID int64, req *datapb.ChannelOperationsRequest) error { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for NotifyChannelOperation") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelOperationsRequest) error); ok { r0 = rf(ctx, nodeID, req) diff --git a/internal/datacoord/mock_trigger_manager.go b/internal/datacoord/mock_trigger_manager.go index 6342dc66aa..4a0b72c460 100644 --- a/internal/datacoord/mock_trigger_manager.go +++ b/internal/datacoord/mock_trigger_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package datacoord @@ -25,6 +25,10 @@ func (_m *MockTriggerManager) EXPECT() *MockTriggerManager_Expecter { func (_m *MockTriggerManager) ManualTrigger(ctx context.Context, collectionID int64, clusteringCompaction bool) (int64, error) { ret := _m.Called(ctx, collectionID, clusteringCompaction) + if len(ret) == 0 { + panic("no return value specified for ManualTrigger") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, bool) (int64, error)); ok { diff --git a/internal/datacoord/session/mock_datanode_manager.go b/internal/datacoord/session/mock_datanode_manager.go index fce3b55236..60da9d2f2b 100644 --- a/internal/datacoord/session/mock_datanode_manager.go +++ b/internal/datacoord/session/mock_datanode_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package session @@ -61,6 +61,10 @@ func (_c *MockDataNodeManager_AddSession_Call) RunAndReturn(run func(*NodeInfo)) func (_m *MockDataNodeManager) CheckChannelOperationProgress(ctx context.Context, nodeID int64, info *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error) { ret := _m.Called(ctx, nodeID, info) + if len(ret) == 0 { + panic("no return value specified for CheckChannelOperationProgress") + } + var r0 *datapb.ChannelOperationProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error)); ok { @@ -117,6 +121,10 @@ func (_c *MockDataNodeManager_CheckChannelOperationProgress_Call) RunAndReturn(r func (_m *MockDataNodeManager) CheckHealth(ctx context.Context) error { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context) error); ok { r0 = rf(ctx) @@ -191,6 +199,10 @@ func (_c *MockDataNodeManager_Close_Call) RunAndReturn(run func()) *MockDataNode func (_m *MockDataNodeManager) Compaction(ctx context.Context, nodeID int64, plan *datapb.CompactionPlan) error { ret := _m.Called(ctx, nodeID, plan) + if len(ret) == 0 { + panic("no return value specified for Compaction") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.CompactionPlan) error); ok { r0 = rf(ctx, nodeID, plan) @@ -268,6 +280,10 @@ func (_c *MockDataNodeManager_DeleteSession_Call) RunAndReturn(run func(*NodeInf func (_m *MockDataNodeManager) DropCompactionPlan(nodeID int64, req *datapb.DropCompactionPlanRequest) error { ret := _m.Called(nodeID, req) + if len(ret) == 0 { + panic("no return value specified for DropCompactionPlan") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.DropCompactionPlanRequest) error); ok { r0 = rf(nodeID, req) @@ -311,6 +327,10 @@ func (_c *MockDataNodeManager_DropCompactionPlan_Call) RunAndReturn(run func(int func (_m *MockDataNodeManager) DropImport(nodeID int64, in *datapb.DropImportRequest) error { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for DropImport") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.DropImportRequest) error); ok { r0 = rf(nodeID, in) @@ -389,6 +409,10 @@ func (_c *MockDataNodeManager_Flush_Call) RunAndReturn(run func(context.Context, func (_m *MockDataNodeManager) FlushChannels(ctx context.Context, nodeID int64, req *datapb.FlushChannelsRequest) error { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for FlushChannels") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.FlushChannelsRequest) error); ok { r0 = rf(ctx, nodeID, req) @@ -433,6 +457,10 @@ func (_c *MockDataNodeManager_FlushChannels_Call) RunAndReturn(run func(context. func (_m *MockDataNodeManager) GetCompactionPlanResult(nodeID int64, planID int64) (*datapb.CompactionPlanResult, error) { ret := _m.Called(nodeID, planID) + if len(ret) == 0 { + panic("no return value specified for GetCompactionPlanResult") + } + var r0 *datapb.CompactionPlanResult var r1 error if rf, ok := ret.Get(0).(func(int64, int64) (*datapb.CompactionPlanResult, error)); ok { @@ -488,6 +516,10 @@ func (_c *MockDataNodeManager_GetCompactionPlanResult_Call) RunAndReturn(run fun func (_m *MockDataNodeManager) GetCompactionPlansResults() (map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult], error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetCompactionPlansResults") + } + var r0 map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult] var r1 error if rf, ok := ret.Get(0).(func() (map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult], error)); ok { @@ -541,6 +573,10 @@ func (_c *MockDataNodeManager_GetCompactionPlansResults_Call) RunAndReturn(run f func (_m *MockDataNodeManager) GetSession(_a0 int64) (*Session, bool) { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for GetSession") + } + var r0 *Session var r1 bool if rf, ok := ret.Get(0).(func(int64) (*Session, bool)); ok { @@ -595,6 +631,10 @@ func (_c *MockDataNodeManager_GetSession_Call) RunAndReturn(run func(int64) (*Se func (_m *MockDataNodeManager) GetSessionIDs() []int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetSessionIDs") + } + var r0 []int64 if rf, ok := ret.Get(0).(func() []int64); ok { r0 = rf() @@ -638,6 +678,10 @@ func (_c *MockDataNodeManager_GetSessionIDs_Call) RunAndReturn(run func() []int6 func (_m *MockDataNodeManager) GetSessions() []*Session { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetSessions") + } + var r0 []*Session if rf, ok := ret.Get(0).(func() []*Session); ok { r0 = rf() @@ -681,6 +725,10 @@ func (_c *MockDataNodeManager_GetSessions_Call) RunAndReturn(run func() []*Sessi func (_m *MockDataNodeManager) ImportV2(nodeID int64, in *datapb.ImportRequest) error { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.ImportRequest) error); ok { r0 = rf(nodeID, in) @@ -724,6 +772,10 @@ func (_c *MockDataNodeManager_ImportV2_Call) RunAndReturn(run func(int64, *datap func (_m *MockDataNodeManager) NotifyChannelOperation(ctx context.Context, nodeID int64, req *datapb.ChannelOperationsRequest) error { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for NotifyChannelOperation") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, *datapb.ChannelOperationsRequest) error); ok { r0 = rf(ctx, nodeID, req) @@ -768,6 +820,10 @@ func (_c *MockDataNodeManager_NotifyChannelOperation_Call) RunAndReturn(run func func (_m *MockDataNodeManager) PreImport(nodeID int64, in *datapb.PreImportRequest) error { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for PreImport") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.PreImportRequest) error); ok { r0 = rf(nodeID, in) @@ -811,6 +867,10 @@ func (_c *MockDataNodeManager_PreImport_Call) RunAndReturn(run func(int64, *data func (_m *MockDataNodeManager) QueryImport(nodeID int64, in *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error) { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for QueryImport") + } + var r0 *datapb.QueryImportResponse var r1 error if rf, ok := ret.Get(0).(func(int64, *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error)); ok { @@ -866,6 +926,10 @@ func (_c *MockDataNodeManager_QueryImport_Call) RunAndReturn(run func(int64, *da func (_m *MockDataNodeManager) QueryPreImport(nodeID int64, in *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error) { ret := _m.Called(nodeID, in) + if len(ret) == 0 { + panic("no return value specified for QueryPreImport") + } + var r0 *datapb.QueryPreImportResponse var r1 error if rf, ok := ret.Get(0).(func(int64, *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error)); ok { @@ -921,6 +985,10 @@ func (_c *MockDataNodeManager_QueryPreImport_Call) RunAndReturn(run func(int64, func (_m *MockDataNodeManager) QuerySlot(nodeID int64) (*datapb.QuerySlotResponse, error) { ret := _m.Called(nodeID) + if len(ret) == 0 { + panic("no return value specified for QuerySlot") + } + var r0 *datapb.QuerySlotResponse var r1 error if rf, ok := ret.Get(0).(func(int64) (*datapb.QuerySlotResponse, error)); ok { @@ -975,6 +1043,10 @@ func (_c *MockDataNodeManager_QuerySlot_Call) RunAndReturn(run func(int64) (*dat func (_m *MockDataNodeManager) SyncSegments(nodeID int64, req *datapb.SyncSegmentsRequest) error { ret := _m.Called(nodeID, req) + if len(ret) == 0 { + panic("no return value specified for SyncSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, *datapb.SyncSegmentsRequest) error); ok { r0 = rf(nodeID, req) diff --git a/internal/datacoord/session/mock_worker_manager.go b/internal/datacoord/session/mock_worker_manager.go index 92630835b3..556f590a10 100644 --- a/internal/datacoord/session/mock_worker_manager.go +++ b/internal/datacoord/session/mock_worker_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package session @@ -24,6 +24,10 @@ func (_m *MockWorkerManager) EXPECT() *MockWorkerManager_Expecter { func (_m *MockWorkerManager) AddNode(nodeID int64, address string) error { ret := _m.Called(nodeID, address) + if len(ret) == 0 { + panic("no return value specified for AddNode") + } + var r0 error if rf, ok := ret.Get(0).(func(int64, string) error); ok { r0 = rf(nodeID, address) @@ -67,6 +71,10 @@ func (_c *MockWorkerManager_AddNode_Call) RunAndReturn(run func(int64, string) e func (_m *MockWorkerManager) ClientSupportDisk() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ClientSupportDisk") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -108,6 +116,10 @@ func (_c *MockWorkerManager_ClientSupportDisk_Call) RunAndReturn(run func() bool func (_m *MockWorkerManager) GetAllClients() map[int64]types.IndexNodeClient { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAllClients") + } + var r0 map[int64]types.IndexNodeClient if rf, ok := ret.Get(0).(func() map[int64]types.IndexNodeClient); ok { r0 = rf() @@ -151,6 +163,10 @@ func (_c *MockWorkerManager_GetAllClients_Call) RunAndReturn(run func() map[int6 func (_m *MockWorkerManager) GetClientByID(nodeID int64) (types.IndexNodeClient, bool) { ret := _m.Called(nodeID) + if len(ret) == 0 { + panic("no return value specified for GetClientByID") + } + var r0 types.IndexNodeClient var r1 bool if rf, ok := ret.Get(0).(func(int64) (types.IndexNodeClient, bool)); ok { @@ -205,6 +221,10 @@ func (_c *MockWorkerManager_GetClientByID_Call) RunAndReturn(run func(int64) (ty func (_m *MockWorkerManager) PickClient() (int64, types.IndexNodeClient) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for PickClient") + } + var r0 int64 var r1 types.IndexNodeClient if rf, ok := ret.Get(0).(func() (int64, types.IndexNodeClient)); ok { diff --git a/internal/kv/mocks/meta_kv.go b/internal/kv/mocks/meta_kv.go index 1119c6a45b..85094ae478 100644 --- a/internal/kv/mocks/meta_kv.go +++ b/internal/kv/mocks/meta_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -56,6 +56,10 @@ func (_c *MetaKv_Close_Call) RunAndReturn(run func()) *MetaKv_Close_Call { func (_m *MetaKv) CompareVersionAndSwap(key string, version int64, target string) (bool, error) { ret := _m.Called(key, version, target) + if len(ret) == 0 { + panic("no return value specified for CompareVersionAndSwap") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string, int64, string) (bool, error)); ok { @@ -110,6 +114,10 @@ func (_c *MetaKv_CompareVersionAndSwap_Call) RunAndReturn(run func(string, int64 func (_m *MetaKv) GetPath(key string) string { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for GetPath") + } + var r0 string if rf, ok := ret.Get(0).(func(string) string); ok { r0 = rf(key) @@ -152,6 +160,10 @@ func (_c *MetaKv_GetPath_Call) RunAndReturn(run func(string) string) *MetaKv_Get func (_m *MetaKv) Has(key string) (bool, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Has") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -204,6 +216,10 @@ func (_c *MetaKv_Has_Call) RunAndReturn(run func(string) (bool, error)) *MetaKv_ func (_m *MetaKv) HasPrefix(prefix string) (bool, error) { ret := _m.Called(prefix) + if len(ret) == 0 { + panic("no return value specified for HasPrefix") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -256,6 +272,10 @@ func (_c *MetaKv_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)) *M func (_m *MetaKv) Load(key string) (string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Load") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(string) (string, error)); ok { @@ -308,6 +328,10 @@ func (_c *MetaKv_Load_Call) RunAndReturn(run func(string) (string, error)) *Meta func (_m *MetaKv) LoadWithPrefix(key string) ([]string, []string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for LoadWithPrefix") + } + var r0 []string var r1 []string var r2 error @@ -371,6 +395,10 @@ func (_c *MetaKv_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]string, [ func (_m *MetaKv) MultiLoad(keys []string) ([]string, error) { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiLoad") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok { @@ -425,6 +453,10 @@ func (_c *MetaKv_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, erro func (_m *MetaKv) MultiRemove(keys []string) error { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiRemove") + } + var r0 error if rf, ok := ret.Get(0).(func([]string) error); ok { r0 = rf(keys) @@ -467,6 +499,10 @@ func (_c *MetaKv_MultiRemove_Call) RunAndReturn(run func([]string) error) *MetaK func (_m *MetaKv) MultiSave(kvs map[string]string) error { ret := _m.Called(kvs) + if len(ret) == 0 { + panic("no return value specified for MultiSave") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string) error); ok { r0 = rf(kvs) @@ -516,6 +552,10 @@ func (_m *MetaKv) MultiSaveAndRemove(saves map[string]string, removals []string, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemove") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -574,6 +614,10 @@ func (_m *MetaKv) MultiSaveAndRemoveWithPrefix(saves map[string]string, removals _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -625,6 +669,10 @@ func (_c *MetaKv_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(map[st func (_m *MetaKv) Remove(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -667,6 +715,10 @@ func (_c *MetaKv_Remove_Call) RunAndReturn(run func(string) error) *MetaKv_Remov func (_m *MetaKv) RemoveWithPrefix(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for RemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -709,6 +761,10 @@ func (_c *MetaKv_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) *Me func (_m *MetaKv) Save(key string, value string) error { ret := _m.Called(key, value) + if len(ret) == 0 { + panic("no return value specified for Save") + } + var r0 error if rf, ok := ret.Get(0).(func(string, string) error); ok { r0 = rf(key, value) @@ -752,6 +808,10 @@ func (_c *MetaKv_Save_Call) RunAndReturn(run func(string, string) error) *MetaKv func (_m *MetaKv) WalkWithPrefix(prefix string, paginationSize int, fn func([]byte, []byte) error) error { ret := _m.Called(prefix, paginationSize, fn) + if len(ret) == 0 { + panic("no return value specified for WalkWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string, int, func([]byte, []byte) error) error); ok { r0 = rf(prefix, paginationSize, fn) diff --git a/internal/kv/mocks/snapshot_kv.go b/internal/kv/mocks/snapshot_kv.go index dc2de1d783..42bc5d7ad9 100644 --- a/internal/kv/mocks/snapshot_kv.go +++ b/internal/kv/mocks/snapshot_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -21,6 +21,10 @@ func (_m *SnapShotKV) EXPECT() *SnapShotKV_Expecter { func (_m *SnapShotKV) Load(key string, ts uint64) (string, error) { ret := _m.Called(key, ts) + if len(ret) == 0 { + panic("no return value specified for Load") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(string, uint64) (string, error)); ok { @@ -74,6 +78,10 @@ func (_c *SnapShotKV_Load_Call) RunAndReturn(run func(string, uint64) (string, e func (_m *SnapShotKV) LoadWithPrefix(key string, ts uint64) ([]string, []string, error) { ret := _m.Called(key, ts) + if len(ret) == 0 { + panic("no return value specified for LoadWithPrefix") + } + var r0 []string var r1 []string var r2 error @@ -138,6 +146,10 @@ func (_c *SnapShotKV_LoadWithPrefix_Call) RunAndReturn(run func(string, uint64) func (_m *SnapShotKV) MultiSave(kvs map[string]string, ts uint64) error { ret := _m.Called(kvs, ts) + if len(ret) == 0 { + panic("no return value specified for MultiSave") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, uint64) error); ok { r0 = rf(kvs, ts) @@ -181,6 +193,10 @@ func (_c *SnapShotKV_MultiSave_Call) RunAndReturn(run func(map[string]string, ui func (_m *SnapShotKV) MultiSaveAndRemove(saves map[string]string, removals []string, ts uint64) error { ret := _m.Called(saves, removals, ts) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemove") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, uint64) error); ok { r0 = rf(saves, removals, ts) @@ -225,6 +241,10 @@ func (_c *SnapShotKV_MultiSaveAndRemove_Call) RunAndReturn(run func(map[string]s func (_m *SnapShotKV) MultiSaveAndRemoveWithPrefix(saves map[string]string, removals []string, ts uint64) error { ret := _m.Called(saves, removals, ts) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, uint64) error); ok { r0 = rf(saves, removals, ts) @@ -269,6 +289,10 @@ func (_c *SnapShotKV_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(ma func (_m *SnapShotKV) Save(key string, value string, ts uint64) error { ret := _m.Called(key, value, ts) + if len(ret) == 0 { + panic("no return value specified for Save") + } + var r0 error if rf, ok := ret.Get(0).(func(string, string, uint64) error); ok { r0 = rf(key, value, ts) diff --git a/internal/kv/mocks/txn_kv.go b/internal/kv/mocks/txn_kv.go index 2168180ce3..76811bbfb7 100644 --- a/internal/kv/mocks/txn_kv.go +++ b/internal/kv/mocks/txn_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -56,6 +56,10 @@ func (_c *TxnKV_Close_Call) RunAndReturn(run func()) *TxnKV_Close_Call { func (_m *TxnKV) Has(key string) (bool, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Has") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -108,6 +112,10 @@ func (_c *TxnKV_Has_Call) RunAndReturn(run func(string) (bool, error)) *TxnKV_Ha func (_m *TxnKV) HasPrefix(prefix string) (bool, error) { ret := _m.Called(prefix) + if len(ret) == 0 { + panic("no return value specified for HasPrefix") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -160,6 +168,10 @@ func (_c *TxnKV_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)) *Tx func (_m *TxnKV) Load(key string) (string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Load") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(string) (string, error)); ok { @@ -212,6 +224,10 @@ func (_c *TxnKV_Load_Call) RunAndReturn(run func(string) (string, error)) *TxnKV func (_m *TxnKV) LoadWithPrefix(key string) ([]string, []string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for LoadWithPrefix") + } + var r0 []string var r1 []string var r2 error @@ -275,6 +291,10 @@ func (_c *TxnKV_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]string, [] func (_m *TxnKV) MultiLoad(keys []string) ([]string, error) { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiLoad") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok { @@ -329,6 +349,10 @@ func (_c *TxnKV_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, error func (_m *TxnKV) MultiRemove(keys []string) error { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiRemove") + } + var r0 error if rf, ok := ret.Get(0).(func([]string) error); ok { r0 = rf(keys) @@ -371,6 +395,10 @@ func (_c *TxnKV_MultiRemove_Call) RunAndReturn(run func([]string) error) *TxnKV_ func (_m *TxnKV) MultiSave(kvs map[string]string) error { ret := _m.Called(kvs) + if len(ret) == 0 { + panic("no return value specified for MultiSave") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string) error); ok { r0 = rf(kvs) @@ -420,6 +448,10 @@ func (_m *TxnKV) MultiSaveAndRemove(saves map[string]string, removals []string, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemove") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -478,6 +510,10 @@ func (_m *TxnKV) MultiSaveAndRemoveWithPrefix(saves map[string]string, removals _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -529,6 +565,10 @@ func (_c *TxnKV_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(map[str func (_m *TxnKV) Remove(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -571,6 +611,10 @@ func (_c *TxnKV_Remove_Call) RunAndReturn(run func(string) error) *TxnKV_Remove_ func (_m *TxnKV) RemoveWithPrefix(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for RemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -613,6 +657,10 @@ func (_c *TxnKV_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) *Txn func (_m *TxnKV) Save(key string, value string) error { ret := _m.Called(key, value) + if len(ret) == 0 { + panic("no return value specified for Save") + } + var r0 error if rf, ok := ret.Get(0).(func(string, string) error); ok { r0 = rf(key, value) diff --git a/internal/kv/mocks/watch_kv.go b/internal/kv/mocks/watch_kv.go index 4fe5aebe5b..af3328ef5d 100644 --- a/internal/kv/mocks/watch_kv.go +++ b/internal/kv/mocks/watch_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -59,6 +59,10 @@ func (_c *WatchKV_Close_Call) RunAndReturn(run func()) *WatchKV_Close_Call { func (_m *WatchKV) CompareVersionAndSwap(key string, version int64, target string) (bool, error) { ret := _m.Called(key, version, target) + if len(ret) == 0 { + panic("no return value specified for CompareVersionAndSwap") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string, int64, string) (bool, error)); ok { @@ -113,6 +117,10 @@ func (_c *WatchKV_CompareVersionAndSwap_Call) RunAndReturn(run func(string, int6 func (_m *WatchKV) GetPath(key string) string { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for GetPath") + } + var r0 string if rf, ok := ret.Get(0).(func(string) string); ok { r0 = rf(key) @@ -155,6 +163,10 @@ func (_c *WatchKV_GetPath_Call) RunAndReturn(run func(string) string) *WatchKV_G func (_m *WatchKV) Has(key string) (bool, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Has") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -207,6 +219,10 @@ func (_c *WatchKV_Has_Call) RunAndReturn(run func(string) (bool, error)) *WatchK func (_m *WatchKV) HasPrefix(prefix string) (bool, error) { ret := _m.Called(prefix) + if len(ret) == 0 { + panic("no return value specified for HasPrefix") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -259,6 +275,10 @@ func (_c *WatchKV_HasPrefix_Call) RunAndReturn(run func(string) (bool, error)) * func (_m *WatchKV) Load(key string) (string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Load") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(string) (string, error)); ok { @@ -311,6 +331,10 @@ func (_c *WatchKV_Load_Call) RunAndReturn(run func(string) (string, error)) *Wat func (_m *WatchKV) LoadWithPrefix(key string) ([]string, []string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for LoadWithPrefix") + } + var r0 []string var r1 []string var r2 error @@ -374,6 +398,10 @@ func (_c *WatchKV_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]string, func (_m *WatchKV) MultiLoad(keys []string) ([]string, error) { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiLoad") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok { @@ -428,6 +456,10 @@ func (_c *WatchKV_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, err func (_m *WatchKV) MultiRemove(keys []string) error { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiRemove") + } + var r0 error if rf, ok := ret.Get(0).(func([]string) error); ok { r0 = rf(keys) @@ -470,6 +502,10 @@ func (_c *WatchKV_MultiRemove_Call) RunAndReturn(run func([]string) error) *Watc func (_m *WatchKV) MultiSave(kvs map[string]string) error { ret := _m.Called(kvs) + if len(ret) == 0 { + panic("no return value specified for MultiSave") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string) error); ok { r0 = rf(kvs) @@ -519,6 +555,10 @@ func (_m *WatchKV) MultiSaveAndRemove(saves map[string]string, removals []string _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemove") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -577,6 +617,10 @@ func (_m *WatchKV) MultiSaveAndRemoveWithPrefix(saves map[string]string, removal _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -628,6 +672,10 @@ func (_c *WatchKV_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(map[s func (_m *WatchKV) Remove(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -670,6 +718,10 @@ func (_c *WatchKV_Remove_Call) RunAndReturn(run func(string) error) *WatchKV_Rem func (_m *WatchKV) RemoveWithPrefix(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for RemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -712,6 +764,10 @@ func (_c *WatchKV_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) *W func (_m *WatchKV) Save(key string, value string) error { ret := _m.Called(key, value) + if len(ret) == 0 { + panic("no return value specified for Save") + } + var r0 error if rf, ok := ret.Get(0).(func(string, string) error); ok { r0 = rf(key, value) @@ -755,6 +811,10 @@ func (_c *WatchKV_Save_Call) RunAndReturn(run func(string, string) error) *Watch func (_m *WatchKV) WalkWithPrefix(prefix string, paginationSize int, fn func([]byte, []byte) error) error { ret := _m.Called(prefix, paginationSize, fn) + if len(ret) == 0 { + panic("no return value specified for WalkWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string, int, func([]byte, []byte) error) error); ok { r0 = rf(prefix, paginationSize, fn) @@ -799,6 +859,10 @@ func (_c *WatchKV_WalkWithPrefix_Call) RunAndReturn(run func(string, int, func([ func (_m *WatchKV) Watch(key string) clientv3.WatchChan { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Watch") + } + var r0 clientv3.WatchChan if rf, ok := ret.Get(0).(func(string) clientv3.WatchChan); ok { r0 = rf(key) @@ -843,6 +907,10 @@ func (_c *WatchKV_Watch_Call) RunAndReturn(run func(string) clientv3.WatchChan) func (_m *WatchKV) WatchWithPrefix(key string) clientv3.WatchChan { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for WatchWithPrefix") + } + var r0 clientv3.WatchChan if rf, ok := ret.Get(0).(func(string) clientv3.WatchChan); ok { r0 = rf(key) @@ -887,6 +955,10 @@ func (_c *WatchKV_WatchWithPrefix_Call) RunAndReturn(run func(string) clientv3.W func (_m *WatchKV) WatchWithRevision(key string, revision int64) clientv3.WatchChan { ret := _m.Called(key, revision) + if len(ret) == 0 { + panic("no return value specified for WatchWithRevision") + } + var r0 clientv3.WatchChan if rf, ok := ret.Get(0).(func(string, int64) clientv3.WatchChan); ok { r0 = rf(key, revision) diff --git a/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go b/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go index b346552c96..6b391ff89a 100644 --- a/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go +++ b/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streaming @@ -37,6 +37,10 @@ func (_m *MockWALAccesser) AppendMessages(ctx context.Context, msgs ...message.M _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AppendMessages") + } + var r0 streaming.AppendResponses if rf, ok := ret.Get(0).(func(context.Context, ...message.MutableMessage) streaming.AppendResponses); ok { r0 = rf(ctx, msgs...) @@ -94,6 +98,10 @@ func (_m *MockWALAccesser) AppendMessagesWithOption(ctx context.Context, opts st _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AppendMessagesWithOption") + } + var r0 streaming.AppendResponses if rf, ok := ret.Get(0).(func(context.Context, streaming.AppendOption, ...message.MutableMessage) streaming.AppendResponses); ok { r0 = rf(ctx, opts, msgs...) @@ -152,6 +160,10 @@ func (_m *MockWALAccesser) RawAppend(ctx context.Context, msgs message.MutableMe _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for RawAppend") + } + var r0 *types.AppendResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage, ...streaming.AppendOption) (*types.AppendResult, error)); ok { @@ -215,6 +227,10 @@ func (_c *MockWALAccesser_RawAppend_Call) RunAndReturn(run func(context.Context, func (_m *MockWALAccesser) Read(ctx context.Context, opts streaming.ReadOption) streaming.Scanner { ret := _m.Called(ctx, opts) + if len(ret) == 0 { + panic("no return value specified for Read") + } + var r0 streaming.Scanner if rf, ok := ret.Get(0).(func(context.Context, streaming.ReadOption) streaming.Scanner); ok { r0 = rf(ctx, opts) @@ -260,6 +276,10 @@ func (_c *MockWALAccesser_Read_Call) RunAndReturn(run func(context.Context, stre func (_m *MockWALAccesser) Txn(ctx context.Context, opts streaming.TxnOption) (streaming.Txn, error) { ret := _m.Called(ctx, opts) + if len(ret) == 0 { + panic("no return value specified for Txn") + } + var r0 streaming.Txn var r1 error if rf, ok := ret.Get(0).(func(context.Context, streaming.TxnOption) (streaming.Txn, error)); ok { diff --git a/internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go b/internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go index 5623d4d127..8ff1fb2d11 100644 --- a/internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go +++ b/internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_balancer @@ -58,6 +58,10 @@ func (_c *MockSubConn_Connect_Call) RunAndReturn(run func()) *MockSubConn_Connec func (_m *MockSubConn) GetOrBuildProducer(_a0 balancer.ProducerBuilder) (balancer.Producer, func()) { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for GetOrBuildProducer") + } + var r0 balancer.Producer var r1 func() if rf, ok := ret.Get(0).(func(balancer.ProducerBuilder) (balancer.Producer, func())); ok { diff --git a/internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go b/internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go index 8f05a876f9..dd32a2d265 100644 --- a/internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go +++ b/internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_resolver @@ -59,6 +59,10 @@ func (_c *MockClientConn_NewAddress_Call) RunAndReturn(run func([]resolver.Addre func (_m *MockClientConn) ParseServiceConfig(serviceConfigJSON string) *serviceconfig.ParseResult { ret := _m.Called(serviceConfigJSON) + if len(ret) == 0 { + panic("no return value specified for ParseServiceConfig") + } + var r0 *serviceconfig.ParseResult if rf, ok := ret.Get(0).(func(string) *serviceconfig.ParseResult); ok { r0 = rf(serviceConfigJSON) @@ -136,6 +140,10 @@ func (_c *MockClientConn_ReportError_Call) RunAndReturn(run func(error)) *MockCl func (_m *MockClientConn) UpdateState(_a0 resolver.State) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for UpdateState") + } + var r0 error if rf, ok := ret.Get(0).(func(resolver.State) error); ok { r0 = rf(_a0) diff --git a/internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go b/internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go index 2bef81aba2..64df110847 100644 --- a/internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go +++ b/internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_grpc @@ -27,6 +27,10 @@ func (_m *MockClientStream) EXPECT() *MockClientStream_Expecter { func (_m *MockClientStream) CloseSend() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -68,6 +72,10 @@ func (_c *MockClientStream_CloseSend_Call) RunAndReturn(run func() error) *MockC func (_m *MockClientStream) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -111,6 +119,10 @@ func (_c *MockClientStream_Context_Call) RunAndReturn(run func() context.Context func (_m *MockClientStream) Header() (metadata.MD, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Header") + } + var r0 metadata.MD var r1 error if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { @@ -164,6 +176,10 @@ func (_c *MockClientStream_Header_Call) RunAndReturn(run func() (metadata.MD, er func (_m *MockClientStream) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -206,6 +222,10 @@ func (_c *MockClientStream_RecvMsg_Call) RunAndReturn(run func(interface{}) erro func (_m *MockClientStream) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -248,6 +268,10 @@ func (_c *MockClientStream_SendMsg_Call) RunAndReturn(run func(interface{}) erro func (_m *MockClientStream) Trailer() metadata.MD { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + var r0 metadata.MD if rf, ok := ret.Get(0).(func() metadata.MD); ok { r0 = rf() diff --git a/internal/mocks/mock_datacoord.go b/internal/mocks/mock_datacoord.go index 95fae6cf51..b33f31f3c1 100644 --- a/internal/mocks/mock_datacoord.go +++ b/internal/mocks/mock_datacoord.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -42,6 +42,10 @@ func (_m *MockDataCoord) EXPECT() *MockDataCoord_Expecter { func (_m *MockDataCoord) AllocSegment(_a0 context.Context, _a1 *datapb.AllocSegmentRequest) (*datapb.AllocSegmentResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AllocSegment") + } + var r0 *datapb.AllocSegmentResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.AllocSegmentRequest) (*datapb.AllocSegmentResponse, error)); ok { @@ -97,6 +101,10 @@ func (_c *MockDataCoord_AllocSegment_Call) RunAndReturn(run func(context.Context func (_m *MockDataCoord) AlterIndex(_a0 context.Context, _a1 *indexpb.AlterIndexRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.AlterIndexRequest) (*commonpb.Status, error)); ok { @@ -152,6 +160,10 @@ func (_c *MockDataCoord_AlterIndex_Call) RunAndReturn(run func(context.Context, func (_m *MockDataCoord) AssignSegmentID(_a0 context.Context, _a1 *datapb.AssignSegmentIDRequest) (*datapb.AssignSegmentIDResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AssignSegmentID") + } + var r0 *datapb.AssignSegmentIDResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.AssignSegmentIDRequest) (*datapb.AssignSegmentIDResponse, error)); ok { @@ -207,6 +219,10 @@ func (_c *MockDataCoord_AssignSegmentID_Call) RunAndReturn(run func(context.Cont func (_m *MockDataCoord) BroadcastAlteredCollection(_a0 context.Context, _a1 *datapb.AlterCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for BroadcastAlteredCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.AlterCollectionRequest) (*commonpb.Status, error)); ok { @@ -262,6 +278,10 @@ func (_c *MockDataCoord_BroadcastAlteredCollection_Call) RunAndReturn(run func(c func (_m *MockDataCoord) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok { @@ -317,6 +337,10 @@ func (_c *MockDataCoord_CheckHealth_Call) RunAndReturn(run func(context.Context, func (_m *MockDataCoord) CreateIndex(_a0 context.Context, _a1 *indexpb.CreateIndexRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.CreateIndexRequest) (*commonpb.Status, error)); ok { @@ -372,6 +396,10 @@ func (_c *MockDataCoord_CreateIndex_Call) RunAndReturn(run func(context.Context, func (_m *MockDataCoord) DescribeIndex(_a0 context.Context, _a1 *indexpb.DescribeIndexRequest) (*indexpb.DescribeIndexResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeIndex") + } + var r0 *indexpb.DescribeIndexResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DescribeIndexRequest) (*indexpb.DescribeIndexResponse, error)); ok { @@ -427,6 +455,10 @@ func (_c *MockDataCoord_DescribeIndex_Call) RunAndReturn(run func(context.Contex func (_m *MockDataCoord) DropIndex(_a0 context.Context, _a1 *indexpb.DropIndexRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DropIndexRequest) (*commonpb.Status, error)); ok { @@ -482,6 +514,10 @@ func (_c *MockDataCoord_DropIndex_Call) RunAndReturn(run func(context.Context, * func (_m *MockDataCoord) DropVirtualChannel(_a0 context.Context, _a1 *datapb.DropVirtualChannelRequest) (*datapb.DropVirtualChannelResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropVirtualChannel") + } + var r0 *datapb.DropVirtualChannelResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropVirtualChannelRequest) (*datapb.DropVirtualChannelResponse, error)); ok { @@ -537,6 +573,10 @@ func (_c *MockDataCoord_DropVirtualChannel_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) Flush(_a0 context.Context, _a1 *datapb.FlushRequest) (*datapb.FlushResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Flush") + } + var r0 *datapb.FlushResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushRequest) (*datapb.FlushResponse, error)); ok { @@ -592,6 +632,10 @@ func (_c *MockDataCoord_Flush_Call) RunAndReturn(run func(context.Context, *data func (_m *MockDataCoord) GcConfirm(_a0 context.Context, _a1 *datapb.GcConfirmRequest) (*datapb.GcConfirmResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GcConfirm") + } + var r0 *datapb.GcConfirmResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcConfirmRequest) (*datapb.GcConfirmResponse, error)); ok { @@ -647,6 +691,10 @@ func (_c *MockDataCoord_GcConfirm_Call) RunAndReturn(run func(context.Context, * func (_m *MockDataCoord) GcControl(_a0 context.Context, _a1 *datapb.GcControlRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GcControl") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcControlRequest) (*commonpb.Status, error)); ok { @@ -702,6 +750,10 @@ func (_c *MockDataCoord_GcControl_Call) RunAndReturn(run func(context.Context, * func (_m *MockDataCoord) GetChannelRecoveryInfo(_a0 context.Context, _a1 *datapb.GetChannelRecoveryInfoRequest) (*datapb.GetChannelRecoveryInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetChannelRecoveryInfo") + } + var r0 *datapb.GetChannelRecoveryInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetChannelRecoveryInfoRequest) (*datapb.GetChannelRecoveryInfoResponse, error)); ok { @@ -757,6 +809,10 @@ func (_c *MockDataCoord_GetChannelRecoveryInfo_Call) RunAndReturn(run func(conte func (_m *MockDataCoord) GetCollectionStatistics(_a0 context.Context, _a1 *datapb.GetCollectionStatisticsRequest) (*datapb.GetCollectionStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCollectionStatistics") + } + var r0 *datapb.GetCollectionStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetCollectionStatisticsRequest) (*datapb.GetCollectionStatisticsResponse, error)); ok { @@ -812,6 +868,10 @@ func (_c *MockDataCoord_GetCollectionStatistics_Call) RunAndReturn(run func(cont func (_m *MockDataCoord) GetCompactionState(_a0 context.Context, _a1 *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCompactionState") + } + var r0 *milvuspb.GetCompactionStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)); ok { @@ -867,6 +927,10 @@ func (_c *MockDataCoord_GetCompactionState_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) GetCompactionStateWithPlans(_a0 context.Context, _a1 *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCompactionStateWithPlans") + } + var r0 *milvuspb.GetCompactionPlansResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)); ok { @@ -922,6 +986,10 @@ func (_c *MockDataCoord_GetCompactionStateWithPlans_Call) RunAndReturn(run func( func (_m *MockDataCoord) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -977,6 +1045,10 @@ func (_c *MockDataCoord_GetComponentStates_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) GetFlushAllState(_a0 context.Context, _a1 *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetFlushAllState") + } + var r0 *milvuspb.GetFlushAllStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)); ok { @@ -1032,6 +1104,10 @@ func (_c *MockDataCoord_GetFlushAllState_Call) RunAndReturn(run func(context.Con func (_m *MockDataCoord) GetFlushState(_a0 context.Context, _a1 *datapb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetFlushState") + } + var r0 *milvuspb.GetFlushStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)); ok { @@ -1087,6 +1163,10 @@ func (_c *MockDataCoord_GetFlushState_Call) RunAndReturn(run func(context.Contex func (_m *MockDataCoord) GetFlushedSegments(_a0 context.Context, _a1 *datapb.GetFlushedSegmentsRequest) (*datapb.GetFlushedSegmentsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetFlushedSegments") + } + var r0 *datapb.GetFlushedSegmentsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushedSegmentsRequest) (*datapb.GetFlushedSegmentsResponse, error)); ok { @@ -1142,6 +1222,10 @@ func (_c *MockDataCoord_GetFlushedSegments_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) GetImportProgress(_a0 context.Context, _a1 *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetImportProgress") + } + var r0 *internalpb.GetImportProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error)); ok { @@ -1197,6 +1281,10 @@ func (_c *MockDataCoord_GetImportProgress_Call) RunAndReturn(run func(context.Co func (_m *MockDataCoord) GetIndexBuildProgress(_a0 context.Context, _a1 *indexpb.GetIndexBuildProgressRequest) (*indexpb.GetIndexBuildProgressResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexBuildProgress") + } + var r0 *indexpb.GetIndexBuildProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexBuildProgressRequest) (*indexpb.GetIndexBuildProgressResponse, error)); ok { @@ -1252,6 +1340,10 @@ func (_c *MockDataCoord_GetIndexBuildProgress_Call) RunAndReturn(run func(contex func (_m *MockDataCoord) GetIndexInfos(_a0 context.Context, _a1 *indexpb.GetIndexInfoRequest) (*indexpb.GetIndexInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexInfos") + } + var r0 *indexpb.GetIndexInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexInfoRequest) (*indexpb.GetIndexInfoResponse, error)); ok { @@ -1307,6 +1399,10 @@ func (_c *MockDataCoord_GetIndexInfos_Call) RunAndReturn(run func(context.Contex func (_m *MockDataCoord) GetIndexState(_a0 context.Context, _a1 *indexpb.GetIndexStateRequest) (*indexpb.GetIndexStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexState") + } + var r0 *indexpb.GetIndexStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStateRequest) (*indexpb.GetIndexStateResponse, error)); ok { @@ -1362,6 +1458,10 @@ func (_c *MockDataCoord_GetIndexState_Call) RunAndReturn(run func(context.Contex func (_m *MockDataCoord) GetIndexStatistics(_a0 context.Context, _a1 *indexpb.GetIndexStatisticsRequest) (*indexpb.GetIndexStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexStatistics") + } + var r0 *indexpb.GetIndexStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStatisticsRequest) (*indexpb.GetIndexStatisticsResponse, error)); ok { @@ -1417,6 +1517,10 @@ func (_c *MockDataCoord_GetIndexStatistics_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) GetInsertBinlogPaths(_a0 context.Context, _a1 *datapb.GetInsertBinlogPathsRequest) (*datapb.GetInsertBinlogPathsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetInsertBinlogPaths") + } + var r0 *datapb.GetInsertBinlogPathsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetInsertBinlogPathsRequest) (*datapb.GetInsertBinlogPathsResponse, error)); ok { @@ -1472,6 +1576,10 @@ func (_c *MockDataCoord_GetInsertBinlogPaths_Call) RunAndReturn(run func(context func (_m *MockDataCoord) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -1527,6 +1635,10 @@ func (_c *MockDataCoord_GetMetrics_Call) RunAndReturn(run func(context.Context, func (_m *MockDataCoord) GetPartitionStatistics(_a0 context.Context, _a1 *datapb.GetPartitionStatisticsRequest) (*datapb.GetPartitionStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetPartitionStatistics") + } + var r0 *datapb.GetPartitionStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetPartitionStatisticsRequest) (*datapb.GetPartitionStatisticsResponse, error)); ok { @@ -1582,6 +1694,10 @@ func (_c *MockDataCoord_GetPartitionStatistics_Call) RunAndReturn(run func(conte func (_m *MockDataCoord) GetRecoveryInfo(_a0 context.Context, _a1 *datapb.GetRecoveryInfoRequest) (*datapb.GetRecoveryInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetRecoveryInfo") + } + var r0 *datapb.GetRecoveryInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequest) (*datapb.GetRecoveryInfoResponse, error)); ok { @@ -1637,6 +1753,10 @@ func (_c *MockDataCoord_GetRecoveryInfo_Call) RunAndReturn(run func(context.Cont func (_m *MockDataCoord) GetRecoveryInfoV2(_a0 context.Context, _a1 *datapb.GetRecoveryInfoRequestV2) (*datapb.GetRecoveryInfoResponseV2, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetRecoveryInfoV2") + } + var r0 *datapb.GetRecoveryInfoResponseV2 var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequestV2) (*datapb.GetRecoveryInfoResponseV2, error)); ok { @@ -1692,6 +1812,10 @@ func (_c *MockDataCoord_GetRecoveryInfoV2_Call) RunAndReturn(run func(context.Co func (_m *MockDataCoord) GetSegmentIndexState(_a0 context.Context, _a1 *indexpb.GetSegmentIndexStateRequest) (*indexpb.GetSegmentIndexStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentIndexState") + } + var r0 *indexpb.GetSegmentIndexStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetSegmentIndexStateRequest) (*indexpb.GetSegmentIndexStateResponse, error)); ok { @@ -1747,6 +1871,10 @@ func (_c *MockDataCoord_GetSegmentIndexState_Call) RunAndReturn(run func(context func (_m *MockDataCoord) GetSegmentInfo(_a0 context.Context, _a1 *datapb.GetSegmentInfoRequest) (*datapb.GetSegmentInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *datapb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoRequest) (*datapb.GetSegmentInfoResponse, error)); ok { @@ -1802,6 +1930,10 @@ func (_c *MockDataCoord_GetSegmentInfo_Call) RunAndReturn(run func(context.Conte func (_m *MockDataCoord) GetSegmentInfoChannel(_a0 context.Context, _a1 *datapb.GetSegmentInfoChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfoChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -1857,6 +1989,10 @@ func (_c *MockDataCoord_GetSegmentInfoChannel_Call) RunAndReturn(run func(contex func (_m *MockDataCoord) GetSegmentStates(_a0 context.Context, _a1 *datapb.GetSegmentStatesRequest) (*datapb.GetSegmentStatesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentStates") + } + var r0 *datapb.GetSegmentStatesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentStatesRequest) (*datapb.GetSegmentStatesResponse, error)); ok { @@ -1912,6 +2048,10 @@ func (_c *MockDataCoord_GetSegmentStates_Call) RunAndReturn(run func(context.Con func (_m *MockDataCoord) GetSegmentsByStates(_a0 context.Context, _a1 *datapb.GetSegmentsByStatesRequest) (*datapb.GetSegmentsByStatesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentsByStates") + } + var r0 *datapb.GetSegmentsByStatesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentsByStatesRequest) (*datapb.GetSegmentsByStatesResponse, error)); ok { @@ -1967,6 +2107,10 @@ func (_c *MockDataCoord_GetSegmentsByStates_Call) RunAndReturn(run func(context. func (_m *MockDataCoord) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -2022,6 +2166,10 @@ func (_c *MockDataCoord_GetStatisticsChannel_Call) RunAndReturn(run func(context func (_m *MockDataCoord) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -2077,6 +2225,10 @@ func (_c *MockDataCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) ImportV2(_a0 context.Context, _a1 *internalpb.ImportRequestInternal) (*internalpb.ImportResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 *internalpb.ImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequestInternal) (*internalpb.ImportResponse, error)); ok { @@ -2132,6 +2284,10 @@ func (_c *MockDataCoord_ImportV2_Call) RunAndReturn(run func(context.Context, *i func (_m *MockDataCoord) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -2173,6 +2329,10 @@ func (_c *MockDataCoord_Init_Call) RunAndReturn(run func() error) *MockDataCoord func (_m *MockDataCoord) ListImports(_a0 context.Context, _a1 *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListImports") + } + var r0 *internalpb.ListImportsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal) (*internalpb.ListImportsResponse, error)); ok { @@ -2228,6 +2388,10 @@ func (_c *MockDataCoord_ListImports_Call) RunAndReturn(run func(context.Context, func (_m *MockDataCoord) ListIndexes(_a0 context.Context, _a1 *indexpb.ListIndexesRequest) (*indexpb.ListIndexesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListIndexes") + } + var r0 *indexpb.ListIndexesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.ListIndexesRequest) (*indexpb.ListIndexesResponse, error)); ok { @@ -2283,6 +2447,10 @@ func (_c *MockDataCoord_ListIndexes_Call) RunAndReturn(run func(context.Context, func (_m *MockDataCoord) ManualCompaction(_a0 context.Context, _a1 *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ManualCompaction") + } + var r0 *milvuspb.ManualCompactionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)); ok { @@ -2338,6 +2506,10 @@ func (_c *MockDataCoord_ManualCompaction_Call) RunAndReturn(run func(context.Con func (_m *MockDataCoord) MarkSegmentsDropped(_a0 context.Context, _a1 *datapb.MarkSegmentsDroppedRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for MarkSegmentsDropped") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.MarkSegmentsDroppedRequest) (*commonpb.Status, error)); ok { @@ -2393,6 +2565,10 @@ func (_c *MockDataCoord_MarkSegmentsDropped_Call) RunAndReturn(run func(context. func (_m *MockDataCoord) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -2467,6 +2643,10 @@ func (_c *MockDataCoord_RegisterStreamingCoordGRPCService_Call) RunAndReturn(run func (_m *MockDataCoord) ReportDataNodeTtMsgs(_a0 context.Context, _a1 *datapb.ReportDataNodeTtMsgsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReportDataNodeTtMsgs") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ReportDataNodeTtMsgsRequest) (*commonpb.Status, error)); ok { @@ -2522,6 +2702,10 @@ func (_c *MockDataCoord_ReportDataNodeTtMsgs_Call) RunAndReturn(run func(context func (_m *MockDataCoord) SaveBinlogPaths(_a0 context.Context, _a1 *datapb.SaveBinlogPathsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SaveBinlogPaths") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.SaveBinlogPathsRequest) (*commonpb.Status, error)); ok { @@ -2742,6 +2926,10 @@ func (_c *MockDataCoord_SetRootCoordClient_Call) RunAndReturn(run func(types.Roo func (_m *MockDataCoord) SetSegmentState(_a0 context.Context, _a1 *datapb.SetSegmentStateRequest) (*datapb.SetSegmentStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SetSegmentState") + } + var r0 *datapb.SetSegmentStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.SetSegmentStateRequest) (*datapb.SetSegmentStateResponse, error)); ok { @@ -2830,6 +3018,10 @@ func (_c *MockDataCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client) func (_m *MockDataCoord) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -2885,6 +3077,10 @@ func (_c *MockDataCoord_ShowConfigurations_Call) RunAndReturn(run func(context.C func (_m *MockDataCoord) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -2926,6 +3122,10 @@ func (_c *MockDataCoord_Start_Call) RunAndReturn(run func() error) *MockDataCoor func (_m *MockDataCoord) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -2967,6 +3167,10 @@ func (_c *MockDataCoord_Stop_Call) RunAndReturn(run func() error) *MockDataCoord func (_m *MockDataCoord) UpdateChannelCheckpoint(_a0 context.Context, _a1 *datapb.UpdateChannelCheckpointRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateChannelCheckpoint") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateChannelCheckpointRequest) (*commonpb.Status, error)); ok { @@ -3022,6 +3226,10 @@ func (_c *MockDataCoord_UpdateChannelCheckpoint_Call) RunAndReturn(run func(cont func (_m *MockDataCoord) UpdateSegmentStatistics(_a0 context.Context, _a1 *datapb.UpdateSegmentStatisticsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateSegmentStatistics") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateSegmentStatisticsRequest) (*commonpb.Status, error)); ok { @@ -3077,6 +3285,10 @@ func (_c *MockDataCoord_UpdateSegmentStatistics_Call) RunAndReturn(run func(cont func (_m *MockDataCoord) WatchChannels(_a0 context.Context, _a1 *datapb.WatchChannelsRequest) (*datapb.WatchChannelsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for WatchChannels") + } + var r0 *datapb.WatchChannelsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchChannelsRequest) (*datapb.WatchChannelsResponse, error)); ok { diff --git a/internal/mocks/mock_datacoord_client.go b/internal/mocks/mock_datacoord_client.go index 2a989a1876..9f42a0f953 100644 --- a/internal/mocks/mock_datacoord_client.go +++ b/internal/mocks/mock_datacoord_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -44,6 +44,10 @@ func (_m *MockDataCoordClient) AllocSegment(ctx context.Context, in *datapb.Allo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AllocSegment") + } + var r0 *datapb.AllocSegmentResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.AllocSegmentRequest, ...grpc.CallOption) (*datapb.AllocSegmentResponse, error)); ok { @@ -114,6 +118,10 @@ func (_m *MockDataCoordClient) AlterIndex(ctx context.Context, in *indexpb.Alter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AlterIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.AlterIndexRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -184,6 +192,10 @@ func (_m *MockDataCoordClient) AssignSegmentID(ctx context.Context, in *datapb.A _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AssignSegmentID") + } + var r0 *datapb.AssignSegmentIDResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.AssignSegmentIDRequest, ...grpc.CallOption) (*datapb.AssignSegmentIDResponse, error)); ok { @@ -254,6 +266,10 @@ func (_m *MockDataCoordClient) BroadcastAlteredCollection(ctx context.Context, i _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for BroadcastAlteredCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.AlterCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -324,6 +340,10 @@ func (_m *MockDataCoordClient) CheckHealth(ctx context.Context, in *milvuspb.Che _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest, ...grpc.CallOption) (*milvuspb.CheckHealthResponse, error)); ok { @@ -387,6 +407,10 @@ func (_c *MockDataCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co func (_m *MockDataCoordClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -435,6 +459,10 @@ func (_m *MockDataCoordClient) CreateIndex(ctx context.Context, in *indexpb.Crea _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.CreateIndexRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -505,6 +533,10 @@ func (_m *MockDataCoordClient) DescribeIndex(ctx context.Context, in *indexpb.De _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DescribeIndex") + } + var r0 *indexpb.DescribeIndexResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DescribeIndexRequest, ...grpc.CallOption) (*indexpb.DescribeIndexResponse, error)); ok { @@ -575,6 +607,10 @@ func (_m *MockDataCoordClient) DropIndex(ctx context.Context, in *indexpb.DropIn _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.DropIndexRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -645,6 +681,10 @@ func (_m *MockDataCoordClient) DropVirtualChannel(ctx context.Context, in *datap _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropVirtualChannel") + } + var r0 *datapb.DropVirtualChannelResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropVirtualChannelRequest, ...grpc.CallOption) (*datapb.DropVirtualChannelResponse, error)); ok { @@ -715,6 +755,10 @@ func (_m *MockDataCoordClient) Flush(ctx context.Context, in *datapb.FlushReques _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Flush") + } + var r0 *datapb.FlushResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushRequest, ...grpc.CallOption) (*datapb.FlushResponse, error)); ok { @@ -785,6 +829,10 @@ func (_m *MockDataCoordClient) GcConfirm(ctx context.Context, in *datapb.GcConfi _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GcConfirm") + } + var r0 *datapb.GcConfirmResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcConfirmRequest, ...grpc.CallOption) (*datapb.GcConfirmResponse, error)); ok { @@ -855,6 +903,10 @@ func (_m *MockDataCoordClient) GcControl(ctx context.Context, in *datapb.GcContr _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GcControl") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GcControlRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -925,6 +977,10 @@ func (_m *MockDataCoordClient) GetChannelRecoveryInfo(ctx context.Context, in *d _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetChannelRecoveryInfo") + } + var r0 *datapb.GetChannelRecoveryInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetChannelRecoveryInfoRequest, ...grpc.CallOption) (*datapb.GetChannelRecoveryInfoResponse, error)); ok { @@ -995,6 +1051,10 @@ func (_m *MockDataCoordClient) GetCollectionStatistics(ctx context.Context, in * _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetCollectionStatistics") + } + var r0 *datapb.GetCollectionStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetCollectionStatisticsRequest, ...grpc.CallOption) (*datapb.GetCollectionStatisticsResponse, error)); ok { @@ -1065,6 +1125,10 @@ func (_m *MockDataCoordClient) GetCompactionState(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetCompactionState") + } + var r0 *milvuspb.GetCompactionStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest, ...grpc.CallOption) (*milvuspb.GetCompactionStateResponse, error)); ok { @@ -1135,6 +1199,10 @@ func (_m *MockDataCoordClient) GetCompactionStateWithPlans(ctx context.Context, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetCompactionStateWithPlans") + } + var r0 *milvuspb.GetCompactionPlansResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest, ...grpc.CallOption) (*milvuspb.GetCompactionPlansResponse, error)); ok { @@ -1205,6 +1273,10 @@ func (_m *MockDataCoordClient) GetComponentStates(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -1275,6 +1347,10 @@ func (_m *MockDataCoordClient) GetFlushAllState(ctx context.Context, in *milvusp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetFlushAllState") + } + var r0 *milvuspb.GetFlushAllStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest, ...grpc.CallOption) (*milvuspb.GetFlushAllStateResponse, error)); ok { @@ -1345,6 +1421,10 @@ func (_m *MockDataCoordClient) GetFlushState(ctx context.Context, in *datapb.Get _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetFlushState") + } + var r0 *milvuspb.GetFlushStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushStateRequest, ...grpc.CallOption) (*milvuspb.GetFlushStateResponse, error)); ok { @@ -1415,6 +1495,10 @@ func (_m *MockDataCoordClient) GetFlushedSegments(ctx context.Context, in *datap _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetFlushedSegments") + } + var r0 *datapb.GetFlushedSegmentsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetFlushedSegmentsRequest, ...grpc.CallOption) (*datapb.GetFlushedSegmentsResponse, error)); ok { @@ -1485,6 +1569,10 @@ func (_m *MockDataCoordClient) GetImportProgress(ctx context.Context, in *intern _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetImportProgress") + } + var r0 *internalpb.GetImportProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error)); ok { @@ -1555,6 +1643,10 @@ func (_m *MockDataCoordClient) GetIndexBuildProgress(ctx context.Context, in *in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetIndexBuildProgress") + } + var r0 *indexpb.GetIndexBuildProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexBuildProgressRequest, ...grpc.CallOption) (*indexpb.GetIndexBuildProgressResponse, error)); ok { @@ -1625,6 +1717,10 @@ func (_m *MockDataCoordClient) GetIndexInfos(ctx context.Context, in *indexpb.Ge _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetIndexInfos") + } + var r0 *indexpb.GetIndexInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexInfoRequest, ...grpc.CallOption) (*indexpb.GetIndexInfoResponse, error)); ok { @@ -1695,6 +1791,10 @@ func (_m *MockDataCoordClient) GetIndexState(ctx context.Context, in *indexpb.Ge _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetIndexState") + } + var r0 *indexpb.GetIndexStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStateRequest, ...grpc.CallOption) (*indexpb.GetIndexStateResponse, error)); ok { @@ -1765,6 +1865,10 @@ func (_m *MockDataCoordClient) GetIndexStatistics(ctx context.Context, in *index _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetIndexStatistics") + } + var r0 *indexpb.GetIndexStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetIndexStatisticsRequest, ...grpc.CallOption) (*indexpb.GetIndexStatisticsResponse, error)); ok { @@ -1835,6 +1939,10 @@ func (_m *MockDataCoordClient) GetInsertBinlogPaths(ctx context.Context, in *dat _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetInsertBinlogPaths") + } + var r0 *datapb.GetInsertBinlogPathsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetInsertBinlogPathsRequest, ...grpc.CallOption) (*datapb.GetInsertBinlogPathsResponse, error)); ok { @@ -1905,6 +2013,10 @@ func (_m *MockDataCoordClient) GetMetrics(ctx context.Context, in *milvuspb.GetM _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -1975,6 +2087,10 @@ func (_m *MockDataCoordClient) GetPartitionStatistics(ctx context.Context, in *d _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetPartitionStatistics") + } + var r0 *datapb.GetPartitionStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetPartitionStatisticsRequest, ...grpc.CallOption) (*datapb.GetPartitionStatisticsResponse, error)); ok { @@ -2045,6 +2161,10 @@ func (_m *MockDataCoordClient) GetRecoveryInfo(ctx context.Context, in *datapb.G _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetRecoveryInfo") + } + var r0 *datapb.GetRecoveryInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequest, ...grpc.CallOption) (*datapb.GetRecoveryInfoResponse, error)); ok { @@ -2115,6 +2235,10 @@ func (_m *MockDataCoordClient) GetRecoveryInfoV2(ctx context.Context, in *datapb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetRecoveryInfoV2") + } + var r0 *datapb.GetRecoveryInfoResponseV2 var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetRecoveryInfoRequestV2, ...grpc.CallOption) (*datapb.GetRecoveryInfoResponseV2, error)); ok { @@ -2185,6 +2309,10 @@ func (_m *MockDataCoordClient) GetSegmentIndexState(ctx context.Context, in *ind _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentIndexState") + } + var r0 *indexpb.GetSegmentIndexStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.GetSegmentIndexStateRequest, ...grpc.CallOption) (*indexpb.GetSegmentIndexStateResponse, error)); ok { @@ -2255,6 +2383,10 @@ func (_m *MockDataCoordClient) GetSegmentInfo(ctx context.Context, in *datapb.Ge _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *datapb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoRequest, ...grpc.CallOption) (*datapb.GetSegmentInfoResponse, error)); ok { @@ -2325,6 +2457,10 @@ func (_m *MockDataCoordClient) GetSegmentInfoChannel(ctx context.Context, in *da _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfoChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentInfoChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -2395,6 +2531,10 @@ func (_m *MockDataCoordClient) GetSegmentStates(ctx context.Context, in *datapb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentStates") + } + var r0 *datapb.GetSegmentStatesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentStatesRequest, ...grpc.CallOption) (*datapb.GetSegmentStatesResponse, error)); ok { @@ -2465,6 +2605,10 @@ func (_m *MockDataCoordClient) GetSegmentsByStates(ctx context.Context, in *data _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentsByStates") + } + var r0 *datapb.GetSegmentsByStatesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.GetSegmentsByStatesRequest, ...grpc.CallOption) (*datapb.GetSegmentsByStatesResponse, error)); ok { @@ -2535,6 +2679,10 @@ func (_m *MockDataCoordClient) GetStatisticsChannel(ctx context.Context, in *int _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -2605,6 +2753,10 @@ func (_m *MockDataCoordClient) GetTimeTickChannel(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -2675,6 +2827,10 @@ func (_m *MockDataCoordClient) ImportV2(ctx context.Context, in *internalpb.Impo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 *internalpb.ImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequestInternal, ...grpc.CallOption) (*internalpb.ImportResponse, error)); ok { @@ -2745,6 +2901,10 @@ func (_m *MockDataCoordClient) ListImports(ctx context.Context, in *internalpb.L _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListImports") + } + var r0 *internalpb.ListImportsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequestInternal, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok { @@ -2815,6 +2975,10 @@ func (_m *MockDataCoordClient) ListIndexes(ctx context.Context, in *indexpb.List _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListIndexes") + } + var r0 *indexpb.ListIndexesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *indexpb.ListIndexesRequest, ...grpc.CallOption) (*indexpb.ListIndexesResponse, error)); ok { @@ -2885,6 +3049,10 @@ func (_m *MockDataCoordClient) ManualCompaction(ctx context.Context, in *milvusp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ManualCompaction") + } + var r0 *milvuspb.ManualCompactionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest, ...grpc.CallOption) (*milvuspb.ManualCompactionResponse, error)); ok { @@ -2955,6 +3123,10 @@ func (_m *MockDataCoordClient) MarkSegmentsDropped(ctx context.Context, in *data _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MarkSegmentsDropped") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.MarkSegmentsDroppedRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -3025,6 +3197,10 @@ func (_m *MockDataCoordClient) ReportDataNodeTtMsgs(ctx context.Context, in *dat _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ReportDataNodeTtMsgs") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ReportDataNodeTtMsgsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -3095,6 +3271,10 @@ func (_m *MockDataCoordClient) SaveBinlogPaths(ctx context.Context, in *datapb.S _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SaveBinlogPaths") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.SaveBinlogPathsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -3165,6 +3345,10 @@ func (_m *MockDataCoordClient) SetSegmentState(ctx context.Context, in *datapb.S _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SetSegmentState") + } + var r0 *datapb.SetSegmentStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.SetSegmentStateRequest, ...grpc.CallOption) (*datapb.SetSegmentStateResponse, error)); ok { @@ -3235,6 +3419,10 @@ func (_m *MockDataCoordClient) ShowConfigurations(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -3305,6 +3493,10 @@ func (_m *MockDataCoordClient) UpdateChannelCheckpoint(ctx context.Context, in * _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateChannelCheckpoint") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateChannelCheckpointRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -3375,6 +3567,10 @@ func (_m *MockDataCoordClient) UpdateSegmentStatistics(ctx context.Context, in * _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateSegmentStatistics") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.UpdateSegmentStatisticsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -3445,6 +3641,10 @@ func (_m *MockDataCoordClient) WatchChannels(ctx context.Context, in *datapb.Wat _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for WatchChannels") + } + var r0 *datapb.WatchChannelsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchChannelsRequest, ...grpc.CallOption) (*datapb.WatchChannelsResponse, error)); ok { diff --git a/internal/mocks/mock_datanode.go b/internal/mocks/mock_datanode.go index 190da75be0..f03f66c8af 100644 --- a/internal/mocks/mock_datanode.go +++ b/internal/mocks/mock_datanode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -36,6 +36,10 @@ func (_m *MockDataNode) EXPECT() *MockDataNode_Expecter { func (_m *MockDataNode) CheckChannelOperationProgress(_a0 context.Context, _a1 *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CheckChannelOperationProgress") + } + var r0 *datapb.ChannelOperationProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelWatchInfo) (*datapb.ChannelOperationProgressResponse, error)); ok { @@ -91,6 +95,10 @@ func (_c *MockDataNode_CheckChannelOperationProgress_Call) RunAndReturn(run func func (_m *MockDataNode) CompactionV2(_a0 context.Context, _a1 *datapb.CompactionPlan) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CompactionV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionPlan) (*commonpb.Status, error)); ok { @@ -146,6 +154,10 @@ func (_c *MockDataNode_CompactionV2_Call) RunAndReturn(run func(context.Context, func (_m *MockDataNode) DropCompactionPlan(_a0 context.Context, _a1 *datapb.DropCompactionPlanRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropCompactionPlan") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropCompactionPlanRequest) (*commonpb.Status, error)); ok { @@ -201,6 +213,10 @@ func (_c *MockDataNode_DropCompactionPlan_Call) RunAndReturn(run func(context.Co func (_m *MockDataNode) DropImport(_a0 context.Context, _a1 *datapb.DropImportRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropImport") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropImportRequest) (*commonpb.Status, error)); ok { @@ -256,6 +272,10 @@ func (_c *MockDataNode_DropImport_Call) RunAndReturn(run func(context.Context, * func (_m *MockDataNode) FlushChannels(_a0 context.Context, _a1 *datapb.FlushChannelsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for FlushChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushChannelsRequest) (*commonpb.Status, error)); ok { @@ -311,6 +331,10 @@ func (_c *MockDataNode_FlushChannels_Call) RunAndReturn(run func(context.Context func (_m *MockDataNode) FlushSegments(_a0 context.Context, _a1 *datapb.FlushSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for FlushSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushSegmentsRequest) (*commonpb.Status, error)); ok { @@ -366,6 +390,10 @@ func (_c *MockDataNode_FlushSegments_Call) RunAndReturn(run func(context.Context func (_m *MockDataNode) GetAddress() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAddress") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -407,6 +435,10 @@ func (_c *MockDataNode_GetAddress_Call) RunAndReturn(run func() string) *MockDat func (_m *MockDataNode) GetCompactionState(_a0 context.Context, _a1 *datapb.CompactionStateRequest) (*datapb.CompactionStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCompactionState") + } + var r0 *datapb.CompactionStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionStateRequest) (*datapb.CompactionStateResponse, error)); ok { @@ -462,6 +494,10 @@ func (_c *MockDataNode_GetCompactionState_Call) RunAndReturn(run func(context.Co func (_m *MockDataNode) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -517,6 +553,10 @@ func (_c *MockDataNode_GetComponentStates_Call) RunAndReturn(run func(context.Co func (_m *MockDataNode) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -572,6 +612,10 @@ func (_c *MockDataNode_GetMetrics_Call) RunAndReturn(run func(context.Context, * func (_m *MockDataNode) GetNodeID() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetNodeID") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -613,6 +657,10 @@ func (_c *MockDataNode_GetNodeID_Call) RunAndReturn(run func() int64) *MockDataN func (_m *MockDataNode) GetStateCode() commonpb.StateCode { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetStateCode") + } + var r0 commonpb.StateCode if rf, ok := ret.Get(0).(func() commonpb.StateCode); ok { r0 = rf() @@ -654,6 +702,10 @@ func (_c *MockDataNode_GetStateCode_Call) RunAndReturn(run func() commonpb.State func (_m *MockDataNode) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -709,6 +761,10 @@ func (_c *MockDataNode_GetStatisticsChannel_Call) RunAndReturn(run func(context. func (_m *MockDataNode) ImportV2(_a0 context.Context, _a1 *datapb.ImportRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ImportRequest) (*commonpb.Status, error)); ok { @@ -764,6 +820,10 @@ func (_c *MockDataNode_ImportV2_Call) RunAndReturn(run func(context.Context, *da func (_m *MockDataNode) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -805,6 +865,10 @@ func (_c *MockDataNode_Init_Call) RunAndReturn(run func() error) *MockDataNode_I func (_m *MockDataNode) NotifyChannelOperation(_a0 context.Context, _a1 *datapb.ChannelOperationsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for NotifyChannelOperation") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelOperationsRequest) (*commonpb.Status, error)); ok { @@ -860,6 +924,10 @@ func (_c *MockDataNode_NotifyChannelOperation_Call) RunAndReturn(run func(contex func (_m *MockDataNode) PreImport(_a0 context.Context, _a1 *datapb.PreImportRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for PreImport") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.PreImportRequest) (*commonpb.Status, error)); ok { @@ -915,6 +983,10 @@ func (_c *MockDataNode_PreImport_Call) RunAndReturn(run func(context.Context, *d func (_m *MockDataNode) QueryImport(_a0 context.Context, _a1 *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryImport") + } + var r0 *datapb.QueryImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryImportRequest) (*datapb.QueryImportResponse, error)); ok { @@ -970,6 +1042,10 @@ func (_c *MockDataNode_QueryImport_Call) RunAndReturn(run func(context.Context, func (_m *MockDataNode) QueryPreImport(_a0 context.Context, _a1 *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryPreImport") + } + var r0 *datapb.QueryPreImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryPreImportRequest) (*datapb.QueryPreImportResponse, error)); ok { @@ -1025,6 +1101,10 @@ func (_c *MockDataNode_QueryPreImport_Call) RunAndReturn(run func(context.Contex func (_m *MockDataNode) QuerySlot(_a0 context.Context, _a1 *datapb.QuerySlotRequest) (*datapb.QuerySlotResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QuerySlot") + } + var r0 *datapb.QuerySlotResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.QuerySlotRequest) (*datapb.QuerySlotResponse, error)); ok { @@ -1080,6 +1160,10 @@ func (_c *MockDataNode_QuerySlot_Call) RunAndReturn(run func(context.Context, *d func (_m *MockDataNode) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1121,6 +1205,10 @@ func (_c *MockDataNode_Register_Call) RunAndReturn(run func() error) *MockDataNo func (_m *MockDataNode) ResendSegmentStats(_a0 context.Context, _a1 *datapb.ResendSegmentStatsRequest) (*datapb.ResendSegmentStatsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ResendSegmentStats") + } + var r0 *datapb.ResendSegmentStatsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ResendSegmentStatsRequest) (*datapb.ResendSegmentStatsResponse, error)); ok { @@ -1209,6 +1297,10 @@ func (_c *MockDataNode_SetAddress_Call) RunAndReturn(run func(string)) *MockData func (_m *MockDataNode) SetDataCoordClient(dataCoord types.DataCoordClient) error { ret := _m.Called(dataCoord) + if len(ret) == 0 { + panic("no return value specified for SetDataCoordClient") + } + var r0 error if rf, ok := ret.Get(0).(func(types.DataCoordClient) error); ok { r0 = rf(dataCoord) @@ -1284,6 +1376,10 @@ func (_c *MockDataNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Clien func (_m *MockDataNode) SetRootCoordClient(rootCoord types.RootCoordClient) error { ret := _m.Called(rootCoord) + if len(ret) == 0 { + panic("no return value specified for SetRootCoordClient") + } + var r0 error if rf, ok := ret.Get(0).(func(types.RootCoordClient) error); ok { r0 = rf(rootCoord) @@ -1326,6 +1422,10 @@ func (_c *MockDataNode_SetRootCoordClient_Call) RunAndReturn(run func(types.Root func (_m *MockDataNode) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -1381,6 +1481,10 @@ func (_c *MockDataNode_ShowConfigurations_Call) RunAndReturn(run func(context.Co func (_m *MockDataNode) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1422,6 +1526,10 @@ func (_c *MockDataNode_Start_Call) RunAndReturn(run func() error) *MockDataNode_ func (_m *MockDataNode) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1463,6 +1571,10 @@ func (_c *MockDataNode_Stop_Call) RunAndReturn(run func() error) *MockDataNode_S func (_m *MockDataNode) SyncSegments(_a0 context.Context, _a1 *datapb.SyncSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SyncSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.SyncSegmentsRequest) (*commonpb.Status, error)); ok { @@ -1551,6 +1663,10 @@ func (_c *MockDataNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.Stat func (_m *MockDataNode) WatchDmChannels(_a0 context.Context, _a1 *datapb.WatchDmChannelsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for WatchDmChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_datanode_client.go b/internal/mocks/mock_datanode_client.go index 91661051c3..fc493844fb 100644 --- a/internal/mocks/mock_datanode_client.go +++ b/internal/mocks/mock_datanode_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -42,6 +42,10 @@ func (_m *MockDataNodeClient) CheckChannelOperationProgress(ctx context.Context, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CheckChannelOperationProgress") + } + var r0 *datapb.ChannelOperationProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelWatchInfo, ...grpc.CallOption) (*datapb.ChannelOperationProgressResponse, error)); ok { @@ -105,6 +109,10 @@ func (_c *MockDataNodeClient_CheckChannelOperationProgress_Call) RunAndReturn(ru func (_m *MockDataNodeClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -153,6 +161,10 @@ func (_m *MockDataNodeClient) CompactionV2(ctx context.Context, in *datapb.Compa _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CompactionV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionPlan, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -223,6 +235,10 @@ func (_m *MockDataNodeClient) DropCompactionPlan(ctx context.Context, in *datapb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropCompactionPlan") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropCompactionPlanRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -293,6 +309,10 @@ func (_m *MockDataNodeClient) DropImport(ctx context.Context, in *datapb.DropImp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropImport") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.DropImportRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -363,6 +383,10 @@ func (_m *MockDataNodeClient) FlushChannels(ctx context.Context, in *datapb.Flus _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for FlushChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushChannelsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -433,6 +457,10 @@ func (_m *MockDataNodeClient) FlushSegments(ctx context.Context, in *datapb.Flus _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for FlushSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.FlushSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -503,6 +531,10 @@ func (_m *MockDataNodeClient) GetCompactionState(ctx context.Context, in *datapb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetCompactionState") + } + var r0 *datapb.CompactionStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.CompactionStateRequest, ...grpc.CallOption) (*datapb.CompactionStateResponse, error)); ok { @@ -573,6 +605,10 @@ func (_m *MockDataNodeClient) GetComponentStates(ctx context.Context, in *milvus _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -643,6 +679,10 @@ func (_m *MockDataNodeClient) GetMetrics(ctx context.Context, in *milvuspb.GetMe _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -713,6 +753,10 @@ func (_m *MockDataNodeClient) GetStatisticsChannel(ctx context.Context, in *inte _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -783,6 +827,10 @@ func (_m *MockDataNodeClient) ImportV2(ctx context.Context, in *datapb.ImportReq _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ImportRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -853,6 +901,10 @@ func (_m *MockDataNodeClient) NotifyChannelOperation(ctx context.Context, in *da _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for NotifyChannelOperation") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ChannelOperationsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -923,6 +975,10 @@ func (_m *MockDataNodeClient) PreImport(ctx context.Context, in *datapb.PreImpor _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for PreImport") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.PreImportRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -993,6 +1049,10 @@ func (_m *MockDataNodeClient) QueryImport(ctx context.Context, in *datapb.QueryI _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QueryImport") + } + var r0 *datapb.QueryImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryImportRequest, ...grpc.CallOption) (*datapb.QueryImportResponse, error)); ok { @@ -1063,6 +1123,10 @@ func (_m *MockDataNodeClient) QueryPreImport(ctx context.Context, in *datapb.Que _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QueryPreImport") + } + var r0 *datapb.QueryPreImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.QueryPreImportRequest, ...grpc.CallOption) (*datapb.QueryPreImportResponse, error)); ok { @@ -1133,6 +1197,10 @@ func (_m *MockDataNodeClient) QuerySlot(ctx context.Context, in *datapb.QuerySlo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QuerySlot") + } + var r0 *datapb.QuerySlotResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.QuerySlotRequest, ...grpc.CallOption) (*datapb.QuerySlotResponse, error)); ok { @@ -1203,6 +1271,10 @@ func (_m *MockDataNodeClient) ResendSegmentStats(ctx context.Context, in *datapb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ResendSegmentStats") + } + var r0 *datapb.ResendSegmentStatsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.ResendSegmentStatsRequest, ...grpc.CallOption) (*datapb.ResendSegmentStatsResponse, error)); ok { @@ -1273,6 +1345,10 @@ func (_m *MockDataNodeClient) ShowConfigurations(ctx context.Context, in *intern _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -1343,6 +1419,10 @@ func (_m *MockDataNodeClient) SyncSegments(ctx context.Context, in *datapb.SyncS _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SyncSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.SyncSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1413,6 +1493,10 @@ func (_m *MockDataNodeClient) WatchDmChannels(ctx context.Context, in *datapb.Wa _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for WatchDmChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *datapb.WatchDmChannelsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_indexnode.go b/internal/mocks/mock_indexnode.go index f4060a471c..31ab4f3572 100644 --- a/internal/mocks/mock_indexnode.go +++ b/internal/mocks/mock_indexnode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -34,6 +34,10 @@ func (_m *MockIndexNode) EXPECT() *MockIndexNode_Expecter { func (_m *MockIndexNode) CreateJob(_a0 context.Context, _a1 *workerpb.CreateJobRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateJob") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobRequest) (*commonpb.Status, error)); ok { @@ -89,6 +93,10 @@ func (_c *MockIndexNode_CreateJob_Call) RunAndReturn(run func(context.Context, * func (_m *MockIndexNode) CreateJobV2(_a0 context.Context, _a1 *workerpb.CreateJobV2Request) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateJobV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobV2Request) (*commonpb.Status, error)); ok { @@ -144,6 +152,10 @@ func (_c *MockIndexNode_CreateJobV2_Call) RunAndReturn(run func(context.Context, func (_m *MockIndexNode) DropJobs(_a0 context.Context, _a1 *workerpb.DropJobsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropJobs") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsRequest) (*commonpb.Status, error)); ok { @@ -199,6 +211,10 @@ func (_c *MockIndexNode_DropJobs_Call) RunAndReturn(run func(context.Context, *w func (_m *MockIndexNode) DropJobsV2(_a0 context.Context, _a1 *workerpb.DropJobsV2Request) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropJobsV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsV2Request) (*commonpb.Status, error)); ok { @@ -254,6 +270,10 @@ func (_c *MockIndexNode_DropJobsV2_Call) RunAndReturn(run func(context.Context, func (_m *MockIndexNode) GetAddress() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAddress") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -295,6 +315,10 @@ func (_c *MockIndexNode_GetAddress_Call) RunAndReturn(run func() string) *MockIn func (_m *MockIndexNode) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -350,6 +374,10 @@ func (_c *MockIndexNode_GetComponentStates_Call) RunAndReturn(run func(context.C func (_m *MockIndexNode) GetJobStats(_a0 context.Context, _a1 *workerpb.GetJobStatsRequest) (*workerpb.GetJobStatsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetJobStats") + } + var r0 *workerpb.GetJobStatsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.GetJobStatsRequest) (*workerpb.GetJobStatsResponse, error)); ok { @@ -405,6 +433,10 @@ func (_c *MockIndexNode_GetJobStats_Call) RunAndReturn(run func(context.Context, func (_m *MockIndexNode) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -460,6 +492,10 @@ func (_c *MockIndexNode_GetMetrics_Call) RunAndReturn(run func(context.Context, func (_m *MockIndexNode) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -515,6 +551,10 @@ func (_c *MockIndexNode_GetStatisticsChannel_Call) RunAndReturn(run func(context func (_m *MockIndexNode) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -556,6 +596,10 @@ func (_c *MockIndexNode_Init_Call) RunAndReturn(run func() error) *MockIndexNode func (_m *MockIndexNode) QueryJobs(_a0 context.Context, _a1 *workerpb.QueryJobsRequest) (*workerpb.QueryJobsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryJobs") + } + var r0 *workerpb.QueryJobsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsRequest) (*workerpb.QueryJobsResponse, error)); ok { @@ -611,6 +655,10 @@ func (_c *MockIndexNode_QueryJobs_Call) RunAndReturn(run func(context.Context, * func (_m *MockIndexNode) QueryJobsV2(_a0 context.Context, _a1 *workerpb.QueryJobsV2Request) (*workerpb.QueryJobsV2Response, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryJobsV2") + } + var r0 *workerpb.QueryJobsV2Response var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsV2Request) (*workerpb.QueryJobsV2Response, error)); ok { @@ -666,6 +714,10 @@ func (_c *MockIndexNode_QueryJobsV2_Call) RunAndReturn(run func(context.Context, func (_m *MockIndexNode) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -773,6 +825,10 @@ func (_c *MockIndexNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Clie func (_m *MockIndexNode) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -828,6 +884,10 @@ func (_c *MockIndexNode_ShowConfigurations_Call) RunAndReturn(run func(context.C func (_m *MockIndexNode) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -869,6 +929,10 @@ func (_c *MockIndexNode_Start_Call) RunAndReturn(run func() error) *MockIndexNod func (_m *MockIndexNode) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() diff --git a/internal/mocks/mock_indexnode_client.go b/internal/mocks/mock_indexnode_client.go index 367a11af02..6e9f8bc81b 100644 --- a/internal/mocks/mock_indexnode_client.go +++ b/internal/mocks/mock_indexnode_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -35,6 +35,10 @@ func (_m *MockIndexNodeClient) EXPECT() *MockIndexNodeClient_Expecter { func (_m *MockIndexNodeClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -83,6 +87,10 @@ func (_m *MockIndexNodeClient) CreateJob(ctx context.Context, in *workerpb.Creat _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateJob") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -153,6 +161,10 @@ func (_m *MockIndexNodeClient) CreateJobV2(ctx context.Context, in *workerpb.Cre _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateJobV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.CreateJobV2Request, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -223,6 +235,10 @@ func (_m *MockIndexNodeClient) DropJobs(ctx context.Context, in *workerpb.DropJo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropJobs") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -293,6 +309,10 @@ func (_m *MockIndexNodeClient) DropJobsV2(ctx context.Context, in *workerpb.Drop _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropJobsV2") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.DropJobsV2Request, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -363,6 +383,10 @@ func (_m *MockIndexNodeClient) GetComponentStates(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -433,6 +457,10 @@ func (_m *MockIndexNodeClient) GetJobStats(ctx context.Context, in *workerpb.Get _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetJobStats") + } + var r0 *workerpb.GetJobStatsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.GetJobStatsRequest, ...grpc.CallOption) (*workerpb.GetJobStatsResponse, error)); ok { @@ -503,6 +531,10 @@ func (_m *MockIndexNodeClient) GetMetrics(ctx context.Context, in *milvuspb.GetM _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -573,6 +605,10 @@ func (_m *MockIndexNodeClient) GetStatisticsChannel(ctx context.Context, in *int _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -643,6 +679,10 @@ func (_m *MockIndexNodeClient) QueryJobs(ctx context.Context, in *workerpb.Query _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QueryJobs") + } + var r0 *workerpb.QueryJobsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsRequest, ...grpc.CallOption) (*workerpb.QueryJobsResponse, error)); ok { @@ -713,6 +753,10 @@ func (_m *MockIndexNodeClient) QueryJobsV2(ctx context.Context, in *workerpb.Que _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QueryJobsV2") + } + var r0 *workerpb.QueryJobsV2Response var r1 error if rf, ok := ret.Get(0).(func(context.Context, *workerpb.QueryJobsV2Request, ...grpc.CallOption) (*workerpb.QueryJobsV2Response, error)); ok { @@ -783,6 +827,10 @@ func (_m *MockIndexNodeClient) ShowConfigurations(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok { diff --git a/internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go b/internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go index a0095b916f..b0bc3b7775 100644 --- a/internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go +++ b/internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_metastore @@ -27,6 +27,10 @@ func (_m *MockStreamingCoordCataLog) EXPECT() *MockStreamingCoordCataLog_Expecte func (_m *MockStreamingCoordCataLog) ListPChannel(ctx context.Context) ([]*streamingpb.PChannelMeta, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for ListPChannel") + } + var r0 []*streamingpb.PChannelMeta var r1 error if rf, ok := ret.Get(0).(func(context.Context) ([]*streamingpb.PChannelMeta, error)); ok { @@ -81,6 +85,10 @@ func (_c *MockStreamingCoordCataLog_ListPChannel_Call) RunAndReturn(run func(con func (_m *MockStreamingCoordCataLog) SavePChannels(ctx context.Context, info []*streamingpb.PChannelMeta) error { ret := _m.Called(ctx, info) + if len(ret) == 0 { + panic("no return value specified for SavePChannels") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []*streamingpb.PChannelMeta) error); ok { r0 = rf(ctx, info) diff --git a/internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go b/internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go index b6e802f2b9..b611f79f84 100644 --- a/internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go +++ b/internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_metastore @@ -27,6 +27,10 @@ func (_m *MockStreamingNodeCataLog) EXPECT() *MockStreamingNodeCataLog_Expecter func (_m *MockStreamingNodeCataLog) ListSegmentAssignment(ctx context.Context, pChannelName string) ([]*streamingpb.SegmentAssignmentMeta, error) { ret := _m.Called(ctx, pChannelName) + if len(ret) == 0 { + panic("no return value specified for ListSegmentAssignment") + } + var r0 []*streamingpb.SegmentAssignmentMeta var r1 error if rf, ok := ret.Get(0).(func(context.Context, string) ([]*streamingpb.SegmentAssignmentMeta, error)); ok { @@ -82,6 +86,10 @@ func (_c *MockStreamingNodeCataLog_ListSegmentAssignment_Call) RunAndReturn(run func (_m *MockStreamingNodeCataLog) SaveSegmentAssignments(ctx context.Context, pChannelName string, infos []*streamingpb.SegmentAssignmentMeta) error { ret := _m.Called(ctx, pChannelName, infos) + if len(ret) == 0 { + panic("no return value specified for SaveSegmentAssignments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, []*streamingpb.SegmentAssignmentMeta) error); ok { r0 = rf(ctx, pChannelName, infos) diff --git a/internal/mocks/mock_proxy.go b/internal/mocks/mock_proxy.go index 36555c92ef..ce6fccf7c1 100644 --- a/internal/mocks/mock_proxy.go +++ b/internal/mocks/mock_proxy.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -38,6 +38,10 @@ func (_m *MockProxy) EXPECT() *MockProxy_Expecter { func (_m *MockProxy) AllocTimestamp(_a0 context.Context, _a1 *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AllocTimestamp") + } + var r0 *milvuspb.AllocTimestampResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AllocTimestampRequest) (*milvuspb.AllocTimestampResponse, error)); ok { @@ -93,6 +97,10 @@ func (_c *MockProxy_AllocTimestamp_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) AlterAlias(_a0 context.Context, _a1 *milvuspb.AlterAliasRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)); ok { @@ -148,6 +156,10 @@ func (_c *MockProxy_AlterAlias_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) AlterCollection(_a0 context.Context, _a1 *milvuspb.AlterCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)); ok { @@ -203,6 +215,10 @@ func (_c *MockProxy_AlterCollection_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) AlterDatabase(_a0 context.Context, _a1 *milvuspb.AlterDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterDatabaseRequest) (*commonpb.Status, error)); ok { @@ -258,6 +274,10 @@ func (_c *MockProxy_AlterDatabase_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) AlterIndex(_a0 context.Context, _a1 *milvuspb.AlterIndexRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterIndexRequest) (*commonpb.Status, error)); ok { @@ -313,6 +333,10 @@ func (_c *MockProxy_AlterIndex_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) BackupRBAC(_a0 context.Context, _a1 *milvuspb.BackupRBACMetaRequest) (*milvuspb.BackupRBACMetaResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for BackupRBAC") + } + var r0 *milvuspb.BackupRBACMetaResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.BackupRBACMetaRequest) (*milvuspb.BackupRBACMetaResponse, error)); ok { @@ -368,6 +392,10 @@ func (_c *MockProxy_BackupRBAC_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) CalcDistance(_a0 context.Context, _a1 *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CalcDistance") + } + var r0 *milvuspb.CalcDistanceResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error)); ok { @@ -423,6 +451,10 @@ func (_c *MockProxy_CalcDistance_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok { @@ -478,6 +510,10 @@ func (_c *MockProxy_CheckHealth_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) Connect(_a0 context.Context, _a1 *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Connect") + } + var r0 *milvuspb.ConnectResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error)); ok { @@ -533,6 +569,10 @@ func (_c *MockProxy_Connect_Call) RunAndReturn(run func(context.Context, *milvus func (_m *MockProxy) CreateAlias(_a0 context.Context, _a1 *milvuspb.CreateAliasRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)); ok { @@ -588,6 +628,10 @@ func (_c *MockProxy_CreateAlias_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) CreateCollection(_a0 context.Context, _a1 *milvuspb.CreateCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)); ok { @@ -643,6 +687,10 @@ func (_c *MockProxy_CreateCollection_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) CreateCredential(_a0 context.Context, _a1 *milvuspb.CreateCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCredentialRequest) (*commonpb.Status, error)); ok { @@ -698,6 +746,10 @@ func (_c *MockProxy_CreateCredential_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) CreateDatabase(_a0 context.Context, _a1 *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)); ok { @@ -753,6 +805,10 @@ func (_c *MockProxy_CreateDatabase_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) CreateIndex(_a0 context.Context, _a1 *milvuspb.CreateIndexRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateIndexRequest) (*commonpb.Status, error)); ok { @@ -808,6 +864,10 @@ func (_c *MockProxy_CreateIndex_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) CreatePartition(_a0 context.Context, _a1 *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreatePartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)); ok { @@ -863,6 +923,10 @@ func (_c *MockProxy_CreatePartition_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) CreateResourceGroup(_a0 context.Context, _a1 *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateResourceGroup") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)); ok { @@ -918,6 +982,10 @@ func (_c *MockProxy_CreateResourceGroup_Call) RunAndReturn(run func(context.Cont func (_m *MockProxy) CreateRole(_a0 context.Context, _a1 *milvuspb.CreateRoleRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)); ok { @@ -973,6 +1041,10 @@ func (_c *MockProxy_CreateRole_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) Delete(_a0 context.Context, _a1 *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Delete") + } + var r0 *milvuspb.MutationResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error)); ok { @@ -1028,6 +1100,10 @@ func (_c *MockProxy_Delete_Call) RunAndReturn(run func(context.Context, *milvusp func (_m *MockProxy) DeleteCredential(_a0 context.Context, _a1 *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DeleteCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)); ok { @@ -1083,6 +1159,10 @@ func (_c *MockProxy_DeleteCredential_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) DescribeAlias(_a0 context.Context, _a1 *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeAlias") + } + var r0 *milvuspb.DescribeAliasResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error)); ok { @@ -1138,6 +1218,10 @@ func (_c *MockProxy_DescribeAlias_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) DescribeCollection(_a0 context.Context, _a1 *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeCollection") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -1193,6 +1277,10 @@ func (_c *MockProxy_DescribeCollection_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) DescribeDatabase(_a0 context.Context, _a1 *milvuspb.DescribeDatabaseRequest) (*milvuspb.DescribeDatabaseResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeDatabase") + } + var r0 *milvuspb.DescribeDatabaseResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeDatabaseRequest) (*milvuspb.DescribeDatabaseResponse, error)); ok { @@ -1248,6 +1336,10 @@ func (_c *MockProxy_DescribeDatabase_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) DescribeIndex(_a0 context.Context, _a1 *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeIndex") + } + var r0 *milvuspb.DescribeIndexResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error)); ok { @@ -1303,6 +1395,10 @@ func (_c *MockProxy_DescribeIndex_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) DescribeResourceGroup(_a0 context.Context, _a1 *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeResourceGroup") + } + var r0 *milvuspb.DescribeResourceGroupResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error)); ok { @@ -1358,6 +1454,10 @@ func (_c *MockProxy_DescribeResourceGroup_Call) RunAndReturn(run func(context.Co func (_m *MockProxy) DescribeSegmentIndexData(_a0 context.Context, _a1 *federpb.DescribeSegmentIndexDataRequest) (*federpb.DescribeSegmentIndexDataResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeSegmentIndexData") + } + var r0 *federpb.DescribeSegmentIndexDataResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *federpb.DescribeSegmentIndexDataRequest) (*federpb.DescribeSegmentIndexDataResponse, error)); ok { @@ -1413,6 +1513,10 @@ func (_c *MockProxy_DescribeSegmentIndexData_Call) RunAndReturn(run func(context func (_m *MockProxy) DropAlias(_a0 context.Context, _a1 *milvuspb.DropAliasRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)); ok { @@ -1468,6 +1572,10 @@ func (_c *MockProxy_DropAlias_Call) RunAndReturn(run func(context.Context, *milv func (_m *MockProxy) DropCollection(_a0 context.Context, _a1 *milvuspb.DropCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)); ok { @@ -1523,6 +1631,10 @@ func (_c *MockProxy_DropCollection_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) DropDatabase(_a0 context.Context, _a1 *milvuspb.DropDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)); ok { @@ -1578,6 +1690,10 @@ func (_c *MockProxy_DropDatabase_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) DropIndex(_a0 context.Context, _a1 *milvuspb.DropIndexRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropIndex") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropIndexRequest) (*commonpb.Status, error)); ok { @@ -1633,6 +1749,10 @@ func (_c *MockProxy_DropIndex_Call) RunAndReturn(run func(context.Context, *milv func (_m *MockProxy) DropPartition(_a0 context.Context, _a1 *milvuspb.DropPartitionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropPartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)); ok { @@ -1688,6 +1808,10 @@ func (_c *MockProxy_DropPartition_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) DropResourceGroup(_a0 context.Context, _a1 *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropResourceGroup") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)); ok { @@ -1743,6 +1867,10 @@ func (_c *MockProxy_DropResourceGroup_Call) RunAndReturn(run func(context.Contex func (_m *MockProxy) DropRole(_a0 context.Context, _a1 *milvuspb.DropRoleRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)); ok { @@ -1798,6 +1926,10 @@ func (_c *MockProxy_DropRole_Call) RunAndReturn(run func(context.Context, *milvu func (_m *MockProxy) Dummy(_a0 context.Context, _a1 *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Dummy") + } + var r0 *milvuspb.DummyResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error)); ok { @@ -1853,6 +1985,10 @@ func (_c *MockProxy_Dummy_Call) RunAndReturn(run func(context.Context, *milvuspb func (_m *MockProxy) Flush(_a0 context.Context, _a1 *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Flush") + } + var r0 *milvuspb.FlushResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error)); ok { @@ -1908,6 +2044,10 @@ func (_c *MockProxy_Flush_Call) RunAndReturn(run func(context.Context, *milvuspb func (_m *MockProxy) FlushAll(_a0 context.Context, _a1 *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for FlushAll") + } + var r0 *milvuspb.FlushAllResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error)); ok { @@ -1963,6 +2103,10 @@ func (_c *MockProxy_FlushAll_Call) RunAndReturn(run func(context.Context, *milvu func (_m *MockProxy) GetAddress() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAddress") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -2004,6 +2148,10 @@ func (_c *MockProxy_GetAddress_Call) RunAndReturn(run func() string) *MockProxy_ func (_m *MockProxy) GetCollectionStatistics(_a0 context.Context, _a1 *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCollectionStatistics") + } + var r0 *milvuspb.GetCollectionStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error)); ok { @@ -2059,6 +2207,10 @@ func (_c *MockProxy_GetCollectionStatistics_Call) RunAndReturn(run func(context. func (_m *MockProxy) GetCompactionState(_a0 context.Context, _a1 *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCompactionState") + } + var r0 *milvuspb.GetCompactionStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error)); ok { @@ -2114,6 +2266,10 @@ func (_c *MockProxy_GetCompactionState_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) GetCompactionStateWithPlans(_a0 context.Context, _a1 *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCompactionStateWithPlans") + } + var r0 *milvuspb.GetCompactionPlansResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error)); ok { @@ -2169,6 +2325,10 @@ func (_c *MockProxy_GetCompactionStateWithPlans_Call) RunAndReturn(run func(cont func (_m *MockProxy) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -2224,6 +2384,10 @@ func (_c *MockProxy_GetComponentStates_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) GetDdChannel(_a0 context.Context, _a1 *internalpb.GetDdChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetDdChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetDdChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -2279,6 +2443,10 @@ func (_c *MockProxy_GetDdChannel_Call) RunAndReturn(run func(context.Context, *i func (_m *MockProxy) GetFlushAllState(_a0 context.Context, _a1 *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetFlushAllState") + } + var r0 *milvuspb.GetFlushAllStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error)); ok { @@ -2334,6 +2502,10 @@ func (_c *MockProxy_GetFlushAllState_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) GetFlushState(_a0 context.Context, _a1 *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetFlushState") + } + var r0 *milvuspb.GetFlushStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error)); ok { @@ -2389,6 +2561,10 @@ func (_c *MockProxy_GetFlushState_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) GetImportProgress(_a0 context.Context, _a1 *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetImportProgress") + } + var r0 *internalpb.GetImportProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest) (*internalpb.GetImportProgressResponse, error)); ok { @@ -2444,6 +2620,10 @@ func (_c *MockProxy_GetImportProgress_Call) RunAndReturn(run func(context.Contex func (_m *MockProxy) GetImportState(_a0 context.Context, _a1 *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetImportState") + } + var r0 *milvuspb.GetImportStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error)); ok { @@ -2499,6 +2679,10 @@ func (_c *MockProxy_GetImportState_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) GetIndexBuildProgress(_a0 context.Context, _a1 *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexBuildProgress") + } + var r0 *milvuspb.GetIndexBuildProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error)); ok { @@ -2554,6 +2738,10 @@ func (_c *MockProxy_GetIndexBuildProgress_Call) RunAndReturn(run func(context.Co func (_m *MockProxy) GetIndexState(_a0 context.Context, _a1 *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexState") + } + var r0 *milvuspb.GetIndexStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error)); ok { @@ -2609,6 +2797,10 @@ func (_c *MockProxy_GetIndexState_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) GetIndexStatistics(_a0 context.Context, _a1 *milvuspb.GetIndexStatisticsRequest) (*milvuspb.GetIndexStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetIndexStatistics") + } + var r0 *milvuspb.GetIndexStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStatisticsRequest) (*milvuspb.GetIndexStatisticsResponse, error)); ok { @@ -2664,6 +2856,10 @@ func (_c *MockProxy_GetIndexStatistics_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) GetLoadState(_a0 context.Context, _a1 *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetLoadState") + } + var r0 *milvuspb.GetLoadStateResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error)); ok { @@ -2719,6 +2915,10 @@ func (_c *MockProxy_GetLoadState_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) GetLoadingProgress(_a0 context.Context, _a1 *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetLoadingProgress") + } + var r0 *milvuspb.GetLoadingProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error)); ok { @@ -2774,6 +2974,10 @@ func (_c *MockProxy_GetLoadingProgress_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -2829,6 +3033,10 @@ func (_c *MockProxy_GetMetrics_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) GetPartitionStatistics(_a0 context.Context, _a1 *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetPartitionStatistics") + } + var r0 *milvuspb.GetPartitionStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error)); ok { @@ -2884,6 +3092,10 @@ func (_c *MockProxy_GetPartitionStatistics_Call) RunAndReturn(run func(context.C func (_m *MockProxy) GetPersistentSegmentInfo(_a0 context.Context, _a1 *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetPersistentSegmentInfo") + } + var r0 *milvuspb.GetPersistentSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error)); ok { @@ -2939,6 +3151,10 @@ func (_c *MockProxy_GetPersistentSegmentInfo_Call) RunAndReturn(run func(context func (_m *MockProxy) GetProxyMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetProxyMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -2994,6 +3210,10 @@ func (_c *MockProxy_GetProxyMetrics_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) GetQuerySegmentInfo(_a0 context.Context, _a1 *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetQuerySegmentInfo") + } + var r0 *milvuspb.GetQuerySegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error)); ok { @@ -3049,6 +3269,10 @@ func (_c *MockProxy_GetQuerySegmentInfo_Call) RunAndReturn(run func(context.Cont func (_m *MockProxy) GetRateLimiter() (types.Limiter, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetRateLimiter") + } + var r0 types.Limiter var r1 error if rf, ok := ret.Get(0).(func() (types.Limiter, error)); ok { @@ -3102,6 +3326,10 @@ func (_c *MockProxy_GetRateLimiter_Call) RunAndReturn(run func() (types.Limiter, func (_m *MockProxy) GetReplicas(_a0 context.Context, _a1 *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetReplicas") + } + var r0 *milvuspb.GetReplicasResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)); ok { @@ -3157,6 +3385,10 @@ func (_c *MockProxy_GetReplicas_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -3212,6 +3444,10 @@ func (_c *MockProxy_GetStatisticsChannel_Call) RunAndReturn(run func(context.Con func (_m *MockProxy) GetVersion(_a0 context.Context, _a1 *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetVersion") + } + var r0 *milvuspb.GetVersionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetVersionRequest) (*milvuspb.GetVersionResponse, error)); ok { @@ -3267,6 +3503,10 @@ func (_c *MockProxy_GetVersion_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) HasCollection(_a0 context.Context, _a1 *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for HasCollection") + } + var r0 *milvuspb.BoolResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)); ok { @@ -3322,6 +3562,10 @@ func (_c *MockProxy_HasCollection_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) HasPartition(_a0 context.Context, _a1 *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for HasPartition") + } + var r0 *milvuspb.BoolResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)); ok { @@ -3377,6 +3621,10 @@ func (_c *MockProxy_HasPartition_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) HybridSearch(_a0 context.Context, _a1 *milvuspb.HybridSearchRequest) (*milvuspb.SearchResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for HybridSearch") + } + var r0 *milvuspb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HybridSearchRequest) (*milvuspb.SearchResults, error)); ok { @@ -3432,6 +3680,10 @@ func (_c *MockProxy_HybridSearch_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) Import(_a0 context.Context, _a1 *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Import") + } + var r0 *milvuspb.ImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error)); ok { @@ -3487,6 +3739,10 @@ func (_c *MockProxy_Import_Call) RunAndReturn(run func(context.Context, *milvusp func (_m *MockProxy) ImportV2(_a0 context.Context, _a1 *internalpb.ImportRequest) (*internalpb.ImportResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 *internalpb.ImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequest) (*internalpb.ImportResponse, error)); ok { @@ -3542,6 +3798,10 @@ func (_c *MockProxy_ImportV2_Call) RunAndReturn(run func(context.Context, *inter func (_m *MockProxy) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -3583,6 +3843,10 @@ func (_c *MockProxy_Init_Call) RunAndReturn(run func() error) *MockProxy_Init_Ca func (_m *MockProxy) Insert(_a0 context.Context, _a1 *milvuspb.InsertRequest) (*milvuspb.MutationResult, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Insert") + } + var r0 *milvuspb.MutationResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.InsertRequest) (*milvuspb.MutationResult, error)); ok { @@ -3638,6 +3902,10 @@ func (_c *MockProxy_Insert_Call) RunAndReturn(run func(context.Context, *milvusp func (_m *MockProxy) InvalidateCollectionMetaCache(_a0 context.Context, _a1 *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for InvalidateCollectionMetaCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error)); ok { @@ -3693,6 +3961,10 @@ func (_c *MockProxy_InvalidateCollectionMetaCache_Call) RunAndReturn(run func(co func (_m *MockProxy) InvalidateCredentialCache(_a0 context.Context, _a1 *proxypb.InvalidateCredCacheRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for InvalidateCredentialCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCredCacheRequest) (*commonpb.Status, error)); ok { @@ -3748,6 +4020,10 @@ func (_c *MockProxy_InvalidateCredentialCache_Call) RunAndReturn(run func(contex func (_m *MockProxy) InvalidateShardLeaderCache(_a0 context.Context, _a1 *proxypb.InvalidateShardLeaderCacheRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for InvalidateShardLeaderCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateShardLeaderCacheRequest) (*commonpb.Status, error)); ok { @@ -3803,6 +4079,10 @@ func (_c *MockProxy_InvalidateShardLeaderCache_Call) RunAndReturn(run func(conte func (_m *MockProxy) ListAliases(_a0 context.Context, _a1 *milvuspb.ListAliasesRequest) (*milvuspb.ListAliasesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListAliases") + } + var r0 *milvuspb.ListAliasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListAliasesRequest) (*milvuspb.ListAliasesResponse, error)); ok { @@ -3858,6 +4138,10 @@ func (_c *MockProxy_ListAliases_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) ListClientInfos(_a0 context.Context, _a1 *proxypb.ListClientInfosRequest) (*proxypb.ListClientInfosResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListClientInfos") + } + var r0 *proxypb.ListClientInfosResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.ListClientInfosRequest) (*proxypb.ListClientInfosResponse, error)); ok { @@ -3913,6 +4197,10 @@ func (_c *MockProxy_ListClientInfos_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) ListCredUsers(_a0 context.Context, _a1 *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListCredUsers") + } + var r0 *milvuspb.ListCredUsersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)); ok { @@ -3968,6 +4256,10 @@ func (_c *MockProxy_ListCredUsers_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) ListDatabases(_a0 context.Context, _a1 *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListDatabases") + } + var r0 *milvuspb.ListDatabasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)); ok { @@ -4023,6 +4315,10 @@ func (_c *MockProxy_ListDatabases_Call) RunAndReturn(run func(context.Context, * func (_m *MockProxy) ListImportTasks(_a0 context.Context, _a1 *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListImportTasks") + } + var r0 *milvuspb.ListImportTasksResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error)); ok { @@ -4078,6 +4374,10 @@ func (_c *MockProxy_ListImportTasks_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) ListImports(_a0 context.Context, _a1 *internalpb.ListImportsRequest) (*internalpb.ListImportsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListImports") + } + var r0 *internalpb.ListImportsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest) (*internalpb.ListImportsResponse, error)); ok { @@ -4133,6 +4433,10 @@ func (_c *MockProxy_ListImports_Call) RunAndReturn(run func(context.Context, *in func (_m *MockProxy) ListIndexedSegment(_a0 context.Context, _a1 *federpb.ListIndexedSegmentRequest) (*federpb.ListIndexedSegmentResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListIndexedSegment") + } + var r0 *federpb.ListIndexedSegmentResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *federpb.ListIndexedSegmentRequest) (*federpb.ListIndexedSegmentResponse, error)); ok { @@ -4188,6 +4492,10 @@ func (_c *MockProxy_ListIndexedSegment_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) ListResourceGroups(_a0 context.Context, _a1 *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListResourceGroups") + } + var r0 *milvuspb.ListResourceGroupsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)); ok { @@ -4243,6 +4551,10 @@ func (_c *MockProxy_ListResourceGroups_Call) RunAndReturn(run func(context.Conte func (_m *MockProxy) LoadBalance(_a0 context.Context, _a1 *milvuspb.LoadBalanceRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadBalanceRequest) (*commonpb.Status, error)); ok { @@ -4298,6 +4610,10 @@ func (_c *MockProxy_LoadBalance_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) LoadCollection(_a0 context.Context, _a1 *milvuspb.LoadCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadCollectionRequest) (*commonpb.Status, error)); ok { @@ -4353,6 +4669,10 @@ func (_c *MockProxy_LoadCollection_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) LoadPartitions(_a0 context.Context, _a1 *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error)); ok { @@ -4408,6 +4728,10 @@ func (_c *MockProxy_LoadPartitions_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) ManualCompaction(_a0 context.Context, _a1 *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ManualCompaction") + } + var r0 *milvuspb.ManualCompactionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error)); ok { @@ -4463,6 +4787,10 @@ func (_c *MockProxy_ManualCompaction_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) OperatePrivilege(_a0 context.Context, _a1 *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for OperatePrivilege") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)); ok { @@ -4518,6 +4846,10 @@ func (_c *MockProxy_OperatePrivilege_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) OperateUserRole(_a0 context.Context, _a1 *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for OperateUserRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)); ok { @@ -4573,6 +4905,10 @@ func (_c *MockProxy_OperateUserRole_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) Query(_a0 context.Context, _a1 *milvuspb.QueryRequest) (*milvuspb.QueryResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Query") + } + var r0 *milvuspb.QueryResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.QueryRequest) (*milvuspb.QueryResults, error)); ok { @@ -4628,6 +4964,10 @@ func (_c *MockProxy_Query_Call) RunAndReturn(run func(context.Context, *milvuspb func (_m *MockProxy) RefreshPolicyInfoCache(_a0 context.Context, _a1 *proxypb.RefreshPolicyInfoCacheRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for RefreshPolicyInfoCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) (*commonpb.Status, error)); ok { @@ -4683,6 +5023,10 @@ func (_c *MockProxy_RefreshPolicyInfoCache_Call) RunAndReturn(run func(context.C func (_m *MockProxy) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -4724,6 +5068,10 @@ func (_c *MockProxy_Register_Call) RunAndReturn(run func() error) *MockProxy_Reg func (_m *MockProxy) RegisterLink(_a0 context.Context, _a1 *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for RegisterLink") + } + var r0 *milvuspb.RegisterLinkResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error)); ok { @@ -4779,6 +5127,10 @@ func (_c *MockProxy_RegisterLink_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) ReleaseCollection(_a0 context.Context, _a1 *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok { @@ -4834,6 +5186,10 @@ func (_c *MockProxy_ReleaseCollection_Call) RunAndReturn(run func(context.Contex func (_m *MockProxy) ReleasePartitions(_a0 context.Context, _a1 *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok { @@ -4889,6 +5245,10 @@ func (_c *MockProxy_ReleasePartitions_Call) RunAndReturn(run func(context.Contex func (_m *MockProxy) RenameCollection(_a0 context.Context, _a1 *milvuspb.RenameCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for RenameCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)); ok { @@ -4944,6 +5304,10 @@ func (_c *MockProxy_RenameCollection_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) ReplicateMessage(_a0 context.Context, _a1 *milvuspb.ReplicateMessageRequest) (*milvuspb.ReplicateMessageResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReplicateMessage") + } + var r0 *milvuspb.ReplicateMessageResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReplicateMessageRequest) (*milvuspb.ReplicateMessageResponse, error)); ok { @@ -4999,6 +5363,10 @@ func (_c *MockProxy_ReplicateMessage_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) RestoreRBAC(_a0 context.Context, _a1 *milvuspb.RestoreRBACMetaRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for RestoreRBAC") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RestoreRBACMetaRequest) (*commonpb.Status, error)); ok { @@ -5054,6 +5422,10 @@ func (_c *MockProxy_RestoreRBAC_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) Search(_a0 context.Context, _a1 *milvuspb.SearchRequest) (*milvuspb.SearchResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Search") + } + var r0 *milvuspb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SearchRequest) (*milvuspb.SearchResults, error)); ok { @@ -5109,6 +5481,10 @@ func (_c *MockProxy_Search_Call) RunAndReturn(run func(context.Context, *milvusp func (_m *MockProxy) SelectGrant(_a0 context.Context, _a1 *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SelectGrant") + } + var r0 *milvuspb.SelectGrantResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)); ok { @@ -5164,6 +5540,10 @@ func (_c *MockProxy_SelectGrant_Call) RunAndReturn(run func(context.Context, *mi func (_m *MockProxy) SelectRole(_a0 context.Context, _a1 *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SelectRole") + } + var r0 *milvuspb.SelectRoleResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)); ok { @@ -5219,6 +5599,10 @@ func (_c *MockProxy_SelectRole_Call) RunAndReturn(run func(context.Context, *mil func (_m *MockProxy) SelectUser(_a0 context.Context, _a1 *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SelectUser") + } + var r0 *milvuspb.SelectUserResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)); ok { @@ -5439,6 +5823,10 @@ func (_c *MockProxy_SetQueryNodeCreator_Call) RunAndReturn(run func(func(context func (_m *MockProxy) SetRates(_a0 context.Context, _a1 *proxypb.SetRatesRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SetRates") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.SetRatesRequest) (*commonpb.Status, error)); ok { @@ -5527,6 +5915,10 @@ func (_c *MockProxy_SetRootCoordClient_Call) RunAndReturn(run func(types.RootCoo func (_m *MockProxy) ShowCollections(_a0 context.Context, _a1 *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowCollections") + } + var r0 *milvuspb.ShowCollectionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)); ok { @@ -5582,6 +5974,10 @@ func (_c *MockProxy_ShowCollections_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) ShowPartitions(_a0 context.Context, _a1 *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowPartitions") + } + var r0 *milvuspb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok { @@ -5637,6 +6033,10 @@ func (_c *MockProxy_ShowPartitions_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -5678,6 +6078,10 @@ func (_c *MockProxy_Start_Call) RunAndReturn(run func() error) *MockProxy_Start_ func (_m *MockProxy) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -5719,6 +6123,10 @@ func (_c *MockProxy_Stop_Call) RunAndReturn(run func() error) *MockProxy_Stop_Ca func (_m *MockProxy) TransferNode(_a0 context.Context, _a1 *milvuspb.TransferNodeRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for TransferNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)); ok { @@ -5774,6 +6182,10 @@ func (_c *MockProxy_TransferNode_Call) RunAndReturn(run func(context.Context, *m func (_m *MockProxy) TransferReplica(_a0 context.Context, _a1 *milvuspb.TransferReplicaRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for TransferReplica") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferReplicaRequest) (*commonpb.Status, error)); ok { @@ -5829,6 +6241,10 @@ func (_c *MockProxy_TransferReplica_Call) RunAndReturn(run func(context.Context, func (_m *MockProxy) UpdateCredential(_a0 context.Context, _a1 *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error)); ok { @@ -5884,6 +6300,10 @@ func (_c *MockProxy_UpdateCredential_Call) RunAndReturn(run func(context.Context func (_m *MockProxy) UpdateCredentialCache(_a0 context.Context, _a1 *proxypb.UpdateCredCacheRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateCredentialCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.UpdateCredCacheRequest) (*commonpb.Status, error)); ok { @@ -5939,6 +6359,10 @@ func (_c *MockProxy_UpdateCredentialCache_Call) RunAndReturn(run func(context.Co func (_m *MockProxy) UpdateResourceGroups(_a0 context.Context, _a1 *milvuspb.UpdateResourceGroupsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateResourceGroups") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateResourceGroupsRequest) (*commonpb.Status, error)); ok { @@ -6027,6 +6451,10 @@ func (_c *MockProxy_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCo func (_m *MockProxy) Upsert(_a0 context.Context, _a1 *milvuspb.UpsertRequest) (*milvuspb.MutationResult, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Upsert") + } + var r0 *milvuspb.MutationResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpsertRequest) (*milvuspb.MutationResult, error)); ok { diff --git a/internal/mocks/mock_proxy_client.go b/internal/mocks/mock_proxy_client.go index 44cad0f23f..7d84eb171c 100644 --- a/internal/mocks/mock_proxy_client.go +++ b/internal/mocks/mock_proxy_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -35,6 +35,10 @@ func (_m *MockProxyClient) EXPECT() *MockProxyClient_Expecter { func (_m *MockProxyClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -83,6 +87,10 @@ func (_m *MockProxyClient) GetComponentStates(ctx context.Context, in *milvuspb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -153,6 +161,10 @@ func (_m *MockProxyClient) GetDdChannel(ctx context.Context, in *internalpb.GetD _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetDdChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetDdChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -223,6 +235,10 @@ func (_m *MockProxyClient) GetImportProgress(ctx context.Context, in *internalpb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetImportProgress") + } + var r0 *internalpb.GetImportProgressResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetImportProgressRequest, ...grpc.CallOption) (*internalpb.GetImportProgressResponse, error)); ok { @@ -293,6 +309,10 @@ func (_m *MockProxyClient) GetProxyMetrics(ctx context.Context, in *milvuspb.Get _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetProxyMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -363,6 +383,10 @@ func (_m *MockProxyClient) GetStatisticsChannel(ctx context.Context, in *interna _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -433,6 +457,10 @@ func (_m *MockProxyClient) ImportV2(ctx context.Context, in *internalpb.ImportRe _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ImportV2") + } + var r0 *internalpb.ImportResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ImportRequest, ...grpc.CallOption) (*internalpb.ImportResponse, error)); ok { @@ -503,6 +531,10 @@ func (_m *MockProxyClient) InvalidateCollectionMetaCache(ctx context.Context, in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for InvalidateCollectionMetaCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -573,6 +605,10 @@ func (_m *MockProxyClient) InvalidateCredentialCache(ctx context.Context, in *pr _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for InvalidateCredentialCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCredCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -643,6 +679,10 @@ func (_m *MockProxyClient) InvalidateShardLeaderCache(ctx context.Context, in *p _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for InvalidateShardLeaderCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateShardLeaderCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -713,6 +753,10 @@ func (_m *MockProxyClient) ListClientInfos(ctx context.Context, in *proxypb.List _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListClientInfos") + } + var r0 *proxypb.ListClientInfosResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.ListClientInfosRequest, ...grpc.CallOption) (*proxypb.ListClientInfosResponse, error)); ok { @@ -783,6 +827,10 @@ func (_m *MockProxyClient) ListImports(ctx context.Context, in *internalpb.ListI _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListImports") + } + var r0 *internalpb.ListImportsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListImportsRequest, ...grpc.CallOption) (*internalpb.ListImportsResponse, error)); ok { @@ -853,6 +901,10 @@ func (_m *MockProxyClient) RefreshPolicyInfoCache(ctx context.Context, in *proxy _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for RefreshPolicyInfoCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -923,6 +975,10 @@ func (_m *MockProxyClient) SetRates(ctx context.Context, in *proxypb.SetRatesReq _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SetRates") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.SetRatesRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -993,6 +1049,10 @@ func (_m *MockProxyClient) UpdateCredentialCache(ctx context.Context, in *proxyp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateCredentialCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.UpdateCredCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_querycoord.go b/internal/mocks/mock_querycoord.go index 5ed0533e63..b5a1caac71 100644 --- a/internal/mocks/mock_querycoord.go +++ b/internal/mocks/mock_querycoord.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -38,6 +38,10 @@ func (_m *MockQueryCoord) EXPECT() *MockQueryCoord_Expecter { func (_m *MockQueryCoord) ActivateChecker(_a0 context.Context, _a1 *querypb.ActivateCheckerRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ActivateChecker") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ActivateCheckerRequest) (*commonpb.Status, error)); ok { @@ -93,6 +97,10 @@ func (_c *MockQueryCoord_ActivateChecker_Call) RunAndReturn(run func(context.Con func (_m *MockQueryCoord) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok { @@ -148,6 +156,10 @@ func (_c *MockQueryCoord_CheckHealth_Call) RunAndReturn(run func(context.Context func (_m *MockQueryCoord) CheckQueryNodeDistribution(_a0 context.Context, _a1 *querypb.CheckQueryNodeDistributionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CheckQueryNodeDistribution") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.CheckQueryNodeDistributionRequest) (*commonpb.Status, error)); ok { @@ -203,6 +215,10 @@ func (_c *MockQueryCoord_CheckQueryNodeDistribution_Call) RunAndReturn(run func( func (_m *MockQueryCoord) CreateResourceGroup(_a0 context.Context, _a1 *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateResourceGroup") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error)); ok { @@ -258,6 +274,10 @@ func (_c *MockQueryCoord_CreateResourceGroup_Call) RunAndReturn(run func(context func (_m *MockQueryCoord) DeactivateChecker(_a0 context.Context, _a1 *querypb.DeactivateCheckerRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DeactivateChecker") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeactivateCheckerRequest) (*commonpb.Status, error)); ok { @@ -313,6 +333,10 @@ func (_c *MockQueryCoord_DeactivateChecker_Call) RunAndReturn(run func(context.C func (_m *MockQueryCoord) DescribeResourceGroup(_a0 context.Context, _a1 *querypb.DescribeResourceGroupRequest) (*querypb.DescribeResourceGroupResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeResourceGroup") + } + var r0 *querypb.DescribeResourceGroupResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DescribeResourceGroupRequest) (*querypb.DescribeResourceGroupResponse, error)); ok { @@ -368,6 +392,10 @@ func (_c *MockQueryCoord_DescribeResourceGroup_Call) RunAndReturn(run func(conte func (_m *MockQueryCoord) DropResourceGroup(_a0 context.Context, _a1 *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropResourceGroup") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error)); ok { @@ -423,6 +451,10 @@ func (_c *MockQueryCoord_DropResourceGroup_Call) RunAndReturn(run func(context.C func (_m *MockQueryCoord) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -478,6 +510,10 @@ func (_c *MockQueryCoord_GetComponentStates_Call) RunAndReturn(run func(context. func (_m *MockQueryCoord) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -533,6 +569,10 @@ func (_c *MockQueryCoord_GetMetrics_Call) RunAndReturn(run func(context.Context, func (_m *MockQueryCoord) GetPartitionStates(_a0 context.Context, _a1 *querypb.GetPartitionStatesRequest) (*querypb.GetPartitionStatesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetPartitionStates") + } + var r0 *querypb.GetPartitionStatesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetPartitionStatesRequest) (*querypb.GetPartitionStatesResponse, error)); ok { @@ -588,6 +628,10 @@ func (_c *MockQueryCoord_GetPartitionStates_Call) RunAndReturn(run func(context. func (_m *MockQueryCoord) GetQueryNodeDistribution(_a0 context.Context, _a1 *querypb.GetQueryNodeDistributionRequest) (*querypb.GetQueryNodeDistributionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetQueryNodeDistribution") + } + var r0 *querypb.GetQueryNodeDistributionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetQueryNodeDistributionRequest) (*querypb.GetQueryNodeDistributionResponse, error)); ok { @@ -643,6 +687,10 @@ func (_c *MockQueryCoord_GetQueryNodeDistribution_Call) RunAndReturn(run func(co func (_m *MockQueryCoord) GetReplicas(_a0 context.Context, _a1 *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetReplicas") + } + var r0 *milvuspb.GetReplicasResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error)); ok { @@ -698,6 +746,10 @@ func (_c *MockQueryCoord_GetReplicas_Call) RunAndReturn(run func(context.Context func (_m *MockQueryCoord) GetSegmentInfo(_a0 context.Context, _a1 *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *querypb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error)); ok { @@ -753,6 +805,10 @@ func (_c *MockQueryCoord_GetSegmentInfo_Call) RunAndReturn(run func(context.Cont func (_m *MockQueryCoord) GetShardLeaders(_a0 context.Context, _a1 *querypb.GetShardLeadersRequest) (*querypb.GetShardLeadersResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetShardLeaders") + } + var r0 *querypb.GetShardLeadersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetShardLeadersRequest) (*querypb.GetShardLeadersResponse, error)); ok { @@ -808,6 +864,10 @@ func (_c *MockQueryCoord_GetShardLeaders_Call) RunAndReturn(run func(context.Con func (_m *MockQueryCoord) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -863,6 +923,10 @@ func (_c *MockQueryCoord_GetStatisticsChannel_Call) RunAndReturn(run func(contex func (_m *MockQueryCoord) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -918,6 +982,10 @@ func (_c *MockQueryCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context. func (_m *MockQueryCoord) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -959,6 +1027,10 @@ func (_c *MockQueryCoord_Init_Call) RunAndReturn(run func() error) *MockQueryCoo func (_m *MockQueryCoord) ListCheckers(_a0 context.Context, _a1 *querypb.ListCheckersRequest) (*querypb.ListCheckersResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListCheckers") + } + var r0 *querypb.ListCheckersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListCheckersRequest) (*querypb.ListCheckersResponse, error)); ok { @@ -1014,6 +1086,10 @@ func (_c *MockQueryCoord_ListCheckers_Call) RunAndReturn(run func(context.Contex func (_m *MockQueryCoord) ListQueryNode(_a0 context.Context, _a1 *querypb.ListQueryNodeRequest) (*querypb.ListQueryNodeResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListQueryNode") + } + var r0 *querypb.ListQueryNodeResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListQueryNodeRequest) (*querypb.ListQueryNodeResponse, error)); ok { @@ -1069,6 +1145,10 @@ func (_c *MockQueryCoord_ListQueryNode_Call) RunAndReturn(run func(context.Conte func (_m *MockQueryCoord) ListResourceGroups(_a0 context.Context, _a1 *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListResourceGroups") + } + var r0 *milvuspb.ListResourceGroupsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error)); ok { @@ -1124,6 +1204,10 @@ func (_c *MockQueryCoord_ListResourceGroups_Call) RunAndReturn(run func(context. func (_m *MockQueryCoord) LoadBalance(_a0 context.Context, _a1 *querypb.LoadBalanceRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadBalanceRequest) (*commonpb.Status, error)); ok { @@ -1179,6 +1263,10 @@ func (_c *MockQueryCoord_LoadBalance_Call) RunAndReturn(run func(context.Context func (_m *MockQueryCoord) LoadCollection(_a0 context.Context, _a1 *querypb.LoadCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadCollectionRequest) (*commonpb.Status, error)); ok { @@ -1234,6 +1322,10 @@ func (_c *MockQueryCoord_LoadCollection_Call) RunAndReturn(run func(context.Cont func (_m *MockQueryCoord) LoadPartitions(_a0 context.Context, _a1 *querypb.LoadPartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok { @@ -1289,6 +1381,10 @@ func (_c *MockQueryCoord_LoadPartitions_Call) RunAndReturn(run func(context.Cont func (_m *MockQueryCoord) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1330,6 +1426,10 @@ func (_c *MockQueryCoord_Register_Call) RunAndReturn(run func() error) *MockQuer func (_m *MockQueryCoord) ReleaseCollection(_a0 context.Context, _a1 *querypb.ReleaseCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok { @@ -1385,6 +1485,10 @@ func (_c *MockQueryCoord_ReleaseCollection_Call) RunAndReturn(run func(context.C func (_m *MockQueryCoord) ReleasePartitions(_a0 context.Context, _a1 *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok { @@ -1440,6 +1544,10 @@ func (_c *MockQueryCoord_ReleasePartitions_Call) RunAndReturn(run func(context.C func (_m *MockQueryCoord) ResumeBalance(_a0 context.Context, _a1 *querypb.ResumeBalanceRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ResumeBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeBalanceRequest) (*commonpb.Status, error)); ok { @@ -1495,6 +1603,10 @@ func (_c *MockQueryCoord_ResumeBalance_Call) RunAndReturn(run func(context.Conte func (_m *MockQueryCoord) ResumeNode(_a0 context.Context, _a1 *querypb.ResumeNodeRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ResumeNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeNodeRequest) (*commonpb.Status, error)); ok { @@ -1583,6 +1695,10 @@ func (_c *MockQueryCoord_SetAddress_Call) RunAndReturn(run func(string)) *MockQu func (_m *MockQueryCoord) SetDataCoordClient(dataCoord types.DataCoordClient) error { ret := _m.Called(dataCoord) + if len(ret) == 0 { + panic("no return value specified for SetDataCoordClient") + } + var r0 error if rf, ok := ret.Get(0).(func(types.DataCoordClient) error); ok { r0 = rf(dataCoord) @@ -1691,6 +1807,10 @@ func (_c *MockQueryCoord_SetQueryNodeCreator_Call) RunAndReturn(run func(func(co func (_m *MockQueryCoord) SetRootCoordClient(rootCoord types.RootCoordClient) error { ret := _m.Called(rootCoord) + if len(ret) == 0 { + panic("no return value specified for SetRootCoordClient") + } + var r0 error if rf, ok := ret.Get(0).(func(types.RootCoordClient) error); ok { r0 = rf(rootCoord) @@ -1766,6 +1886,10 @@ func (_c *MockQueryCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client func (_m *MockQueryCoord) ShowCollections(_a0 context.Context, _a1 *querypb.ShowCollectionsRequest) (*querypb.ShowCollectionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowCollections") + } + var r0 *querypb.ShowCollectionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowCollectionsRequest) (*querypb.ShowCollectionsResponse, error)); ok { @@ -1821,6 +1945,10 @@ func (_c *MockQueryCoord_ShowCollections_Call) RunAndReturn(run func(context.Con func (_m *MockQueryCoord) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -1876,6 +2004,10 @@ func (_c *MockQueryCoord_ShowConfigurations_Call) RunAndReturn(run func(context. func (_m *MockQueryCoord) ShowPartitions(_a0 context.Context, _a1 *querypb.ShowPartitionsRequest) (*querypb.ShowPartitionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowPartitions") + } + var r0 *querypb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowPartitionsRequest) (*querypb.ShowPartitionsResponse, error)); ok { @@ -1931,6 +2063,10 @@ func (_c *MockQueryCoord_ShowPartitions_Call) RunAndReturn(run func(context.Cont func (_m *MockQueryCoord) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1972,6 +2108,10 @@ func (_c *MockQueryCoord_Start_Call) RunAndReturn(run func() error) *MockQueryCo func (_m *MockQueryCoord) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -2013,6 +2153,10 @@ func (_c *MockQueryCoord_Stop_Call) RunAndReturn(run func() error) *MockQueryCoo func (_m *MockQueryCoord) SuspendBalance(_a0 context.Context, _a1 *querypb.SuspendBalanceRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SuspendBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendBalanceRequest) (*commonpb.Status, error)); ok { @@ -2068,6 +2212,10 @@ func (_c *MockQueryCoord_SuspendBalance_Call) RunAndReturn(run func(context.Cont func (_m *MockQueryCoord) SuspendNode(_a0 context.Context, _a1 *querypb.SuspendNodeRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SuspendNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendNodeRequest) (*commonpb.Status, error)); ok { @@ -2123,6 +2271,10 @@ func (_c *MockQueryCoord_SuspendNode_Call) RunAndReturn(run func(context.Context func (_m *MockQueryCoord) SyncNewCreatedPartition(_a0 context.Context, _a1 *querypb.SyncNewCreatedPartitionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SyncNewCreatedPartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncNewCreatedPartitionRequest) (*commonpb.Status, error)); ok { @@ -2178,6 +2330,10 @@ func (_c *MockQueryCoord_SyncNewCreatedPartition_Call) RunAndReturn(run func(con func (_m *MockQueryCoord) TransferChannel(_a0 context.Context, _a1 *querypb.TransferChannelRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for TransferChannel") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferChannelRequest) (*commonpb.Status, error)); ok { @@ -2233,6 +2389,10 @@ func (_c *MockQueryCoord_TransferChannel_Call) RunAndReturn(run func(context.Con func (_m *MockQueryCoord) TransferNode(_a0 context.Context, _a1 *milvuspb.TransferNodeRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for TransferNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) (*commonpb.Status, error)); ok { @@ -2288,6 +2448,10 @@ func (_c *MockQueryCoord_TransferNode_Call) RunAndReturn(run func(context.Contex func (_m *MockQueryCoord) TransferReplica(_a0 context.Context, _a1 *querypb.TransferReplicaRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for TransferReplica") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferReplicaRequest) (*commonpb.Status, error)); ok { @@ -2343,6 +2507,10 @@ func (_c *MockQueryCoord_TransferReplica_Call) RunAndReturn(run func(context.Con func (_m *MockQueryCoord) TransferSegment(_a0 context.Context, _a1 *querypb.TransferSegmentRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for TransferSegment") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferSegmentRequest) (*commonpb.Status, error)); ok { @@ -2398,6 +2566,10 @@ func (_c *MockQueryCoord_TransferSegment_Call) RunAndReturn(run func(context.Con func (_m *MockQueryCoord) UpdateLoadConfig(_a0 context.Context, _a1 *querypb.UpdateLoadConfigRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateLoadConfig") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateLoadConfigRequest) (*commonpb.Status, error)); ok { @@ -2453,6 +2625,10 @@ func (_c *MockQueryCoord_UpdateLoadConfig_Call) RunAndReturn(run func(context.Co func (_m *MockQueryCoord) UpdateResourceGroups(_a0 context.Context, _a1 *querypb.UpdateResourceGroupsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateResourceGroups") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateResourceGroupsRequest) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_querycoord_client.go b/internal/mocks/mock_querycoord_client.go index 7d99a324bb..3285f867a6 100644 --- a/internal/mocks/mock_querycoord_client.go +++ b/internal/mocks/mock_querycoord_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -42,6 +42,10 @@ func (_m *MockQueryCoordClient) ActivateChecker(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ActivateChecker") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ActivateCheckerRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -112,6 +116,10 @@ func (_m *MockQueryCoordClient) CheckHealth(ctx context.Context, in *milvuspb.Ch _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest, ...grpc.CallOption) (*milvuspb.CheckHealthResponse, error)); ok { @@ -182,6 +190,10 @@ func (_m *MockQueryCoordClient) CheckQueryNodeDistribution(ctx context.Context, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CheckQueryNodeDistribution") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.CheckQueryNodeDistributionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -245,6 +257,10 @@ func (_c *MockQueryCoordClient_CheckQueryNodeDistribution_Call) RunAndReturn(run func (_m *MockQueryCoordClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -293,6 +309,10 @@ func (_m *MockQueryCoordClient) CreateResourceGroup(ctx context.Context, in *mil _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateResourceGroup") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -363,6 +383,10 @@ func (_m *MockQueryCoordClient) DeactivateChecker(ctx context.Context, in *query _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DeactivateChecker") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeactivateCheckerRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -433,6 +457,10 @@ func (_m *MockQueryCoordClient) DescribeResourceGroup(ctx context.Context, in *q _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DescribeResourceGroup") + } + var r0 *querypb.DescribeResourceGroupResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DescribeResourceGroupRequest, ...grpc.CallOption) (*querypb.DescribeResourceGroupResponse, error)); ok { @@ -503,6 +531,10 @@ func (_m *MockQueryCoordClient) DropResourceGroup(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropResourceGroup") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -573,6 +605,10 @@ func (_m *MockQueryCoordClient) GetComponentStates(ctx context.Context, in *milv _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -643,6 +679,10 @@ func (_m *MockQueryCoordClient) GetMetrics(ctx context.Context, in *milvuspb.Get _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -713,6 +753,10 @@ func (_m *MockQueryCoordClient) GetPartitionStates(ctx context.Context, in *quer _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetPartitionStates") + } + var r0 *querypb.GetPartitionStatesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetPartitionStatesRequest, ...grpc.CallOption) (*querypb.GetPartitionStatesResponse, error)); ok { @@ -783,6 +827,10 @@ func (_m *MockQueryCoordClient) GetQueryNodeDistribution(ctx context.Context, in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetQueryNodeDistribution") + } + var r0 *querypb.GetQueryNodeDistributionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetQueryNodeDistributionRequest, ...grpc.CallOption) (*querypb.GetQueryNodeDistributionResponse, error)); ok { @@ -853,6 +901,10 @@ func (_m *MockQueryCoordClient) GetReplicas(ctx context.Context, in *milvuspb.Ge _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetReplicas") + } + var r0 *milvuspb.GetReplicasResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest, ...grpc.CallOption) (*milvuspb.GetReplicasResponse, error)); ok { @@ -923,6 +975,10 @@ func (_m *MockQueryCoordClient) GetSegmentInfo(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *querypb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest, ...grpc.CallOption) (*querypb.GetSegmentInfoResponse, error)); ok { @@ -993,6 +1049,10 @@ func (_m *MockQueryCoordClient) GetShardLeaders(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetShardLeaders") + } + var r0 *querypb.GetShardLeadersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetShardLeadersRequest, ...grpc.CallOption) (*querypb.GetShardLeadersResponse, error)); ok { @@ -1063,6 +1123,10 @@ func (_m *MockQueryCoordClient) GetStatisticsChannel(ctx context.Context, in *in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -1133,6 +1197,10 @@ func (_m *MockQueryCoordClient) GetTimeTickChannel(ctx context.Context, in *inte _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -1203,6 +1271,10 @@ func (_m *MockQueryCoordClient) ListCheckers(ctx context.Context, in *querypb.Li _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListCheckers") + } + var r0 *querypb.ListCheckersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListCheckersRequest, ...grpc.CallOption) (*querypb.ListCheckersResponse, error)); ok { @@ -1273,6 +1345,10 @@ func (_m *MockQueryCoordClient) ListQueryNode(ctx context.Context, in *querypb.L _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListQueryNode") + } + var r0 *querypb.ListQueryNodeResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ListQueryNodeRequest, ...grpc.CallOption) (*querypb.ListQueryNodeResponse, error)); ok { @@ -1343,6 +1419,10 @@ func (_m *MockQueryCoordClient) ListResourceGroups(ctx context.Context, in *milv _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListResourceGroups") + } + var r0 *milvuspb.ListResourceGroupsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest, ...grpc.CallOption) (*milvuspb.ListResourceGroupsResponse, error)); ok { @@ -1413,6 +1493,10 @@ func (_m *MockQueryCoordClient) LoadBalance(ctx context.Context, in *querypb.Loa _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadBalanceRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1483,6 +1567,10 @@ func (_m *MockQueryCoordClient) LoadCollection(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1553,6 +1641,10 @@ func (_m *MockQueryCoordClient) LoadPartitions(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1623,6 +1715,10 @@ func (_m *MockQueryCoordClient) ReleaseCollection(ctx context.Context, in *query _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ReleaseCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1693,6 +1789,10 @@ func (_m *MockQueryCoordClient) ReleasePartitions(ctx context.Context, in *query _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1763,6 +1863,10 @@ func (_m *MockQueryCoordClient) ResumeBalance(ctx context.Context, in *querypb.R _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ResumeBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeBalanceRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1833,6 +1937,10 @@ func (_m *MockQueryCoordClient) ResumeNode(ctx context.Context, in *querypb.Resu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ResumeNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ResumeNodeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1903,6 +2011,10 @@ func (_m *MockQueryCoordClient) ShowCollections(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowCollections") + } + var r0 *querypb.ShowCollectionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowCollectionsRequest, ...grpc.CallOption) (*querypb.ShowCollectionsResponse, error)); ok { @@ -1973,6 +2085,10 @@ func (_m *MockQueryCoordClient) ShowConfigurations(ctx context.Context, in *inte _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -2043,6 +2159,10 @@ func (_m *MockQueryCoordClient) ShowPartitions(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowPartitions") + } + var r0 *querypb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ShowPartitionsRequest, ...grpc.CallOption) (*querypb.ShowPartitionsResponse, error)); ok { @@ -2113,6 +2233,10 @@ func (_m *MockQueryCoordClient) SuspendBalance(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SuspendBalance") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendBalanceRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2183,6 +2307,10 @@ func (_m *MockQueryCoordClient) SuspendNode(ctx context.Context, in *querypb.Sus _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SuspendNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SuspendNodeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2253,6 +2381,10 @@ func (_m *MockQueryCoordClient) SyncNewCreatedPartition(ctx context.Context, in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SyncNewCreatedPartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncNewCreatedPartitionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2323,6 +2455,10 @@ func (_m *MockQueryCoordClient) TransferChannel(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for TransferChannel") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferChannelRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2393,6 +2529,10 @@ func (_m *MockQueryCoordClient) TransferNode(ctx context.Context, in *milvuspb.T _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for TransferNode") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2463,6 +2603,10 @@ func (_m *MockQueryCoordClient) TransferReplica(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for TransferReplica") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferReplicaRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2533,6 +2677,10 @@ func (_m *MockQueryCoordClient) TransferSegment(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for TransferSegment") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.TransferSegmentRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2603,6 +2751,10 @@ func (_m *MockQueryCoordClient) UpdateLoadConfig(ctx context.Context, in *queryp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateLoadConfig") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateLoadConfigRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2673,6 +2825,10 @@ func (_m *MockQueryCoordClient) UpdateResourceGroups(ctx context.Context, in *qu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateResourceGroups") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UpdateResourceGroupsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_querynode.go b/internal/mocks/mock_querynode.go index d84ee0d2a8..451f651a7e 100644 --- a/internal/mocks/mock_querynode.go +++ b/internal/mocks/mock_querynode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -34,6 +34,10 @@ func (_m *MockQueryNode) EXPECT() *MockQueryNode_Expecter { func (_m *MockQueryNode) Delete(_a0 context.Context, _a1 *querypb.DeleteRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Delete") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest) (*commonpb.Status, error)); ok { @@ -89,6 +93,10 @@ func (_c *MockQueryNode_Delete_Call) RunAndReturn(run func(context.Context, *que func (_m *MockQueryNode) GetAddress() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAddress") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -130,6 +138,10 @@ func (_c *MockQueryNode_GetAddress_Call) RunAndReturn(run func() string) *MockQu func (_m *MockQueryNode) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -185,6 +197,10 @@ func (_c *MockQueryNode_GetComponentStates_Call) RunAndReturn(run func(context.C func (_m *MockQueryNode) GetDataDistribution(_a0 context.Context, _a1 *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetDataDistribution") + } + var r0 *querypb.GetDataDistributionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error)); ok { @@ -240,6 +256,10 @@ func (_c *MockQueryNode_GetDataDistribution_Call) RunAndReturn(run func(context. func (_m *MockQueryNode) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -295,6 +315,10 @@ func (_c *MockQueryNode_GetMetrics_Call) RunAndReturn(run func(context.Context, func (_m *MockQueryNode) GetNodeID() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetNodeID") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -336,6 +360,10 @@ func (_c *MockQueryNode_GetNodeID_Call) RunAndReturn(run func() int64) *MockQuer func (_m *MockQueryNode) GetSegmentInfo(_a0 context.Context, _a1 *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *querypb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error)); ok { @@ -391,6 +419,10 @@ func (_c *MockQueryNode_GetSegmentInfo_Call) RunAndReturn(run func(context.Conte func (_m *MockQueryNode) GetStatistics(_a0 context.Context, _a1 *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatistics") + } + var r0 *internalpb.GetStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error)); ok { @@ -446,6 +478,10 @@ func (_c *MockQueryNode_GetStatistics_Call) RunAndReturn(run func(context.Contex func (_m *MockQueryNode) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -501,6 +537,10 @@ func (_c *MockQueryNode_GetStatisticsChannel_Call) RunAndReturn(run func(context func (_m *MockQueryNode) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -556,6 +596,10 @@ func (_c *MockQueryNode_GetTimeTickChannel_Call) RunAndReturn(run func(context.C func (_m *MockQueryNode) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -597,6 +641,10 @@ func (_c *MockQueryNode_Init_Call) RunAndReturn(run func() error) *MockQueryNode func (_m *MockQueryNode) LoadPartitions(_a0 context.Context, _a1 *querypb.LoadPartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok { @@ -652,6 +700,10 @@ func (_c *MockQueryNode_LoadPartitions_Call) RunAndReturn(run func(context.Conte func (_m *MockQueryNode) LoadSegments(_a0 context.Context, _a1 *querypb.LoadSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) (*commonpb.Status, error)); ok { @@ -707,6 +759,10 @@ func (_c *MockQueryNode_LoadSegments_Call) RunAndReturn(run func(context.Context func (_m *MockQueryNode) Query(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Query") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok { @@ -762,6 +818,10 @@ func (_c *MockQueryNode_Query_Call) RunAndReturn(run func(context.Context, *quer func (_m *MockQueryNode) QuerySegments(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QuerySegments") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok { @@ -817,6 +877,10 @@ func (_c *MockQueryNode_QuerySegments_Call) RunAndReturn(run func(context.Contex func (_m *MockQueryNode) QueryStream(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamServer) error { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryStream") + } + var r0 error if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamServer) error); ok { r0 = rf(_a0, _a1) @@ -860,6 +924,10 @@ func (_c *MockQueryNode_QueryStream_Call) RunAndReturn(run func(*querypb.QueryRe func (_m *MockQueryNode) QueryStreamSegments(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamSegmentsServer) error { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryStreamSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamSegmentsServer) error); ok { r0 = rf(_a0, _a1) @@ -903,6 +971,10 @@ func (_c *MockQueryNode_QueryStreamSegments_Call) RunAndReturn(run func(*querypb func (_m *MockQueryNode) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -944,6 +1016,10 @@ func (_c *MockQueryNode_Register_Call) RunAndReturn(run func() error) *MockQuery func (_m *MockQueryNode) ReleaseCollection(_a0 context.Context, _a1 *querypb.ReleaseCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok { @@ -999,6 +1075,10 @@ func (_c *MockQueryNode_ReleaseCollection_Call) RunAndReturn(run func(context.Co func (_m *MockQueryNode) ReleasePartitions(_a0 context.Context, _a1 *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok { @@ -1054,6 +1134,10 @@ func (_c *MockQueryNode_ReleasePartitions_Call) RunAndReturn(run func(context.Co func (_m *MockQueryNode) ReleaseSegments(_a0 context.Context, _a1 *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error)); ok { @@ -1109,6 +1193,10 @@ func (_c *MockQueryNode_ReleaseSegments_Call) RunAndReturn(run func(context.Cont func (_m *MockQueryNode) Search(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Search") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok { @@ -1164,6 +1252,10 @@ func (_c *MockQueryNode_Search_Call) RunAndReturn(run func(context.Context, *que func (_m *MockQueryNode) SearchSegments(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SearchSegments") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok { @@ -1285,6 +1377,10 @@ func (_c *MockQueryNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Clie func (_m *MockQueryNode) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -1340,6 +1436,10 @@ func (_c *MockQueryNode_ShowConfigurations_Call) RunAndReturn(run func(context.C func (_m *MockQueryNode) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1381,6 +1481,10 @@ func (_c *MockQueryNode_Start_Call) RunAndReturn(run func() error) *MockQueryNod func (_m *MockQueryNode) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1422,6 +1526,10 @@ func (_c *MockQueryNode_Stop_Call) RunAndReturn(run func() error) *MockQueryNode func (_m *MockQueryNode) SyncDistribution(_a0 context.Context, _a1 *querypb.SyncDistributionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SyncDistribution") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncDistributionRequest) (*commonpb.Status, error)); ok { @@ -1477,6 +1585,10 @@ func (_c *MockQueryNode_SyncDistribution_Call) RunAndReturn(run func(context.Con func (_m *MockQueryNode) SyncReplicaSegments(_a0 context.Context, _a1 *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SyncReplicaSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error)); ok { @@ -1532,6 +1644,10 @@ func (_c *MockQueryNode_SyncReplicaSegments_Call) RunAndReturn(run func(context. func (_m *MockQueryNode) UnsubDmChannel(_a0 context.Context, _a1 *querypb.UnsubDmChannelRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UnsubDmChannel") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UnsubDmChannelRequest) (*commonpb.Status, error)); ok { @@ -1620,6 +1736,10 @@ func (_c *MockQueryNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.Sta func (_m *MockQueryNode) WatchDmChannels(_a0 context.Context, _a1 *querypb.WatchDmChannelsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for WatchDmChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_querynode_client.go b/internal/mocks/mock_querynode_client.go index 3621a87884..fcd8ea591d 100644 --- a/internal/mocks/mock_querynode_client.go +++ b/internal/mocks/mock_querynode_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -35,6 +35,10 @@ func (_m *MockQueryNodeClient) EXPECT() *MockQueryNodeClient_Expecter { func (_m *MockQueryNodeClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -83,6 +87,10 @@ func (_m *MockQueryNodeClient) Delete(ctx context.Context, in *querypb.DeleteReq _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Delete") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -153,6 +161,10 @@ func (_m *MockQueryNodeClient) GetComponentStates(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -223,6 +235,10 @@ func (_m *MockQueryNodeClient) GetDataDistribution(ctx context.Context, in *quer _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetDataDistribution") + } + var r0 *querypb.GetDataDistributionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetDataDistributionRequest, ...grpc.CallOption) (*querypb.GetDataDistributionResponse, error)); ok { @@ -293,6 +309,10 @@ func (_m *MockQueryNodeClient) GetMetrics(ctx context.Context, in *milvuspb.GetM _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -363,6 +383,10 @@ func (_m *MockQueryNodeClient) GetSegmentInfo(ctx context.Context, in *querypb.G _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *querypb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest, ...grpc.CallOption) (*querypb.GetSegmentInfoResponse, error)); ok { @@ -433,6 +457,10 @@ func (_m *MockQueryNodeClient) GetStatistics(ctx context.Context, in *querypb.Ge _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatistics") + } + var r0 *internalpb.GetStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest, ...grpc.CallOption) (*internalpb.GetStatisticsResponse, error)); ok { @@ -503,6 +531,10 @@ func (_m *MockQueryNodeClient) GetStatisticsChannel(ctx context.Context, in *int _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -573,6 +605,10 @@ func (_m *MockQueryNodeClient) GetTimeTickChannel(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -643,6 +679,10 @@ func (_m *MockQueryNodeClient) LoadPartitions(ctx context.Context, in *querypb.L _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -713,6 +753,10 @@ func (_m *MockQueryNodeClient) LoadSegments(ctx context.Context, in *querypb.Loa _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -783,6 +827,10 @@ func (_m *MockQueryNodeClient) Query(ctx context.Context, in *querypb.QueryReque _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Query") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (*internalpb.RetrieveResults, error)); ok { @@ -853,6 +901,10 @@ func (_m *MockQueryNodeClient) QuerySegments(ctx context.Context, in *querypb.Qu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QuerySegments") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (*internalpb.RetrieveResults, error)); ok { @@ -923,6 +975,10 @@ func (_m *MockQueryNodeClient) QueryStream(ctx context.Context, in *querypb.Quer _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QueryStream") + } + var r0 querypb.QueryNode_QueryStreamClient var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (querypb.QueryNode_QueryStreamClient, error)); ok { @@ -993,6 +1049,10 @@ func (_m *MockQueryNodeClient) QueryStreamSegments(ctx context.Context, in *quer _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for QueryStreamSegments") + } + var r0 querypb.QueryNode_QueryStreamSegmentsClient var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, ...grpc.CallOption) (querypb.QueryNode_QueryStreamSegmentsClient, error)); ok { @@ -1063,6 +1123,10 @@ func (_m *MockQueryNodeClient) ReleaseCollection(ctx context.Context, in *queryp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ReleaseCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1133,6 +1197,10 @@ func (_m *MockQueryNodeClient) ReleasePartitions(ctx context.Context, in *queryp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1203,6 +1271,10 @@ func (_m *MockQueryNodeClient) ReleaseSegments(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ReleaseSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1273,6 +1345,10 @@ func (_m *MockQueryNodeClient) Search(ctx context.Context, in *querypb.SearchReq _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Search") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest, ...grpc.CallOption) (*internalpb.SearchResults, error)); ok { @@ -1343,6 +1419,10 @@ func (_m *MockQueryNodeClient) SearchSegments(ctx context.Context, in *querypb.S _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SearchSegments") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest, ...grpc.CallOption) (*internalpb.SearchResults, error)); ok { @@ -1413,6 +1493,10 @@ func (_m *MockQueryNodeClient) ShowConfigurations(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -1483,6 +1567,10 @@ func (_m *MockQueryNodeClient) SyncDistribution(ctx context.Context, in *querypb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SyncDistribution") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncDistributionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1553,6 +1641,10 @@ func (_m *MockQueryNodeClient) SyncReplicaSegments(ctx context.Context, in *quer _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SyncReplicaSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncReplicaSegmentsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1623,6 +1715,10 @@ func (_m *MockQueryNodeClient) UnsubDmChannel(ctx context.Context, in *querypb.U _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UnsubDmChannel") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UnsubDmChannelRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1693,6 +1789,10 @@ func (_m *MockQueryNodeClient) WatchDmChannels(ctx context.Context, in *querypb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for WatchDmChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.WatchDmChannelsRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_rootcoord.go b/internal/mocks/mock_rootcoord.go index 12a6d5c5ac..a9fe3cfcce 100644 --- a/internal/mocks/mock_rootcoord.go +++ b/internal/mocks/mock_rootcoord.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -40,6 +40,10 @@ func (_m *RootCoord) EXPECT() *RootCoord_Expecter { func (_m *RootCoord) AllocID(_a0 context.Context, _a1 *rootcoordpb.AllocIDRequest) (*rootcoordpb.AllocIDResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AllocID") + } + var r0 *rootcoordpb.AllocIDResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocIDRequest) (*rootcoordpb.AllocIDResponse, error)); ok { @@ -95,6 +99,10 @@ func (_c *RootCoord_AllocID_Call) RunAndReturn(run func(context.Context, *rootco func (_m *RootCoord) AllocTimestamp(_a0 context.Context, _a1 *rootcoordpb.AllocTimestampRequest) (*rootcoordpb.AllocTimestampResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AllocTimestamp") + } + var r0 *rootcoordpb.AllocTimestampResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocTimestampRequest) (*rootcoordpb.AllocTimestampResponse, error)); ok { @@ -150,6 +158,10 @@ func (_c *RootCoord_AllocTimestamp_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) AlterAlias(_a0 context.Context, _a1 *milvuspb.AlterAliasRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) (*commonpb.Status, error)); ok { @@ -205,6 +217,10 @@ func (_c *RootCoord_AlterAlias_Call) RunAndReturn(run func(context.Context, *mil func (_m *RootCoord) AlterCollection(_a0 context.Context, _a1 *milvuspb.AlterCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) (*commonpb.Status, error)); ok { @@ -260,6 +276,10 @@ func (_c *RootCoord_AlterCollection_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) AlterDatabase(_a0 context.Context, _a1 *rootcoordpb.AlterDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for AlterDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AlterDatabaseRequest) (*commonpb.Status, error)); ok { @@ -315,6 +335,10 @@ func (_c *RootCoord_AlterDatabase_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) BackupRBAC(_a0 context.Context, _a1 *milvuspb.BackupRBACMetaRequest) (*milvuspb.BackupRBACMetaResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for BackupRBAC") + } + var r0 *milvuspb.BackupRBACMetaResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.BackupRBACMetaRequest) (*milvuspb.BackupRBACMetaResponse, error)); ok { @@ -370,6 +394,10 @@ func (_c *RootCoord_BackupRBAC_Call) RunAndReturn(run func(context.Context, *mil func (_m *RootCoord) CheckHealth(_a0 context.Context, _a1 *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error)); ok { @@ -425,6 +453,10 @@ func (_c *RootCoord_CheckHealth_Call) RunAndReturn(run func(context.Context, *mi func (_m *RootCoord) CreateAlias(_a0 context.Context, _a1 *milvuspb.CreateAliasRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) (*commonpb.Status, error)); ok { @@ -480,6 +512,10 @@ func (_c *RootCoord_CreateAlias_Call) RunAndReturn(run func(context.Context, *mi func (_m *RootCoord) CreateCollection(_a0 context.Context, _a1 *milvuspb.CreateCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) (*commonpb.Status, error)); ok { @@ -535,6 +571,10 @@ func (_c *RootCoord_CreateCollection_Call) RunAndReturn(run func(context.Context func (_m *RootCoord) CreateCredential(_a0 context.Context, _a1 *internalpb.CredentialInfo) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo) (*commonpb.Status, error)); ok { @@ -590,6 +630,10 @@ func (_c *RootCoord_CreateCredential_Call) RunAndReturn(run func(context.Context func (_m *RootCoord) CreateDatabase(_a0 context.Context, _a1 *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest) (*commonpb.Status, error)); ok { @@ -645,6 +689,10 @@ func (_c *RootCoord_CreateDatabase_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) CreatePartition(_a0 context.Context, _a1 *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreatePartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) (*commonpb.Status, error)); ok { @@ -700,6 +748,10 @@ func (_c *RootCoord_CreatePartition_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) CreateRole(_a0 context.Context, _a1 *milvuspb.CreateRoleRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CreateRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) (*commonpb.Status, error)); ok { @@ -755,6 +807,10 @@ func (_c *RootCoord_CreateRole_Call) RunAndReturn(run func(context.Context, *mil func (_m *RootCoord) DeleteCredential(_a0 context.Context, _a1 *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DeleteCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error)); ok { @@ -810,6 +866,10 @@ func (_c *RootCoord_DeleteCredential_Call) RunAndReturn(run func(context.Context func (_m *RootCoord) DescribeAlias(_a0 context.Context, _a1 *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeAlias") + } + var r0 *milvuspb.DescribeAliasResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeAliasRequest) (*milvuspb.DescribeAliasResponse, error)); ok { @@ -865,6 +925,10 @@ func (_c *RootCoord_DescribeAlias_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) DescribeCollection(_a0 context.Context, _a1 *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeCollection") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -920,6 +984,10 @@ func (_c *RootCoord_DescribeCollection_Call) RunAndReturn(run func(context.Conte func (_m *RootCoord) DescribeCollectionInternal(_a0 context.Context, _a1 *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeCollectionInternal") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -975,6 +1043,10 @@ func (_c *RootCoord_DescribeCollectionInternal_Call) RunAndReturn(run func(conte func (_m *RootCoord) DescribeDatabase(_a0 context.Context, _a1 *rootcoordpb.DescribeDatabaseRequest) (*rootcoordpb.DescribeDatabaseResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DescribeDatabase") + } + var r0 *rootcoordpb.DescribeDatabaseResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.DescribeDatabaseRequest) (*rootcoordpb.DescribeDatabaseResponse, error)); ok { @@ -1030,6 +1102,10 @@ func (_c *RootCoord_DescribeDatabase_Call) RunAndReturn(run func(context.Context func (_m *RootCoord) DropAlias(_a0 context.Context, _a1 *milvuspb.DropAliasRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) (*commonpb.Status, error)); ok { @@ -1085,6 +1161,10 @@ func (_c *RootCoord_DropAlias_Call) RunAndReturn(run func(context.Context, *milv func (_m *RootCoord) DropCollection(_a0 context.Context, _a1 *milvuspb.DropCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) (*commonpb.Status, error)); ok { @@ -1140,6 +1220,10 @@ func (_c *RootCoord_DropCollection_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) DropDatabase(_a0 context.Context, _a1 *milvuspb.DropDatabaseRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest) (*commonpb.Status, error)); ok { @@ -1195,6 +1279,10 @@ func (_c *RootCoord_DropDatabase_Call) RunAndReturn(run func(context.Context, *m func (_m *RootCoord) DropPartition(_a0 context.Context, _a1 *milvuspb.DropPartitionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropPartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) (*commonpb.Status, error)); ok { @@ -1250,6 +1338,10 @@ func (_c *RootCoord_DropPartition_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) DropRole(_a0 context.Context, _a1 *milvuspb.DropRoleRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for DropRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) (*commonpb.Status, error)); ok { @@ -1305,6 +1397,10 @@ func (_c *RootCoord_DropRole_Call) RunAndReturn(run func(context.Context, *milvu func (_m *RootCoord) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -1360,6 +1456,10 @@ func (_c *RootCoord_GetComponentStates_Call) RunAndReturn(run func(context.Conte func (_m *RootCoord) GetCredential(_a0 context.Context, _a1 *rootcoordpb.GetCredentialRequest) (*rootcoordpb.GetCredentialResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetCredential") + } + var r0 *rootcoordpb.GetCredentialResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetCredentialRequest) (*rootcoordpb.GetCredentialResponse, error)); ok { @@ -1415,6 +1515,10 @@ func (_c *RootCoord_GetCredential_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) GetMetrics(ctx context.Context, req *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -1470,6 +1574,10 @@ func (_c *RootCoord_GetMetrics_Call) RunAndReturn(run func(context.Context, *mil func (_m *RootCoord) GetPChannelInfo(_a0 context.Context, _a1 *rootcoordpb.GetPChannelInfoRequest) (*rootcoordpb.GetPChannelInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetPChannelInfo") + } + var r0 *rootcoordpb.GetPChannelInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetPChannelInfoRequest) (*rootcoordpb.GetPChannelInfoResponse, error)); ok { @@ -1525,6 +1633,10 @@ func (_c *RootCoord_GetPChannelInfo_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -1580,6 +1692,10 @@ func (_c *RootCoord_GetStatisticsChannel_Call) RunAndReturn(run func(context.Con func (_m *RootCoord) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -1635,6 +1751,10 @@ func (_c *RootCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.Conte func (_m *RootCoord) HasCollection(_a0 context.Context, _a1 *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for HasCollection") + } + var r0 *milvuspb.BoolResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error)); ok { @@ -1690,6 +1810,10 @@ func (_c *RootCoord_HasCollection_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) HasPartition(_a0 context.Context, _a1 *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for HasPartition") + } + var r0 *milvuspb.BoolResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error)); ok { @@ -1745,6 +1869,10 @@ func (_c *RootCoord_HasPartition_Call) RunAndReturn(run func(context.Context, *m func (_m *RootCoord) Init() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1786,6 +1914,10 @@ func (_c *RootCoord_Init_Call) RunAndReturn(run func() error) *RootCoord_Init_Ca func (_m *RootCoord) InvalidateCollectionMetaCache(_a0 context.Context, _a1 *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for InvalidateCollectionMetaCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error)); ok { @@ -1841,6 +1973,10 @@ func (_c *RootCoord_InvalidateCollectionMetaCache_Call) RunAndReturn(run func(co func (_m *RootCoord) ListAliases(_a0 context.Context, _a1 *milvuspb.ListAliasesRequest) (*milvuspb.ListAliasesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListAliases") + } + var r0 *milvuspb.ListAliasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListAliasesRequest) (*milvuspb.ListAliasesResponse, error)); ok { @@ -1896,6 +2032,10 @@ func (_c *RootCoord_ListAliases_Call) RunAndReturn(run func(context.Context, *mi func (_m *RootCoord) ListCredUsers(_a0 context.Context, _a1 *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListCredUsers") + } + var r0 *milvuspb.ListCredUsersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error)); ok { @@ -1951,6 +2091,10 @@ func (_c *RootCoord_ListCredUsers_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) ListDatabases(_a0 context.Context, _a1 *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListDatabases") + } + var r0 *milvuspb.ListDatabasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest) (*milvuspb.ListDatabasesResponse, error)); ok { @@ -2006,6 +2150,10 @@ func (_c *RootCoord_ListDatabases_Call) RunAndReturn(run func(context.Context, * func (_m *RootCoord) ListPolicy(_a0 context.Context, _a1 *internalpb.ListPolicyRequest) (*internalpb.ListPolicyResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ListPolicy") + } + var r0 *internalpb.ListPolicyResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListPolicyRequest) (*internalpb.ListPolicyResponse, error)); ok { @@ -2061,6 +2209,10 @@ func (_c *RootCoord_ListPolicy_Call) RunAndReturn(run func(context.Context, *int func (_m *RootCoord) OperatePrivilege(_a0 context.Context, _a1 *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for OperatePrivilege") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error)); ok { @@ -2116,6 +2268,10 @@ func (_c *RootCoord_OperatePrivilege_Call) RunAndReturn(run func(context.Context func (_m *RootCoord) OperateUserRole(_a0 context.Context, _a1 *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for OperateUserRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error)); ok { @@ -2171,6 +2327,10 @@ func (_c *RootCoord_OperateUserRole_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) Register() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -2212,6 +2372,10 @@ func (_c *RootCoord_Register_Call) RunAndReturn(run func() error) *RootCoord_Reg func (_m *RootCoord) RenameCollection(_a0 context.Context, _a1 *milvuspb.RenameCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for RenameCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) (*commonpb.Status, error)); ok { @@ -2267,6 +2431,10 @@ func (_c *RootCoord_RenameCollection_Call) RunAndReturn(run func(context.Context func (_m *RootCoord) RestoreRBAC(_a0 context.Context, _a1 *milvuspb.RestoreRBACMetaRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for RestoreRBAC") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RestoreRBACMetaRequest) (*commonpb.Status, error)); ok { @@ -2322,6 +2490,10 @@ func (_c *RootCoord_RestoreRBAC_Call) RunAndReturn(run func(context.Context, *mi func (_m *RootCoord) SelectGrant(_a0 context.Context, _a1 *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SelectGrant") + } + var r0 *milvuspb.SelectGrantResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error)); ok { @@ -2377,6 +2549,10 @@ func (_c *RootCoord_SelectGrant_Call) RunAndReturn(run func(context.Context, *mi func (_m *RootCoord) SelectRole(_a0 context.Context, _a1 *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SelectRole") + } + var r0 *milvuspb.SelectRoleResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error)); ok { @@ -2432,6 +2608,10 @@ func (_c *RootCoord_SelectRole_Call) RunAndReturn(run func(context.Context, *mil func (_m *RootCoord) SelectUser(_a0 context.Context, _a1 *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SelectUser") + } + var r0 *milvuspb.SelectUserResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error)); ok { @@ -2520,6 +2700,10 @@ func (_c *RootCoord_SetAddress_Call) RunAndReturn(run func(string)) *RootCoord_S func (_m *RootCoord) SetDataCoordClient(dataCoord types.DataCoordClient) error { ret := _m.Called(dataCoord) + if len(ret) == 0 { + panic("no return value specified for SetDataCoordClient") + } + var r0 error if rf, ok := ret.Get(0).(func(types.DataCoordClient) error); ok { r0 = rf(dataCoord) @@ -2628,6 +2812,10 @@ func (_c *RootCoord_SetProxyCreator_Call) RunAndReturn(run func(func(context.Con func (_m *RootCoord) SetQueryCoordClient(queryCoord types.QueryCoordClient) error { ret := _m.Called(queryCoord) + if len(ret) == 0 { + panic("no return value specified for SetQueryCoordClient") + } + var r0 error if rf, ok := ret.Get(0).(func(types.QueryCoordClient) error); ok { r0 = rf(queryCoord) @@ -2703,6 +2891,10 @@ func (_c *RootCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *R func (_m *RootCoord) ShowCollections(_a0 context.Context, _a1 *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowCollections") + } + var r0 *milvuspb.ShowCollectionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error)); ok { @@ -2758,6 +2950,10 @@ func (_c *RootCoord_ShowCollections_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -2813,6 +3009,10 @@ func (_c *RootCoord_ShowConfigurations_Call) RunAndReturn(run func(context.Conte func (_m *RootCoord) ShowPartitions(_a0 context.Context, _a1 *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowPartitions") + } + var r0 *milvuspb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok { @@ -2868,6 +3068,10 @@ func (_c *RootCoord_ShowPartitions_Call) RunAndReturn(run func(context.Context, func (_m *RootCoord) ShowPartitionsInternal(_a0 context.Context, _a1 *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowPartitionsInternal") + } + var r0 *milvuspb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error)); ok { @@ -2923,6 +3127,10 @@ func (_c *RootCoord_ShowPartitionsInternal_Call) RunAndReturn(run func(context.C func (_m *RootCoord) ShowSegments(_a0 context.Context, _a1 *milvuspb.ShowSegmentsRequest) (*milvuspb.ShowSegmentsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowSegments") + } + var r0 *milvuspb.ShowSegmentsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowSegmentsRequest) (*milvuspb.ShowSegmentsResponse, error)); ok { @@ -2978,6 +3186,10 @@ func (_c *RootCoord_ShowSegments_Call) RunAndReturn(run func(context.Context, *m func (_m *RootCoord) Start() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Start") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -3019,6 +3231,10 @@ func (_c *RootCoord_Start_Call) RunAndReturn(run func() error) *RootCoord_Start_ func (_m *RootCoord) Stop() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Stop") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -3060,6 +3276,10 @@ func (_c *RootCoord_Stop_Call) RunAndReturn(run func() error) *RootCoord_Stop_Ca func (_m *RootCoord) UpdateChannelTimeTick(_a0 context.Context, _a1 *internalpb.ChannelTimeTickMsg) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateChannelTimeTick") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ChannelTimeTickMsg) (*commonpb.Status, error)); ok { @@ -3115,6 +3335,10 @@ func (_c *RootCoord_UpdateChannelTimeTick_Call) RunAndReturn(run func(context.Co func (_m *RootCoord) UpdateCredential(_a0 context.Context, _a1 *internalpb.CredentialInfo) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UpdateCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/mock_rootcoord_client.go b/internal/mocks/mock_rootcoord_client.go index 61dbc772c2..58b0a40671 100644 --- a/internal/mocks/mock_rootcoord_client.go +++ b/internal/mocks/mock_rootcoord_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -44,6 +44,10 @@ func (_m *MockRootCoordClient) AllocID(ctx context.Context, in *rootcoordpb.Allo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AllocID") + } + var r0 *rootcoordpb.AllocIDResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocIDRequest, ...grpc.CallOption) (*rootcoordpb.AllocIDResponse, error)); ok { @@ -114,6 +118,10 @@ func (_m *MockRootCoordClient) AllocTimestamp(ctx context.Context, in *rootcoord _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AllocTimestamp") + } + var r0 *rootcoordpb.AllocTimestampResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocTimestampRequest, ...grpc.CallOption) (*rootcoordpb.AllocTimestampResponse, error)); ok { @@ -184,6 +192,10 @@ func (_m *MockRootCoordClient) AlterAlias(ctx context.Context, in *milvuspb.Alte _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AlterAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -254,6 +266,10 @@ func (_m *MockRootCoordClient) AlterCollection(ctx context.Context, in *milvuspb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AlterCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -324,6 +340,10 @@ func (_m *MockRootCoordClient) AlterDatabase(ctx context.Context, in *rootcoordp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AlterDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AlterDatabaseRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -394,6 +414,10 @@ func (_m *MockRootCoordClient) BackupRBAC(ctx context.Context, in *milvuspb.Back _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for BackupRBAC") + } + var r0 *milvuspb.BackupRBACMetaResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.BackupRBACMetaRequest, ...grpc.CallOption) (*milvuspb.BackupRBACMetaResponse, error)); ok { @@ -464,6 +488,10 @@ func (_m *MockRootCoordClient) CheckHealth(ctx context.Context, in *milvuspb.Che _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CheckHealth") + } + var r0 *milvuspb.CheckHealthResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest, ...grpc.CallOption) (*milvuspb.CheckHealthResponse, error)); ok { @@ -527,6 +555,10 @@ func (_c *MockRootCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co func (_m *MockRootCoordClient) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -575,6 +607,10 @@ func (_m *MockRootCoordClient) CreateAlias(ctx context.Context, in *milvuspb.Cre _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -645,6 +681,10 @@ func (_m *MockRootCoordClient) CreateCollection(ctx context.Context, in *milvusp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -715,6 +755,10 @@ func (_m *MockRootCoordClient) CreateCredential(ctx context.Context, in *interna _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -785,6 +829,10 @@ func (_m *MockRootCoordClient) CreateDatabase(ctx context.Context, in *milvuspb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateDatabaseRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -855,6 +903,10 @@ func (_m *MockRootCoordClient) CreatePartition(ctx context.Context, in *milvuspb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreatePartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -925,6 +977,10 @@ func (_m *MockRootCoordClient) CreateRole(ctx context.Context, in *milvuspb.Crea _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CreateRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -995,6 +1051,10 @@ func (_m *MockRootCoordClient) DeleteCredential(ctx context.Context, in *milvusp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DeleteCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1065,6 +1125,10 @@ func (_m *MockRootCoordClient) DescribeAlias(ctx context.Context, in *milvuspb.D _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DescribeAlias") + } + var r0 *milvuspb.DescribeAliasResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeAliasRequest, ...grpc.CallOption) (*milvuspb.DescribeAliasResponse, error)); ok { @@ -1135,6 +1199,10 @@ func (_m *MockRootCoordClient) DescribeCollection(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DescribeCollection") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest, ...grpc.CallOption) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -1205,6 +1273,10 @@ func (_m *MockRootCoordClient) DescribeCollectionInternal(ctx context.Context, i _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DescribeCollectionInternal") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest, ...grpc.CallOption) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -1275,6 +1347,10 @@ func (_m *MockRootCoordClient) DescribeDatabase(ctx context.Context, in *rootcoo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DescribeDatabase") + } + var r0 *rootcoordpb.DescribeDatabaseResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.DescribeDatabaseRequest, ...grpc.CallOption) (*rootcoordpb.DescribeDatabaseResponse, error)); ok { @@ -1345,6 +1421,10 @@ func (_m *MockRootCoordClient) DropAlias(ctx context.Context, in *milvuspb.DropA _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropAlias") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1415,6 +1495,10 @@ func (_m *MockRootCoordClient) DropCollection(ctx context.Context, in *milvuspb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1485,6 +1569,10 @@ func (_m *MockRootCoordClient) DropDatabase(ctx context.Context, in *milvuspb.Dr _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropDatabase") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropDatabaseRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1555,6 +1643,10 @@ func (_m *MockRootCoordClient) DropPartition(ctx context.Context, in *milvuspb.D _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropPartition") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1625,6 +1717,10 @@ func (_m *MockRootCoordClient) DropRole(ctx context.Context, in *milvuspb.DropRo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for DropRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -1695,6 +1791,10 @@ func (_m *MockRootCoordClient) GetComponentStates(ctx context.Context, in *milvu _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest, ...grpc.CallOption) (*milvuspb.ComponentStates, error)); ok { @@ -1765,6 +1865,10 @@ func (_m *MockRootCoordClient) GetCredential(ctx context.Context, in *rootcoordp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetCredential") + } + var r0 *rootcoordpb.GetCredentialResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetCredentialRequest, ...grpc.CallOption) (*rootcoordpb.GetCredentialResponse, error)); ok { @@ -1835,6 +1939,10 @@ func (_m *MockRootCoordClient) GetMetrics(ctx context.Context, in *milvuspb.GetM _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest, ...grpc.CallOption) (*milvuspb.GetMetricsResponse, error)); ok { @@ -1905,6 +2013,10 @@ func (_m *MockRootCoordClient) GetPChannelInfo(ctx context.Context, in *rootcoor _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetPChannelInfo") + } + var r0 *rootcoordpb.GetPChannelInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.GetPChannelInfoRequest, ...grpc.CallOption) (*rootcoordpb.GetPChannelInfoResponse, error)); ok { @@ -1975,6 +2087,10 @@ func (_m *MockRootCoordClient) GetStatisticsChannel(ctx context.Context, in *int _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -2045,6 +2161,10 @@ func (_m *MockRootCoordClient) GetTimeTickChannel(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest, ...grpc.CallOption) (*milvuspb.StringResponse, error)); ok { @@ -2115,6 +2235,10 @@ func (_m *MockRootCoordClient) HasCollection(ctx context.Context, in *milvuspb.H _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for HasCollection") + } + var r0 *milvuspb.BoolResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest, ...grpc.CallOption) (*milvuspb.BoolResponse, error)); ok { @@ -2185,6 +2309,10 @@ func (_m *MockRootCoordClient) HasPartition(ctx context.Context, in *milvuspb.Ha _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for HasPartition") + } + var r0 *milvuspb.BoolResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest, ...grpc.CallOption) (*milvuspb.BoolResponse, error)); ok { @@ -2255,6 +2383,10 @@ func (_m *MockRootCoordClient) InvalidateCollectionMetaCache(ctx context.Context _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for InvalidateCollectionMetaCache") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2325,6 +2457,10 @@ func (_m *MockRootCoordClient) ListAliases(ctx context.Context, in *milvuspb.Lis _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListAliases") + } + var r0 *milvuspb.ListAliasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListAliasesRequest, ...grpc.CallOption) (*milvuspb.ListAliasesResponse, error)); ok { @@ -2395,6 +2531,10 @@ func (_m *MockRootCoordClient) ListCredUsers(ctx context.Context, in *milvuspb.L _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListCredUsers") + } + var r0 *milvuspb.ListCredUsersResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest, ...grpc.CallOption) (*milvuspb.ListCredUsersResponse, error)); ok { @@ -2465,6 +2605,10 @@ func (_m *MockRootCoordClient) ListDatabases(ctx context.Context, in *milvuspb.L _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListDatabases") + } + var r0 *milvuspb.ListDatabasesResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListDatabasesRequest, ...grpc.CallOption) (*milvuspb.ListDatabasesResponse, error)); ok { @@ -2535,6 +2679,10 @@ func (_m *MockRootCoordClient) ListPolicy(ctx context.Context, in *internalpb.Li _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ListPolicy") + } + var r0 *internalpb.ListPolicyResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ListPolicyRequest, ...grpc.CallOption) (*internalpb.ListPolicyResponse, error)); ok { @@ -2605,6 +2753,10 @@ func (_m *MockRootCoordClient) OperatePrivilege(ctx context.Context, in *milvusp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for OperatePrivilege") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2675,6 +2827,10 @@ func (_m *MockRootCoordClient) OperateUserRole(ctx context.Context, in *milvuspb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for OperateUserRole") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2745,6 +2901,10 @@ func (_m *MockRootCoordClient) RenameCollection(ctx context.Context, in *milvusp _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for RenameCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2815,6 +2975,10 @@ func (_m *MockRootCoordClient) RestoreRBAC(ctx context.Context, in *milvuspb.Res _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for RestoreRBAC") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RestoreRBACMetaRequest, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -2885,6 +3049,10 @@ func (_m *MockRootCoordClient) SelectGrant(ctx context.Context, in *milvuspb.Sel _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SelectGrant") + } + var r0 *milvuspb.SelectGrantResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest, ...grpc.CallOption) (*milvuspb.SelectGrantResponse, error)); ok { @@ -2955,6 +3123,10 @@ func (_m *MockRootCoordClient) SelectRole(ctx context.Context, in *milvuspb.Sele _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SelectRole") + } + var r0 *milvuspb.SelectRoleResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest, ...grpc.CallOption) (*milvuspb.SelectRoleResponse, error)); ok { @@ -3025,6 +3197,10 @@ func (_m *MockRootCoordClient) SelectUser(ctx context.Context, in *milvuspb.Sele _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for SelectUser") + } + var r0 *milvuspb.SelectUserResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest, ...grpc.CallOption) (*milvuspb.SelectUserResponse, error)); ok { @@ -3095,6 +3271,10 @@ func (_m *MockRootCoordClient) ShowCollections(ctx context.Context, in *milvuspb _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowCollections") + } + var r0 *milvuspb.ShowCollectionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest, ...grpc.CallOption) (*milvuspb.ShowCollectionsResponse, error)); ok { @@ -3165,6 +3345,10 @@ func (_m *MockRootCoordClient) ShowConfigurations(ctx context.Context, in *inter _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest, ...grpc.CallOption) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -3235,6 +3419,10 @@ func (_m *MockRootCoordClient) ShowPartitions(ctx context.Context, in *milvuspb. _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowPartitions") + } + var r0 *milvuspb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest, ...grpc.CallOption) (*milvuspb.ShowPartitionsResponse, error)); ok { @@ -3305,6 +3493,10 @@ func (_m *MockRootCoordClient) ShowPartitionsInternal(ctx context.Context, in *m _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowPartitionsInternal") + } + var r0 *milvuspb.ShowPartitionsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest, ...grpc.CallOption) (*milvuspb.ShowPartitionsResponse, error)); ok { @@ -3375,6 +3567,10 @@ func (_m *MockRootCoordClient) ShowSegments(ctx context.Context, in *milvuspb.Sh _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for ShowSegments") + } + var r0 *milvuspb.ShowSegmentsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowSegmentsRequest, ...grpc.CallOption) (*milvuspb.ShowSegmentsResponse, error)); ok { @@ -3445,6 +3641,10 @@ func (_m *MockRootCoordClient) UpdateChannelTimeTick(ctx context.Context, in *in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateChannelTimeTick") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ChannelTimeTickMsg, ...grpc.CallOption) (*commonpb.Status, error)); ok { @@ -3515,6 +3715,10 @@ func (_m *MockRootCoordClient) UpdateCredential(ctx context.Context, in *interna _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateCredential") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.CredentialInfo, ...grpc.CallOption) (*commonpb.Status, error)); ok { diff --git a/internal/mocks/streamingcoord/mock_client/mock_Client.go b/internal/mocks/streamingcoord/mock_client/mock_Client.go index 719b05c716..02923644d2 100644 --- a/internal/mocks/streamingcoord/mock_client/mock_Client.go +++ b/internal/mocks/streamingcoord/mock_client/mock_Client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_client @@ -24,6 +24,10 @@ func (_m *MockClient) EXPECT() *MockClient_Expecter { func (_m *MockClient) Assignment() client.AssignmentService { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Assignment") + } + var r0 client.AssignmentService if rf, ok := ret.Get(0).(func() client.AssignmentService); ok { r0 = rf() diff --git a/internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go b/internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go index dce56cf458..eb48a01a75 100644 --- a/internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go +++ b/internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_balancer @@ -60,6 +60,10 @@ func (_c *MockBalancer_Close_Call) RunAndReturn(run func()) *MockBalancer_Close_ func (_m *MockBalancer) MarkAsUnavailable(ctx context.Context, pChannels []types.PChannelInfo) error { ret := _m.Called(ctx, pChannels) + if len(ret) == 0 { + panic("no return value specified for MarkAsUnavailable") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []types.PChannelInfo) error); ok { r0 = rf(ctx, pChannels) @@ -103,6 +107,10 @@ func (_c *MockBalancer_MarkAsUnavailable_Call) RunAndReturn(run func(context.Con func (_m *MockBalancer) Trigger(ctx context.Context) error { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for Trigger") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context) error); ok { r0 = rf(ctx) @@ -145,6 +153,10 @@ func (_c *MockBalancer_Trigger_Call) RunAndReturn(run func(context.Context) erro func (_m *MockBalancer) WatchChannelAssignments(ctx context.Context, cb func(typeutil.VersionInt64Pair, []types.PChannelInfoAssigned) error) error { ret := _m.Called(ctx, cb) + if len(ret) == 0 { + panic("no return value specified for WatchChannelAssignments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, func(typeutil.VersionInt64Pair, []types.PChannelInfoAssigned) error) error); ok { r0 = rf(ctx, cb) diff --git a/internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go b/internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go index f51798aeea..084a0ad6a2 100644 --- a/internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go +++ b/internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_assignment @@ -58,6 +58,10 @@ func (_c *MockWatcher_Close_Call) RunAndReturn(run func()) *MockWatcher_Close_Ca func (_m *MockWatcher) Get(ctx context.Context, channel string) *types.PChannelInfoAssigned { ret := _m.Called(ctx, channel) + if len(ret) == 0 { + panic("no return value specified for Get") + } + var r0 *types.PChannelInfoAssigned if rf, ok := ret.Get(0).(func(context.Context, string) *types.PChannelInfoAssigned); ok { r0 = rf(ctx, channel) @@ -103,6 +107,10 @@ func (_c *MockWatcher_Get_Call) RunAndReturn(run func(context.Context, string) * func (_m *MockWatcher) Watch(ctx context.Context, channel string, previous *types.PChannelInfoAssigned) error { ret := _m.Called(ctx, channel, previous) + if len(ret) == 0 { + panic("no return value specified for Watch") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, *types.PChannelInfoAssigned) error); ok { r0 = rf(ctx, channel, previous) diff --git a/internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go b/internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go index f5c8ca5925..efa7eb0f7f 100644 --- a/internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go +++ b/internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_consumer @@ -53,6 +53,10 @@ func (_c *MockConsumer_Close_Call) RunAndReturn(run func()) *MockConsumer_Close_ func (_m *MockConsumer) Done() <-chan struct{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Done") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { r0 = rf() @@ -96,6 +100,10 @@ func (_c *MockConsumer_Done_Call) RunAndReturn(run func() <-chan struct{}) *Mock func (_m *MockConsumer) Error() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Error") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() diff --git a/internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go b/internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go index 7f639bf1ea..b215ccd60c 100644 --- a/internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go +++ b/internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_producer @@ -28,6 +28,10 @@ func (_m *MockProducer) EXPECT() *MockProducer_Expecter { func (_m *MockProducer) Assignment() types.PChannelInfoAssigned { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Assignment") + } + var r0 types.PChannelInfoAssigned if rf, ok := ret.Get(0).(func() types.PChannelInfoAssigned); ok { r0 = rf() @@ -69,6 +73,10 @@ func (_c *MockProducer_Assignment_Call) RunAndReturn(run func() types.PChannelIn func (_m *MockProducer) Available() <-chan struct{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Available") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { r0 = rf() @@ -144,6 +152,10 @@ func (_c *MockProducer_Close_Call) RunAndReturn(run func()) *MockProducer_Close_ func (_m *MockProducer) IsAvailable() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for IsAvailable") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -185,6 +197,10 @@ func (_c *MockProducer_IsAvailable_Call) RunAndReturn(run func() bool) *MockProd func (_m *MockProducer) Produce(ctx context.Context, msg message.MutableMessage) (*types.AppendResult, error) { ret := _m.Called(ctx, msg) + if len(ret) == 0 { + panic("no return value specified for Produce") + } + var r0 *types.AppendResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage) (*types.AppendResult, error)); ok { diff --git a/internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go b/internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go index 3cf3fb40b9..dd5d140160 100644 --- a/internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go +++ b/internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_handler @@ -63,6 +63,10 @@ func (_c *MockHandlerClient_Close_Call) RunAndReturn(run func()) *MockHandlerCli func (_m *MockHandlerClient) CreateConsumer(ctx context.Context, opts *handler.ConsumerOptions) (consumer.Consumer, error) { ret := _m.Called(ctx, opts) + if len(ret) == 0 { + panic("no return value specified for CreateConsumer") + } + var r0 consumer.Consumer var r1 error if rf, ok := ret.Get(0).(func(context.Context, *handler.ConsumerOptions) (consumer.Consumer, error)); ok { @@ -118,6 +122,10 @@ func (_c *MockHandlerClient_CreateConsumer_Call) RunAndReturn(run func(context.C func (_m *MockHandlerClient) CreateProducer(ctx context.Context, opts *handler.ProducerOptions) (producer.Producer, error) { ret := _m.Called(ctx, opts) + if len(ret) == 0 { + panic("no return value specified for CreateProducer") + } + var r0 producer.Producer var r1 error if rf, ok := ret.Get(0).(func(context.Context, *handler.ProducerOptions) (producer.Producer, error)); ok { diff --git a/internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go b/internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go index 3ded946964..0fd793afa7 100644 --- a/internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go +++ b/internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_manager @@ -27,6 +27,10 @@ func (_m *MockManagerClient) EXPECT() *MockManagerClient_Expecter { func (_m *MockManagerClient) Assign(ctx context.Context, pchannel types.PChannelInfoAssigned) error { ret := _m.Called(ctx, pchannel) + if len(ret) == 0 { + panic("no return value specified for Assign") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfoAssigned) error); ok { r0 = rf(ctx, pchannel) @@ -102,6 +106,10 @@ func (_c *MockManagerClient_Close_Call) RunAndReturn(run func()) *MockManagerCli func (_m *MockManagerClient) CollectAllStatus(ctx context.Context) (map[int64]*types.StreamingNodeStatus, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for CollectAllStatus") + } + var r0 map[int64]*types.StreamingNodeStatus var r1 error if rf, ok := ret.Get(0).(func(context.Context) (map[int64]*types.StreamingNodeStatus, error)); ok { @@ -156,6 +164,10 @@ func (_c *MockManagerClient_CollectAllStatus_Call) RunAndReturn(run func(context func (_m *MockManagerClient) Remove(ctx context.Context, pchannel types.PChannelInfoAssigned) error { ret := _m.Called(ctx, pchannel) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfoAssigned) error); ok { r0 = rf(ctx, pchannel) @@ -199,6 +211,10 @@ func (_c *MockManagerClient_Remove_Call) RunAndReturn(run func(context.Context, func (_m *MockManagerClient) WatchNodeChanged(ctx context.Context) (<-chan struct{}, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for WatchNodeChanged") + } + var r0 <-chan struct{} var r1 error if rf, ok := ret.Get(0).(func(context.Context) (<-chan struct{}, error)); ok { diff --git a/internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go b/internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go index 813e5dfac1..a1ab70673b 100644 --- a/internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go +++ b/internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_flusher @@ -24,6 +24,10 @@ func (_m *MockFlusher) EXPECT() *MockFlusher_Expecter { func (_m *MockFlusher) RegisterPChannel(pchannel string, w wal.WAL) error { ret := _m.Called(pchannel, w) + if len(ret) == 0 { + panic("no return value specified for RegisterPChannel") + } + var r0 error if rf, ok := ret.Get(0).(func(string, wal.WAL) error); ok { r0 = rf(pchannel, w) diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_Opener.go b/internal/mocks/streamingnode/server/mock_wal/mock_Opener.go index 868a981dcd..808d1bd10f 100644 --- a/internal/mocks/streamingnode/server/mock_wal/mock_Opener.go +++ b/internal/mocks/streamingnode/server/mock_wal/mock_Opener.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_wal @@ -58,6 +58,10 @@ func (_c *MockOpener_Close_Call) RunAndReturn(run func()) *MockOpener_Close_Call func (_m *MockOpener) Open(ctx context.Context, opt *wal.OpenOption) (wal.WAL, error) { ret := _m.Called(ctx, opt) + if len(ret) == 0 { + panic("no return value specified for Open") + } + var r0 wal.WAL var r1 error if rf, ok := ret.Get(0).(func(context.Context, *wal.OpenOption) (wal.WAL, error)); ok { diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go b/internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go index 2f2b3d0c76..92a243cb82 100644 --- a/internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go +++ b/internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_wal @@ -24,6 +24,10 @@ func (_m *MockOpenerBuilder) EXPECT() *MockOpenerBuilder_Expecter { func (_m *MockOpenerBuilder) Build() (wal.Opener, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Build") + } + var r0 wal.Opener var r1 error if rf, ok := ret.Get(0).(func() (wal.Opener, error)); ok { @@ -77,6 +81,10 @@ func (_c *MockOpenerBuilder_Build_Call) RunAndReturn(run func() (wal.Opener, err func (_m *MockOpenerBuilder) Name() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Name") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go b/internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go index 25fd0a2f9f..5f86b858ad 100644 --- a/internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go +++ b/internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_wal @@ -26,6 +26,10 @@ func (_m *MockScanner) EXPECT() *MockScanner_Expecter { func (_m *MockScanner) Chan() <-chan message.ImmutableMessage { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Chan") + } + var r0 <-chan message.ImmutableMessage if rf, ok := ret.Get(0).(func() <-chan message.ImmutableMessage); ok { r0 = rf() @@ -69,6 +73,10 @@ func (_c *MockScanner_Chan_Call) RunAndReturn(run func() <-chan message.Immutabl func (_m *MockScanner) Channel() types.PChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Channel") + } + var r0 types.PChannelInfo if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok { r0 = rf() @@ -110,6 +118,10 @@ func (_c *MockScanner_Channel_Call) RunAndReturn(run func() types.PChannelInfo) func (_m *MockScanner) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -151,6 +163,10 @@ func (_c *MockScanner_Close_Call) RunAndReturn(run func() error) *MockScanner_Cl func (_m *MockScanner) Done() <-chan struct{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Done") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { r0 = rf() @@ -194,6 +210,10 @@ func (_c *MockScanner_Done_Call) RunAndReturn(run func() <-chan struct{}) *MockS func (_m *MockScanner) Error() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Error") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_WAL.go b/internal/mocks/streamingnode/server/mock_wal/mock_WAL.go index 0a650a3d9c..ce54d77c61 100644 --- a/internal/mocks/streamingnode/server/mock_wal/mock_WAL.go +++ b/internal/mocks/streamingnode/server/mock_wal/mock_WAL.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_wal @@ -30,6 +30,10 @@ func (_m *MockWAL) EXPECT() *MockWAL_Expecter { func (_m *MockWAL) Append(ctx context.Context, msg message.MutableMessage) (*types.AppendResult, error) { ret := _m.Called(ctx, msg) + if len(ret) == 0 { + panic("no return value specified for Append") + } + var r0 *types.AppendResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage) (*types.AppendResult, error)); ok { @@ -120,6 +124,10 @@ func (_c *MockWAL_AppendAsync_Call) RunAndReturn(run func(context.Context, messa func (_m *MockWAL) Available() <-chan struct{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Available") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { r0 = rf() @@ -163,6 +171,10 @@ func (_c *MockWAL_Available_Call) RunAndReturn(run func() <-chan struct{}) *Mock func (_m *MockWAL) Channel() types.PChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Channel") + } + var r0 types.PChannelInfo if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok { r0 = rf() @@ -236,6 +248,10 @@ func (_c *MockWAL_Close_Call) RunAndReturn(run func()) *MockWAL_Close_Call { func (_m *MockWAL) IsAvailable() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for IsAvailable") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -277,6 +293,10 @@ func (_c *MockWAL_IsAvailable_Call) RunAndReturn(run func() bool) *MockWAL_IsAva func (_m *MockWAL) Read(ctx context.Context, deliverPolicy wal.ReadOption) (wal.Scanner, error) { ret := _m.Called(ctx, deliverPolicy) + if len(ret) == 0 { + panic("no return value specified for Read") + } + var r0 wal.Scanner var r1 error if rf, ok := ret.Get(0).(func(context.Context, wal.ReadOption) (wal.Scanner, error)); ok { @@ -332,6 +352,10 @@ func (_c *MockWAL_Read_Call) RunAndReturn(run func(context.Context, wal.ReadOpti func (_m *MockWAL) WALName() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for WALName") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go b/internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go index 4c12954e6c..7d06eacb29 100644 --- a/internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go +++ b/internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_walmanager @@ -60,6 +60,10 @@ func (_c *MockManager_Close_Call) RunAndReturn(run func()) *MockManager_Close_Ca func (_m *MockManager) GetAllAvailableChannels() ([]types.PChannelInfo, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetAllAvailableChannels") + } + var r0 []types.PChannelInfo var r1 error if rf, ok := ret.Get(0).(func() ([]types.PChannelInfo, error)); ok { @@ -113,6 +117,10 @@ func (_c *MockManager_GetAllAvailableChannels_Call) RunAndReturn(run func() ([]t func (_m *MockManager) GetAvailableWAL(channel types.PChannelInfo) (wal.WAL, error) { ret := _m.Called(channel) + if len(ret) == 0 { + panic("no return value specified for GetAvailableWAL") + } + var r0 wal.WAL var r1 error if rf, ok := ret.Get(0).(func(types.PChannelInfo) (wal.WAL, error)); ok { @@ -167,6 +175,10 @@ func (_c *MockManager_GetAvailableWAL_Call) RunAndReturn(run func(types.PChannel func (_m *MockManager) Open(ctx context.Context, channel types.PChannelInfo) error { ret := _m.Called(ctx, channel) + if len(ret) == 0 { + panic("no return value specified for Open") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfo) error); ok { r0 = rf(ctx, channel) @@ -210,6 +222,10 @@ func (_c *MockManager_Open_Call) RunAndReturn(run func(context.Context, types.PC func (_m *MockManager) Remove(ctx context.Context, channel types.PChannelInfo) error { ret := _m.Called(ctx, channel) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfo) error); ok { r0 = rf(ctx, channel) diff --git a/internal/mocks/streamingnode/server/wal/interceptors/segment/mock_inspector/mock_SealOperator.go b/internal/mocks/streamingnode/server/wal/interceptors/segment/mock_inspector/mock_SealOperator.go index 2346b3cd09..33e1babadc 100644 --- a/internal/mocks/streamingnode/server/wal/interceptors/segment/mock_inspector/mock_SealOperator.go +++ b/internal/mocks/streamingnode/server/wal/interceptors/segment/mock_inspector/mock_SealOperator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_inspector @@ -29,6 +29,10 @@ func (_m *MockSealOperator) EXPECT() *MockSealOperator_Expecter { func (_m *MockSealOperator) Channel() types.PChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Channel") + } + var r0 types.PChannelInfo if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok { r0 = rf() @@ -70,6 +74,10 @@ func (_c *MockSealOperator_Channel_Call) RunAndReturn(run func() types.PChannelI func (_m *MockSealOperator) IsNoWaitSeal() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for IsNoWaitSeal") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() diff --git a/internal/mocks/streamingnode/server/wal/interceptors/timetick/mock_inspector/mock_TimeTickSyncOperator.go b/internal/mocks/streamingnode/server/wal/interceptors/timetick/mock_inspector/mock_TimeTickSyncOperator.go index 22ed2dde74..b273f7da7a 100644 --- a/internal/mocks/streamingnode/server/wal/interceptors/timetick/mock_inspector/mock_TimeTickSyncOperator.go +++ b/internal/mocks/streamingnode/server/wal/interceptors/timetick/mock_inspector/mock_TimeTickSyncOperator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_inspector @@ -28,6 +28,10 @@ func (_m *MockTimeTickSyncOperator) EXPECT() *MockTimeTickSyncOperator_Expecter func (_m *MockTimeTickSyncOperator) Channel() types.PChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Channel") + } + var r0 types.PChannelInfo if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok { r0 = rf() @@ -102,6 +106,10 @@ func (_c *MockTimeTickSyncOperator_Sync_Call) RunAndReturn(run func(context.Cont func (_m *MockTimeTickSyncOperator) TimeTickNotifier() *inspector.TimeTickNotifier { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TimeTickNotifier") + } + var r0 *inspector.TimeTickNotifier if rf, ok := ret.Get(0).(func() *inspector.TimeTickNotifier); ok { r0 = rf() diff --git a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go index ad228f9de5..72811537a7 100644 --- a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go +++ b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_interceptors @@ -59,6 +59,10 @@ func (_c *MockInterceptor_Close_Call) RunAndReturn(run func()) *MockInterceptor_ func (_m *MockInterceptor) DoAppend(ctx context.Context, msg message.MutableMessage, append func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error) { ret := _m.Called(ctx, msg, append) + if len(ret) == 0 { + panic("no return value specified for DoAppend") + } + var r0 message.MessageID var r1 error if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage, func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error)); ok { diff --git a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go index 95bdc5c1be..512d567e02 100644 --- a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go +++ b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_interceptors @@ -24,6 +24,10 @@ func (_m *MockInterceptorBuilder) EXPECT() *MockInterceptorBuilder_Expecter { func (_m *MockInterceptorBuilder) Build(param interceptors.InterceptorBuildParam) interceptors.Interceptor { ret := _m.Called(param) + if len(ret) == 0 { + panic("no return value specified for Build") + } + var r0 interceptors.Interceptor if rf, ok := ret.Get(0).(func(interceptors.InterceptorBuildParam) interceptors.Interceptor); ok { r0 = rf(param) diff --git a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go index 6ff37f6d22..ccc49f8eef 100644 --- a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go +++ b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_interceptors @@ -59,6 +59,10 @@ func (_c *MockInterceptorWithReady_Close_Call) RunAndReturn(run func()) *MockInt func (_m *MockInterceptorWithReady) DoAppend(ctx context.Context, msg message.MutableMessage, append func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error) { ret := _m.Called(ctx, msg, append) + if len(ret) == 0 { + panic("no return value specified for DoAppend") + } + var r0 message.MessageID var r1 error if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage, func(context.Context, message.MutableMessage) (message.MessageID, error)) (message.MessageID, error)); ok { @@ -115,6 +119,10 @@ func (_c *MockInterceptorWithReady_DoAppend_Call) RunAndReturn(run func(context. func (_m *MockInterceptorWithReady) Ready() <-chan struct{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Ready") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { r0 = rf() diff --git a/internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go b/internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go index 0a1f0d1374..1a62b68a19 100644 --- a/internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go +++ b/internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_discoverer @@ -26,6 +26,10 @@ func (_m *MockDiscoverer) EXPECT() *MockDiscoverer_Expecter { func (_m *MockDiscoverer) Discover(ctx context.Context, cb func(discoverer.VersionedState) error) error { ret := _m.Called(ctx, cb) + if len(ret) == 0 { + panic("no return value specified for Discover") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, func(discoverer.VersionedState) error) error); ok { r0 = rf(ctx, cb) @@ -69,6 +73,10 @@ func (_c *MockDiscoverer_Discover_Call) RunAndReturn(run func(context.Context, f func (_m *MockDiscoverer) NewVersionedState() discoverer.VersionedState { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for NewVersionedState") + } + var r0 discoverer.VersionedState if rf, ok := ret.Get(0).(func() discoverer.VersionedState); ok { r0 = rf() diff --git a/internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go b/internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go index 2c8d838751..2c66a694c5 100644 --- a/internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go +++ b/internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_lazygrpc @@ -59,6 +59,10 @@ func (_c *MockService_Close_Call[T]) RunAndReturn(run func()) *MockService_Close func (_m *MockService[T]) GetConn(ctx context.Context) (*grpc.ClientConn, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for GetConn") + } + var r0 *grpc.ClientConn var r1 error if rf, ok := ret.Get(0).(func(context.Context) (*grpc.ClientConn, error)); ok { @@ -113,6 +117,10 @@ func (_c *MockService_GetConn_Call[T]) RunAndReturn(run func(context.Context) (* func (_m *MockService[T]) GetService(ctx context.Context) (T, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for GetService") + } + var r0 T var r1 error if rf, ok := ret.Get(0).(func(context.Context) (T, error)); ok { diff --git a/internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go b/internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go index 774a548cde..8dd8a5d03a 100644 --- a/internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go +++ b/internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_resolver @@ -26,6 +26,10 @@ func (_m *MockBuilder) EXPECT() *MockBuilder_Expecter { func (_m *MockBuilder) Build(target resolver.Target, cc resolver.ClientConn, opts resolver.BuildOptions) (resolver.Resolver, error) { ret := _m.Called(target, cc, opts) + if len(ret) == 0 { + panic("no return value specified for Build") + } + var r0 resolver.Resolver var r1 error if rf, ok := ret.Get(0).(func(resolver.Target, resolver.ClientConn, resolver.BuildOptions) (resolver.Resolver, error)); ok { @@ -114,6 +118,10 @@ func (_c *MockBuilder_Close_Call) RunAndReturn(run func()) *MockBuilder_Close_Ca func (_m *MockBuilder) Resolver() serviceresolver.Resolver { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Resolver") + } + var r0 serviceresolver.Resolver if rf, ok := ret.Get(0).(func() serviceresolver.Resolver); ok { r0 = rf() @@ -157,6 +165,10 @@ func (_c *MockBuilder_Resolver_Call) RunAndReturn(run func() serviceresolver.Res func (_m *MockBuilder) Scheme() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Scheme") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go b/internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go index ff666a739b..e974b9acb8 100644 --- a/internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go +++ b/internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_resolver @@ -26,6 +26,10 @@ func (_m *MockResolver) EXPECT() *MockResolver_Expecter { func (_m *MockResolver) GetLatestState() discoverer.VersionedState { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetLatestState") + } + var r0 discoverer.VersionedState if rf, ok := ret.Get(0).(func() discoverer.VersionedState); ok { r0 = rf() @@ -67,6 +71,10 @@ func (_c *MockResolver_GetLatestState_Call) RunAndReturn(run func() discoverer.V func (_m *MockResolver) Watch(ctx context.Context, cb func(discoverer.VersionedState) error) error { ret := _m.Called(ctx, cb) + if len(ret) == 0 { + panic("no return value specified for Watch") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, func(discoverer.VersionedState) error) error); ok { r0 = rf(ctx, cb) diff --git a/internal/proxy/mock_cache.go b/internal/proxy/mock_cache.go index f1ef527f8e..cb1697b7f2 100644 --- a/internal/proxy/mock_cache.go +++ b/internal/proxy/mock_cache.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package proxy @@ -28,6 +28,10 @@ func (_m *MockCache) EXPECT() *MockCache_Expecter { func (_m *MockCache) AllocID(ctx context.Context) (int64, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for AllocID") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context) (int64, error)); ok { @@ -114,6 +118,10 @@ func (_c *MockCache_DeprecateShardCache_Call) RunAndReturn(run func(string, stri func (_m *MockCache) GetCollectionID(ctx context.Context, database string, collectionName string) (int64, error) { ret := _m.Called(ctx, database, collectionName) + if len(ret) == 0 { + panic("no return value specified for GetCollectionID") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string) (int64, error)); ok { @@ -168,6 +176,10 @@ func (_c *MockCache_GetCollectionID_Call) RunAndReturn(run func(context.Context, func (_m *MockCache) GetCollectionInfo(ctx context.Context, database string, collectionName string, collectionID int64) (*collectionBasicInfo, error) { ret := _m.Called(ctx, database, collectionName, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetCollectionInfo") + } + var r0 *collectionBasicInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string, int64) (*collectionBasicInfo, error)); ok { @@ -225,6 +237,10 @@ func (_c *MockCache_GetCollectionInfo_Call) RunAndReturn(run func(context.Contex func (_m *MockCache) GetCollectionName(ctx context.Context, database string, collectionID int64) (string, error) { ret := _m.Called(ctx, database, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetCollectionName") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, int64) (string, error)); ok { @@ -279,6 +295,10 @@ func (_c *MockCache_GetCollectionName_Call) RunAndReturn(run func(context.Contex func (_m *MockCache) GetCollectionSchema(ctx context.Context, database string, collectionName string) (*schemaInfo, error) { ret := _m.Called(ctx, database, collectionName) + if len(ret) == 0 { + panic("no return value specified for GetCollectionSchema") + } + var r0 *schemaInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string) (*schemaInfo, error)); ok { @@ -335,6 +355,10 @@ func (_c *MockCache_GetCollectionSchema_Call) RunAndReturn(run func(context.Cont func (_m *MockCache) GetCredentialInfo(ctx context.Context, username string) (*internalpb.CredentialInfo, error) { ret := _m.Called(ctx, username) + if len(ret) == 0 { + panic("no return value specified for GetCredentialInfo") + } + var r0 *internalpb.CredentialInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, string) (*internalpb.CredentialInfo, error)); ok { @@ -390,6 +414,10 @@ func (_c *MockCache_GetCredentialInfo_Call) RunAndReturn(run func(context.Contex func (_m *MockCache) GetDatabaseInfo(ctx context.Context, database string) (*databaseInfo, error) { ret := _m.Called(ctx, database) + if len(ret) == 0 { + panic("no return value specified for GetDatabaseInfo") + } + var r0 *databaseInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, string) (*databaseInfo, error)); ok { @@ -445,6 +473,10 @@ func (_c *MockCache_GetDatabaseInfo_Call) RunAndReturn(run func(context.Context, func (_m *MockCache) GetPartitionID(ctx context.Context, database string, collectionName string, partitionName string) (int64, error) { ret := _m.Called(ctx, database, collectionName, partitionName) + if len(ret) == 0 { + panic("no return value specified for GetPartitionID") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string, string) (int64, error)); ok { @@ -500,6 +532,10 @@ func (_c *MockCache_GetPartitionID_Call) RunAndReturn(run func(context.Context, func (_m *MockCache) GetPartitionInfo(ctx context.Context, database string, collectionName string, partitionName string) (*partitionInfo, error) { ret := _m.Called(ctx, database, collectionName, partitionName) + if len(ret) == 0 { + panic("no return value specified for GetPartitionInfo") + } + var r0 *partitionInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string, string) (*partitionInfo, error)); ok { @@ -557,6 +593,10 @@ func (_c *MockCache_GetPartitionInfo_Call) RunAndReturn(run func(context.Context func (_m *MockCache) GetPartitions(ctx context.Context, database string, collectionName string) (map[string]int64, error) { ret := _m.Called(ctx, database, collectionName) + if len(ret) == 0 { + panic("no return value specified for GetPartitions") + } + var r0 map[string]int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string) (map[string]int64, error)); ok { @@ -613,6 +653,10 @@ func (_c *MockCache_GetPartitions_Call) RunAndReturn(run func(context.Context, s func (_m *MockCache) GetPartitionsIndex(ctx context.Context, database string, collectionName string) ([]string, error) { ret := _m.Called(ctx, database, collectionName) + if len(ret) == 0 { + panic("no return value specified for GetPartitionsIndex") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string) ([]string, error)); ok { @@ -669,6 +713,10 @@ func (_c *MockCache_GetPartitionsIndex_Call) RunAndReturn(run func(context.Conte func (_m *MockCache) GetPrivilegeInfo(ctx context.Context) []string { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for GetPrivilegeInfo") + } + var r0 []string if rf, ok := ret.Get(0).(func(context.Context) []string); ok { r0 = rf(ctx) @@ -713,6 +761,10 @@ func (_c *MockCache_GetPrivilegeInfo_Call) RunAndReturn(run func(context.Context func (_m *MockCache) GetShards(ctx context.Context, withCache bool, database string, collectionName string, collectionID int64) (map[string][]nodeInfo, error) { ret := _m.Called(ctx, withCache, database, collectionName, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetShards") + } + var r0 map[string][]nodeInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, bool, string, string, int64) (map[string][]nodeInfo, error)); ok { @@ -771,6 +823,10 @@ func (_c *MockCache_GetShards_Call) RunAndReturn(run func(context.Context, bool, func (_m *MockCache) GetUserRole(username string) []string { ret := _m.Called(username) + if len(ret) == 0 { + panic("no return value specified for GetUserRole") + } + var r0 []string if rf, ok := ret.Get(0).(func(string) []string); ok { r0 = rf(username) @@ -815,6 +871,10 @@ func (_c *MockCache_GetUserRole_Call) RunAndReturn(run func(string) []string) *M func (_m *MockCache) HasDatabase(ctx context.Context, database string) bool { ret := _m.Called(ctx, database) + if len(ret) == 0 { + panic("no return value specified for HasDatabase") + } + var r0 bool if rf, ok := ret.Get(0).(func(context.Context, string) bool); ok { r0 = rf(ctx, database) @@ -925,6 +985,10 @@ func (_c *MockCache_InvalidateShardLeaderCache_Call) RunAndReturn(run func([]int func (_m *MockCache) RefreshPolicyInfo(op typeutil.CacheOp) error { ret := _m.Called(op) + if len(ret) == 0 { + panic("no return value specified for RefreshPolicyInfo") + } + var r0 error if rf, ok := ret.Get(0).(func(typeutil.CacheOp) error); ok { r0 = rf(op) @@ -1002,6 +1066,10 @@ func (_c *MockCache_RemoveCollection_Call) RunAndReturn(run func(context.Context func (_m *MockCache) RemoveCollectionsByID(ctx context.Context, collectionID int64) []string { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for RemoveCollectionsByID") + } + var r0 []string if rf, ok := ret.Get(0).(func(context.Context, int64) []string); ok { r0 = rf(ctx, collectionID) diff --git a/internal/proxy/mock_channels_manager.go b/internal/proxy/mock_channels_manager.go index 79b8c6015d..7342014623 100644 --- a/internal/proxy/mock_channels_manager.go +++ b/internal/proxy/mock_channels_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package proxy @@ -24,6 +24,10 @@ func (_m *MockChannelsMgr) EXPECT() *MockChannelsMgr_Expecter { func (_m *MockChannelsMgr) getChannels(collectionID int64) ([]string, error) { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for getChannels") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func(int64) ([]string, error)); ok { @@ -78,6 +82,10 @@ func (_c *MockChannelsMgr_getChannels_Call) RunAndReturn(run func(int64) ([]stri func (_m *MockChannelsMgr) getOrCreateDmlStream(collectionID int64) (msgstream.MsgStream, error) { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for getOrCreateDmlStream") + } + var r0 msgstream.MsgStream var r1 error if rf, ok := ret.Get(0).(func(int64) (msgstream.MsgStream, error)); ok { @@ -132,6 +140,10 @@ func (_c *MockChannelsMgr_getOrCreateDmlStream_Call) RunAndReturn(run func(int64 func (_m *MockChannelsMgr) getVChannels(collectionID int64) ([]string, error) { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for getVChannels") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func(int64) ([]string, error)); ok { diff --git a/internal/proxy/mock_lb_balancer.go b/internal/proxy/mock_lb_balancer.go index 368d6bbbd2..8fdbf9e5bb 100644 --- a/internal/proxy/mock_lb_balancer.go +++ b/internal/proxy/mock_lb_balancer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package proxy @@ -92,6 +92,10 @@ func (_c *MockLBBalancer_Close_Call) RunAndReturn(run func()) *MockLBBalancer_Cl func (_m *MockLBBalancer) SelectNode(ctx context.Context, availableNodes []int64, nq int64) (int64, error) { ret := _m.Called(ctx, availableNodes, nq) + if len(ret) == 0 { + panic("no return value specified for SelectNode") + } + var r0 int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, []int64, int64) (int64, error)); ok { diff --git a/internal/proxy/mock_lb_policy.go b/internal/proxy/mock_lb_policy.go index 0ba62a53c0..44c30bdbef 100644 --- a/internal/proxy/mock_lb_policy.go +++ b/internal/proxy/mock_lb_policy.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package proxy @@ -58,6 +58,10 @@ func (_c *MockLBPolicy_Close_Call) RunAndReturn(run func()) *MockLBPolicy_Close_ func (_m *MockLBPolicy) Execute(ctx context.Context, workload CollectionWorkLoad) error { ret := _m.Called(ctx, workload) + if len(ret) == 0 { + panic("no return value specified for Execute") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, CollectionWorkLoad) error); ok { r0 = rf(ctx, workload) @@ -101,6 +105,10 @@ func (_c *MockLBPolicy_Execute_Call) RunAndReturn(run func(context.Context, Coll func (_m *MockLBPolicy) ExecuteWithRetry(ctx context.Context, workload ChannelWorkload) error { ret := _m.Called(ctx, workload) + if len(ret) == 0 { + panic("no return value specified for ExecuteWithRetry") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, ChannelWorkload) error); ok { r0 = rf(ctx, workload) diff --git a/internal/proxy/mock_shardclient_manager.go b/internal/proxy/mock_shardclient_manager.go index 33d886a18d..edb6205f46 100644 --- a/internal/proxy/mock_shardclient_manager.go +++ b/internal/proxy/mock_shardclient_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package proxy @@ -58,6 +58,10 @@ func (_c *MockShardClientManager_Close_Call) RunAndReturn(run func()) *MockShard func (_m *MockShardClientManager) GetClient(ctx context.Context, nodeID int64) (types.QueryNodeClient, error) { ret := _m.Called(ctx, nodeID) + if len(ret) == 0 { + panic("no return value specified for GetClient") + } + var r0 types.QueryNodeClient var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (types.QueryNodeClient, error)); ok { @@ -146,6 +150,10 @@ func (_c *MockShardClientManager_SetClientCreatorFunc_Call) RunAndReturn(run fun func (_m *MockShardClientManager) UpdateShardLeaders(oldLeaders map[string][]nodeInfo, newLeaders map[string][]nodeInfo) error { ret := _m.Called(oldLeaders, newLeaders) + if len(ret) == 0 { + panic("no return value specified for UpdateShardLeaders") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string][]nodeInfo, map[string][]nodeInfo) error); ok { r0 = rf(oldLeaders, newLeaders) diff --git a/internal/proxy/mock_tso_test.go b/internal/proxy/mock_tso_test.go index 10b0b9655c..131e9ab688 100644 --- a/internal/proxy/mock_tso_test.go +++ b/internal/proxy/mock_tso_test.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package proxy @@ -36,6 +36,10 @@ func (_m *mockTimestampAllocator) AllocTimestamp(ctx context.Context, req *rootc _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AllocTimestamp") + } + var r0 *rootcoordpb.AllocTimestampResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *rootcoordpb.AllocTimestampRequest, ...grpc.CallOption) (*rootcoordpb.AllocTimestampResponse, error)); ok { diff --git a/internal/querycoordv2/balance/mock_balancer.go b/internal/querycoordv2/balance/mock_balancer.go index f1f2250e30..003451a9ca 100644 --- a/internal/querycoordv2/balance/mock_balancer.go +++ b/internal/querycoordv2/balance/mock_balancer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package balance @@ -24,6 +24,10 @@ func (_m *MockBalancer) EXPECT() *MockBalancer_Expecter { func (_m *MockBalancer) AssignChannel(channels []*meta.DmChannel, nodes []int64, manualBalance bool) []ChannelAssignPlan { ret := _m.Called(channels, nodes, manualBalance) + if len(ret) == 0 { + panic("no return value specified for AssignChannel") + } + var r0 []ChannelAssignPlan if rf, ok := ret.Get(0).(func([]*meta.DmChannel, []int64, bool) []ChannelAssignPlan); ok { r0 = rf(channels, nodes, manualBalance) @@ -70,6 +74,10 @@ func (_c *MockBalancer_AssignChannel_Call) RunAndReturn(run func([]*meta.DmChann func (_m *MockBalancer) AssignSegment(collectionID int64, segments []*meta.Segment, nodes []int64, manualBalance bool) []SegmentAssignPlan { ret := _m.Called(collectionID, segments, nodes, manualBalance) + if len(ret) == 0 { + panic("no return value specified for AssignSegment") + } + var r0 []SegmentAssignPlan if rf, ok := ret.Get(0).(func(int64, []*meta.Segment, []int64, bool) []SegmentAssignPlan); ok { r0 = rf(collectionID, segments, nodes, manualBalance) @@ -117,6 +125,10 @@ func (_c *MockBalancer_AssignSegment_Call) RunAndReturn(run func(int64, []*meta. func (_m *MockBalancer) BalanceReplica(replica *meta.Replica) ([]SegmentAssignPlan, []ChannelAssignPlan) { ret := _m.Called(replica) + if len(ret) == 0 { + panic("no return value specified for BalanceReplica") + } + var r0 []SegmentAssignPlan var r1 []ChannelAssignPlan if rf, ok := ret.Get(0).(func(*meta.Replica) ([]SegmentAssignPlan, []ChannelAssignPlan)); ok { diff --git a/internal/querycoordv2/dist/mock_controller.go b/internal/querycoordv2/dist/mock_controller.go index 420fe2c871..5a33c07cc9 100644 --- a/internal/querycoordv2/dist/mock_controller.go +++ b/internal/querycoordv2/dist/mock_controller.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package dist diff --git a/internal/querycoordv2/meta/mock_broker.go b/internal/querycoordv2/meta/mock_broker.go index 88389fced0..40618e2ace 100644 --- a/internal/querycoordv2/meta/mock_broker.go +++ b/internal/querycoordv2/meta/mock_broker.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package meta @@ -34,6 +34,10 @@ func (_m *MockBroker) EXPECT() *MockBroker_Expecter { func (_m *MockBroker) DescribeCollection(ctx context.Context, collectionID int64) (*milvuspb.DescribeCollectionResponse, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for DescribeCollection") + } + var r0 *milvuspb.DescribeCollectionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (*milvuspb.DescribeCollectionResponse, error)); ok { @@ -89,6 +93,10 @@ func (_c *MockBroker_DescribeCollection_Call) RunAndReturn(run func(context.Cont func (_m *MockBroker) DescribeDatabase(ctx context.Context, dbName string) (*rootcoordpb.DescribeDatabaseResponse, error) { ret := _m.Called(ctx, dbName) + if len(ret) == 0 { + panic("no return value specified for DescribeDatabase") + } + var r0 *rootcoordpb.DescribeDatabaseResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, string) (*rootcoordpb.DescribeDatabaseResponse, error)); ok { @@ -144,6 +152,10 @@ func (_c *MockBroker_DescribeDatabase_Call) RunAndReturn(run func(context.Contex func (_m *MockBroker) GetCollectionLoadInfo(ctx context.Context, collectionID int64) ([]string, int64, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetCollectionLoadInfo") + } + var r0 []string var r1 int64 var r2 error @@ -206,6 +218,10 @@ func (_c *MockBroker_GetCollectionLoadInfo_Call) RunAndReturn(run func(context.C func (_m *MockBroker) GetIndexInfo(ctx context.Context, collectionID int64, segmentID int64) ([]*querypb.FieldIndexInfo, error) { ret := _m.Called(ctx, collectionID, segmentID) + if len(ret) == 0 { + panic("no return value specified for GetIndexInfo") + } + var r0 []*querypb.FieldIndexInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, int64) ([]*querypb.FieldIndexInfo, error)); ok { @@ -262,6 +278,10 @@ func (_c *MockBroker_GetIndexInfo_Call) RunAndReturn(run func(context.Context, i func (_m *MockBroker) GetPartitions(ctx context.Context, collectionID int64) ([]int64, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetPartitions") + } + var r0 []int64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) ([]int64, error)); ok { @@ -317,6 +337,10 @@ func (_c *MockBroker_GetPartitions_Call) RunAndReturn(run func(context.Context, func (_m *MockBroker) GetRecoveryInfo(ctx context.Context, collectionID int64, partitionID int64) ([]*datapb.VchannelInfo, []*datapb.SegmentBinlogs, error) { ret := _m.Called(ctx, collectionID, partitionID) + if len(ret) == 0 { + panic("no return value specified for GetRecoveryInfo") + } + var r0 []*datapb.VchannelInfo var r1 []*datapb.SegmentBinlogs var r2 error @@ -389,6 +413,10 @@ func (_m *MockBroker) GetRecoveryInfoV2(ctx context.Context, collectionID int64, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetRecoveryInfoV2") + } + var r0 []*datapb.VchannelInfo var r1 []*datapb.SegmentInfo var r2 error @@ -468,6 +496,10 @@ func (_m *MockBroker) GetSegmentInfo(ctx context.Context, segmentID ...int64) ([ _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 []*datapb.SegmentInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, ...int64) ([]*datapb.SegmentInfo, error)); ok { @@ -530,6 +562,10 @@ func (_c *MockBroker_GetSegmentInfo_Call) RunAndReturn(run func(context.Context, func (_m *MockBroker) ListIndexes(ctx context.Context, collectionID int64) ([]*indexpb.IndexInfo, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for ListIndexes") + } + var r0 []*indexpb.IndexInfo var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) ([]*indexpb.IndexInfo, error)); ok { diff --git a/internal/querycoordv2/meta/mock_target_manager.go b/internal/querycoordv2/meta/mock_target_manager.go index 508e7cf3eb..95c8a1175d 100644 --- a/internal/querycoordv2/meta/mock_target_manager.go +++ b/internal/querycoordv2/meta/mock_target_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package meta @@ -28,6 +28,10 @@ func (_m *MockTargetManager) EXPECT() *MockTargetManager_Expecter { func (_m *MockTargetManager) CanSegmentBeMoved(collectionID int64, segmentID int64) bool { ret := _m.Called(collectionID, segmentID) + if len(ret) == 0 { + panic("no return value specified for CanSegmentBeMoved") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, int64) bool); ok { r0 = rf(collectionID, segmentID) @@ -71,6 +75,10 @@ func (_c *MockTargetManager_CanSegmentBeMoved_Call) RunAndReturn(run func(int64, func (_m *MockTargetManager) GetCollectionTargetVersion(collectionID int64, scope int32) int64 { ret := _m.Called(collectionID, scope) + if len(ret) == 0 { + panic("no return value specified for GetCollectionTargetVersion") + } + var r0 int64 if rf, ok := ret.Get(0).(func(int64, int32) int64); ok { r0 = rf(collectionID, scope) @@ -114,6 +122,10 @@ func (_c *MockTargetManager_GetCollectionTargetVersion_Call) RunAndReturn(run fu func (_m *MockTargetManager) GetDmChannel(collectionID int64, channel string, scope int32) *DmChannel { ret := _m.Called(collectionID, channel, scope) + if len(ret) == 0 { + panic("no return value specified for GetDmChannel") + } + var r0 *DmChannel if rf, ok := ret.Get(0).(func(int64, string, int32) *DmChannel); ok { r0 = rf(collectionID, channel, scope) @@ -160,6 +172,10 @@ func (_c *MockTargetManager_GetDmChannel_Call) RunAndReturn(run func(int64, stri func (_m *MockTargetManager) GetDmChannelsByCollection(collectionID int64, scope int32) map[string]*DmChannel { ret := _m.Called(collectionID, scope) + if len(ret) == 0 { + panic("no return value specified for GetDmChannelsByCollection") + } + var r0 map[string]*DmChannel if rf, ok := ret.Get(0).(func(int64, int32) map[string]*DmChannel); ok { r0 = rf(collectionID, scope) @@ -205,6 +221,10 @@ func (_c *MockTargetManager_GetDmChannelsByCollection_Call) RunAndReturn(run fun func (_m *MockTargetManager) GetDroppedSegmentsByChannel(collectionID int64, channelName string, scope int32) []int64 { ret := _m.Called(collectionID, channelName, scope) + if len(ret) == 0 { + panic("no return value specified for GetDroppedSegmentsByChannel") + } + var r0 []int64 if rf, ok := ret.Get(0).(func(int64, string, int32) []int64); ok { r0 = rf(collectionID, channelName, scope) @@ -251,6 +271,10 @@ func (_c *MockTargetManager_GetDroppedSegmentsByChannel_Call) RunAndReturn(run f func (_m *MockTargetManager) GetGrowingSegmentsByChannel(collectionID int64, channelName string, scope int32) typeutil.Set[int64] { ret := _m.Called(collectionID, channelName, scope) + if len(ret) == 0 { + panic("no return value specified for GetGrowingSegmentsByChannel") + } + var r0 typeutil.Set[int64] if rf, ok := ret.Get(0).(func(int64, string, int32) typeutil.Set[int64]); ok { r0 = rf(collectionID, channelName, scope) @@ -297,6 +321,10 @@ func (_c *MockTargetManager_GetGrowingSegmentsByChannel_Call) RunAndReturn(run f func (_m *MockTargetManager) GetGrowingSegmentsByCollection(collectionID int64, scope int32) typeutil.Set[int64] { ret := _m.Called(collectionID, scope) + if len(ret) == 0 { + panic("no return value specified for GetGrowingSegmentsByCollection") + } + var r0 typeutil.Set[int64] if rf, ok := ret.Get(0).(func(int64, int32) typeutil.Set[int64]); ok { r0 = rf(collectionID, scope) @@ -342,6 +370,10 @@ func (_c *MockTargetManager_GetGrowingSegmentsByCollection_Call) RunAndReturn(ru func (_m *MockTargetManager) GetSealedSegment(collectionID int64, id int64, scope int32) *datapb.SegmentInfo { ret := _m.Called(collectionID, id, scope) + if len(ret) == 0 { + panic("no return value specified for GetSealedSegment") + } + var r0 *datapb.SegmentInfo if rf, ok := ret.Get(0).(func(int64, int64, int32) *datapb.SegmentInfo); ok { r0 = rf(collectionID, id, scope) @@ -388,6 +420,10 @@ func (_c *MockTargetManager_GetSealedSegment_Call) RunAndReturn(run func(int64, func (_m *MockTargetManager) GetSealedSegmentsByChannel(collectionID int64, channelName string, scope int32) map[int64]*datapb.SegmentInfo { ret := _m.Called(collectionID, channelName, scope) + if len(ret) == 0 { + panic("no return value specified for GetSealedSegmentsByChannel") + } + var r0 map[int64]*datapb.SegmentInfo if rf, ok := ret.Get(0).(func(int64, string, int32) map[int64]*datapb.SegmentInfo); ok { r0 = rf(collectionID, channelName, scope) @@ -434,6 +470,10 @@ func (_c *MockTargetManager_GetSealedSegmentsByChannel_Call) RunAndReturn(run fu func (_m *MockTargetManager) GetSealedSegmentsByCollection(collectionID int64, scope int32) map[int64]*datapb.SegmentInfo { ret := _m.Called(collectionID, scope) + if len(ret) == 0 { + panic("no return value specified for GetSealedSegmentsByCollection") + } + var r0 map[int64]*datapb.SegmentInfo if rf, ok := ret.Get(0).(func(int64, int32) map[int64]*datapb.SegmentInfo); ok { r0 = rf(collectionID, scope) @@ -479,6 +519,10 @@ func (_c *MockTargetManager_GetSealedSegmentsByCollection_Call) RunAndReturn(run func (_m *MockTargetManager) GetSealedSegmentsByPartition(collectionID int64, partitionID int64, scope int32) map[int64]*datapb.SegmentInfo { ret := _m.Called(collectionID, partitionID, scope) + if len(ret) == 0 { + panic("no return value specified for GetSealedSegmentsByPartition") + } + var r0 map[int64]*datapb.SegmentInfo if rf, ok := ret.Get(0).(func(int64, int64, int32) map[int64]*datapb.SegmentInfo); ok { r0 = rf(collectionID, partitionID, scope) @@ -525,6 +569,10 @@ func (_c *MockTargetManager_GetSealedSegmentsByPartition_Call) RunAndReturn(run func (_m *MockTargetManager) IsCurrentTargetExist(collectionID int64, partitionID int64) bool { ret := _m.Called(collectionID, partitionID) + if len(ret) == 0 { + panic("no return value specified for IsCurrentTargetExist") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, int64) bool); ok { r0 = rf(collectionID, partitionID) @@ -568,6 +616,10 @@ func (_c *MockTargetManager_IsCurrentTargetExist_Call) RunAndReturn(run func(int func (_m *MockTargetManager) IsNextTargetExist(collectionID int64) bool { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for IsNextTargetExist") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64) bool); ok { r0 = rf(collectionID) @@ -617,6 +669,10 @@ func (_m *MockTargetManager) PullNextTargetV1(broker Broker, collectionID int64, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for PullNextTargetV1") + } + var r0 map[int64]*datapb.SegmentInfo var r1 map[string]*DmChannel var r2 error @@ -696,6 +752,10 @@ func (_m *MockTargetManager) PullNextTargetV2(broker Broker, collectionID int64, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for PullNextTargetV2") + } + var r0 map[int64]*datapb.SegmentInfo var r1 map[string]*DmChannel var r2 error @@ -768,6 +828,10 @@ func (_c *MockTargetManager_PullNextTargetV2_Call) RunAndReturn(run func(Broker, func (_m *MockTargetManager) Recover(catalog metastore.QueryCoordCatalog) error { ret := _m.Called(catalog) + if len(ret) == 0 { + panic("no return value specified for Recover") + } + var r0 error if rf, ok := ret.Get(0).(func(metastore.QueryCoordCatalog) error); ok { r0 = rf(catalog) @@ -924,6 +988,10 @@ func (_c *MockTargetManager_SaveCurrentTarget_Call) RunAndReturn(run func(metast func (_m *MockTargetManager) UpdateCollectionCurrentTarget(collectionID int64) bool { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for UpdateCollectionCurrentTarget") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64) bool); ok { r0 = rf(collectionID) @@ -966,6 +1034,10 @@ func (_c *MockTargetManager_UpdateCollectionCurrentTarget_Call) RunAndReturn(run func (_m *MockTargetManager) UpdateCollectionNextTarget(collectionID int64) error { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for UpdateCollectionNextTarget") + } + var r0 error if rf, ok := ret.Get(0).(func(int64) error); ok { r0 = rf(collectionID) diff --git a/internal/querycoordv2/mocks/mock_querynode.go b/internal/querycoordv2/mocks/mock_querynode.go index 039d03ee6b..4bf70286ac 100644 --- a/internal/querycoordv2/mocks/mock_querynode.go +++ b/internal/querycoordv2/mocks/mock_querynode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mocks @@ -33,6 +33,10 @@ func (_m *MockQueryNodeServer) EXPECT() *MockQueryNodeServer_Expecter { func (_m *MockQueryNodeServer) Delete(_a0 context.Context, _a1 *querypb.DeleteRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Delete") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest) (*commonpb.Status, error)); ok { @@ -88,6 +92,10 @@ func (_c *MockQueryNodeServer_Delete_Call) RunAndReturn(run func(context.Context func (_m *MockQueryNodeServer) GetComponentStates(_a0 context.Context, _a1 *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetComponentStatesRequest) (*milvuspb.ComponentStates, error)); ok { @@ -143,6 +151,10 @@ func (_c *MockQueryNodeServer_GetComponentStates_Call) RunAndReturn(run func(con func (_m *MockQueryNodeServer) GetDataDistribution(_a0 context.Context, _a1 *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetDataDistribution") + } + var r0 *querypb.GetDataDistributionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error)); ok { @@ -198,6 +210,10 @@ func (_c *MockQueryNodeServer_GetDataDistribution_Call) RunAndReturn(run func(co func (_m *MockQueryNodeServer) GetMetrics(_a0 context.Context, _a1 *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -253,6 +269,10 @@ func (_c *MockQueryNodeServer_GetMetrics_Call) RunAndReturn(run func(context.Con func (_m *MockQueryNodeServer) GetSegmentInfo(_a0 context.Context, _a1 *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 *querypb.GetSegmentInfoResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetSegmentInfoRequest) (*querypb.GetSegmentInfoResponse, error)); ok { @@ -308,6 +328,10 @@ func (_c *MockQueryNodeServer_GetSegmentInfo_Call) RunAndReturn(run func(context func (_m *MockQueryNodeServer) GetStatistics(_a0 context.Context, _a1 *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatistics") + } + var r0 *internalpb.GetStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error)); ok { @@ -363,6 +387,10 @@ func (_c *MockQueryNodeServer_GetStatistics_Call) RunAndReturn(run func(context. func (_m *MockQueryNodeServer) GetStatisticsChannel(_a0 context.Context, _a1 *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetStatisticsChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetStatisticsChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -418,6 +446,10 @@ func (_c *MockQueryNodeServer_GetStatisticsChannel_Call) RunAndReturn(run func(c func (_m *MockQueryNodeServer) GetTimeTickChannel(_a0 context.Context, _a1 *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for GetTimeTickChannel") + } + var r0 *milvuspb.StringResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetTimeTickChannelRequest) (*milvuspb.StringResponse, error)); ok { @@ -473,6 +505,10 @@ func (_c *MockQueryNodeServer_GetTimeTickChannel_Call) RunAndReturn(run func(con func (_m *MockQueryNodeServer) LoadPartitions(_a0 context.Context, _a1 *querypb.LoadPartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok { @@ -528,6 +564,10 @@ func (_c *MockQueryNodeServer_LoadPartitions_Call) RunAndReturn(run func(context func (_m *MockQueryNodeServer) LoadSegments(_a0 context.Context, _a1 *querypb.LoadSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) (*commonpb.Status, error)); ok { @@ -583,6 +623,10 @@ func (_c *MockQueryNodeServer_LoadSegments_Call) RunAndReturn(run func(context.C func (_m *MockQueryNodeServer) Query(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Query") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok { @@ -638,6 +682,10 @@ func (_c *MockQueryNodeServer_Query_Call) RunAndReturn(run func(context.Context, func (_m *MockQueryNodeServer) QuerySegments(_a0 context.Context, _a1 *querypb.QueryRequest) (*internalpb.RetrieveResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QuerySegments") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok { @@ -693,6 +741,10 @@ func (_c *MockQueryNodeServer_QuerySegments_Call) RunAndReturn(run func(context. func (_m *MockQueryNodeServer) QueryStream(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamServer) error { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryStream") + } + var r0 error if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamServer) error); ok { r0 = rf(_a0, _a1) @@ -736,6 +788,10 @@ func (_c *MockQueryNodeServer_QueryStream_Call) RunAndReturn(run func(*querypb.Q func (_m *MockQueryNodeServer) QueryStreamSegments(_a0 *querypb.QueryRequest, _a1 querypb.QueryNode_QueryStreamSegmentsServer) error { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for QueryStreamSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(*querypb.QueryRequest, querypb.QueryNode_QueryStreamSegmentsServer) error); ok { r0 = rf(_a0, _a1) @@ -779,6 +835,10 @@ func (_c *MockQueryNodeServer_QueryStreamSegments_Call) RunAndReturn(run func(*q func (_m *MockQueryNodeServer) ReleaseCollection(_a0 context.Context, _a1 *querypb.ReleaseCollectionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseCollection") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseCollectionRequest) (*commonpb.Status, error)); ok { @@ -834,6 +894,10 @@ func (_c *MockQueryNodeServer_ReleaseCollection_Call) RunAndReturn(run func(cont func (_m *MockQueryNodeServer) ReleasePartitions(_a0 context.Context, _a1 *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok { @@ -889,6 +953,10 @@ func (_c *MockQueryNodeServer_ReleasePartitions_Call) RunAndReturn(run func(cont func (_m *MockQueryNodeServer) ReleaseSegments(_a0 context.Context, _a1 *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error)); ok { @@ -944,6 +1012,10 @@ func (_c *MockQueryNodeServer_ReleaseSegments_Call) RunAndReturn(run func(contex func (_m *MockQueryNodeServer) Search(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for Search") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok { @@ -999,6 +1071,10 @@ func (_c *MockQueryNodeServer_Search_Call) RunAndReturn(run func(context.Context func (_m *MockQueryNodeServer) SearchSegments(_a0 context.Context, _a1 *querypb.SearchRequest) (*internalpb.SearchResults, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SearchSegments") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok { @@ -1054,6 +1130,10 @@ func (_c *MockQueryNodeServer_SearchSegments_Call) RunAndReturn(run func(context func (_m *MockQueryNodeServer) ShowConfigurations(_a0 context.Context, _a1 *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ShowConfigurations") + } + var r0 *internalpb.ShowConfigurationsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *internalpb.ShowConfigurationsRequest) (*internalpb.ShowConfigurationsResponse, error)); ok { @@ -1109,6 +1189,10 @@ func (_c *MockQueryNodeServer_ShowConfigurations_Call) RunAndReturn(run func(con func (_m *MockQueryNodeServer) SyncDistribution(_a0 context.Context, _a1 *querypb.SyncDistributionRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SyncDistribution") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncDistributionRequest) (*commonpb.Status, error)); ok { @@ -1164,6 +1248,10 @@ func (_c *MockQueryNodeServer_SyncDistribution_Call) RunAndReturn(run func(conte func (_m *MockQueryNodeServer) SyncReplicaSegments(_a0 context.Context, _a1 *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for SyncReplicaSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SyncReplicaSegmentsRequest) (*commonpb.Status, error)); ok { @@ -1219,6 +1307,10 @@ func (_c *MockQueryNodeServer_SyncReplicaSegments_Call) RunAndReturn(run func(co func (_m *MockQueryNodeServer) UnsubDmChannel(_a0 context.Context, _a1 *querypb.UnsubDmChannelRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for UnsubDmChannel") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.UnsubDmChannelRequest) (*commonpb.Status, error)); ok { @@ -1274,6 +1366,10 @@ func (_c *MockQueryNodeServer_UnsubDmChannel_Call) RunAndReturn(run func(context func (_m *MockQueryNodeServer) WatchDmChannels(_a0 context.Context, _a1 *querypb.WatchDmChannelsRequest) (*commonpb.Status, error) { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for WatchDmChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok { diff --git a/internal/querycoordv2/session/mock_cluster.go b/internal/querycoordv2/session/mock_cluster.go index dbc14c720c..bd7d3b3eae 100644 --- a/internal/querycoordv2/session/mock_cluster.go +++ b/internal/querycoordv2/session/mock_cluster.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package session @@ -31,6 +31,10 @@ func (_m *MockCluster) EXPECT() *MockCluster_Expecter { func (_m *MockCluster) GetComponentStates(ctx context.Context, nodeID int64) (*milvuspb.ComponentStates, error) { ret := _m.Called(ctx, nodeID) + if len(ret) == 0 { + panic("no return value specified for GetComponentStates") + } + var r0 *milvuspb.ComponentStates var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (*milvuspb.ComponentStates, error)); ok { @@ -86,6 +90,10 @@ func (_c *MockCluster_GetComponentStates_Call) RunAndReturn(run func(context.Con func (_m *MockCluster) GetDataDistribution(ctx context.Context, nodeID int64, req *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for GetDataDistribution") + } + var r0 *querypb.GetDataDistributionResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.GetDataDistributionRequest) (*querypb.GetDataDistributionResponse, error)); ok { @@ -142,6 +150,10 @@ func (_c *MockCluster_GetDataDistribution_Call) RunAndReturn(run func(context.Co func (_m *MockCluster) GetMetrics(ctx context.Context, nodeID int64, req *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for GetMetrics") + } + var r0 *milvuspb.GetMetricsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error)); ok { @@ -198,6 +210,10 @@ func (_c *MockCluster_GetMetrics_Call) RunAndReturn(run func(context.Context, in func (_m *MockCluster) LoadPartitions(ctx context.Context, nodeID int64, req *querypb.LoadPartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for LoadPartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.LoadPartitionsRequest) (*commonpb.Status, error)); ok { @@ -254,6 +270,10 @@ func (_c *MockCluster_LoadPartitions_Call) RunAndReturn(run func(context.Context func (_m *MockCluster) LoadSegments(ctx context.Context, nodeID int64, req *querypb.LoadSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for LoadSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.LoadSegmentsRequest) (*commonpb.Status, error)); ok { @@ -310,6 +330,10 @@ func (_c *MockCluster_LoadSegments_Call) RunAndReturn(run func(context.Context, func (_m *MockCluster) ReleasePartitions(ctx context.Context, nodeID int64, req *querypb.ReleasePartitionsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for ReleasePartitions") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.ReleasePartitionsRequest) (*commonpb.Status, error)); ok { @@ -366,6 +390,10 @@ func (_c *MockCluster_ReleasePartitions_Call) RunAndReturn(run func(context.Cont func (_m *MockCluster) ReleaseSegments(ctx context.Context, nodeID int64, req *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for ReleaseSegments") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.ReleaseSegmentsRequest) (*commonpb.Status, error)); ok { @@ -486,6 +514,10 @@ func (_c *MockCluster_Stop_Call) RunAndReturn(run func()) *MockCluster_Stop_Call func (_m *MockCluster) SyncDistribution(ctx context.Context, nodeID int64, req *querypb.SyncDistributionRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for SyncDistribution") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.SyncDistributionRequest) (*commonpb.Status, error)); ok { @@ -542,6 +574,10 @@ func (_c *MockCluster_SyncDistribution_Call) RunAndReturn(run func(context.Conte func (_m *MockCluster) UnsubDmChannel(ctx context.Context, nodeID int64, req *querypb.UnsubDmChannelRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for UnsubDmChannel") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.UnsubDmChannelRequest) (*commonpb.Status, error)); ok { @@ -598,6 +634,10 @@ func (_c *MockCluster_UnsubDmChannel_Call) RunAndReturn(run func(context.Context func (_m *MockCluster) WatchDmChannels(ctx context.Context, nodeID int64, req *querypb.WatchDmChannelsRequest) (*commonpb.Status, error) { ret := _m.Called(ctx, nodeID, req) + if len(ret) == 0 { + panic("no return value specified for WatchDmChannels") + } + var r0 *commonpb.Status var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, *querypb.WatchDmChannelsRequest) (*commonpb.Status, error)); ok { diff --git a/internal/querycoordv2/task/mock_scheduler.go b/internal/querycoordv2/task/mock_scheduler.go index f3eb7bd69e..5b5b5678f5 100644 --- a/internal/querycoordv2/task/mock_scheduler.go +++ b/internal/querycoordv2/task/mock_scheduler.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package task @@ -21,6 +21,10 @@ func (_m *MockScheduler) EXPECT() *MockScheduler_Expecter { func (_m *MockScheduler) Add(task Task) error { ret := _m.Called(task) + if len(ret) == 0 { + panic("no return value specified for Add") + } + var r0 error if rf, ok := ret.Get(0).(func(Task) error); ok { r0 = rf(task) @@ -129,6 +133,10 @@ func (_c *MockScheduler_Dispatch_Call) RunAndReturn(run func(int64)) *MockSchedu func (_m *MockScheduler) GetChannelTaskDelta(nodeID int64, collectionID int64) int { ret := _m.Called(nodeID, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetChannelTaskDelta") + } + var r0 int if rf, ok := ret.Get(0).(func(int64, int64) int); ok { r0 = rf(nodeID, collectionID) @@ -172,6 +180,10 @@ func (_c *MockScheduler_GetChannelTaskDelta_Call) RunAndReturn(run func(int64, i func (_m *MockScheduler) GetChannelTaskNum() int { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetChannelTaskNum") + } + var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() @@ -213,6 +225,10 @@ func (_c *MockScheduler_GetChannelTaskNum_Call) RunAndReturn(run func() int) *Mo func (_m *MockScheduler) GetExecutedFlag(nodeID int64) <-chan struct{} { ret := _m.Called(nodeID) + if len(ret) == 0 { + panic("no return value specified for GetExecutedFlag") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func(int64) <-chan struct{}); ok { r0 = rf(nodeID) @@ -257,6 +273,10 @@ func (_c *MockScheduler_GetExecutedFlag_Call) RunAndReturn(run func(int64) <-cha func (_m *MockScheduler) GetSegmentTaskDelta(nodeID int64, collectionID int64) int { ret := _m.Called(nodeID, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetSegmentTaskDelta") + } + var r0 int if rf, ok := ret.Get(0).(func(int64, int64) int); ok { r0 = rf(nodeID, collectionID) @@ -300,6 +320,10 @@ func (_c *MockScheduler_GetSegmentTaskDelta_Call) RunAndReturn(run func(int64, i func (_m *MockScheduler) GetSegmentTaskNum() int { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetSegmentTaskNum") + } + var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() diff --git a/internal/querynodev2/cluster/mock_manager.go b/internal/querynodev2/cluster/mock_manager.go index 3b032f1bce..06593f204c 100644 --- a/internal/querynodev2/cluster/mock_manager.go +++ b/internal/querynodev2/cluster/mock_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package cluster @@ -25,6 +25,10 @@ func (_m *MockManager) EXPECT() *MockManager_Expecter { func (_m *MockManager) GetWorker(ctx context.Context, nodeID int64) (Worker, error) { ret := _m.Called(ctx, nodeID) + if len(ret) == 0 { + panic("no return value specified for GetWorker") + } + var r0 Worker var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (Worker, error)); ok { diff --git a/internal/querynodev2/cluster/mock_worker.go b/internal/querynodev2/cluster/mock_worker.go index ba573bf89d..0eb5104b90 100644 --- a/internal/querynodev2/cluster/mock_worker.go +++ b/internal/querynodev2/cluster/mock_worker.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package cluster @@ -30,6 +30,10 @@ func (_m *MockWorker) EXPECT() *MockWorker_Expecter { func (_m *MockWorker) Delete(ctx context.Context, req *querypb.DeleteRequest) error { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for Delete") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.DeleteRequest) error); ok { r0 = rf(ctx, req) @@ -73,6 +77,10 @@ func (_c *MockWorker_Delete_Call) RunAndReturn(run func(context.Context, *queryp func (_m *MockWorker) GetStatistics(ctx context.Context, req *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for GetStatistics") + } + var r0 *internalpb.GetStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) (*internalpb.GetStatisticsResponse, error)); ok { @@ -128,6 +136,10 @@ func (_c *MockWorker_GetStatistics_Call) RunAndReturn(run func(context.Context, func (_m *MockWorker) IsHealthy() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for IsHealthy") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -169,6 +181,10 @@ func (_c *MockWorker_IsHealthy_Call) RunAndReturn(run func() bool) *MockWorker_I func (_m *MockWorker) LoadSegments(_a0 context.Context, _a1 *querypb.LoadSegmentsRequest) error { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for LoadSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) error); ok { r0 = rf(_a0, _a1) @@ -212,6 +228,10 @@ func (_c *MockWorker_LoadSegments_Call) RunAndReturn(run func(context.Context, * func (_m *MockWorker) QuerySegments(ctx context.Context, req *querypb.QueryRequest) (*internalpb.RetrieveResults, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for QuerySegments") + } + var r0 *internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) (*internalpb.RetrieveResults, error)); ok { @@ -267,6 +287,10 @@ func (_c *MockWorker_QuerySegments_Call) RunAndReturn(run func(context.Context, func (_m *MockWorker) QueryStreamSegments(ctx context.Context, req *querypb.QueryRequest, srv streamrpc.QueryStreamServer) error { ret := _m.Called(ctx, req, srv) + if len(ret) == 0 { + panic("no return value specified for QueryStreamSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, streamrpc.QueryStreamServer) error); ok { r0 = rf(ctx, req, srv) @@ -311,6 +335,10 @@ func (_c *MockWorker_QueryStreamSegments_Call) RunAndReturn(run func(context.Con func (_m *MockWorker) ReleaseSegments(_a0 context.Context, _a1 *querypb.ReleaseSegmentsRequest) error { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for ReleaseSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest) error); ok { r0 = rf(_a0, _a1) @@ -354,6 +382,10 @@ func (_c *MockWorker_ReleaseSegments_Call) RunAndReturn(run func(context.Context func (_m *MockWorker) SearchSegments(ctx context.Context, req *querypb.SearchRequest) (*internalpb.SearchResults, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for SearchSegments") + } + var r0 *internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) (*internalpb.SearchResults, error)); ok { diff --git a/internal/querynodev2/delegator/mock_delegator.go b/internal/querynodev2/delegator/mock_delegator.go index 4cc49950a9..3ef1a03ea8 100644 --- a/internal/querynodev2/delegator/mock_delegator.go +++ b/internal/querynodev2/delegator/mock_delegator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package delegator @@ -97,6 +97,10 @@ func (_c *MockShardDelegator_Close_Call) RunAndReturn(run func()) *MockShardDele func (_m *MockShardDelegator) Collection() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Collection") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -138,6 +142,10 @@ func (_c *MockShardDelegator_Collection_Call) RunAndReturn(run func() int64) *Mo func (_m *MockShardDelegator) GetDeleteBufferSize() (int64, int64) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetDeleteBufferSize") + } + var r0 int64 var r1 int64 if rf, ok := ret.Get(0).(func() (int64, int64)); ok { @@ -189,6 +197,10 @@ func (_c *MockShardDelegator_GetDeleteBufferSize_Call) RunAndReturn(run func() ( func (_m *MockShardDelegator) GetPartitionStatsVersions(ctx context.Context) map[int64]int64 { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for GetPartitionStatsVersions") + } + var r0 map[int64]int64 if rf, ok := ret.Get(0).(func(context.Context) map[int64]int64); ok { r0 = rf(ctx) @@ -233,6 +245,10 @@ func (_c *MockShardDelegator_GetPartitionStatsVersions_Call) RunAndReturn(run fu func (_m *MockShardDelegator) GetSegmentInfo(readable bool) ([]SnapshotItem, []SegmentEntry) { ret := _m.Called(readable) + if len(ret) == 0 { + panic("no return value specified for GetSegmentInfo") + } + var r0 []SnapshotItem var r1 []SegmentEntry if rf, ok := ret.Get(0).(func(bool) ([]SnapshotItem, []SegmentEntry)); ok { @@ -289,6 +305,10 @@ func (_c *MockShardDelegator_GetSegmentInfo_Call) RunAndReturn(run func(bool) ([ func (_m *MockShardDelegator) GetStatistics(ctx context.Context, req *querypb.GetStatisticsRequest) ([]*internalpb.GetStatisticsResponse, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for GetStatistics") + } + var r0 []*internalpb.GetStatisticsResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.GetStatisticsRequest) ([]*internalpb.GetStatisticsResponse, error)); ok { @@ -344,6 +364,10 @@ func (_c *MockShardDelegator_GetStatistics_Call) RunAndReturn(run func(context.C func (_m *MockShardDelegator) GetTargetVersion() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetTargetVersion") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -385,6 +409,10 @@ func (_c *MockShardDelegator_GetTargetVersion_Call) RunAndReturn(run func() int6 func (_m *MockShardDelegator) LoadGrowing(ctx context.Context, infos []*querypb.SegmentLoadInfo, version int64) error { ret := _m.Called(ctx, infos, version) + if len(ret) == 0 { + panic("no return value specified for LoadGrowing") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []*querypb.SegmentLoadInfo, int64) error); ok { r0 = rf(ctx, infos, version) @@ -429,6 +457,10 @@ func (_c *MockShardDelegator_LoadGrowing_Call) RunAndReturn(run func(context.Con func (_m *MockShardDelegator) LoadSegments(ctx context.Context, req *querypb.LoadSegmentsRequest) error { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for LoadSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.LoadSegmentsRequest) error); ok { r0 = rf(ctx, req) @@ -539,6 +571,10 @@ func (_c *MockShardDelegator_ProcessInsert_Call) RunAndReturn(run func(map[int64 func (_m *MockShardDelegator) Query(ctx context.Context, req *querypb.QueryRequest) ([]*internalpb.RetrieveResults, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for Query") + } + var r0 []*internalpb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest) ([]*internalpb.RetrieveResults, error)); ok { @@ -594,6 +630,10 @@ func (_c *MockShardDelegator_Query_Call) RunAndReturn(run func(context.Context, func (_m *MockShardDelegator) QueryStream(ctx context.Context, req *querypb.QueryRequest, srv streamrpc.QueryStreamServer) error { ret := _m.Called(ctx, req, srv) + if len(ret) == 0 { + panic("no return value specified for QueryStream") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.QueryRequest, streamrpc.QueryStreamServer) error); ok { r0 = rf(ctx, req, srv) @@ -638,6 +678,10 @@ func (_c *MockShardDelegator_QueryStream_Call) RunAndReturn(run func(context.Con func (_m *MockShardDelegator) ReleaseSegments(ctx context.Context, req *querypb.ReleaseSegmentsRequest, force bool) error { ret := _m.Called(ctx, req, force) + if len(ret) == 0 { + panic("no return value specified for ReleaseSegments") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.ReleaseSegmentsRequest, bool) error); ok { r0 = rf(ctx, req, force) @@ -682,6 +726,10 @@ func (_c *MockShardDelegator_ReleaseSegments_Call) RunAndReturn(run func(context func (_m *MockShardDelegator) Search(ctx context.Context, req *querypb.SearchRequest) ([]*internalpb.SearchResults, error) { ret := _m.Called(ctx, req) + if len(ret) == 0 { + panic("no return value specified for Search") + } + var r0 []*internalpb.SearchResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *querypb.SearchRequest) ([]*internalpb.SearchResults, error)); ok { @@ -737,6 +785,10 @@ func (_c *MockShardDelegator_Search_Call) RunAndReturn(run func(context.Context, func (_m *MockShardDelegator) Serviceable() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Serviceable") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -962,6 +1014,10 @@ func (_c *MockShardDelegator_TryCleanExcludedSegments_Call) RunAndReturn(run fun func (_m *MockShardDelegator) VerifyExcludedSegments(segmentID int64, ts uint64) bool { ret := _m.Called(segmentID, ts) + if len(ret) == 0 { + panic("no return value specified for VerifyExcludedSegments") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, uint64) bool); ok { r0 = rf(segmentID, ts) @@ -1005,6 +1061,10 @@ func (_c *MockShardDelegator_VerifyExcludedSegments_Call) RunAndReturn(run func( func (_m *MockShardDelegator) Version() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Version") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() diff --git a/internal/querynodev2/optimizers/mock_query_hook.go b/internal/querynodev2/optimizers/mock_query_hook.go index 7c9f5dab88..6b084d7098 100644 --- a/internal/querynodev2/optimizers/mock_query_hook.go +++ b/internal/querynodev2/optimizers/mock_query_hook.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package optimizers @@ -21,6 +21,10 @@ func (_m *MockQueryHook) EXPECT() *MockQueryHook_Expecter { func (_m *MockQueryHook) DeleteTuningConfig(_a0 string) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for DeleteTuningConfig") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(_a0) @@ -63,6 +67,10 @@ func (_c *MockQueryHook_DeleteTuningConfig_Call) RunAndReturn(run func(string) e func (_m *MockQueryHook) Init(_a0 string) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Init") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(_a0) @@ -105,6 +113,10 @@ func (_c *MockQueryHook_Init_Call) RunAndReturn(run func(string) error) *MockQue func (_m *MockQueryHook) InitTuningConfig(_a0 map[string]string) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for InitTuningConfig") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string) error); ok { r0 = rf(_a0) @@ -147,6 +159,10 @@ func (_c *MockQueryHook_InitTuningConfig_Call) RunAndReturn(run func(map[string] func (_m *MockQueryHook) Run(_a0 map[string]interface{}) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Run") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]interface{}) error); ok { r0 = rf(_a0) diff --git a/internal/querynodev2/segments/mock_collection_manager.go b/internal/querynodev2/segments/mock_collection_manager.go index 049b326c71..496134100f 100644 --- a/internal/querynodev2/segments/mock_collection_manager.go +++ b/internal/querynodev2/segments/mock_collection_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package segments @@ -27,6 +27,10 @@ func (_m *MockCollectionManager) EXPECT() *MockCollectionManager_Expecter { func (_m *MockCollectionManager) Get(collectionID int64) *Collection { ret := _m.Called(collectionID) + if len(ret) == 0 { + panic("no return value specified for Get") + } + var r0 *Collection if rf, ok := ret.Get(0).(func(int64) *Collection); ok { r0 = rf(collectionID) @@ -71,6 +75,10 @@ func (_c *MockCollectionManager_Get_Call) RunAndReturn(run func(int64) *Collecti func (_m *MockCollectionManager) List() []int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for List") + } + var r0 []int64 if rf, ok := ret.Get(0).(func() []int64); ok { r0 = rf() @@ -150,6 +158,10 @@ func (_c *MockCollectionManager_PutOrRef_Call) RunAndReturn(run func(int64, *sch func (_m *MockCollectionManager) Ref(collectionID int64, count uint32) bool { ret := _m.Called(collectionID, count) + if len(ret) == 0 { + panic("no return value specified for Ref") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, uint32) bool); ok { r0 = rf(collectionID, count) @@ -193,6 +205,10 @@ func (_c *MockCollectionManager_Ref_Call) RunAndReturn(run func(int64, uint32) b func (_m *MockCollectionManager) Unref(collectionID int64, count uint32) bool { ret := _m.Called(collectionID, count) + if len(ret) == 0 { + panic("no return value specified for Unref") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, uint32) bool); ok { r0 = rf(collectionID, count) diff --git a/internal/querynodev2/segments/mock_loader.go b/internal/querynodev2/segments/mock_loader.go index 7eb0a369db..d50d52b9c0 100644 --- a/internal/querynodev2/segments/mock_loader.go +++ b/internal/querynodev2/segments/mock_loader.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package segments @@ -44,6 +44,10 @@ func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Load") + } + var r0 []Segment var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) ([]Segment, error)); ok { @@ -116,6 +120,10 @@ func (_m *MockLoader) LoadBM25Stats(ctx context.Context, collectionID int64, inf _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadBM25Stats") + } + var r0 *typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats] var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, ...*querypb.SegmentLoadInfo) (*typeutil.ConcurrentMap[int64, map[int64]*storage.BM25Stats], error)); ok { @@ -186,6 +194,10 @@ func (_m *MockLoader) LoadBloomFilterSet(ctx context.Context, collectionID int64 _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for LoadBloomFilterSet") + } + var r0 []*pkoracle.BloomFilterSet var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, int64, ...*querypb.SegmentLoadInfo) ([]*pkoracle.BloomFilterSet, error)); ok { @@ -250,6 +262,10 @@ func (_c *MockLoader_LoadBloomFilterSet_Call) RunAndReturn(run func(context.Cont func (_m *MockLoader) LoadDeltaLogs(ctx context.Context, segment Segment, deltaLogs []*datapb.FieldBinlog) error { ret := _m.Called(ctx, segment, deltaLogs) + if len(ret) == 0 { + panic("no return value specified for LoadDeltaLogs") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, Segment, []*datapb.FieldBinlog) error); ok { r0 = rf(ctx, segment, deltaLogs) @@ -294,6 +310,10 @@ func (_c *MockLoader_LoadDeltaLogs_Call) RunAndReturn(run func(context.Context, func (_m *MockLoader) LoadIndex(ctx context.Context, segment Segment, info *querypb.SegmentLoadInfo, version int64) error { ret := _m.Called(ctx, segment, info, version) + if len(ret) == 0 { + panic("no return value specified for LoadIndex") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, Segment, *querypb.SegmentLoadInfo, int64) error); ok { r0 = rf(ctx, segment, info, version) @@ -339,6 +359,10 @@ func (_c *MockLoader_LoadIndex_Call) RunAndReturn(run func(context.Context, Segm func (_m *MockLoader) LoadLazySegment(ctx context.Context, segment Segment, loadInfo *querypb.SegmentLoadInfo) error { ret := _m.Called(ctx, segment, loadInfo) + if len(ret) == 0 { + panic("no return value specified for LoadLazySegment") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, Segment, *querypb.SegmentLoadInfo) error); ok { r0 = rf(ctx, segment, loadInfo) diff --git a/internal/querynodev2/segments/mock_segment.go b/internal/querynodev2/segments/mock_segment.go index 6253da14e7..c6e7bdd407 100644 --- a/internal/querynodev2/segments/mock_segment.go +++ b/internal/querynodev2/segments/mock_segment.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package segments @@ -39,6 +39,10 @@ func (_m *MockSegment) EXPECT() *MockSegment_Expecter { func (_m *MockSegment) BatchPkExist(lc *storage.BatchLocationsCache) []bool { ret := _m.Called(lc) + if len(ret) == 0 { + panic("no return value specified for BatchPkExist") + } + var r0 []bool if rf, ok := ret.Get(0).(func(*storage.BatchLocationsCache) []bool); ok { r0 = rf(lc) @@ -83,6 +87,10 @@ func (_c *MockSegment_BatchPkExist_Call) RunAndReturn(run func(*storage.BatchLoc func (_m *MockSegment) CASVersion(_a0 int64, _a1 int64) bool { ret := _m.Called(_a0, _a1) + if len(ret) == 0 { + panic("no return value specified for CASVersion") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, int64) bool); ok { r0 = rf(_a0, _a1) @@ -126,6 +134,10 @@ func (_c *MockSegment_CASVersion_Call) RunAndReturn(run func(int64, int64) bool) func (_m *MockSegment) Collection() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Collection") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -167,6 +179,10 @@ func (_c *MockSegment_Collection_Call) RunAndReturn(run func() int64) *MockSegme func (_m *MockSegment) DatabaseName() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for DatabaseName") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -208,6 +224,10 @@ func (_c *MockSegment_DatabaseName_Call) RunAndReturn(run func() string) *MockSe func (_m *MockSegment) Delete(ctx context.Context, primaryKeys []storage.PrimaryKey, timestamps []uint64) error { ret := _m.Called(ctx, primaryKeys, timestamps) + if len(ret) == 0 { + panic("no return value specified for Delete") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []storage.PrimaryKey, []uint64) error); ok { r0 = rf(ctx, primaryKeys, timestamps) @@ -252,6 +272,10 @@ func (_c *MockSegment_Delete_Call) RunAndReturn(run func(context.Context, []stor func (_m *MockSegment) ExistIndex(fieldID int64) bool { ret := _m.Called(fieldID) + if len(ret) == 0 { + panic("no return value specified for ExistIndex") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64) bool); ok { r0 = rf(fieldID) @@ -294,6 +318,10 @@ func (_c *MockSegment_ExistIndex_Call) RunAndReturn(run func(int64) bool) *MockS func (_m *MockSegment) GetBM25Stats() map[int64]*storage.BM25Stats { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetBM25Stats") + } + var r0 map[int64]*storage.BM25Stats if rf, ok := ret.Get(0).(func() map[int64]*storage.BM25Stats); ok { r0 = rf() @@ -337,6 +365,10 @@ func (_c *MockSegment_GetBM25Stats_Call) RunAndReturn(run func() map[int64]*stor func (_m *MockSegment) GetIndex(fieldID int64) *IndexedFieldInfo { ret := _m.Called(fieldID) + if len(ret) == 0 { + panic("no return value specified for GetIndex") + } + var r0 *IndexedFieldInfo if rf, ok := ret.Get(0).(func(int64) *IndexedFieldInfo); ok { r0 = rf(fieldID) @@ -381,6 +413,10 @@ func (_c *MockSegment_GetIndex_Call) RunAndReturn(run func(int64) *IndexedFieldI func (_m *MockSegment) HasRawData(fieldID int64) bool { ret := _m.Called(fieldID) + if len(ret) == 0 { + panic("no return value specified for HasRawData") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64) bool); ok { r0 = rf(fieldID) @@ -423,6 +459,10 @@ func (_c *MockSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *MockS func (_m *MockSegment) ID() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ID") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -464,6 +504,10 @@ func (_c *MockSegment_ID_Call) RunAndReturn(run func() int64) *MockSegment_ID_Ca func (_m *MockSegment) Indexes() []*IndexedFieldInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Indexes") + } + var r0 []*IndexedFieldInfo if rf, ok := ret.Get(0).(func() []*IndexedFieldInfo); ok { r0 = rf() @@ -507,6 +551,10 @@ func (_c *MockSegment_Indexes_Call) RunAndReturn(run func() []*IndexedFieldInfo) func (_m *MockSegment) Insert(ctx context.Context, rowIDs []int64, timestamps []uint64, record *segcorepb.InsertRecord) error { ret := _m.Called(ctx, rowIDs, timestamps, record) + if len(ret) == 0 { + panic("no return value specified for Insert") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []int64, []uint64, *segcorepb.InsertRecord) error); ok { r0 = rf(ctx, rowIDs, timestamps, record) @@ -552,6 +600,10 @@ func (_c *MockSegment_Insert_Call) RunAndReturn(run func(context.Context, []int6 func (_m *MockSegment) InsertCount() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for InsertCount") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -593,6 +645,10 @@ func (_c *MockSegment_InsertCount_Call) RunAndReturn(run func() int64) *MockSegm func (_m *MockSegment) IsLazyLoad() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for IsLazyLoad") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -634,6 +690,10 @@ func (_c *MockSegment_IsLazyLoad_Call) RunAndReturn(run func() bool) *MockSegmen func (_m *MockSegment) IsSorted() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for IsSorted") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -675,6 +735,10 @@ func (_c *MockSegment_IsSorted_Call) RunAndReturn(run func() bool) *MockSegment_ func (_m *MockSegment) LastDeltaTimestamp() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for LastDeltaTimestamp") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -716,6 +780,10 @@ func (_c *MockSegment_LastDeltaTimestamp_Call) RunAndReturn(run func() uint64) * func (_m *MockSegment) Level() datapb.SegmentLevel { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Level") + } + var r0 datapb.SegmentLevel if rf, ok := ret.Get(0).(func() datapb.SegmentLevel); ok { r0 = rf() @@ -757,6 +825,10 @@ func (_c *MockSegment_Level_Call) RunAndReturn(run func() datapb.SegmentLevel) * func (_m *MockSegment) LoadDeltaData(ctx context.Context, deltaData *storage.DeleteData) error { ret := _m.Called(ctx, deltaData) + if len(ret) == 0 { + panic("no return value specified for LoadDeltaData") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *storage.DeleteData) error); ok { r0 = rf(ctx, deltaData) @@ -800,6 +872,10 @@ func (_c *MockSegment_LoadDeltaData_Call) RunAndReturn(run func(context.Context, func (_m *MockSegment) LoadInfo() *querypb.SegmentLoadInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for LoadInfo") + } + var r0 *querypb.SegmentLoadInfo if rf, ok := ret.Get(0).(func() *querypb.SegmentLoadInfo); ok { r0 = rf() @@ -843,6 +919,10 @@ func (_c *MockSegment_LoadInfo_Call) RunAndReturn(run func() *querypb.SegmentLoa func (_m *MockSegment) MayPkExist(lc *storage.LocationsCache) bool { ret := _m.Called(lc) + if len(ret) == 0 { + panic("no return value specified for MayPkExist") + } + var r0 bool if rf, ok := ret.Get(0).(func(*storage.LocationsCache) bool); ok { r0 = rf(lc) @@ -885,6 +965,10 @@ func (_c *MockSegment_MayPkExist_Call) RunAndReturn(run func(*storage.LocationsC func (_m *MockSegment) MemSize() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for MemSize") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -926,6 +1010,10 @@ func (_c *MockSegment_MemSize_Call) RunAndReturn(run func() int64) *MockSegment_ func (_m *MockSegment) NeedUpdatedVersion() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for NeedUpdatedVersion") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -967,6 +1055,10 @@ func (_c *MockSegment_NeedUpdatedVersion_Call) RunAndReturn(run func() int64) *M func (_m *MockSegment) Partition() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Partition") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -1008,6 +1100,10 @@ func (_c *MockSegment_Partition_Call) RunAndReturn(run func() int64) *MockSegmen func (_m *MockSegment) PinIfNotReleased() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for PinIfNotReleased") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1097,6 +1193,10 @@ func (_c *MockSegment_Release_Call) RunAndReturn(run func(context.Context, ...re func (_m *MockSegment) RemoveUnusedFieldFiles() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for RemoveUnusedFieldFiles") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -1171,6 +1271,10 @@ func (_c *MockSegment_ResetIndexesLazyLoad_Call) RunAndReturn(run func(bool)) *M func (_m *MockSegment) ResourceGroup() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ResourceGroup") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -1212,6 +1316,10 @@ func (_c *MockSegment_ResourceGroup_Call) RunAndReturn(run func() string) *MockS func (_m *MockSegment) ResourceUsageEstimate() ResourceUsage { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ResourceUsageEstimate") + } + var r0 ResourceUsage if rf, ok := ret.Get(0).(func() ResourceUsage); ok { r0 = rf() @@ -1253,6 +1361,10 @@ func (_c *MockSegment_ResourceUsageEstimate_Call) RunAndReturn(run func() Resour func (_m *MockSegment) Retrieve(ctx context.Context, plan *RetrievePlan) (*segcorepb.RetrieveResults, error) { ret := _m.Called(ctx, plan) + if len(ret) == 0 { + panic("no return value specified for Retrieve") + } + var r0 *segcorepb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *RetrievePlan) (*segcorepb.RetrieveResults, error)); ok { @@ -1308,6 +1420,10 @@ func (_c *MockSegment_Retrieve_Call) RunAndReturn(run func(context.Context, *Ret func (_m *MockSegment) RetrieveByOffsets(ctx context.Context, plan *RetrievePlan, offsets []int64) (*segcorepb.RetrieveResults, error) { ret := _m.Called(ctx, plan, offsets) + if len(ret) == 0 { + panic("no return value specified for RetrieveByOffsets") + } + var r0 *segcorepb.RetrieveResults var r1 error if rf, ok := ret.Get(0).(func(context.Context, *RetrievePlan, []int64) (*segcorepb.RetrieveResults, error)); ok { @@ -1364,6 +1480,10 @@ func (_c *MockSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Cont func (_m *MockSegment) RowNum() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for RowNum") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() @@ -1405,6 +1525,10 @@ func (_c *MockSegment_RowNum_Call) RunAndReturn(run func() int64) *MockSegment_R func (_m *MockSegment) Search(ctx context.Context, searchReq *SearchRequest) (*SearchResult, error) { ret := _m.Called(ctx, searchReq) + if len(ret) == 0 { + panic("no return value specified for Search") + } + var r0 *SearchResult var r1 error if rf, ok := ret.Get(0).(func(context.Context, *SearchRequest) (*SearchResult, error)); ok { @@ -1460,6 +1584,10 @@ func (_c *MockSegment_Search_Call) RunAndReturn(run func(context.Context, *Searc func (_m *MockSegment) Shard() metautil.Channel { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Shard") + } + var r0 metautil.Channel if rf, ok := ret.Get(0).(func() metautil.Channel); ok { r0 = rf() @@ -1501,6 +1629,10 @@ func (_c *MockSegment_Shard_Call) RunAndReturn(run func() metautil.Channel) *Moc func (_m *MockSegment) StartPosition() *msgpb.MsgPosition { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for StartPosition") + } + var r0 *msgpb.MsgPosition if rf, ok := ret.Get(0).(func() *msgpb.MsgPosition); ok { r0 = rf() @@ -1544,6 +1676,10 @@ func (_c *MockSegment_StartPosition_Call) RunAndReturn(run func() *msgpb.MsgPosi func (_m *MockSegment) Type() commonpb.SegmentState { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Type") + } + var r0 commonpb.SegmentState if rf, ok := ret.Get(0).(func() commonpb.SegmentState); ok { r0 = rf() @@ -1683,6 +1819,10 @@ func (_c *MockSegment_UpdateBloomFilter_Call) RunAndReturn(run func([]storage.Pr func (_m *MockSegment) Version() int64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Version") + } + var r0 int64 if rf, ok := ret.Get(0).(func() int64); ok { r0 = rf() diff --git a/internal/querynodev2/segments/mock_segment_manager.go b/internal/querynodev2/segments/mock_segment_manager.go index 01d1cb6cca..a4598e665d 100644 --- a/internal/querynodev2/segments/mock_segment_manager.go +++ b/internal/querynodev2/segments/mock_segment_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package segments @@ -62,6 +62,10 @@ func (_c *MockSegmentManager_Clear_Call) RunAndReturn(run func(context.Context)) func (_m *MockSegmentManager) Empty() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Empty") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -103,6 +107,10 @@ func (_c *MockSegmentManager_Empty_Call) RunAndReturn(run func() bool) *MockSegm func (_m *MockSegmentManager) Exist(segmentID int64, typ commonpb.SegmentState) bool { ret := _m.Called(segmentID, typ) + if len(ret) == 0 { + panic("no return value specified for Exist") + } + var r0 bool if rf, ok := ret.Get(0).(func(int64, commonpb.SegmentState) bool); ok { r0 = rf(segmentID, typ) @@ -146,6 +154,10 @@ func (_c *MockSegmentManager_Exist_Call) RunAndReturn(run func(int64, commonpb.S func (_m *MockSegmentManager) Get(segmentID int64) Segment { ret := _m.Called(segmentID) + if len(ret) == 0 { + panic("no return value specified for Get") + } + var r0 Segment if rf, ok := ret.Get(0).(func(int64) Segment); ok { r0 = rf(segmentID) @@ -197,6 +209,10 @@ func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilt _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetAndPin") + } + var r0 []Segment var r1 error if rf, ok := ret.Get(0).(func([]int64, ...SegmentFilter) ([]Segment, error)); ok { @@ -265,6 +281,10 @@ func (_m *MockSegmentManager) GetAndPinBy(filters ...SegmentFilter) ([]Segment, _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetAndPinBy") + } + var r0 []Segment var r1 error if rf, ok := ret.Get(0).(func(...SegmentFilter) ([]Segment, error)); ok { @@ -332,6 +352,10 @@ func (_m *MockSegmentManager) GetBy(filters ...SegmentFilter) []Segment { _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for GetBy") + } + var r0 []Segment if rf, ok := ret.Get(0).(func(...SegmentFilter) []Segment); ok { r0 = rf(filters...) @@ -383,6 +407,10 @@ func (_c *MockSegmentManager_GetBy_Call) RunAndReturn(run func(...SegmentFilter) func (_m *MockSegmentManager) GetGrowing(segmentID int64) Segment { ret := _m.Called(segmentID) + if len(ret) == 0 { + panic("no return value specified for GetGrowing") + } + var r0 Segment if rf, ok := ret.Get(0).(func(int64) Segment); ok { r0 = rf(segmentID) @@ -427,6 +455,10 @@ func (_c *MockSegmentManager_GetGrowing_Call) RunAndReturn(run func(int64) Segme func (_m *MockSegmentManager) GetSealed(segmentID int64) Segment { ret := _m.Called(segmentID) + if len(ret) == 0 { + panic("no return value specified for GetSealed") + } + var r0 Segment if rf, ok := ret.Get(0).(func(int64) Segment); ok { r0 = rf(segmentID) @@ -471,6 +503,10 @@ func (_c *MockSegmentManager_GetSealed_Call) RunAndReturn(run func(int64) Segmen func (_m *MockSegmentManager) GetWithType(segmentID int64, typ commonpb.SegmentState) Segment { ret := _m.Called(segmentID, typ) + if len(ret) == 0 { + panic("no return value specified for GetWithType") + } + var r0 Segment if rf, ok := ret.Get(0).(func(int64, commonpb.SegmentState) Segment); ok { r0 = rf(segmentID, typ) @@ -565,6 +601,10 @@ func (_c *MockSegmentManager_Put_Call) RunAndReturn(run func(context.Context, co func (_m *MockSegmentManager) Remove(ctx context.Context, segmentID int64, scope querypb.DataScope) (int, int) { ret := _m.Called(ctx, segmentID, scope) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 int var r1 int if rf, ok := ret.Get(0).(func(context.Context, int64, querypb.DataScope) (int, int)); ok { @@ -626,6 +666,10 @@ func (_m *MockSegmentManager) RemoveBy(ctx context.Context, filters ...SegmentFi _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for RemoveBy") + } + var r0 int var r1 int if rf, ok := ret.Get(0).(func(context.Context, ...SegmentFilter) (int, int)); ok { @@ -726,6 +770,10 @@ func (_m *MockSegmentManager) UpdateBy(action SegmentAction, filters ...SegmentF _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for UpdateBy") + } + var r0 int if rf, ok := ret.Get(0).(func(SegmentAction, ...SegmentFilter) int); ok { r0 = rf(action, filters...) diff --git a/internal/rootcoord/mocks/garbage_collector.go b/internal/rootcoord/mocks/garbage_collector.go index 4378dd21eb..7c23345380 100644 --- a/internal/rootcoord/mocks/garbage_collector.go +++ b/internal/rootcoord/mocks/garbage_collector.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mockrootcoord @@ -26,6 +26,10 @@ func (_m *GarbageCollector) EXPECT() *GarbageCollector_Expecter { func (_m *GarbageCollector) GcCollectionData(ctx context.Context, coll *model.Collection) (uint64, error) { ret := _m.Called(ctx, coll) + if len(ret) == 0 { + panic("no return value specified for GcCollectionData") + } + var r0 uint64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) (uint64, error)); ok { @@ -79,6 +83,10 @@ func (_c *GarbageCollector_GcCollectionData_Call) RunAndReturn(run func(context. func (_m *GarbageCollector) GcPartitionData(ctx context.Context, pChannels []string, vchannels []string, partition *model.Partition) (uint64, error) { ret := _m.Called(ctx, pChannels, vchannels, partition) + if len(ret) == 0 { + panic("no return value specified for GcPartitionData") + } + var r0 uint64 var r1 error if rf, ok := ret.Get(0).(func(context.Context, []string, []string, *model.Partition) (uint64, error)); ok { diff --git a/internal/rootcoord/mocks/meta_table.go b/internal/rootcoord/mocks/meta_table.go index 69c7cc1743..243adbd726 100644 --- a/internal/rootcoord/mocks/meta_table.go +++ b/internal/rootcoord/mocks/meta_table.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mockrootcoord @@ -34,6 +34,10 @@ func (_m *IMetaTable) EXPECT() *IMetaTable_Expecter { func (_m *IMetaTable) AddCollection(ctx context.Context, coll *model.Collection) error { ret := _m.Called(ctx, coll) + if len(ret) == 0 { + panic("no return value specified for AddCollection") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *model.Collection) error); ok { r0 = rf(ctx, coll) @@ -77,6 +81,10 @@ func (_c *IMetaTable_AddCollection_Call) RunAndReturn(run func(context.Context, func (_m *IMetaTable) AddCredential(credInfo *internalpb.CredentialInfo) error { ret := _m.Called(credInfo) + if len(ret) == 0 { + panic("no return value specified for AddCredential") + } + var r0 error if rf, ok := ret.Get(0).(func(*internalpb.CredentialInfo) error); ok { r0 = rf(credInfo) @@ -119,6 +127,10 @@ func (_c *IMetaTable_AddCredential_Call) RunAndReturn(run func(*internalpb.Crede func (_m *IMetaTable) AddPartition(ctx context.Context, partition *model.Partition) error { ret := _m.Called(ctx, partition) + if len(ret) == 0 { + panic("no return value specified for AddPartition") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *model.Partition) error); ok { r0 = rf(ctx, partition) @@ -162,6 +174,10 @@ func (_c *IMetaTable_AddPartition_Call) RunAndReturn(run func(context.Context, * func (_m *IMetaTable) AlterAlias(ctx context.Context, dbName string, alias string, collectionName string, ts uint64) error { ret := _m.Called(ctx, dbName, alias, collectionName, ts) + if len(ret) == 0 { + panic("no return value specified for AlterAlias") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, string, string, uint64) error); ok { r0 = rf(ctx, dbName, alias, collectionName, ts) @@ -208,6 +224,10 @@ func (_c *IMetaTable_AlterAlias_Call) RunAndReturn(run func(context.Context, str func (_m *IMetaTable) AlterCollection(ctx context.Context, oldColl *model.Collection, newColl *model.Collection, ts uint64) error { ret := _m.Called(ctx, oldColl, newColl, ts) + if len(ret) == 0 { + panic("no return value specified for AlterCollection") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *model.Collection, *model.Collection, uint64) error); ok { r0 = rf(ctx, oldColl, newColl, ts) @@ -253,6 +273,10 @@ func (_c *IMetaTable_AlterCollection_Call) RunAndReturn(run func(context.Context func (_m *IMetaTable) AlterCredential(credInfo *internalpb.CredentialInfo) error { ret := _m.Called(credInfo) + if len(ret) == 0 { + panic("no return value specified for AlterCredential") + } + var r0 error if rf, ok := ret.Get(0).(func(*internalpb.CredentialInfo) error); ok { r0 = rf(credInfo) @@ -295,6 +319,10 @@ func (_c *IMetaTable_AlterCredential_Call) RunAndReturn(run func(*internalpb.Cre func (_m *IMetaTable) AlterDatabase(ctx context.Context, oldDB *model.Database, newDB *model.Database, ts uint64) error { ret := _m.Called(ctx, oldDB, newDB, ts) + if len(ret) == 0 { + panic("no return value specified for AlterDatabase") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *model.Database, *model.Database, uint64) error); ok { r0 = rf(ctx, oldDB, newDB, ts) @@ -340,6 +368,10 @@ func (_c *IMetaTable_AlterDatabase_Call) RunAndReturn(run func(context.Context, func (_m *IMetaTable) BackupRBAC(ctx context.Context, tenant string) (*milvuspb.RBACMeta, error) { ret := _m.Called(ctx, tenant) + if len(ret) == 0 { + panic("no return value specified for BackupRBAC") + } + var r0 *milvuspb.RBACMeta var r1 error if rf, ok := ret.Get(0).(func(context.Context, string) (*milvuspb.RBACMeta, error)); ok { @@ -395,6 +427,10 @@ func (_c *IMetaTable_BackupRBAC_Call) RunAndReturn(run func(context.Context, str func (_m *IMetaTable) ChangeCollectionState(ctx context.Context, collectionID int64, state etcdpb.CollectionState, ts uint64) error { ret := _m.Called(ctx, collectionID, state, ts) + if len(ret) == 0 { + panic("no return value specified for ChangeCollectionState") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, etcdpb.CollectionState, uint64) error); ok { r0 = rf(ctx, collectionID, state, ts) @@ -440,6 +476,10 @@ func (_c *IMetaTable_ChangeCollectionState_Call) RunAndReturn(run func(context.C func (_m *IMetaTable) ChangePartitionState(ctx context.Context, collectionID int64, partitionID int64, state etcdpb.PartitionState, ts uint64) error { ret := _m.Called(ctx, collectionID, partitionID, state, ts) + if len(ret) == 0 { + panic("no return value specified for ChangePartitionState") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, int64, etcdpb.PartitionState, uint64) error); ok { r0 = rf(ctx, collectionID, partitionID, state, ts) @@ -486,6 +526,10 @@ func (_c *IMetaTable_ChangePartitionState_Call) RunAndReturn(run func(context.Co func (_m *IMetaTable) CreateAlias(ctx context.Context, dbName string, alias string, collectionName string, ts uint64) error { ret := _m.Called(ctx, dbName, alias, collectionName, ts) + if len(ret) == 0 { + panic("no return value specified for CreateAlias") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, string, string, uint64) error); ok { r0 = rf(ctx, dbName, alias, collectionName, ts) @@ -532,6 +576,10 @@ func (_c *IMetaTable_CreateAlias_Call) RunAndReturn(run func(context.Context, st func (_m *IMetaTable) CreateDatabase(ctx context.Context, db *model.Database, ts uint64) error { ret := _m.Called(ctx, db, ts) + if len(ret) == 0 { + panic("no return value specified for CreateDatabase") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, *model.Database, uint64) error); ok { r0 = rf(ctx, db, ts) @@ -576,6 +624,10 @@ func (_c *IMetaTable_CreateDatabase_Call) RunAndReturn(run func(context.Context, func (_m *IMetaTable) CreateRole(tenant string, entity *milvuspb.RoleEntity) error { ret := _m.Called(tenant, entity) + if len(ret) == 0 { + panic("no return value specified for CreateRole") + } + var r0 error if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity) error); ok { r0 = rf(tenant, entity) @@ -619,6 +671,10 @@ func (_c *IMetaTable_CreateRole_Call) RunAndReturn(run func(string, *milvuspb.Ro func (_m *IMetaTable) DeleteCredential(username string) error { ret := _m.Called(username) + if len(ret) == 0 { + panic("no return value specified for DeleteCredential") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(username) @@ -661,6 +717,10 @@ func (_c *IMetaTable_DeleteCredential_Call) RunAndReturn(run func(string) error) func (_m *IMetaTable) DescribeAlias(ctx context.Context, dbName string, alias string, ts uint64) (string, error) { ret := _m.Called(ctx, dbName, alias, ts) + if len(ret) == 0 { + panic("no return value specified for DescribeAlias") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) (string, error)); ok { @@ -716,6 +776,10 @@ func (_c *IMetaTable_DescribeAlias_Call) RunAndReturn(run func(context.Context, func (_m *IMetaTable) DropAlias(ctx context.Context, dbName string, alias string, ts uint64) error { ret := _m.Called(ctx, dbName, alias, ts) + if len(ret) == 0 { + panic("no return value specified for DropAlias") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) error); ok { r0 = rf(ctx, dbName, alias, ts) @@ -761,6 +825,10 @@ func (_c *IMetaTable_DropAlias_Call) RunAndReturn(run func(context.Context, stri func (_m *IMetaTable) DropDatabase(ctx context.Context, dbName string, ts uint64) error { ret := _m.Called(ctx, dbName, ts) + if len(ret) == 0 { + panic("no return value specified for DropDatabase") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, uint64) error); ok { r0 = rf(ctx, dbName, ts) @@ -805,6 +873,10 @@ func (_c *IMetaTable_DropDatabase_Call) RunAndReturn(run func(context.Context, s func (_m *IMetaTable) DropGrant(tenant string, role *milvuspb.RoleEntity) error { ret := _m.Called(tenant, role) + if len(ret) == 0 { + panic("no return value specified for DropGrant") + } + var r0 error if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity) error); ok { r0 = rf(tenant, role) @@ -848,6 +920,10 @@ func (_c *IMetaTable_DropGrant_Call) RunAndReturn(run func(string, *milvuspb.Rol func (_m *IMetaTable) DropRole(tenant string, roleName string) error { ret := _m.Called(tenant, roleName) + if len(ret) == 0 { + panic("no return value specified for DropRole") + } + var r0 error if rf, ok := ret.Get(0).(func(string, string) error); ok { r0 = rf(tenant, roleName) @@ -891,6 +967,10 @@ func (_c *IMetaTable_DropRole_Call) RunAndReturn(run func(string, string) error) func (_m *IMetaTable) GetCollectionByID(ctx context.Context, dbName string, collectionID int64, ts uint64, allowUnavailable bool) (*model.Collection, error) { ret := _m.Called(ctx, dbName, collectionID, ts, allowUnavailable) + if len(ret) == 0 { + panic("no return value specified for GetCollectionByID") + } + var r0 *model.Collection var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, int64, uint64, bool) (*model.Collection, error)); ok { @@ -949,6 +1029,10 @@ func (_c *IMetaTable_GetCollectionByID_Call) RunAndReturn(run func(context.Conte func (_m *IMetaTable) GetCollectionByIDWithMaxTs(ctx context.Context, collectionID int64) (*model.Collection, error) { ret := _m.Called(ctx, collectionID) + if len(ret) == 0 { + panic("no return value specified for GetCollectionByIDWithMaxTs") + } + var r0 *model.Collection var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64) (*model.Collection, error)); ok { @@ -1004,6 +1088,10 @@ func (_c *IMetaTable_GetCollectionByIDWithMaxTs_Call) RunAndReturn(run func(cont func (_m *IMetaTable) GetCollectionByName(ctx context.Context, dbName string, collectionName string, ts uint64) (*model.Collection, error) { ret := _m.Called(ctx, dbName, collectionName, ts) + if len(ret) == 0 { + panic("no return value specified for GetCollectionByName") + } + var r0 *model.Collection var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) (*model.Collection, error)); ok { @@ -1061,6 +1149,10 @@ func (_c *IMetaTable_GetCollectionByName_Call) RunAndReturn(run func(context.Con func (_m *IMetaTable) GetCollectionVirtualChannels(colID int64) []string { ret := _m.Called(colID) + if len(ret) == 0 { + panic("no return value specified for GetCollectionVirtualChannels") + } + var r0 []string if rf, ok := ret.Get(0).(func(int64) []string); ok { r0 = rf(colID) @@ -1105,6 +1197,10 @@ func (_c *IMetaTable_GetCollectionVirtualChannels_Call) RunAndReturn(run func(in func (_m *IMetaTable) GetCredential(username string) (*internalpb.CredentialInfo, error) { ret := _m.Called(username) + if len(ret) == 0 { + panic("no return value specified for GetCredential") + } + var r0 *internalpb.CredentialInfo var r1 error if rf, ok := ret.Get(0).(func(string) (*internalpb.CredentialInfo, error)); ok { @@ -1159,6 +1255,10 @@ func (_c *IMetaTable_GetCredential_Call) RunAndReturn(run func(string) (*interna func (_m *IMetaTable) GetDatabaseByID(ctx context.Context, dbID int64, ts uint64) (*model.Database, error) { ret := _m.Called(ctx, dbID, ts) + if len(ret) == 0 { + panic("no return value specified for GetDatabaseByID") + } + var r0 *model.Database var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, uint64) (*model.Database, error)); ok { @@ -1215,6 +1315,10 @@ func (_c *IMetaTable_GetDatabaseByID_Call) RunAndReturn(run func(context.Context func (_m *IMetaTable) GetDatabaseByName(ctx context.Context, dbName string, ts uint64) (*model.Database, error) { ret := _m.Called(ctx, dbName, ts) + if len(ret) == 0 { + panic("no return value specified for GetDatabaseByName") + } + var r0 *model.Database var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, uint64) (*model.Database, error)); ok { @@ -1271,6 +1375,10 @@ func (_c *IMetaTable_GetDatabaseByName_Call) RunAndReturn(run func(context.Conte func (_m *IMetaTable) GetPChannelInfo(pchannel string) *rootcoordpb.GetPChannelInfoResponse { ret := _m.Called(pchannel) + if len(ret) == 0 { + panic("no return value specified for GetPChannelInfo") + } + var r0 *rootcoordpb.GetPChannelInfoResponse if rf, ok := ret.Get(0).(func(string) *rootcoordpb.GetPChannelInfoResponse); ok { r0 = rf(pchannel) @@ -1315,6 +1423,10 @@ func (_c *IMetaTable_GetPChannelInfo_Call) RunAndReturn(run func(string) *rootco func (_m *IMetaTable) IsAlias(db string, name string) bool { ret := _m.Called(db, name) + if len(ret) == 0 { + panic("no return value specified for IsAlias") + } + var r0 bool if rf, ok := ret.Get(0).(func(string, string) bool); ok { r0 = rf(db, name) @@ -1358,6 +1470,10 @@ func (_c *IMetaTable_IsAlias_Call) RunAndReturn(run func(string, string) bool) * func (_m *IMetaTable) ListAliases(ctx context.Context, dbName string, collectionName string, ts uint64) ([]string, error) { ret := _m.Called(ctx, dbName, collectionName, ts) + if len(ret) == 0 { + panic("no return value specified for ListAliases") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, string, uint64) ([]string, error)); ok { @@ -1415,6 +1531,10 @@ func (_c *IMetaTable_ListAliases_Call) RunAndReturn(run func(context.Context, st func (_m *IMetaTable) ListAliasesByID(collID int64) []string { ret := _m.Called(collID) + if len(ret) == 0 { + panic("no return value specified for ListAliasesByID") + } + var r0 []string if rf, ok := ret.Get(0).(func(int64) []string); ok { r0 = rf(collID) @@ -1459,6 +1579,10 @@ func (_c *IMetaTable_ListAliasesByID_Call) RunAndReturn(run func(int64) []string func (_m *IMetaTable) ListAllAvailCollections(ctx context.Context) map[int64][]int64 { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for ListAllAvailCollections") + } + var r0 map[int64][]int64 if rf, ok := ret.Get(0).(func(context.Context) map[int64][]int64); ok { r0 = rf(ctx) @@ -1503,6 +1627,10 @@ func (_c *IMetaTable_ListAllAvailCollections_Call) RunAndReturn(run func(context func (_m *IMetaTable) ListCollectionPhysicalChannels() map[int64][]string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ListCollectionPhysicalChannels") + } + var r0 map[int64][]string if rf, ok := ret.Get(0).(func() map[int64][]string); ok { r0 = rf() @@ -1546,6 +1674,10 @@ func (_c *IMetaTable_ListCollectionPhysicalChannels_Call) RunAndReturn(run func( func (_m *IMetaTable) ListCollections(ctx context.Context, dbName string, ts uint64, onlyAvail bool) ([]*model.Collection, error) { ret := _m.Called(ctx, dbName, ts, onlyAvail) + if len(ret) == 0 { + panic("no return value specified for ListCollections") + } + var r0 []*model.Collection var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, uint64, bool) ([]*model.Collection, error)); ok { @@ -1603,6 +1735,10 @@ func (_c *IMetaTable_ListCollections_Call) RunAndReturn(run func(context.Context func (_m *IMetaTable) ListCredentialUsernames() (*milvuspb.ListCredUsersResponse, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ListCredentialUsernames") + } + var r0 *milvuspb.ListCredUsersResponse var r1 error if rf, ok := ret.Get(0).(func() (*milvuspb.ListCredUsersResponse, error)); ok { @@ -1656,6 +1792,10 @@ func (_c *IMetaTable_ListCredentialUsernames_Call) RunAndReturn(run func() (*mil func (_m *IMetaTable) ListDatabases(ctx context.Context, ts uint64) ([]*model.Database, error) { ret := _m.Called(ctx, ts) + if len(ret) == 0 { + panic("no return value specified for ListDatabases") + } + var r0 []*model.Database var r1 error if rf, ok := ret.Get(0).(func(context.Context, uint64) ([]*model.Database, error)); ok { @@ -1711,6 +1851,10 @@ func (_c *IMetaTable_ListDatabases_Call) RunAndReturn(run func(context.Context, func (_m *IMetaTable) ListPolicy(tenant string) ([]string, error) { ret := _m.Called(tenant) + if len(ret) == 0 { + panic("no return value specified for ListPolicy") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func(string) ([]string, error)); ok { @@ -1765,6 +1909,10 @@ func (_c *IMetaTable_ListPolicy_Call) RunAndReturn(run func(string) ([]string, e func (_m *IMetaTable) ListUserRole(tenant string) ([]string, error) { ret := _m.Called(tenant) + if len(ret) == 0 { + panic("no return value specified for ListUserRole") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func(string) ([]string, error)); ok { @@ -1819,6 +1967,10 @@ func (_c *IMetaTable_ListUserRole_Call) RunAndReturn(run func(string) ([]string, func (_m *IMetaTable) OperatePrivilege(tenant string, entity *milvuspb.GrantEntity, operateType milvuspb.OperatePrivilegeType) error { ret := _m.Called(tenant, entity, operateType) + if len(ret) == 0 { + panic("no return value specified for OperatePrivilege") + } + var r0 error if rf, ok := ret.Get(0).(func(string, *milvuspb.GrantEntity, milvuspb.OperatePrivilegeType) error); ok { r0 = rf(tenant, entity, operateType) @@ -1863,6 +2015,10 @@ func (_c *IMetaTable_OperatePrivilege_Call) RunAndReturn(run func(string, *milvu func (_m *IMetaTable) OperateUserRole(tenant string, userEntity *milvuspb.UserEntity, roleEntity *milvuspb.RoleEntity, operateType milvuspb.OperateUserRoleType) error { ret := _m.Called(tenant, userEntity, roleEntity, operateType) + if len(ret) == 0 { + panic("no return value specified for OperateUserRole") + } + var r0 error if rf, ok := ret.Get(0).(func(string, *milvuspb.UserEntity, *milvuspb.RoleEntity, milvuspb.OperateUserRoleType) error); ok { r0 = rf(tenant, userEntity, roleEntity, operateType) @@ -1908,6 +2064,10 @@ func (_c *IMetaTable_OperateUserRole_Call) RunAndReturn(run func(string, *milvus func (_m *IMetaTable) RemoveCollection(ctx context.Context, collectionID int64, ts uint64) error { ret := _m.Called(ctx, collectionID, ts) + if len(ret) == 0 { + panic("no return value specified for RemoveCollection") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, uint64) error); ok { r0 = rf(ctx, collectionID, ts) @@ -1952,6 +2112,10 @@ func (_c *IMetaTable_RemoveCollection_Call) RunAndReturn(run func(context.Contex func (_m *IMetaTable) RemovePartition(ctx context.Context, dbID int64, collectionID int64, partitionID int64, ts uint64) error { ret := _m.Called(ctx, dbID, collectionID, partitionID, ts) + if len(ret) == 0 { + panic("no return value specified for RemovePartition") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, int64, int64, int64, uint64) error); ok { r0 = rf(ctx, dbID, collectionID, partitionID, ts) @@ -1998,6 +2162,10 @@ func (_c *IMetaTable_RemovePartition_Call) RunAndReturn(run func(context.Context func (_m *IMetaTable) RenameCollection(ctx context.Context, dbName string, oldName string, newDBName string, newName string, ts uint64) error { ret := _m.Called(ctx, dbName, oldName, newDBName, newName, ts) + if len(ret) == 0 { + panic("no return value specified for RenameCollection") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, string, string, string, uint64) error); ok { r0 = rf(ctx, dbName, oldName, newDBName, newName, ts) @@ -2045,6 +2213,10 @@ func (_c *IMetaTable_RenameCollection_Call) RunAndReturn(run func(context.Contex func (_m *IMetaTable) RestoreRBAC(ctx context.Context, tenant string, meta *milvuspb.RBACMeta) error { ret := _m.Called(ctx, tenant, meta) + if len(ret) == 0 { + panic("no return value specified for RestoreRBAC") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, string, *milvuspb.RBACMeta) error); ok { r0 = rf(ctx, tenant, meta) @@ -2089,6 +2261,10 @@ func (_c *IMetaTable_RestoreRBAC_Call) RunAndReturn(run func(context.Context, st func (_m *IMetaTable) SelectGrant(tenant string, entity *milvuspb.GrantEntity) ([]*milvuspb.GrantEntity, error) { ret := _m.Called(tenant, entity) + if len(ret) == 0 { + panic("no return value specified for SelectGrant") + } + var r0 []*milvuspb.GrantEntity var r1 error if rf, ok := ret.Get(0).(func(string, *milvuspb.GrantEntity) ([]*milvuspb.GrantEntity, error)); ok { @@ -2144,6 +2320,10 @@ func (_c *IMetaTable_SelectGrant_Call) RunAndReturn(run func(string, *milvuspb.G func (_m *IMetaTable) SelectRole(tenant string, entity *milvuspb.RoleEntity, includeUserInfo bool) ([]*milvuspb.RoleResult, error) { ret := _m.Called(tenant, entity, includeUserInfo) + if len(ret) == 0 { + panic("no return value specified for SelectRole") + } + var r0 []*milvuspb.RoleResult var r1 error if rf, ok := ret.Get(0).(func(string, *milvuspb.RoleEntity, bool) ([]*milvuspb.RoleResult, error)); ok { @@ -2200,6 +2380,10 @@ func (_c *IMetaTable_SelectRole_Call) RunAndReturn(run func(string, *milvuspb.Ro func (_m *IMetaTable) SelectUser(tenant string, entity *milvuspb.UserEntity, includeRoleInfo bool) ([]*milvuspb.UserResult, error) { ret := _m.Called(tenant, entity, includeRoleInfo) + if len(ret) == 0 { + panic("no return value specified for SelectUser") + } + var r0 []*milvuspb.UserResult var r1 error if rf, ok := ret.Get(0).(func(string, *milvuspb.UserEntity, bool) ([]*milvuspb.UserResult, error)); ok { diff --git a/pkg/eventlog/mock_logger.go b/pkg/eventlog/mock_logger.go index 8d5c8c3306..41ded424f6 100644 --- a/pkg/eventlog/mock_logger.go +++ b/pkg/eventlog/mock_logger.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package eventlog @@ -21,6 +21,10 @@ func (_m *MockLogger) EXPECT() *MockLogger_Expecter { func (_m *MockLogger) Flush() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Flush") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() diff --git a/pkg/kv/predicates/mock_predicate.go b/pkg/kv/predicates/mock_predicate.go index 1183a47881..eace7337f2 100644 --- a/pkg/kv/predicates/mock_predicate.go +++ b/pkg/kv/predicates/mock_predicate.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package predicates @@ -21,6 +21,10 @@ func (_m *MockPredicate) EXPECT() *MockPredicate_Expecter { func (_m *MockPredicate) IsTrue(_a0 interface{}) bool { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for IsTrue") + } + var r0 bool if rf, ok := ret.Get(0).(func(interface{}) bool); ok { r0 = rf(_a0) @@ -63,6 +67,10 @@ func (_c *MockPredicate_IsTrue_Call) RunAndReturn(run func(interface{}) bool) *M func (_m *MockPredicate) Key() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Key") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -104,6 +112,10 @@ func (_c *MockPredicate_Key_Call) RunAndReturn(run func() string) *MockPredicate func (_m *MockPredicate) Target() PredicateTarget { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Target") + } + var r0 PredicateTarget if rf, ok := ret.Get(0).(func() PredicateTarget); ok { r0 = rf() @@ -145,6 +157,10 @@ func (_c *MockPredicate_Target_Call) RunAndReturn(run func() PredicateTarget) *M func (_m *MockPredicate) TargetValue() interface{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TargetValue") + } + var r0 interface{} if rf, ok := ret.Get(0).(func() interface{}); ok { r0 = rf() @@ -188,6 +204,10 @@ func (_c *MockPredicate_TargetValue_Call) RunAndReturn(run func() interface{}) * func (_m *MockPredicate) Type() PredicateType { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Type") + } + var r0 PredicateType if rf, ok := ret.Get(0).(func() PredicateType); ok { r0 = rf() diff --git a/pkg/mocks/mock_kv/mock_MetaKv.go b/pkg/mocks/mock_kv/mock_MetaKv.go index 5744be0fc7..52702fd8cf 100644 --- a/pkg/mocks/mock_kv/mock_MetaKv.go +++ b/pkg/mocks/mock_kv/mock_MetaKv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_kv @@ -56,6 +56,10 @@ func (_c *MockMetaKv_Close_Call) RunAndReturn(run func()) *MockMetaKv_Close_Call func (_m *MockMetaKv) CompareVersionAndSwap(key string, version int64, target string) (bool, error) { ret := _m.Called(key, version, target) + if len(ret) == 0 { + panic("no return value specified for CompareVersionAndSwap") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string, int64, string) (bool, error)); ok { @@ -110,6 +114,10 @@ func (_c *MockMetaKv_CompareVersionAndSwap_Call) RunAndReturn(run func(string, i func (_m *MockMetaKv) GetPath(key string) string { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for GetPath") + } + var r0 string if rf, ok := ret.Get(0).(func(string) string); ok { r0 = rf(key) @@ -152,6 +160,10 @@ func (_c *MockMetaKv_GetPath_Call) RunAndReturn(run func(string) string) *MockMe func (_m *MockMetaKv) Has(key string) (bool, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Has") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -204,6 +216,10 @@ func (_c *MockMetaKv_Has_Call) RunAndReturn(run func(string) (bool, error)) *Moc func (_m *MockMetaKv) HasPrefix(prefix string) (bool, error) { ret := _m.Called(prefix) + if len(ret) == 0 { + panic("no return value specified for HasPrefix") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func(string) (bool, error)); ok { @@ -256,6 +272,10 @@ func (_c *MockMetaKv_HasPrefix_Call) RunAndReturn(run func(string) (bool, error) func (_m *MockMetaKv) Load(key string) (string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Load") + } + var r0 string var r1 error if rf, ok := ret.Get(0).(func(string) (string, error)); ok { @@ -308,6 +328,10 @@ func (_c *MockMetaKv_Load_Call) RunAndReturn(run func(string) (string, error)) * func (_m *MockMetaKv) LoadWithPrefix(key string) ([]string, []string, error) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for LoadWithPrefix") + } + var r0 []string var r1 []string var r2 error @@ -371,6 +395,10 @@ func (_c *MockMetaKv_LoadWithPrefix_Call) RunAndReturn(run func(string) ([]strin func (_m *MockMetaKv) MultiLoad(keys []string) ([]string, error) { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiLoad") + } + var r0 []string var r1 error if rf, ok := ret.Get(0).(func([]string) ([]string, error)); ok { @@ -425,6 +453,10 @@ func (_c *MockMetaKv_MultiLoad_Call) RunAndReturn(run func([]string) ([]string, func (_m *MockMetaKv) MultiRemove(keys []string) error { ret := _m.Called(keys) + if len(ret) == 0 { + panic("no return value specified for MultiRemove") + } + var r0 error if rf, ok := ret.Get(0).(func([]string) error); ok { r0 = rf(keys) @@ -467,6 +499,10 @@ func (_c *MockMetaKv_MultiRemove_Call) RunAndReturn(run func([]string) error) *M func (_m *MockMetaKv) MultiSave(kvs map[string]string) error { ret := _m.Called(kvs) + if len(ret) == 0 { + panic("no return value specified for MultiSave") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string) error); ok { r0 = rf(kvs) @@ -516,6 +552,10 @@ func (_m *MockMetaKv) MultiSaveAndRemove(saves map[string]string, removals []str _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemove") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -574,6 +614,10 @@ func (_m *MockMetaKv) MultiSaveAndRemoveWithPrefix(saves map[string]string, remo _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for MultiSaveAndRemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(map[string]string, []string, ...predicates.Predicate) error); ok { r0 = rf(saves, removals, preds...) @@ -625,6 +669,10 @@ func (_c *MockMetaKv_MultiSaveAndRemoveWithPrefix_Call) RunAndReturn(run func(ma func (_m *MockMetaKv) Remove(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -667,6 +715,10 @@ func (_c *MockMetaKv_Remove_Call) RunAndReturn(run func(string) error) *MockMeta func (_m *MockMetaKv) RemoveWithPrefix(key string) error { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for RemoveWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(key) @@ -709,6 +761,10 @@ func (_c *MockMetaKv_RemoveWithPrefix_Call) RunAndReturn(run func(string) error) func (_m *MockMetaKv) Save(key string, value string) error { ret := _m.Called(key, value) + if len(ret) == 0 { + panic("no return value specified for Save") + } + var r0 error if rf, ok := ret.Get(0).(func(string, string) error); ok { r0 = rf(key, value) @@ -752,6 +808,10 @@ func (_c *MockMetaKv_Save_Call) RunAndReturn(run func(string, string) error) *Mo func (_m *MockMetaKv) WalkWithPrefix(prefix string, paginationSize int, fn func([]byte, []byte) error) error { ret := _m.Called(prefix, paginationSize, fn) + if len(ret) == 0 { + panic("no return value specified for WalkWithPrefix") + } + var r0 error if rf, ok := ret.Get(0).(func(string, int, func([]byte, []byte) error) error); ok { r0 = rf(prefix, paginationSize, fn) diff --git a/pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go index c99c0a7e74..04682f4d50 100644 --- a/pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go +++ b/pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_walimpls @@ -24,6 +24,10 @@ func (_m *MockOpenerBuilderImpls) EXPECT() *MockOpenerBuilderImpls_Expecter { func (_m *MockOpenerBuilderImpls) Build() (walimpls.OpenerImpls, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Build") + } + var r0 walimpls.OpenerImpls var r1 error if rf, ok := ret.Get(0).(func() (walimpls.OpenerImpls, error)); ok { @@ -77,6 +81,10 @@ func (_c *MockOpenerBuilderImpls_Build_Call) RunAndReturn(run func() (walimpls.O func (_m *MockOpenerBuilderImpls) Name() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Name") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go index 1cc66433fd..0f5c5f8550 100644 --- a/pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go +++ b/pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_walimpls @@ -58,6 +58,10 @@ func (_c *MockOpenerImpls_Close_Call) RunAndReturn(run func()) *MockOpenerImpls_ func (_m *MockOpenerImpls) Open(ctx context.Context, opt *walimpls.OpenOption) (walimpls.WALImpls, error) { ret := _m.Called(ctx, opt) + if len(ret) == 0 { + panic("no return value specified for Open") + } + var r0 walimpls.WALImpls var r1 error if rf, ok := ret.Get(0).(func(context.Context, *walimpls.OpenOption) (walimpls.WALImpls, error)); ok { diff --git a/pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go index 14feb42926..783a63b5eb 100644 --- a/pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go +++ b/pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_walimpls @@ -24,6 +24,10 @@ func (_m *MockScannerImpls) EXPECT() *MockScannerImpls_Expecter { func (_m *MockScannerImpls) Chan() <-chan message.ImmutableMessage { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Chan") + } + var r0 <-chan message.ImmutableMessage if rf, ok := ret.Get(0).(func() <-chan message.ImmutableMessage); ok { r0 = rf() @@ -67,6 +71,10 @@ func (_c *MockScannerImpls_Chan_Call) RunAndReturn(run func() <-chan message.Imm func (_m *MockScannerImpls) Close() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Close") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -108,6 +116,10 @@ func (_c *MockScannerImpls_Close_Call) RunAndReturn(run func() error) *MockScann func (_m *MockScannerImpls) Done() <-chan struct{} { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Done") + } + var r0 <-chan struct{} if rf, ok := ret.Get(0).(func() <-chan struct{}); ok { r0 = rf() @@ -151,6 +163,10 @@ func (_c *MockScannerImpls_Done_Call) RunAndReturn(run func() <-chan struct{}) * func (_m *MockScannerImpls) Error() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Error") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -192,6 +208,10 @@ func (_c *MockScannerImpls_Error_Call) RunAndReturn(run func() error) *MockScann func (_m *MockScannerImpls) Name() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Name") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go index f85f320cb8..2b2b77d261 100644 --- a/pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go +++ b/pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_walimpls @@ -30,6 +30,10 @@ func (_m *MockWALImpls) EXPECT() *MockWALImpls_Expecter { func (_m *MockWALImpls) Append(ctx context.Context, msg message.MutableMessage) (message.MessageID, error) { ret := _m.Called(ctx, msg) + if len(ret) == 0 { + panic("no return value specified for Append") + } + var r0 message.MessageID var r1 error if rf, ok := ret.Get(0).(func(context.Context, message.MutableMessage) (message.MessageID, error)); ok { @@ -85,6 +89,10 @@ func (_c *MockWALImpls_Append_Call) RunAndReturn(run func(context.Context, messa func (_m *MockWALImpls) Channel() types.PChannelInfo { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Channel") + } + var r0 types.PChannelInfo if rf, ok := ret.Get(0).(func() types.PChannelInfo); ok { r0 = rf() @@ -158,6 +166,10 @@ func (_c *MockWALImpls_Close_Call) RunAndReturn(run func()) *MockWALImpls_Close_ func (_m *MockWALImpls) Read(ctx context.Context, opts walimpls.ReadOption) (walimpls.ScannerImpls, error) { ret := _m.Called(ctx, opts) + if len(ret) == 0 { + panic("no return value specified for Read") + } + var r0 walimpls.ScannerImpls var r1 error if rf, ok := ret.Get(0).(func(context.Context, walimpls.ReadOption) (walimpls.ScannerImpls, error)); ok { @@ -213,6 +225,10 @@ func (_c *MockWALImpls_Read_Call) RunAndReturn(run func(context.Context, walimpl func (_m *MockWALImpls) WALName() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for WALName") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go index 494efb8188..904bd2b08a 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -36,6 +36,10 @@ func (_m *MockStreamingCoordAssignmentServiceClient) AssignmentDiscover(ctx cont _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for AssignmentDiscover") + } + var r0 streamingpb.StreamingCoordAssignmentService_AssignmentDiscoverClient var r1 error if rf, ok := ret.Get(0).(func(context.Context, ...grpc.CallOption) (streamingpb.StreamingCoordAssignmentService_AssignmentDiscoverClient, error)); ok { diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go index 9cbc322f3c..a9e201ab88 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -28,6 +28,10 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) EXPECT() func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) CloseSend() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -69,6 +73,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_CloseSend func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -112,6 +120,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Context_C func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Header() (metadata.MD, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Header") + } + var r0 metadata.MD var r1 error if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { @@ -165,6 +177,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Header_Ca func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Recv() (*streamingpb.AssignmentDiscoverResponse, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Recv") + } + var r0 *streamingpb.AssignmentDiscoverResponse var r1 error if rf, ok := ret.Get(0).(func() (*streamingpb.AssignmentDiscoverResponse, error)); ok { @@ -218,6 +234,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Recv_Call func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -260,6 +280,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_RecvMsg_C func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Send(_a0 *streamingpb.AssignmentDiscoverRequest) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Send") + } + var r0 error if rf, ok := ret.Get(0).(func(*streamingpb.AssignmentDiscoverRequest) error); ok { r0 = rf(_a0) @@ -302,6 +326,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Send_Call func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -344,6 +372,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_SendMsg_C func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Trailer() metadata.MD { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + var r0 metadata.MD if rf, ok := ret.Get(0).(func() metadata.MD); ok { r0 = rf() diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go index 7fb9e2e2b1..2b130b409d 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -28,6 +28,10 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) EXPECT() func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -71,6 +75,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Context_C func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Recv() (*streamingpb.AssignmentDiscoverRequest, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Recv") + } + var r0 *streamingpb.AssignmentDiscoverRequest var r1 error if rf, ok := ret.Get(0).(func() (*streamingpb.AssignmentDiscoverRequest, error)); ok { @@ -124,6 +132,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Recv_Call func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -166,6 +178,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_RecvMsg_C func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Send(_a0 *streamingpb.AssignmentDiscoverResponse) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Send") + } + var r0 error if rf, ok := ret.Get(0).(func(*streamingpb.AssignmentDiscoverResponse) error); ok { r0 = rf(_a0) @@ -208,6 +224,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Send_Call func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) SendHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for SendHeader") + } + var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) @@ -250,6 +270,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SendHeade func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -292,6 +316,10 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SendMsg_C func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) SetHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for SetHeader") + } + var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go index 16c3a26efb..54b30eddfe 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -36,6 +36,10 @@ func (_m *MockStreamingNodeHandlerServiceClient) Consume(ctx context.Context, op _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Consume") + } + var r0 streamingpb.StreamingNodeHandlerService_ConsumeClient var r1 error if rf, ok := ret.Get(0).(func(context.Context, ...grpc.CallOption) (streamingpb.StreamingNodeHandlerService_ConsumeClient, error)); ok { @@ -105,6 +109,10 @@ func (_m *MockStreamingNodeHandlerServiceClient) Produce(ctx context.Context, op _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Produce") + } + var r0 streamingpb.StreamingNodeHandlerService_ProduceClient var r1 error if rf, ok := ret.Get(0).(func(context.Context, ...grpc.CallOption) (streamingpb.StreamingNodeHandlerService_ProduceClient, error)); ok { diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go index 181ecee161..335f43af0c 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ConsumeClient) EXPECT() *MockStreaming func (_m *MockStreamingNodeHandlerService_ConsumeClient) CloseSend() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -69,6 +73,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_CloseSend_Call) RunAndRe func (_m *MockStreamingNodeHandlerService_ConsumeClient) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -112,6 +120,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Context_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ConsumeClient) Header() (metadata.MD, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Header") + } + var r0 metadata.MD var r1 error if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { @@ -165,6 +177,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Header_Call) RunAndRetur func (_m *MockStreamingNodeHandlerService_ConsumeClient) Recv() (*streamingpb.ConsumeResponse, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Recv") + } + var r0 *streamingpb.ConsumeResponse var r1 error if rf, ok := ret.Get(0).(func() (*streamingpb.ConsumeResponse, error)); ok { @@ -218,6 +234,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Recv_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ConsumeClient) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -260,6 +280,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_RecvMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ConsumeClient) Send(_a0 *streamingpb.ConsumeRequest) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Send") + } + var r0 error if rf, ok := ret.Get(0).(func(*streamingpb.ConsumeRequest) error); ok { r0 = rf(_a0) @@ -302,6 +326,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Send_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ConsumeClient) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -344,6 +372,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_SendMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ConsumeClient) Trailer() metadata.MD { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + var r0 metadata.MD if rf, ok := ret.Get(0).(func() metadata.MD); ok { r0 = rf() diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go index 35f9f22e8d..fa6c99ddbf 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ConsumeServer) EXPECT() *MockStreaming func (_m *MockStreamingNodeHandlerService_ConsumeServer) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -71,6 +75,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Context_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ConsumeServer) Recv() (*streamingpb.ConsumeRequest, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Recv") + } + var r0 *streamingpb.ConsumeRequest var r1 error if rf, ok := ret.Get(0).(func() (*streamingpb.ConsumeRequest, error)); ok { @@ -124,6 +132,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Recv_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ConsumeServer) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -166,6 +178,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_RecvMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ConsumeServer) Send(_a0 *streamingpb.ConsumeResponse) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Send") + } + var r0 error if rf, ok := ret.Get(0).(func(*streamingpb.ConsumeResponse) error); ok { r0 = rf(_a0) @@ -208,6 +224,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Send_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ConsumeServer) SendHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for SendHeader") + } + var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) @@ -250,6 +270,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_SendHeader_Call) RunAndR func (_m *MockStreamingNodeHandlerService_ConsumeServer) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -292,6 +316,10 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_SendMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ConsumeServer) SetHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for SetHeader") + } + var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go index be833cb96e..784e313b26 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ProduceClient) EXPECT() *MockStreaming func (_m *MockStreamingNodeHandlerService_ProduceClient) CloseSend() error { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + var r0 error if rf, ok := ret.Get(0).(func() error); ok { r0 = rf() @@ -69,6 +73,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_CloseSend_Call) RunAndRe func (_m *MockStreamingNodeHandlerService_ProduceClient) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -112,6 +120,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Context_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ProduceClient) Header() (metadata.MD, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Header") + } + var r0 metadata.MD var r1 error if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { @@ -165,6 +177,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Header_Call) RunAndRetur func (_m *MockStreamingNodeHandlerService_ProduceClient) Recv() (*streamingpb.ProduceResponse, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Recv") + } + var r0 *streamingpb.ProduceResponse var r1 error if rf, ok := ret.Get(0).(func() (*streamingpb.ProduceResponse, error)); ok { @@ -218,6 +234,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Recv_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ProduceClient) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -260,6 +280,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_RecvMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ProduceClient) Send(_a0 *streamingpb.ProduceRequest) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Send") + } + var r0 error if rf, ok := ret.Get(0).(func(*streamingpb.ProduceRequest) error); ok { r0 = rf(_a0) @@ -302,6 +326,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Send_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ProduceClient) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -344,6 +372,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_SendMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ProduceClient) Trailer() metadata.MD { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + var r0 metadata.MD if rf, ok := ret.Get(0).(func() metadata.MD); ok { r0 = rf() diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go index b6bf7eb764..7126e7a3ca 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -28,6 +28,10 @@ func (_m *MockStreamingNodeHandlerService_ProduceServer) EXPECT() *MockStreaming func (_m *MockStreamingNodeHandlerService_ProduceServer) Context() context.Context { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Context") + } + var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() @@ -71,6 +75,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Context_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ProduceServer) Recv() (*streamingpb.ProduceRequest, error) { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Recv") + } + var r0 *streamingpb.ProduceRequest var r1 error if rf, ok := ret.Get(0).(func() (*streamingpb.ProduceRequest, error)); ok { @@ -124,6 +132,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Recv_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ProduceServer) RecvMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -166,6 +178,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_RecvMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ProduceServer) Send(_a0 *streamingpb.ProduceResponse) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Send") + } + var r0 error if rf, ok := ret.Get(0).(func(*streamingpb.ProduceResponse) error); ok { r0 = rf(_a0) @@ -208,6 +224,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Send_Call) RunAndReturn( func (_m *MockStreamingNodeHandlerService_ProduceServer) SendHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for SendHeader") + } + var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) @@ -250,6 +270,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_SendHeader_Call) RunAndR func (_m *MockStreamingNodeHandlerService_ProduceServer) SendMsg(m interface{}) error { ret := _m.Called(m) + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + var r0 error if rf, ok := ret.Get(0).(func(interface{}) error); ok { r0 = rf(m) @@ -292,6 +316,10 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_SendMsg_Call) RunAndRetu func (_m *MockStreamingNodeHandlerService_ProduceServer) SetHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for SetHeader") + } + var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go index e0d8cf280f..e278c6588c 100644 --- a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go +++ b/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_streamingpb @@ -36,6 +36,10 @@ func (_m *MockStreamingNodeManagerServiceClient) Assign(ctx context.Context, in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Assign") + } + var r0 *streamingpb.StreamingNodeManagerAssignResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *streamingpb.StreamingNodeManagerAssignRequest, ...grpc.CallOption) (*streamingpb.StreamingNodeManagerAssignResponse, error)); ok { @@ -106,6 +110,10 @@ func (_m *MockStreamingNodeManagerServiceClient) CollectStatus(ctx context.Conte _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for CollectStatus") + } + var r0 *streamingpb.StreamingNodeManagerCollectStatusResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *streamingpb.StreamingNodeManagerCollectStatusRequest, ...grpc.CallOption) (*streamingpb.StreamingNodeManagerCollectStatusResponse, error)); ok { @@ -176,6 +184,10 @@ func (_m *MockStreamingNodeManagerServiceClient) Remove(ctx context.Context, in _ca = append(_ca, _va...) ret := _m.Called(_ca...) + if len(ret) == 0 { + panic("no return value specified for Remove") + } + var r0 *streamingpb.StreamingNodeManagerRemoveResponse var r1 error if rf, ok := ret.Get(0).(func(context.Context, *streamingpb.StreamingNodeManagerRemoveRequest, ...grpc.CallOption) (*streamingpb.StreamingNodeManagerRemoveResponse, error)); ok { diff --git a/pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go b/pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go index 3b3551268d..7d13ac551b 100644 --- a/pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go +++ b/pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_message @@ -24,6 +24,10 @@ func (_m *MockImmutableMessage) EXPECT() *MockImmutableMessage_Expecter { func (_m *MockImmutableMessage) BarrierTimeTick() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for BarrierTimeTick") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -65,6 +69,10 @@ func (_c *MockImmutableMessage_BarrierTimeTick_Call) RunAndReturn(run func() uin func (_m *MockImmutableMessage) EstimateSize() int { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for EstimateSize") + } + var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() @@ -106,6 +114,10 @@ func (_c *MockImmutableMessage_EstimateSize_Call) RunAndReturn(run func() int) * func (_m *MockImmutableMessage) LastConfirmedMessageID() message.MessageID { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for LastConfirmedMessageID") + } + var r0 message.MessageID if rf, ok := ret.Get(0).(func() message.MessageID); ok { r0 = rf() @@ -149,6 +161,10 @@ func (_c *MockImmutableMessage_LastConfirmedMessageID_Call) RunAndReturn(run fun func (_m *MockImmutableMessage) MessageID() message.MessageID { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for MessageID") + } + var r0 message.MessageID if rf, ok := ret.Get(0).(func() message.MessageID); ok { r0 = rf() @@ -192,6 +208,10 @@ func (_c *MockImmutableMessage_MessageID_Call) RunAndReturn(run func() message.M func (_m *MockImmutableMessage) MessageType() message.MessageType { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for MessageType") + } + var r0 message.MessageType if rf, ok := ret.Get(0).(func() message.MessageType); ok { r0 = rf() @@ -233,6 +253,10 @@ func (_c *MockImmutableMessage_MessageType_Call) RunAndReturn(run func() message func (_m *MockImmutableMessage) Payload() []byte { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Payload") + } + var r0 []byte if rf, ok := ret.Get(0).(func() []byte); ok { r0 = rf() @@ -276,6 +300,10 @@ func (_c *MockImmutableMessage_Payload_Call) RunAndReturn(run func() []byte) *Mo func (_m *MockImmutableMessage) Properties() message.RProperties { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Properties") + } + var r0 message.RProperties if rf, ok := ret.Get(0).(func() message.RProperties); ok { r0 = rf() @@ -319,6 +347,10 @@ func (_c *MockImmutableMessage_Properties_Call) RunAndReturn(run func() message. func (_m *MockImmutableMessage) TimeTick() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TimeTick") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -360,6 +392,10 @@ func (_c *MockImmutableMessage_TimeTick_Call) RunAndReturn(run func() uint64) *M func (_m *MockImmutableMessage) TxnContext() *message.TxnContext { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TxnContext") + } + var r0 *message.TxnContext if rf, ok := ret.Get(0).(func() *message.TxnContext); ok { r0 = rf() @@ -403,6 +439,10 @@ func (_c *MockImmutableMessage_TxnContext_Call) RunAndReturn(run func() *message func (_m *MockImmutableMessage) VChannel() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for VChannel") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -444,6 +484,10 @@ func (_c *MockImmutableMessage_VChannel_Call) RunAndReturn(run func() string) *M func (_m *MockImmutableMessage) Version() message.Version { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Version") + } + var r0 message.Version if rf, ok := ret.Get(0).(func() message.Version); ok { r0 = rf() @@ -485,6 +529,10 @@ func (_c *MockImmutableMessage_Version_Call) RunAndReturn(run func() message.Ver func (_m *MockImmutableMessage) WALName() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for WALName") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go b/pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go index 2e69295279..21f8b21c3a 100644 --- a/pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go +++ b/pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_message @@ -24,6 +24,10 @@ func (_m *MockImmutableTxnMessage) EXPECT() *MockImmutableTxnMessage_Expecter { func (_m *MockImmutableTxnMessage) BarrierTimeTick() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for BarrierTimeTick") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -65,6 +69,10 @@ func (_c *MockImmutableTxnMessage_BarrierTimeTick_Call) RunAndReturn(run func() func (_m *MockImmutableTxnMessage) Begin() message.ImmutableMessage { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Begin") + } + var r0 message.ImmutableMessage if rf, ok := ret.Get(0).(func() message.ImmutableMessage); ok { r0 = rf() @@ -108,6 +116,10 @@ func (_c *MockImmutableTxnMessage_Begin_Call) RunAndReturn(run func() message.Im func (_m *MockImmutableTxnMessage) Commit() message.ImmutableMessage { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Commit") + } + var r0 message.ImmutableMessage if rf, ok := ret.Get(0).(func() message.ImmutableMessage); ok { r0 = rf() @@ -151,6 +163,10 @@ func (_c *MockImmutableTxnMessage_Commit_Call) RunAndReturn(run func() message.I func (_m *MockImmutableTxnMessage) EstimateSize() int { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for EstimateSize") + } + var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() @@ -192,6 +208,10 @@ func (_c *MockImmutableTxnMessage_EstimateSize_Call) RunAndReturn(run func() int func (_m *MockImmutableTxnMessage) LastConfirmedMessageID() message.MessageID { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for LastConfirmedMessageID") + } + var r0 message.MessageID if rf, ok := ret.Get(0).(func() message.MessageID); ok { r0 = rf() @@ -235,6 +255,10 @@ func (_c *MockImmutableTxnMessage_LastConfirmedMessageID_Call) RunAndReturn(run func (_m *MockImmutableTxnMessage) MessageID() message.MessageID { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for MessageID") + } + var r0 message.MessageID if rf, ok := ret.Get(0).(func() message.MessageID); ok { r0 = rf() @@ -278,6 +302,10 @@ func (_c *MockImmutableTxnMessage_MessageID_Call) RunAndReturn(run func() messag func (_m *MockImmutableTxnMessage) MessageType() message.MessageType { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for MessageType") + } + var r0 message.MessageType if rf, ok := ret.Get(0).(func() message.MessageType); ok { r0 = rf() @@ -319,6 +347,10 @@ func (_c *MockImmutableTxnMessage_MessageType_Call) RunAndReturn(run func() mess func (_m *MockImmutableTxnMessage) Payload() []byte { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Payload") + } + var r0 []byte if rf, ok := ret.Get(0).(func() []byte); ok { r0 = rf() @@ -362,6 +394,10 @@ func (_c *MockImmutableTxnMessage_Payload_Call) RunAndReturn(run func() []byte) func (_m *MockImmutableTxnMessage) Properties() message.RProperties { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Properties") + } + var r0 message.RProperties if rf, ok := ret.Get(0).(func() message.RProperties); ok { r0 = rf() @@ -405,6 +441,10 @@ func (_c *MockImmutableTxnMessage_Properties_Call) RunAndReturn(run func() messa func (_m *MockImmutableTxnMessage) RangeOver(visitor func(message.ImmutableMessage) error) error { ret := _m.Called(visitor) + if len(ret) == 0 { + panic("no return value specified for RangeOver") + } + var r0 error if rf, ok := ret.Get(0).(func(func(message.ImmutableMessage) error) error); ok { r0 = rf(visitor) @@ -447,6 +487,10 @@ func (_c *MockImmutableTxnMessage_RangeOver_Call) RunAndReturn(run func(func(mes func (_m *MockImmutableTxnMessage) Size() int { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Size") + } + var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() @@ -488,6 +532,10 @@ func (_c *MockImmutableTxnMessage_Size_Call) RunAndReturn(run func() int) *MockI func (_m *MockImmutableTxnMessage) TimeTick() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TimeTick") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -529,6 +577,10 @@ func (_c *MockImmutableTxnMessage_TimeTick_Call) RunAndReturn(run func() uint64) func (_m *MockImmutableTxnMessage) TxnContext() *message.TxnContext { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TxnContext") + } + var r0 *message.TxnContext if rf, ok := ret.Get(0).(func() *message.TxnContext); ok { r0 = rf() @@ -572,6 +624,10 @@ func (_c *MockImmutableTxnMessage_TxnContext_Call) RunAndReturn(run func() *mess func (_m *MockImmutableTxnMessage) VChannel() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for VChannel") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -613,6 +669,10 @@ func (_c *MockImmutableTxnMessage_VChannel_Call) RunAndReturn(run func() string) func (_m *MockImmutableTxnMessage) Version() message.Version { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Version") + } + var r0 message.Version if rf, ok := ret.Get(0).(func() message.Version); ok { r0 = rf() @@ -654,6 +714,10 @@ func (_c *MockImmutableTxnMessage_Version_Call) RunAndReturn(run func() message. func (_m *MockImmutableTxnMessage) WALName() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for WALName") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/util/mock_message/mock_MessageID.go b/pkg/mocks/streaming/util/mock_message/mock_MessageID.go index 92222732d9..ba042676f8 100644 --- a/pkg/mocks/streaming/util/mock_message/mock_MessageID.go +++ b/pkg/mocks/streaming/util/mock_message/mock_MessageID.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_message @@ -24,6 +24,10 @@ func (_m *MockMessageID) EXPECT() *MockMessageID_Expecter { func (_m *MockMessageID) EQ(_a0 message.MessageID) bool { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for EQ") + } + var r0 bool if rf, ok := ret.Get(0).(func(message.MessageID) bool); ok { r0 = rf(_a0) @@ -66,6 +70,10 @@ func (_c *MockMessageID_EQ_Call) RunAndReturn(run func(message.MessageID) bool) func (_m *MockMessageID) LT(_a0 message.MessageID) bool { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for LT") + } + var r0 bool if rf, ok := ret.Get(0).(func(message.MessageID) bool); ok { r0 = rf(_a0) @@ -108,6 +116,10 @@ func (_c *MockMessageID_LT_Call) RunAndReturn(run func(message.MessageID) bool) func (_m *MockMessageID) LTE(_a0 message.MessageID) bool { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for LTE") + } + var r0 bool if rf, ok := ret.Get(0).(func(message.MessageID) bool); ok { r0 = rf(_a0) @@ -150,6 +162,10 @@ func (_c *MockMessageID_LTE_Call) RunAndReturn(run func(message.MessageID) bool) func (_m *MockMessageID) Marshal() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Marshal") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -191,6 +207,10 @@ func (_c *MockMessageID_Marshal_Call) RunAndReturn(run func() string) *MockMessa func (_m *MockMessageID) String() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for String") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -232,6 +252,10 @@ func (_c *MockMessageID_String_Call) RunAndReturn(run func() string) *MockMessag func (_m *MockMessageID) WALName() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for WALName") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go b/pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go index 960c012e46..c3fbdc02d4 100644 --- a/pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go +++ b/pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_message @@ -24,6 +24,10 @@ func (_m *MockMutableMessage) EXPECT() *MockMutableMessage_Expecter { func (_m *MockMutableMessage) BarrierTimeTick() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for BarrierTimeTick") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -65,6 +69,10 @@ func (_c *MockMutableMessage_BarrierTimeTick_Call) RunAndReturn(run func() uint6 func (_m *MockMutableMessage) EstimateSize() int { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for EstimateSize") + } + var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() @@ -106,6 +114,10 @@ func (_c *MockMutableMessage_EstimateSize_Call) RunAndReturn(run func() int) *Mo func (_m *MockMutableMessage) IntoImmutableMessage(msgID message.MessageID) message.ImmutableMessage { ret := _m.Called(msgID) + if len(ret) == 0 { + panic("no return value specified for IntoImmutableMessage") + } + var r0 message.ImmutableMessage if rf, ok := ret.Get(0).(func(message.MessageID) message.ImmutableMessage); ok { r0 = rf(msgID) @@ -150,6 +162,10 @@ func (_c *MockMutableMessage_IntoImmutableMessage_Call) RunAndReturn(run func(me func (_m *MockMutableMessage) MessageType() message.MessageType { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for MessageType") + } + var r0 message.MessageType if rf, ok := ret.Get(0).(func() message.MessageType); ok { r0 = rf() @@ -191,6 +207,10 @@ func (_c *MockMutableMessage_MessageType_Call) RunAndReturn(run func() message.M func (_m *MockMutableMessage) Payload() []byte { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Payload") + } + var r0 []byte if rf, ok := ret.Get(0).(func() []byte); ok { r0 = rf() @@ -234,6 +254,10 @@ func (_c *MockMutableMessage_Payload_Call) RunAndReturn(run func() []byte) *Mock func (_m *MockMutableMessage) Properties() message.RProperties { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Properties") + } + var r0 message.RProperties if rf, ok := ret.Get(0).(func() message.RProperties); ok { r0 = rf() @@ -277,6 +301,10 @@ func (_c *MockMutableMessage_Properties_Call) RunAndReturn(run func() message.RP func (_m *MockMutableMessage) TimeTick() uint64 { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TimeTick") + } + var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() @@ -318,6 +346,10 @@ func (_c *MockMutableMessage_TimeTick_Call) RunAndReturn(run func() uint64) *Moc func (_m *MockMutableMessage) TxnContext() *message.TxnContext { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for TxnContext") + } + var r0 *message.TxnContext if rf, ok := ret.Get(0).(func() *message.TxnContext); ok { r0 = rf() @@ -361,6 +393,10 @@ func (_c *MockMutableMessage_TxnContext_Call) RunAndReturn(run func() *message.T func (_m *MockMutableMessage) VChannel() string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for VChannel") + } + var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() @@ -402,6 +438,10 @@ func (_c *MockMutableMessage_VChannel_Call) RunAndReturn(run func() string) *Moc func (_m *MockMutableMessage) Version() message.Version { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Version") + } + var r0 message.Version if rf, ok := ret.Get(0).(func() message.Version); ok { r0 = rf() @@ -443,6 +483,10 @@ func (_c *MockMutableMessage_Version_Call) RunAndReturn(run func() message.Versi func (_m *MockMutableMessage) WithBarrierTimeTick(tt uint64) message.MutableMessage { ret := _m.Called(tt) + if len(ret) == 0 { + panic("no return value specified for WithBarrierTimeTick") + } + var r0 message.MutableMessage if rf, ok := ret.Get(0).(func(uint64) message.MutableMessage); ok { r0 = rf(tt) @@ -487,6 +531,10 @@ func (_c *MockMutableMessage_WithBarrierTimeTick_Call) RunAndReturn(run func(uin func (_m *MockMutableMessage) WithLastConfirmed(id message.MessageID) message.MutableMessage { ret := _m.Called(id) + if len(ret) == 0 { + panic("no return value specified for WithLastConfirmed") + } + var r0 message.MutableMessage if rf, ok := ret.Get(0).(func(message.MessageID) message.MutableMessage); ok { r0 = rf(id) @@ -531,6 +579,10 @@ func (_c *MockMutableMessage_WithLastConfirmed_Call) RunAndReturn(run func(messa func (_m *MockMutableMessage) WithLastConfirmedUseMessageID() message.MutableMessage { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for WithLastConfirmedUseMessageID") + } + var r0 message.MutableMessage if rf, ok := ret.Get(0).(func() message.MutableMessage); ok { r0 = rf() @@ -574,6 +626,10 @@ func (_c *MockMutableMessage_WithLastConfirmedUseMessageID_Call) RunAndReturn(ru func (_m *MockMutableMessage) WithTimeTick(tt uint64) message.MutableMessage { ret := _m.Called(tt) + if len(ret) == 0 { + panic("no return value specified for WithTimeTick") + } + var r0 message.MutableMessage if rf, ok := ret.Get(0).(func(uint64) message.MutableMessage); ok { r0 = rf(tt) @@ -618,6 +674,10 @@ func (_c *MockMutableMessage_WithTimeTick_Call) RunAndReturn(run func(uint64) me func (_m *MockMutableMessage) WithTxnContext(txnCtx message.TxnContext) message.MutableMessage { ret := _m.Called(txnCtx) + if len(ret) == 0 { + panic("no return value specified for WithTxnContext") + } + var r0 message.MutableMessage if rf, ok := ret.Get(0).(func(message.TxnContext) message.MutableMessage); ok { r0 = rf(txnCtx) @@ -662,6 +722,10 @@ func (_c *MockMutableMessage_WithTxnContext_Call) RunAndReturn(run func(message. func (_m *MockMutableMessage) WithWALTerm(term int64) message.MutableMessage { ret := _m.Called(term) + if len(ret) == 0 { + panic("no return value specified for WithWALTerm") + } + var r0 message.MutableMessage if rf, ok := ret.Get(0).(func(int64) message.MutableMessage); ok { r0 = rf(term) diff --git a/pkg/mocks/streaming/util/mock_message/mock_RProperties.go b/pkg/mocks/streaming/util/mock_message/mock_RProperties.go index 5df87240b1..21bd3b14e2 100644 --- a/pkg/mocks/streaming/util/mock_message/mock_RProperties.go +++ b/pkg/mocks/streaming/util/mock_message/mock_RProperties.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_message @@ -21,6 +21,10 @@ func (_m *MockRProperties) EXPECT() *MockRProperties_Expecter { func (_m *MockRProperties) Exist(key string) bool { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Exist") + } + var r0 bool if rf, ok := ret.Get(0).(func(string) bool); ok { r0 = rf(key) @@ -63,6 +67,10 @@ func (_c *MockRProperties_Exist_Call) RunAndReturn(run func(string) bool) *MockR func (_m *MockRProperties) Get(key string) (string, bool) { ret := _m.Called(key) + if len(ret) == 0 { + panic("no return value specified for Get") + } + var r0 string var r1 bool if rf, ok := ret.Get(0).(func(string) (string, bool)); ok { @@ -115,6 +123,10 @@ func (_c *MockRProperties_Get_Call) RunAndReturn(run func(string) (string, bool) func (_m *MockRProperties) ToRawMap() map[string]string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for ToRawMap") + } + var r0 map[string]string if rf, ok := ret.Get(0).(func() map[string]string); ok { r0 = rf() diff --git a/pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go b/pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go index 849c0c31b0..69f69706ac 100644 --- a/pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go +++ b/pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_types @@ -26,6 +26,10 @@ func (_m *MockAssignmentDiscoverWatcher) EXPECT() *MockAssignmentDiscoverWatcher func (_m *MockAssignmentDiscoverWatcher) AssignmentDiscover(ctx context.Context, cb func(*types.VersionedStreamingNodeAssignments) error) error { ret := _m.Called(ctx, cb) + if len(ret) == 0 { + panic("no return value specified for AssignmentDiscover") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, func(*types.VersionedStreamingNodeAssignments) error) error); ok { r0 = rf(ctx, cb) @@ -69,6 +73,10 @@ func (_c *MockAssignmentDiscoverWatcher_AssignmentDiscover_Call) RunAndReturn(ru func (_m *MockAssignmentDiscoverWatcher) ReportAssignmentError(ctx context.Context, pchannel types.PChannelInfo, err error) error { ret := _m.Called(ctx, pchannel, err) + if len(ret) == 0 { + panic("no return value specified for ReportAssignmentError") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfo, error) error); ok { r0 = rf(ctx, pchannel, err) diff --git a/pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go b/pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go index d5fbed96a8..dfb71a4995 100644 --- a/pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go +++ b/pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package mock_types @@ -26,6 +26,10 @@ func (_m *MockAssignmentRebalanceTrigger) EXPECT() *MockAssignmentRebalanceTrigg func (_m *MockAssignmentRebalanceTrigger) ReportAssignmentError(ctx context.Context, pchannel types.PChannelInfo, err error) error { ret := _m.Called(ctx, pchannel, err) + if len(ret) == 0 { + panic("no return value specified for ReportAssignmentError") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, types.PChannelInfo, error) error); ok { r0 = rf(ctx, pchannel, err) diff --git a/pkg/mq/common/mock_id.go b/pkg/mq/common/mock_id.go index 0b642448a6..6041d5ee52 100644 --- a/pkg/mq/common/mock_id.go +++ b/pkg/mq/common/mock_id.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package common @@ -21,6 +21,10 @@ func (_m *MockMessageID) EXPECT() *MockMessageID_Expecter { func (_m *MockMessageID) AtEarliestPosition() bool { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for AtEarliestPosition") + } + var r0 bool if rf, ok := ret.Get(0).(func() bool); ok { r0 = rf() @@ -62,6 +66,10 @@ func (_c *MockMessageID_AtEarliestPosition_Call) RunAndReturn(run func() bool) * func (_m *MockMessageID) Equal(msgID []byte) (bool, error) { ret := _m.Called(msgID) + if len(ret) == 0 { + panic("no return value specified for Equal") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func([]byte) (bool, error)); ok { @@ -114,6 +122,10 @@ func (_c *MockMessageID_Equal_Call) RunAndReturn(run func([]byte) (bool, error)) func (_m *MockMessageID) LessOrEqualThan(msgID []byte) (bool, error) { ret := _m.Called(msgID) + if len(ret) == 0 { + panic("no return value specified for LessOrEqualThan") + } + var r0 bool var r1 error if rf, ok := ret.Get(0).(func([]byte) (bool, error)); ok { @@ -166,6 +178,10 @@ func (_c *MockMessageID_LessOrEqualThan_Call) RunAndReturn(run func([]byte) (boo func (_m *MockMessageID) Serialize() []byte { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Serialize") + } + var r0 []byte if rf, ok := ret.Get(0).(func() []byte); ok { r0 = rf() diff --git a/pkg/mq/msgdispatcher/mock_client.go b/pkg/mq/msgdispatcher/mock_client.go index 883a16de98..bc279459d2 100644 --- a/pkg/mq/msgdispatcher/mock_client.go +++ b/pkg/mq/msgdispatcher/mock_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package msgdispatcher @@ -96,6 +96,10 @@ func (_c *MockClient_Deregister_Call) RunAndReturn(run func(string)) *MockClient func (_m *MockClient) Register(ctx context.Context, vchannel string, pos *msgpb.MsgPosition, subPos common.SubscriptionInitialPosition) (<-chan *msgstream.MsgPack, error) { ret := _m.Called(ctx, vchannel, pos, subPos) + if len(ret) == 0 { + panic("no return value specified for Register") + } + var r0 <-chan *msgstream.MsgPack var r1 error if rf, ok := ret.Get(0).(func(context.Context, string, *msgpb.MsgPosition, common.SubscriptionInitialPosition) (<-chan *msgstream.MsgPack, error)); ok { diff --git a/pkg/mq/msgstream/mock_msgstream.go b/pkg/mq/msgstream/mock_msgstream.go index 84fb325260..50b403d939 100644 --- a/pkg/mq/msgstream/mock_msgstream.go +++ b/pkg/mq/msgstream/mock_msgstream.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package msgstream @@ -29,6 +29,10 @@ func (_m *MockMsgStream) EXPECT() *MockMsgStream_Expecter { func (_m *MockMsgStream) AsConsumer(ctx context.Context, channels []string, subName string, position common.SubscriptionInitialPosition) error { ret := _m.Called(ctx, channels, subName, position) + if len(ret) == 0 { + panic("no return value specified for AsConsumer") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []string, string, common.SubscriptionInitialPosition) error); ok { r0 = rf(ctx, channels, subName, position) @@ -107,6 +111,10 @@ func (_c *MockMsgStream_AsProducer_Call) RunAndReturn(run func([]string)) *MockM func (_m *MockMsgStream) Broadcast(_a0 *MsgPack) (map[string][]common.MessageID, error) { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Broadcast") + } + var r0 map[string][]common.MessageID var r1 error if rf, ok := ret.Get(0).(func(*MsgPack) (map[string][]common.MessageID, error)); ok { @@ -161,6 +169,10 @@ func (_c *MockMsgStream_Broadcast_Call) RunAndReturn(run func(*MsgPack) (map[str func (_m *MockMsgStream) Chan() <-chan *MsgPack { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for Chan") + } + var r0 <-chan *MsgPack if rf, ok := ret.Get(0).(func() <-chan *MsgPack); ok { r0 = rf() @@ -204,6 +216,10 @@ func (_c *MockMsgStream_Chan_Call) RunAndReturn(run func() <-chan *MsgPack) *Moc func (_m *MockMsgStream) CheckTopicValid(channel string) error { ret := _m.Called(channel) + if len(ret) == 0 { + panic("no return value specified for CheckTopicValid") + } + var r0 error if rf, ok := ret.Get(0).(func(string) error); ok { r0 = rf(channel) @@ -311,6 +327,10 @@ func (_c *MockMsgStream_EnableProduce_Call) RunAndReturn(run func(bool)) *MockMs func (_m *MockMsgStream) GetLatestMsgID(channel string) (common.MessageID, error) { ret := _m.Called(channel) + if len(ret) == 0 { + panic("no return value specified for GetLatestMsgID") + } + var r0 common.MessageID var r1 error if rf, ok := ret.Get(0).(func(string) (common.MessageID, error)); ok { @@ -365,6 +385,10 @@ func (_c *MockMsgStream_GetLatestMsgID_Call) RunAndReturn(run func(string) (comm func (_m *MockMsgStream) GetProduceChannels() []string { ret := _m.Called() + if len(ret) == 0 { + panic("no return value specified for GetProduceChannels") + } + var r0 []string if rf, ok := ret.Get(0).(func() []string); ok { r0 = rf() @@ -408,6 +432,10 @@ func (_c *MockMsgStream_GetProduceChannels_Call) RunAndReturn(run func() []strin func (_m *MockMsgStream) Produce(_a0 *MsgPack) error { ret := _m.Called(_a0) + if len(ret) == 0 { + panic("no return value specified for Produce") + } + var r0 error if rf, ok := ret.Get(0).(func(*MsgPack) error); ok { r0 = rf(_a0) @@ -450,6 +478,10 @@ func (_c *MockMsgStream_Produce_Call) RunAndReturn(run func(*MsgPack) error) *Mo func (_m *MockMsgStream) Seek(ctx context.Context, msgPositions []*msgpb.MsgPosition, includeCurrentMsg bool) error { ret := _m.Called(ctx, msgPositions, includeCurrentMsg) + if len(ret) == 0 { + panic("no return value specified for Seek") + } + var r0 error if rf, ok := ret.Get(0).(func(context.Context, []*msgpb.MsgPosition, bool) error); ok { r0 = rf(ctx, msgPositions, includeCurrentMsg) diff --git a/pkg/mq/msgstream/mock_msgstream_factory.go b/pkg/mq/msgstream/mock_msgstream_factory.go index 6f5f4e7b86..fe6c8a6587 100644 --- a/pkg/mq/msgstream/mock_msgstream_factory.go +++ b/pkg/mq/msgstream/mock_msgstream_factory.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.32.4. DO NOT EDIT. +// Code generated by mockery v2.46.0. DO NOT EDIT. package msgstream @@ -25,6 +25,10 @@ func (_m *MockFactory) EXPECT() *MockFactory_Expecter { func (_m *MockFactory) NewMsgStream(ctx context.Context) (MsgStream, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for NewMsgStream") + } + var r0 MsgStream var r1 error if rf, ok := ret.Get(0).(func(context.Context) (MsgStream, error)); ok { @@ -79,6 +83,10 @@ func (_c *MockFactory_NewMsgStream_Call) RunAndReturn(run func(context.Context) func (_m *MockFactory) NewMsgStreamDisposer(ctx context.Context) func([]string, string) error { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for NewMsgStreamDisposer") + } + var r0 func([]string, string) error if rf, ok := ret.Get(0).(func(context.Context) func([]string, string) error); ok { r0 = rf(ctx) @@ -123,6 +131,10 @@ func (_c *MockFactory_NewMsgStreamDisposer_Call) RunAndReturn(run func(context.C func (_m *MockFactory) NewTtMsgStream(ctx context.Context) (MsgStream, error) { ret := _m.Called(ctx) + if len(ret) == 0 { + panic("no return value specified for NewTtMsgStream") + } + var r0 MsgStream var r1 error if rf, ok := ret.Get(0).(func(context.Context) (MsgStream, error)); ok {