github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/worker/workerfakes/fake_pool.go

// Code generated by counterfeiter. DO NOT EDIT.
package workerfakes

import (
	"context"
	"sync"

	"code.cloudfoundry.org/lager"
	"github.com/pf-qiu/concourse/v6/atc/db"
	"github.com/pf-qiu/concourse/v6/atc/worker"
)

type FakePool struct {
	ContainerInWorkerStub        func(lager.Logger, db.ContainerOwner, worker.WorkerSpec) (bool, error)
	containerInWorkerMutex       sync.RWMutex
	containerInWorkerArgsForCall []struct {
		arg1 lager.Logger
		arg2 db.ContainerOwner
		arg3 worker.WorkerSpec
	}
	containerInWorkerReturns struct {
		result1 bool
		result2 error
	}
	containerInWorkerReturnsOnCall map[int]struct {
		result1 bool
		result2 error
	}
	FindOrChooseWorkerStub        func(lager.Logger, worker.WorkerSpec) (worker.Worker, error)
	findOrChooseWorkerMutex       sync.RWMutex
	findOrChooseWorkerArgsForCall []struct {
		arg1 lager.Logger
		arg2 worker.WorkerSpec
	}
	findOrChooseWorkerReturns struct {
		result1 worker.Worker
		result2 error
	}
	findOrChooseWorkerReturnsOnCall map[int]struct {
		result1 worker.Worker
		result2 error
	}
	FindOrChooseWorkerForContainerStub        func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy) (worker.Worker, error)
	findOrChooseWorkerForContainerMutex       sync.RWMutex
	findOrChooseWorkerForContainerArgsForCall []struct {
		arg1 context.Context
		arg2 lager.Logger
		arg3 db.ContainerOwner
		arg4 worker.ContainerSpec
		arg5 worker.WorkerSpec
		arg6 worker.ContainerPlacementStrategy
	}
	findOrChooseWorkerForContainerReturns struct {
		result1 worker.Worker
		result2 error
	}
	findOrChooseWorkerForContainerReturnsOnCall map[int]struct {
		result1 worker.Worker
		result2 error
	}
	invocations      map[string][][]interface{}
	invocationsMutex sync.RWMutex
}

func (fake *FakePool) ContainerInWorker(arg1 lager.Logger, arg2 db.ContainerOwner, arg3 worker.WorkerSpec) (bool, error) {
	fake.containerInWorkerMutex.Lock()
	ret, specificReturn := fake.containerInWorkerReturnsOnCall[len(fake.containerInWorkerArgsForCall)]
	fake.containerInWorkerArgsForCall = append(fake.containerInWorkerArgsForCall, struct {
		arg1 lager.Logger
		arg2 db.ContainerOwner
		arg3 worker.WorkerSpec
	}{arg1, arg2, arg3})
	fake.recordInvocation("ContainerInWorker", []interface{}{arg1, arg2, arg3})
	fake.containerInWorkerMutex.Unlock()
	if fake.ContainerInWorkerStub != nil {
		return fake.ContainerInWorkerStub(arg1, arg2, arg3)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.containerInWorkerReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakePool) ContainerInWorkerCallCount() int {
	fake.containerInWorkerMutex.RLock()
	defer fake.containerInWorkerMutex.RUnlock()
	return len(fake.containerInWorkerArgsForCall)
}

func (fake *FakePool) ContainerInWorkerCalls(stub func(lager.Logger, db.ContainerOwner, worker.WorkerSpec) (bool, error)) {
	fake.containerInWorkerMutex.Lock()
	defer fake.containerInWorkerMutex.Unlock()
	fake.ContainerInWorkerStub = stub
}

func (fake *FakePool) ContainerInWorkerArgsForCall(i int) (lager.Logger, db.ContainerOwner, worker.WorkerSpec) {
	fake.containerInWorkerMutex.RLock()
	defer fake.containerInWorkerMutex.RUnlock()
	argsForCall := fake.containerInWorkerArgsForCall[i]
	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
}

func (fake *FakePool) ContainerInWorkerReturns(result1 bool, result2 error) {
	fake.containerInWorkerMutex.Lock()
	defer fake.containerInWorkerMutex.Unlock()
	fake.ContainerInWorkerStub = nil
	fake.containerInWorkerReturns = struct {
		result1 bool
		result2 error
	}{result1, result2}
}

func (fake *FakePool) ContainerInWorkerReturnsOnCall(i int, result1 bool, result2 error) {
	fake.containerInWorkerMutex.Lock()
	defer fake.containerInWorkerMutex.Unlock()
	fake.ContainerInWorkerStub = nil
	if fake.containerInWorkerReturnsOnCall == nil {
		fake.containerInWorkerReturnsOnCall = make(map[int]struct {
			result1 bool
			result2 error
		})
	}
	fake.containerInWorkerReturnsOnCall[i] = struct {
		result1 bool
		result2 error
	}{result1, result2}
}

func (fake *FakePool) FindOrChooseWorker(arg1 lager.Logger, arg2 worker.WorkerSpec) (worker.Worker, error) {
	fake.findOrChooseWorkerMutex.Lock()
	ret, specificReturn := fake.findOrChooseWorkerReturnsOnCall[len(fake.findOrChooseWorkerArgsForCall)]
	fake.findOrChooseWorkerArgsForCall = append(fake.findOrChooseWorkerArgsForCall, struct {
		arg1 lager.Logger
		arg2 worker.WorkerSpec
	}{arg1, arg2})
	fake.recordInvocation("FindOrChooseWorker", []interface{}{arg1, arg2})
	fake.findOrChooseWorkerMutex.Unlock()
	if fake.FindOrChooseWorkerStub != nil {
		return fake.FindOrChooseWorkerStub(arg1, arg2)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.findOrChooseWorkerReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakePool) FindOrChooseWorkerCallCount() int {
	fake.findOrChooseWorkerMutex.RLock()
	defer fake.findOrChooseWorkerMutex.RUnlock()
	return len(fake.findOrChooseWorkerArgsForCall)
}

func (fake *FakePool) FindOrChooseWorkerCalls(stub func(lager.Logger, worker.WorkerSpec) (worker.Worker, error)) {
	fake.findOrChooseWorkerMutex.Lock()
	defer fake.findOrChooseWorkerMutex.Unlock()
	fake.FindOrChooseWorkerStub = stub
}

func (fake *FakePool) FindOrChooseWorkerArgsForCall(i int) (lager.Logger, worker.WorkerSpec) {
	fake.findOrChooseWorkerMutex.RLock()
	defer fake.findOrChooseWorkerMutex.RUnlock()
	argsForCall := fake.findOrChooseWorkerArgsForCall[i]
	return argsForCall.arg1, argsForCall.arg2
}

func (fake *FakePool) FindOrChooseWorkerReturns(result1 worker.Worker, result2 error) {
	fake.findOrChooseWorkerMutex.Lock()
	defer fake.findOrChooseWorkerMutex.Unlock()
	fake.FindOrChooseWorkerStub = nil
	fake.findOrChooseWorkerReturns = struct {
		result1 worker.Worker
		result2 error
	}{result1, result2}
}

func (fake *FakePool) FindOrChooseWorkerReturnsOnCall(i int, result1 worker.Worker, result2 error) {
	fake.findOrChooseWorkerMutex.Lock()
	defer fake.findOrChooseWorkerMutex.Unlock()
	fake.FindOrChooseWorkerStub = nil
	if fake.findOrChooseWorkerReturnsOnCall == nil {
		fake.findOrChooseWorkerReturnsOnCall = make(map[int]struct {
			result1 worker.Worker
			result2 error
		})
	}
	fake.findOrChooseWorkerReturnsOnCall[i] = struct {
		result1 worker.Worker
		result2 error
	}{result1, result2}
}

func (fake *FakePool) FindOrChooseWorkerForContainer(arg1 context.Context, arg2 lager.Logger, arg3 db.ContainerOwner, arg4 worker.ContainerSpec, arg5 worker.WorkerSpec, arg6 worker.ContainerPlacementStrategy) (worker.Worker, error) {
	fake.findOrChooseWorkerForContainerMutex.Lock()
	ret, specificReturn := fake.findOrChooseWorkerForContainerReturnsOnCall[len(fake.findOrChooseWorkerForContainerArgsForCall)]
	fake.findOrChooseWorkerForContainerArgsForCall = append(fake.findOrChooseWorkerForContainerArgsForCall, struct {
		arg1 context.Context
		arg2 lager.Logger
		arg3 db.ContainerOwner
		arg4 worker.ContainerSpec
		arg5 worker.WorkerSpec
		arg6 worker.ContainerPlacementStrategy
	}{arg1, arg2, arg3, arg4, arg5, arg6})
	fake.recordInvocation("FindOrChooseWorkerForContainer", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6})
	fake.findOrChooseWorkerForContainerMutex.Unlock()
	if fake.FindOrChooseWorkerForContainerStub != nil {
		return fake.FindOrChooseWorkerForContainerStub(arg1, arg2, arg3, arg4, arg5, arg6)
	}
	if specificReturn {
		return ret.result1, ret.result2
	}
	fakeReturns := fake.findOrChooseWorkerForContainerReturns
	return fakeReturns.result1, fakeReturns.result2
}

func (fake *FakePool) FindOrChooseWorkerForContainerCallCount() int {
	fake.findOrChooseWorkerForContainerMutex.RLock()
	defer fake.findOrChooseWorkerForContainerMutex.RUnlock()
	return len(fake.findOrChooseWorkerForContainerArgsForCall)
}

func (fake *FakePool) FindOrChooseWorkerForContainerCalls(stub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy) (worker.Worker, error)) {
	fake.findOrChooseWorkerForContainerMutex.Lock()
	defer fake.findOrChooseWorkerForContainerMutex.Unlock()
	fake.FindOrChooseWorkerForContainerStub = stub
}

func (fake *FakePool) FindOrChooseWorkerForContainerArgsForCall(i int) (context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy) {
	fake.findOrChooseWorkerForContainerMutex.RLock()
	defer fake.findOrChooseWorkerForContainerMutex.RUnlock()
	argsForCall := fake.findOrChooseWorkerForContainerArgsForCall[i]
	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6
}

func (fake *FakePool) FindOrChooseWorkerForContainerReturns(result1 worker.Worker, result2 error) {
	fake.findOrChooseWorkerForContainerMutex.Lock()
	defer fake.findOrChooseWorkerForContainerMutex.Unlock()
	fake.FindOrChooseWorkerForContainerStub = nil
	fake.findOrChooseWorkerForContainerReturns = struct {
		result1 worker.Worker
		result2 error
	}{result1, result2}
}

func (fake *FakePool) FindOrChooseWorkerForContainerReturnsOnCall(i int, result1 worker.Worker, result2 error) {
	fake.findOrChooseWorkerForContainerMutex.Lock()
	defer fake.findOrChooseWorkerForContainerMutex.Unlock()
	fake.FindOrChooseWorkerForContainerStub = nil
	if fake.findOrChooseWorkerForContainerReturnsOnCall == nil {
		fake.findOrChooseWorkerForContainerReturnsOnCall = make(map[int]struct {
			result1 worker.Worker
			result2 error
		})
	}
	fake.findOrChooseWorkerForContainerReturnsOnCall[i] = struct {
		result1 worker.Worker
		result2 error
	}{result1, result2}
}

func (fake *FakePool) Invocations() map[string][][]interface{} {
	fake.invocationsMutex.RLock()
	defer fake.invocationsMutex.RUnlock()
	fake.containerInWorkerMutex.RLock()
	defer fake.containerInWorkerMutex.RUnlock()
	fake.findOrChooseWorkerMutex.RLock()
	defer fake.findOrChooseWorkerMutex.RUnlock()
	fake.findOrChooseWorkerForContainerMutex.RLock()
	defer fake.findOrChooseWorkerForContainerMutex.RUnlock()
	copiedInvocations := map[string][][]interface{}{}
	for key, value := range fake.invocations {
		copiedInvocations[key] = value
	}
	return copiedInvocations
}

func (fake *FakePool) 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.Pool = new(FakePool)