1// Code generated by counterfeiter. DO NOT EDIT.
2package tracingfakes
3
4import (
5	"context"
6	"sync"
7	"time"
8
9	"go.opentelemetry.io/otel/api/core"
10	"go.opentelemetry.io/otel/api/trace"
11	"google.golang.org/grpc/codes"
12)
13
14type FakeSpan struct {
15	AddEventStub        func(context.Context, string, ...core.KeyValue)
16	addEventMutex       sync.RWMutex
17	addEventArgsForCall []struct {
18		arg1 context.Context
19		arg2 string
20		arg3 []core.KeyValue
21	}
22	AddEventWithTimestampStub        func(context.Context, time.Time, string, ...core.KeyValue)
23	addEventWithTimestampMutex       sync.RWMutex
24	addEventWithTimestampArgsForCall []struct {
25		arg1 context.Context
26		arg2 time.Time
27		arg3 string
28		arg4 []core.KeyValue
29	}
30	EndStub        func(...trace.EndOption)
31	endMutex       sync.RWMutex
32	endArgsForCall []struct {
33		arg1 []trace.EndOption
34	}
35	IsRecordingStub        func() bool
36	isRecordingMutex       sync.RWMutex
37	isRecordingArgsForCall []struct {
38	}
39	isRecordingReturns struct {
40		result1 bool
41	}
42	isRecordingReturnsOnCall map[int]struct {
43		result1 bool
44	}
45	SetAttributesStub        func(...core.KeyValue)
46	setAttributesMutex       sync.RWMutex
47	setAttributesArgsForCall []struct {
48		arg1 []core.KeyValue
49	}
50	SetNameStub        func(string)
51	setNameMutex       sync.RWMutex
52	setNameArgsForCall []struct {
53		arg1 string
54	}
55	SetStatusStub        func(codes.Code)
56	setStatusMutex       sync.RWMutex
57	setStatusArgsForCall []struct {
58		arg1 codes.Code
59	}
60	SpanContextStub        func() core.SpanContext
61	spanContextMutex       sync.RWMutex
62	spanContextArgsForCall []struct {
63	}
64	spanContextReturns struct {
65		result1 core.SpanContext
66	}
67	spanContextReturnsOnCall map[int]struct {
68		result1 core.SpanContext
69	}
70	TracerStub        func() trace.Tracer
71	tracerMutex       sync.RWMutex
72	tracerArgsForCall []struct {
73	}
74	tracerReturns struct {
75		result1 trace.Tracer
76	}
77	tracerReturnsOnCall map[int]struct {
78		result1 trace.Tracer
79	}
80	invocations      map[string][][]interface{}
81	invocationsMutex sync.RWMutex
82}
83
84func (fake *FakeSpan) AddEvent(arg1 context.Context, arg2 string, arg3 ...core.KeyValue) {
85	fake.addEventMutex.Lock()
86	fake.addEventArgsForCall = append(fake.addEventArgsForCall, struct {
87		arg1 context.Context
88		arg2 string
89		arg3 []core.KeyValue
90	}{arg1, arg2, arg3})
91	fake.recordInvocation("AddEvent", []interface{}{arg1, arg2, arg3})
92	fake.addEventMutex.Unlock()
93	if fake.AddEventStub != nil {
94		fake.AddEventStub(arg1, arg2, arg3...)
95	}
96}
97
98func (fake *FakeSpan) AddEventCallCount() int {
99	fake.addEventMutex.RLock()
100	defer fake.addEventMutex.RUnlock()
101	return len(fake.addEventArgsForCall)
102}
103
104func (fake *FakeSpan) AddEventCalls(stub func(context.Context, string, ...core.KeyValue)) {
105	fake.addEventMutex.Lock()
106	defer fake.addEventMutex.Unlock()
107	fake.AddEventStub = stub
108}
109
110func (fake *FakeSpan) AddEventArgsForCall(i int) (context.Context, string, []core.KeyValue) {
111	fake.addEventMutex.RLock()
112	defer fake.addEventMutex.RUnlock()
113	argsForCall := fake.addEventArgsForCall[i]
114	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3
115}
116
117func (fake *FakeSpan) AddEventWithTimestamp(arg1 context.Context, arg2 time.Time, arg3 string, arg4 ...core.KeyValue) {
118	fake.addEventWithTimestampMutex.Lock()
119	fake.addEventWithTimestampArgsForCall = append(fake.addEventWithTimestampArgsForCall, struct {
120		arg1 context.Context
121		arg2 time.Time
122		arg3 string
123		arg4 []core.KeyValue
124	}{arg1, arg2, arg3, arg4})
125	fake.recordInvocation("AddEventWithTimestamp", []interface{}{arg1, arg2, arg3, arg4})
126	fake.addEventWithTimestampMutex.Unlock()
127	if fake.AddEventWithTimestampStub != nil {
128		fake.AddEventWithTimestampStub(arg1, arg2, arg3, arg4...)
129	}
130}
131
132func (fake *FakeSpan) AddEventWithTimestampCallCount() int {
133	fake.addEventWithTimestampMutex.RLock()
134	defer fake.addEventWithTimestampMutex.RUnlock()
135	return len(fake.addEventWithTimestampArgsForCall)
136}
137
138func (fake *FakeSpan) AddEventWithTimestampCalls(stub func(context.Context, time.Time, string, ...core.KeyValue)) {
139	fake.addEventWithTimestampMutex.Lock()
140	defer fake.addEventWithTimestampMutex.Unlock()
141	fake.AddEventWithTimestampStub = stub
142}
143
144func (fake *FakeSpan) AddEventWithTimestampArgsForCall(i int) (context.Context, time.Time, string, []core.KeyValue) {
145	fake.addEventWithTimestampMutex.RLock()
146	defer fake.addEventWithTimestampMutex.RUnlock()
147	argsForCall := fake.addEventWithTimestampArgsForCall[i]
148	return argsForCall.arg1, argsForCall.arg2, argsForCall.arg3, argsForCall.arg4
149}
150
151func (fake *FakeSpan) End(arg1 ...trace.EndOption) {
152	fake.endMutex.Lock()
153	fake.endArgsForCall = append(fake.endArgsForCall, struct {
154		arg1 []trace.EndOption
155	}{arg1})
156	fake.recordInvocation("End", []interface{}{arg1})
157	fake.endMutex.Unlock()
158	if fake.EndStub != nil {
159		fake.EndStub(arg1...)
160	}
161}
162
163func (fake *FakeSpan) EndCallCount() int {
164	fake.endMutex.RLock()
165	defer fake.endMutex.RUnlock()
166	return len(fake.endArgsForCall)
167}
168
169func (fake *FakeSpan) EndCalls(stub func(...trace.EndOption)) {
170	fake.endMutex.Lock()
171	defer fake.endMutex.Unlock()
172	fake.EndStub = stub
173}
174
175func (fake *FakeSpan) EndArgsForCall(i int) []trace.EndOption {
176	fake.endMutex.RLock()
177	defer fake.endMutex.RUnlock()
178	argsForCall := fake.endArgsForCall[i]
179	return argsForCall.arg1
180}
181
182func (fake *FakeSpan) IsRecording() bool {
183	fake.isRecordingMutex.Lock()
184	ret, specificReturn := fake.isRecordingReturnsOnCall[len(fake.isRecordingArgsForCall)]
185	fake.isRecordingArgsForCall = append(fake.isRecordingArgsForCall, struct {
186	}{})
187	fake.recordInvocation("IsRecording", []interface{}{})
188	fake.isRecordingMutex.Unlock()
189	if fake.IsRecordingStub != nil {
190		return fake.IsRecordingStub()
191	}
192	if specificReturn {
193		return ret.result1
194	}
195	fakeReturns := fake.isRecordingReturns
196	return fakeReturns.result1
197}
198
199func (fake *FakeSpan) IsRecordingCallCount() int {
200	fake.isRecordingMutex.RLock()
201	defer fake.isRecordingMutex.RUnlock()
202	return len(fake.isRecordingArgsForCall)
203}
204
205func (fake *FakeSpan) IsRecordingCalls(stub func() bool) {
206	fake.isRecordingMutex.Lock()
207	defer fake.isRecordingMutex.Unlock()
208	fake.IsRecordingStub = stub
209}
210
211func (fake *FakeSpan) IsRecordingReturns(result1 bool) {
212	fake.isRecordingMutex.Lock()
213	defer fake.isRecordingMutex.Unlock()
214	fake.IsRecordingStub = nil
215	fake.isRecordingReturns = struct {
216		result1 bool
217	}{result1}
218}
219
220func (fake *FakeSpan) IsRecordingReturnsOnCall(i int, result1 bool) {
221	fake.isRecordingMutex.Lock()
222	defer fake.isRecordingMutex.Unlock()
223	fake.IsRecordingStub = nil
224	if fake.isRecordingReturnsOnCall == nil {
225		fake.isRecordingReturnsOnCall = make(map[int]struct {
226			result1 bool
227		})
228	}
229	fake.isRecordingReturnsOnCall[i] = struct {
230		result1 bool
231	}{result1}
232}
233
234func (fake *FakeSpan) SetAttributes(arg1 ...core.KeyValue) {
235	fake.setAttributesMutex.Lock()
236	fake.setAttributesArgsForCall = append(fake.setAttributesArgsForCall, struct {
237		arg1 []core.KeyValue
238	}{arg1})
239	fake.recordInvocation("SetAttributes", []interface{}{arg1})
240	fake.setAttributesMutex.Unlock()
241	if fake.SetAttributesStub != nil {
242		fake.SetAttributesStub(arg1...)
243	}
244}
245
246func (fake *FakeSpan) SetAttributesCallCount() int {
247	fake.setAttributesMutex.RLock()
248	defer fake.setAttributesMutex.RUnlock()
249	return len(fake.setAttributesArgsForCall)
250}
251
252func (fake *FakeSpan) SetAttributesCalls(stub func(...core.KeyValue)) {
253	fake.setAttributesMutex.Lock()
254	defer fake.setAttributesMutex.Unlock()
255	fake.SetAttributesStub = stub
256}
257
258func (fake *FakeSpan) SetAttributesArgsForCall(i int) []core.KeyValue {
259	fake.setAttributesMutex.RLock()
260	defer fake.setAttributesMutex.RUnlock()
261	argsForCall := fake.setAttributesArgsForCall[i]
262	return argsForCall.arg1
263}
264
265func (fake *FakeSpan) SetName(arg1 string) {
266	fake.setNameMutex.Lock()
267	fake.setNameArgsForCall = append(fake.setNameArgsForCall, struct {
268		arg1 string
269	}{arg1})
270	fake.recordInvocation("SetName", []interface{}{arg1})
271	fake.setNameMutex.Unlock()
272	if fake.SetNameStub != nil {
273		fake.SetNameStub(arg1)
274	}
275}
276
277func (fake *FakeSpan) SetNameCallCount() int {
278	fake.setNameMutex.RLock()
279	defer fake.setNameMutex.RUnlock()
280	return len(fake.setNameArgsForCall)
281}
282
283func (fake *FakeSpan) SetNameCalls(stub func(string)) {
284	fake.setNameMutex.Lock()
285	defer fake.setNameMutex.Unlock()
286	fake.SetNameStub = stub
287}
288
289func (fake *FakeSpan) SetNameArgsForCall(i int) string {
290	fake.setNameMutex.RLock()
291	defer fake.setNameMutex.RUnlock()
292	argsForCall := fake.setNameArgsForCall[i]
293	return argsForCall.arg1
294}
295
296func (fake *FakeSpan) SetStatus(arg1 codes.Code) {
297	fake.setStatusMutex.Lock()
298	fake.setStatusArgsForCall = append(fake.setStatusArgsForCall, struct {
299		arg1 codes.Code
300	}{arg1})
301	fake.recordInvocation("SetStatus", []interface{}{arg1})
302	fake.setStatusMutex.Unlock()
303	if fake.SetStatusStub != nil {
304		fake.SetStatusStub(arg1)
305	}
306}
307
308func (fake *FakeSpan) SetStatusCallCount() int {
309	fake.setStatusMutex.RLock()
310	defer fake.setStatusMutex.RUnlock()
311	return len(fake.setStatusArgsForCall)
312}
313
314func (fake *FakeSpan) SetStatusCalls(stub func(codes.Code)) {
315	fake.setStatusMutex.Lock()
316	defer fake.setStatusMutex.Unlock()
317	fake.SetStatusStub = stub
318}
319
320func (fake *FakeSpan) SetStatusArgsForCall(i int) codes.Code {
321	fake.setStatusMutex.RLock()
322	defer fake.setStatusMutex.RUnlock()
323	argsForCall := fake.setStatusArgsForCall[i]
324	return argsForCall.arg1
325}
326
327func (fake *FakeSpan) SpanContext() core.SpanContext {
328	fake.spanContextMutex.Lock()
329	ret, specificReturn := fake.spanContextReturnsOnCall[len(fake.spanContextArgsForCall)]
330	fake.spanContextArgsForCall = append(fake.spanContextArgsForCall, struct {
331	}{})
332	fake.recordInvocation("SpanContext", []interface{}{})
333	fake.spanContextMutex.Unlock()
334	if fake.SpanContextStub != nil {
335		return fake.SpanContextStub()
336	}
337	if specificReturn {
338		return ret.result1
339	}
340	fakeReturns := fake.spanContextReturns
341	return fakeReturns.result1
342}
343
344func (fake *FakeSpan) SpanContextCallCount() int {
345	fake.spanContextMutex.RLock()
346	defer fake.spanContextMutex.RUnlock()
347	return len(fake.spanContextArgsForCall)
348}
349
350func (fake *FakeSpan) SpanContextCalls(stub func() core.SpanContext) {
351	fake.spanContextMutex.Lock()
352	defer fake.spanContextMutex.Unlock()
353	fake.SpanContextStub = stub
354}
355
356func (fake *FakeSpan) SpanContextReturns(result1 core.SpanContext) {
357	fake.spanContextMutex.Lock()
358	defer fake.spanContextMutex.Unlock()
359	fake.SpanContextStub = nil
360	fake.spanContextReturns = struct {
361		result1 core.SpanContext
362	}{result1}
363}
364
365func (fake *FakeSpan) SpanContextReturnsOnCall(i int, result1 core.SpanContext) {
366	fake.spanContextMutex.Lock()
367	defer fake.spanContextMutex.Unlock()
368	fake.SpanContextStub = nil
369	if fake.spanContextReturnsOnCall == nil {
370		fake.spanContextReturnsOnCall = make(map[int]struct {
371			result1 core.SpanContext
372		})
373	}
374	fake.spanContextReturnsOnCall[i] = struct {
375		result1 core.SpanContext
376	}{result1}
377}
378
379func (fake *FakeSpan) Tracer() trace.Tracer {
380	fake.tracerMutex.Lock()
381	ret, specificReturn := fake.tracerReturnsOnCall[len(fake.tracerArgsForCall)]
382	fake.tracerArgsForCall = append(fake.tracerArgsForCall, struct {
383	}{})
384	fake.recordInvocation("Tracer", []interface{}{})
385	fake.tracerMutex.Unlock()
386	if fake.TracerStub != nil {
387		return fake.TracerStub()
388	}
389	if specificReturn {
390		return ret.result1
391	}
392	fakeReturns := fake.tracerReturns
393	return fakeReturns.result1
394}
395
396func (fake *FakeSpan) TracerCallCount() int {
397	fake.tracerMutex.RLock()
398	defer fake.tracerMutex.RUnlock()
399	return len(fake.tracerArgsForCall)
400}
401
402func (fake *FakeSpan) TracerCalls(stub func() trace.Tracer) {
403	fake.tracerMutex.Lock()
404	defer fake.tracerMutex.Unlock()
405	fake.TracerStub = stub
406}
407
408func (fake *FakeSpan) TracerReturns(result1 trace.Tracer) {
409	fake.tracerMutex.Lock()
410	defer fake.tracerMutex.Unlock()
411	fake.TracerStub = nil
412	fake.tracerReturns = struct {
413		result1 trace.Tracer
414	}{result1}
415}
416
417func (fake *FakeSpan) TracerReturnsOnCall(i int, result1 trace.Tracer) {
418	fake.tracerMutex.Lock()
419	defer fake.tracerMutex.Unlock()
420	fake.TracerStub = nil
421	if fake.tracerReturnsOnCall == nil {
422		fake.tracerReturnsOnCall = make(map[int]struct {
423			result1 trace.Tracer
424		})
425	}
426	fake.tracerReturnsOnCall[i] = struct {
427		result1 trace.Tracer
428	}{result1}
429}
430
431func (fake *FakeSpan) Invocations() map[string][][]interface{} {
432	fake.invocationsMutex.RLock()
433	defer fake.invocationsMutex.RUnlock()
434	fake.addEventMutex.RLock()
435	defer fake.addEventMutex.RUnlock()
436	fake.addEventWithTimestampMutex.RLock()
437	defer fake.addEventWithTimestampMutex.RUnlock()
438	fake.endMutex.RLock()
439	defer fake.endMutex.RUnlock()
440	fake.isRecordingMutex.RLock()
441	defer fake.isRecordingMutex.RUnlock()
442	fake.setAttributesMutex.RLock()
443	defer fake.setAttributesMutex.RUnlock()
444	fake.setNameMutex.RLock()
445	defer fake.setNameMutex.RUnlock()
446	fake.setStatusMutex.RLock()
447	defer fake.setStatusMutex.RUnlock()
448	fake.spanContextMutex.RLock()
449	defer fake.spanContextMutex.RUnlock()
450	fake.tracerMutex.RLock()
451	defer fake.tracerMutex.RUnlock()
452	copiedInvocations := map[string][][]interface{}{}
453	for key, value := range fake.invocations {
454		copiedInvocations[key] = value
455	}
456	return copiedInvocations
457}
458
459func (fake *FakeSpan) recordInvocation(key string, args []interface{}) {
460	fake.invocationsMutex.Lock()
461	defer fake.invocationsMutex.Unlock()
462	if fake.invocations == nil {
463		fake.invocations = map[string][][]interface{}{}
464	}
465	if fake.invocations[key] == nil {
466		fake.invocations[key] = [][]interface{}{}
467	}
468	fake.invocations[key] = append(fake.invocations[key], args)
469}
470
471var _ trace.Span = new(FakeSpan)
472