// Code generated by MockGen. DO NOT EDIT.
// Source: github.com/kweaver-ai/kweaver-core/decision-agent/agent-backend/agent-factory/src/port/driven/idbaccess (interfaces: IPubedAgentRepo)
//
// Generated by this command:
//
//	mockgen -package idbaccessmock -destination ./idbaccessmock/pubed_agent.go github.com/kweaver-ai/kweaver-core/decision-agent/agent-backend/agent-factory/src/port/driven/idbaccess IPubedAgentRepo
//

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

import (
	context "context"
	sql "database/sql"
	reflect "reflect"

	padbarg "github.com/kweaver-ai/kweaver-core/decision-agent/agent-backend/agent-factory/src/drivenadapter/dbaccess/pubedagentdbacc/padbarg"
	padbret "github.com/kweaver-ai/kweaver-core/decision-agent/agent-backend/agent-factory/src/drivenadapter/dbaccess/pubedagentdbacc/padbret"
	pubedreq "github.com/kweaver-ai/kweaver-core/decision-agent/agent-backend/agent-factory/src/driveradapter/api/rdto/published/pubedreq"
	dapo "github.com/kweaver-ai/kweaver-core/decision-agent/agent-backend/agent-factory/src/infra/persistence/dapo"
	sqlx "github.com/kweaver-ai/proton-rds-sdk-go/sqlx"
	gomock "go.uber.org/mock/gomock"
)

// MockIPubedAgentRepo is a mock of IPubedAgentRepo interface.
type MockIPubedAgentRepo struct {
	ctrl     *gomock.Controller
	recorder *MockIPubedAgentRepoMockRecorder
	isgomock struct{}
}

// MockIPubedAgentRepoMockRecorder is the mock recorder for MockIPubedAgentRepo.
type MockIPubedAgentRepoMockRecorder struct {
	mock *MockIPubedAgentRepo
}

// NewMockIPubedAgentRepo creates a new mock instance.
func NewMockIPubedAgentRepo(ctrl *gomock.Controller) *MockIPubedAgentRepo {
	mock := &MockIPubedAgentRepo{ctrl: ctrl}
	mock.recorder = &MockIPubedAgentRepoMockRecorder{mock}
	return mock
}

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

// BeginTx mocks base method.
func (m *MockIPubedAgentRepo) BeginTx(ctx context.Context) (*sql.Tx, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "BeginTx", ctx)
	ret0, _ := ret[0].(*sql.Tx)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// BeginTx indicates an expected call of BeginTx.
func (mr *MockIPubedAgentRepoMockRecorder) BeginTx(ctx any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BeginTx", reflect.TypeOf((*MockIPubedAgentRepo)(nil).BeginTx), ctx)
}

// GetDB mocks base method.
func (m *MockIPubedAgentRepo) GetDB() *sqlx.DB {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetDB")
	ret0, _ := ret[0].(*sqlx.DB)
	return ret0
}

// GetDB indicates an expected call of GetDB.
func (mr *MockIPubedAgentRepoMockRecorder) GetDB() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDB", reflect.TypeOf((*MockIPubedAgentRepo)(nil).GetDB))
}

// GetPubedList mocks base method.
func (m *MockIPubedAgentRepo) GetPubedList(ctx context.Context, req *pubedreq.PubedAgentListReq) ([]*dapo.PublishedJoinPo, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPubedList", ctx, req)
	ret0, _ := ret[0].([]*dapo.PublishedJoinPo)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPubedList indicates an expected call of GetPubedList.
func (mr *MockIPubedAgentRepoMockRecorder) GetPubedList(ctx, req any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPubedList", reflect.TypeOf((*MockIPubedAgentRepo)(nil).GetPubedList), ctx, req)
}

// GetPubedListByXx mocks base method.
func (m *MockIPubedAgentRepo) GetPubedListByXx(ctx context.Context, arg *padbarg.GetPaPoListByXxArg) (*padbret.GetPaPoListByXxRet, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPubedListByXx", ctx, arg)
	ret0, _ := ret[0].(*padbret.GetPaPoListByXxRet)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPubedListByXx indicates an expected call of GetPubedListByXx.
func (mr *MockIPubedAgentRepoMockRecorder) GetPubedListByXx(ctx, arg any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPubedListByXx", reflect.TypeOf((*MockIPubedAgentRepo)(nil).GetPubedListByXx), ctx, arg)
}

// GetPubedPoMapByXx mocks base method.
func (m *MockIPubedAgentRepo) GetPubedPoMapByXx(ctx context.Context, arg *padbarg.GetPaPoListByXxArg) (*padbret.GetPaPoMapByXxRet, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPubedPoMapByXx", ctx, arg)
	ret0, _ := ret[0].(*padbret.GetPaPoMapByXxRet)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPubedPoMapByXx indicates an expected call of GetPubedPoMapByXx.
func (mr *MockIPubedAgentRepoMockRecorder) GetPubedPoMapByXx(ctx, arg any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPubedPoMapByXx", reflect.TypeOf((*MockIPubedAgentRepo)(nil).GetPubedPoMapByXx), ctx, arg)
}
