admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-06 09:08:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: mockery too unavailable after upgrade golang version (#41481)

Browse Source 
issue: #41291
pr: #41318

Signed-off-by: chyezh <chyezh@outlook.com>
This commit is contained in:
Zhen Ye 2025-04-24 10:46:43 +08:00 committed by GitHub
parent f23df95a77
commit 5fd47c3c89
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
100 changed files with 497 additions and 495 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -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

6
internal/kv/mocks/meta_kv.go
View File

@ -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
}

2
internal/kv/mocks/snapshot_kv.go
View File

@ -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

6
internal/kv/mocks/txn_kv.go
View File

@ -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
}

6
internal/kv/mocks/watch_kv.go
View File

@ -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
}

2
internal/mocks/distributed/mock_streaming/mock_Broadcast.go
View File

@ -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

6
internal/mocks/distributed/mock_streaming/mock_WALAccesser.go
View File

@ -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()

12
internal/mocks/google.golang.org/grpc/mock_balancer/mock_SubConn.go
View File

@ -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
}

6
internal/mocks/google.golang.org/grpc/mock_resolver/mock_ClientConn.go
View File

@ -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
}

10
internal/mocks/google.golang.org/mock_grpc/mock_ClientStream.go
View File

@ -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()

22
internal/mocks/mock_datanode.go
View File

@ -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
}

4
internal/mocks/mock_datanode_client.go
View File

@ -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()

2
internal/mocks/mock_metastore/mock_StreamingCoordCataLog.go
View File

@ -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

2
internal/mocks/mock_metastore/mock_StreamingNodeCataLog.go
View File

@ -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

26
internal/mocks/mock_mixcoord.go
View File

@ -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
}

4
internal/mocks/mock_mixcoord_client.go
View File

@ -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()

24
internal/mocks/mock_proxy.go
View File

@ -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
}

4
internal/mocks/mock_proxy_client.go
View File

@ -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()

20
internal/mocks/mock_querynode.go
View File

@ -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
}

4
internal/mocks/mock_querynode_client.go
View File

@ -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()

4
internal/mocks/mock_storage/mock_ChunkManager.go
View File

@ -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()

6
internal/mocks/streamingcoord/mock_client/mock_BroadcastService.go
View File

@ -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
}

10
internal/mocks/streamingcoord/mock_client/mock_Client.go
View File

@ -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
}

6
internal/mocks/streamingcoord/server/mock_balancer/mock_Balancer.go
View File

@ -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
}

16
internal/mocks/streamingcoord/server/mock_broadcaster/mock_AppendOperator.go
View File

@ -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
}

6
internal/mocks/streamingnode/client/handler/mock_assignment/mock_Watcher.go
View File

@ -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
}

8
internal/mocks/streamingnode/client/handler/mock_consumer/mock_Consumer.go
View File

@ -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()

10
internal/mocks/streamingnode/client/handler/mock_producer/mock_Producer.go
View File

@ -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()

6
internal/mocks/streamingnode/client/mock_handler/mock_HandlerClient.go
View File

@ -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
}

6
internal/mocks/streamingnode/client/mock_manager/mock_ManagerClient.go
View File

@ -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
}

6
internal/mocks/streamingnode/server/mock_wal/mock_Opener.go
View File

@ -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
}

6
internal/mocks/streamingnode/server/mock_wal/mock_OpenerBuilder.go
View File

@ -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()

12
internal/mocks/streamingnode/server/mock_wal/mock_Scanner.go
View File

@ -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()

16
internal/mocks/streamingnode/server/mock_wal/mock_WAL.go
View File

@ -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()

8
internal/mocks/streamingnode/server/mock_walmanager/mock_Manager.go
View File

@ -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()

2
internal/mocks/streamingnode/server/wal/interceptors/mock_wab/mock_ROWriteAheadBuffer.go
View File

@ -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

12
internal/mocks/streamingnode/server/wal/interceptors/segment/mock_inspector/mock_SealOperator.go
View File

@ -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
}

10
internal/mocks/streamingnode/server/wal/interceptors/timetick/mock_inspector/mock_TimeTickSyncOperator.go
View File

@ -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()

6
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_Interceptor.go
View File

@ -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
}

2
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorBuilder.go
View File

@ -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

8
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithMetrics.go
View File

@ -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()

8
internal/mocks/streamingnode/server/wal/mock_interceptors/mock_InterceptorWithReady.go
View File

@ -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()

14
internal/mocks/util/mock_segcore/mock_CSegment.go
View File

@ -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()

2
internal/mocks/util/searchutil/mock_optimizers/mock_QueryHook.go
View File

@ -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

2
internal/mocks/util/streamingutil/service/mock_discoverer/mock_Discoverer.go
View File

@ -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

10
internal/mocks/util/streamingutil/service/mock_lazygrpc/mock_Service.go
View File

@ -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 {

10
internal/mocks/util/streamingutil/service/mock_resolver/mock_Builder.go
View File

@ -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()

2
internal/mocks/util/streamingutil/service/mock_resolver/mock_Resolver.go
View File

@ -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

18
internal/proxy/mock_cache.go
View File

@ -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
}

14
internal/proxy/mock_lb_balancer.go
View File

@ -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
}

10
internal/proxy/mock_lb_policy.go
View File

@ -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
}

8
internal/proxy/mock_shardclient_manager.go
View File

@ -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
}

2
internal/proxy/mock_tso_test.go
View File

@ -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

2
internal/querycoordv2/balance/mock_balancer.go
View File

@ -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

12
internal/querycoordv2/dist/mock_controller.go vendored
View File

@ -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
}

2
internal/querycoordv2/meta/mock_broker.go
View File

@ -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

8
internal/querycoordv2/meta/mock_target_manager.go
View File

@ -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
}

2
internal/querycoordv2/mocks/mock_querynode.go
View File

@ -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

10
internal/querycoordv2/session/mock_cluster.go
View File

@ -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
}

36
internal/querycoordv2/task/mock_scheduler.go
View File

@ -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
}

2
internal/querynodev2/cluster/mock_manager.go
View File

@ -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

8
internal/querynodev2/cluster/mock_worker.go
View File

@ -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
}

38
internal/querynodev2/delegator/mock_delegator.go
View File

@ -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()

6
internal/querynodev2/segments/mock_collection_manager.go
View File

@ -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()

26
internal/querynodev2/segments/mock_loader.go
View File

@ -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:] {

62
internal/querynodev2/segments/mock_segment.go
View File

@ -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()

62
internal/querynodev2/segments/mock_segment_manager.go
View File

@ -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
}

10
internal/rootcoord/mocks/garbage_collector.go
View File

@ -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
}

2
internal/rootcoord/mocks/meta_table.go
View File

@ -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

8
pkg/eventlog/mock_logger.go
View File

@ -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
}

10
pkg/kv/predicates/mock_predicate.go
View File

@ -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()

2
pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Cipher.go
View File

@ -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

2
pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Decryptor.go
View File

@ -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

2
pkg/mocks/github.com/milvus-io/milvus-proto/go-api/v2/mock_hook/mock_Encryptor.go
View File

@ -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

6
pkg/mocks/mock_kv/mock_MetaKv.go
View File

@ -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
}

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go
View File

@ -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

12
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go
View File

@ -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()

8
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go
View File

@ -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
}

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordBroadcastServiceClient.go
View File

@ -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

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go
View File

@ -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

12
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go
View File

@ -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()

8
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go
View File

@ -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
}

12
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go
View File

@ -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()

8
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go
View File

@ -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
}

2
pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go
View File

@ -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

6
pkg/mocks/streaming/mock_walimpls/mock_OpenerBuilderImpls.go
View File

@ -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()

6
pkg/mocks/streaming/mock_walimpls/mock_OpenerImpls.go
View File

@ -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
}

12
pkg/mocks/streaming/mock_walimpls/mock_ScannerImpls.go
View File

@ -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()

10
pkg/mocks/streaming/mock_walimpls/mock_WALImpls.go
View File

@ -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()

28
pkg/mocks/streaming/util/mock_message/mock_ImmutableMessage.go
View File

@ -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()

34
pkg/mocks/streaming/util/mock_message/mock_ImmutableTxnMessage.go
View File

@ -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()

8
pkg/mocks/streaming/util/mock_message/mock_MessageID.go
View File

@ -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()

24
pkg/mocks/streaming/util/mock_message/mock_MutableMessage.go
View File

@ -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()

4
pkg/mocks/streaming/util/mock_message/mock_RProperties.go
View File

@ -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()

2
pkg/mocks/streaming/util/mock_types/mock_AssignmentDiscoverWatcher.go
View File

@ -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

2
pkg/mocks/streaming/util/mock_types/mock_AssignmentRebalanceTrigger.go
View File

@ -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

6
pkg/mq/common/mock_id.go
View File

@ -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()

8
pkg/mq/msgdispatcher/mock_client.go
View File

@ -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
}

18
pkg/mq/msgstream/mock_msgstream.go
View File

@ -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
}

2
pkg/mq/msgstream/mock_msgstream_factory.go
View File

@ -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