From 6e18ededab94a768e87a8cd48743fe6511615a7c Mon Sep 17 00:00:00 2001 From: SimFG Date: Fri, 25 Apr 2025 14:40:40 +0800 Subject: [PATCH] fix: [2.5] mockery too unavailable after upgrade golang version (#41522) - issue: ##41291 - pr: #41481 Signed-off-by: SimFG --- Makefile | 2 +- .../datacoord/allocator/mock_allocator.go | 2 +- internal/datacoord/broker/mock_broker.go | 2 +- internal/datacoord/mock_channel_manager.go | 8 +-- internal/datacoord/mock_channel_store.go | 18 +++--- internal/datacoord/mock_cluster.go | 10 +-- internal/datacoord/mock_compaction_meta.go | 14 ++--- .../datacoord/mock_compaction_plan_context.go | 16 ++--- internal/datacoord/mock_handler.go | 2 +- .../mock_index_engine_version_manager.go | 20 +++--- internal/datacoord/mock_segment_manager.go | 10 +-- internal/datacoord/mock_stats_job_manager.go | 10 +-- internal/datacoord/mock_sub_cluster.go | 2 +- internal/datacoord/mock_trigger.go | 10 +-- internal/datacoord/mock_trigger_manager.go | 12 ++-- .../session/mock_data_node_manager.go | 18 +++--- .../datacoord/session/mock_worker_manager.go | 12 ++-- internal/kv/mocks/meta_kv.go | 6 +- internal/kv/mocks/snapshot_kv.go | 2 +- internal/kv/mocks/txn_kv.go | 6 +- internal/kv/mocks/watch_kv.go | 6 +- .../mock_streaming/mock_WALAccesser.go | 4 +- .../grpc/mock_balancer/mock_SubConn.go | 12 ++-- .../grpc/mock_resolver/mock_ClientConn.go | 6 +- .../mock_grpc/mock_ClientStream.go | 10 +-- internal/mocks/mock_datacoord.go | 22 +++---- internal/mocks/mock_datacoord_client.go | 4 +- internal/mocks/mock_datanode.go | 22 +++---- internal/mocks/mock_datanode_client.go | 4 +- internal/mocks/mock_indexnode.go | 18 +++--- internal/mocks/mock_indexnode_client.go | 4 +- .../mock_StreamingCoordCataLog.go | 2 +- .../mock_StreamingNodeCataLog.go | 2 +- internal/mocks/mock_proxy.go | 28 ++++----- internal/mocks/mock_proxy_client.go | 4 +- internal/mocks/mock_querycoord.go | 20 +++--- internal/mocks/mock_querycoord_client.go | 4 +- internal/mocks/mock_querynode.go | 20 +++--- internal/mocks/mock_querynode_client.go | 4 +- internal/mocks/mock_rootcoord.go | 26 ++++---- internal/mocks/mock_rootcoord_client.go | 4 +- .../mock_client/mock_BroadcastService.go | 6 +- .../streamingcoord/mock_client/mock_Client.go | 10 +-- .../server/mock_balancer/mock_Balancer.go | 6 +- .../mock_broadcaster/mock_AppendOperator.go | 16 ++--- .../handler/mock_assignment/mock_Watcher.go | 6 +- .../handler/mock_consumer/mock_Consumer.go | 10 +-- .../handler/mock_producer/mock_Producer.go | 12 ++-- .../client/mock_handler/mock_HandlerClient.go | 6 +- .../client/mock_manager/mock_ManagerClient.go | 6 +- .../server/mock_flusher/mock_Flusher.go | 16 ++--- .../server/mock_wal/mock_Opener.go | 6 +- .../server/mock_wal/mock_OpenerBuilder.go | 6 +- .../server/mock_wal/mock_Scanner.go | 12 ++-- .../streamingnode/server/mock_wal/mock_WAL.go | 16 ++--- .../server/mock_walmanager/mock_Manager.go | 8 +-- .../mock_inspector/mock_SealOperator.go | 12 ++-- .../mock_TimeTickSyncOperator.go | 8 +-- .../wal/mock_interceptors/mock_Interceptor.go | 6 +- .../mock_InterceptorBuilder.go | 2 +- .../mock_InterceptorWithReady.go | 8 +-- .../mocks/util/mock_segcore/mock_CSegment.go | 14 ++--- .../mock_optimizers/mock_QueryHook.go | 2 +- .../mock_discoverer/mock_Discoverer.go | 4 +- .../service/mock_lazygrpc/mock_Service.go | 10 +-- .../service/mock_resolver/mock_Builder.go | 10 +-- .../service/mock_resolver/mock_Resolver.go | 4 +- internal/proxy/mock_cache.go | 18 +++--- internal/proxy/mock_channels_manager.go | 8 +-- internal/proxy/mock_lb_balancer.go | 14 ++--- internal/proxy/mock_lb_policy.go | 10 +-- internal/proxy/mock_shardclient_manager.go | 8 +-- internal/proxy/mock_tso_test.go | 2 +- .../querycoordv2/balance/mock_balancer.go | 2 +- internal/querycoordv2/dist/mock_controller.go | 12 ++-- internal/querycoordv2/meta/mock_broker.go | 2 +- .../querycoordv2/meta/mock_target_manager.go | 8 +-- internal/querycoordv2/session/mock_cluster.go | 10 +-- internal/querycoordv2/task/mock_scheduler.go | 36 +++++------ internal/querynodev2/cluster/mock_manager.go | 2 +- internal/querynodev2/cluster/mock_worker.go | 8 +-- .../querynodev2/delegator/mock_delegator.go | 38 ++++++------ .../segments/mock_collection_manager.go | 6 +- internal/querynodev2/segments/mock_loader.go | 26 ++++---- internal/querynodev2/segments/mock_segment.go | 62 +++++++++---------- .../segments/mock_segment_manager.go | 62 +++++++++---------- internal/rootcoord/mocks/garbage_collector.go | 10 +-- internal/rootcoord/mocks/meta_table.go | 2 +- pkg/eventlog/mock_logger.go | 8 +-- pkg/kv/predicates/mock_predicate.go | 10 +-- pkg/mocks/mock_kv/mock_MetaKv.go | 6 +- ...k_StreamingCoordAssignmentServiceClient.go | 2 +- ...ignmentService_AssignmentDiscoverClient.go | 12 ++-- ...ignmentService_AssignmentDiscoverServer.go | 8 +-- ...ck_StreamingCoordBroadcastServiceClient.go | 2 +- .../mock_StreamingNodeHandlerServiceClient.go | 2 +- ...reamingNodeHandlerService_ConsumeClient.go | 12 ++-- ...reamingNodeHandlerService_ConsumeServer.go | 8 +-- ...reamingNodeHandlerService_ProduceClient.go | 12 ++-- ...reamingNodeHandlerService_ProduceServer.go | 8 +-- .../mock_StreamingNodeManagerServiceClient.go | 2 +- .../mock_walimpls/mock_OpenerBuilderImpls.go | 6 +- .../mock_walimpls/mock_OpenerImpls.go | 6 +- .../mock_walimpls/mock_ScannerImpls.go | 12 ++-- .../streaming/mock_walimpls/mock_WALImpls.go | 10 +-- .../mock_message/mock_ImmutableMessage.go | 28 ++++----- .../mock_message/mock_ImmutableTxnMessage.go | 34 +++++----- .../util/mock_message/mock_MessageID.go | 8 +-- .../util/mock_message/mock_MutableMessage.go | 24 +++---- .../util/mock_message/mock_RProperties.go | 4 +- .../mock_AssignmentDiscoverWatcher.go | 2 +- .../mock_AssignmentRebalanceTrigger.go | 2 +- pkg/mq/common/mock_id.go | 6 +- pkg/mq/msgdispatcher/mock_client.go | 8 +-- pkg/mq/msgstream/mock_msgstream.go | 16 ++--- pkg/mq/msgstream/mock_msgstream_factory.go | 2 +- 116 files changed, 619 insertions(+), 617 deletions(-) diff --git a/Makefile b/Makefile index da8b767c53..d152cf72af 100644 --- a/Makefile +++ b/Makefile @@ -54,7 +54,7 @@ GOLANGCI_LINT_VERSION := 1.64.7 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.46.0 +MOCKERY_VERSION := 2.53.3 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 ab76ebc76e..93c107575a 100644 --- a/internal/datacoord/allocator/mock_allocator.go +++ b/internal/datacoord/allocator/mock_allocator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package allocator diff --git a/internal/datacoord/broker/mock_broker.go b/internal/datacoord/broker/mock_broker.go index bf73466d52..7e5c47de00 100644 --- a/internal/datacoord/broker/mock_broker.go +++ b/internal/datacoord/broker/mock_broker.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package broker diff --git a/internal/datacoord/mock_channel_manager.go b/internal/datacoord/mock_channel_manager.go index 2586c1e196..c609344048 100644 --- a/internal/datacoord/mock_channel_manager.go +++ b/internal/datacoord/mock_channel_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -68,7 +68,7 @@ func (_c *MockChannelManager_AddNode_Call) RunAndReturn(run func(int64) error) * return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockChannelManager) Close() { _m.Called() } @@ -96,7 +96,7 @@ func (_c *MockChannelManager_Close_Call) Return() *MockChannelManager_Close_Call } func (_c *MockChannelManager_Close_Call) RunAndReturn(run func()) *MockChannelManager_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -309,7 +309,7 @@ func (_c *MockChannelManager_GetChannelNamesByCollectionID_Call) RunAndReturn(ru return _c } -// GetChannelWatchInfos provides a mock function with given fields: +// GetChannelWatchInfos provides a mock function with no fields func (_m *MockChannelManager) GetChannelWatchInfos() map[int64]map[string]*datapb.ChannelWatchInfo { ret := _m.Called() diff --git a/internal/datacoord/mock_channel_store.go b/internal/datacoord/mock_channel_store.go index 86eac95d42..acda3845eb 100644 --- a/internal/datacoord/mock_channel_store.go +++ b/internal/datacoord/mock_channel_store.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -46,11 +46,11 @@ func (_c *MockRWChannelStore_AddNode_Call) Return() *MockRWChannelStore_AddNode_ } func (_c *MockRWChannelStore_AddNode_Call) RunAndReturn(run func(int64)) *MockRWChannelStore_AddNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// GetBufferChannelInfo provides a mock function with given fields: +// GetBufferChannelInfo provides a mock function with no fields func (_m *MockRWChannelStore) GetBufferChannelInfo() *NodeChannelInfo { ret := _m.Called() @@ -256,7 +256,7 @@ func (_c *MockRWChannelStore_GetNodeChannelsByCollectionID_Call) RunAndReturn(ru return _c } -// GetNodes provides a mock function with given fields: +// GetNodes provides a mock function with no fields func (_m *MockRWChannelStore) GetNodes() []int64 { ret := _m.Called() @@ -303,7 +303,7 @@ func (_c *MockRWChannelStore_GetNodes_Call) RunAndReturn(run func() []int64) *Mo return _c } -// GetNodesChannels provides a mock function with given fields: +// GetNodesChannels provides a mock function with no fields func (_m *MockRWChannelStore) GetNodesChannels() []*NodeChannelInfo { ret := _m.Called() @@ -396,7 +396,7 @@ func (_c *MockRWChannelStore_HasChannel_Call) RunAndReturn(run func(string) bool return _c } -// Reload provides a mock function with given fields: +// Reload provides a mock function with no fields func (_m *MockRWChannelStore) Reload() error { ret := _m.Called() @@ -470,7 +470,7 @@ func (_c *MockRWChannelStore_RemoveNode_Call) Return() *MockRWChannelStore_Remov } func (_c *MockRWChannelStore_RemoveNode_Call) RunAndReturn(run func(int64)) *MockRWChannelStore_RemoveNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -516,7 +516,7 @@ func (_c *MockRWChannelStore_SetLegacyChannelByNode_Call) Return() *MockRWChanne } func (_c *MockRWChannelStore_SetLegacyChannelByNode_Call) RunAndReturn(run func(...int64)) *MockRWChannelStore_SetLegacyChannelByNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -598,7 +598,7 @@ func (_c *MockRWChannelStore_UpdateState_Call) Return() *MockRWChannelStore_Upda } func (_c *MockRWChannelStore_UpdateState_Call) RunAndReturn(run func(bool, int64, RWChannel, int64)) *MockRWChannelStore_UpdateState_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_cluster.go b/internal/datacoord/mock_cluster.go index eb41111ed1..87581eea50 100644 --- a/internal/datacoord/mock_cluster.go +++ b/internal/datacoord/mock_cluster.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -24,7 +24,7 @@ func (_m *MockCluster) EXPECT() *MockCluster_Expecter { return &MockCluster_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockCluster) Close() { _m.Called() } @@ -52,7 +52,7 @@ func (_c *MockCluster_Close_Call) Return() *MockCluster_Close_Call { } func (_c *MockCluster_Close_Call) RunAndReturn(run func()) *MockCluster_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -201,7 +201,7 @@ func (_c *MockCluster_FlushChannels_Call) RunAndReturn(run func(context.Context, return _c } -// GetSessions provides a mock function with given fields: +// GetSessions provides a mock function with no fields func (_m *MockCluster) GetSessions() []*session.Session { ret := _m.Called() @@ -460,7 +460,7 @@ func (_c *MockCluster_QueryPreImport_Call) RunAndReturn(run func(int64, *datapb. return _c } -// QuerySlots provides a mock function with given fields: +// QuerySlots provides a mock function with no fields func (_m *MockCluster) QuerySlots() map[int64]int64 { ret := _m.Called() diff --git a/internal/datacoord/mock_compaction_meta.go b/internal/datacoord/mock_compaction_meta.go index 5ba535ab70..ec4cd2a389 100644 --- a/internal/datacoord/mock_compaction_meta.go +++ b/internal/datacoord/mock_compaction_meta.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -299,7 +299,7 @@ func (_c *MockCompactionMeta_DropCompactionTask_Call) RunAndReturn(run func(cont return _c } -// GetAnalyzeMeta provides a mock function with given fields: +// GetAnalyzeMeta provides a mock function with no fields func (_m *MockCompactionMeta) GetAnalyzeMeta() *analyzeMeta { ret := _m.Called() @@ -346,7 +346,7 @@ func (_c *MockCompactionMeta_GetAnalyzeMeta_Call) RunAndReturn(run func() *analy return _c } -// GetCompactionTaskMeta provides a mock function with given fields: +// GetCompactionTaskMeta provides a mock function with no fields func (_m *MockCompactionMeta) GetCompactionTaskMeta() *compactionTaskMeta { ret := _m.Called() @@ -539,7 +539,7 @@ func (_c *MockCompactionMeta_GetHealthySegment_Call) RunAndReturn(run func(conte return _c } -// GetIndexMeta provides a mock function with given fields: +// GetIndexMeta provides a mock function with no fields func (_m *MockCompactionMeta) GetIndexMeta() *indexMeta { ret := _m.Called() @@ -586,7 +586,7 @@ func (_c *MockCompactionMeta_GetIndexMeta_Call) RunAndReturn(run func() *indexMe return _c } -// GetPartitionStatsMeta provides a mock function with given fields: +// GetPartitionStatsMeta provides a mock function with no fields func (_m *MockCompactionMeta) GetPartitionStatsMeta() *partitionStatsMeta { ret := _m.Called() @@ -870,7 +870,7 @@ func (_c *MockCompactionMeta_SetSegmentStating_Call) Return() *MockCompactionMet } func (_c *MockCompactionMeta_SetSegmentStating_Call) RunAndReturn(run func(int64, bool)) *MockCompactionMeta_SetSegmentStating_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -905,7 +905,7 @@ func (_c *MockCompactionMeta_SetSegmentsCompacting_Call) Return() *MockCompactio } func (_c *MockCompactionMeta_SetSegmentsCompacting_Call) RunAndReturn(run func(context.Context, []int64, bool)) *MockCompactionMeta_SetSegmentsCompacting_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_compaction_plan_context.go b/internal/datacoord/mock_compaction_plan_context.go index 586b31854e..83096cb4be 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -210,7 +210,7 @@ func (_c *MockCompactionPlanContext_getCompactionTasksNumBySignalID_Call) RunAnd return _c } -// isFull provides a mock function with given fields: +// isFull provides a mock function with no fields func (_m *MockCompactionPlanContext) isFull() bool { ret := _m.Called() @@ -284,7 +284,7 @@ func (_c *MockCompactionPlanContext_removeTasksByChannel_Call) Return() *MockCom } func (_c *MockCompactionPlanContext_removeTasksByChannel_Call) RunAndReturn(run func(string)) *MockCompactionPlanContext_removeTasksByChannel_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -317,11 +317,11 @@ func (_c *MockCompactionPlanContext_setTaskScheduler_Call) Return() *MockCompact } func (_c *MockCompactionPlanContext_setTaskScheduler_Call) RunAndReturn(run func(*taskScheduler)) *MockCompactionPlanContext_setTaskScheduler_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// start provides a mock function with given fields: +// start provides a mock function with no fields func (_m *MockCompactionPlanContext) start() { _m.Called() } @@ -349,11 +349,11 @@ func (_c *MockCompactionPlanContext_start_Call) Return() *MockCompactionPlanCont } func (_c *MockCompactionPlanContext_start_Call) RunAndReturn(run func()) *MockCompactionPlanContext_start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// stop provides a mock function with given fields: +// stop provides a mock function with no fields func (_m *MockCompactionPlanContext) stop() { _m.Called() } @@ -381,7 +381,7 @@ func (_c *MockCompactionPlanContext_stop_Call) Return() *MockCompactionPlanConte } func (_c *MockCompactionPlanContext_stop_Call) RunAndReturn(run func()) *MockCompactionPlanContext_stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_handler.go b/internal/datacoord/mock_handler.go index cb5fe55c3d..9417a8b6ff 100644 --- a/internal/datacoord/mock_handler.go +++ b/internal/datacoord/mock_handler.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord diff --git a/internal/datacoord/mock_index_engine_version_manager.go b/internal/datacoord/mock_index_engine_version_manager.go index faaae2f382..275c4cca7d 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -49,11 +49,11 @@ func (_c *MockVersionManager_AddNode_Call) Return() *MockVersionManager_AddNode_ } func (_c *MockVersionManager_AddNode_Call) RunAndReturn(run func(*sessionutil.Session)) *MockVersionManager_AddNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// GetCurrentIndexEngineVersion provides a mock function with given fields: +// GetCurrentIndexEngineVersion provides a mock function with no fields func (_m *MockVersionManager) GetCurrentIndexEngineVersion() int32 { ret := _m.Called() @@ -98,7 +98,7 @@ func (_c *MockVersionManager_GetCurrentIndexEngineVersion_Call) RunAndReturn(run return _c } -// GetCurrentScalarIndexEngineVersion provides a mock function with given fields: +// GetCurrentScalarIndexEngineVersion provides a mock function with no fields func (_m *MockVersionManager) GetCurrentScalarIndexEngineVersion() int32 { ret := _m.Called() @@ -143,7 +143,7 @@ func (_c *MockVersionManager_GetCurrentScalarIndexEngineVersion_Call) RunAndRetu return _c } -// GetIndexNonEncoding provides a mock function with given fields: +// GetIndexNonEncoding provides a mock function with no fields func (_m *MockVersionManager) GetIndexNonEncoding() bool { ret := _m.Called() @@ -188,7 +188,7 @@ func (_c *MockVersionManager_GetIndexNonEncoding_Call) RunAndReturn(run func() b return _c } -// GetMinimalIndexEngineVersion provides a mock function with given fields: +// GetMinimalIndexEngineVersion provides a mock function with no fields func (_m *MockVersionManager) GetMinimalIndexEngineVersion() int32 { ret := _m.Called() @@ -233,7 +233,7 @@ func (_c *MockVersionManager_GetMinimalIndexEngineVersion_Call) RunAndReturn(run return _c } -// GetMinimalScalarIndexEngineVersion provides a mock function with given fields: +// GetMinimalScalarIndexEngineVersion provides a mock function with no fields func (_m *MockVersionManager) GetMinimalScalarIndexEngineVersion() int32 { ret := _m.Called() @@ -307,7 +307,7 @@ func (_c *MockVersionManager_RemoveNode_Call) Return() *MockVersionManager_Remov } func (_c *MockVersionManager_RemoveNode_Call) RunAndReturn(run func(*sessionutil.Session)) *MockVersionManager_RemoveNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -340,7 +340,7 @@ func (_c *MockVersionManager_Startup_Call) Return() *MockVersionManager_Startup_ } func (_c *MockVersionManager_Startup_Call) RunAndReturn(run func(map[string]*sessionutil.Session)) *MockVersionManager_Startup_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -373,7 +373,7 @@ func (_c *MockVersionManager_Update_Call) Return() *MockVersionManager_Update_Ca } func (_c *MockVersionManager_Update_Call) RunAndReturn(run func(*sessionutil.Session)) *MockVersionManager_Update_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_segment_manager.go b/internal/datacoord/mock_segment_manager.go index 3f8e2392b1..de9c14423d 100644 --- a/internal/datacoord/mock_segment_manager.go +++ b/internal/datacoord/mock_segment_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -176,7 +176,7 @@ func (_c *MockManager_CleanZeroSealedSegmentsOfChannel_Call) Return() *MockManag } func (_c *MockManager_CleanZeroSealedSegmentsOfChannel_Call) RunAndReturn(run func(context.Context, string, uint64)) *MockManager_CleanZeroSealedSegmentsOfChannel_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -211,7 +211,7 @@ func (_c *MockManager_DropSegment_Call) Return() *MockManager_DropSegment_Call { } func (_c *MockManager_DropSegment_Call) RunAndReturn(run func(context.Context, string, int64)) *MockManager_DropSegment_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -245,7 +245,7 @@ func (_c *MockManager_DropSegmentsOfChannel_Call) Return() *MockManager_DropSegm } func (_c *MockManager_DropSegmentsOfChannel_Call) RunAndReturn(run func(context.Context, string)) *MockManager_DropSegmentsOfChannel_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -280,7 +280,7 @@ func (_c *MockManager_ExpireAllocations_Call) Return() *MockManager_ExpireAlloca } func (_c *MockManager_ExpireAllocations_Call) RunAndReturn(run func(context.Context, string, uint64)) *MockManager_ExpireAllocations_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_stats_job_manager.go b/internal/datacoord/mock_stats_job_manager.go index 19a1013028..e180728a29 100644 --- a/internal/datacoord/mock_stats_job_manager.go +++ b/internal/datacoord/mock_stats_job_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -116,7 +116,7 @@ func (_c *MockStatsJobManager_GetStatsTask_Call) RunAndReturn(run func(int64, in return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockStatsJobManager) Start() { _m.Called() } @@ -144,11 +144,11 @@ func (_c *MockStatsJobManager_Start_Call) Return() *MockStatsJobManager_Start_Ca } func (_c *MockStatsJobManager_Start_Call) RunAndReturn(run func()) *MockStatsJobManager_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockStatsJobManager) Stop() { _m.Called() } @@ -176,7 +176,7 @@ func (_c *MockStatsJobManager_Stop_Call) Return() *MockStatsJobManager_Stop_Call } func (_c *MockStatsJobManager_Stop_Call) RunAndReturn(run func()) *MockStatsJobManager_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_sub_cluster.go b/internal/datacoord/mock_sub_cluster.go index 7ae9424625..f80abf7cf0 100644 --- a/internal/datacoord/mock_sub_cluster.go +++ b/internal/datacoord/mock_sub_cluster.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord diff --git a/internal/datacoord/mock_trigger.go b/internal/datacoord/mock_trigger.go index 0a042529b7..b609ff2487 100644 --- a/internal/datacoord/mock_trigger.go +++ b/internal/datacoord/mock_trigger.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -78,7 +78,7 @@ func (_c *MockTrigger_TriggerCompaction_Call) RunAndReturn(run func(context.Cont return _c } -// start provides a mock function with given fields: +// start provides a mock function with no fields func (_m *MockTrigger) start() { _m.Called() } @@ -106,11 +106,11 @@ func (_c *MockTrigger_start_Call) Return() *MockTrigger_start_Call { } func (_c *MockTrigger_start_Call) RunAndReturn(run func()) *MockTrigger_start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// stop provides a mock function with given fields: +// stop provides a mock function with no fields func (_m *MockTrigger) stop() { _m.Called() } @@ -138,7 +138,7 @@ func (_c *MockTrigger_stop_Call) Return() *MockTrigger_stop_Call { } func (_c *MockTrigger_stop_Call) RunAndReturn(run func()) *MockTrigger_stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/mock_trigger_manager.go b/internal/datacoord/mock_trigger_manager.go index fdb093a8e4..cdec8c5d0b 100644 --- a/internal/datacoord/mock_trigger_manager.go +++ b/internal/datacoord/mock_trigger_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package datacoord @@ -108,11 +108,11 @@ func (_c *MockTriggerManager_OnCollectionUpdate_Call) Return() *MockTriggerManag } func (_c *MockTriggerManager_OnCollectionUpdate_Call) RunAndReturn(run func(int64)) *MockTriggerManager_OnCollectionUpdate_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockTriggerManager) Start() { _m.Called() } @@ -140,11 +140,11 @@ func (_c *MockTriggerManager_Start_Call) Return() *MockTriggerManager_Start_Call } func (_c *MockTriggerManager_Start_Call) RunAndReturn(run func()) *MockTriggerManager_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockTriggerManager) Stop() { _m.Called() } @@ -172,7 +172,7 @@ func (_c *MockTriggerManager_Stop_Call) Return() *MockTriggerManager_Stop_Call { } func (_c *MockTriggerManager_Stop_Call) RunAndReturn(run func()) *MockTriggerManager_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/datacoord/session/mock_data_node_manager.go b/internal/datacoord/session/mock_data_node_manager.go index c53fa7f770..bbf022badf 100644 --- a/internal/datacoord/session/mock_data_node_manager.go +++ b/internal/datacoord/session/mock_data_node_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package session @@ -53,7 +53,7 @@ func (_c *MockDataNodeManager_AddSession_Call) Return() *MockDataNodeManager_Add } func (_c *MockDataNodeManager_AddSession_Call) RunAndReturn(run func(*NodeInfo)) *MockDataNodeManager_AddSession_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -163,7 +163,7 @@ func (_c *MockDataNodeManager_CheckHealth_Call) RunAndReturn(run func(context.Co return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockDataNodeManager) Close() { _m.Called() } @@ -191,7 +191,7 @@ func (_c *MockDataNodeManager_Close_Call) Return() *MockDataNodeManager_Close_Ca } func (_c *MockDataNodeManager_Close_Call) RunAndReturn(run func()) *MockDataNodeManager_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -272,7 +272,7 @@ func (_c *MockDataNodeManager_DeleteSession_Call) Return() *MockDataNodeManager_ } func (_c *MockDataNodeManager_DeleteSession_Call) RunAndReturn(run func(*NodeInfo)) *MockDataNodeManager_DeleteSession_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -401,7 +401,7 @@ func (_c *MockDataNodeManager_Flush_Call) Return() *MockDataNodeManager_Flush_Ca } func (_c *MockDataNodeManager_Flush_Call) RunAndReturn(run func(context.Context, int64, *datapb.FlushSegmentsRequest)) *MockDataNodeManager_Flush_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -512,7 +512,7 @@ func (_c *MockDataNodeManager_GetCompactionPlanResult_Call) RunAndReturn(run fun return _c } -// GetCompactionPlansResults provides a mock function with given fields: +// GetCompactionPlansResults provides a mock function with no fields func (_m *MockDataNodeManager) GetCompactionPlansResults() (map[int64]*typeutil.Pair[int64, *datapb.CompactionPlanResult], error) { ret := _m.Called() @@ -627,7 +627,7 @@ func (_c *MockDataNodeManager_GetSession_Call) RunAndReturn(run func(int64) (*Se return _c } -// GetSessionIDs provides a mock function with given fields: +// GetSessionIDs provides a mock function with no fields func (_m *MockDataNodeManager) GetSessionIDs() []int64 { ret := _m.Called() @@ -674,7 +674,7 @@ func (_c *MockDataNodeManager_GetSessionIDs_Call) RunAndReturn(run func() []int6 return _c } -// GetSessions provides a mock function with given fields: +// GetSessions provides a mock function with no fields func (_m *MockDataNodeManager) GetSessions() []*Session { ret := _m.Called() diff --git a/internal/datacoord/session/mock_worker_manager.go b/internal/datacoord/session/mock_worker_manager.go index 3a88f6bd50..4190219e92 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package session @@ -67,7 +67,7 @@ func (_c *MockWorkerManager_AddNode_Call) RunAndReturn(run func(int64, string) e return _c } -// GetAllClients provides a mock function with given fields: +// GetAllClients provides a mock function with no fields func (_m *MockWorkerManager) GetAllClients() map[int64]types.IndexNodeClient { ret := _m.Called() @@ -172,7 +172,7 @@ func (_c *MockWorkerManager_GetClientByID_Call) RunAndReturn(run func(int64) (ty return _c } -// PickClient provides a mock function with given fields: +// PickClient provides a mock function with no fields func (_m *MockWorkerManager) PickClient() (int64, types.IndexNodeClient) { ret := _m.Called() @@ -229,7 +229,7 @@ func (_c *MockWorkerManager_PickClient_Call) RunAndReturn(run func() (int64, typ return _c } -// QuerySlots provides a mock function with given fields: +// QuerySlots provides a mock function with no fields func (_m *MockWorkerManager) QuerySlots() map[int64]*WorkerSlots { ret := _m.Called() @@ -305,7 +305,7 @@ func (_c *MockWorkerManager_RemoveNode_Call) Return() *MockWorkerManager_RemoveN } func (_c *MockWorkerManager_RemoveNode_Call) RunAndReturn(run func(int64)) *MockWorkerManager_RemoveNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -338,7 +338,7 @@ func (_c *MockWorkerManager_StoppingNode_Call) Return() *MockWorkerManager_Stopp } func (_c *MockWorkerManager_StoppingNode_Call) RunAndReturn(run func(int64)) *MockWorkerManager_StoppingNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/kv/mocks/meta_kv.go b/internal/kv/mocks/meta_kv.go index ece71aad3a..93df37e3d5 100644 --- a/internal/kv/mocks/meta_kv.go +++ b/internal/kv/mocks/meta_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -23,7 +23,7 @@ func (_m *MetaKv) EXPECT() *MetaKv_Expecter { return &MetaKv_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MetaKv) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *MetaKv_Close_Call) Return() *MetaKv_Close_Call { } func (_c *MetaKv_Close_Call) RunAndReturn(run func()) *MetaKv_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/kv/mocks/snapshot_kv.go b/internal/kv/mocks/snapshot_kv.go index 220385636b..6a084c91a5 100644 --- a/internal/kv/mocks/snapshot_kv.go +++ b/internal/kv/mocks/snapshot_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks diff --git a/internal/kv/mocks/txn_kv.go b/internal/kv/mocks/txn_kv.go index 9bc41776ff..93d7582411 100644 --- a/internal/kv/mocks/txn_kv.go +++ b/internal/kv/mocks/txn_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -23,7 +23,7 @@ func (_m *TxnKV) EXPECT() *TxnKV_Expecter { return &TxnKV_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *TxnKV) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *TxnKV_Close_Call) Return() *TxnKV_Close_Call { } func (_c *TxnKV_Close_Call) RunAndReturn(run func()) *TxnKV_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/kv/mocks/watch_kv.go b/internal/kv/mocks/watch_kv.go index 1685d18d93..6ef9a7d095 100644 --- a/internal/kv/mocks/watch_kv.go +++ b/internal/kv/mocks/watch_kv.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -25,7 +25,7 @@ func (_m *WatchKV) EXPECT() *WatchKV_Expecter { return &WatchKV_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *WatchKV) Close() { _m.Called() } @@ -53,7 +53,7 @@ func (_c *WatchKV_Close_Call) Return() *WatchKV_Close_Call { } func (_c *WatchKV_Close_Call) RunAndReturn(run func()) *WatchKV_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go b/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go index 20baa397e7..d787f74447 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streaming @@ -149,7 +149,7 @@ func (_c *MockWALAccesser_AppendMessagesWithOption_Call) RunAndReturn(run func(c return _c } -// Broadcast provides a mock function with given fields: +// Broadcast provides a mock function with no fields func (_m *MockWALAccesser) Broadcast() streaming.Broadcast { ret := _m.Called() 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 8ff1fb2d11..70217492e9 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_balancer @@ -22,7 +22,7 @@ func (_m *MockSubConn) EXPECT() *MockSubConn_Expecter { return &MockSubConn_Expecter{mock: &_m.Mock} } -// Connect provides a mock function with given fields: +// Connect provides a mock function with no fields func (_m *MockSubConn) Connect() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockSubConn_Connect_Call) Return() *MockSubConn_Connect_Call { } func (_c *MockSubConn_Connect_Call) RunAndReturn(run func()) *MockSubConn_Connect_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -114,7 +114,7 @@ func (_c *MockSubConn_GetOrBuildProducer_Call) RunAndReturn(run func(balancer.Pr return _c } -// Shutdown provides a mock function with given fields: +// Shutdown provides a mock function with no fields func (_m *MockSubConn) Shutdown() { _m.Called() } @@ -142,7 +142,7 @@ func (_c *MockSubConn_Shutdown_Call) Return() *MockSubConn_Shutdown_Call { } func (_c *MockSubConn_Shutdown_Call) RunAndReturn(run func()) *MockSubConn_Shutdown_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -175,7 +175,7 @@ func (_c *MockSubConn_UpdateAddresses_Call) Return() *MockSubConn_UpdateAddresse } func (_c *MockSubConn_UpdateAddresses_Call) RunAndReturn(run func([]resolver.Address)) *MockSubConn_UpdateAddresses_Call { - _c.Call.Return(run) + _c.Run(run) return _c } 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 dd32a2d265..ef733d7756 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_resolver @@ -51,7 +51,7 @@ func (_c *MockClientConn_NewAddress_Call) Return() *MockClientConn_NewAddress_Ca } func (_c *MockClientConn_NewAddress_Call) RunAndReturn(run func([]resolver.Address)) *MockClientConn_NewAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -132,7 +132,7 @@ func (_c *MockClientConn_ReportError_Call) Return() *MockClientConn_ReportError_ } func (_c *MockClientConn_ReportError_Call) RunAndReturn(run func(error)) *MockClientConn_ReportError_Call { - _c.Call.Return(run) + _c.Run(run) return _c } 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 64df110847..c98c9d4856 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_grpc @@ -23,7 +23,7 @@ func (_m *MockClientStream) EXPECT() *MockClientStream_Expecter { return &MockClientStream_Expecter{mock: &_m.Mock} } -// CloseSend provides a mock function with given fields: +// CloseSend provides a mock function with no fields func (_m *MockClientStream) CloseSend() error { ret := _m.Called() @@ -68,7 +68,7 @@ func (_c *MockClientStream_CloseSend_Call) RunAndReturn(run func() error) *MockC return _c } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockClientStream) Context() context.Context { ret := _m.Called() @@ -115,7 +115,7 @@ func (_c *MockClientStream_Context_Call) RunAndReturn(run func() context.Context return _c } -// Header provides a mock function with given fields: +// Header provides a mock function with no fields func (_m *MockClientStream) Header() (metadata.MD, error) { ret := _m.Called() @@ -264,7 +264,7 @@ func (_c *MockClientStream_SendMsg_Call) RunAndReturn(run func(interface{}) erro return _c } -// Trailer provides a mock function with given fields: +// Trailer provides a mock function with no fields func (_m *MockClientStream) Trailer() metadata.MD { ret := _m.Called() diff --git a/internal/mocks/mock_datacoord.go b/internal/mocks/mock_datacoord.go index fa06349688..1db4935d9f 100644 --- a/internal/mocks/mock_datacoord.go +++ b/internal/mocks/mock_datacoord.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -2278,7 +2278,7 @@ func (_c *MockDataCoord_ImportV2_Call) RunAndReturn(run func(context.Context, *i return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MockDataCoord) Init() error { ret := _m.Called() @@ -2559,7 +2559,7 @@ func (_c *MockDataCoord_MarkSegmentsDropped_Call) RunAndReturn(run func(context. return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *MockDataCoord) Register() error { ret := _m.Called() @@ -2751,7 +2751,7 @@ func (_c *MockDataCoord_SetAddress_Call) Return() *MockDataCoord_SetAddress_Call } func (_c *MockDataCoord_SetAddress_Call) RunAndReturn(run func(string)) *MockDataCoord_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2784,7 +2784,7 @@ func (_c *MockDataCoord_SetDataNodeCreator_Call) Return() *MockDataCoord_SetData } func (_c *MockDataCoord_SetDataNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.DataNodeClient, error))) *MockDataCoord_SetDataNodeCreator_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2817,7 +2817,7 @@ func (_c *MockDataCoord_SetEtcdClient_Call) Return() *MockDataCoord_SetEtcdClien } func (_c *MockDataCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockDataCoord_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2850,7 +2850,7 @@ func (_c *MockDataCoord_SetIndexNodeCreator_Call) Return() *MockDataCoord_SetInd } func (_c *MockDataCoord_SetIndexNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.IndexNodeClient, error))) *MockDataCoord_SetIndexNodeCreator_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2883,7 +2883,7 @@ func (_c *MockDataCoord_SetRootCoordClient_Call) Return() *MockDataCoord_SetRoot } func (_c *MockDataCoord_SetRootCoordClient_Call) RunAndReturn(run func(types.RootCoordClient)) *MockDataCoord_SetRootCoordClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2975,7 +2975,7 @@ func (_c *MockDataCoord_SetTiKVClient_Call) Return() *MockDataCoord_SetTiKVClien } func (_c *MockDataCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *MockDataCoord_SetTiKVClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -3038,7 +3038,7 @@ func (_c *MockDataCoord_ShowConfigurations_Call) RunAndReturn(run func(context.C return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockDataCoord) Start() error { ret := _m.Called() @@ -3083,7 +3083,7 @@ func (_c *MockDataCoord_Start_Call) RunAndReturn(run func() error) *MockDataCoor return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockDataCoord) Stop() error { ret := _m.Called() diff --git a/internal/mocks/mock_datacoord_client.go b/internal/mocks/mock_datacoord_client.go index 15197837a2..d45595e49e 100644 --- a/internal/mocks/mock_datacoord_client.go +++ b/internal/mocks/mock_datacoord_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -403,7 +403,7 @@ func (_c *MockDataCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockDataCoordClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_datanode.go b/internal/mocks/mock_datanode.go index 9c3d6aac90..a64667060d 100644 --- a/internal/mocks/mock_datanode.go +++ b/internal/mocks/mock_datanode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -386,7 +386,7 @@ func (_c *MockDataNode_FlushSegments_Call) RunAndReturn(run func(context.Context return _c } -// GetAddress provides a mock function with given fields: +// GetAddress provides a mock function with no fields func (_m *MockDataNode) GetAddress() string { ret := _m.Called() @@ -608,7 +608,7 @@ func (_c *MockDataNode_GetMetrics_Call) RunAndReturn(run func(context.Context, * return _c } -// GetNodeID provides a mock function with given fields: +// GetNodeID provides a mock function with no fields func (_m *MockDataNode) GetNodeID() int64 { ret := _m.Called() @@ -653,7 +653,7 @@ func (_c *MockDataNode_GetNodeID_Call) RunAndReturn(run func() int64) *MockDataN return _c } -// GetStateCode provides a mock function with given fields: +// GetStateCode provides a mock function with no fields func (_m *MockDataNode) GetStateCode() commonpb.StateCode { ret := _m.Called() @@ -816,7 +816,7 @@ func (_c *MockDataNode_ImportV2_Call) RunAndReturn(run func(context.Context, *da return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MockDataNode) Init() error { ret := _m.Called() @@ -1156,7 +1156,7 @@ func (_c *MockDataNode_QuerySlot_Call) RunAndReturn(run func(context.Context, *d return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *MockDataNode) Register() error { ret := _m.Called() @@ -1289,7 +1289,7 @@ func (_c *MockDataNode_SetAddress_Call) Return() *MockDataNode_SetAddress_Call { } func (_c *MockDataNode_SetAddress_Call) RunAndReturn(run func(string)) *MockDataNode_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1368,7 +1368,7 @@ func (_c *MockDataNode_SetEtcdClient_Call) Return() *MockDataNode_SetEtcdClient_ } func (_c *MockDataNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockDataNode_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1477,7 +1477,7 @@ func (_c *MockDataNode_ShowConfigurations_Call) RunAndReturn(run func(context.Co return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockDataNode) Start() error { ret := _m.Called() @@ -1522,7 +1522,7 @@ func (_c *MockDataNode_Start_Call) RunAndReturn(run func() error) *MockDataNode_ return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockDataNode) Stop() error { ret := _m.Called() @@ -1655,7 +1655,7 @@ func (_c *MockDataNode_UpdateStateCode_Call) Return() *MockDataNode_UpdateStateC } func (_c *MockDataNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockDataNode_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_datanode_client.go b/internal/mocks/mock_datanode_client.go index 948db62946..1e094df5df 100644 --- a/internal/mocks/mock_datanode_client.go +++ b/internal/mocks/mock_datanode_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -105,7 +105,7 @@ func (_c *MockDataNodeClient_CheckChannelOperationProgress_Call) RunAndReturn(ru return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockDataNodeClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_indexnode.go b/internal/mocks/mock_indexnode.go index 3f726a49c2..42b4c74ff8 100644 --- a/internal/mocks/mock_indexnode.go +++ b/internal/mocks/mock_indexnode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -266,7 +266,7 @@ func (_c *MockIndexNode_DropJobsV2_Call) RunAndReturn(run func(context.Context, return _c } -// GetAddress provides a mock function with given fields: +// GetAddress provides a mock function with no fields func (_m *MockIndexNode) GetAddress() string { ret := _m.Called() @@ -547,7 +547,7 @@ func (_c *MockIndexNode_GetStatisticsChannel_Call) RunAndReturn(run func(context return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MockIndexNode) Init() error { ret := _m.Called() @@ -710,7 +710,7 @@ func (_c *MockIndexNode_QueryJobsV2_Call) RunAndReturn(run func(context.Context, return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *MockIndexNode) Register() error { ret := _m.Called() @@ -784,7 +784,7 @@ func (_c *MockIndexNode_SetAddress_Call) Return() *MockIndexNode_SetAddress_Call } func (_c *MockIndexNode_SetAddress_Call) RunAndReturn(run func(string)) *MockIndexNode_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -817,7 +817,7 @@ func (_c *MockIndexNode_SetEtcdClient_Call) Return() *MockIndexNode_SetEtcdClien } func (_c *MockIndexNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockIndexNode_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -880,7 +880,7 @@ func (_c *MockIndexNode_ShowConfigurations_Call) RunAndReturn(run func(context.C return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockIndexNode) Start() error { ret := _m.Called() @@ -925,7 +925,7 @@ func (_c *MockIndexNode_Start_Call) RunAndReturn(run func() error) *MockIndexNod return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockIndexNode) Stop() error { ret := _m.Called() @@ -999,7 +999,7 @@ func (_c *MockIndexNode_UpdateStateCode_Call) Return() *MockIndexNode_UpdateStat } func (_c *MockIndexNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockIndexNode_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_indexnode_client.go b/internal/mocks/mock_indexnode_client.go index 5090a8944d..f1b3814759 100644 --- a/internal/mocks/mock_indexnode_client.go +++ b/internal/mocks/mock_indexnode_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -31,7 +31,7 @@ func (_m *MockIndexNodeClient) EXPECT() *MockIndexNodeClient_Expecter { return &MockIndexNodeClient_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockIndexNodeClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go b/internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go index 6321eacbd5..3cc96f9ea5 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_metastore diff --git a/internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go b/internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go index 18d3145d3c..42d84fa214 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_metastore diff --git a/internal/mocks/mock_proxy.go b/internal/mocks/mock_proxy.go index c753f6ec96..ce3feb012b 100644 --- a/internal/mocks/mock_proxy.go +++ b/internal/mocks/mock_proxy.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -2276,7 +2276,7 @@ func (_c *MockProxy_FlushAll_Call) RunAndReturn(run func(context.Context, *milvu return _c } -// GetAddress provides a mock function with given fields: +// GetAddress provides a mock function with no fields func (_m *MockProxy) GetAddress() string { ret := _m.Called() @@ -3442,7 +3442,7 @@ func (_c *MockProxy_GetQuerySegmentInfo_Call) RunAndReturn(run func(context.Cont return _c } -// GetRateLimiter provides a mock function with given fields: +// GetRateLimiter provides a mock function with no fields func (_m *MockProxy) GetRateLimiter() (types.Limiter, error) { ret := _m.Called() @@ -4030,7 +4030,7 @@ func (_c *MockProxy_ImportV2_Call) RunAndReturn(run func(context.Context, *inter return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MockProxy) Init() error { ret := _m.Called() @@ -5432,7 +5432,7 @@ func (_c *MockProxy_RefreshPolicyInfoCache_Call) RunAndReturn(run func(context.C return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *MockProxy) Register() error { ret := _m.Called() @@ -6155,7 +6155,7 @@ func (_c *MockProxy_SetAddress_Call) Return() *MockProxy_SetAddress_Call { } func (_c *MockProxy_SetAddress_Call) RunAndReturn(run func(string)) *MockProxy_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6188,7 +6188,7 @@ func (_c *MockProxy_SetDataCoordClient_Call) Return() *MockProxy_SetDataCoordCli } func (_c *MockProxy_SetDataCoordClient_Call) RunAndReturn(run func(types.DataCoordClient)) *MockProxy_SetDataCoordClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6221,7 +6221,7 @@ func (_c *MockProxy_SetEtcdClient_Call) Return() *MockProxy_SetEtcdClient_Call { } func (_c *MockProxy_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockProxy_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6254,7 +6254,7 @@ func (_c *MockProxy_SetQueryCoordClient_Call) Return() *MockProxy_SetQueryCoordC } func (_c *MockProxy_SetQueryCoordClient_Call) RunAndReturn(run func(types.QueryCoordClient)) *MockProxy_SetQueryCoordClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6287,7 +6287,7 @@ func (_c *MockProxy_SetQueryNodeCreator_Call) Return() *MockProxy_SetQueryNodeCr } func (_c *MockProxy_SetQueryNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.QueryNodeClient, error))) *MockProxy_SetQueryNodeCreator_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6379,7 +6379,7 @@ func (_c *MockProxy_SetRootCoordClient_Call) Return() *MockProxy_SetRootCoordCli } func (_c *MockProxy_SetRootCoordClient_Call) RunAndReturn(run func(types.RootCoordClient)) *MockProxy_SetRootCoordClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6501,7 +6501,7 @@ func (_c *MockProxy_ShowPartitions_Call) RunAndReturn(run func(context.Context, return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockProxy) Start() error { ret := _m.Called() @@ -6546,7 +6546,7 @@ func (_c *MockProxy_Start_Call) RunAndReturn(run func() error) *MockProxy_Start_ return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockProxy) Stop() error { ret := _m.Called() @@ -6915,7 +6915,7 @@ func (_c *MockProxy_UpdateStateCode_Call) Return() *MockProxy_UpdateStateCode_Ca } func (_c *MockProxy_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockProxy_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_proxy_client.go b/internal/mocks/mock_proxy_client.go index ecea90b1ea..4b13afe1bd 100644 --- a/internal/mocks/mock_proxy_client.go +++ b/internal/mocks/mock_proxy_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -31,7 +31,7 @@ func (_m *MockProxyClient) EXPECT() *MockProxyClient_Expecter { return &MockProxyClient_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockProxyClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_querycoord.go b/internal/mocks/mock_querycoord.go index 0410121b23..f4d1417d7f 100644 --- a/internal/mocks/mock_querycoord.go +++ b/internal/mocks/mock_querycoord.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -1037,7 +1037,7 @@ func (_c *MockQueryCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context. return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MockQueryCoord) Init() error { ret := _m.Called() @@ -1436,7 +1436,7 @@ func (_c *MockQueryCoord_LoadPartitions_Call) RunAndReturn(run func(context.Cont return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *MockQueryCoord) Register() error { ret := _m.Called() @@ -1746,7 +1746,7 @@ func (_c *MockQueryCoord_SetAddress_Call) Return() *MockQueryCoord_SetAddress_Ca } func (_c *MockQueryCoord_SetAddress_Call) RunAndReturn(run func(string)) *MockQueryCoord_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1825,7 +1825,7 @@ func (_c *MockQueryCoord_SetEtcdClient_Call) Return() *MockQueryCoord_SetEtcdCli } func (_c *MockQueryCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockQueryCoord_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1858,7 +1858,7 @@ func (_c *MockQueryCoord_SetQueryNodeCreator_Call) Return() *MockQueryCoord_SetQ } func (_c *MockQueryCoord_SetQueryNodeCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.QueryNodeClient, error))) *MockQueryCoord_SetQueryNodeCreator_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1937,7 +1937,7 @@ func (_c *MockQueryCoord_SetTiKVClient_Call) Return() *MockQueryCoord_SetTiKVCli } func (_c *MockQueryCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *MockQueryCoord_SetTiKVClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2118,7 +2118,7 @@ func (_c *MockQueryCoord_ShowPartitions_Call) RunAndReturn(run func(context.Cont return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockQueryCoord) Start() error { ret := _m.Called() @@ -2163,7 +2163,7 @@ func (_c *MockQueryCoord_Start_Call) RunAndReturn(run func() error) *MockQueryCo return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockQueryCoord) Stop() error { ret := _m.Called() @@ -2768,7 +2768,7 @@ func (_c *MockQueryCoord_UpdateStateCode_Call) Return() *MockQueryCoord_UpdateSt } func (_c *MockQueryCoord_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockQueryCoord_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_querycoord_client.go b/internal/mocks/mock_querycoord_client.go index edf426923a..1fff89c4e6 100644 --- a/internal/mocks/mock_querycoord_client.go +++ b/internal/mocks/mock_querycoord_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -327,7 +327,7 @@ func (_c *MockQueryCoordClient_CheckQueryNodeDistribution_Call) RunAndReturn(run return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockQueryCoordClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_querynode.go b/internal/mocks/mock_querynode.go index aead5a45d6..a7e58de332 100644 --- a/internal/mocks/mock_querynode.go +++ b/internal/mocks/mock_querynode.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -148,7 +148,7 @@ func (_c *MockQueryNode_DeleteBatch_Call) RunAndReturn(run func(context.Context, return _c } -// GetAddress provides a mock function with given fields: +// GetAddress provides a mock function with no fields func (_m *MockQueryNode) GetAddress() string { ret := _m.Called() @@ -370,7 +370,7 @@ func (_c *MockQueryNode_GetMetrics_Call) RunAndReturn(run func(context.Context, return _c } -// GetNodeID provides a mock function with given fields: +// GetNodeID provides a mock function with no fields func (_m *MockQueryNode) GetNodeID() int64 { ret := _m.Called() @@ -651,7 +651,7 @@ func (_c *MockQueryNode_GetTimeTickChannel_Call) RunAndReturn(run func(context.C return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MockQueryNode) Init() error { ret := _m.Called() @@ -1026,7 +1026,7 @@ func (_c *MockQueryNode_QueryStreamSegments_Call) RunAndReturn(run func(*querypb return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *MockQueryNode) Register() error { ret := _m.Called() @@ -1395,7 +1395,7 @@ func (_c *MockQueryNode_SetAddress_Call) Return() *MockQueryNode_SetAddress_Call } func (_c *MockQueryNode_SetAddress_Call) RunAndReturn(run func(string)) *MockQueryNode_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1428,7 +1428,7 @@ func (_c *MockQueryNode_SetEtcdClient_Call) Return() *MockQueryNode_SetEtcdClien } func (_c *MockQueryNode_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MockQueryNode_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1491,7 +1491,7 @@ func (_c *MockQueryNode_ShowConfigurations_Call) RunAndReturn(run func(context.C return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockQueryNode) Start() error { ret := _m.Called() @@ -1536,7 +1536,7 @@ func (_c *MockQueryNode_Start_Call) RunAndReturn(run func() error) *MockQueryNod return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockQueryNode) Stop() error { ret := _m.Called() @@ -1787,7 +1787,7 @@ func (_c *MockQueryNode_UpdateStateCode_Call) Return() *MockQueryNode_UpdateStat } func (_c *MockQueryNode_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MockQueryNode_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_querynode_client.go b/internal/mocks/mock_querynode_client.go index 9924fda6fe..0cf8ba5e07 100644 --- a/internal/mocks/mock_querynode_client.go +++ b/internal/mocks/mock_querynode_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -31,7 +31,7 @@ func (_m *MockQueryNodeClient) EXPECT() *MockQueryNodeClient_Expecter { return &MockQueryNodeClient_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockQueryNodeClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_rootcoord.go b/internal/mocks/mock_rootcoord.go index 36c061fd5e..958b0af25c 100644 --- a/internal/mocks/mock_rootcoord.go +++ b/internal/mocks/mock_rootcoord.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -1926,7 +1926,7 @@ func (_c *RootCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.Conte return _c } -// GracefulStop provides a mock function with given fields: +// GracefulStop provides a mock function with no fields func (_m *RootCoord) GracefulStop() { _m.Called() } @@ -1954,7 +1954,7 @@ func (_c *RootCoord_GracefulStop_Call) Return() *RootCoord_GracefulStop_Call { } func (_c *RootCoord_GracefulStop_Call) RunAndReturn(run func()) *RootCoord_GracefulStop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -2076,7 +2076,7 @@ func (_c *RootCoord_HasPartition_Call) RunAndReturn(run func(context.Context, *m return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *RootCoord) Init() error { ret := _m.Called() @@ -2652,7 +2652,7 @@ func (_c *RootCoord_OperateUserRole_Call) RunAndReturn(run func(context.Context, return _c } -// Register provides a mock function with given fields: +// Register provides a mock function with no fields func (_m *RootCoord) Register() error { ret := _m.Called() @@ -2726,7 +2726,7 @@ func (_c *RootCoord_RegisterStreamingCoordGRPCService_Call) Return() *RootCoord_ } func (_c *RootCoord_RegisterStreamingCoordGRPCService_Call) RunAndReturn(run func(*grpc.Server)) *RootCoord_RegisterStreamingCoordGRPCService_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -3054,7 +3054,7 @@ func (_c *RootCoord_SetAddress_Call) Return() *RootCoord_SetAddress_Call { } func (_c *RootCoord_SetAddress_Call) RunAndReturn(run func(string)) *RootCoord_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -3133,7 +3133,7 @@ func (_c *RootCoord_SetEtcdClient_Call) Return() *RootCoord_SetEtcdClient_Call { } func (_c *RootCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *RootCoord_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -3166,7 +3166,7 @@ func (_c *RootCoord_SetProxyCreator_Call) Return() *RootCoord_SetProxyCreator_Ca } func (_c *RootCoord_SetProxyCreator_Call) RunAndReturn(run func(func(context.Context, string, int64) (types.ProxyClient, error))) *RootCoord_SetProxyCreator_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -3245,7 +3245,7 @@ func (_c *RootCoord_SetTiKVClient_Call) Return() *RootCoord_SetTiKVClient_Call { } func (_c *RootCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *RootCoord_SetTiKVClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -3603,7 +3603,7 @@ func (_c *RootCoord_ShowSegments_Call) RunAndReturn(run func(context.Context, *m return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *RootCoord) Start() error { ret := _m.Called() @@ -3648,7 +3648,7 @@ func (_c *RootCoord_Start_Call) RunAndReturn(run func() error) *RootCoord_Start_ return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *RootCoord) Stop() error { ret := _m.Called() @@ -3840,7 +3840,7 @@ func (_c *RootCoord_UpdateStateCode_Call) Return() *RootCoord_UpdateStateCode_Ca } func (_c *RootCoord_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *RootCoord_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_rootcoord_client.go b/internal/mocks/mock_rootcoord_client.go index d797d17aa8..e5fbba3ac3 100644 --- a/internal/mocks/mock_rootcoord_client.go +++ b/internal/mocks/mock_rootcoord_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mocks @@ -625,7 +625,7 @@ func (_c *MockRootCoordClient_CheckHealth_Call) RunAndReturn(run func(context.Co return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockRootCoordClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/streamingcoord/mock_client/mock_BroadcastService.go b/internal/mocks/streamingcoord/mock_client/mock_BroadcastService.go index 9544c099cd..f28b3ba1e7 100644 --- a/internal/mocks/streamingcoord/mock_client/mock_BroadcastService.go +++ b/internal/mocks/streamingcoord/mock_client/mock_BroadcastService.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_client @@ -130,7 +130,7 @@ func (_c *MockBroadcastService_Broadcast_Call) RunAndReturn(run func(context.Con return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockBroadcastService) Close() { _m.Called() } @@ -158,7 +158,7 @@ func (_c *MockBroadcastService_Close_Call) Return() *MockBroadcastService_Close_ } func (_c *MockBroadcastService_Close_Call) RunAndReturn(run func()) *MockBroadcastService_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingcoord/mock_client/mock_Client.go b/internal/mocks/streamingcoord/mock_client/mock_Client.go index 574e08d015..c81cf2ed6f 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_client @@ -20,7 +20,7 @@ func (_m *MockClient) EXPECT() *MockClient_Expecter { return &MockClient_Expecter{mock: &_m.Mock} } -// Assignment provides a mock function with given fields: +// Assignment provides a mock function with no fields func (_m *MockClient) Assignment() client.AssignmentService { ret := _m.Called() @@ -67,7 +67,7 @@ func (_c *MockClient_Assignment_Call) RunAndReturn(run func() client.AssignmentS return _c } -// Broadcast provides a mock function with given fields: +// Broadcast provides a mock function with no fields func (_m *MockClient) Broadcast() client.BroadcastService { ret := _m.Called() @@ -114,7 +114,7 @@ func (_c *MockClient_Broadcast_Call) RunAndReturn(run func() client.BroadcastSer return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockClient) Close() { _m.Called() } @@ -142,7 +142,7 @@ func (_c *MockClient_Close_Call) Return() *MockClient_Close_Call { } func (_c *MockClient_Close_Call) RunAndReturn(run func()) *MockClient_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go b/internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go index dbf89e90c4..9a35e157f1 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_balancer @@ -24,7 +24,7 @@ func (_m *MockBalancer) EXPECT() *MockBalancer_Expecter { return &MockBalancer_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockBalancer) Close() { _m.Called() } @@ -52,7 +52,7 @@ func (_c *MockBalancer_Close_Call) Return() *MockBalancer_Close_Call { } func (_c *MockBalancer_Close_Call) RunAndReturn(run func()) *MockBalancer_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingcoord/server/mock_broadcaster/mock_AppendOperator.go b/internal/mocks/streamingcoord/server/mock_broadcaster/mock_AppendOperator.go index 0b0daf9572..fb82d5a5aa 100644 --- a/internal/mocks/streamingcoord/server/mock_broadcaster/mock_AppendOperator.go +++ b/internal/mocks/streamingcoord/server/mock_broadcaster/mock_AppendOperator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_broadcaster @@ -8,7 +8,7 @@ import ( message "github.com/milvus-io/milvus/pkg/v2/streaming/util/message" mock "github.com/stretchr/testify/mock" - streaming "github.com/milvus-io/milvus/internal/distributed/streaming" + types "github.com/milvus-io/milvus/pkg/v2/streaming/util/types" ) // MockAppendOperator is an autogenerated mock type for the AppendOperator type @@ -25,7 +25,7 @@ func (_m *MockAppendOperator) EXPECT() *MockAppendOperator_Expecter { } // AppendMessages provides a mock function with given fields: ctx, msgs -func (_m *MockAppendOperator) AppendMessages(ctx context.Context, msgs ...message.MutableMessage) streaming.AppendResponses { +func (_m *MockAppendOperator) AppendMessages(ctx context.Context, msgs ...message.MutableMessage) types.AppendResponses { _va := make([]interface{}, len(msgs)) for _i := range msgs { _va[_i] = msgs[_i] @@ -39,11 +39,11 @@ func (_m *MockAppendOperator) AppendMessages(ctx context.Context, msgs ...messag 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 { + var r0 types.AppendResponses + if rf, ok := ret.Get(0).(func(context.Context, ...message.MutableMessage) types.AppendResponses); ok { r0 = rf(ctx, msgs...) } else { - r0 = ret.Get(0).(streaming.AppendResponses) + r0 = ret.Get(0).(types.AppendResponses) } return r0 @@ -75,12 +75,12 @@ func (_c *MockAppendOperator_AppendMessages_Call) Run(run func(ctx context.Conte return _c } -func (_c *MockAppendOperator_AppendMessages_Call) Return(_a0 streaming.AppendResponses) *MockAppendOperator_AppendMessages_Call { +func (_c *MockAppendOperator_AppendMessages_Call) Return(_a0 types.AppendResponses) *MockAppendOperator_AppendMessages_Call { _c.Call.Return(_a0) return _c } -func (_c *MockAppendOperator_AppendMessages_Call) RunAndReturn(run func(context.Context, ...message.MutableMessage) streaming.AppendResponses) *MockAppendOperator_AppendMessages_Call { +func (_c *MockAppendOperator_AppendMessages_Call) RunAndReturn(run func(context.Context, ...message.MutableMessage) types.AppendResponses) *MockAppendOperator_AppendMessages_Call { _c.Call.Return(run) return _c } 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 b582820cb1..d3e07c0b5e 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_assignment @@ -22,7 +22,7 @@ func (_m *MockWatcher) EXPECT() *MockWatcher_Expecter { return &MockWatcher_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockWatcher) Close() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockWatcher_Close_Call) Return() *MockWatcher_Close_Call { } func (_c *MockWatcher_Close_Call) RunAndReturn(run func()) *MockWatcher_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } 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 efa7eb0f7f..ebfd641977 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_consumer @@ -17,7 +17,7 @@ func (_m *MockConsumer) EXPECT() *MockConsumer_Expecter { return &MockConsumer_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockConsumer) Close() { _m.Called() } @@ -45,11 +45,11 @@ func (_c *MockConsumer_Close_Call) Return() *MockConsumer_Close_Call { } func (_c *MockConsumer_Close_Call) RunAndReturn(run func()) *MockConsumer_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Done provides a mock function with given fields: +// Done provides a mock function with no fields func (_m *MockConsumer) Done() <-chan struct{} { ret := _m.Called() @@ -96,7 +96,7 @@ func (_c *MockConsumer_Done_Call) RunAndReturn(run func() <-chan struct{}) *Mock return _c } -// Error provides a mock function with given fields: +// Error provides a mock function with no fields func (_m *MockConsumer) Error() error { ret := _m.Called() 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 39178a28a6..dbfc00a47e 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_producer @@ -24,7 +24,7 @@ func (_m *MockProducer) EXPECT() *MockProducer_Expecter { return &MockProducer_Expecter{mock: &_m.Mock} } -// Assignment provides a mock function with given fields: +// Assignment provides a mock function with no fields func (_m *MockProducer) Assignment() types.PChannelInfoAssigned { ret := _m.Called() @@ -69,7 +69,7 @@ func (_c *MockProducer_Assignment_Call) RunAndReturn(run func() types.PChannelIn return _c } -// Available provides a mock function with given fields: +// Available provides a mock function with no fields func (_m *MockProducer) Available() <-chan struct{} { ret := _m.Called() @@ -116,7 +116,7 @@ func (_c *MockProducer_Available_Call) RunAndReturn(run func() <-chan struct{}) return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockProducer) Close() { _m.Called() } @@ -144,11 +144,11 @@ func (_c *MockProducer_Close_Call) Return() *MockProducer_Close_Call { } func (_c *MockProducer_Close_Call) RunAndReturn(run func()) *MockProducer_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// IsAvailable provides a mock function with given fields: +// IsAvailable provides a mock function with no fields func (_m *MockProducer) IsAvailable() bool { ret := _m.Called() diff --git a/internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go b/internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go index dd5d140160..f2326ab39e 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_handler @@ -27,7 +27,7 @@ func (_m *MockHandlerClient) EXPECT() *MockHandlerClient_Expecter { return &MockHandlerClient_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockHandlerClient) Close() { _m.Called() } @@ -55,7 +55,7 @@ func (_c *MockHandlerClient_Close_Call) Return() *MockHandlerClient_Close_Call { } func (_c *MockHandlerClient_Close_Call) RunAndReturn(run func()) *MockHandlerClient_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go b/internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go index f4e1bda989..4747109d9b 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_manager @@ -70,7 +70,7 @@ func (_c *MockManagerClient_Assign_Call) RunAndReturn(run func(context.Context, return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockManagerClient) Close() { _m.Called() } @@ -98,7 +98,7 @@ func (_c *MockManagerClient_Close_Call) Return() *MockManagerClient_Close_Call { } func (_c *MockManagerClient_Close_Call) RunAndReturn(run func()) *MockManagerClient_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go b/internal/mocks/streamingnode/server/mock_flusher/mock_Flusher.go index a1ab70673b..d2d04875b5 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_flusher @@ -97,11 +97,11 @@ func (_c *MockFlusher_RegisterVChannel_Call) Return() *MockFlusher_RegisterVChan } func (_c *MockFlusher_RegisterVChannel_Call) RunAndReturn(run func(string, wal.WAL)) *MockFlusher_RegisterVChannel_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockFlusher) Start() { _m.Called() } @@ -129,11 +129,11 @@ func (_c *MockFlusher_Start_Call) Return() *MockFlusher_Start_Call { } func (_c *MockFlusher_Start_Call) RunAndReturn(run func()) *MockFlusher_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockFlusher) Stop() { _m.Called() } @@ -161,7 +161,7 @@ func (_c *MockFlusher_Stop_Call) Return() *MockFlusher_Stop_Call { } func (_c *MockFlusher_Stop_Call) RunAndReturn(run func()) *MockFlusher_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -194,7 +194,7 @@ func (_c *MockFlusher_UnregisterPChannel_Call) Return() *MockFlusher_UnregisterP } func (_c *MockFlusher_UnregisterPChannel_Call) RunAndReturn(run func(string)) *MockFlusher_UnregisterPChannel_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -227,7 +227,7 @@ func (_c *MockFlusher_UnregisterVChannel_Call) Return() *MockFlusher_UnregisterV } func (_c *MockFlusher_UnregisterVChannel_Call) RunAndReturn(run func(string)) *MockFlusher_UnregisterVChannel_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_Opener.go b/internal/mocks/streamingnode/server/mock_wal/mock_Opener.go index 808d1bd10f..7489e4f608 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_wal @@ -22,7 +22,7 @@ func (_m *MockOpener) EXPECT() *MockOpener_Expecter { return &MockOpener_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockOpener) Close() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockOpener_Close_Call) Return() *MockOpener_Close_Call { } func (_c *MockOpener_Close_Call) RunAndReturn(run func()) *MockOpener_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go b/internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go index 92a243cb82..5dec8e7926 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_wal @@ -20,7 +20,7 @@ func (_m *MockOpenerBuilder) EXPECT() *MockOpenerBuilder_Expecter { return &MockOpenerBuilder_Expecter{mock: &_m.Mock} } -// Build provides a mock function with given fields: +// Build provides a mock function with no fields func (_m *MockOpenerBuilder) Build() (wal.Opener, error) { ret := _m.Called() @@ -77,7 +77,7 @@ func (_c *MockOpenerBuilder_Build_Call) RunAndReturn(run func() (wal.Opener, err return _c } -// Name provides a mock function with given fields: +// Name provides a mock function with no fields func (_m *MockOpenerBuilder) Name() string { ret := _m.Called() diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go b/internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go index 468e46dda3..10bf560f52 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_wal @@ -22,7 +22,7 @@ func (_m *MockScanner) EXPECT() *MockScanner_Expecter { return &MockScanner_Expecter{mock: &_m.Mock} } -// Chan provides a mock function with given fields: +// Chan provides a mock function with no fields func (_m *MockScanner) Chan() <-chan message.ImmutableMessage { ret := _m.Called() @@ -69,7 +69,7 @@ func (_c *MockScanner_Chan_Call) RunAndReturn(run func() <-chan message.Immutabl return _c } -// Channel provides a mock function with given fields: +// Channel provides a mock function with no fields func (_m *MockScanner) Channel() types.PChannelInfo { ret := _m.Called() @@ -114,7 +114,7 @@ func (_c *MockScanner_Channel_Call) RunAndReturn(run func() types.PChannelInfo) return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockScanner) Close() error { ret := _m.Called() @@ -159,7 +159,7 @@ func (_c *MockScanner_Close_Call) RunAndReturn(run func() error) *MockScanner_Cl return _c } -// Done provides a mock function with given fields: +// Done provides a mock function with no fields func (_m *MockScanner) Done() <-chan struct{} { ret := _m.Called() @@ -206,7 +206,7 @@ func (_c *MockScanner_Done_Call) RunAndReturn(run func() <-chan struct{}) *MockS return _c } -// Error provides a mock function with given fields: +// Error provides a mock function with no fields func (_m *MockScanner) Error() error { ret := _m.Called() diff --git a/internal/mocks/streamingnode/server/mock_wal/mock_WAL.go b/internal/mocks/streamingnode/server/mock_wal/mock_WAL.go index f195c6160e..7e509515d2 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_wal @@ -116,11 +116,11 @@ func (_c *MockWAL_AppendAsync_Call) Return() *MockWAL_AppendAsync_Call { } func (_c *MockWAL_AppendAsync_Call) RunAndReturn(run func(context.Context, message.MutableMessage, func(*types.AppendResult, error))) *MockWAL_AppendAsync_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Available provides a mock function with given fields: +// Available provides a mock function with no fields func (_m *MockWAL) Available() <-chan struct{} { ret := _m.Called() @@ -167,7 +167,7 @@ func (_c *MockWAL_Available_Call) RunAndReturn(run func() <-chan struct{}) *Mock return _c } -// Channel provides a mock function with given fields: +// Channel provides a mock function with no fields func (_m *MockWAL) Channel() types.PChannelInfo { ret := _m.Called() @@ -212,7 +212,7 @@ func (_c *MockWAL_Channel_Call) RunAndReturn(run func() types.PChannelInfo) *Moc return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockWAL) Close() { _m.Called() } @@ -240,11 +240,11 @@ func (_c *MockWAL_Close_Call) Return() *MockWAL_Close_Call { } func (_c *MockWAL_Close_Call) RunAndReturn(run func()) *MockWAL_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// IsAvailable provides a mock function with given fields: +// IsAvailable provides a mock function with no fields func (_m *MockWAL) IsAvailable() bool { ret := _m.Called() @@ -348,7 +348,7 @@ func (_c *MockWAL_Read_Call) RunAndReturn(run func(context.Context, wal.ReadOpti return _c } -// WALName provides a mock function with given fields: +// WALName provides a mock function with no fields func (_m *MockWAL) WALName() string { ret := _m.Called() diff --git a/internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go b/internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go index 038faaca59..c67b364d06 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_walmanager @@ -24,7 +24,7 @@ func (_m *MockManager) EXPECT() *MockManager_Expecter { return &MockManager_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockManager) Close() { _m.Called() } @@ -52,11 +52,11 @@ func (_c *MockManager_Close_Call) Return() *MockManager_Close_Call { } func (_c *MockManager_Close_Call) RunAndReturn(run func()) *MockManager_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// GetAllAvailableChannels provides a mock function with given fields: +// GetAllAvailableChannels provides a mock function with no fields func (_m *MockManager) GetAllAvailableChannels() ([]types.PChannelInfo, error) { ret := _m.Called() 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 e120692282..f8697fb36d 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_inspector @@ -25,7 +25,7 @@ func (_m *MockSealOperator) EXPECT() *MockSealOperator_Expecter { return &MockSealOperator_Expecter{mock: &_m.Mock} } -// Channel provides a mock function with given fields: +// Channel provides a mock function with no fields func (_m *MockSealOperator) Channel() types.PChannelInfo { ret := _m.Called() @@ -70,7 +70,7 @@ func (_c *MockSealOperator_Channel_Call) RunAndReturn(run func() types.PChannelI return _c } -// IsNoWaitSeal provides a mock function with given fields: +// IsNoWaitSeal provides a mock function with no fields func (_m *MockSealOperator) IsNoWaitSeal() bool { ret := _m.Called() @@ -159,7 +159,7 @@ func (_c *MockSealOperator_MustSealSegments_Call) Return() *MockSealOperator_Mus } func (_c *MockSealOperator_MustSealSegments_Call) RunAndReturn(run func(context.Context, ...stats.SegmentBelongs)) *MockSealOperator_MustSealSegments_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -207,7 +207,7 @@ func (_c *MockSealOperator_TryToSealSegments_Call) Return() *MockSealOperator_Tr } func (_c *MockSealOperator_TryToSealSegments_Call) RunAndReturn(run func(context.Context, ...stats.SegmentBelongs)) *MockSealOperator_TryToSealSegments_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -240,7 +240,7 @@ func (_c *MockSealOperator_TryToSealWaitedSegment_Call) Return() *MockSealOperat } func (_c *MockSealOperator_TryToSealWaitedSegment_Call) RunAndReturn(run func(context.Context)) *MockSealOperator_TryToSealWaitedSegment_Call { - _c.Call.Return(run) + _c.Run(run) return _c } 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 8af21523a7..b7dc005ee2 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_inspector @@ -24,7 +24,7 @@ func (_m *MockTimeTickSyncOperator) EXPECT() *MockTimeTickSyncOperator_Expecter return &MockTimeTickSyncOperator_Expecter{mock: &_m.Mock} } -// Channel provides a mock function with given fields: +// Channel provides a mock function with no fields func (_m *MockTimeTickSyncOperator) Channel() types.PChannelInfo { ret := _m.Called() @@ -98,11 +98,11 @@ func (_c *MockTimeTickSyncOperator_Sync_Call) Return() *MockTimeTickSyncOperator } func (_c *MockTimeTickSyncOperator_Sync_Call) RunAndReturn(run func(context.Context)) *MockTimeTickSyncOperator_Sync_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// TimeTickNotifier provides a mock function with given fields: +// TimeTickNotifier provides a mock function with no fields func (_m *MockTimeTickSyncOperator) TimeTickNotifier() *inspector.TimeTickNotifier { ret := _m.Called() 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 4f78455948..8336aef1d4 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_interceptors @@ -23,7 +23,7 @@ func (_m *MockInterceptor) EXPECT() *MockInterceptor_Expecter { return &MockInterceptor_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockInterceptor) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *MockInterceptor_Close_Call) Return() *MockInterceptor_Close_Call { } func (_c *MockInterceptor_Close_Call) RunAndReturn(run func()) *MockInterceptor_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } 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 512d567e02..f341f93fac 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_interceptors 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 de2e8d9a26..0150e70160 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_interceptors @@ -23,7 +23,7 @@ func (_m *MockInterceptorWithReady) EXPECT() *MockInterceptorWithReady_Expecter return &MockInterceptorWithReady_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockInterceptorWithReady) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *MockInterceptorWithReady_Close_Call) Return() *MockInterceptorWithRead } func (_c *MockInterceptorWithReady_Close_Call) RunAndReturn(run func()) *MockInterceptorWithReady_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -115,7 +115,7 @@ func (_c *MockInterceptorWithReady_DoAppend_Call) RunAndReturn(run func(context. return _c } -// Ready provides a mock function with given fields: +// Ready provides a mock function with no fields func (_m *MockInterceptorWithReady) Ready() <-chan struct{} { ret := _m.Called() diff --git a/internal/mocks/util/mock_segcore/mock_CSegment.go b/internal/mocks/util/mock_segcore/mock_CSegment.go index dc77513b2c..14b7ac6a87 100644 --- a/internal/mocks/util/mock_segcore/mock_CSegment.go +++ b/internal/mocks/util/mock_segcore/mock_CSegment.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_segcore @@ -186,7 +186,7 @@ func (_c *MockCSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *Mock return _c } -// ID provides a mock function with given fields: +// ID provides a mock function with no fields func (_m *MockCSegment) ID() int64 { ret := _m.Called() @@ -349,7 +349,7 @@ func (_c *MockCSegment_LoadFieldData_Call) RunAndReturn(run func(context.Context return _c } -// MemSize provides a mock function with given fields: +// MemSize provides a mock function with no fields func (_m *MockCSegment) MemSize() int64 { ret := _m.Called() @@ -394,7 +394,7 @@ func (_c *MockCSegment_MemSize_Call) RunAndReturn(run func() int64) *MockCSegmen return _c } -// RawPointer provides a mock function with given fields: +// RawPointer provides a mock function with no fields func (_m *MockCSegment) RawPointer() segcore.CSegmentInterface { ret := _m.Called() @@ -439,7 +439,7 @@ func (_c *MockCSegment_RawPointer_Call) RunAndReturn(run func() segcore.CSegment return _c } -// Release provides a mock function with given fields: +// Release provides a mock function with no fields func (_m *MockCSegment) Release() { _m.Called() } @@ -467,7 +467,7 @@ func (_c *MockCSegment_Release_Call) Return() *MockCSegment_Release_Call { } func (_c *MockCSegment_Release_Call) RunAndReturn(run func()) *MockCSegment_Release_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -589,7 +589,7 @@ func (_c *MockCSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Con return _c } -// RowNum provides a mock function with given fields: +// RowNum provides a mock function with no fields func (_m *MockCSegment) RowNum() int64 { ret := _m.Called() diff --git a/internal/mocks/util/searchutil/mock_optimizers/mock_QueryHook.go b/internal/mocks/util/searchutil/mock_optimizers/mock_QueryHook.go index 5521418b11..57bd0bd6d6 100644 --- a/internal/mocks/util/searchutil/mock_optimizers/mock_QueryHook.go +++ b/internal/mocks/util/searchutil/mock_optimizers/mock_QueryHook.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_optimizers 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 1a62b68a19..5e53288ca9 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_discoverer @@ -69,7 +69,7 @@ func (_c *MockDiscoverer_Discover_Call) RunAndReturn(run func(context.Context, f return _c } -// NewVersionedState provides a mock function with given fields: +// NewVersionedState provides a mock function with no fields func (_m *MockDiscoverer) NewVersionedState() discoverer.VersionedState { ret := _m.Called() 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 2c66a694c5..f4c9882496 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_lazygrpc @@ -23,7 +23,7 @@ func (_m *MockService[T]) EXPECT() *MockService_Expecter[T] { return &MockService_Expecter[T]{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockService[T]) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *MockService_Close_Call[T]) Return() *MockService_Close_Call[T] { } func (_c *MockService_Close_Call[T]) RunAndReturn(run func()) *MockService_Close_Call[T] { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -129,7 +129,9 @@ func (_m *MockService[T]) GetService(ctx context.Context) (T, error) { if rf, ok := ret.Get(0).(func(context.Context) T); ok { r0 = rf(ctx) } else { - r0 = ret.Get(0).(T) + if ret.Get(0) != nil { + r0 = ret.Get(0).(T) + } } if rf, ok := ret.Get(1).(func(context.Context) 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 8dd8a5d03a..ba8e1e94d6 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_resolver @@ -82,7 +82,7 @@ func (_c *MockBuilder_Build_Call) RunAndReturn(run func(resolver.Target, resolve return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockBuilder) Close() { _m.Called() } @@ -110,11 +110,11 @@ func (_c *MockBuilder_Close_Call) Return() *MockBuilder_Close_Call { } func (_c *MockBuilder_Close_Call) RunAndReturn(run func()) *MockBuilder_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Resolver provides a mock function with given fields: +// Resolver provides a mock function with no fields func (_m *MockBuilder) Resolver() serviceresolver.Resolver { ret := _m.Called() @@ -161,7 +161,7 @@ func (_c *MockBuilder_Resolver_Call) RunAndReturn(run func() serviceresolver.Res return _c } -// Scheme provides a mock function with given fields: +// Scheme provides a mock function with no fields func (_m *MockBuilder) Scheme() string { ret := _m.Called() 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 e974b9acb8..aebd71e6cf 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_resolver @@ -22,7 +22,7 @@ func (_m *MockResolver) EXPECT() *MockResolver_Expecter { return &MockResolver_Expecter{mock: &_m.Mock} } -// GetLatestState provides a mock function with given fields: +// GetLatestState provides a mock function with no fields func (_m *MockResolver) GetLatestState() discoverer.VersionedState { ret := _m.Called() diff --git a/internal/proxy/mock_cache.go b/internal/proxy/mock_cache.go index eb5c3d3a12..1bd04c3185 100644 --- a/internal/proxy/mock_cache.go +++ b/internal/proxy/mock_cache.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package proxy @@ -110,7 +110,7 @@ func (_c *MockCache_DeprecateShardCache_Call) Return() *MockCache_DeprecateShard } func (_c *MockCache_DeprecateShardCache_Call) RunAndReturn(run func(string, string)) *MockCache_DeprecateShardCache_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -944,7 +944,7 @@ func (_c *MockCache_InitPolicyInfo_Call) Return() *MockCache_InitPolicyInfo_Call } func (_c *MockCache_InitPolicyInfo_Call) RunAndReturn(run func([]string, []string)) *MockCache_InitPolicyInfo_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -977,11 +977,11 @@ func (_c *MockCache_InvalidateShardLeaderCache_Call) Return() *MockCache_Invalid } func (_c *MockCache_InvalidateShardLeaderCache_Call) RunAndReturn(run func([]int64)) *MockCache_InvalidateShardLeaderCache_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// ListShardLocation provides a mock function with given fields: +// ListShardLocation provides a mock function with no fields func (_m *MockCache) ListShardLocation() map[int64]nodeInfo { ret := _m.Called() @@ -1105,7 +1105,7 @@ func (_c *MockCache_RemoveCollection_Call) Return() *MockCache_RemoveCollection_ } func (_c *MockCache_RemoveCollection_Call) RunAndReturn(run func(context.Context, string, string)) *MockCache_RemoveCollection_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1189,7 +1189,7 @@ func (_c *MockCache_RemoveCredential_Call) Return() *MockCache_RemoveCredential_ } func (_c *MockCache_RemoveCredential_Call) RunAndReturn(run func(string)) *MockCache_RemoveCredential_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1223,7 +1223,7 @@ func (_c *MockCache_RemoveDatabase_Call) Return() *MockCache_RemoveDatabase_Call } func (_c *MockCache_RemoveDatabase_Call) RunAndReturn(run func(context.Context, string)) *MockCache_RemoveDatabase_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1256,7 +1256,7 @@ func (_c *MockCache_UpdateCredential_Call) Return() *MockCache_UpdateCredential_ } func (_c *MockCache_UpdateCredential_Call) RunAndReturn(run func(*internalpb.CredentialInfo)) *MockCache_UpdateCredential_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/proxy/mock_channels_manager.go b/internal/proxy/mock_channels_manager.go index a21686b1ab..54a672d0a8 100644 --- a/internal/proxy/mock_channels_manager.go +++ b/internal/proxy/mock_channels_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package proxy @@ -197,7 +197,7 @@ func (_c *MockChannelsMgr_getVChannels_Call) RunAndReturn(run func(int64) ([]str return _c } -// removeAllDMLStream provides a mock function with given fields: +// removeAllDMLStream provides a mock function with no fields func (_m *MockChannelsMgr) removeAllDMLStream() { _m.Called() } @@ -225,7 +225,7 @@ func (_c *MockChannelsMgr_removeAllDMLStream_Call) Return() *MockChannelsMgr_rem } func (_c *MockChannelsMgr_removeAllDMLStream_Call) RunAndReturn(run func()) *MockChannelsMgr_removeAllDMLStream_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -258,7 +258,7 @@ func (_c *MockChannelsMgr_removeDMLStream_Call) Return() *MockChannelsMgr_remove } func (_c *MockChannelsMgr_removeDMLStream_Call) RunAndReturn(run func(int64)) *MockChannelsMgr_removeDMLStream_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/proxy/mock_lb_balancer.go b/internal/proxy/mock_lb_balancer.go index 556ede06c2..0d441b1ecf 100644 --- a/internal/proxy/mock_lb_balancer.go +++ b/internal/proxy/mock_lb_balancer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package proxy @@ -52,11 +52,11 @@ func (_c *MockLBBalancer_CancelWorkload_Call) Return() *MockLBBalancer_CancelWor } func (_c *MockLBBalancer_CancelWorkload_Call) RunAndReturn(run func(int64, int64)) *MockLBBalancer_CancelWorkload_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockLBBalancer) Close() { _m.Called() } @@ -84,7 +84,7 @@ func (_c *MockLBBalancer_Close_Call) Return() *MockLBBalancer_Close_Call { } func (_c *MockLBBalancer_Close_Call) RunAndReturn(run func()) *MockLBBalancer_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -117,7 +117,7 @@ func (_c *MockLBBalancer_RegisterNodeInfo_Call) Return() *MockLBBalancer_Registe } func (_c *MockLBBalancer_RegisterNodeInfo_Call) RunAndReturn(run func([]nodeInfo)) *MockLBBalancer_RegisterNodeInfo_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -208,7 +208,7 @@ func (_c *MockLBBalancer_Start_Call) Return() *MockLBBalancer_Start_Call { } func (_c *MockLBBalancer_Start_Call) RunAndReturn(run func(context.Context)) *MockLBBalancer_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -242,7 +242,7 @@ func (_c *MockLBBalancer_UpdateCostMetrics_Call) Return() *MockLBBalancer_Update } func (_c *MockLBBalancer_UpdateCostMetrics_Call) RunAndReturn(run func(int64, *internalpb.CostAggregation)) *MockLBBalancer_UpdateCostMetrics_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/proxy/mock_lb_policy.go b/internal/proxy/mock_lb_policy.go index 541d545385..227b404a4e 100644 --- a/internal/proxy/mock_lb_policy.go +++ b/internal/proxy/mock_lb_policy.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package proxy @@ -22,7 +22,7 @@ func (_m *MockLBPolicy) EXPECT() *MockLBPolicy_Expecter { return &MockLBPolicy_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockLBPolicy) Close() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockLBPolicy_Close_Call) Return() *MockLBPolicy_Close_Call { } func (_c *MockLBPolicy_Close_Call) RunAndReturn(run func()) *MockLBPolicy_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -177,7 +177,7 @@ func (_c *MockLBPolicy_Start_Call) Return() *MockLBPolicy_Start_Call { } func (_c *MockLBPolicy_Start_Call) RunAndReturn(run func(context.Context)) *MockLBPolicy_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -211,7 +211,7 @@ func (_c *MockLBPolicy_UpdateCostMetrics_Call) Return() *MockLBPolicy_UpdateCost } func (_c *MockLBPolicy_UpdateCostMetrics_Call) RunAndReturn(run func(int64, *internalpb.CostAggregation)) *MockLBPolicy_UpdateCostMetrics_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/proxy/mock_shardclient_manager.go b/internal/proxy/mock_shardclient_manager.go index a9c7921efb..3ed2c3367d 100644 --- a/internal/proxy/mock_shardclient_manager.go +++ b/internal/proxy/mock_shardclient_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package proxy @@ -22,7 +22,7 @@ func (_m *MockShardClientManager) EXPECT() *MockShardClientManager_Expecter { return &MockShardClientManager_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockShardClientManager) Close() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockShardClientManager_Close_Call) Return() *MockShardClientManager_Cl } func (_c *MockShardClientManager_Close_Call) RunAndReturn(run func()) *MockShardClientManager_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -142,7 +142,7 @@ func (_c *MockShardClientManager_SetClientCreatorFunc_Call) Return() *MockShardC } func (_c *MockShardClientManager_SetClientCreatorFunc_Call) RunAndReturn(run func(queryNodeCreatorFunc)) *MockShardClientManager_SetClientCreatorFunc_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/proxy/mock_tso_test.go b/internal/proxy/mock_tso_test.go index f10ac4ec2a..8b53e444cf 100644 --- a/internal/proxy/mock_tso_test.go +++ b/internal/proxy/mock_tso_test.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package proxy diff --git a/internal/querycoordv2/balance/mock_balancer.go b/internal/querycoordv2/balance/mock_balancer.go index e7f082f667..7b07415ea3 100644 --- a/internal/querycoordv2/balance/mock_balancer.go +++ b/internal/querycoordv2/balance/mock_balancer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package balance diff --git a/internal/querycoordv2/dist/mock_controller.go b/internal/querycoordv2/dist/mock_controller.go index 5a33c07cc9..862db366f4 100644 --- a/internal/querycoordv2/dist/mock_controller.go +++ b/internal/querycoordv2/dist/mock_controller.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package dist @@ -50,7 +50,7 @@ func (_c *MockController_Remove_Call) Return() *MockController_Remove_Call { } func (_c *MockController_Remove_Call) RunAndReturn(run func(int64)) *MockController_Remove_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -84,11 +84,11 @@ func (_c *MockController_StartDistInstance_Call) Return() *MockController_StartD } func (_c *MockController_StartDistInstance_Call) RunAndReturn(run func(context.Context, int64)) *MockController_StartDistInstance_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockController) Stop() { _m.Called() } @@ -116,7 +116,7 @@ func (_c *MockController_Stop_Call) Return() *MockController_Stop_Call { } func (_c *MockController_Stop_Call) RunAndReturn(run func()) *MockController_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -149,7 +149,7 @@ func (_c *MockController_SyncAll_Call) Return() *MockController_SyncAll_Call { } func (_c *MockController_SyncAll_Call) RunAndReturn(run func(context.Context)) *MockController_SyncAll_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/querycoordv2/meta/mock_broker.go b/internal/querycoordv2/meta/mock_broker.go index 5eb877d94f..6f79bd0344 100644 --- a/internal/querycoordv2/meta/mock_broker.go +++ b/internal/querycoordv2/meta/mock_broker.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package meta diff --git a/internal/querycoordv2/meta/mock_target_manager.go b/internal/querycoordv2/meta/mock_target_manager.go index 1ffeb2dfd5..9ed5d3d7e9 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package meta @@ -953,7 +953,7 @@ func (_c *MockTargetManager_RemoveCollection_Call) Return() *MockTargetManager_R } func (_c *MockTargetManager_RemoveCollection_Call) RunAndReturn(run func(context.Context, int64)) *MockTargetManager_RemoveCollection_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1002,7 +1002,7 @@ func (_c *MockTargetManager_RemovePartition_Call) Return() *MockTargetManager_Re } func (_c *MockTargetManager_RemovePartition_Call) RunAndReturn(run func(context.Context, int64, ...int64)) *MockTargetManager_RemovePartition_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1036,7 +1036,7 @@ func (_c *MockTargetManager_SaveCurrentTarget_Call) Return() *MockTargetManager_ } func (_c *MockTargetManager_SaveCurrentTarget_Call) RunAndReturn(run func(context.Context, metastore.QueryCoordCatalog)) *MockTargetManager_SaveCurrentTarget_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/querycoordv2/session/mock_cluster.go b/internal/querycoordv2/session/mock_cluster.go index 6700569c31..b1e2ccbe27 100644 --- a/internal/querycoordv2/session/mock_cluster.go +++ b/internal/querycoordv2/session/mock_cluster.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package session @@ -446,7 +446,7 @@ func (_c *MockCluster_ReleaseSegments_Call) RunAndReturn(run func(context.Contex return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockCluster) Start() { _m.Called() } @@ -474,11 +474,11 @@ func (_c *MockCluster_Start_Call) Return() *MockCluster_Start_Call { } func (_c *MockCluster_Start_Call) RunAndReturn(run func()) *MockCluster_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockCluster) Stop() { _m.Called() } @@ -506,7 +506,7 @@ func (_c *MockCluster_Stop_Call) Return() *MockCluster_Stop_Call { } func (_c *MockCluster_Stop_Call) RunAndReturn(run func()) *MockCluster_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/querycoordv2/task/mock_scheduler.go b/internal/querycoordv2/task/mock_scheduler.go index ff0cd6e0e7..454606f9a8 100644 --- a/internal/querycoordv2/task/mock_scheduler.go +++ b/internal/querycoordv2/task/mock_scheduler.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package task @@ -17,9 +17,9 @@ func (_m *MockScheduler) EXPECT() *MockScheduler_Expecter { return &MockScheduler_Expecter{mock: &_m.Mock} } -// Add provides a mock function with given fields: task -func (_m *MockScheduler) Add(task Task) error { - ret := _m.Called(task) +// Add provides a mock function with given fields: _a0 +func (_m *MockScheduler) Add(_a0 Task) error { + ret := _m.Called(_a0) if len(ret) == 0 { panic("no return value specified for Add") @@ -27,7 +27,7 @@ func (_m *MockScheduler) Add(task Task) error { var r0 error if rf, ok := ret.Get(0).(func(Task) error); ok { - r0 = rf(task) + r0 = rf(_a0) } else { r0 = ret.Error(0) } @@ -41,12 +41,12 @@ type MockScheduler_Add_Call struct { } // Add is a helper method to define mock.On call -// - task Task -func (_e *MockScheduler_Expecter) Add(task interface{}) *MockScheduler_Add_Call { - return &MockScheduler_Add_Call{Call: _e.mock.On("Add", task)} +// - _a0 Task +func (_e *MockScheduler_Expecter) Add(_a0 interface{}) *MockScheduler_Add_Call { + return &MockScheduler_Add_Call{Call: _e.mock.On("Add", _a0)} } -func (_c *MockScheduler_Add_Call) Run(run func(task Task)) *MockScheduler_Add_Call { +func (_c *MockScheduler_Add_Call) Run(run func(_a0 Task)) *MockScheduler_Add_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(Task)) }) @@ -92,7 +92,7 @@ func (_c *MockScheduler_AddExecutor_Call) Return() *MockScheduler_AddExecutor_Ca } func (_c *MockScheduler_AddExecutor_Call) RunAndReturn(run func(int64)) *MockScheduler_AddExecutor_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -125,7 +125,7 @@ func (_c *MockScheduler_Dispatch_Call) Return() *MockScheduler_Dispatch_Call { } func (_c *MockScheduler_Dispatch_Call) RunAndReturn(run func(int64)) *MockScheduler_Dispatch_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -389,7 +389,7 @@ func (_c *MockScheduler_GetSegmentTaskNum_Call) RunAndReturn(run func(...TaskFil return _c } -// GetTasksJSON provides a mock function with given fields: +// GetTasksJSON provides a mock function with no fields func (_m *MockScheduler) GetTasksJSON() string { ret := _m.Called() @@ -463,7 +463,7 @@ func (_c *MockScheduler_RemoveByNode_Call) Return() *MockScheduler_RemoveByNode_ } func (_c *MockScheduler_RemoveByNode_Call) RunAndReturn(run func(int64)) *MockScheduler_RemoveByNode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -496,11 +496,11 @@ func (_c *MockScheduler_RemoveExecutor_Call) Return() *MockScheduler_RemoveExecu } func (_c *MockScheduler_RemoveExecutor_Call) RunAndReturn(run func(int64)) *MockScheduler_RemoveExecutor_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockScheduler) Start() { _m.Called() } @@ -528,11 +528,11 @@ func (_c *MockScheduler_Start_Call) Return() *MockScheduler_Start_Call { } func (_c *MockScheduler_Start_Call) RunAndReturn(run func()) *MockScheduler_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockScheduler) Stop() { _m.Called() } @@ -560,7 +560,7 @@ func (_c *MockScheduler_Stop_Call) Return() *MockScheduler_Stop_Call { } func (_c *MockScheduler_Stop_Call) RunAndReturn(run func()) *MockScheduler_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/querynodev2/cluster/mock_manager.go b/internal/querynodev2/cluster/mock_manager.go index 06593f204c..e20a2fbb68 100644 --- a/internal/querynodev2/cluster/mock_manager.go +++ b/internal/querynodev2/cluster/mock_manager.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package cluster diff --git a/internal/querynodev2/cluster/mock_worker.go b/internal/querynodev2/cluster/mock_worker.go index f710f517aa..923f61ed10 100644 --- a/internal/querynodev2/cluster/mock_worker.go +++ b/internal/querynodev2/cluster/mock_worker.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package cluster @@ -191,7 +191,7 @@ func (_c *MockWorker_GetStatistics_Call) RunAndReturn(run func(context.Context, return _c } -// IsHealthy provides a mock function with given fields: +// IsHealthy provides a mock function with no fields func (_m *MockWorker) IsHealthy() bool { ret := _m.Called() @@ -496,7 +496,7 @@ func (_c *MockWorker_SearchSegments_Call) RunAndReturn(run func(context.Context, return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MockWorker) Stop() { _m.Called() } @@ -524,7 +524,7 @@ func (_c *MockWorker_Stop_Call) Return() *MockWorker_Stop_Call { } func (_c *MockWorker_Stop_Call) RunAndReturn(run func()) *MockWorker_Stop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/querynodev2/delegator/mock_delegator.go b/internal/querynodev2/delegator/mock_delegator.go index 3df854a310..f954f6c379 100644 --- a/internal/querynodev2/delegator/mock_delegator.go +++ b/internal/querynodev2/delegator/mock_delegator.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package delegator @@ -57,11 +57,11 @@ func (_c *MockShardDelegator_AddExcludedSegments_Call) Return() *MockShardDelega } func (_c *MockShardDelegator_AddExcludedSegments_Call) RunAndReturn(run func(map[int64]uint64)) *MockShardDelegator_AddExcludedSegments_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockShardDelegator) Close() { _m.Called() } @@ -89,11 +89,11 @@ func (_c *MockShardDelegator_Close_Call) Return() *MockShardDelegator_Close_Call } func (_c *MockShardDelegator_Close_Call) RunAndReturn(run func()) *MockShardDelegator_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Collection provides a mock function with given fields: +// Collection provides a mock function with no fields func (_m *MockShardDelegator) Collection() int64 { ret := _m.Called() @@ -138,7 +138,7 @@ func (_c *MockShardDelegator_Collection_Call) RunAndReturn(run func() int64) *Mo return _c } -// GetDeleteBufferSize provides a mock function with given fields: +// GetDeleteBufferSize provides a mock function with no fields func (_m *MockShardDelegator) GetDeleteBufferSize() (int64, int64) { ret := _m.Called() @@ -360,7 +360,7 @@ func (_c *MockShardDelegator_GetStatistics_Call) RunAndReturn(run func(context.C return _c } -// GetTSafe provides a mock function with given fields: +// GetTSafe provides a mock function with no fields func (_m *MockShardDelegator) GetTSafe() uint64 { ret := _m.Called() @@ -405,7 +405,7 @@ func (_c *MockShardDelegator_GetTSafe_Call) RunAndReturn(run func() uint64) *Moc return _c } -// GetTargetVersion provides a mock function with given fields: +// GetTargetVersion provides a mock function with no fields func (_m *MockShardDelegator) GetTargetVersion() int64 { ret := _m.Called() @@ -575,7 +575,7 @@ func (_c *MockShardDelegator_ProcessDelete_Call) Return() *MockShardDelegator_Pr } func (_c *MockShardDelegator_ProcessDelete_Call) RunAndReturn(run func([]*DeleteData, uint64)) *MockShardDelegator_ProcessDelete_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -608,7 +608,7 @@ func (_c *MockShardDelegator_ProcessInsert_Call) Return() *MockShardDelegator_Pr } func (_c *MockShardDelegator_ProcessInsert_Call) RunAndReturn(run func(map[int64]*InsertData)) *MockShardDelegator_ProcessInsert_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -826,7 +826,7 @@ func (_c *MockShardDelegator_Search_Call) RunAndReturn(run func(context.Context, return _c } -// Serviceable provides a mock function with given fields: +// Serviceable provides a mock function with no fields func (_m *MockShardDelegator) Serviceable() bool { ret := _m.Called() @@ -871,7 +871,7 @@ func (_c *MockShardDelegator_Serviceable_Call) RunAndReturn(run func() bool) *Mo return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MockShardDelegator) Start() { _m.Called() } @@ -899,7 +899,7 @@ func (_c *MockShardDelegator_Start_Call) Return() *MockShardDelegator_Start_Call } func (_c *MockShardDelegator_Start_Call) RunAndReturn(run func()) *MockShardDelegator_Start_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -947,7 +947,7 @@ func (_c *MockShardDelegator_SyncDistribution_Call) Return() *MockShardDelegator } func (_c *MockShardDelegator_SyncDistribution_Call) RunAndReturn(run func(context.Context, ...SegmentEntry)) *MockShardDelegator_SyncDistribution_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -981,7 +981,7 @@ func (_c *MockShardDelegator_SyncPartitionStats_Call) Return() *MockShardDelegat } func (_c *MockShardDelegator_SyncPartitionStats_Call) RunAndReturn(run func(context.Context, map[int64]int64)) *MockShardDelegator_SyncPartitionStats_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1019,7 +1019,7 @@ func (_c *MockShardDelegator_SyncTargetVersion_Call) Return() *MockShardDelegato } func (_c *MockShardDelegator_SyncTargetVersion_Call) RunAndReturn(run func(int64, []int64, []int64, []int64, []int64, *msgpb.MsgPosition)) *MockShardDelegator_SyncTargetVersion_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1052,7 +1052,7 @@ func (_c *MockShardDelegator_TryCleanExcludedSegments_Call) Return() *MockShardD } func (_c *MockShardDelegator_TryCleanExcludedSegments_Call) RunAndReturn(run func(uint64)) *MockShardDelegator_TryCleanExcludedSegments_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1085,7 +1085,7 @@ func (_c *MockShardDelegator_UpdateTSafe_Call) Return() *MockShardDelegator_Upda } func (_c *MockShardDelegator_UpdateTSafe_Call) RunAndReturn(run func(uint64)) *MockShardDelegator_UpdateTSafe_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1136,7 +1136,7 @@ func (_c *MockShardDelegator_VerifyExcludedSegments_Call) RunAndReturn(run func( return _c } -// Version provides a mock function with given fields: +// Version provides a mock function with no fields func (_m *MockShardDelegator) Version() int64 { ret := _m.Called() diff --git a/internal/querynodev2/segments/mock_collection_manager.go b/internal/querynodev2/segments/mock_collection_manager.go index fe8368a514..c41829447b 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package segments @@ -71,7 +71,7 @@ func (_c *MockCollectionManager_Get_Call) RunAndReturn(run func(int64) *Collecti return _c } -// List provides a mock function with given fields: +// List provides a mock function with no fields func (_m *MockCollectionManager) List() []int64 { ret := _m.Called() @@ -118,7 +118,7 @@ func (_c *MockCollectionManager_List_Call) RunAndReturn(run func() []int64) *Moc return _c } -// ListWithName provides a mock function with given fields: +// ListWithName provides a mock function with no fields func (_m *MockCollectionManager) ListWithName() map[int64]string { ret := _m.Called() diff --git a/internal/querynodev2/segments/mock_loader.go b/internal/querynodev2/segments/mock_loader.go index f304836fc3..a619ca21c5 100644 --- a/internal/querynodev2/segments/mock_loader.go +++ b/internal/querynodev2/segments/mock_loader.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package segments @@ -33,11 +33,11 @@ func (_m *MockLoader) EXPECT() *MockLoader_Expecter { return &MockLoader_Expecter{mock: &_m.Mock} } -// Load provides a mock function with given fields: ctx, collectionID, segmentType, version, segments -func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, segments ...*querypb.SegmentLoadInfo) ([]Segment, error) { - _va := make([]interface{}, len(segments)) - for _i := range segments { - _va[_i] = segments[_i] +// Load provides a mock function with given fields: ctx, collectionID, segmentType, version, _a4 +func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, _a4 ...*querypb.SegmentLoadInfo) ([]Segment, error) { + _va := make([]interface{}, len(_a4)) + for _i := range _a4 { + _va[_i] = _a4[_i] } var _ca []interface{} _ca = append(_ca, ctx, collectionID, segmentType, version) @@ -51,10 +51,10 @@ func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType var r0 []Segment var r1 error if rf, ok := ret.Get(0).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) ([]Segment, error)); ok { - return rf(ctx, collectionID, segmentType, version, segments...) + return rf(ctx, collectionID, segmentType, version, _a4...) } if rf, ok := ret.Get(0).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) []Segment); ok { - r0 = rf(ctx, collectionID, segmentType, version, segments...) + r0 = rf(ctx, collectionID, segmentType, version, _a4...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]Segment) @@ -62,7 +62,7 @@ func (_m *MockLoader) Load(ctx context.Context, collectionID int64, segmentType } if rf, ok := ret.Get(1).(func(context.Context, int64, commonpb.SegmentState, int64, ...*querypb.SegmentLoadInfo) error); ok { - r1 = rf(ctx, collectionID, segmentType, version, segments...) + r1 = rf(ctx, collectionID, segmentType, version, _a4...) } else { r1 = ret.Error(1) } @@ -80,13 +80,13 @@ type MockLoader_Load_Call struct { // - collectionID int64 // - segmentType commonpb.SegmentState // - version int64 -// - segments ...*querypb.SegmentLoadInfo -func (_e *MockLoader_Expecter) Load(ctx interface{}, collectionID interface{}, segmentType interface{}, version interface{}, segments ...interface{}) *MockLoader_Load_Call { +// - _a4 ...*querypb.SegmentLoadInfo +func (_e *MockLoader_Expecter) Load(ctx interface{}, collectionID interface{}, segmentType interface{}, version interface{}, _a4 ...interface{}) *MockLoader_Load_Call { return &MockLoader_Load_Call{Call: _e.mock.On("Load", - append([]interface{}{ctx, collectionID, segmentType, version}, segments...)...)} + append([]interface{}{ctx, collectionID, segmentType, version}, _a4...)...)} } -func (_c *MockLoader_Load_Call) Run(run func(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, segments ...*querypb.SegmentLoadInfo)) *MockLoader_Load_Call { +func (_c *MockLoader_Load_Call) Run(run func(ctx context.Context, collectionID int64, segmentType commonpb.SegmentState, version int64, _a4 ...*querypb.SegmentLoadInfo)) *MockLoader_Load_Call { _c.Call.Run(func(args mock.Arguments) { variadicArgs := make([]*querypb.SegmentLoadInfo, len(args)-4) for i, a := range args[4:] { diff --git a/internal/querynodev2/segments/mock_segment.go b/internal/querynodev2/segments/mock_segment.go index 360d27a3ad..94150c257a 100644 --- a/internal/querynodev2/segments/mock_segment.go +++ b/internal/querynodev2/segments/mock_segment.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package segments @@ -132,7 +132,7 @@ func (_c *MockSegment_CASVersion_Call) RunAndReturn(run func(int64, int64) bool) return _c } -// Collection provides a mock function with given fields: +// Collection provides a mock function with no fields func (_m *MockSegment) Collection() int64 { ret := _m.Called() @@ -177,7 +177,7 @@ func (_c *MockSegment_Collection_Call) RunAndReturn(run func() int64) *MockSegme return _c } -// DatabaseName provides a mock function with given fields: +// DatabaseName provides a mock function with no fields func (_m *MockSegment) DatabaseName() string { ret := _m.Called() @@ -316,7 +316,7 @@ func (_c *MockSegment_ExistIndex_Call) RunAndReturn(run func(int64) bool) *MockS return _c } -// GetBM25Stats provides a mock function with given fields: +// GetBM25Stats provides a mock function with no fields func (_m *MockSegment) GetBM25Stats() map[int64]*storage.BM25Stats { ret := _m.Called() @@ -363,7 +363,7 @@ func (_c *MockSegment_GetBM25Stats_Call) RunAndReturn(run func() map[int64]*stor return _c } -// GetFieldJSONIndexStats provides a mock function with given fields: +// GetFieldJSONIndexStats provides a mock function with no fields func (_m *MockSegment) GetFieldJSONIndexStats() []int64 { ret := _m.Called() @@ -552,7 +552,7 @@ func (_c *MockSegment_HasRawData_Call) RunAndReturn(run func(int64) bool) *MockS return _c } -// ID provides a mock function with given fields: +// ID provides a mock function with no fields func (_m *MockSegment) ID() int64 { ret := _m.Called() @@ -597,7 +597,7 @@ func (_c *MockSegment_ID_Call) RunAndReturn(run func() int64) *MockSegment_ID_Ca return _c } -// Indexes provides a mock function with given fields: +// Indexes provides a mock function with no fields func (_m *MockSegment) Indexes() []*IndexedFieldInfo { ret := _m.Called() @@ -693,7 +693,7 @@ func (_c *MockSegment_Insert_Call) RunAndReturn(run func(context.Context, []int6 return _c } -// InsertCount provides a mock function with given fields: +// InsertCount provides a mock function with no fields func (_m *MockSegment) InsertCount() int64 { ret := _m.Called() @@ -738,7 +738,7 @@ func (_c *MockSegment_InsertCount_Call) RunAndReturn(run func() int64) *MockSegm return _c } -// IsLazyLoad provides a mock function with given fields: +// IsLazyLoad provides a mock function with no fields func (_m *MockSegment) IsLazyLoad() bool { ret := _m.Called() @@ -783,7 +783,7 @@ func (_c *MockSegment_IsLazyLoad_Call) RunAndReturn(run func() bool) *MockSegmen return _c } -// IsSorted provides a mock function with given fields: +// IsSorted provides a mock function with no fields func (_m *MockSegment) IsSorted() bool { ret := _m.Called() @@ -828,7 +828,7 @@ func (_c *MockSegment_IsSorted_Call) RunAndReturn(run func() bool) *MockSegment_ return _c } -// LastDeltaTimestamp provides a mock function with given fields: +// LastDeltaTimestamp provides a mock function with no fields func (_m *MockSegment) LastDeltaTimestamp() uint64 { ret := _m.Called() @@ -873,7 +873,7 @@ func (_c *MockSegment_LastDeltaTimestamp_Call) RunAndReturn(run func() uint64) * return _c } -// Level provides a mock function with given fields: +// Level provides a mock function with no fields func (_m *MockSegment) Level() datapb.SegmentLevel { ret := _m.Called() @@ -965,7 +965,7 @@ func (_c *MockSegment_LoadDeltaData_Call) RunAndReturn(run func(context.Context, return _c } -// LoadInfo provides a mock function with given fields: +// LoadInfo provides a mock function with no fields func (_m *MockSegment) LoadInfo() *querypb.SegmentLoadInfo { ret := _m.Called() @@ -1058,7 +1058,7 @@ func (_c *MockSegment_MayPkExist_Call) RunAndReturn(run func(*storage.LocationsC return _c } -// MemSize provides a mock function with given fields: +// MemSize provides a mock function with no fields func (_m *MockSegment) MemSize() int64 { ret := _m.Called() @@ -1103,7 +1103,7 @@ func (_c *MockSegment_MemSize_Call) RunAndReturn(run func() int64) *MockSegment_ return _c } -// NeedUpdatedVersion provides a mock function with given fields: +// NeedUpdatedVersion provides a mock function with no fields func (_m *MockSegment) NeedUpdatedVersion() int64 { ret := _m.Called() @@ -1148,7 +1148,7 @@ func (_c *MockSegment_NeedUpdatedVersion_Call) RunAndReturn(run func() int64) *M return _c } -// Partition provides a mock function with given fields: +// Partition provides a mock function with no fields func (_m *MockSegment) Partition() int64 { ret := _m.Called() @@ -1193,7 +1193,7 @@ func (_c *MockSegment_Partition_Call) RunAndReturn(run func() int64) *MockSegmen return _c } -// PinIfNotReleased provides a mock function with given fields: +// PinIfNotReleased provides a mock function with no fields func (_m *MockSegment) PinIfNotReleased() error { ret := _m.Called() @@ -1282,11 +1282,11 @@ func (_c *MockSegment_Release_Call) Return() *MockSegment_Release_Call { } func (_c *MockSegment_Release_Call) RunAndReturn(run func(context.Context, ...releaseOption)) *MockSegment_Release_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// RemoveUnusedFieldFiles provides a mock function with given fields: +// RemoveUnusedFieldFiles provides a mock function with no fields func (_m *MockSegment) RemoveUnusedFieldFiles() error { ret := _m.Called() @@ -1360,11 +1360,11 @@ func (_c *MockSegment_ResetIndexesLazyLoad_Call) Return() *MockSegment_ResetInde } func (_c *MockSegment_ResetIndexesLazyLoad_Call) RunAndReturn(run func(bool)) *MockSegment_ResetIndexesLazyLoad_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// ResourceGroup provides a mock function with given fields: +// ResourceGroup provides a mock function with no fields func (_m *MockSegment) ResourceGroup() string { ret := _m.Called() @@ -1409,7 +1409,7 @@ func (_c *MockSegment_ResourceGroup_Call) RunAndReturn(run func() string) *MockS return _c } -// ResourceUsageEstimate provides a mock function with given fields: +// ResourceUsageEstimate provides a mock function with no fields func (_m *MockSegment) ResourceUsageEstimate() ResourceUsage { ret := _m.Called() @@ -1572,7 +1572,7 @@ func (_c *MockSegment_RetrieveByOffsets_Call) RunAndReturn(run func(context.Cont return _c } -// RowNum provides a mock function with given fields: +// RowNum provides a mock function with no fields func (_m *MockSegment) RowNum() int64 { ret := _m.Called() @@ -1676,7 +1676,7 @@ func (_c *MockSegment_Search_Call) RunAndReturn(run func(context.Context, *segco return _c } -// Shard provides a mock function with given fields: +// Shard provides a mock function with no fields func (_m *MockSegment) Shard() metautil.Channel { ret := _m.Called() @@ -1721,7 +1721,7 @@ func (_c *MockSegment_Shard_Call) RunAndReturn(run func() metautil.Channel) *Moc return _c } -// StartPosition provides a mock function with given fields: +// StartPosition provides a mock function with no fields func (_m *MockSegment) StartPosition() *msgpb.MsgPosition { ret := _m.Called() @@ -1768,7 +1768,7 @@ func (_c *MockSegment_StartPosition_Call) RunAndReturn(run func() *msgpb.MsgPosi return _c } -// Type provides a mock function with given fields: +// Type provides a mock function with no fields func (_m *MockSegment) Type() commonpb.SegmentState { ret := _m.Called() @@ -1813,7 +1813,7 @@ func (_c *MockSegment_Type_Call) RunAndReturn(run func() commonpb.SegmentState) return _c } -// Unpin provides a mock function with given fields: +// Unpin provides a mock function with no fields func (_m *MockSegment) Unpin() { _m.Called() } @@ -1841,7 +1841,7 @@ func (_c *MockSegment_Unpin_Call) Return() *MockSegment_Unpin_Call { } func (_c *MockSegment_Unpin_Call) RunAndReturn(run func()) *MockSegment_Unpin_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1874,7 +1874,7 @@ func (_c *MockSegment_UpdateBM25Stats_Call) Return() *MockSegment_UpdateBM25Stat } func (_c *MockSegment_UpdateBM25Stats_Call) RunAndReturn(run func(map[int64]*storage.BM25Stats)) *MockSegment_UpdateBM25Stats_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1907,11 +1907,11 @@ func (_c *MockSegment_UpdateBloomFilter_Call) Return() *MockSegment_UpdateBloomF } func (_c *MockSegment_UpdateBloomFilter_Call) RunAndReturn(run func([]storage.PrimaryKey)) *MockSegment_UpdateBloomFilter_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Version provides a mock function with given fields: +// Version provides a mock function with no fields func (_m *MockSegment) Version() int64 { ret := _m.Called() diff --git a/internal/querynodev2/segments/mock_segment_manager.go b/internal/querynodev2/segments/mock_segment_manager.go index 4d6cbe2579..b069a54926 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package segments @@ -54,11 +54,11 @@ func (_c *MockSegmentManager_Clear_Call) Return() *MockSegmentManager_Clear_Call } func (_c *MockSegmentManager_Clear_Call) RunAndReturn(run func(context.Context)) *MockSegmentManager_Clear_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// Empty provides a mock function with given fields: +// Empty provides a mock function with no fields func (_m *MockSegmentManager) Empty() bool { ret := _m.Called() @@ -198,14 +198,14 @@ func (_c *MockSegmentManager_Get_Call) RunAndReturn(run func(int64) Segment) *Mo return _c } -// GetAndPin provides a mock function with given fields: segments, filters -func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilter) ([]Segment, error) { +// GetAndPin provides a mock function with given fields: _a0, filters +func (_m *MockSegmentManager) GetAndPin(_a0 []int64, filters ...SegmentFilter) ([]Segment, error) { _va := make([]interface{}, len(filters)) for _i := range filters { _va[_i] = filters[_i] } var _ca []interface{} - _ca = append(_ca, segments) + _ca = append(_ca, _a0) _ca = append(_ca, _va...) ret := _m.Called(_ca...) @@ -216,10 +216,10 @@ func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilt var r0 []Segment var r1 error if rf, ok := ret.Get(0).(func([]int64, ...SegmentFilter) ([]Segment, error)); ok { - return rf(segments, filters...) + return rf(_a0, filters...) } if rf, ok := ret.Get(0).(func([]int64, ...SegmentFilter) []Segment); ok { - r0 = rf(segments, filters...) + r0 = rf(_a0, filters...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]Segment) @@ -227,7 +227,7 @@ func (_m *MockSegmentManager) GetAndPin(segments []int64, filters ...SegmentFilt } if rf, ok := ret.Get(1).(func([]int64, ...SegmentFilter) error); ok { - r1 = rf(segments, filters...) + r1 = rf(_a0, filters...) } else { r1 = ret.Error(1) } @@ -241,14 +241,14 @@ type MockSegmentManager_GetAndPin_Call struct { } // GetAndPin is a helper method to define mock.On call -// - segments []int64 +// - _a0 []int64 // - filters ...SegmentFilter -func (_e *MockSegmentManager_Expecter) GetAndPin(segments interface{}, filters ...interface{}) *MockSegmentManager_GetAndPin_Call { +func (_e *MockSegmentManager_Expecter) GetAndPin(_a0 interface{}, filters ...interface{}) *MockSegmentManager_GetAndPin_Call { return &MockSegmentManager_GetAndPin_Call{Call: _e.mock.On("GetAndPin", - append([]interface{}{segments}, filters...)...)} + append([]interface{}{_a0}, filters...)...)} } -func (_c *MockSegmentManager_GetAndPin_Call) Run(run func(segments []int64, filters ...SegmentFilter)) *MockSegmentManager_GetAndPin_Call { +func (_c *MockSegmentManager_GetAndPin_Call) Run(run func(_a0 []int64, filters ...SegmentFilter)) *MockSegmentManager_GetAndPin_Call { _c.Call.Run(func(args mock.Arguments) { variadicArgs := make([]SegmentFilter, len(args)-1) for i, a := range args[1:] { @@ -548,11 +548,11 @@ func (_c *MockSegmentManager_GetWithType_Call) RunAndReturn(run func(int64, comm return _c } -// Put provides a mock function with given fields: ctx, segmentType, segments -func (_m *MockSegmentManager) Put(ctx context.Context, segmentType commonpb.SegmentState, segments ...Segment) { - _va := make([]interface{}, len(segments)) - for _i := range segments { - _va[_i] = segments[_i] +// Put provides a mock function with given fields: ctx, segmentType, _a2 +func (_m *MockSegmentManager) Put(ctx context.Context, segmentType commonpb.SegmentState, _a2 ...Segment) { + _va := make([]interface{}, len(_a2)) + for _i := range _a2 { + _va[_i] = _a2[_i] } var _ca []interface{} _ca = append(_ca, ctx, segmentType) @@ -568,13 +568,13 @@ type MockSegmentManager_Put_Call struct { // Put is a helper method to define mock.On call // - ctx context.Context // - segmentType commonpb.SegmentState -// - segments ...Segment -func (_e *MockSegmentManager_Expecter) Put(ctx interface{}, segmentType interface{}, segments ...interface{}) *MockSegmentManager_Put_Call { +// - _a2 ...Segment +func (_e *MockSegmentManager_Expecter) Put(ctx interface{}, segmentType interface{}, _a2 ...interface{}) *MockSegmentManager_Put_Call { return &MockSegmentManager_Put_Call{Call: _e.mock.On("Put", - append([]interface{}{ctx, segmentType}, segments...)...)} + append([]interface{}{ctx, segmentType}, _a2...)...)} } -func (_c *MockSegmentManager_Put_Call) Run(run func(ctx context.Context, segmentType commonpb.SegmentState, segments ...Segment)) *MockSegmentManager_Put_Call { +func (_c *MockSegmentManager_Put_Call) Run(run func(ctx context.Context, segmentType commonpb.SegmentState, _a2 ...Segment)) *MockSegmentManager_Put_Call { _c.Call.Run(func(args mock.Arguments) { variadicArgs := make([]Segment, len(args)-2) for i, a := range args[2:] { @@ -593,7 +593,7 @@ func (_c *MockSegmentManager_Put_Call) Return() *MockSegmentManager_Put_Call { } func (_c *MockSegmentManager_Put_Call) RunAndReturn(run func(context.Context, commonpb.SegmentState, ...Segment)) *MockSegmentManager_Put_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -726,9 +726,9 @@ func (_c *MockSegmentManager_RemoveBy_Call) RunAndReturn(run func(context.Contex return _c } -// Unpin provides a mock function with given fields: segments -func (_m *MockSegmentManager) Unpin(segments []Segment) { - _m.Called(segments) +// Unpin provides a mock function with given fields: _a0 +func (_m *MockSegmentManager) Unpin(_a0 []Segment) { + _m.Called(_a0) } // MockSegmentManager_Unpin_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Unpin' @@ -737,12 +737,12 @@ type MockSegmentManager_Unpin_Call struct { } // Unpin is a helper method to define mock.On call -// - segments []Segment -func (_e *MockSegmentManager_Expecter) Unpin(segments interface{}) *MockSegmentManager_Unpin_Call { - return &MockSegmentManager_Unpin_Call{Call: _e.mock.On("Unpin", segments)} +// - _a0 []Segment +func (_e *MockSegmentManager_Expecter) Unpin(_a0 interface{}) *MockSegmentManager_Unpin_Call { + return &MockSegmentManager_Unpin_Call{Call: _e.mock.On("Unpin", _a0)} } -func (_c *MockSegmentManager_Unpin_Call) Run(run func(segments []Segment)) *MockSegmentManager_Unpin_Call { +func (_c *MockSegmentManager_Unpin_Call) Run(run func(_a0 []Segment)) *MockSegmentManager_Unpin_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].([]Segment)) }) @@ -755,7 +755,7 @@ func (_c *MockSegmentManager_Unpin_Call) Return() *MockSegmentManager_Unpin_Call } func (_c *MockSegmentManager_Unpin_Call) RunAndReturn(run func([]Segment)) *MockSegmentManager_Unpin_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/rootcoord/mocks/garbage_collector.go b/internal/rootcoord/mocks/garbage_collector.go index 7c23345380..2e5dc366c6 100644 --- a/internal/rootcoord/mocks/garbage_collector.go +++ b/internal/rootcoord/mocks/garbage_collector.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mockrootcoord @@ -168,7 +168,7 @@ func (_c *GarbageCollector_ReDropCollection_Call) Return() *GarbageCollector_ReD } func (_c *GarbageCollector_ReDropCollection_Call) RunAndReturn(run func(*model.Collection, uint64)) *GarbageCollector_ReDropCollection_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -205,7 +205,7 @@ func (_c *GarbageCollector_ReDropPartition_Call) Return() *GarbageCollector_ReDr } func (_c *GarbageCollector_ReDropPartition_Call) RunAndReturn(run func(int64, []string, []string, *model.Partition, uint64)) *GarbageCollector_ReDropPartition_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -238,7 +238,7 @@ func (_c *GarbageCollector_RemoveCreatingCollection_Call) Return() *GarbageColle } func (_c *GarbageCollector_RemoveCreatingCollection_Call) RunAndReturn(run func(*model.Collection)) *GarbageCollector_RemoveCreatingCollection_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -273,7 +273,7 @@ func (_c *GarbageCollector_RemoveCreatingPartition_Call) Return() *GarbageCollec } func (_c *GarbageCollector_RemoveCreatingPartition_Call) RunAndReturn(run func(int64, *model.Partition, uint64)) *GarbageCollector_RemoveCreatingPartition_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/rootcoord/mocks/meta_table.go b/internal/rootcoord/mocks/meta_table.go index 95970ae88b..f51af94253 100644 --- a/internal/rootcoord/mocks/meta_table.go +++ b/internal/rootcoord/mocks/meta_table.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mockrootcoord diff --git a/pkg/eventlog/mock_logger.go b/pkg/eventlog/mock_logger.go index 41ded424f6..dd6d00a53e 100644 --- a/pkg/eventlog/mock_logger.go +++ b/pkg/eventlog/mock_logger.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package eventlog @@ -17,7 +17,7 @@ func (_m *MockLogger) EXPECT() *MockLogger_Expecter { return &MockLogger_Expecter{mock: &_m.Mock} } -// Flush provides a mock function with given fields: +// Flush provides a mock function with no fields func (_m *MockLogger) Flush() error { ret := _m.Called() @@ -91,7 +91,7 @@ func (_c *MockLogger_Record_Call) Return() *MockLogger_Record_Call { } func (_c *MockLogger_Record_Call) RunAndReturn(run func(Evt)) *MockLogger_Record_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -125,7 +125,7 @@ func (_c *MockLogger_RecordFunc_Call) Return() *MockLogger_RecordFunc_Call { } func (_c *MockLogger_RecordFunc_Call) RunAndReturn(run func(Level, func() Evt)) *MockLogger_RecordFunc_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/kv/predicates/mock_predicate.go b/pkg/kv/predicates/mock_predicate.go index eace7337f2..6fb5932a55 100644 --- a/pkg/kv/predicates/mock_predicate.go +++ b/pkg/kv/predicates/mock_predicate.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package predicates @@ -63,7 +63,7 @@ func (_c *MockPredicate_IsTrue_Call) RunAndReturn(run func(interface{}) bool) *M return _c } -// Key provides a mock function with given fields: +// Key provides a mock function with no fields func (_m *MockPredicate) Key() string { ret := _m.Called() @@ -108,7 +108,7 @@ func (_c *MockPredicate_Key_Call) RunAndReturn(run func() string) *MockPredicate return _c } -// Target provides a mock function with given fields: +// Target provides a mock function with no fields func (_m *MockPredicate) Target() PredicateTarget { ret := _m.Called() @@ -153,7 +153,7 @@ func (_c *MockPredicate_Target_Call) RunAndReturn(run func() PredicateTarget) *M return _c } -// TargetValue provides a mock function with given fields: +// TargetValue provides a mock function with no fields func (_m *MockPredicate) TargetValue() interface{} { ret := _m.Called() @@ -200,7 +200,7 @@ func (_c *MockPredicate_TargetValue_Call) RunAndReturn(run func() interface{}) * return _c } -// Type provides a mock function with given fields: +// Type provides a mock function with no fields func (_m *MockPredicate) Type() PredicateType { ret := _m.Called() diff --git a/pkg/mocks/mock_kv/mock_MetaKv.go b/pkg/mocks/mock_kv/mock_MetaKv.go index 4cda102275..7823040eff 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_kv @@ -23,7 +23,7 @@ func (_m *MockMetaKv) EXPECT() *MockMetaKv_Expecter { return &MockMetaKv_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockMetaKv) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *MockMetaKv_Close_Call) Return() *MockMetaKv_Close_Call { } func (_c *MockMetaKv_Close_Call) RunAndReturn(run func()) *MockMetaKv_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go index 56f9628554..75690c0d82 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go index 854635ecb6..9b1195d657 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb @@ -24,7 +24,7 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) EXPECT() return &MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Expecter{mock: &_m.Mock} } -// CloseSend provides a mock function with given fields: +// CloseSend provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) CloseSend() error { ret := _m.Called() @@ -69,7 +69,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_CloseSend return _c } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Context() context.Context { ret := _m.Called() @@ -116,7 +116,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Context_C return _c } -// Header provides a mock function with given fields: +// Header provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Header() (metadata.MD, error) { ret := _m.Called() @@ -173,7 +173,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_Header_Ca return _c } -// Recv provides a mock function with given fields: +// Recv provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Recv() (*streamingpb.AssignmentDiscoverResponse, error) { ret := _m.Called() @@ -368,7 +368,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverClient_SendMsg_C return _c } -// Trailer provides a mock function with given fields: +// Trailer provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverClient) Trailer() metadata.MD { ret := _m.Called() diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go index 2e276686e1..4954df384e 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb @@ -24,7 +24,7 @@ func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) EXPECT() return &MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Expecter{mock: &_m.Mock} } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Context() context.Context { ret := _m.Called() @@ -71,7 +71,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_Context_C return _c } -// Recv provides a mock function with given fields: +// Recv provides a mock function with no fields func (_m *MockStreamingCoordAssignmentService_AssignmentDiscoverServer) Recv() (*streamingpb.AssignmentDiscoverRequest, error) { ret := _m.Called() @@ -387,7 +387,7 @@ func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SetTraile } func (_c *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamingCoordAssignmentService_AssignmentDiscoverServer_SetTrailer_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordBroadcastServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordBroadcastServiceClient.go index a475026d41..c0fe15cd49 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordBroadcastServiceClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordBroadcastServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go index 04d5f2e857..158374ae41 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go index ec6de7974f..9e5ed6e5bf 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb @@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ConsumeClient) EXPECT() *MockStreaming return &MockStreamingNodeHandlerService_ConsumeClient_Expecter{mock: &_m.Mock} } -// CloseSend provides a mock function with given fields: +// CloseSend provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeClient) CloseSend() error { ret := _m.Called() @@ -69,7 +69,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_CloseSend_Call) RunAndRe return _c } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeClient) Context() context.Context { ret := _m.Called() @@ -116,7 +116,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Context_Call) RunAndRetu return _c } -// Header provides a mock function with given fields: +// Header provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeClient) Header() (metadata.MD, error) { ret := _m.Called() @@ -173,7 +173,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_Header_Call) RunAndRetur return _c } -// Recv provides a mock function with given fields: +// Recv provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeClient) Recv() (*streamingpb.ConsumeResponse, error) { ret := _m.Called() @@ -368,7 +368,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeClient_SendMsg_Call) RunAndRetu return _c } -// Trailer provides a mock function with given fields: +// Trailer provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeClient) Trailer() metadata.MD { ret := _m.Called() diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go index bc490bf31b..7e82825d7c 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb @@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ConsumeServer) EXPECT() *MockStreaming return &MockStreamingNodeHandlerService_ConsumeServer_Expecter{mock: &_m.Mock} } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeServer) Context() context.Context { ret := _m.Called() @@ -71,7 +71,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_Context_Call) RunAndRetu return _c } -// Recv provides a mock function with given fields: +// Recv provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ConsumeServer) Recv() (*streamingpb.ConsumeRequest, error) { ret := _m.Called() @@ -387,7 +387,7 @@ func (_c *MockStreamingNodeHandlerService_ConsumeServer_SetTrailer_Call) Return( } func (_c *MockStreamingNodeHandlerService_ConsumeServer_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamingNodeHandlerService_ConsumeServer_SetTrailer_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go index 53e21e0e80..ab0aad017c 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb @@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ProduceClient) EXPECT() *MockStreaming return &MockStreamingNodeHandlerService_ProduceClient_Expecter{mock: &_m.Mock} } -// CloseSend provides a mock function with given fields: +// CloseSend provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceClient) CloseSend() error { ret := _m.Called() @@ -69,7 +69,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_CloseSend_Call) RunAndRe return _c } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceClient) Context() context.Context { ret := _m.Called() @@ -116,7 +116,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Context_Call) RunAndRetu return _c } -// Header provides a mock function with given fields: +// Header provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceClient) Header() (metadata.MD, error) { ret := _m.Called() @@ -173,7 +173,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_Header_Call) RunAndRetur return _c } -// Recv provides a mock function with given fields: +// Recv provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceClient) Recv() (*streamingpb.ProduceResponse, error) { ret := _m.Called() @@ -368,7 +368,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceClient_SendMsg_Call) RunAndRetu return _c } -// Trailer provides a mock function with given fields: +// Trailer provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceClient) Trailer() metadata.MD { ret := _m.Called() diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go index e9e0953bcc..ac1b1495c2 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb @@ -24,7 +24,7 @@ func (_m *MockStreamingNodeHandlerService_ProduceServer) EXPECT() *MockStreaming return &MockStreamingNodeHandlerService_ProduceServer_Expecter{mock: &_m.Mock} } -// Context provides a mock function with given fields: +// Context provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceServer) Context() context.Context { ret := _m.Called() @@ -71,7 +71,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_Context_Call) RunAndRetu return _c } -// Recv provides a mock function with given fields: +// Recv provides a mock function with no fields func (_m *MockStreamingNodeHandlerService_ProduceServer) Recv() (*streamingpb.ProduceRequest, error) { ret := _m.Called() @@ -387,7 +387,7 @@ func (_c *MockStreamingNodeHandlerService_ProduceServer_SetTrailer_Call) Return( } func (_c *MockStreamingNodeHandlerService_ProduceServer_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamingNodeHandlerService_ProduceServer_SetTrailer_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go index 12ba55533f..86e4d41a9c 100644 --- a/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go +++ b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_streamingpb diff --git a/pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go index 70e5708eb1..bf091a466e 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_walimpls @@ -20,7 +20,7 @@ func (_m *MockOpenerBuilderImpls) EXPECT() *MockOpenerBuilderImpls_Expecter { return &MockOpenerBuilderImpls_Expecter{mock: &_m.Mock} } -// Build provides a mock function with given fields: +// Build provides a mock function with no fields func (_m *MockOpenerBuilderImpls) Build() (walimpls.OpenerImpls, error) { ret := _m.Called() @@ -77,7 +77,7 @@ func (_c *MockOpenerBuilderImpls_Build_Call) RunAndReturn(run func() (walimpls.O return _c } -// Name provides a mock function with given fields: +// Name provides a mock function with no fields func (_m *MockOpenerBuilderImpls) Name() string { ret := _m.Called() diff --git a/pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go index 1e7d91347d..d76775a2f7 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_walimpls @@ -22,7 +22,7 @@ func (_m *MockOpenerImpls) EXPECT() *MockOpenerImpls_Expecter { return &MockOpenerImpls_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockOpenerImpls) Close() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockOpenerImpls_Close_Call) Return() *MockOpenerImpls_Close_Call { } func (_c *MockOpenerImpls_Close_Call) RunAndReturn(run func()) *MockOpenerImpls_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go index 45c3a4ae79..2ab9affe06 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_walimpls @@ -20,7 +20,7 @@ func (_m *MockScannerImpls) EXPECT() *MockScannerImpls_Expecter { return &MockScannerImpls_Expecter{mock: &_m.Mock} } -// Chan provides a mock function with given fields: +// Chan provides a mock function with no fields func (_m *MockScannerImpls) Chan() <-chan message.ImmutableMessage { ret := _m.Called() @@ -67,7 +67,7 @@ func (_c *MockScannerImpls_Chan_Call) RunAndReturn(run func() <-chan message.Imm return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockScannerImpls) Close() error { ret := _m.Called() @@ -112,7 +112,7 @@ func (_c *MockScannerImpls_Close_Call) RunAndReturn(run func() error) *MockScann return _c } -// Done provides a mock function with given fields: +// Done provides a mock function with no fields func (_m *MockScannerImpls) Done() <-chan struct{} { ret := _m.Called() @@ -159,7 +159,7 @@ func (_c *MockScannerImpls_Done_Call) RunAndReturn(run func() <-chan struct{}) * return _c } -// Error provides a mock function with given fields: +// Error provides a mock function with no fields func (_m *MockScannerImpls) Error() error { ret := _m.Called() @@ -204,7 +204,7 @@ func (_c *MockScannerImpls_Error_Call) RunAndReturn(run func() error) *MockScann return _c } -// Name provides a mock function with given fields: +// Name provides a mock function with no fields func (_m *MockScannerImpls) Name() string { ret := _m.Called() diff --git a/pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go b/pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go index d898714d84..9f627489a2 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_walimpls @@ -85,7 +85,7 @@ func (_c *MockWALImpls_Append_Call) RunAndReturn(run func(context.Context, messa return _c } -// Channel provides a mock function with given fields: +// Channel provides a mock function with no fields func (_m *MockWALImpls) Channel() types.PChannelInfo { ret := _m.Called() @@ -130,7 +130,7 @@ func (_c *MockWALImpls_Channel_Call) RunAndReturn(run func() types.PChannelInfo) return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockWALImpls) Close() { _m.Called() } @@ -158,7 +158,7 @@ func (_c *MockWALImpls_Close_Call) Return() *MockWALImpls_Close_Call { } func (_c *MockWALImpls_Close_Call) RunAndReturn(run func()) *MockWALImpls_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -221,7 +221,7 @@ func (_c *MockWALImpls_Read_Call) RunAndReturn(run func(context.Context, walimpl return _c } -// WALName provides a mock function with given fields: +// WALName provides a mock function with no fields func (_m *MockWALImpls) WALName() string { ret := _m.Called() diff --git a/pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go b/pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go index 54081ac876..616dd17850 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_message @@ -20,7 +20,7 @@ func (_m *MockImmutableMessage) EXPECT() *MockImmutableMessage_Expecter { return &MockImmutableMessage_Expecter{mock: &_m.Mock} } -// BarrierTimeTick provides a mock function with given fields: +// BarrierTimeTick provides a mock function with no fields func (_m *MockImmutableMessage) BarrierTimeTick() uint64 { ret := _m.Called() @@ -65,7 +65,7 @@ func (_c *MockImmutableMessage_BarrierTimeTick_Call) RunAndReturn(run func() uin return _c } -// BroadcastHeader provides a mock function with given fields: +// BroadcastHeader provides a mock function with no fields func (_m *MockImmutableMessage) BroadcastHeader() *message.BroadcastHeader { ret := _m.Called() @@ -112,7 +112,7 @@ func (_c *MockImmutableMessage_BroadcastHeader_Call) RunAndReturn(run func() *me return _c } -// EstimateSize provides a mock function with given fields: +// EstimateSize provides a mock function with no fields func (_m *MockImmutableMessage) EstimateSize() int { ret := _m.Called() @@ -157,7 +157,7 @@ func (_c *MockImmutableMessage_EstimateSize_Call) RunAndReturn(run func() int) * return _c } -// LastConfirmedMessageID provides a mock function with given fields: +// LastConfirmedMessageID provides a mock function with no fields func (_m *MockImmutableMessage) LastConfirmedMessageID() message.MessageID { ret := _m.Called() @@ -204,7 +204,7 @@ func (_c *MockImmutableMessage_LastConfirmedMessageID_Call) RunAndReturn(run fun return _c } -// MessageID provides a mock function with given fields: +// MessageID provides a mock function with no fields func (_m *MockImmutableMessage) MessageID() message.MessageID { ret := _m.Called() @@ -251,7 +251,7 @@ func (_c *MockImmutableMessage_MessageID_Call) RunAndReturn(run func() message.M return _c } -// MessageType provides a mock function with given fields: +// MessageType provides a mock function with no fields func (_m *MockImmutableMessage) MessageType() message.MessageType { ret := _m.Called() @@ -296,7 +296,7 @@ func (_c *MockImmutableMessage_MessageType_Call) RunAndReturn(run func() message return _c } -// Payload provides a mock function with given fields: +// Payload provides a mock function with no fields func (_m *MockImmutableMessage) Payload() []byte { ret := _m.Called() @@ -343,7 +343,7 @@ func (_c *MockImmutableMessage_Payload_Call) RunAndReturn(run func() []byte) *Mo return _c } -// Properties provides a mock function with given fields: +// Properties provides a mock function with no fields func (_m *MockImmutableMessage) Properties() message.RProperties { ret := _m.Called() @@ -390,7 +390,7 @@ func (_c *MockImmutableMessage_Properties_Call) RunAndReturn(run func() message. return _c } -// TimeTick provides a mock function with given fields: +// TimeTick provides a mock function with no fields func (_m *MockImmutableMessage) TimeTick() uint64 { ret := _m.Called() @@ -435,7 +435,7 @@ func (_c *MockImmutableMessage_TimeTick_Call) RunAndReturn(run func() uint64) *M return _c } -// TxnContext provides a mock function with given fields: +// TxnContext provides a mock function with no fields func (_m *MockImmutableMessage) TxnContext() *message.TxnContext { ret := _m.Called() @@ -482,7 +482,7 @@ func (_c *MockImmutableMessage_TxnContext_Call) RunAndReturn(run func() *message return _c } -// VChannel provides a mock function with given fields: +// VChannel provides a mock function with no fields func (_m *MockImmutableMessage) VChannel() string { ret := _m.Called() @@ -527,7 +527,7 @@ func (_c *MockImmutableMessage_VChannel_Call) RunAndReturn(run func() string) *M return _c } -// Version provides a mock function with given fields: +// Version provides a mock function with no fields func (_m *MockImmutableMessage) Version() message.Version { ret := _m.Called() @@ -572,7 +572,7 @@ func (_c *MockImmutableMessage_Version_Call) RunAndReturn(run func() message.Ver return _c } -// WALName provides a mock function with given fields: +// WALName provides a mock function with no fields func (_m *MockImmutableMessage) WALName() string { ret := _m.Called() diff --git a/pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go b/pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go index 59366fadb9..1bab47e8a1 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_message @@ -20,7 +20,7 @@ func (_m *MockImmutableTxnMessage) EXPECT() *MockImmutableTxnMessage_Expecter { return &MockImmutableTxnMessage_Expecter{mock: &_m.Mock} } -// BarrierTimeTick provides a mock function with given fields: +// BarrierTimeTick provides a mock function with no fields func (_m *MockImmutableTxnMessage) BarrierTimeTick() uint64 { ret := _m.Called() @@ -65,7 +65,7 @@ func (_c *MockImmutableTxnMessage_BarrierTimeTick_Call) RunAndReturn(run func() return _c } -// Begin provides a mock function with given fields: +// Begin provides a mock function with no fields func (_m *MockImmutableTxnMessage) Begin() message.ImmutableMessage { ret := _m.Called() @@ -112,7 +112,7 @@ func (_c *MockImmutableTxnMessage_Begin_Call) RunAndReturn(run func() message.Im return _c } -// BroadcastHeader provides a mock function with given fields: +// BroadcastHeader provides a mock function with no fields func (_m *MockImmutableTxnMessage) BroadcastHeader() *message.BroadcastHeader { ret := _m.Called() @@ -159,7 +159,7 @@ func (_c *MockImmutableTxnMessage_BroadcastHeader_Call) RunAndReturn(run func() return _c } -// Commit provides a mock function with given fields: +// Commit provides a mock function with no fields func (_m *MockImmutableTxnMessage) Commit() message.ImmutableMessage { ret := _m.Called() @@ -206,7 +206,7 @@ func (_c *MockImmutableTxnMessage_Commit_Call) RunAndReturn(run func() message.I return _c } -// EstimateSize provides a mock function with given fields: +// EstimateSize provides a mock function with no fields func (_m *MockImmutableTxnMessage) EstimateSize() int { ret := _m.Called() @@ -251,7 +251,7 @@ func (_c *MockImmutableTxnMessage_EstimateSize_Call) RunAndReturn(run func() int return _c } -// LastConfirmedMessageID provides a mock function with given fields: +// LastConfirmedMessageID provides a mock function with no fields func (_m *MockImmutableTxnMessage) LastConfirmedMessageID() message.MessageID { ret := _m.Called() @@ -298,7 +298,7 @@ func (_c *MockImmutableTxnMessage_LastConfirmedMessageID_Call) RunAndReturn(run return _c } -// MessageID provides a mock function with given fields: +// MessageID provides a mock function with no fields func (_m *MockImmutableTxnMessage) MessageID() message.MessageID { ret := _m.Called() @@ -345,7 +345,7 @@ func (_c *MockImmutableTxnMessage_MessageID_Call) RunAndReturn(run func() messag return _c } -// MessageType provides a mock function with given fields: +// MessageType provides a mock function with no fields func (_m *MockImmutableTxnMessage) MessageType() message.MessageType { ret := _m.Called() @@ -390,7 +390,7 @@ func (_c *MockImmutableTxnMessage_MessageType_Call) RunAndReturn(run func() mess return _c } -// Payload provides a mock function with given fields: +// Payload provides a mock function with no fields func (_m *MockImmutableTxnMessage) Payload() []byte { ret := _m.Called() @@ -437,7 +437,7 @@ func (_c *MockImmutableTxnMessage_Payload_Call) RunAndReturn(run func() []byte) return _c } -// Properties provides a mock function with given fields: +// Properties provides a mock function with no fields func (_m *MockImmutableTxnMessage) Properties() message.RProperties { ret := _m.Called() @@ -530,7 +530,7 @@ func (_c *MockImmutableTxnMessage_RangeOver_Call) RunAndReturn(run func(func(mes return _c } -// Size provides a mock function with given fields: +// Size provides a mock function with no fields func (_m *MockImmutableTxnMessage) Size() int { ret := _m.Called() @@ -575,7 +575,7 @@ func (_c *MockImmutableTxnMessage_Size_Call) RunAndReturn(run func() int) *MockI return _c } -// TimeTick provides a mock function with given fields: +// TimeTick provides a mock function with no fields func (_m *MockImmutableTxnMessage) TimeTick() uint64 { ret := _m.Called() @@ -620,7 +620,7 @@ func (_c *MockImmutableTxnMessage_TimeTick_Call) RunAndReturn(run func() uint64) return _c } -// TxnContext provides a mock function with given fields: +// TxnContext provides a mock function with no fields func (_m *MockImmutableTxnMessage) TxnContext() *message.TxnContext { ret := _m.Called() @@ -667,7 +667,7 @@ func (_c *MockImmutableTxnMessage_TxnContext_Call) RunAndReturn(run func() *mess return _c } -// VChannel provides a mock function with given fields: +// VChannel provides a mock function with no fields func (_m *MockImmutableTxnMessage) VChannel() string { ret := _m.Called() @@ -712,7 +712,7 @@ func (_c *MockImmutableTxnMessage_VChannel_Call) RunAndReturn(run func() string) return _c } -// Version provides a mock function with given fields: +// Version provides a mock function with no fields func (_m *MockImmutableTxnMessage) Version() message.Version { ret := _m.Called() @@ -757,7 +757,7 @@ func (_c *MockImmutableTxnMessage_Version_Call) RunAndReturn(run func() message. return _c } -// WALName provides a mock function with given fields: +// WALName provides a mock function with no fields func (_m *MockImmutableTxnMessage) WALName() string { ret := _m.Called() diff --git a/pkg/mocks/streaming/util/mock_message/mock_MessageID.go b/pkg/mocks/streaming/util/mock_message/mock_MessageID.go index 080ab742e8..b3fc0cd7ac 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_message @@ -158,7 +158,7 @@ func (_c *MockMessageID_LTE_Call) RunAndReturn(run func(message.MessageID) bool) return _c } -// Marshal provides a mock function with given fields: +// Marshal provides a mock function with no fields func (_m *MockMessageID) Marshal() string { ret := _m.Called() @@ -203,7 +203,7 @@ func (_c *MockMessageID_Marshal_Call) RunAndReturn(run func() string) *MockMessa return _c } -// String provides a mock function with given fields: +// String provides a mock function with no fields func (_m *MockMessageID) String() string { ret := _m.Called() @@ -248,7 +248,7 @@ func (_c *MockMessageID_String_Call) RunAndReturn(run func() string) *MockMessag return _c } -// WALName provides a mock function with given fields: +// WALName provides a mock function with no fields func (_m *MockMessageID) WALName() string { ret := _m.Called() diff --git a/pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go b/pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go index f38ef13d09..a7c3cd4b98 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_message @@ -20,7 +20,7 @@ func (_m *MockMutableMessage) EXPECT() *MockMutableMessage_Expecter { return &MockMutableMessage_Expecter{mock: &_m.Mock} } -// BarrierTimeTick provides a mock function with given fields: +// BarrierTimeTick provides a mock function with no fields func (_m *MockMutableMessage) BarrierTimeTick() uint64 { ret := _m.Called() @@ -65,7 +65,7 @@ func (_c *MockMutableMessage_BarrierTimeTick_Call) RunAndReturn(run func() uint6 return _c } -// BroadcastHeader provides a mock function with given fields: +// BroadcastHeader provides a mock function with no fields func (_m *MockMutableMessage) BroadcastHeader() *message.BroadcastHeader { ret := _m.Called() @@ -112,7 +112,7 @@ func (_c *MockMutableMessage_BroadcastHeader_Call) RunAndReturn(run func() *mess return _c } -// EstimateSize provides a mock function with given fields: +// EstimateSize provides a mock function with no fields func (_m *MockMutableMessage) EstimateSize() int { ret := _m.Called() @@ -205,7 +205,7 @@ func (_c *MockMutableMessage_IntoImmutableMessage_Call) RunAndReturn(run func(me return _c } -// MessageType provides a mock function with given fields: +// MessageType provides a mock function with no fields func (_m *MockMutableMessage) MessageType() message.MessageType { ret := _m.Called() @@ -250,7 +250,7 @@ func (_c *MockMutableMessage_MessageType_Call) RunAndReturn(run func() message.M return _c } -// Payload provides a mock function with given fields: +// Payload provides a mock function with no fields func (_m *MockMutableMessage) Payload() []byte { ret := _m.Called() @@ -297,7 +297,7 @@ func (_c *MockMutableMessage_Payload_Call) RunAndReturn(run func() []byte) *Mock return _c } -// Properties provides a mock function with given fields: +// Properties provides a mock function with no fields func (_m *MockMutableMessage) Properties() message.RProperties { ret := _m.Called() @@ -344,7 +344,7 @@ func (_c *MockMutableMessage_Properties_Call) RunAndReturn(run func() message.RP return _c } -// TimeTick provides a mock function with given fields: +// TimeTick provides a mock function with no fields func (_m *MockMutableMessage) TimeTick() uint64 { ret := _m.Called() @@ -389,7 +389,7 @@ func (_c *MockMutableMessage_TimeTick_Call) RunAndReturn(run func() uint64) *Moc return _c } -// TxnContext provides a mock function with given fields: +// TxnContext provides a mock function with no fields func (_m *MockMutableMessage) TxnContext() *message.TxnContext { ret := _m.Called() @@ -436,7 +436,7 @@ func (_c *MockMutableMessage_TxnContext_Call) RunAndReturn(run func() *message.T return _c } -// VChannel provides a mock function with given fields: +// VChannel provides a mock function with no fields func (_m *MockMutableMessage) VChannel() string { ret := _m.Called() @@ -481,7 +481,7 @@ func (_c *MockMutableMessage_VChannel_Call) RunAndReturn(run func() string) *Moc return _c } -// Version provides a mock function with given fields: +// Version provides a mock function with no fields func (_m *MockMutableMessage) Version() message.Version { ret := _m.Called() @@ -622,7 +622,7 @@ func (_c *MockMutableMessage_WithLastConfirmed_Call) RunAndReturn(run func(messa return _c } -// WithLastConfirmedUseMessageID provides a mock function with given fields: +// WithLastConfirmedUseMessageID provides a mock function with no fields func (_m *MockMutableMessage) WithLastConfirmedUseMessageID() message.MutableMessage { ret := _m.Called() diff --git a/pkg/mocks/streaming/util/mock_message/mock_RProperties.go b/pkg/mocks/streaming/util/mock_message/mock_RProperties.go index 21bd3b14e2..9455b1b2e2 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_message @@ -119,7 +119,7 @@ func (_c *MockRProperties_Get_Call) RunAndReturn(run func(string) (string, bool) return _c } -// ToRawMap provides a mock function with given fields: +// ToRawMap provides a mock function with no fields func (_m *MockRProperties) ToRawMap() map[string]string { ret := _m.Called() diff --git a/pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go b/pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go index 824618e0fc..cb03ce2622 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_types diff --git a/pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go b/pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go index 9e1ca2c1f7..9d93f4a939 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package mock_types diff --git a/pkg/mq/common/mock_id.go b/pkg/mq/common/mock_id.go index 6041d5ee52..cdd6567e6c 100644 --- a/pkg/mq/common/mock_id.go +++ b/pkg/mq/common/mock_id.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package common @@ -17,7 +17,7 @@ func (_m *MockMessageID) EXPECT() *MockMessageID_Expecter { return &MockMessageID_Expecter{mock: &_m.Mock} } -// AtEarliestPosition provides a mock function with given fields: +// AtEarliestPosition provides a mock function with no fields func (_m *MockMessageID) AtEarliestPosition() bool { ret := _m.Called() @@ -174,7 +174,7 @@ func (_c *MockMessageID_LessOrEqualThan_Call) RunAndReturn(run func([]byte) (boo return _c } -// Serialize provides a mock function with given fields: +// Serialize provides a mock function with no fields func (_m *MockMessageID) Serialize() []byte { ret := _m.Called() diff --git a/pkg/mq/msgdispatcher/mock_client.go b/pkg/mq/msgdispatcher/mock_client.go index dbaafb0c8c..aa47ecac63 100644 --- a/pkg/mq/msgdispatcher/mock_client.go +++ b/pkg/mq/msgdispatcher/mock_client.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package msgdispatcher @@ -22,7 +22,7 @@ func (_m *MockClient) EXPECT() *MockClient_Expecter { return &MockClient_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockClient) Close() { _m.Called() } @@ -50,7 +50,7 @@ func (_c *MockClient_Close_Call) Return() *MockClient_Close_Call { } func (_c *MockClient_Close_Call) RunAndReturn(run func()) *MockClient_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -83,7 +83,7 @@ func (_c *MockClient_Deregister_Call) Return() *MockClient_Deregister_Call { } func (_c *MockClient_Deregister_Call) RunAndReturn(run func(string)) *MockClient_Deregister_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mq/msgstream/mock_msgstream.go b/pkg/mq/msgstream/mock_msgstream.go index 43cd4977fc..a40ffcbd2e 100644 --- a/pkg/mq/msgstream/mock_msgstream.go +++ b/pkg/mq/msgstream/mock_msgstream.go @@ -1,4 +1,4 @@ -// Code generated by mockery v2.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package msgstream @@ -104,7 +104,7 @@ func (_c *MockMsgStream_AsProducer_Call) Return() *MockMsgStream_AsProducer_Call } func (_c *MockMsgStream_AsProducer_Call) RunAndReturn(run func(context.Context, []string)) *MockMsgStream_AsProducer_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -167,7 +167,7 @@ func (_c *MockMsgStream_Broadcast_Call) RunAndReturn(run func(context.Context, * return _c } -// Chan provides a mock function with given fields: +// Chan provides a mock function with no fields func (_m *MockMsgStream) Chan() <-chan *MsgPack { ret := _m.Called() @@ -260,7 +260,7 @@ func (_c *MockMsgStream_CheckTopicValid_Call) RunAndReturn(run func(string) erro return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockMsgStream) Close() { _m.Called() } @@ -288,7 +288,7 @@ func (_c *MockMsgStream_Close_Call) Return() *MockMsgStream_Close_Call { } func (_c *MockMsgStream_Close_Call) RunAndReturn(run func()) *MockMsgStream_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -321,7 +321,7 @@ func (_c *MockMsgStream_ForceEnableProduce_Call) Return() *MockMsgStream_ForceEn } func (_c *MockMsgStream_ForceEnableProduce_Call) RunAndReturn(run func(bool)) *MockMsgStream_ForceEnableProduce_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -383,7 +383,7 @@ func (_c *MockMsgStream_GetLatestMsgID_Call) RunAndReturn(run func(string) (comm return _c } -// GetProduceChannels provides a mock function with given fields: +// GetProduceChannels provides a mock function with no fields func (_m *MockMsgStream) GetProduceChannels() []string { ret := _m.Called() @@ -554,7 +554,7 @@ func (_c *MockMsgStream_SetRepackFunc_Call) Return() *MockMsgStream_SetRepackFun } func (_c *MockMsgStream_SetRepackFunc_Call) RunAndReturn(run func(RepackFunc)) *MockMsgStream_SetRepackFunc_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/pkg/mq/msgstream/mock_msgstream_factory.go b/pkg/mq/msgstream/mock_msgstream_factory.go index fe6c8a6587..1be94da4a5 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.46.0. DO NOT EDIT. +// Code generated by mockery v2.53.3. DO NOT EDIT. package msgstream