// Code generated by MockGen. DO NOT EDIT.
// Source: ./central/repository/trial.go

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

import (
	context "context"
	reflect "reflect"
	time "time"

	gomock "github.com/golang/mock/gomock"
	entity "github.com/pingcap/dbaas/central/db/entity"
)

// MockTrialRepository is a mock of TrialRepository interface.
type MockTrialRepository struct {
	ctrl     *gomock.Controller
	recorder *MockTrialRepositoryMockRecorder
}

// MockTrialRepositoryMockRecorder is the mock recorder for MockTrialRepository.
type MockTrialRepositoryMockRecorder struct {
	mock *MockTrialRepository
}

// NewMockTrialRepository creates a new mock instance.
func NewMockTrialRepository(ctrl *gomock.Controller) *MockTrialRepository {
	mock := &MockTrialRepository{ctrl: ctrl}
	mock.recorder = &MockTrialRepositoryMockRecorder{mock}
	return mock
}

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

// QueryCloseToEnd mocks base method.
func (m *MockTrialRepository) QueryCloseToEnd(ctx context.Context, startedAt time.Time) ([]entity.Trial, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "QueryCloseToEnd", ctx, startedAt)
	ret0, _ := ret[0].([]entity.Trial)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// QueryCloseToEnd indicates an expected call of QueryCloseToEnd.
func (mr *MockTrialRepositoryMockRecorder) QueryCloseToEnd(ctx, startedAt interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryCloseToEnd", reflect.TypeOf((*MockTrialRepository)(nil).QueryCloseToEnd), ctx, startedAt)
}

// Save mocks base method.
func (m *MockTrialRepository) Save(ctx context.Context, en *entity.Trial) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Save", ctx, en)
	ret0, _ := ret[0].(error)
	return ret0
}

// Save indicates an expected call of Save.
func (mr *MockTrialRepositoryMockRecorder) Save(ctx, en interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Save", reflect.TypeOf((*MockTrialRepository)(nil).Save), ctx, en)
}