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