fake_worker_provider.go

// This file was generated by counterfeiter
package workerfakes

import (
	"sync"

	"code.cloudfoundry.org/lager"
	"github.com/concourse/atc"
	"github.com/concourse/atc/dbng"
	"github.com/concourse/atc/worker"
)

type FakeWorkerProvider struct {
	RunningWorkersStub        func() ([]worker.Worker, error)
	runningWorkersMutex       sync.RWMutex
	runningWorkersArgsForCall []struct{}
	runningWorkersReturns     struct {
		result1 []worker.Worker
		result2 error
	}
	runningWorkersReturnsOnCall map[int]struct {
		result1 []worker.Worker
		result2 error
	}
	GetWorkerStub        func(string) (worker.Worker, bool, error)
	getWorkerMutex       sync.RWMutex
	getWorkerArgsForCall []struct {
		arg1 string
	}
	getWorkerReturns struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	getWorkerReturnsOnCall map[int]struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	FindWorkerForContainerStub        func(logger lager.Logger, teamID int, handle string) (worker.Worker, bool, error)
	findWorkerForContainerMutex       sync.RWMutex
	findWorkerForContainerArgsForCall []struct {
		logger lager.Logger
		teamID int
		handle string
	}
	findWorkerForContainerReturns struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	findWorkerForContainerReturnsOnCall map[int]struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	FindWorkerForResourceCheckContainerStub        func(logger lager.Logger, teamID int, resourceUser dbng.ResourceUser, resourceType string, resourceSource atc.Source, types atc.VersionedResourceTypes) (worker.Worker, bool, error)
	findWorkerForResourceCheckContainerMutex       sync.RWMutex
	findWorkerForResourceCheckContainerArgsForCall []struct {
		logger         lager.Logger
		teamID         int
		resourceUser   dbng.ResourceUser
		resourceType   string
		resourceSource atc.Source
		types          atc.VersionedResourceTypes
	}
	findWorkerForResourceCheckContainerReturns struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	findWorkerForResourceCheckContainerReturnsOnCall map[int]struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	FindWorkerForBuildContainerStub        func(logger lager.Logger, teamID int, buildID int, planID atc.PlanID) (worker.Worker, bool, error)
	findWorkerForBuildContainerMutex       sync.RWMutex
	findWorkerForBuildContainerArgsForCall []struct {
		logger  lager.Logger
		teamID  int
		buildID int
		planID  atc.PlanID
	}
	findWorkerForBuildContainerReturns struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	findWorkerForBuildContainerReturnsOnCall map[int]struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}
	invocations      map[string][][]interface{}
	invocationsMutex sync.RWMutex
}

func (fake *FakeWorkerProvider) RunningWorkers() ([]worker.Worker, error) {
	fake.runningWorkersMutex.Lock()
	ret, specificReturn := fake.runningWorkersReturnsOnCall[len(fake.runningWorkersArgsForCall)]
	fake.runningWorkersArgsForCall = append(fake.runningWorkersArgsForCall, struct{}{})
	fake.recordInvocation("RunningWorkers", []interface{}{})
	fake.runningWorkersMutex.Unlock()
	if fake.RunningWorkersStub != nil {
		return fake.RunningWorkersStub()
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	return fake.runningWorkersReturns.result1, fake.runningWorkersReturns.result2
}

func (fake *FakeWorkerProvider) RunningWorkersCallCount() int {
	fake.runningWorkersMutex.RLock()
	defer fake.runningWorkersMutex.RUnlock()
	return len(fake.runningWorkersArgsForCall)
}

func (fake *FakeWorkerProvider) RunningWorkersReturns(result1 []worker.Worker, result2 error) {
	fake.RunningWorkersStub = nil
	fake.runningWorkersReturns = struct {
		result1 []worker.Worker
		result2 error
	}{result1, result2}
}

func (fake *FakeWorkerProvider) RunningWorkersReturnsOnCall(i int, result1 []worker.Worker, result2 error) {
	fake.RunningWorkersStub = nil
	if fake.runningWorkersReturnsOnCall == nil {
		fake.runningWorkersReturnsOnCall = make(map[int]struct {
			result1 []worker.Worker
			result2 error
		})
	}
	fake.runningWorkersReturnsOnCall[i] = struct {
		result1 []worker.Worker
		result2 error
	}{result1, result2}
}

func (fake *FakeWorkerProvider) GetWorker(arg1 string) (worker.Worker, bool, error) {
	fake.getWorkerMutex.Lock()
	ret, specificReturn := fake.getWorkerReturnsOnCall[len(fake.getWorkerArgsForCall)]
	fake.getWorkerArgsForCall = append(fake.getWorkerArgsForCall, struct {
		arg1 string
	}{arg1})
	fake.recordInvocation("GetWorker", []interface{}{arg1})
	fake.getWorkerMutex.Unlock()
	if fake.GetWorkerStub != nil {
		return fake.GetWorkerStub(arg1)
	}
	if specificReturn {
		return ret.result1, ret.result2, ret.result3
	}
	return fake.getWorkerReturns.result1, fake.getWorkerReturns.result2, fake.getWorkerReturns.result3
}

func (fake *FakeWorkerProvider) GetWorkerCallCount() int {
	fake.getWorkerMutex.RLock()
	defer fake.getWorkerMutex.RUnlock()
	return len(fake.getWorkerArgsForCall)
}

func (fake *FakeWorkerProvider) GetWorkerArgsForCall(i int) string {
	fake.getWorkerMutex.RLock()
	defer fake.getWorkerMutex.RUnlock()
	return fake.getWorkerArgsForCall[i].arg1
}

func (fake *FakeWorkerProvider) GetWorkerReturns(result1 worker.Worker, result2 bool, result3 error) {
	fake.GetWorkerStub = nil
	fake.getWorkerReturns = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) GetWorkerReturnsOnCall(i int, result1 worker.Worker, result2 bool, result3 error) {
	fake.GetWorkerStub = nil
	if fake.getWorkerReturnsOnCall == nil {
		fake.getWorkerReturnsOnCall = make(map[int]struct {
			result1 worker.Worker
			result2 bool
			result3 error
		})
	}
	fake.getWorkerReturnsOnCall[i] = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) FindWorkerForContainer(logger lager.Logger, teamID int, handle string) (worker.Worker, bool, error) {
	fake.findWorkerForContainerMutex.Lock()
	ret, specificReturn := fake.findWorkerForContainerReturnsOnCall[len(fake.findWorkerForContainerArgsForCall)]
	fake.findWorkerForContainerArgsForCall = append(fake.findWorkerForContainerArgsForCall, struct {
		logger lager.Logger
		teamID int
		handle string
	}{logger, teamID, handle})
	fake.recordInvocation("FindWorkerForContainer", []interface{}{logger, teamID, handle})
	fake.findWorkerForContainerMutex.Unlock()
	if fake.FindWorkerForContainerStub != nil {
		return fake.FindWorkerForContainerStub(logger, teamID, handle)
	}
	if specificReturn {
		return ret.result1, ret.result2, ret.result3
	}
	return fake.findWorkerForContainerReturns.result1, fake.findWorkerForContainerReturns.result2, fake.findWorkerForContainerReturns.result3
}

func (fake *FakeWorkerProvider) FindWorkerForContainerCallCount() int {
	fake.findWorkerForContainerMutex.RLock()
	defer fake.findWorkerForContainerMutex.RUnlock()
	return len(fake.findWorkerForContainerArgsForCall)
}

func (fake *FakeWorkerProvider) FindWorkerForContainerArgsForCall(i int) (lager.Logger, int, string) {
	fake.findWorkerForContainerMutex.RLock()
	defer fake.findWorkerForContainerMutex.RUnlock()
	return fake.findWorkerForContainerArgsForCall[i].logger, fake.findWorkerForContainerArgsForCall[i].teamID, fake.findWorkerForContainerArgsForCall[i].handle
}

func (fake *FakeWorkerProvider) FindWorkerForContainerReturns(result1 worker.Worker, result2 bool, result3 error) {
	fake.FindWorkerForContainerStub = nil
	fake.findWorkerForContainerReturns = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) FindWorkerForContainerReturnsOnCall(i int, result1 worker.Worker, result2 bool, result3 error) {
	fake.FindWorkerForContainerStub = nil
	if fake.findWorkerForContainerReturnsOnCall == nil {
		fake.findWorkerForContainerReturnsOnCall = make(map[int]struct {
			result1 worker.Worker
			result2 bool
			result3 error
		})
	}
	fake.findWorkerForContainerReturnsOnCall[i] = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) FindWorkerForResourceCheckContainer(logger lager.Logger, teamID int, resourceUser dbng.ResourceUser, resourceType string, resourceSource atc.Source, types atc.VersionedResourceTypes) (worker.Worker, bool, error) {
	fake.findWorkerForResourceCheckContainerMutex.Lock()
	ret, specificReturn := fake.findWorkerForResourceCheckContainerReturnsOnCall[len(fake.findWorkerForResourceCheckContainerArgsForCall)]
	fake.findWorkerForResourceCheckContainerArgsForCall = append(fake.findWorkerForResourceCheckContainerArgsForCall, struct {
		logger         lager.Logger
		teamID         int
		resourceUser   dbng.ResourceUser
		resourceType   string
		resourceSource atc.Source
		types          atc.VersionedResourceTypes
	}{logger, teamID, resourceUser, resourceType, resourceSource, types})
	fake.recordInvocation("FindWorkerForResourceCheckContainer", []interface{}{logger, teamID, resourceUser, resourceType, resourceSource, types})
	fake.findWorkerForResourceCheckContainerMutex.Unlock()
	if fake.FindWorkerForResourceCheckContainerStub != nil {
		return fake.FindWorkerForResourceCheckContainerStub(logger, teamID, resourceUser, resourceType, resourceSource, types)
	}
	if specificReturn {
		return ret.result1, ret.result2, ret.result3
	}
	return fake.findWorkerForResourceCheckContainerReturns.result1, fake.findWorkerForResourceCheckContainerReturns.result2, fake.findWorkerForResourceCheckContainerReturns.result3
}

func (fake *FakeWorkerProvider) FindWorkerForResourceCheckContainerCallCount() int {
	fake.findWorkerForResourceCheckContainerMutex.RLock()
	defer fake.findWorkerForResourceCheckContainerMutex.RUnlock()
	return len(fake.findWorkerForResourceCheckContainerArgsForCall)
}

func (fake *FakeWorkerProvider) FindWorkerForResourceCheckContainerArgsForCall(i int) (lager.Logger, int, dbng.ResourceUser, string, atc.Source, atc.VersionedResourceTypes) {
	fake.findWorkerForResourceCheckContainerMutex.RLock()
	defer fake.findWorkerForResourceCheckContainerMutex.RUnlock()
	return fake.findWorkerForResourceCheckContainerArgsForCall[i].logger, fake.findWorkerForResourceCheckContainerArgsForCall[i].teamID, fake.findWorkerForResourceCheckContainerArgsForCall[i].resourceUser, fake.findWorkerForResourceCheckContainerArgsForCall[i].resourceType, fake.findWorkerForResourceCheckContainerArgsForCall[i].resourceSource, fake.findWorkerForResourceCheckContainerArgsForCall[i].types
}

func (fake *FakeWorkerProvider) FindWorkerForResourceCheckContainerReturns(result1 worker.Worker, result2 bool, result3 error) {
	fake.FindWorkerForResourceCheckContainerStub = nil
	fake.findWorkerForResourceCheckContainerReturns = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) FindWorkerForResourceCheckContainerReturnsOnCall(i int, result1 worker.Worker, result2 bool, result3 error) {
	fake.FindWorkerForResourceCheckContainerStub = nil
	if fake.findWorkerForResourceCheckContainerReturnsOnCall == nil {
		fake.findWorkerForResourceCheckContainerReturnsOnCall = make(map[int]struct {
			result1 worker.Worker
			result2 bool
			result3 error
		})
	}
	fake.findWorkerForResourceCheckContainerReturnsOnCall[i] = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) FindWorkerForBuildContainer(logger lager.Logger, teamID int, buildID int, planID atc.PlanID) (worker.Worker, bool, error) {
	fake.findWorkerForBuildContainerMutex.Lock()
	ret, specificReturn := fake.findWorkerForBuildContainerReturnsOnCall[len(fake.findWorkerForBuildContainerArgsForCall)]
	fake.findWorkerForBuildContainerArgsForCall = append(fake.findWorkerForBuildContainerArgsForCall, struct {
		logger  lager.Logger
		teamID  int
		buildID int
		planID  atc.PlanID
	}{logger, teamID, buildID, planID})
	fake.recordInvocation("FindWorkerForBuildContainer", []interface{}{logger, teamID, buildID, planID})
	fake.findWorkerForBuildContainerMutex.Unlock()
	if fake.FindWorkerForBuildContainerStub != nil {
		return fake.FindWorkerForBuildContainerStub(logger, teamID, buildID, planID)
	}
	if specificReturn {
		return ret.result1, ret.result2, ret.result3
	}
	return fake.findWorkerForBuildContainerReturns.result1, fake.findWorkerForBuildContainerReturns.result2, fake.findWorkerForBuildContainerReturns.result3
}

func (fake *FakeWorkerProvider) FindWorkerForBuildContainerCallCount() int {
	fake.findWorkerForBuildContainerMutex.RLock()
	defer fake.findWorkerForBuildContainerMutex.RUnlock()
	return len(fake.findWorkerForBuildContainerArgsForCall)
}

func (fake *FakeWorkerProvider) FindWorkerForBuildContainerArgsForCall(i int) (lager.Logger, int, int, atc.PlanID) {
	fake.findWorkerForBuildContainerMutex.RLock()
	defer fake.findWorkerForBuildContainerMutex.RUnlock()
	return fake.findWorkerForBuildContainerArgsForCall[i].logger, fake.findWorkerForBuildContainerArgsForCall[i].teamID, fake.findWorkerForBuildContainerArgsForCall[i].buildID, fake.findWorkerForBuildContainerArgsForCall[i].planID
}

func (fake *FakeWorkerProvider) FindWorkerForBuildContainerReturns(result1 worker.Worker, result2 bool, result3 error) {
	fake.FindWorkerForBuildContainerStub = nil
	fake.findWorkerForBuildContainerReturns = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) FindWorkerForBuildContainerReturnsOnCall(i int, result1 worker.Worker, result2 bool, result3 error) {
	fake.FindWorkerForBuildContainerStub = nil
	if fake.findWorkerForBuildContainerReturnsOnCall == nil {
		fake.findWorkerForBuildContainerReturnsOnCall = make(map[int]struct {
			result1 worker.Worker
			result2 bool
			result3 error
		})
	}
	fake.findWorkerForBuildContainerReturnsOnCall[i] = struct {
		result1 worker.Worker
		result2 bool
		result3 error
	}{result1, result2, result3}
}

func (fake *FakeWorkerProvider) Invocations() map[string][][]interface{} {
	fake.invocationsMutex.RLock()
	defer fake.invocationsMutex.RUnlock()
	fake.runningWorkersMutex.RLock()
	defer fake.runningWorkersMutex.RUnlock()
	fake.getWorkerMutex.RLock()
	defer fake.getWorkerMutex.RUnlock()
	fake.findWorkerForContainerMutex.RLock()
	defer fake.findWorkerForContainerMutex.RUnlock()
	fake.findWorkerForResourceCheckContainerMutex.RLock()
	defer fake.findWorkerForResourceCheckContainerMutex.RUnlock()
	fake.findWorkerForBuildContainerMutex.RLock()
	defer fake.findWorkerForBuildContainerMutex.RUnlock()
	return fake.invocations
}

func (fake *FakeWorkerProvider) recordInvocation(key string, args []interface{}) {
	fake.invocationsMutex.Lock()
	defer fake.invocationsMutex.Unlock()
	if fake.invocations == nil {
		fake.invocations = map[string][][]interface{}{}
	}
	if fake.invocations[key] == nil {
		fake.invocations[key] = [][]interface{}{}
	}
	fake.invocations[key] = append(fake.invocations[key], args)
}

var _ worker.WorkerProvider = new(FakeWorkerProvider)