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)