// Code generated by MockGen. DO NOT EDIT.
// Source: github.com/tidbcloud/pkgv2/db (interfaces: TransProvider)
//
// Generated by this command:
//
//	mockgen -destination db_mock.go -package=test github.com/tidbcloud/pkgv2/db TransProvider
//

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

import (
	context "context"
	reflect "reflect"

	gomock "go.uber.org/mock/gomock"
	gorm "gorm.io/gorm"
)

// MockTransProvider is a mock of TransProvider interface.
type MockTransProvider struct {
	ctrl     *gomock.Controller
	recorder *MockTransProviderMockRecorder
	isgomock struct{}
}

// MockTransProviderMockRecorder is the mock recorder for MockTransProvider.
type MockTransProviderMockRecorder struct {
	mock *MockTransProvider
}

// NewMockTransProvider creates a new mock instance.
func NewMockTransProvider(ctrl *gomock.Controller) *MockTransProvider {
	mock := &MockTransProvider{ctrl: ctrl}
	mock.recorder = &MockTransProviderMockRecorder{mock}
	return mock
}

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

// ReadDB mocks base method.
func (m *MockTransProvider) ReadDB(ctx context.Context) *gorm.DB {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ReadDB", ctx)
	ret0, _ := ret[0].(*gorm.DB)
	return ret0
}

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

// Transaction mocks base method.
func (m *MockTransProvider) Transaction(ctx context.Context, cb func(context.Context) error) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Transaction", ctx, cb)
	ret0, _ := ret[0].(error)
	return ret0
}

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

// WriteDB mocks base method.
func (m *MockTransProvider) WriteDB(ctx context.Context) *gorm.DB {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "WriteDB", ctx)
	ret0, _ := ret[0].(*gorm.DB)
	return ret0
}

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