github.com/pf-qiu/concourse/v6@v6.7.3-0.20201207032516-1f455d73275f/tracing/tracingfakes/fake_span.go (about)

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