github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/atc/db/dbfakes/fake_task_cache_factory.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 FakeTaskCacheFactory struct { 11 FindStub func(int, string, string) (db.UsedTaskCache, bool, error) 12 findMutex sync.RWMutex 13 findArgsForCall []struct { 14 arg1 int 15 arg2 string 16 arg3 string 17 } 18 findReturns struct { 19 result1 db.UsedTaskCache 20 result2 bool 21 result3 error 22 } 23 findReturnsOnCall map[int]struct { 24 result1 db.UsedTaskCache 25 result2 bool 26 result3 error 27 } 28 FindOrCreateStub func(int, string, string) (db.UsedTaskCache, error) 29 findOrCreateMutex sync.RWMutex 30 findOrCreateArgsForCall []struct { 31 arg1 int 32 arg2 string 33 arg3 string 34 } 35 findOrCreateReturns struct { 36 result1 db.UsedTaskCache 37 result2 error 38 } 39 findOrCreateReturnsOnCall map[int]struct { 40 result1 db.UsedTaskCache 41 result2 error 42 } 43 invocations map[string][][]interface{} 44 invocationsMutex sync.RWMutex 45 } 46 47 func (fake *FakeTaskCacheFactory) Find(arg1 int, arg2 string, arg3 string) (db.UsedTaskCache, bool, error) { 48 fake.findMutex.Lock() 49 ret, specificReturn := fake.findReturnsOnCall[len(fake.findArgsForCall)] 50 fake.findArgsForCall = append(fake.findArgsForCall, struct { 51 arg1 int 52 arg2 string 53 arg3 string 54 }{arg1, arg2, arg3}) 55 fake.recordInvocation("Find", []interface{}{arg1, arg2, arg3}) 56 fake.findMutex.Unlock() 57 if fake.FindStub != nil { 58 return fake.FindStub(arg1, arg2, arg3) 59 } 60 if specificReturn { 61 return ret.result1, ret.result2, ret.result3 62 } 63 fakeReturns := fake.findReturns 64 return fakeReturns.result1, fakeReturns.result2, fakeReturns.result3 65 } 66 67 func (fake *FakeTaskCacheFactory) FindCallCount() int { 68 fake.findMutex.RLock() 69 defer fake.findMutex.RUnlock() 70 return len(fake.findArgsForCall) 71 } 72 73 func (fake *FakeTaskCacheFactory) FindCalls(stub func(int, string, string) (db.UsedTaskCache, bool, error)) { 74 fake.findMutex.Lock() 75 defer fake.findMutex.Unlock() 76 fake.FindStub = stub 77 } 78 79 func (fake *FakeTaskCacheFactory) FindArgsForCall(i int) (int, string, string) { 80 fake.findMutex.RLock() 81 defer fake.findMutex.RUnlock() 82 argsForCall := fake.findArgsForCall[i] 83 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 84 } 85 86 func (fake *FakeTaskCacheFactory) FindReturns(result1 db.UsedTaskCache, result2 bool, result3 error) { 87 fake.findMutex.Lock() 88 defer fake.findMutex.Unlock() 89 fake.FindStub = nil 90 fake.findReturns = struct { 91 result1 db.UsedTaskCache 92 result2 bool 93 result3 error 94 }{result1, result2, result3} 95 } 96 97 func (fake *FakeTaskCacheFactory) FindReturnsOnCall(i int, result1 db.UsedTaskCache, result2 bool, result3 error) { 98 fake.findMutex.Lock() 99 defer fake.findMutex.Unlock() 100 fake.FindStub = nil 101 if fake.findReturnsOnCall == nil { 102 fake.findReturnsOnCall = make(map[int]struct { 103 result1 db.UsedTaskCache 104 result2 bool 105 result3 error 106 }) 107 } 108 fake.findReturnsOnCall[i] = struct { 109 result1 db.UsedTaskCache 110 result2 bool 111 result3 error 112 }{result1, result2, result3} 113 } 114 115 func (fake *FakeTaskCacheFactory) FindOrCreate(arg1 int, arg2 string, arg3 string) (db.UsedTaskCache, error) { 116 fake.findOrCreateMutex.Lock() 117 ret, specificReturn := fake.findOrCreateReturnsOnCall[len(fake.findOrCreateArgsForCall)] 118 fake.findOrCreateArgsForCall = append(fake.findOrCreateArgsForCall, struct { 119 arg1 int 120 arg2 string 121 arg3 string 122 }{arg1, arg2, arg3}) 123 fake.recordInvocation("FindOrCreate", []interface{}{arg1, arg2, arg3}) 124 fake.findOrCreateMutex.Unlock() 125 if fake.FindOrCreateStub != nil { 126 return fake.FindOrCreateStub(arg1, arg2, arg3) 127 } 128 if specificReturn { 129 return ret.result1, ret.result2 130 } 131 fakeReturns := fake.findOrCreateReturns 132 return fakeReturns.result1, fakeReturns.result2 133 } 134 135 func (fake *FakeTaskCacheFactory) FindOrCreateCallCount() int { 136 fake.findOrCreateMutex.RLock() 137 defer fake.findOrCreateMutex.RUnlock() 138 return len(fake.findOrCreateArgsForCall) 139 } 140 141 func (fake *FakeTaskCacheFactory) FindOrCreateCalls(stub func(int, string, string) (db.UsedTaskCache, error)) { 142 fake.findOrCreateMutex.Lock() 143 defer fake.findOrCreateMutex.Unlock() 144 fake.FindOrCreateStub = stub 145 } 146 147 func (fake *FakeTaskCacheFactory) FindOrCreateArgsForCall(i int) (int, string, string) { 148 fake.findOrCreateMutex.RLock() 149 defer fake.findOrCreateMutex.RUnlock() 150 argsForCall := fake.findOrCreateArgsForCall[i] 151 return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3 152 } 153 154 func (fake *FakeTaskCacheFactory) FindOrCreateReturns(result1 db.UsedTaskCache, result2 error) { 155 fake.findOrCreateMutex.Lock() 156 defer fake.findOrCreateMutex.Unlock() 157 fake.FindOrCreateStub = nil 158 fake.findOrCreateReturns = struct { 159 result1 db.UsedTaskCache 160 result2 error 161 }{result1, result2} 162 } 163 164 func (fake *FakeTaskCacheFactory) FindOrCreateReturnsOnCall(i int, result1 db.UsedTaskCache, result2 error) { 165 fake.findOrCreateMutex.Lock() 166 defer fake.findOrCreateMutex.Unlock() 167 fake.FindOrCreateStub = nil 168 if fake.findOrCreateReturnsOnCall == nil { 169 fake.findOrCreateReturnsOnCall = make(map[int]struct { 170 result1 db.UsedTaskCache 171 result2 error 172 }) 173 } 174 fake.findOrCreateReturnsOnCall[i] = struct { 175 result1 db.UsedTaskCache 176 result2 error 177 }{result1, result2} 178 } 179 180 func (fake *FakeTaskCacheFactory) Invocations() map[string][][]interface{} { 181 fake.invocationsMutex.RLock() 182 defer fake.invocationsMutex.RUnlock() 183 fake.findMutex.RLock() 184 defer fake.findMutex.RUnlock() 185 fake.findOrCreateMutex.RLock() 186 defer fake.findOrCreateMutex.RUnlock() 187 copiedInvocations := map[string][][]interface{}{} 188 for key, value := range fake.invocations { 189 copiedInvocations[key] = value 190 } 191 return copiedInvocations 192 } 193 194 func (fake *FakeTaskCacheFactory) recordInvocation(key string, args []interface{}) { 195 fake.invocationsMutex.Lock() 196 defer fake.invocationsMutex.Unlock() 197 if fake.invocations == nil { 198 fake.invocations = map[string][][]interface{}{} 199 } 200 if fake.invocations[key] == nil { 201 fake.invocations[key] = [][]interface{}{} 202 } 203 fake.invocations[key] = append(fake.invocations[key], args) 204 } 205 206 var _ db.TaskCacheFactory = new(FakeTaskCacheFactory)