// Code generated by mockery v2.32.4. DO NOT EDIT. package mock_message import ( message "github.com/milvus-io/milvus/internal/util/streamingutil/message" mock "github.com/stretchr/testify/mock" ) // MockImmutableMessage is an autogenerated mock type for the ImmutableMessage type type MockImmutableMessage struct { mock.Mock } type MockImmutableMessage_Expecter struct { mock *mock.Mock } func (_m *MockImmutableMessage) EXPECT() *MockImmutableMessage_Expecter { return &MockImmutableMessage_Expecter{mock: &_m.Mock} } // EstimateSize provides a mock function with given fields: func (_m *MockImmutableMessage) EstimateSize() int { ret := _m.Called() var r0 int if rf, ok := ret.Get(0).(func() int); ok { r0 = rf() } else { r0 = ret.Get(0).(int) } return r0 } // MockImmutableMessage_EstimateSize_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'EstimateSize' type MockImmutableMessage_EstimateSize_Call struct { *mock.Call } // EstimateSize is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) EstimateSize() *MockImmutableMessage_EstimateSize_Call { return &MockImmutableMessage_EstimateSize_Call{Call: _e.mock.On("EstimateSize")} } func (_c *MockImmutableMessage_EstimateSize_Call) Run(run func()) *MockImmutableMessage_EstimateSize_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_EstimateSize_Call) Return(_a0 int) *MockImmutableMessage_EstimateSize_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_EstimateSize_Call) RunAndReturn(run func() int) *MockImmutableMessage_EstimateSize_Call { _c.Call.Return(run) return _c } // LastConfirmedMessageID provides a mock function with given fields: func (_m *MockImmutableMessage) LastConfirmedMessageID() message.MessageID { ret := _m.Called() var r0 message.MessageID if rf, ok := ret.Get(0).(func() message.MessageID); ok { r0 = rf() } else { if ret.Get(0) != nil { r0 = ret.Get(0).(message.MessageID) } } return r0 } // MockImmutableMessage_LastConfirmedMessageID_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LastConfirmedMessageID' type MockImmutableMessage_LastConfirmedMessageID_Call struct { *mock.Call } // LastConfirmedMessageID is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) LastConfirmedMessageID() *MockImmutableMessage_LastConfirmedMessageID_Call { return &MockImmutableMessage_LastConfirmedMessageID_Call{Call: _e.mock.On("LastConfirmedMessageID")} } func (_c *MockImmutableMessage_LastConfirmedMessageID_Call) Run(run func()) *MockImmutableMessage_LastConfirmedMessageID_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_LastConfirmedMessageID_Call) Return(_a0 message.MessageID) *MockImmutableMessage_LastConfirmedMessageID_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_LastConfirmedMessageID_Call) RunAndReturn(run func() message.MessageID) *MockImmutableMessage_LastConfirmedMessageID_Call { _c.Call.Return(run) return _c } // MessageID provides a mock function with given fields: func (_m *MockImmutableMessage) MessageID() message.MessageID { ret := _m.Called() var r0 message.MessageID if rf, ok := ret.Get(0).(func() message.MessageID); ok { r0 = rf() } else { if ret.Get(0) != nil { r0 = ret.Get(0).(message.MessageID) } } return r0 } // MockImmutableMessage_MessageID_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'MessageID' type MockImmutableMessage_MessageID_Call struct { *mock.Call } // MessageID is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) MessageID() *MockImmutableMessage_MessageID_Call { return &MockImmutableMessage_MessageID_Call{Call: _e.mock.On("MessageID")} } func (_c *MockImmutableMessage_MessageID_Call) Run(run func()) *MockImmutableMessage_MessageID_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_MessageID_Call) Return(_a0 message.MessageID) *MockImmutableMessage_MessageID_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_MessageID_Call) RunAndReturn(run func() message.MessageID) *MockImmutableMessage_MessageID_Call { _c.Call.Return(run) return _c } // MessageType provides a mock function with given fields: func (_m *MockImmutableMessage) MessageType() message.MessageType { ret := _m.Called() var r0 message.MessageType if rf, ok := ret.Get(0).(func() message.MessageType); ok { r0 = rf() } else { r0 = ret.Get(0).(message.MessageType) } return r0 } // MockImmutableMessage_MessageType_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'MessageType' type MockImmutableMessage_MessageType_Call struct { *mock.Call } // MessageType is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) MessageType() *MockImmutableMessage_MessageType_Call { return &MockImmutableMessage_MessageType_Call{Call: _e.mock.On("MessageType")} } func (_c *MockImmutableMessage_MessageType_Call) Run(run func()) *MockImmutableMessage_MessageType_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_MessageType_Call) Return(_a0 message.MessageType) *MockImmutableMessage_MessageType_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_MessageType_Call) RunAndReturn(run func() message.MessageType) *MockImmutableMessage_MessageType_Call { _c.Call.Return(run) return _c } // Payload provides a mock function with given fields: func (_m *MockImmutableMessage) Payload() []byte { ret := _m.Called() var r0 []byte if rf, ok := ret.Get(0).(func() []byte); ok { r0 = rf() } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } return r0 } // MockImmutableMessage_Payload_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Payload' type MockImmutableMessage_Payload_Call struct { *mock.Call } // Payload is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) Payload() *MockImmutableMessage_Payload_Call { return &MockImmutableMessage_Payload_Call{Call: _e.mock.On("Payload")} } func (_c *MockImmutableMessage_Payload_Call) Run(run func()) *MockImmutableMessage_Payload_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_Payload_Call) Return(_a0 []byte) *MockImmutableMessage_Payload_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_Payload_Call) RunAndReturn(run func() []byte) *MockImmutableMessage_Payload_Call { _c.Call.Return(run) return _c } // Properties provides a mock function with given fields: func (_m *MockImmutableMessage) Properties() message.RProperties { ret := _m.Called() var r0 message.RProperties if rf, ok := ret.Get(0).(func() message.RProperties); ok { r0 = rf() } else { if ret.Get(0) != nil { r0 = ret.Get(0).(message.RProperties) } } return r0 } // MockImmutableMessage_Properties_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Properties' type MockImmutableMessage_Properties_Call struct { *mock.Call } // Properties is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) Properties() *MockImmutableMessage_Properties_Call { return &MockImmutableMessage_Properties_Call{Call: _e.mock.On("Properties")} } func (_c *MockImmutableMessage_Properties_Call) Run(run func()) *MockImmutableMessage_Properties_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_Properties_Call) Return(_a0 message.RProperties) *MockImmutableMessage_Properties_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_Properties_Call) RunAndReturn(run func() message.RProperties) *MockImmutableMessage_Properties_Call { _c.Call.Return(run) return _c } // TimeTick provides a mock function with given fields: func (_m *MockImmutableMessage) TimeTick() uint64 { ret := _m.Called() var r0 uint64 if rf, ok := ret.Get(0).(func() uint64); ok { r0 = rf() } else { r0 = ret.Get(0).(uint64) } return r0 } // MockImmutableMessage_TimeTick_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'TimeTick' type MockImmutableMessage_TimeTick_Call struct { *mock.Call } // TimeTick is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) TimeTick() *MockImmutableMessage_TimeTick_Call { return &MockImmutableMessage_TimeTick_Call{Call: _e.mock.On("TimeTick")} } func (_c *MockImmutableMessage_TimeTick_Call) Run(run func()) *MockImmutableMessage_TimeTick_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_TimeTick_Call) Return(_a0 uint64) *MockImmutableMessage_TimeTick_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_TimeTick_Call) RunAndReturn(run func() uint64) *MockImmutableMessage_TimeTick_Call { _c.Call.Return(run) return _c } // Version provides a mock function with given fields: func (_m *MockImmutableMessage) Version() message.Version { ret := _m.Called() var r0 message.Version if rf, ok := ret.Get(0).(func() message.Version); ok { r0 = rf() } else { r0 = ret.Get(0).(message.Version) } return r0 } // MockImmutableMessage_Version_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Version' type MockImmutableMessage_Version_Call struct { *mock.Call } // Version is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) Version() *MockImmutableMessage_Version_Call { return &MockImmutableMessage_Version_Call{Call: _e.mock.On("Version")} } func (_c *MockImmutableMessage_Version_Call) Run(run func()) *MockImmutableMessage_Version_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_Version_Call) Return(_a0 message.Version) *MockImmutableMessage_Version_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_Version_Call) RunAndReturn(run func() message.Version) *MockImmutableMessage_Version_Call { _c.Call.Return(run) return _c } // WALName provides a mock function with given fields: func (_m *MockImmutableMessage) WALName() string { ret := _m.Called() var r0 string if rf, ok := ret.Get(0).(func() string); ok { r0 = rf() } else { r0 = ret.Get(0).(string) } return r0 } // MockImmutableMessage_WALName_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'WALName' type MockImmutableMessage_WALName_Call struct { *mock.Call } // WALName is a helper method to define mock.On call func (_e *MockImmutableMessage_Expecter) WALName() *MockImmutableMessage_WALName_Call { return &MockImmutableMessage_WALName_Call{Call: _e.mock.On("WALName")} } func (_c *MockImmutableMessage_WALName_Call) Run(run func()) *MockImmutableMessage_WALName_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockImmutableMessage_WALName_Call) Return(_a0 string) *MockImmutableMessage_WALName_Call { _c.Call.Return(_a0) return _c } func (_c *MockImmutableMessage_WALName_Call) RunAndReturn(run func() string) *MockImmutableMessage_WALName_Call { _c.Call.Return(run) return _c } // NewMockImmutableMessage creates a new instance of MockImmutableMessage. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. // The first argument is typically a *testing.T value. func NewMockImmutableMessage(t interface { mock.TestingT Cleanup(func()) }) *MockImmutableMessage { mock := &MockImmutableMessage{} mock.Mock.Test(t) t.Cleanup(func() { mock.AssertExpectations(t) }) return mock }