From 5fd47c3c89311a52310305aa21280a0d7efce647 Mon Sep 17 00:00:00 2001 From: Zhen Ye Date: Thu, 24 Apr 2025 10:46:43 +0800 Subject: [PATCH] fix: mockery too unavailable after upgrade golang version (#41481) issue: #41291 pr: #41318 Signed-off-by: chyezh --- Makefile | 2 +- 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_Broadcast.go | 2 +- .../mock_streaming/mock_WALAccesser.go | 6 +- .../grpc/mock_balancer/mock_SubConn.go | 12 ++-- .../grpc/mock_resolver/mock_ClientConn.go | 6 +- .../mock_grpc/mock_ClientStream.go | 10 +-- internal/mocks/mock_datanode.go | 22 +++---- internal/mocks/mock_datanode_client.go | 4 +- .../mock_StreamingCoordCataLog.go | 2 +- .../mock_StreamingNodeCataLog.go | 2 +- internal/mocks/mock_mixcoord.go | 26 ++++---- internal/mocks/mock_mixcoord_client.go | 4 +- internal/mocks/mock_proxy.go | 24 +++---- internal/mocks/mock_proxy_client.go | 4 +- internal/mocks/mock_querynode.go | 20 +++--- internal/mocks/mock_querynode_client.go | 4 +- .../mocks/mock_storage/mock_ChunkManager.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 | 8 +-- .../handler/mock_producer/mock_Producer.go | 10 +-- .../client/mock_handler/mock_HandlerClient.go | 6 +- .../client/mock_manager/mock_ManagerClient.go | 6 +- .../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_wab/mock_ROWriteAheadBuffer.go | 2 +- .../mock_inspector/mock_SealOperator.go | 12 ++-- .../mock_TimeTickSyncOperator.go | 10 +-- .../wal/mock_interceptors/mock_Interceptor.go | 6 +- .../mock_InterceptorBuilder.go | 2 +- .../mock_InterceptorWithMetrics.go | 8 +-- .../mock_InterceptorWithReady.go | 8 +-- .../mocks/util/mock_segcore/mock_CSegment.go | 14 ++--- .../mock_optimizers/mock_QueryHook.go | 2 +- .../mock_discoverer/mock_Discoverer.go | 2 +- .../service/mock_lazygrpc/mock_Service.go | 10 +-- .../service/mock_resolver/mock_Builder.go | 10 +-- .../service/mock_resolver/mock_Resolver.go | 2 +- internal/proxy/mock_cache.go | 18 +++--- 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/mocks/mock_querynode.go | 2 +- 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 +-- .../go-api/v2/mock_hook/mock_Cipher.go | 2 +- .../go-api/v2/mock_hook/mock_Decryptor.go | 2 +- .../go-api/v2/mock_hook/mock_Encryptor.go | 2 +- 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 | 18 +++--- pkg/mq/msgstream/mock_msgstream_factory.go | 2 +- 100 files changed, 497 insertions(+), 495 deletions(-) diff --git a/Makefile b/Makefile index 71185186b5..02277b2e45 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/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_Broadcast.go b/internal/mocks/distributed/mock_streaming/mock_Broadcast.go index 1e78402966..dec3b85294 100644 --- a/internal/mocks/distributed/mock_streaming/mock_Broadcast.go +++ b/internal/mocks/distributed/mock_streaming/mock_Broadcast.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 diff --git a/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go b/internal/mocks/distributed/mock_streaming/mock_WALAccesser.go index 6f9d178849..258c945d32 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() @@ -435,7 +435,7 @@ func (_c *MockWALAccesser_Txn_Call) RunAndReturn(run func(context.Context, strea return _c } -// WALName provides a mock function with given fields: +// WALName provides a mock function with no fields func (_m *MockWALAccesser) WALName() string { 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_datanode.go b/internal/mocks/mock_datanode.go index d53f2187c8..427e645ce4 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 @@ -624,7 +624,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() @@ -905,7 +905,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() @@ -950,7 +950,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() @@ -1113,7 +1113,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() @@ -1571,7 +1571,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() @@ -1704,7 +1704,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 } @@ -1737,7 +1737,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 } @@ -1846,7 +1846,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() @@ -1891,7 +1891,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() @@ -2024,7 +2024,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 36e3a3ce86..eb62d5d41b 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 @@ -107,7 +107,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_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 7650371903..092d73d480 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_mixcoord.go b/internal/mocks/mock_mixcoord.go index 1209a6e2d0..7a3544c3e7 100644 --- a/internal/mocks/mock_mixcoord.go +++ b/internal/mocks/mock_mixcoord.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 @@ -4705,7 +4705,7 @@ func (_c *MixCoord_GetTimeTickChannel_Call) RunAndReturn(run func(context.Contex return _c } -// GracefulStop provides a mock function with given fields: +// GracefulStop provides a mock function with no fields func (_m *MixCoord) GracefulStop() { _m.Called() } @@ -4733,7 +4733,7 @@ func (_c *MixCoord_GracefulStop_Call) Return() *MixCoord_GracefulStop_Call { } func (_c *MixCoord_GracefulStop_Call) RunAndReturn(run func()) *MixCoord_GracefulStop_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -4914,7 +4914,7 @@ func (_c *MixCoord_ImportV2_Call) RunAndReturn(run func(context.Context, *intern return _c } -// Init provides a mock function with given fields: +// Init provides a mock function with no fields func (_m *MixCoord) Init() error { ret := _m.Called() @@ -6080,7 +6080,7 @@ func (_c *MixCoord_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 *MixCoord) Register() error { ret := _m.Called() @@ -6154,7 +6154,7 @@ func (_c *MixCoord_RegisterStreamingCoordGRPCService_Call) Return() *MixCoord_Re } func (_c *MixCoord_RegisterStreamingCoordGRPCService_Call) RunAndReturn(run func(*grpc.Server)) *MixCoord_RegisterStreamingCoordGRPCService_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6836,7 +6836,7 @@ func (_c *MixCoord_SetAddress_Call) Return() *MixCoord_SetAddress_Call { } func (_c *MixCoord_SetAddress_Call) RunAndReturn(run func(string)) *MixCoord_SetAddress_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6869,7 +6869,7 @@ func (_c *MixCoord_SetEtcdClient_Call) Return() *MixCoord_SetEtcdClient_Call { } func (_c *MixCoord_SetEtcdClient_Call) RunAndReturn(run func(*clientv3.Client)) *MixCoord_SetEtcdClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6902,7 +6902,7 @@ func (_c *MixCoord_SetMixCoordClient_Call) Return() *MixCoord_SetMixCoordClient_ } func (_c *MixCoord_SetMixCoordClient_Call) RunAndReturn(run func(types.MixCoordClient)) *MixCoord_SetMixCoordClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6994,7 +6994,7 @@ func (_c *MixCoord_SetTiKVClient_Call) Return() *MixCoord_SetTiKVClient_Call { } func (_c *MixCoord_SetTiKVClient_Call) RunAndReturn(run func(*txnkv.Client)) *MixCoord_SetTiKVClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -7470,7 +7470,7 @@ func (_c *MixCoord_ShowSegments_Call) RunAndReturn(run func(context.Context, *mi return _c } -// Start provides a mock function with given fields: +// Start provides a mock function with no fields func (_m *MixCoord) Start() error { ret := _m.Called() @@ -7515,7 +7515,7 @@ func (_c *MixCoord_Start_Call) RunAndReturn(run func() error) *MixCoord_Start_Ca return _c } -// Stop provides a mock function with given fields: +// Stop provides a mock function with no fields func (_m *MixCoord) Stop() error { ret := _m.Called() @@ -8356,7 +8356,7 @@ func (_c *MixCoord_UpdateStateCode_Call) Return() *MixCoord_UpdateStateCode_Call } func (_c *MixCoord_UpdateStateCode_Call) RunAndReturn(run func(commonpb.StateCode)) *MixCoord_UpdateStateCode_Call { - _c.Call.Return(run) + _c.Run(run) return _c } diff --git a/internal/mocks/mock_mixcoord_client.go b/internal/mocks/mock_mixcoord_client.go index 572c57dfca..95c41e66c8 100644 --- a/internal/mocks/mock_mixcoord_client.go +++ b/internal/mocks/mock_mixcoord_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 @@ -1223,7 +1223,7 @@ func (_c *MockMixCoordClient_CheckQueryNodeDistribution_Call) RunAndReturn(run f return _c } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockMixCoordClient) Close() error { ret := _m.Called() diff --git a/internal/mocks/mock_proxy.go b/internal/mocks/mock_proxy.go index 3a91b5b7e8..de29768ff6 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 @@ -2335,7 +2335,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() @@ -3501,7 +3501,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() @@ -4089,7 +4089,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() @@ -5491,7 +5491,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() @@ -6214,7 +6214,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 } @@ -6247,7 +6247,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 } @@ -6280,7 +6280,7 @@ func (_c *MockProxy_SetMixCoordClient_Call) Return() *MockProxy_SetMixCoordClien } func (_c *MockProxy_SetMixCoordClient_Call) RunAndReturn(run func(types.MixCoordClient)) *MockProxy_SetMixCoordClient_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -6313,7 +6313,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 } @@ -6494,7 +6494,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() @@ -6539,7 +6539,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() @@ -6908,7 +6908,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_querynode.go b/internal/mocks/mock_querynode.go index f293f6ad6e..363c3c17ef 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() @@ -1846,7 +1846,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 9d6ca80c62..e70f1631e3 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_storage/mock_ChunkManager.go b/internal/mocks/mock_storage/mock_ChunkManager.go index 8ac4ce03f3..e35e2c7edd 100644 --- a/internal/mocks/mock_storage/mock_ChunkManager.go +++ b/internal/mocks/mock_storage/mock_ChunkManager.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_storage @@ -624,7 +624,7 @@ func (_c *MockChunkManager_RemoveWithPrefix_Call) RunAndReturn(run func(context. return _c } -// RootPath provides a mock function with given fields: +// RootPath provides a mock function with no fields func (_m *MockChunkManager) RootPath() string { 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 662204bc28..6707850ac6 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 e9328568f4..a8473bb679 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() error { ret := _m.Called() @@ -62,7 +62,7 @@ func (_c *MockConsumer_Close_Call) RunAndReturn(run func() error) *MockConsumer_ 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() @@ -109,7 +109,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 82dbc50b48..6268f90002 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 @@ -83,7 +83,7 @@ func (_c *MockProducer_Append_Call) RunAndReturn(run func(context.Context, messa 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() @@ -130,7 +130,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() } @@ -158,11 +158,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 5f4793e7fb..fc87965622 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_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 18b20c45ba..e15f37492a 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,7 +240,7 @@ 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 } @@ -301,7 +301,7 @@ func (_c *MockWAL_GetLatestMVCCTimestamp_Call) RunAndReturn(run func(context.Con 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() @@ -405,7 +405,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/mock_wab/mock_ROWriteAheadBuffer.go b/internal/mocks/streamingnode/server/wal/interceptors/mock_wab/mock_ROWriteAheadBuffer.go index bad93162ad..e6601e84e7 100644 --- a/internal/mocks/streamingnode/server/wal/interceptors/mock_wab/mock_ROWriteAheadBuffer.go +++ b/internal/mocks/streamingnode/server/wal/interceptors/mock_wab/mock_ROWriteAheadBuffer.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_wab 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 37cb396dce..741daef633 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 @@ -27,7 +27,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() @@ -72,7 +72,7 @@ func (_c *MockTimeTickSyncOperator_Channel_Call) RunAndReturn(run func() types.P return _c } -// MVCCManager provides a mock function with given fields: +// MVCCManager provides a mock function with no fields func (_m *MockTimeTickSyncOperator) MVCCManager() *mvcc.MVCCManager { ret := _m.Called() @@ -149,11 +149,11 @@ func (_c *MockTimeTickSyncOperator_Sync_Call) Return() *MockTimeTickSyncOperator } func (_c *MockTimeTickSyncOperator_Sync_Call) RunAndReturn(run func(context.Context, bool)) *MockTimeTickSyncOperator_Sync_Call { - _c.Call.Return(run) + _c.Run(run) return _c } -// WriteAheadBuffer provides a mock function with given fields: +// WriteAheadBuffer provides a mock function with no fields func (_m *MockTimeTickSyncOperator) WriteAheadBuffer() wab.ROWriteAheadBuffer { 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 03e6e85cfc..f165eda5e4 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_InterceptorWithMetrics.go b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithMetrics.go index d144865340..59e5d5c8f1 100644 --- a/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithMetrics.go +++ b/internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithMetrics.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 *MockInterceptorWithMetrics) EXPECT() *MockInterceptorWithMetrics_Expec return &MockInterceptorWithMetrics_Expecter{mock: &_m.Mock} } -// Close provides a mock function with given fields: +// Close provides a mock function with no fields func (_m *MockInterceptorWithMetrics) Close() { _m.Called() } @@ -51,7 +51,7 @@ func (_c *MockInterceptorWithMetrics_Close_Call) Return() *MockInterceptorWithMe } func (_c *MockInterceptorWithMetrics_Close_Call) RunAndReturn(run func()) *MockInterceptorWithMetrics_Close_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -115,7 +115,7 @@ func (_c *MockInterceptorWithMetrics_DoAppend_Call) RunAndReturn(run func(contex return _c } -// Name provides a mock function with given fields: +// Name provides a mock function with no fields func (_m *MockInterceptorWithMetrics) Name() string { ret := _m.Called() 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 061968946c..8249d30233 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 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 cbc7adb751..41de328920 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 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_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/mocks/mock_querynode.go b/internal/querycoordv2/mocks/mock_querynode.go index 43e482cc43..1bf5c04904 100644 --- a/internal/querycoordv2/mocks/mock_querynode.go +++ b/internal/querycoordv2/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 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 a4aaa5d68d..b60d358440 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 @@ -194,7 +194,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() @@ -499,7 +499,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() } @@ -527,7 +527,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 513eaecc33..921a6542f0 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 @@ -59,11 +59,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() } @@ -91,11 +91,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() @@ -140,7 +140,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() @@ -362,7 +362,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() @@ -407,7 +407,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() @@ -625,7 +625,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 } @@ -658,7 +658,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 } @@ -876,7 +876,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() @@ -921,7 +921,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() } @@ -949,7 +949,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 } @@ -997,7 +997,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 } @@ -1031,7 +1031,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 } @@ -1070,7 +1070,7 @@ func (_c *MockShardDelegator_SyncTargetVersion_Call) Return() *MockShardDelegato } func (_c *MockShardDelegator_SyncTargetVersion_Call) RunAndReturn(run func(int64, []int64, []int64, []int64, []int64, *msgpb.MsgPosition, *msgpb.MsgPosition)) *MockShardDelegator_SyncTargetVersion_Call { - _c.Call.Return(run) + _c.Run(run) return _c } @@ -1103,7 +1103,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 } @@ -1183,7 +1183,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 } @@ -1234,7 +1234,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 5baf3f5f22..428a433869 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/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Cipher.go b/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Cipher.go index 435fe4b28b..bdffbf82b4 100644 --- a/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Cipher.go +++ b/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Cipher.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_hook diff --git a/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Decryptor.go b/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Decryptor.go index a48afdb28e..f74d01b2d6 100644 --- a/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Decryptor.go +++ b/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Decryptor.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_hook diff --git a/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Encryptor.go b/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Encryptor.go index 0350ccc666..bfa4e4df34 100644 --- a/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Encryptor.go +++ b/pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Encryptor.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_hook 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 111cd4bf1a..f06e6df9be 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 *ConsumeMsgPack { 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() @@ -430,7 +430,7 @@ func (_c *MockMsgStream_GetProduceChannels_Call) RunAndReturn(run func() []strin return _c } -// GetUnmarshalDispatcher provides a mock function with given fields: +// GetUnmarshalDispatcher provides a mock function with no fields func (_m *MockMsgStream) GetUnmarshalDispatcher() UnmarshalDispatcher { ret := _m.Called() @@ -601,7 +601,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