// Code generated by mockery v2.53.0. DO NOT EDIT. package grpc import ( context "context" metadata "google.golang.org/grpc/metadata" mock "github.com/stretchr/testify/mock" ) // MockStreamContextWrapper is an autogenerated mock type for the StreamContextWrapper type type MockStreamContextWrapper struct { mock.Mock } type MockStreamContextWrapper_Expecter struct { mock *mock.Mock } func (_m *MockStreamContextWrapper) EXPECT() *MockStreamContextWrapper_Expecter { return &MockStreamContextWrapper_Expecter{mock: &_m.Mock} } // Context provides a mock function with no fields func (_m *MockStreamContextWrapper) Context() context.Context { ret := _m.Called() if len(ret) == 0 { panic("no return value specified for Context") } var r0 context.Context if rf, ok := ret.Get(0).(func() context.Context); ok { r0 = rf() } else { if ret.Get(0) != nil { r0 = ret.Get(0).(context.Context) } } return r0 } // MockStreamContextWrapper_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' type MockStreamContextWrapper_Context_Call struct { *mock.Call } // Context is a helper method to define mock.On call func (_e *MockStreamContextWrapper_Expecter) Context() *MockStreamContextWrapper_Context_Call { return &MockStreamContextWrapper_Context_Call{Call: _e.mock.On("Context")} } func (_c *MockStreamContextWrapper_Context_Call) Run(run func()) *MockStreamContextWrapper_Context_Call { _c.Call.Run(func(args mock.Arguments) { run() }) return _c } func (_c *MockStreamContextWrapper_Context_Call) Return(_a0 context.Context) *MockStreamContextWrapper_Context_Call { _c.Call.Return(_a0) return _c } func (_c *MockStreamContextWrapper_Context_Call) RunAndReturn(run func() context.Context) *MockStreamContextWrapper_Context_Call { _c.Call.Return(run) return _c } // RecvMsg provides a mock function with given fields: m func (_m *MockStreamContextWrapper) RecvMsg(m any) error { ret := _m.Called(m) if len(ret) == 0 { panic("no return value specified for RecvMsg") } var r0 error if rf, ok := ret.Get(0).(func(any) error); ok { r0 = rf(m) } else { r0 = ret.Error(0) } return r0 } // MockStreamContextWrapper_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' type MockStreamContextWrapper_RecvMsg_Call struct { *mock.Call } // RecvMsg is a helper method to define mock.On call // - m any func (_e *MockStreamContextWrapper_Expecter) RecvMsg(m interface{}) *MockStreamContextWrapper_RecvMsg_Call { return &MockStreamContextWrapper_RecvMsg_Call{Call: _e.mock.On("RecvMsg", m)} } func (_c *MockStreamContextWrapper_RecvMsg_Call) Run(run func(m any)) *MockStreamContextWrapper_RecvMsg_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(any)) }) return _c } func (_c *MockStreamContextWrapper_RecvMsg_Call) Return(_a0 error) *MockStreamContextWrapper_RecvMsg_Call { _c.Call.Return(_a0) return _c } func (_c *MockStreamContextWrapper_RecvMsg_Call) RunAndReturn(run func(any) error) *MockStreamContextWrapper_RecvMsg_Call { _c.Call.Return(run) return _c } // SendHeader provides a mock function with given fields: _a0 func (_m *MockStreamContextWrapper) SendHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) if len(ret) == 0 { panic("no return value specified for SendHeader") } var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) } else { r0 = ret.Error(0) } return r0 } // MockStreamContextWrapper_SendHeader_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendHeader' type MockStreamContextWrapper_SendHeader_Call struct { *mock.Call } // SendHeader is a helper method to define mock.On call // - _a0 metadata.MD func (_e *MockStreamContextWrapper_Expecter) SendHeader(_a0 interface{}) *MockStreamContextWrapper_SendHeader_Call { return &MockStreamContextWrapper_SendHeader_Call{Call: _e.mock.On("SendHeader", _a0)} } func (_c *MockStreamContextWrapper_SendHeader_Call) Run(run func(_a0 metadata.MD)) *MockStreamContextWrapper_SendHeader_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(metadata.MD)) }) return _c } func (_c *MockStreamContextWrapper_SendHeader_Call) Return(_a0 error) *MockStreamContextWrapper_SendHeader_Call { _c.Call.Return(_a0) return _c } func (_c *MockStreamContextWrapper_SendHeader_Call) RunAndReturn(run func(metadata.MD) error) *MockStreamContextWrapper_SendHeader_Call { _c.Call.Return(run) return _c } // SendMsg provides a mock function with given fields: m func (_m *MockStreamContextWrapper) SendMsg(m any) error { ret := _m.Called(m) if len(ret) == 0 { panic("no return value specified for SendMsg") } var r0 error if rf, ok := ret.Get(0).(func(any) error); ok { r0 = rf(m) } else { r0 = ret.Error(0) } return r0 } // MockStreamContextWrapper_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' type MockStreamContextWrapper_SendMsg_Call struct { *mock.Call } // SendMsg is a helper method to define mock.On call // - m any func (_e *MockStreamContextWrapper_Expecter) SendMsg(m interface{}) *MockStreamContextWrapper_SendMsg_Call { return &MockStreamContextWrapper_SendMsg_Call{Call: _e.mock.On("SendMsg", m)} } func (_c *MockStreamContextWrapper_SendMsg_Call) Run(run func(m any)) *MockStreamContextWrapper_SendMsg_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(any)) }) return _c } func (_c *MockStreamContextWrapper_SendMsg_Call) Return(_a0 error) *MockStreamContextWrapper_SendMsg_Call { _c.Call.Return(_a0) return _c } func (_c *MockStreamContextWrapper_SendMsg_Call) RunAndReturn(run func(any) error) *MockStreamContextWrapper_SendMsg_Call { _c.Call.Return(run) return _c } // SetContext provides a mock function with given fields: _a0 func (_m *MockStreamContextWrapper) SetContext(_a0 context.Context) { _m.Called(_a0) } // MockStreamContextWrapper_SetContext_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetContext' type MockStreamContextWrapper_SetContext_Call struct { *mock.Call } // SetContext is a helper method to define mock.On call // - _a0 context.Context func (_e *MockStreamContextWrapper_Expecter) SetContext(_a0 interface{}) *MockStreamContextWrapper_SetContext_Call { return &MockStreamContextWrapper_SetContext_Call{Call: _e.mock.On("SetContext", _a0)} } func (_c *MockStreamContextWrapper_SetContext_Call) Run(run func(_a0 context.Context)) *MockStreamContextWrapper_SetContext_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(context.Context)) }) return _c } func (_c *MockStreamContextWrapper_SetContext_Call) Return() *MockStreamContextWrapper_SetContext_Call { _c.Call.Return() return _c } func (_c *MockStreamContextWrapper_SetContext_Call) RunAndReturn(run func(context.Context)) *MockStreamContextWrapper_SetContext_Call { _c.Run(run) return _c } // SetHeader provides a mock function with given fields: _a0 func (_m *MockStreamContextWrapper) SetHeader(_a0 metadata.MD) error { ret := _m.Called(_a0) if len(ret) == 0 { panic("no return value specified for SetHeader") } var r0 error if rf, ok := ret.Get(0).(func(metadata.MD) error); ok { r0 = rf(_a0) } else { r0 = ret.Error(0) } return r0 } // MockStreamContextWrapper_SetHeader_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetHeader' type MockStreamContextWrapper_SetHeader_Call struct { *mock.Call } // SetHeader is a helper method to define mock.On call // - _a0 metadata.MD func (_e *MockStreamContextWrapper_Expecter) SetHeader(_a0 interface{}) *MockStreamContextWrapper_SetHeader_Call { return &MockStreamContextWrapper_SetHeader_Call{Call: _e.mock.On("SetHeader", _a0)} } func (_c *MockStreamContextWrapper_SetHeader_Call) Run(run func(_a0 metadata.MD)) *MockStreamContextWrapper_SetHeader_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(metadata.MD)) }) return _c } func (_c *MockStreamContextWrapper_SetHeader_Call) Return(_a0 error) *MockStreamContextWrapper_SetHeader_Call { _c.Call.Return(_a0) return _c } func (_c *MockStreamContextWrapper_SetHeader_Call) RunAndReturn(run func(metadata.MD) error) *MockStreamContextWrapper_SetHeader_Call { _c.Call.Return(run) return _c } // SetTrailer provides a mock function with given fields: _a0 func (_m *MockStreamContextWrapper) SetTrailer(_a0 metadata.MD) { _m.Called(_a0) } // MockStreamContextWrapper_SetTrailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetTrailer' type MockStreamContextWrapper_SetTrailer_Call struct { *mock.Call } // SetTrailer is a helper method to define mock.On call // - _a0 metadata.MD func (_e *MockStreamContextWrapper_Expecter) SetTrailer(_a0 interface{}) *MockStreamContextWrapper_SetTrailer_Call { return &MockStreamContextWrapper_SetTrailer_Call{Call: _e.mock.On("SetTrailer", _a0)} } func (_c *MockStreamContextWrapper_SetTrailer_Call) Run(run func(_a0 metadata.MD)) *MockStreamContextWrapper_SetTrailer_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(metadata.MD)) }) return _c } func (_c *MockStreamContextWrapper_SetTrailer_Call) Return() *MockStreamContextWrapper_SetTrailer_Call { _c.Call.Return() return _c } func (_c *MockStreamContextWrapper_SetTrailer_Call) RunAndReturn(run func(metadata.MD)) *MockStreamContextWrapper_SetTrailer_Call { _c.Run(run) return _c } // NewMockStreamContextWrapper creates a new instance of MockStreamContextWrapper. 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 NewMockStreamContextWrapper(t interface { mock.TestingT Cleanup(func()) }) *MockStreamContextWrapper { mock := &MockStreamContextWrapper{} mock.Mock.Test(t) t.Cleanup(func() { mock.AssertExpectations(t) }) return mock }