// Code generated by MockGen. DO NOT EDIT.
// Source: application.go
//
// Generated by this command:
//
//	mockgen -source application.go -destination application_mock.go -package=application
//

// Package application is a generated GoMock package.
package application

import (
	context "context"
	reflect "reflect"

	gatewayv1 "github.com/tidbcloud/aisupport/packages/sdk-go/contracts/api/agent-gateway/gateway/v1"
	gatewayv10 "github.com/tidbcloud/aisupport/packages/sdk-go/contracts/entity/agent-gateway/gateway/v1"
	gomock "go.uber.org/mock/gomock"
)

// MockApplication is a mock of Application interface.
type MockApplication struct {
	ctrl     *gomock.Controller
	recorder *MockApplicationMockRecorder
	isgomock struct{}
}

// MockApplicationMockRecorder is the mock recorder for MockApplication.
type MockApplicationMockRecorder struct {
	mock *MockApplication
}

// NewMockApplication creates a new mock instance.
func NewMockApplication(ctrl *gomock.Controller) *MockApplication {
	mock := &MockApplication{ctrl: ctrl}
	mock.recorder = &MockApplicationMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockApplication) EXPECT() *MockApplicationMockRecorder {
	return m.recorder
}

// HandleApplicationTaskEvent mocks base method.
func (m *MockApplication) HandleApplicationTaskEvent(arg0 context.Context, arg1 *gatewayv10.TaskEvent) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "HandleApplicationTaskEvent", arg0, arg1)
	ret0, _ := ret[0].(error)
	return ret0
}

// HandleApplicationTaskEvent indicates an expected call of HandleApplicationTaskEvent.
func (mr *MockApplicationMockRecorder) HandleApplicationTaskEvent(arg0, arg1 any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleApplicationTaskEvent", reflect.TypeOf((*MockApplication)(nil).HandleApplicationTaskEvent), arg0, arg1)
}

// RegisterApplicationTaskEventHandler mocks base method.
func (m *MockApplication) RegisterApplicationTaskEventHandler(arg0 string, arg1 ApplicationTaskEventHandler) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RegisterApplicationTaskEventHandler", arg0, arg1)
	ret0, _ := ret[0].(error)
	return ret0
}

// RegisterApplicationTaskEventHandler indicates an expected call of RegisterApplicationTaskEventHandler.
func (mr *MockApplicationMockRecorder) RegisterApplicationTaskEventHandler(arg0, arg1 any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterApplicationTaskEventHandler", reflect.TypeOf((*MockApplication)(nil).RegisterApplicationTaskEventHandler), arg0, arg1)
}

// SendApplicationMessage mocks base method.
func (m *MockApplication) SendApplicationMessage(arg0 context.Context, arg1 *gatewayv1.ProcessMessageRequest) (*gatewayv1.ProcessMessageResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "SendApplicationMessage", arg0, arg1)
	ret0, _ := ret[0].(*gatewayv1.ProcessMessageResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// SendApplicationMessage indicates an expected call of SendApplicationMessage.
func (mr *MockApplicationMockRecorder) SendApplicationMessage(arg0, arg1 any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendApplicationMessage", reflect.TypeOf((*MockApplication)(nil).SendApplicationMessage), arg0, arg1)
}