// Code generated by MockGen. DO NOT EDIT.
// Source: ./internel/dao/org/org.go

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

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

	gomock "github.com/golang/mock/gomock"
	entity "github.com/tidbcloud/account/internel/dao/entity"
	gorm "gorm.io/gorm"
)

// MockOrgDaoIface is a mock of OrgDaoIface interface.
type MockOrgDaoIface struct {
	ctrl     *gomock.Controller
	recorder *MockOrgDaoIfaceMockRecorder
}

// MockOrgDaoIfaceMockRecorder is the mock recorder for MockOrgDaoIface.
type MockOrgDaoIfaceMockRecorder struct {
	mock *MockOrgDaoIface
}

// NewMockOrgDaoIface creates a new mock instance.
func NewMockOrgDaoIface(ctrl *gomock.Controller) *MockOrgDaoIface {
	mock := &MockOrgDaoIface{ctrl: ctrl}
	mock.recorder = &MockOrgDaoIfaceMockRecorder{mock}
	return mock
}

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

// CreateOrg mocks base method.
func (m *MockOrgDaoIface) CreateOrg(ctx context.Context, tx *gorm.DB, tenant entity.Tenant) (*entity.Tenant, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreateOrg", ctx, tx, tenant)
	ret0, _ := ret[0].(*entity.Tenant)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// CreateOrg indicates an expected call of CreateOrg.
func (mr *MockOrgDaoIfaceMockRecorder) CreateOrg(ctx, tx, tenant interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrg", reflect.TypeOf((*MockOrgDaoIface)(nil).CreateOrg), ctx, tx, tenant)
}

// CreateTrial mocks base method.
func (m *MockOrgDaoIface) CreateTrial(ctx context.Context, tx *gorm.DB, creatorUserID, orgID uint64, startedAt time.Time, estimatedEndedAt, endedAt *time.Time, trialType string) (*entity.Trial, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "CreateTrial", ctx, tx, creatorUserID, orgID, startedAt, estimatedEndedAt, endedAt, trialType)
	ret0, _ := ret[0].(*entity.Trial)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// CreateTrial indicates an expected call of CreateTrial.
func (mr *MockOrgDaoIfaceMockRecorder) CreateTrial(ctx, tx, creatorUserID, orgID, startedAt, estimatedEndedAt, endedAt, trialType interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTrial", reflect.TypeOf((*MockOrgDaoIface)(nil).CreateTrial), ctx, tx, creatorUserID, orgID, startedAt, estimatedEndedAt, endedAt, trialType)
}

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

// DeleteOrgFromPartner indicates an expected call of DeleteOrgFromPartner.
func (mr *MockOrgDaoIfaceMockRecorder) DeleteOrgFromPartner(ctx, tx, orgId interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrgFromPartner", reflect.TypeOf((*MockOrgDaoIface)(nil).DeleteOrgFromPartner), ctx, tx, orgId)
}

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

// DeleteOrgImpersonates indicates an expected call of DeleteOrgImpersonates.
func (mr *MockOrgDaoIfaceMockRecorder) DeleteOrgImpersonates(ctx, tx, orgID interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteOrgImpersonates", reflect.TypeOf((*MockOrgDaoIface)(nil).DeleteOrgImpersonates), ctx, tx, orgID)
}

// GetOrgByCompanyName mocks base method.
func (m *MockOrgDaoIface) GetOrgByCompanyName(ctx context.Context, tx *gorm.DB, companyName string) (*entity.Tenant, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetOrgByCompanyName", ctx, tx, companyName)
	ret0, _ := ret[0].(*entity.Tenant)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetOrgByCompanyName indicates an expected call of GetOrgByCompanyName.
func (mr *MockOrgDaoIfaceMockRecorder) GetOrgByCompanyName(ctx, tx, companyName interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgByCompanyName", reflect.TypeOf((*MockOrgDaoIface)(nil).GetOrgByCompanyName), ctx, tx, companyName)
}

// GetOrgByOrgID mocks base method.
func (m *MockOrgDaoIface) GetOrgByOrgID(ctx context.Context, tx *gorm.DB, orgID uint64) (*entity.Tenant, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetOrgByOrgID", ctx, tx, orgID)
	ret0, _ := ret[0].(*entity.Tenant)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetOrgByOrgID indicates an expected call of GetOrgByOrgID.
func (mr *MockOrgDaoIfaceMockRecorder) GetOrgByOrgID(ctx, tx, orgID interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOrgByOrgID", reflect.TypeOf((*MockOrgDaoIface)(nil).GetOrgByOrgID), ctx, tx, orgID)
}

// ListOrgs mocks base method.
func (m *MockOrgDaoIface) ListOrgs(ctx context.Context, tx *gorm.DB, param ListOrgsParam) (*ListOrgsResult, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListOrgs", ctx, tx, param)
	ret0, _ := ret[0].(*ListOrgsResult)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListOrgs indicates an expected call of ListOrgs.
func (mr *MockOrgDaoIfaceMockRecorder) ListOrgs(ctx, tx, param interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListOrgs", reflect.TypeOf((*MockOrgDaoIface)(nil).ListOrgs), ctx, tx, param)
}

// ListOrgsByFilter mocks base method.
func (m *MockOrgDaoIface) ListOrgsByFilter(ctx context.Context, db *gorm.DB, filter ListOrgsFilter) ([]entity.Tenant, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListOrgsByFilter", ctx, db, filter)
	ret0, _ := ret[0].([]entity.Tenant)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListOrgsByFilter indicates an expected call of ListOrgsByFilter.
func (mr *MockOrgDaoIfaceMockRecorder) ListOrgsByFilter(ctx, db, filter interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListOrgsByFilter", reflect.TypeOf((*MockOrgDaoIface)(nil).ListOrgsByFilter), ctx, db, filter)
}

// ListTrials mocks base method.
func (m *MockOrgDaoIface) ListTrials(ctx context.Context, tx *gorm.DB, orgID uint64, trialType string) ([]entity.Trial, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListTrials", ctx, tx, orgID, trialType)
	ret0, _ := ret[0].([]entity.Trial)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ListTrials indicates an expected call of ListTrials.
func (mr *MockOrgDaoIfaceMockRecorder) ListTrials(ctx, tx, orgID, trialType interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListTrials", reflect.TypeOf((*MockOrgDaoIface)(nil).ListTrials), ctx, tx, orgID, trialType)
}

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

// MarkOrgAsDeleted indicates an expected call of MarkOrgAsDeleted.
func (mr *MockOrgDaoIfaceMockRecorder) MarkOrgAsDeleted(ctx, tx, orgID interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MarkOrgAsDeleted", reflect.TypeOf((*MockOrgDaoIface)(nil).MarkOrgAsDeleted), ctx, tx, orgID)
}

// OpenOrgSso mocks base method.
func (m *MockOrgDaoIface) OpenOrgSso(ctx context.Context, tx *gorm.DB, orgID uint64, orgType, companyName string) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "OpenOrgSso", ctx, tx, orgID, orgType, companyName)
	ret0, _ := ret[0].(error)
	return ret0
}

// OpenOrgSso indicates an expected call of OpenOrgSso.
func (mr *MockOrgDaoIfaceMockRecorder) OpenOrgSso(ctx, tx, orgID, orgType, companyName interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenOrgSso", reflect.TypeOf((*MockOrgDaoIface)(nil).OpenOrgSso), ctx, tx, orgID, orgType, companyName)
}

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

// UpdateOrg indicates an expected call of UpdateOrg.
func (mr *MockOrgDaoIfaceMockRecorder) UpdateOrg(ctx, tx, param interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateOrg", reflect.TypeOf((*MockOrgDaoIface)(nil).UpdateOrg), ctx, tx, param)
}