github.com/rohankumardubey/aresdb@v0.0.2-0.20190517170215-e54e3ca06b9c/memstore/mocks/Scheduler.go

//  Copyright (c) 2017-2018 Uber Technologies, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Code generated by mockery v1.0.0. DO NOT EDIT.
package mocks

import common "github.com/uber/aresdb/memstore/common"
import memstore "github.com/uber/aresdb/memstore"
import mock "github.com/stretchr/testify/mock"

// Scheduler is an autogenerated mock type for the Scheduler type
type Scheduler struct {
	mock.Mock
}

// DeleteTable provides a mock function with given fields: table, isFactTable
func (_m *Scheduler) DeleteTable(table string, isFactTable bool) {
	_m.Called(table, isFactTable)
}

// EnableJobType provides a mock function with given fields: jobType, enable
func (_m *Scheduler) EnableJobType(jobType common.JobType, enable bool) {
	_m.Called(jobType, enable)
}

// GetJobDetails provides a mock function with given fields: jobType
func (_m *Scheduler) GetJobDetails(jobType common.JobType) interface{} {
	ret := _m.Called(jobType)

	var r0 interface{}
	if rf, ok := ret.Get(0).(func(common.JobType) interface{}); ok {
		r0 = rf(jobType)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(interface{})
		}
	}

	return r0
}

// IsJobTypeEnabled provides a mock function with given fields: jobType
func (_m *Scheduler) IsJobTypeEnabled(jobType common.JobType) bool {
	ret := _m.Called(jobType)

	var r0 bool
	if rf, ok := ret.Get(0).(func(common.JobType) bool); ok {
		r0 = rf(jobType)
	} else {
		r0 = ret.Get(0).(bool)
	}

	return r0
}

// Lock provides a mock function with given fields:
func (_m *Scheduler) Lock() {
	_m.Called()
}

// NewArchivingJob provides a mock function with given fields: tableName, shardID, cutoff
func (_m *Scheduler) NewArchivingJob(tableName string, shardID int, cutoff uint32) memstore.Job {
	ret := _m.Called(tableName, shardID, cutoff)

	var r0 memstore.Job
	if rf, ok := ret.Get(0).(func(string, int, uint32) memstore.Job); ok {
		r0 = rf(tableName, shardID, cutoff)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(memstore.Job)
		}
	}

	return r0
}

// NewBackfillJob provides a mock function with given fields: tableName, shardID
func (_m *Scheduler) NewBackfillJob(tableName string, shardID int) memstore.Job {
	ret := _m.Called(tableName, shardID)

	var r0 memstore.Job
	if rf, ok := ret.Get(0).(func(string, int) memstore.Job); ok {
		r0 = rf(tableName, shardID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(memstore.Job)
		}
	}

	return r0
}

// NewPurgeJob provides a mock function with given fields: tableName, shardID, batchIDStart, batchIDEnd
func (_m *Scheduler) NewPurgeJob(tableName string, shardID int, batchIDStart int, batchIDEnd int) memstore.Job {
	ret := _m.Called(tableName, shardID, batchIDStart, batchIDEnd)

	var r0 memstore.Job
	if rf, ok := ret.Get(0).(func(string, int, int, int) memstore.Job); ok {
		r0 = rf(tableName, shardID, batchIDStart, batchIDEnd)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(memstore.Job)
		}
	}

	return r0
}

// NewSnapshotJob provides a mock function with given fields: tableName, shardID
func (_m *Scheduler) NewSnapshotJob(tableName string, shardID int) memstore.Job {
	ret := _m.Called(tableName, shardID)

	var r0 memstore.Job
	if rf, ok := ret.Get(0).(func(string, int) memstore.Job); ok {
		r0 = rf(tableName, shardID)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(memstore.Job)
		}
	}

	return r0
}

// RLock provides a mock function with given fields:
func (_m *Scheduler) RLock() {
	_m.Called()
}

// RUnlock provides a mock function with given fields:
func (_m *Scheduler) RUnlock() {
	_m.Called()
}

// Start provides a mock function with given fields:
func (_m *Scheduler) Start() {
	_m.Called()
}

// Stop provides a mock function with given fields:
func (_m *Scheduler) Stop() {
	_m.Called()
}

// SubmitJob provides a mock function with given fields: job
func (_m *Scheduler) SubmitJob(job memstore.Job) (error, chan error) {
	ret := _m.Called(job)

	var r0 error
	if rf, ok := ret.Get(0).(func(memstore.Job) error); ok {
		r0 = rf(job)
	} else {
		r0 = ret.Error(0)
	}

	var r1 chan error
	if rf, ok := ret.Get(1).(func(memstore.Job) chan error); ok {
		r1 = rf(job)
	} else {
		if ret.Get(1) != nil {
			r1 = ret.Get(1).(chan error)
		}
	}

	return r0, r1
}

// Unlock provides a mock function with given fields:
func (_m *Scheduler) Unlock() {
	_m.Called()
}