github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/db/dbfakes/fake_worker_lifecycle.go (about)

     1  // Code generated by counterfeiter. DO NOT EDIT.
     2  package dbfakes
     3  
     4  import (
     5  	"sync"
     6  
     7  	"github.com/pf-qiu/concourse/v6/atc/db"
     8  )
     9  
    10  type FakeWorkerLifecycle struct {
    11  	DeleteFinishedRetiringWorkersStub        func() ([]string, error)
    12  	deleteFinishedRetiringWorkersMutex       sync.RWMutex
    13  	deleteFinishedRetiringWorkersArgsForCall []struct {
    14  	}
    15  	deleteFinishedRetiringWorkersReturns struct {
    16  		result1 []string
    17  		result2 error
    18  	}
    19  	deleteFinishedRetiringWorkersReturnsOnCall map[int]struct {
    20  		result1 []string
    21  		result2 error
    22  	}
    23  	DeleteUnresponsiveEphemeralWorkersStub        func() ([]string, error)
    24  	deleteUnresponsiveEphemeralWorkersMutex       sync.RWMutex
    25  	deleteUnresponsiveEphemeralWorkersArgsForCall []struct {
    26  	}
    27  	deleteUnresponsiveEphemeralWorkersReturns struct {
    28  		result1 []string
    29  		result2 error
    30  	}
    31  	deleteUnresponsiveEphemeralWorkersReturnsOnCall map[int]struct {
    32  		result1 []string
    33  		result2 error
    34  	}
    35  	GetWorkerStateByNameStub        func() (map[string]db.WorkerState, error)
    36  	getWorkerStateByNameMutex       sync.RWMutex
    37  	getWorkerStateByNameArgsForCall []struct {
    38  	}
    39  	getWorkerStateByNameReturns struct {
    40  		result1 map[string]db.WorkerState
    41  		result2 error
    42  	}
    43  	getWorkerStateByNameReturnsOnCall map[int]struct {
    44  		result1 map[string]db.WorkerState
    45  		result2 error
    46  	}
    47  	LandFinishedLandingWorkersStub        func() ([]string, error)
    48  	landFinishedLandingWorkersMutex       sync.RWMutex
    49  	landFinishedLandingWorkersArgsForCall []struct {
    50  	}
    51  	landFinishedLandingWorkersReturns struct {
    52  		result1 []string
    53  		result2 error
    54  	}
    55  	landFinishedLandingWorkersReturnsOnCall map[int]struct {
    56  		result1 []string
    57  		result2 error
    58  	}
    59  	StallUnresponsiveWorkersStub        func() ([]string, error)
    60  	stallUnresponsiveWorkersMutex       sync.RWMutex
    61  	stallUnresponsiveWorkersArgsForCall []struct {
    62  	}
    63  	stallUnresponsiveWorkersReturns struct {
    64  		result1 []string
    65  		result2 error
    66  	}
    67  	stallUnresponsiveWorkersReturnsOnCall map[int]struct {
    68  		result1 []string
    69  		result2 error
    70  	}
    71  	invocations      map[string][][]interface{}
    72  	invocationsMutex sync.RWMutex
    73  }
    74  
    75  func (fake *FakeWorkerLifecycle) DeleteFinishedRetiringWorkers() ([]string, error) {
    76  	fake.deleteFinishedRetiringWorkersMutex.Lock()
    77  	ret, specificReturn := fake.deleteFinishedRetiringWorkersReturnsOnCall[len(fake.deleteFinishedRetiringWorkersArgsForCall)]
    78  	fake.deleteFinishedRetiringWorkersArgsForCall = append(fake.deleteFinishedRetiringWorkersArgsForCall, struct {
    79  	}{})
    80  	fake.recordInvocation("DeleteFinishedRetiringWorkers", []interface{}{})
    81  	fake.deleteFinishedRetiringWorkersMutex.Unlock()
    82  	if fake.DeleteFinishedRetiringWorkersStub != nil {
    83  		return fake.DeleteFinishedRetiringWorkersStub()
    84  	}
    85  	if specificReturn {
    86  		return ret.result1, ret.result2
    87  	}
    88  	fakeReturns := fake.deleteFinishedRetiringWorkersReturns
    89  	return fakeReturns.result1, fakeReturns.result2
    90  }
    91  
    92  func (fake *FakeWorkerLifecycle) DeleteFinishedRetiringWorkersCallCount() int {
    93  	fake.deleteFinishedRetiringWorkersMutex.RLock()
    94  	defer fake.deleteFinishedRetiringWorkersMutex.RUnlock()
    95  	return len(fake.deleteFinishedRetiringWorkersArgsForCall)
    96  }
    97  
    98  func (fake *FakeWorkerLifecycle) DeleteFinishedRetiringWorkersCalls(stub func() ([]string, error)) {
    99  	fake.deleteFinishedRetiringWorkersMutex.Lock()
   100  	defer fake.deleteFinishedRetiringWorkersMutex.Unlock()
   101  	fake.DeleteFinishedRetiringWorkersStub = stub
   102  }
   103  
   104  func (fake *FakeWorkerLifecycle) DeleteFinishedRetiringWorkersReturns(result1 []string, result2 error) {
   105  	fake.deleteFinishedRetiringWorkersMutex.Lock()
   106  	defer fake.deleteFinishedRetiringWorkersMutex.Unlock()
   107  	fake.DeleteFinishedRetiringWorkersStub = nil
   108  	fake.deleteFinishedRetiringWorkersReturns = struct {
   109  		result1 []string
   110  		result2 error
   111  	}{result1, result2}
   112  }
   113  
   114  func (fake *FakeWorkerLifecycle) DeleteFinishedRetiringWorkersReturnsOnCall(i int, result1 []string, result2 error) {
   115  	fake.deleteFinishedRetiringWorkersMutex.Lock()
   116  	defer fake.deleteFinishedRetiringWorkersMutex.Unlock()
   117  	fake.DeleteFinishedRetiringWorkersStub = nil
   118  	if fake.deleteFinishedRetiringWorkersReturnsOnCall == nil {
   119  		fake.deleteFinishedRetiringWorkersReturnsOnCall = make(map[int]struct {
   120  			result1 []string
   121  			result2 error
   122  		})
   123  	}
   124  	fake.deleteFinishedRetiringWorkersReturnsOnCall[i] = struct {
   125  		result1 []string
   126  		result2 error
   127  	}{result1, result2}
   128  }
   129  
   130  func (fake *FakeWorkerLifecycle) DeleteUnresponsiveEphemeralWorkers() ([]string, error) {
   131  	fake.deleteUnresponsiveEphemeralWorkersMutex.Lock()
   132  	ret, specificReturn := fake.deleteUnresponsiveEphemeralWorkersReturnsOnCall[len(fake.deleteUnresponsiveEphemeralWorkersArgsForCall)]
   133  	fake.deleteUnresponsiveEphemeralWorkersArgsForCall = append(fake.deleteUnresponsiveEphemeralWorkersArgsForCall, struct {
   134  	}{})
   135  	fake.recordInvocation("DeleteUnresponsiveEphemeralWorkers", []interface{}{})
   136  	fake.deleteUnresponsiveEphemeralWorkersMutex.Unlock()
   137  	if fake.DeleteUnresponsiveEphemeralWorkersStub != nil {
   138  		return fake.DeleteUnresponsiveEphemeralWorkersStub()
   139  	}
   140  	if specificReturn {
   141  		return ret.result1, ret.result2
   142  	}
   143  	fakeReturns := fake.deleteUnresponsiveEphemeralWorkersReturns
   144  	return fakeReturns.result1, fakeReturns.result2
   145  }
   146  
   147  func (fake *FakeWorkerLifecycle) DeleteUnresponsiveEphemeralWorkersCallCount() int {
   148  	fake.deleteUnresponsiveEphemeralWorkersMutex.RLock()
   149  	defer fake.deleteUnresponsiveEphemeralWorkersMutex.RUnlock()
   150  	return len(fake.deleteUnresponsiveEphemeralWorkersArgsForCall)
   151  }
   152  
   153  func (fake *FakeWorkerLifecycle) DeleteUnresponsiveEphemeralWorkersCalls(stub func() ([]string, error)) {
   154  	fake.deleteUnresponsiveEphemeralWorkersMutex.Lock()
   155  	defer fake.deleteUnresponsiveEphemeralWorkersMutex.Unlock()
   156  	fake.DeleteUnresponsiveEphemeralWorkersStub = stub
   157  }
   158  
   159  func (fake *FakeWorkerLifecycle) DeleteUnresponsiveEphemeralWorkersReturns(result1 []string, result2 error) {
   160  	fake.deleteUnresponsiveEphemeralWorkersMutex.Lock()
   161  	defer fake.deleteUnresponsiveEphemeralWorkersMutex.Unlock()
   162  	fake.DeleteUnresponsiveEphemeralWorkersStub = nil
   163  	fake.deleteUnresponsiveEphemeralWorkersReturns = struct {
   164  		result1 []string
   165  		result2 error
   166  	}{result1, result2}
   167  }
   168  
   169  func (fake *FakeWorkerLifecycle) DeleteUnresponsiveEphemeralWorkersReturnsOnCall(i int, result1 []string, result2 error) {
   170  	fake.deleteUnresponsiveEphemeralWorkersMutex.Lock()
   171  	defer fake.deleteUnresponsiveEphemeralWorkersMutex.Unlock()
   172  	fake.DeleteUnresponsiveEphemeralWorkersStub = nil
   173  	if fake.deleteUnresponsiveEphemeralWorkersReturnsOnCall == nil {
   174  		fake.deleteUnresponsiveEphemeralWorkersReturnsOnCall = make(map[int]struct {
   175  			result1 []string
   176  			result2 error
   177  		})
   178  	}
   179  	fake.deleteUnresponsiveEphemeralWorkersReturnsOnCall[i] = struct {
   180  		result1 []string
   181  		result2 error
   182  	}{result1, result2}
   183  }
   184  
   185  func (fake *FakeWorkerLifecycle) GetWorkerStateByName() (map[string]db.WorkerState, error) {
   186  	fake.getWorkerStateByNameMutex.Lock()
   187  	ret, specificReturn := fake.getWorkerStateByNameReturnsOnCall[len(fake.getWorkerStateByNameArgsForCall)]
   188  	fake.getWorkerStateByNameArgsForCall = append(fake.getWorkerStateByNameArgsForCall, struct {
   189  	}{})
   190  	fake.recordInvocation("GetWorkerStateByName", []interface{}{})
   191  	fake.getWorkerStateByNameMutex.Unlock()
   192  	if fake.GetWorkerStateByNameStub != nil {
   193  		return fake.GetWorkerStateByNameStub()
   194  	}
   195  	if specificReturn {
   196  		return ret.result1, ret.result2
   197  	}
   198  	fakeReturns := fake.getWorkerStateByNameReturns
   199  	return fakeReturns.result1, fakeReturns.result2
   200  }
   201  
   202  func (fake *FakeWorkerLifecycle) GetWorkerStateByNameCallCount() int {
   203  	fake.getWorkerStateByNameMutex.RLock()
   204  	defer fake.getWorkerStateByNameMutex.RUnlock()
   205  	return len(fake.getWorkerStateByNameArgsForCall)
   206  }
   207  
   208  func (fake *FakeWorkerLifecycle) GetWorkerStateByNameCalls(stub func() (map[string]db.WorkerState, error)) {
   209  	fake.getWorkerStateByNameMutex.Lock()
   210  	defer fake.getWorkerStateByNameMutex.Unlock()
   211  	fake.GetWorkerStateByNameStub = stub
   212  }
   213  
   214  func (fake *FakeWorkerLifecycle) GetWorkerStateByNameReturns(result1 map[string]db.WorkerState, result2 error) {
   215  	fake.getWorkerStateByNameMutex.Lock()
   216  	defer fake.getWorkerStateByNameMutex.Unlock()
   217  	fake.GetWorkerStateByNameStub = nil
   218  	fake.getWorkerStateByNameReturns = struct {
   219  		result1 map[string]db.WorkerState
   220  		result2 error
   221  	}{result1, result2}
   222  }
   223  
   224  func (fake *FakeWorkerLifecycle) GetWorkerStateByNameReturnsOnCall(i int, result1 map[string]db.WorkerState, result2 error) {
   225  	fake.getWorkerStateByNameMutex.Lock()
   226  	defer fake.getWorkerStateByNameMutex.Unlock()
   227  	fake.GetWorkerStateByNameStub = nil
   228  	if fake.getWorkerStateByNameReturnsOnCall == nil {
   229  		fake.getWorkerStateByNameReturnsOnCall = make(map[int]struct {
   230  			result1 map[string]db.WorkerState
   231  			result2 error
   232  		})
   233  	}
   234  	fake.getWorkerStateByNameReturnsOnCall[i] = struct {
   235  		result1 map[string]db.WorkerState
   236  		result2 error
   237  	}{result1, result2}
   238  }
   239  
   240  func (fake *FakeWorkerLifecycle) LandFinishedLandingWorkers() ([]string, error) {
   241  	fake.landFinishedLandingWorkersMutex.Lock()
   242  	ret, specificReturn := fake.landFinishedLandingWorkersReturnsOnCall[len(fake.landFinishedLandingWorkersArgsForCall)]
   243  	fake.landFinishedLandingWorkersArgsForCall = append(fake.landFinishedLandingWorkersArgsForCall, struct {
   244  	}{})
   245  	fake.recordInvocation("LandFinishedLandingWorkers", []interface{}{})
   246  	fake.landFinishedLandingWorkersMutex.Unlock()
   247  	if fake.LandFinishedLandingWorkersStub != nil {
   248  		return fake.LandFinishedLandingWorkersStub()
   249  	}
   250  	if specificReturn {
   251  		return ret.result1, ret.result2
   252  	}
   253  	fakeReturns := fake.landFinishedLandingWorkersReturns
   254  	return fakeReturns.result1, fakeReturns.result2
   255  }
   256  
   257  func (fake *FakeWorkerLifecycle) LandFinishedLandingWorkersCallCount() int {
   258  	fake.landFinishedLandingWorkersMutex.RLock()
   259  	defer fake.landFinishedLandingWorkersMutex.RUnlock()
   260  	return len(fake.landFinishedLandingWorkersArgsForCall)
   261  }
   262  
   263  func (fake *FakeWorkerLifecycle) LandFinishedLandingWorkersCalls(stub func() ([]string, error)) {
   264  	fake.landFinishedLandingWorkersMutex.Lock()
   265  	defer fake.landFinishedLandingWorkersMutex.Unlock()
   266  	fake.LandFinishedLandingWorkersStub = stub
   267  }
   268  
   269  func (fake *FakeWorkerLifecycle) LandFinishedLandingWorkersReturns(result1 []string, result2 error) {
   270  	fake.landFinishedLandingWorkersMutex.Lock()
   271  	defer fake.landFinishedLandingWorkersMutex.Unlock()
   272  	fake.LandFinishedLandingWorkersStub = nil
   273  	fake.landFinishedLandingWorkersReturns = struct {
   274  		result1 []string
   275  		result2 error
   276  	}{result1, result2}
   277  }
   278  
   279  func (fake *FakeWorkerLifecycle) LandFinishedLandingWorkersReturnsOnCall(i int, result1 []string, result2 error) {
   280  	fake.landFinishedLandingWorkersMutex.Lock()
   281  	defer fake.landFinishedLandingWorkersMutex.Unlock()
   282  	fake.LandFinishedLandingWorkersStub = nil
   283  	if fake.landFinishedLandingWorkersReturnsOnCall == nil {
   284  		fake.landFinishedLandingWorkersReturnsOnCall = make(map[int]struct {
   285  			result1 []string
   286  			result2 error
   287  		})
   288  	}
   289  	fake.landFinishedLandingWorkersReturnsOnCall[i] = struct {
   290  		result1 []string
   291  		result2 error
   292  	}{result1, result2}
   293  }
   294  
   295  func (fake *FakeWorkerLifecycle) StallUnresponsiveWorkers() ([]string, error) {
   296  	fake.stallUnresponsiveWorkersMutex.Lock()
   297  	ret, specificReturn := fake.stallUnresponsiveWorkersReturnsOnCall[len(fake.stallUnresponsiveWorkersArgsForCall)]
   298  	fake.stallUnresponsiveWorkersArgsForCall = append(fake.stallUnresponsiveWorkersArgsForCall, struct {
   299  	}{})
   300  	fake.recordInvocation("StallUnresponsiveWorkers", []interface{}{})
   301  	fake.stallUnresponsiveWorkersMutex.Unlock()
   302  	if fake.StallUnresponsiveWorkersStub != nil {
   303  		return fake.StallUnresponsiveWorkersStub()
   304  	}
   305  	if specificReturn {
   306  		return ret.result1, ret.result2
   307  	}
   308  	fakeReturns := fake.stallUnresponsiveWorkersReturns
   309  	return fakeReturns.result1, fakeReturns.result2
   310  }
   311  
   312  func (fake *FakeWorkerLifecycle) StallUnresponsiveWorkersCallCount() int {
   313  	fake.stallUnresponsiveWorkersMutex.RLock()
   314  	defer fake.stallUnresponsiveWorkersMutex.RUnlock()
   315  	return len(fake.stallUnresponsiveWorkersArgsForCall)
   316  }
   317  
   318  func (fake *FakeWorkerLifecycle) StallUnresponsiveWorkersCalls(stub func() ([]string, error)) {
   319  	fake.stallUnresponsiveWorkersMutex.Lock()
   320  	defer fake.stallUnresponsiveWorkersMutex.Unlock()
   321  	fake.StallUnresponsiveWorkersStub = stub
   322  }
   323  
   324  func (fake *FakeWorkerLifecycle) StallUnresponsiveWorkersReturns(result1 []string, result2 error) {
   325  	fake.stallUnresponsiveWorkersMutex.Lock()
   326  	defer fake.stallUnresponsiveWorkersMutex.Unlock()
   327  	fake.StallUnresponsiveWorkersStub = nil
   328  	fake.stallUnresponsiveWorkersReturns = struct {
   329  		result1 []string
   330  		result2 error
   331  	}{result1, result2}
   332  }
   333  
   334  func (fake *FakeWorkerLifecycle) StallUnresponsiveWorkersReturnsOnCall(i int, result1 []string, result2 error) {
   335  	fake.stallUnresponsiveWorkersMutex.Lock()
   336  	defer fake.stallUnresponsiveWorkersMutex.Unlock()
   337  	fake.StallUnresponsiveWorkersStub = nil
   338  	if fake.stallUnresponsiveWorkersReturnsOnCall == nil {
   339  		fake.stallUnresponsiveWorkersReturnsOnCall = make(map[int]struct {
   340  			result1 []string
   341  			result2 error
   342  		})
   343  	}
   344  	fake.stallUnresponsiveWorkersReturnsOnCall[i] = struct {
   345  		result1 []string
   346  		result2 error
   347  	}{result1, result2}
   348  }
   349  
   350  func (fake *FakeWorkerLifecycle) Invocations() map[string][][]interface{} {
   351  	fake.invocationsMutex.RLock()
   352  	defer fake.invocationsMutex.RUnlock()
   353  	fake.deleteFinishedRetiringWorkersMutex.RLock()
   354  	defer fake.deleteFinishedRetiringWorkersMutex.RUnlock()
   355  	fake.deleteUnresponsiveEphemeralWorkersMutex.RLock()
   356  	defer fake.deleteUnresponsiveEphemeralWorkersMutex.RUnlock()
   357  	fake.getWorkerStateByNameMutex.RLock()
   358  	defer fake.getWorkerStateByNameMutex.RUnlock()
   359  	fake.landFinishedLandingWorkersMutex.RLock()
   360  	defer fake.landFinishedLandingWorkersMutex.RUnlock()
   361  	fake.stallUnresponsiveWorkersMutex.RLock()
   362  	defer fake.stallUnresponsiveWorkersMutex.RUnlock()
   363  	copiedInvocations := map[string][][]interface{}{}
   364  	for key, value := range fake.invocations {
   365  		copiedInvocations[key] = value
   366  	}
   367  	return copiedInvocations
   368  }
   369  
   370  func (fake *FakeWorkerLifecycle) recordInvocation(key string, args []interface{}) {
   371  	fake.invocationsMutex.Lock()
   372  	defer fake.invocationsMutex.Unlock()
   373  	if fake.invocations == nil {
   374  		fake.invocations = map[string][][]interface{}{}
   375  	}
   376  	if fake.invocations[key] == nil {
   377  		fake.invocations[key] = [][]interface{}{}
   378  	}
   379  	fake.invocations[key] = append(fake.invocations[key], args)
   380  }
   381  
   382  var _ db.WorkerLifecycle = new(FakeWorkerLifecycle)