github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/engine/enginefakes/fake_stepper_factory.go (about) 1 // Code generated by counterfeiter. DO NOT EDIT. 2 package enginefakes 3 4 import ( 5 "sync" 6 7 "github.com/pf-qiu/concourse/v6/atc/db" 8 "github.com/pf-qiu/concourse/v6/atc/engine" 9 "github.com/pf-qiu/concourse/v6/atc/exec" 10 ) 11 12 type FakeStepperFactory struct { 13 StepperForBuildStub func(db.Build) (exec.Stepper, error) 14 stepperForBuildMutex sync.RWMutex 15 stepperForBuildArgsForCall []struct { 16 arg1 db.Build 17 } 18 stepperForBuildReturns struct { 19 result1 exec.Stepper 20 result2 error 21 } 22 stepperForBuildReturnsOnCall map[int]struct { 23 result1 exec.Stepper 24 result2 error 25 } 26 invocations map[string][][]interface{} 27 invocationsMutex sync.RWMutex 28 } 29 30 func (fake *FakeStepperFactory) StepperForBuild(arg1 db.Build) (exec.Stepper, error) { 31 fake.stepperForBuildMutex.Lock() 32 ret, specificReturn := fake.stepperForBuildReturnsOnCall[len(fake.stepperForBuildArgsForCall)] 33 fake.stepperForBuildArgsForCall = append(fake.stepperForBuildArgsForCall, struct { 34 arg1 db.Build 35 }{arg1}) 36 fake.recordInvocation("StepperForBuild", []interface{}{arg1}) 37 fake.stepperForBuildMutex.Unlock() 38 if fake.StepperForBuildStub != nil { 39 return fake.StepperForBuildStub(arg1) 40 } 41 if specificReturn { 42 return ret.result1, ret.result2 43 } 44 fakeReturns := fake.stepperForBuildReturns 45 return fakeReturns.result1, fakeReturns.result2 46 } 47 48 func (fake *FakeStepperFactory) StepperForBuildCallCount() int { 49 fake.stepperForBuildMutex.RLock() 50 defer fake.stepperForBuildMutex.RUnlock() 51 return len(fake.stepperForBuildArgsForCall) 52 } 53 54 func (fake *FakeStepperFactory) StepperForBuildCalls(stub func(db.Build) (exec.Stepper, error)) { 55 fake.stepperForBuildMutex.Lock() 56 defer fake.stepperForBuildMutex.Unlock() 57 fake.StepperForBuildStub = stub 58 } 59 60 func (fake *FakeStepperFactory) StepperForBuildArgsForCall(i int) db.Build { 61 fake.stepperForBuildMutex.RLock() 62 defer fake.stepperForBuildMutex.RUnlock() 63 argsForCall := fake.stepperForBuildArgsForCall[i] 64 return argsForCall.arg1 65 } 66 67 func (fake *FakeStepperFactory) StepperForBuildReturns(result1 exec.Stepper, result2 error) { 68 fake.stepperForBuildMutex.Lock() 69 defer fake.stepperForBuildMutex.Unlock() 70 fake.StepperForBuildStub = nil 71 fake.stepperForBuildReturns = struct { 72 result1 exec.Stepper 73 result2 error 74 }{result1, result2} 75 } 76 77 func (fake *FakeStepperFactory) StepperForBuildReturnsOnCall(i int, result1 exec.Stepper, result2 error) { 78 fake.stepperForBuildMutex.Lock() 79 defer fake.stepperForBuildMutex.Unlock() 80 fake.StepperForBuildStub = nil 81 if fake.stepperForBuildReturnsOnCall == nil { 82 fake.stepperForBuildReturnsOnCall = make(map[int]struct { 83 result1 exec.Stepper 84 result2 error 85 }) 86 } 87 fake.stepperForBuildReturnsOnCall[i] = struct { 88 result1 exec.Stepper 89 result2 error 90 }{result1, result2} 91 } 92 93 func (fake *FakeStepperFactory) Invocations() map[string][][]interface{} { 94 fake.invocationsMutex.RLock() 95 defer fake.invocationsMutex.RUnlock() 96 fake.stepperForBuildMutex.RLock() 97 defer fake.stepperForBuildMutex.RUnlock() 98 copiedInvocations := map[string][][]interface{}{} 99 for key, value := range fake.invocations { 100 copiedInvocations[key] = value 101 } 102 return copiedInvocations 103 } 104 105 func (fake *FakeStepperFactory) recordInvocation(key string, args []interface{}) { 106 fake.invocationsMutex.Lock() 107 defer fake.invocationsMutex.Unlock() 108 if fake.invocations == nil { 109 fake.invocations = map[string][][]interface{}{} 110 } 111 if fake.invocations[key] == nil { 112 fake.invocations[key] = [][]interface{}{} 113 } 114 fake.invocations[key] = append(fake.invocations[key], args) 115 } 116 117 var _ engine.StepperFactory = new(FakeStepperFactory)