// Code generated by MockGen. DO NOT EDIT. // Source: eventbus.go // // Generated by this command: // // mockgen -destination ../../internal/mock/infra/eventbus/eventbus_mock.go -package mock -source eventbus.go Factory // // Package mock is a generated GoMock package. package mock import ( context "context" reflect "reflect" eventbus "github.com/coze-dev/coze-studio/backend/infra/eventbus" gomock "go.uber.org/mock/gomock" ) // MockProducer is a mock of Producer interface. type MockProducer struct { ctrl *gomock.Controller recorder *MockProducerMockRecorder isgomock struct{} } // MockProducerMockRecorder is the mock recorder for MockProducer. type MockProducerMockRecorder struct { mock *MockProducer } // NewMockProducer creates a new mock instance. func NewMockProducer(ctrl *gomock.Controller) *MockProducer { mock := &MockProducer{ctrl: ctrl} mock.recorder = &MockProducerMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockProducer) EXPECT() *MockProducerMockRecorder { return m.recorder } // BatchSend mocks base method. func (m *MockProducer) BatchSend(ctx context.Context, bodyArr [][]byte, opts ...eventbus.SendOpt) error { m.ctrl.T.Helper() varargs := []any{ctx, bodyArr} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "BatchSend", varargs...) ret0, _ := ret[0].(error) return ret0 } // BatchSend indicates an expected call of BatchSend. func (mr *MockProducerMockRecorder) BatchSend(ctx, bodyArr any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]any{ctx, bodyArr}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BatchSend", reflect.TypeOf((*MockProducer)(nil).BatchSend), varargs...) } // Send mocks base method. func (m *MockProducer) Send(ctx context.Context, body []byte, opts ...eventbus.SendOpt) error { m.ctrl.T.Helper() varargs := []any{ctx, body} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "Send", varargs...) ret0, _ := ret[0].(error) return ret0 } // Send indicates an expected call of Send. func (mr *MockProducerMockRecorder) Send(ctx, body any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]any{ctx, body}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockProducer)(nil).Send), varargs...) } // MockConsumerService is a mock of ConsumerService interface. type MockConsumerService struct { ctrl *gomock.Controller recorder *MockConsumerServiceMockRecorder isgomock struct{} } // MockConsumerServiceMockRecorder is the mock recorder for MockConsumerService. type MockConsumerServiceMockRecorder struct { mock *MockConsumerService } // NewMockConsumerService creates a new mock instance. func NewMockConsumerService(ctrl *gomock.Controller) *MockConsumerService { mock := &MockConsumerService{ctrl: ctrl} mock.recorder = &MockConsumerServiceMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockConsumerService) EXPECT() *MockConsumerServiceMockRecorder { return m.recorder } // RegisterConsumer mocks base method. func (m *MockConsumerService) RegisterConsumer(nameServer, topic, group string, consumerHandler eventbus.ConsumerHandler, opts ...eventbus.ConsumerOpt) error { m.ctrl.T.Helper() varargs := []any{nameServer, topic, group, consumerHandler} for _, a := range opts { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "RegisterConsumer", varargs...) ret0, _ := ret[0].(error) return ret0 } // RegisterConsumer indicates an expected call of RegisterConsumer. func (mr *MockConsumerServiceMockRecorder) RegisterConsumer(nameServer, topic, group, consumerHandler any, opts ...any) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]any{nameServer, topic, group, consumerHandler}, opts...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterConsumer", reflect.TypeOf((*MockConsumerService)(nil).RegisterConsumer), varargs...) } // MockConsumerHandler is a mock of ConsumerHandler interface. type MockConsumerHandler struct { ctrl *gomock.Controller recorder *MockConsumerHandlerMockRecorder isgomock struct{} } // MockConsumerHandlerMockRecorder is the mock recorder for MockConsumerHandler. type MockConsumerHandlerMockRecorder struct { mock *MockConsumerHandler } // NewMockConsumerHandler creates a new mock instance. func NewMockConsumerHandler(ctrl *gomock.Controller) *MockConsumerHandler { mock := &MockConsumerHandler{ctrl: ctrl} mock.recorder = &MockConsumerHandlerMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockConsumerHandler) EXPECT() *MockConsumerHandlerMockRecorder { return m.recorder } // HandleMessage mocks base method. func (m *MockConsumerHandler) HandleMessage(ctx context.Context, msg *eventbus.Message) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "HandleMessage", ctx, msg) ret0, _ := ret[0].(error) return ret0 } // HandleMessage indicates an expected call of HandleMessage. func (mr *MockConsumerHandlerMockRecorder) HandleMessage(ctx, msg any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleMessage", reflect.TypeOf((*MockConsumerHandler)(nil).HandleMessage), ctx, msg) }