github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/worker/workerfakes/fake_client.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package workerfakes 3 4 import ( 5 "context" 6 "io" 7 "sync" 8 "time" 9 10 "code.cloudfoundry.org/lager" 11 "github.com/pf-qiu/concourse/v6/atc/db" 12 "github.com/pf-qiu/concourse/v6/atc/db/lock" 13 "github.com/pf-qiu/concourse/v6/atc/resource" 14 "github.com/pf-qiu/concourse/v6/atc/runtime" 15 "github.com/pf-qiu/concourse/v6/atc/worker" 16 ) 17 18 type FakeClient struct { 19 CreateVolumeStub func(lager.Logger, worker.VolumeSpec, worker.WorkerSpec, db.VolumeType) (worker.Volume, error) 20 createVolumeMutex sync.RWMutex 21 createVolumeArgsForCall []struct { 22 arg1 lager.Logger 23 arg2 worker.VolumeSpec 24 arg3 worker.WorkerSpec 25 arg4 db.VolumeType 26 } 27 createVolumeReturns struct { 28 result1 worker.Volume 29 result2 error 30 } 31 createVolumeReturnsOnCall map[int]struct { 32 result1 worker.Volume 33 result2 error 34 } 35 FindContainerStub func(lager.Logger, int, string) (worker.Container, bool, error) 36 findContainerMutex sync.RWMutex 37 findContainerArgsForCall []struct { 38 arg1 lager.Logger 39 arg2 int 40 arg3 string 41 } 42 findContainerReturns struct { 43 result1 worker.Container 44 result2 bool 45 result3 error 46 } 47 findContainerReturnsOnCall map[int]struct { 48 result1 worker.Container 49 result2 bool 50 result3 error 51 } 52 FindVolumeStub func(lager.Logger, int, string) (worker.Volume, bool, error) 53 findVolumeMutex sync.RWMutex 54 findVolumeArgsForCall []struct { 55 arg1 lager.Logger 56 arg2 int 57 arg3 string 58 } 59 findVolumeReturns struct { 60 result1 worker.Volume 61 result2 bool 62 result3 error 63 } 64 findVolumeReturnsOnCall map[int]struct { 65 result1 worker.Volume 66 result2 bool 67 result3 error 68 } 69 RunCheckStepStub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, resource.Resource, time.Duration) (worker.CheckResult, error) 70 runCheckStepMutex sync.RWMutex 71 runCheckStepArgsForCall []struct { 72 arg1 context.Context 73 arg2 lager.Logger 74 arg3 db.ContainerOwner 75 arg4 worker.ContainerSpec 76 arg5 worker.WorkerSpec 77 arg6 worker.ContainerPlacementStrategy 78 arg7 db.ContainerMetadata 79 arg8 runtime.ProcessSpec 80 arg9 runtime.StartingEventDelegate 81 arg10 resource.Resource 82 arg11 time.Duration 83 } 84 runCheckStepReturns struct { 85 result1 worker.CheckResult 86 result2 error 87 } 88 runCheckStepReturnsOnCall map[int]struct { 89 result1 worker.CheckResult 90 result2 error 91 } 92 RunGetStepStub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, db.UsedResourceCache, resource.Resource) (worker.GetResult, error) 93 runGetStepMutex sync.RWMutex 94 runGetStepArgsForCall []struct { 95 arg1 context.Context 96 arg2 lager.Logger 97 arg3 db.ContainerOwner 98 arg4 worker.ContainerSpec 99 arg5 worker.WorkerSpec 100 arg6 worker.ContainerPlacementStrategy 101 arg7 db.ContainerMetadata 102 arg8 runtime.ProcessSpec 103 arg9 runtime.StartingEventDelegate 104 arg10 db.UsedResourceCache 105 arg11 resource.Resource 106 } 107 runGetStepReturns struct { 108 result1 worker.GetResult 109 result2 error 110 } 111 runGetStepReturnsOnCall map[int]struct { 112 result1 worker.GetResult 113 result2 error 114 } 115 RunPutStepStub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, resource.Resource) (worker.PutResult, error) 116 runPutStepMutex sync.RWMutex 117 runPutStepArgsForCall []struct { 118 arg1 context.Context 119 arg2 lager.Logger 120 arg3 db.ContainerOwner 121 arg4 worker.ContainerSpec 122 arg5 worker.WorkerSpec 123 arg6 worker.ContainerPlacementStrategy 124 arg7 db.ContainerMetadata 125 arg8 runtime.ProcessSpec 126 arg9 runtime.StartingEventDelegate 127 arg10 resource.Resource 128 } 129 runPutStepReturns struct { 130 result1 worker.PutResult 131 result2 error 132 } 133 runPutStepReturnsOnCall map[int]struct { 134 result1 worker.PutResult 135 result2 error 136 } 137 RunTaskStepStub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, lock.LockFactory) (worker.TaskResult, error) 138 runTaskStepMutex sync.RWMutex 139 runTaskStepArgsForCall []struct { 140 arg1 context.Context 141 arg2 lager.Logger 142 arg3 db.ContainerOwner 143 arg4 worker.ContainerSpec 144 arg5 worker.WorkerSpec 145 arg6 worker.ContainerPlacementStrategy 146 arg7 db.ContainerMetadata 147 arg8 runtime.ProcessSpec 148 arg9 runtime.StartingEventDelegate 149 arg10 lock.LockFactory 150 } 151 runTaskStepReturns struct { 152 result1 worker.TaskResult 153 result2 error 154 } 155 runTaskStepReturnsOnCall map[int]struct { 156 result1 worker.TaskResult 157 result2 error 158 } 159 StreamFileFromArtifactStub func(context.Context, lager.Logger, runtime.Artifact, string) (io.ReadCloser, error) 160 streamFileFromArtifactMutex sync.RWMutex 161 streamFileFromArtifactArgsForCall []struct { 162 arg1 context.Context 163 arg2 lager.Logger 164 arg3 runtime.Artifact 165 arg4 string 166 } 167 streamFileFromArtifactReturns struct { 168 result1 io.ReadCloser 169 result2 error 170 } 171 streamFileFromArtifactReturnsOnCall map[int]struct { 172 result1 io.ReadCloser 173 result2 error 174 } 175 invocations map[string][][]interface{} 176 invocationsMutex sync.RWMutex 177 } 178 179 func (fake *FakeClient) CreateVolume(arg1 lager.Logger, arg2 worker.VolumeSpec, arg3 worker.WorkerSpec, arg4 db.VolumeType) (worker.Volume, error) { 180 fake.createVolumeMutex.Lock() 181 ret, specificReturn := fake.createVolumeReturnsOnCall[len(fake.createVolumeArgsForCall)] 182 fake.createVolumeArgsForCall = append(fake.createVolumeArgsForCall, struct { 183 arg1 lager.Logger 184 arg2 worker.VolumeSpec 185 arg3 worker.WorkerSpec 186 arg4 db.VolumeType 187 }{arg1, arg2, arg3, arg4}) 188 fake.recordInvocation("CreateVolume", []interface{}{arg1, arg2, arg3, arg4}) 189 fake.createVolumeMutex.Unlock() 190 if fake.CreateVolumeStub != nil { 191 return fake.CreateVolumeStub(arg1, arg2, arg3, arg4) 192 } 193 if specificReturn { 194 return ret.result1, ret.result2 195 } 196 fakeReturns := fake.createVolumeReturns 197 return fakeReturns.result1, fakeReturns.result2 198 } 199 200 func (fake *FakeClient) CreateVolumeCallCount() int { 201 fake.createVolumeMutex.RLock() 202 defer fake.createVolumeMutex.RUnlock() 203 return len(fake.createVolumeArgsForCall) 204 } 205 206 func (fake *FakeClient) CreateVolumeCalls(stub func(lager.Logger, worker.VolumeSpec, worker.WorkerSpec, db.VolumeType) (worker.Volume, error)) { 207 fake.createVolumeMutex.Lock() 208 defer fake.createVolumeMutex.Unlock() 209 fake.CreateVolumeStub = stub 210 } 211 212 func (fake *FakeClient) CreateVolumeArgsForCall(i int) (lager.Logger, worker.VolumeSpec, worker.WorkerSpec, db.VolumeType) { 213 fake.createVolumeMutex.RLock() 214 defer fake.createVolumeMutex.RUnlock() 215 argsForCall := fake.createVolumeArgsForCall[i] 216 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4 217 } 218 219 func (fake *FakeClient) CreateVolumeReturns(result1 worker.Volume, result2 error) { 220 fake.createVolumeMutex.Lock() 221 defer fake.createVolumeMutex.Unlock() 222 fake.CreateVolumeStub = nil 223 fake.createVolumeReturns = struct { 224 result1 worker.Volume 225 result2 error 226 }{result1, result2} 227 } 228 229 func (fake *FakeClient) CreateVolumeReturnsOnCall(i int, result1 worker.Volume, result2 error) { 230 fake.createVolumeMutex.Lock() 231 defer fake.createVolumeMutex.Unlock() 232 fake.CreateVolumeStub = nil 233 if fake.createVolumeReturnsOnCall == nil { 234 fake.createVolumeReturnsOnCall = make(map[int]struct { 235 result1 worker.Volume 236 result2 error 237 }) 238 } 239 fake.createVolumeReturnsOnCall[i] = struct { 240 result1 worker.Volume 241 result2 error 242 }{result1, result2} 243 } 244 245 func (fake *FakeClient) FindContainer(arg1 lager.Logger, arg2 int, arg3 string) (worker.Container, bool, error) { 246 fake.findContainerMutex.Lock() 247 ret, specificReturn := fake.findContainerReturnsOnCall[len(fake.findContainerArgsForCall)] 248 fake.findContainerArgsForCall = append(fake.findContainerArgsForCall, struct { 249 arg1 lager.Logger 250 arg2 int 251 arg3 string 252 }{arg1, arg2, arg3}) 253 fake.recordInvocation("FindContainer", []interface{}{arg1, arg2, arg3}) 254 fake.findContainerMutex.Unlock() 255 if fake.FindContainerStub != nil { 256 return fake.FindContainerStub(arg1, arg2, arg3) 257 } 258 if specificReturn { 259 return ret.result1, ret.result2, ret.result3 260 } 261 fakeReturns := fake.findContainerReturns 262 return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3 263 } 264 265 func (fake *FakeClient) FindContainerCallCount() int { 266 fake.findContainerMutex.RLock() 267 defer fake.findContainerMutex.RUnlock() 268 return len(fake.findContainerArgsForCall) 269 } 270 271 func (fake *FakeClient) FindContainerCalls(stub func(lager.Logger, int, string) (worker.Container, bool, error)) { 272 fake.findContainerMutex.Lock() 273 defer fake.findContainerMutex.Unlock() 274 fake.FindContainerStub = stub 275 } 276 277 func (fake *FakeClient) FindContainerArgsForCall(i int) (lager.Logger, int, string) { 278 fake.findContainerMutex.RLock() 279 defer fake.findContainerMutex.RUnlock() 280 argsForCall := fake.findContainerArgsForCall[i] 281 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 282 } 283 284 func (fake *FakeClient) FindContainerReturns(result1 worker.Container, result2 bool, result3 error) { 285 fake.findContainerMutex.Lock() 286 defer fake.findContainerMutex.Unlock() 287 fake.FindContainerStub = nil 288 fake.findContainerReturns = struct { 289 result1 worker.Container 290 result2 bool 291 result3 error 292 }{result1, result2, result3} 293 } 294 295 func (fake *FakeClient) FindContainerReturnsOnCall(i int, result1 worker.Container, result2 bool, result3 error) { 296 fake.findContainerMutex.Lock() 297 defer fake.findContainerMutex.Unlock() 298 fake.FindContainerStub = nil 299 if fake.findContainerReturnsOnCall == nil { 300 fake.findContainerReturnsOnCall = make(map[int]struct { 301 result1 worker.Container 302 result2 bool 303 result3 error 304 }) 305 } 306 fake.findContainerReturnsOnCall[i] = struct { 307 result1 worker.Container 308 result2 bool 309 result3 error 310 }{result1, result2, result3} 311 } 312 313 func (fake *FakeClient) FindVolume(arg1 lager.Logger, arg2 int, arg3 string) (worker.Volume, bool, error) { 314 fake.findVolumeMutex.Lock() 315 ret, specificReturn := fake.findVolumeReturnsOnCall[len(fake.findVolumeArgsForCall)] 316 fake.findVolumeArgsForCall = append(fake.findVolumeArgsForCall, struct { 317 arg1 lager.Logger 318 arg2 int 319 arg3 string 320 }{arg1, arg2, arg3}) 321 fake.recordInvocation("FindVolume", []interface{}{arg1, arg2, arg3}) 322 fake.findVolumeMutex.Unlock() 323 if fake.FindVolumeStub != nil { 324 return fake.FindVolumeStub(arg1, arg2, arg3) 325 } 326 if specificReturn { 327 return ret.result1, ret.result2, ret.result3 328 } 329 fakeReturns := fake.findVolumeReturns 330 return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3 331 } 332 333 func (fake *FakeClient) FindVolumeCallCount() int { 334 fake.findVolumeMutex.RLock() 335 defer fake.findVolumeMutex.RUnlock() 336 return len(fake.findVolumeArgsForCall) 337 } 338 339 func (fake *FakeClient) FindVolumeCalls(stub func(lager.Logger, int, string) (worker.Volume, bool, error)) { 340 fake.findVolumeMutex.Lock() 341 defer fake.findVolumeMutex.Unlock() 342 fake.FindVolumeStub = stub 343 } 344 345 func (fake *FakeClient) FindVolumeArgsForCall(i int) (lager.Logger, int, string) { 346 fake.findVolumeMutex.RLock() 347 defer fake.findVolumeMutex.RUnlock() 348 argsForCall := fake.findVolumeArgsForCall[i] 349 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 350 } 351 352 func (fake *FakeClient) FindVolumeReturns(result1 worker.Volume, result2 bool, result3 error) { 353 fake.findVolumeMutex.Lock() 354 defer fake.findVolumeMutex.Unlock() 355 fake.FindVolumeStub = nil 356 fake.findVolumeReturns = struct { 357 result1 worker.Volume 358 result2 bool 359 result3 error 360 }{result1, result2, result3} 361 } 362 363 func (fake *FakeClient) FindVolumeReturnsOnCall(i int, result1 worker.Volume, result2 bool, result3 error) { 364 fake.findVolumeMutex.Lock() 365 defer fake.findVolumeMutex.Unlock() 366 fake.FindVolumeStub = nil 367 if fake.findVolumeReturnsOnCall == nil { 368 fake.findVolumeReturnsOnCall = make(map[int]struct { 369 result1 worker.Volume 370 result2 bool 371 result3 error 372 }) 373 } 374 fake.findVolumeReturnsOnCall[i] = struct { 375 result1 worker.Volume 376 result2 bool 377 result3 error 378 }{result1, result2, result3} 379 } 380 381 func (fake *FakeClient) RunCheckStep(arg1 context.Context, arg2 lager.Logger, arg3 db.ContainerOwner, arg4 worker.ContainerSpec, arg5 worker.WorkerSpec, arg6 worker.ContainerPlacementStrategy, arg7 db.ContainerMetadata, arg8 runtime.ProcessSpec, arg9 runtime.StartingEventDelegate, arg10 resource.Resource, arg11 time.Duration) (worker.CheckResult, error) { 382 fake.runCheckStepMutex.Lock() 383 ret, specificReturn := fake.runCheckStepReturnsOnCall[len(fake.runCheckStepArgsForCall)] 384 fake.runCheckStepArgsForCall = append(fake.runCheckStepArgsForCall, struct { 385 arg1 context.Context 386 arg2 lager.Logger 387 arg3 db.ContainerOwner 388 arg4 worker.ContainerSpec 389 arg5 worker.WorkerSpec 390 arg6 worker.ContainerPlacementStrategy 391 arg7 db.ContainerMetadata 392 arg8 runtime.ProcessSpec 393 arg9 runtime.StartingEventDelegate 394 arg10 resource.Resource 395 arg11 time.Duration 396 }{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10, arg11}) 397 fake.recordInvocation("RunCheckStep", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10, arg11}) 398 fake.runCheckStepMutex.Unlock() 399 if fake.RunCheckStepStub != nil { 400 return fake.RunCheckStepStub(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10, arg11) 401 } 402 if specificReturn { 403 return ret.result1, ret.result2 404 } 405 fakeReturns := fake.runCheckStepReturns 406 return fakeReturns.result1, fakeReturns.result2 407 } 408 409 func (fake *FakeClient) RunCheckStepCallCount() int { 410 fake.runCheckStepMutex.RLock() 411 defer fake.runCheckStepMutex.RUnlock() 412 return len(fake.runCheckStepArgsForCall) 413 } 414 415 func (fake *FakeClient) RunCheckStepCalls(stub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, resource.Resource, time.Duration) (worker.CheckResult, error)) { 416 fake.runCheckStepMutex.Lock() 417 defer fake.runCheckStepMutex.Unlock() 418 fake.RunCheckStepStub = stub 419 } 420 421 func (fake *FakeClient) RunCheckStepArgsForCall(i int) (context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, resource.Resource, time.Duration) { 422 fake.runCheckStepMutex.RLock() 423 defer fake.runCheckStepMutex.RUnlock() 424 argsForCall := fake.runCheckStepArgsForCall[i] 425 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6, argsForCall.arg7, argsForCall.arg8, argsForCall.arg9, argsForCall.arg10, argsForCall.arg11 426 } 427 428 func (fake *FakeClient) RunCheckStepReturns(result1 worker.CheckResult, result2 error) { 429 fake.runCheckStepMutex.Lock() 430 defer fake.runCheckStepMutex.Unlock() 431 fake.RunCheckStepStub = nil 432 fake.runCheckStepReturns = struct { 433 result1 worker.CheckResult 434 result2 error 435 }{result1, result2} 436 } 437 438 func (fake *FakeClient) RunCheckStepReturnsOnCall(i int, result1 worker.CheckResult, result2 error) { 439 fake.runCheckStepMutex.Lock() 440 defer fake.runCheckStepMutex.Unlock() 441 fake.RunCheckStepStub = nil 442 if fake.runCheckStepReturnsOnCall == nil { 443 fake.runCheckStepReturnsOnCall = make(map[int]struct { 444 result1 worker.CheckResult 445 result2 error 446 }) 447 } 448 fake.runCheckStepReturnsOnCall[i] = struct { 449 result1 worker.CheckResult 450 result2 error 451 }{result1, result2} 452 } 453 454 func (fake *FakeClient) RunGetStep(arg1 context.Context, arg2 lager.Logger, arg3 db.ContainerOwner, arg4 worker.ContainerSpec, arg5 worker.WorkerSpec, arg6 worker.ContainerPlacementStrategy, arg7 db.ContainerMetadata, arg8 runtime.ProcessSpec, arg9 runtime.StartingEventDelegate, arg10 db.UsedResourceCache, arg11 resource.Resource) (worker.GetResult, error) { 455 fake.runGetStepMutex.Lock() 456 ret, specificReturn := fake.runGetStepReturnsOnCall[len(fake.runGetStepArgsForCall)] 457 fake.runGetStepArgsForCall = append(fake.runGetStepArgsForCall, struct { 458 arg1 context.Context 459 arg2 lager.Logger 460 arg3 db.ContainerOwner 461 arg4 worker.ContainerSpec 462 arg5 worker.WorkerSpec 463 arg6 worker.ContainerPlacementStrategy 464 arg7 db.ContainerMetadata 465 arg8 runtime.ProcessSpec 466 arg9 runtime.StartingEventDelegate 467 arg10 db.UsedResourceCache 468 arg11 resource.Resource 469 }{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10, arg11}) 470 fake.recordInvocation("RunGetStep", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10, arg11}) 471 fake.runGetStepMutex.Unlock() 472 if fake.RunGetStepStub != nil { 473 return fake.RunGetStepStub(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10, arg11) 474 } 475 if specificReturn { 476 return ret.result1, ret.result2 477 } 478 fakeReturns := fake.runGetStepReturns 479 return fakeReturns.result1, fakeReturns.result2 480 } 481 482 func (fake *FakeClient) RunGetStepCallCount() int { 483 fake.runGetStepMutex.RLock() 484 defer fake.runGetStepMutex.RUnlock() 485 return len(fake.runGetStepArgsForCall) 486 } 487 488 func (fake *FakeClient) RunGetStepCalls(stub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, db.UsedResourceCache, resource.Resource) (worker.GetResult, error)) { 489 fake.runGetStepMutex.Lock() 490 defer fake.runGetStepMutex.Unlock() 491 fake.RunGetStepStub = stub 492 } 493 494 func (fake *FakeClient) RunGetStepArgsForCall(i int) (context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, db.UsedResourceCache, resource.Resource) { 495 fake.runGetStepMutex.RLock() 496 defer fake.runGetStepMutex.RUnlock() 497 argsForCall := fake.runGetStepArgsForCall[i] 498 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6, argsForCall.arg7, argsForCall.arg8, argsForCall.arg9, argsForCall.arg10, argsForCall.arg11 499 } 500 501 func (fake *FakeClient) RunGetStepReturns(result1 worker.GetResult, result2 error) { 502 fake.runGetStepMutex.Lock() 503 defer fake.runGetStepMutex.Unlock() 504 fake.RunGetStepStub = nil 505 fake.runGetStepReturns = struct { 506 result1 worker.GetResult 507 result2 error 508 }{result1, result2} 509 } 510 511 func (fake *FakeClient) RunGetStepReturnsOnCall(i int, result1 worker.GetResult, result2 error) { 512 fake.runGetStepMutex.Lock() 513 defer fake.runGetStepMutex.Unlock() 514 fake.RunGetStepStub = nil 515 if fake.runGetStepReturnsOnCall == nil { 516 fake.runGetStepReturnsOnCall = make(map[int]struct { 517 result1 worker.GetResult 518 result2 error 519 }) 520 } 521 fake.runGetStepReturnsOnCall[i] = struct { 522 result1 worker.GetResult 523 result2 error 524 }{result1, result2} 525 } 526 527 func (fake *FakeClient) RunPutStep(arg1 context.Context, arg2 lager.Logger, arg3 db.ContainerOwner, arg4 worker.ContainerSpec, arg5 worker.WorkerSpec, arg6 worker.ContainerPlacementStrategy, arg7 db.ContainerMetadata, arg8 runtime.ProcessSpec, arg9 runtime.StartingEventDelegate, arg10 resource.Resource) (worker.PutResult, error) { 528 fake.runPutStepMutex.Lock() 529 ret, specificReturn := fake.runPutStepReturnsOnCall[len(fake.runPutStepArgsForCall)] 530 fake.runPutStepArgsForCall = append(fake.runPutStepArgsForCall, struct { 531 arg1 context.Context 532 arg2 lager.Logger 533 arg3 db.ContainerOwner 534 arg4 worker.ContainerSpec 535 arg5 worker.WorkerSpec 536 arg6 worker.ContainerPlacementStrategy 537 arg7 db.ContainerMetadata 538 arg8 runtime.ProcessSpec 539 arg9 runtime.StartingEventDelegate 540 arg10 resource.Resource 541 }{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10}) 542 fake.recordInvocation("RunPutStep", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10}) 543 fake.runPutStepMutex.Unlock() 544 if fake.RunPutStepStub != nil { 545 return fake.RunPutStepStub(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10) 546 } 547 if specificReturn { 548 return ret.result1, ret.result2 549 } 550 fakeReturns := fake.runPutStepReturns 551 return fakeReturns.result1, fakeReturns.result2 552 } 553 554 func (fake *FakeClient) RunPutStepCallCount() int { 555 fake.runPutStepMutex.RLock() 556 defer fake.runPutStepMutex.RUnlock() 557 return len(fake.runPutStepArgsForCall) 558 } 559 560 func (fake *FakeClient) RunPutStepCalls(stub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, resource.Resource) (worker.PutResult, error)) { 561 fake.runPutStepMutex.Lock() 562 defer fake.runPutStepMutex.Unlock() 563 fake.RunPutStepStub = stub 564 } 565 566 func (fake *FakeClient) RunPutStepArgsForCall(i int) (context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, resource.Resource) { 567 fake.runPutStepMutex.RLock() 568 defer fake.runPutStepMutex.RUnlock() 569 argsForCall := fake.runPutStepArgsForCall[i] 570 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6, argsForCall.arg7, argsForCall.arg8, argsForCall.arg9, argsForCall.arg10 571 } 572 573 func (fake *FakeClient) RunPutStepReturns(result1 worker.PutResult, result2 error) { 574 fake.runPutStepMutex.Lock() 575 defer fake.runPutStepMutex.Unlock() 576 fake.RunPutStepStub = nil 577 fake.runPutStepReturns = struct { 578 result1 worker.PutResult 579 result2 error 580 }{result1, result2} 581 } 582 583 func (fake *FakeClient) RunPutStepReturnsOnCall(i int, result1 worker.PutResult, result2 error) { 584 fake.runPutStepMutex.Lock() 585 defer fake.runPutStepMutex.Unlock() 586 fake.RunPutStepStub = nil 587 if fake.runPutStepReturnsOnCall == nil { 588 fake.runPutStepReturnsOnCall = make(map[int]struct { 589 result1 worker.PutResult 590 result2 error 591 }) 592 } 593 fake.runPutStepReturnsOnCall[i] = struct { 594 result1 worker.PutResult 595 result2 error 596 }{result1, result2} 597 } 598 599 func (fake *FakeClient) RunTaskStep(arg1 context.Context, arg2 lager.Logger, arg3 db.ContainerOwner, arg4 worker.ContainerSpec, arg5 worker.WorkerSpec, arg6 worker.ContainerPlacementStrategy, arg7 db.ContainerMetadata, arg8 runtime.ProcessSpec, arg9 runtime.StartingEventDelegate, arg10 lock.LockFactory) (worker.TaskResult, error) { 600 fake.runTaskStepMutex.Lock() 601 ret, specificReturn := fake.runTaskStepReturnsOnCall[len(fake.runTaskStepArgsForCall)] 602 fake.runTaskStepArgsForCall = append(fake.runTaskStepArgsForCall, struct { 603 arg1 context.Context 604 arg2 lager.Logger 605 arg3 db.ContainerOwner 606 arg4 worker.ContainerSpec 607 arg5 worker.WorkerSpec 608 arg6 worker.ContainerPlacementStrategy 609 arg7 db.ContainerMetadata 610 arg8 runtime.ProcessSpec 611 arg9 runtime.StartingEventDelegate 612 arg10 lock.LockFactory 613 }{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10}) 614 fake.recordInvocation("RunTaskStep", []interface{}{arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10}) 615 fake.runTaskStepMutex.Unlock() 616 if fake.RunTaskStepStub != nil { 617 return fake.RunTaskStepStub(arg1, arg2, arg3, arg4, arg5, arg6, arg7, arg8, arg9, arg10) 618 } 619 if specificReturn { 620 return ret.result1, ret.result2 621 } 622 fakeReturns := fake.runTaskStepReturns 623 return fakeReturns.result1, fakeReturns.result2 624 } 625 626 func (fake *FakeClient) RunTaskStepCallCount() int { 627 fake.runTaskStepMutex.RLock() 628 defer fake.runTaskStepMutex.RUnlock() 629 return len(fake.runTaskStepArgsForCall) 630 } 631 632 func (fake *FakeClient) RunTaskStepCalls(stub func(context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, lock.LockFactory) (worker.TaskResult, error)) { 633 fake.runTaskStepMutex.Lock() 634 defer fake.runTaskStepMutex.Unlock() 635 fake.RunTaskStepStub = stub 636 } 637 638 func (fake *FakeClient) RunTaskStepArgsForCall(i int) (context.Context, lager.Logger, db.ContainerOwner, worker.ContainerSpec, worker.WorkerSpec, worker.ContainerPlacementStrategy, db.ContainerMetadata, runtime.ProcessSpec, runtime.StartingEventDelegate, lock.LockFactory) { 639 fake.runTaskStepMutex.RLock() 640 defer fake.runTaskStepMutex.RUnlock() 641 argsForCall := fake.runTaskStepArgsForCall[i] 642 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4, argsForCall.arg5, argsForCall.arg6, argsForCall.arg7, argsForCall.arg8, argsForCall.arg9, argsForCall.arg10 643 } 644 645 func (fake *FakeClient) RunTaskStepReturns(result1 worker.TaskResult, result2 error) { 646 fake.runTaskStepMutex.Lock() 647 defer fake.runTaskStepMutex.Unlock() 648 fake.RunTaskStepStub = nil 649 fake.runTaskStepReturns = struct { 650 result1 worker.TaskResult 651 result2 error 652 }{result1, result2} 653 } 654 655 func (fake *FakeClient) RunTaskStepReturnsOnCall(i int, result1 worker.TaskResult, result2 error) { 656 fake.runTaskStepMutex.Lock() 657 defer fake.runTaskStepMutex.Unlock() 658 fake.RunTaskStepStub = nil 659 if fake.runTaskStepReturnsOnCall == nil { 660 fake.runTaskStepReturnsOnCall = make(map[int]struct { 661 result1 worker.TaskResult 662 result2 error 663 }) 664 } 665 fake.runTaskStepReturnsOnCall[i] = struct { 666 result1 worker.TaskResult 667 result2 error 668 }{result1, result2} 669 } 670 671 func (fake *FakeClient) StreamFileFromArtifact(arg1 context.Context, arg2 lager.Logger, arg3 runtime.Artifact, arg4 string) (io.ReadCloser, error) { 672 fake.streamFileFromArtifactMutex.Lock() 673 ret, specificReturn := fake.streamFileFromArtifactReturnsOnCall[len(fake.streamFileFromArtifactArgsForCall)] 674 fake.streamFileFromArtifactArgsForCall = append(fake.streamFileFromArtifactArgsForCall, struct { 675 arg1 context.Context 676 arg2 lager.Logger 677 arg3 runtime.Artifact 678 arg4 string 679 }{arg1, arg2, arg3, arg4}) 680 fake.recordInvocation("StreamFileFromArtifact", []interface{}{arg1, arg2, arg3, arg4}) 681 fake.streamFileFromArtifactMutex.Unlock() 682 if fake.StreamFileFromArtifactStub != nil { 683 return fake.StreamFileFromArtifactStub(arg1, arg2, arg3, arg4) 684 } 685 if specificReturn { 686 return ret.result1, ret.result2 687 } 688 fakeReturns := fake.streamFileFromArtifactReturns 689 return fakeReturns.result1, fakeReturns.result2 690 } 691 692 func (fake *FakeClient) StreamFileFromArtifactCallCount() int { 693 fake.streamFileFromArtifactMutex.RLock() 694 defer fake.streamFileFromArtifactMutex.RUnlock() 695 return len(fake.streamFileFromArtifactArgsForCall) 696 } 697 698 func (fake *FakeClient) StreamFileFromArtifactCalls(stub func(context.Context, lager.Logger, runtime.Artifact, string) (io.ReadCloser, error)) { 699 fake.streamFileFromArtifactMutex.Lock() 700 defer fake.streamFileFromArtifactMutex.Unlock() 701 fake.StreamFileFromArtifactStub = stub 702 } 703 704 func (fake *FakeClient) StreamFileFromArtifactArgsForCall(i int) (context.Context, lager.Logger, runtime.Artifact, string) { 705 fake.streamFileFromArtifactMutex.RLock() 706 defer fake.streamFileFromArtifactMutex.RUnlock() 707 argsForCall := fake.streamFileFromArtifactArgsForCall[i] 708 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4 709 } 710 711 func (fake *FakeClient) StreamFileFromArtifactReturns(result1 io.ReadCloser, result2 error) { 712 fake.streamFileFromArtifactMutex.Lock() 713 defer fake.streamFileFromArtifactMutex.Unlock() 714 fake.StreamFileFromArtifactStub = nil 715 fake.streamFileFromArtifactReturns = struct { 716 result1 io.ReadCloser 717 result2 error 718 }{result1, result2} 719 } 720 721 func (fake *FakeClient) StreamFileFromArtifactReturnsOnCall(i int, result1 io.ReadCloser, result2 error) { 722 fake.streamFileFromArtifactMutex.Lock() 723 defer fake.streamFileFromArtifactMutex.Unlock() 724 fake.StreamFileFromArtifactStub = nil 725 if fake.streamFileFromArtifactReturnsOnCall == nil { 726 fake.streamFileFromArtifactReturnsOnCall = make(map[int]struct { 727 result1 io.ReadCloser 728 result2 error 729 }) 730 } 731 fake.streamFileFromArtifactReturnsOnCall[i] = struct { 732 result1 io.ReadCloser 733 result2 error 734 }{result1, result2} 735 } 736 737 func (fake *FakeClient) Invocations() map[string][][]interface{} { 738 fake.invocationsMutex.RLock() 739 defer fake.invocationsMutex.RUnlock() 740 fake.createVolumeMutex.RLock() 741 defer fake.createVolumeMutex.RUnlock() 742 fake.findContainerMutex.RLock() 743 defer fake.findContainerMutex.RUnlock() 744 fake.findVolumeMutex.RLock() 745 defer fake.findVolumeMutex.RUnlock() 746 fake.runCheckStepMutex.RLock() 747 defer fake.runCheckStepMutex.RUnlock() 748 fake.runGetStepMutex.RLock() 749 defer fake.runGetStepMutex.RUnlock() 750 fake.runPutStepMutex.RLock() 751 defer fake.runPutStepMutex.RUnlock() 752 fake.runTaskStepMutex.RLock() 753 defer fake.runTaskStepMutex.RUnlock() 754 fake.streamFileFromArtifactMutex.RLock() 755 defer fake.streamFileFromArtifactMutex.RUnlock() 756 copiedInvocations := map[string][][]interface{}{} 757 for key, value := range fake.invocations { 758 copiedInvocations[key] = value 759 } 760 return copiedInvocations 761 } 762 763 func (fake *FakeClient) recordInvocation(key string, args []interface{}) { 764 fake.invocationsMutex.Lock() 765 defer fake.invocationsMutex.Unlock() 766 if fake.invocations == nil { 767 fake.invocations = map[string][][]interface{}{} 768 } 769 if fake.invocations[key] == nil { 770 fake.invocations[key] = [][]interface{}{} 771 } 772 fake.invocations[key] = append(fake.invocations[key], args) 773 } 774 775 var _ worker.Client = new(FakeClient)