1// Code generated by counterfeiter. DO NOT EDIT. 2package workerfakes 3 4import ( 5 "context" 6 "sync" 7 8 "github.com/concourse/concourse/atc/worker" 9) 10 11type FakeFetchSource struct { 12 CreateStub func(context.Context) (worker.GetResult, worker.Volume, error) 13 createMutex sync.RWMutex 14 createArgsForCall []struct { 15 arg1 context.Context 16 } 17 createReturns struct { 18 result1 worker.GetResult 19 result2 worker.Volume 20 result3 error 21 } 22 createReturnsOnCall map[int]struct { 23 result1 worker.GetResult 24 result2 worker.Volume 25 result3 error 26 } 27 FindStub func() (worker.GetResult, worker.Volume, bool, error) 28 findMutex sync.RWMutex 29 findArgsForCall []struct { 30 } 31 findReturns struct { 32 result1 worker.GetResult 33 result2 worker.Volume 34 result3 bool 35 result4 error 36 } 37 findReturnsOnCall map[int]struct { 38 result1 worker.GetResult 39 result2 worker.Volume 40 result3 bool 41 result4 error 42 } 43 invocations map[string][][]interface{} 44 invocationsMutex sync.RWMutex 45} 46 47func (fake *FakeFetchSource) Create(arg1 context.Context) (worker.GetResult, worker.Volume, error) { 48 fake.createMutex.Lock() 49 ret, specificReturn := fake.createReturnsOnCall[len(fake.createArgsForCall)] 50 fake.createArgsForCall = append(fake.createArgsForCall, struct { 51 arg1 context.Context 52 }{arg1}) 53 fake.recordInvocation("Create", []interface{}{arg1}) 54 fake.createMutex.Unlock() 55 if fake.CreateStub != nil { 56 return fake.CreateStub(arg1) 57 } 58 if specificReturn { 59 return ret.result1, ret.result2, ret.result3 60 } 61 fakeReturns := fake.createReturns 62 return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3 63} 64 65func (fake *FakeFetchSource) CreateCallCount() int { 66 fake.createMutex.RLock() 67 defer fake.createMutex.RUnlock() 68 return len(fake.createArgsForCall) 69} 70 71func (fake *FakeFetchSource) CreateCalls(stub func(context.Context) (worker.GetResult, worker.Volume, error)) { 72 fake.createMutex.Lock() 73 defer fake.createMutex.Unlock() 74 fake.CreateStub = stub 75} 76 77func (fake *FakeFetchSource) CreateArgsForCall(i int) context.Context { 78 fake.createMutex.RLock() 79 defer fake.createMutex.RUnlock() 80 argsForCall := fake.createArgsForCall[i] 81 return argsForCall.arg1 82} 83 84func (fake *FakeFetchSource) CreateReturns(result1 worker.GetResult, result2 worker.Volume, result3 error) { 85 fake.createMutex.Lock() 86 defer fake.createMutex.Unlock() 87 fake.CreateStub = nil 88 fake.createReturns = struct { 89 result1 worker.GetResult 90 result2 worker.Volume 91 result3 error 92 }{result1, result2, result3} 93} 94 95func (fake *FakeFetchSource) CreateReturnsOnCall(i int, result1 worker.GetResult, result2 worker.Volume, result3 error) { 96 fake.createMutex.Lock() 97 defer fake.createMutex.Unlock() 98 fake.CreateStub = nil 99 if fake.createReturnsOnCall == nil { 100 fake.createReturnsOnCall = make(map[int]struct { 101 result1 worker.GetResult 102 result2 worker.Volume 103 result3 error 104 }) 105 } 106 fake.createReturnsOnCall[i] = struct { 107 result1 worker.GetResult 108 result2 worker.Volume 109 result3 error 110 }{result1, result2, result3} 111} 112 113func (fake *FakeFetchSource) Find() (worker.GetResult, worker.Volume, bool, error) { 114 fake.findMutex.Lock() 115 ret, specificReturn := fake.findReturnsOnCall[len(fake.findArgsForCall)] 116 fake.findArgsForCall = append(fake.findArgsForCall, struct { 117 }{}) 118 fake.recordInvocation("Find", []interface{}{}) 119 fake.findMutex.Unlock() 120 if fake.FindStub != nil { 121 return fake.FindStub() 122 } 123 if specificReturn { 124 return ret.result1, ret.result2, ret.result3, ret.result4 125 } 126 fakeReturns := fake.findReturns 127 return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3, fakeReturns.result4 128} 129 130func (fake *FakeFetchSource) FindCallCount() int { 131 fake.findMutex.RLock() 132 defer fake.findMutex.RUnlock() 133 return len(fake.findArgsForCall) 134} 135 136func (fake *FakeFetchSource) FindCalls(stub func() (worker.GetResult, worker.Volume, bool, error)) { 137 fake.findMutex.Lock() 138 defer fake.findMutex.Unlock() 139 fake.FindStub = stub 140} 141 142func (fake *FakeFetchSource) FindReturns(result1 worker.GetResult, result2 worker.Volume, result3 bool, result4 error) { 143 fake.findMutex.Lock() 144 defer fake.findMutex.Unlock() 145 fake.FindStub = nil 146 fake.findReturns = struct { 147 result1 worker.GetResult 148 result2 worker.Volume 149 result3 bool 150 result4 error 151 }{result1, result2, result3, result4} 152} 153 154func (fake *FakeFetchSource) FindReturnsOnCall(i int, result1 worker.GetResult, result2 worker.Volume, result3 bool, result4 error) { 155 fake.findMutex.Lock() 156 defer fake.findMutex.Unlock() 157 fake.FindStub = nil 158 if fake.findReturnsOnCall == nil { 159 fake.findReturnsOnCall = make(map[int]struct { 160 result1 worker.GetResult 161 result2 worker.Volume 162 result3 bool 163 result4 error 164 }) 165 } 166 fake.findReturnsOnCall[i] = struct { 167 result1 worker.GetResult 168 result2 worker.Volume 169 result3 bool 170 result4 error 171 }{result1, result2, result3, result4} 172} 173 174func (fake *FakeFetchSource) Invocations() map[string][][]interface{} { 175 fake.invocationsMutex.RLock() 176 defer fake.invocationsMutex.RUnlock() 177 fake.createMutex.RLock() 178 defer fake.createMutex.RUnlock() 179 fake.findMutex.RLock() 180 defer fake.findMutex.RUnlock() 181 copiedInvocations := map[string][][]interface{}{} 182 for key, value := range fake.invocations { 183 copiedInvocations[key] = value 184 } 185 return copiedInvocations 186} 187 188func (fake *FakeFetchSource) recordInvocation(key string, args []interface{}) { 189 fake.invocationsMutex.Lock() 190 defer fake.invocationsMutex.Unlock() 191 if fake.invocations == nil { 192 fake.invocations = map[string][][]interface{}{} 193 } 194 if fake.invocations[key] == nil { 195 fake.invocations[key] = [][]interface{}{} 196 } 197 fake.invocations[key] = append(fake.invocations[key], args) 198} 199 200var _ worker.FetchSource = new(FakeFetchSource) 201