github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/worker/workerfakes/fake_container_placement_strategy.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package workerfakes
     3  
     4  import (
     5  	"sync"
     6  
     7  	"code.cloudfoundry.org/lager"
     8  	"github.com/pf-qiu/concourse/v6/atc/worker"
     9  )
    10  
    11  type FakeContainerPlacementStrategy struct {
    12  	ChooseStub        func(lager.Logger, []worker.Worker, worker.ContainerSpec) (worker.Worker, error)
    13  	chooseMutex       sync.RWMutex
    14  	chooseArgsForCall []struct {
    15  		arg1 lager.Logger
    16  		arg2 []worker.Worker
    17  		arg3 worker.ContainerSpec
    18  	}
    19  	chooseReturns struct {
    20  		result1 worker.Worker
    21  		result2 error
    22  	}
    23  	chooseReturnsOnCall map[int]struct {
    24  		result1 worker.Worker
    25  		result2 error
    26  	}
    27  	ModifiesActiveTasksStub        func() bool
    28  	modifiesActiveTasksMutex       sync.RWMutex
    29  	modifiesActiveTasksArgsForCall []struct {
    30  	}
    31  	modifiesActiveTasksReturns struct {
    32  		result1 bool
    33  	}
    34  	modifiesActiveTasksReturnsOnCall map[int]struct {
    35  		result1 bool
    36  	}
    37  	invocations      map[string][][]interface{}
    38  	invocationsMutex sync.RWMutex
    39  }
    40  
    41  func (fake *FakeContainerPlacementStrategy) Choose(arg1 lager.Logger, arg2 []worker.Worker, arg3 worker.ContainerSpec) (worker.Worker, error) {
    42  	var arg2Copy []worker.Worker
    43  	if arg2 != nil {
    44  		arg2Copy = make([]worker.Worker, len(arg2))
    45  		copy(arg2Copy, arg2)
    46  	}
    47  	fake.chooseMutex.Lock()
    48  	ret, specificReturn := fake.chooseReturnsOnCall[len(fake.chooseArgsForCall)]
    49  	fake.chooseArgsForCall = append(fake.chooseArgsForCall, struct {
    50  		arg1 lager.Logger
    51  		arg2 []worker.Worker
    52  		arg3 worker.ContainerSpec
    53  	}{arg1, arg2Copy, arg3})
    54  	fake.recordInvocation("Choose", []interface{}{arg1, arg2Copy, arg3})
    55  	fake.chooseMutex.Unlock()
    56  	if fake.ChooseStub != nil {
    57  		return fake.ChooseStub(arg1, arg2, arg3)
    58  	}
    59  	if specificReturn {
    60  		return ret.result1, ret.result2
    61  	}
    62  	fakeReturns := fake.chooseReturns
    63  	return fakeReturns.result1, fakeReturns.result2
    64  }
    65  
    66  func (fake *FakeContainerPlacementStrategy) ChooseCallCount() int {
    67  	fake.chooseMutex.RLock()
    68  	defer fake.chooseMutex.RUnlock()
    69  	return len(fake.chooseArgsForCall)
    70  }
    71  
    72  func (fake *FakeContainerPlacementStrategy) ChooseCalls(stub func(lager.Logger, []worker.Worker, worker.ContainerSpec) (worker.Worker, error)) {
    73  	fake.chooseMutex.Lock()
    74  	defer fake.chooseMutex.Unlock()
    75  	fake.ChooseStub = stub
    76  }
    77  
    78  func (fake *FakeContainerPlacementStrategy) ChooseArgsForCall(i int) (lager.Logger, []worker.Worker, worker.ContainerSpec) {
    79  	fake.chooseMutex.RLock()
    80  	defer fake.chooseMutex.RUnlock()
    81  	argsForCall := fake.chooseArgsForCall[i]
    82  	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
    83  }
    84  
    85  func (fake *FakeContainerPlacementStrategy) ChooseReturns(result1 worker.Worker, result2 error) {
    86  	fake.chooseMutex.Lock()
    87  	defer fake.chooseMutex.Unlock()
    88  	fake.ChooseStub = nil
    89  	fake.chooseReturns = struct {
    90  		result1 worker.Worker
    91  		result2 error
    92  	}{result1, result2}
    93  }
    94  
    95  func (fake *FakeContainerPlacementStrategy) ChooseReturnsOnCall(i int, result1 worker.Worker, result2 error) {
    96  	fake.chooseMutex.Lock()
    97  	defer fake.chooseMutex.Unlock()
    98  	fake.ChooseStub = nil
    99  	if fake.chooseReturnsOnCall == nil {
   100  		fake.chooseReturnsOnCall = make(map[int]struct {
   101  			result1 worker.Worker
   102  			result2 error
   103  		})
   104  	}
   105  	fake.chooseReturnsOnCall[i] = struct {
   106  		result1 worker.Worker
   107  		result2 error
   108  	}{result1, result2}
   109  }
   110  
   111  func (fake *FakeContainerPlacementStrategy) ModifiesActiveTasks() bool {
   112  	fake.modifiesActiveTasksMutex.Lock()
   113  	ret, specificReturn := fake.modifiesActiveTasksReturnsOnCall[len(fake.modifiesActiveTasksArgsForCall)]
   114  	fake.modifiesActiveTasksArgsForCall = append(fake.modifiesActiveTasksArgsForCall, struct {
   115  	}{})
   116  	fake.recordInvocation("ModifiesActiveTasks", []interface{}{})
   117  	fake.modifiesActiveTasksMutex.Unlock()
   118  	if fake.ModifiesActiveTasksStub != nil {
   119  		return fake.ModifiesActiveTasksStub()
   120  	}
   121  	if specificReturn {
   122  		return ret.result1
   123  	}
   124  	fakeReturns := fake.modifiesActiveTasksReturns
   125  	return fakeReturns.result1
   126  }
   127  
   128  func (fake *FakeContainerPlacementStrategy) ModifiesActiveTasksCallCount() int {
   129  	fake.modifiesActiveTasksMutex.RLock()
   130  	defer fake.modifiesActiveTasksMutex.RUnlock()
   131  	return len(fake.modifiesActiveTasksArgsForCall)
   132  }
   133  
   134  func (fake *FakeContainerPlacementStrategy) ModifiesActiveTasksCalls(stub func() bool) {
   135  	fake.modifiesActiveTasksMutex.Lock()
   136  	defer fake.modifiesActiveTasksMutex.Unlock()
   137  	fake.ModifiesActiveTasksStub = stub
   138  }
   139  
   140  func (fake *FakeContainerPlacementStrategy) ModifiesActiveTasksReturns(result1 bool) {
   141  	fake.modifiesActiveTasksMutex.Lock()
   142  	defer fake.modifiesActiveTasksMutex.Unlock()
   143  	fake.ModifiesActiveTasksStub = nil
   144  	fake.modifiesActiveTasksReturns = struct {
   145  		result1 bool
   146  	}{result1}
   147  }
   148  
   149  func (fake *FakeContainerPlacementStrategy) ModifiesActiveTasksReturnsOnCall(i int, result1 bool) {
   150  	fake.modifiesActiveTasksMutex.Lock()
   151  	defer fake.modifiesActiveTasksMutex.Unlock()
   152  	fake.ModifiesActiveTasksStub = nil
   153  	if fake.modifiesActiveTasksReturnsOnCall == nil {
   154  		fake.modifiesActiveTasksReturnsOnCall = make(map[int]struct {
   155  			result1 bool
   156  		})
   157  	}
   158  	fake.modifiesActiveTasksReturnsOnCall[i] = struct {
   159  		result1 bool
   160  	}{result1}
   161  }
   162  
   163  func (fake *FakeContainerPlacementStrategy) Invocations() map[string][][]interface{} {
   164  	fake.invocationsMutex.RLock()
   165  	defer fake.invocationsMutex.RUnlock()
   166  	fake.chooseMutex.RLock()
   167  	defer fake.chooseMutex.RUnlock()
   168  	fake.modifiesActiveTasksMutex.RLock()
   169  	defer fake.modifiesActiveTasksMutex.RUnlock()
   170  	copiedInvocations := map[string][][]interface{}{}
   171  	for key, value := range fake.invocations {
   172  		copiedInvocations[key] = value
   173  	}
   174  	return copiedInvocations
   175  }
   176  
   177  func (fake *FakeContainerPlacementStrategy) recordInvocation(key string, args []interface{}) {
   178  	fake.invocationsMutex.Lock()
   179  	defer fake.invocationsMutex.Unlock()
   180  	if fake.invocations == nil {
   181  		fake.invocations = map[string][][]interface{}{}
   182  	}
   183  	if fake.invocations[key] == nil {
   184  		fake.invocations[key] = [][]interface{}{}
   185  	}
   186  	fake.invocations[key] = append(fake.invocations[key], args)
   187  }
   188  
   189  var _ worker.ContainerPlacementStrategy = new(FakeContainerPlacementStrategy)