diff --git a/go.mod b/go.mod index 64a69b8..3845e3e 100644 --- a/go.mod +++ b/go.mod @@ -13,8 +13,6 @@ require ( github.com/twitchtv/twirp v5.8.0+incompatible go.elastic.co/apm/v2 v2.2.0 go.opencensus.io v0.23.0 - go.opentelemetry.io/otel v1.24.0 - go.opentelemetry.io/otel/trace v1.24.0 golang.org/x/tools v0.1.11-0.20220316014157-77aa08bb151a google.golang.org/grpc v1.45.0 ) @@ -29,10 +27,9 @@ require ( github.com/elastic/go-licenser v0.4.0 // indirect github.com/elastic/go-sysinfo v1.7.1 // indirect github.com/elastic/go-windows v1.0.1 // indirect - github.com/go-logr/logr v1.4.1 // indirect - github.com/go-logr/stdr v1.2.2 // indirect github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect github.com/golang/protobuf v1.5.2 // indirect + github.com/google/go-cmp v0.6.0 // indirect github.com/google/uuid v1.3.0 // indirect github.com/huandu/xstrings v1.3.2 // indirect github.com/imdario/mergo v0.3.12 // indirect @@ -51,7 +48,6 @@ require ( github.com/shopspring/decimal v1.2.0 // indirect github.com/spf13/cast v1.4.1 // indirect go.elastic.co/fastjson v1.1.0 // indirect - go.opentelemetry.io/otel/metric v1.24.0 // indirect golang.org/x/crypto v0.0.0-20220411220226-7b82a4e95df4 // indirect golang.org/x/mod v0.6.0-dev.0.20220106191415-9b9b3d81d5e3 // indirect golang.org/x/net v0.7.0 // indirect diff --git a/go.sum b/go.sum index 9d0f7b9..51672e3 100644 --- a/go.sum +++ b/go.sum @@ -92,11 +92,6 @@ github.com/go-kit/log v0.1.0/go.mod h1:zbhenjAZHb184qTLMA9ZjW7ThYL0H2mk7Q6pNt4vb github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE= github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk= github.com/go-logfmt/logfmt v0.5.0/go.mod h1:wCYkCAKZfumFQihp8CzCvQ3paCTfi41vtzG1KdI/P7A= -github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= -github.com/go-logr/logr v1.4.1 h1:pKouT5E8xu9zeFC39JXRDukb6JFQPXM5p5I91188VAQ= -github.com/go-logr/logr v1.4.1/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY= -github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= -github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= github.com/go-stack/stack v1.8.0/go.mod h1:v0f6uXyyMGvRgIKkXu+yp6POWl0qKG85gN/melR3HDY= github.com/gogo/protobuf v1.1.1/go.mod h1:r8qH/GZQm5c6nD/R0oafs1akxWv10x8SbQlK7atdtwQ= github.com/gojuno/minimock/v3 v3.0.4/go.mod h1:HqeqnwV8mAABn3pO5hqF+RE7gjA0jsN8cbbSogoGrzI= @@ -295,12 +290,6 @@ go.opencensus.io v0.22.3/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw= go.opencensus.io v0.22.4/go.mod h1:yxeiOL68Rb0Xd1ddK5vPZ/oVn4vY4Ynel7k9FzqtOIw= go.opencensus.io v0.23.0 h1:gqCw0LfLxScz8irSi8exQc7fyQ0fKQU/qnC/X8+V/1M= go.opencensus.io v0.23.0/go.mod h1:XItmlyltB5F7CS4xOC1DcqMoFqwtC6OG2xF7mCv7P7E= -go.opentelemetry.io/otel v1.24.0 h1:0LAOdjNmQeSTzGBzduGe/rU4tZhMwL5rWgtp9Ku5Jfo= -go.opentelemetry.io/otel v1.24.0/go.mod h1:W7b9Ozg4nkF5tWI5zsXkaKKDjdVjpD4oAt9Qi/MArHo= -go.opentelemetry.io/otel/metric v1.24.0 h1:6EhoGWWK28x1fbpA4tYTOWBkPefTDQnb8WSGXlc88kI= -go.opentelemetry.io/otel/metric v1.24.0/go.mod h1:VYhLe1rFfxuTXLgj4CBiyz+9WYBA8pNGJgDcSFRKBco= -go.opentelemetry.io/otel/trace v1.24.0 h1:CsKnnL4dUAr/0llH9FKuc698G04IrpWV0MQA/Y1YELI= -go.opentelemetry.io/otel/trace v1.24.0/go.mod h1:HPc3Xr/cOApsBI154IU0OI0HJexz+aw5uPdbs3UCjNU= go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI= golang.org/x/crypto v0.0.0-20180904163835-0709b304e793/go.mod h1:6SG95UA2DQfeDnfUPMdvaQW0Q7yPrPDi9nlGo2tz2b4= golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= diff --git a/templates_tests/interface_with_opentelemetry.go b/templates_tests/interface_with_opentelemetry.go deleted file mode 100755 index 24a4b0b..0000000 --- a/templates_tests/interface_with_opentelemetry.go +++ /dev/null @@ -1,77 +0,0 @@ -// Code generated by gowrap. DO NOT EDIT. -// template: ../templates/opentelemetry -// gowrap: http://github.com/hexdigest/gowrap - -package templatestests - -//go:generate gowrap gen -p github.com/hexdigest/gowrap/templates_tests -i TestInterface -t ../templates/opentelemetry -o interface_with_opentelemetry.go -v DecoratorName=TestInterfaceWithOpentelemetry -l "" - -import ( - "context" - - "go.opentelemetry.io/otel" - "go.opentelemetry.io/otel/attribute" - "go.opentelemetry.io/otel/trace" -) - -// TestInterfaceWithOpentelemetry implements TestInterface interface instrumented with opentracing spans -type TestInterfaceWithOpentelemetry struct { - TestInterface - _instance string - _spanDecorator func(span trace.Span, params, results map[string]interface{}) -} - -// NewTestInterfaceWithOpentelemetry returns TestInterfaceWithOpentelemetry -func NewTestInterfaceWithOpentelemetry(base TestInterface, instance string, spanDecorator ...func(span trace.Span, params, results map[string]interface{})) TestInterfaceWithOpentelemetry { - d := TestInterfaceWithOpentelemetry{ - TestInterface: base, - _instance: instance, - } - - if len(spanDecorator) > 0 && spanDecorator[0] != nil { - d._spanDecorator = spanDecorator[0] - } - - return d -} - -// ContextNoError implements TestInterface -func (_d TestInterfaceWithOpentelemetry) ContextNoError(ctx context.Context, a1 string, a2 string) { - ctx, _span := otel.Tracer(_d._instance).Start(ctx, "TestInterface.ContextNoError") - defer func() { - if _d._spanDecorator != nil { - _d._spanDecorator(_span, map[string]interface{}{ - "ctx": ctx, - "a1": a1, - "a2": a2}, map[string]interface{}{}) - } - _span.End() - }() - _d.TestInterface.ContextNoError(ctx, a1, a2) - return -} - -// F implements TestInterface -func (_d TestInterfaceWithOpentelemetry) F(ctx context.Context, a1 string, a2 ...string) (result1 string, result2 string, err error) { - ctx, _span := otel.Tracer(_d._instance).Start(ctx, "TestInterface.F") - defer func() { - if _d._spanDecorator != nil { - _d._spanDecorator(_span, map[string]interface{}{ - "ctx": ctx, - "a1": a1, - "a2": a2}, map[string]interface{}{ - "result1": result1, - "result2": result2, - "err": err}) - } else if err != nil { - _span.RecordError(err) - _span.SetAttributes( - attribute.String("event", "error"), - attribute.String("message", err.Error()), - ) - } - - _span.End() - }() - return _d.TestInterface.F(ctx, a1, a2...) -} diff --git a/templates_tests/interface_with_opentelemetry_test.go b/templates_tests/interface_with_opentelemetry_test.go deleted file mode 100644 index b2e81c2..0000000 --- a/templates_tests/interface_with_opentelemetry_test.go +++ /dev/null @@ -1,74 +0,0 @@ -package templatestests - -import ( - "context" - "errors" - "testing" - - "github.com/gojuno/minimock/v3" - "github.com/stretchr/testify/assert" - "go.opentelemetry.io/otel" - "go.opentelemetry.io/otel/attribute" -) - -func TestTestInterfaceWithOpenTelemetryTracing_F(t *testing.T) { - t.Run("no error", func(t *testing.T) { - impl := &testImpl{r1: "1", r2: "2"} - wrapped := NewTestInterfaceWithOpentelemetry(impl, "test") - - mc := minimock.NewController(t) - defer mc.Finish() - - span := NewOpentelemetrySpanMock(mc) - span.EndMock.Expect().Return() - - tr := NewOpentelemetryTracerMock(mc) - tr.StartMock.Expect(context.Background(), "TestInterface.F").Return(context.Background(), span) - - tp := NewOpentelemetryTracerProviderMock(mc) - tp.TracerMock.Expect("test").Return(tr) - - otel.SetTracerProvider(tp) - - r1, r2, err := wrapped.F(context.Background(), "a1", "a2") - assert.NoError(t, err) - assert.Equal(t, "1", r1) - assert.Equal(t, "2", r2) - }) - - t.Run("error", func(t *testing.T) { - err := errors.New("unexpected error") - - impl := &testImpl{r1: "1", r2: "2", err: err} - wrapped := NewTestInterfaceWithOpentelemetry(impl, "test") - - mc := minimock.NewController(t) - defer mc.Finish() - - span := NewOpentelemetrySpanMock(mc) - span.RecordErrorMock.Expect(err) - span.EndMock.Expect().Return() - span.SetAttributesMock. - Inspect(func(kv ...attribute.KeyValue) { - assert.Equal(t, string(kv[0].Key), "event") - assert.Equal(t, kv[0].Value.AsString(), "error") - assert.Equal(t, string(kv[1].Key), "message") - assert.Equal(t, kv[1].Value.AsString(), err.Error()) - }).Return() - - tr := NewOpentelemetryTracerMock(mc) - tr.StartMock.Expect(context.Background(), "TestInterface.F").Return(context.Background(), span) - - tp := NewOpentelemetryTracerProviderMock(mc) - tp.TracerMock.Expect("test").Return(tr) - - otel.SetTracerProvider(tp) - - r1, r2, err := wrapped.F(context.Background(), "a1", "a2") - - assert.Error(t, err) - assert.Equal(t, "1", r1) - assert.Equal(t, "2", r2) - - }) -} diff --git a/templates_tests/span_opentelemery_mock_test.go b/templates_tests/span_opentelemery_mock_test.go deleted file mode 100644 index a6c5765..0000000 --- a/templates_tests/span_opentelemery_mock_test.go +++ /dev/null @@ -1,1717 +0,0 @@ -package templatestests - -// Code generated by http://github.com/gojuno/minimock (dev). DO NOT EDIT. - -//go:generate minimock -i go.opentelemetry.io/otel/trace.Span -o ./span_opentelemery_mock_test.go -n OpentelemetrySpanMock - -import ( - "sync" - mm_atomic "sync/atomic" - mm_time "time" - - "github.com/gojuno/minimock/v3" - "go.opentelemetry.io/otel/attribute" - "go.opentelemetry.io/otel/codes" - mm_trace "go.opentelemetry.io/otel/trace" -) - -// OpentelemetrySpanMock implements trace.Span -type OpentelemetrySpanMock struct { - t minimock.Tester - - funcAddEvent func(name string, options ...mm_trace.EventOption) - inspectFuncAddEvent func(name string, options ...mm_trace.EventOption) - afterAddEventCounter uint64 - beforeAddEventCounter uint64 - AddEventMock mOpentelemetrySpanMockAddEvent - - funcEnd func(options ...mm_trace.SpanEndOption) - inspectFuncEnd func(options ...mm_trace.SpanEndOption) - afterEndCounter uint64 - beforeEndCounter uint64 - EndMock mOpentelemetrySpanMockEnd - - funcIsRecording func() (b1 bool) - inspectFuncIsRecording func() - afterIsRecordingCounter uint64 - beforeIsRecordingCounter uint64 - IsRecordingMock mOpentelemetrySpanMockIsRecording - - funcRecordError func(err error, options ...mm_trace.EventOption) - inspectFuncRecordError func(err error, options ...mm_trace.EventOption) - afterRecordErrorCounter uint64 - beforeRecordErrorCounter uint64 - RecordErrorMock mOpentelemetrySpanMockRecordError - - funcSetAttributes func(kv ...attribute.KeyValue) - inspectFuncSetAttributes func(kv ...attribute.KeyValue) - afterSetAttributesCounter uint64 - beforeSetAttributesCounter uint64 - SetAttributesMock mOpentelemetrySpanMockSetAttributes - - funcSetName func(name string) - inspectFuncSetName func(name string) - afterSetNameCounter uint64 - beforeSetNameCounter uint64 - SetNameMock mOpentelemetrySpanMockSetName - - funcSetStatus func(code codes.Code, description string) - inspectFuncSetStatus func(code codes.Code, description string) - afterSetStatusCounter uint64 - beforeSetStatusCounter uint64 - SetStatusMock mOpentelemetrySpanMockSetStatus - - funcSpanContext func() (s1 mm_trace.SpanContext) - inspectFuncSpanContext func() - afterSpanContextCounter uint64 - beforeSpanContextCounter uint64 - SpanContextMock mOpentelemetrySpanMockSpanContext - - funcTracerProvider func() (t1 mm_trace.TracerProvider) - inspectFuncTracerProvider func() - afterTracerProviderCounter uint64 - beforeTracerProviderCounter uint64 - TracerProviderMock mOpentelemetrySpanMockTracerProvider -} - -// NewOpentelemetrySpanMock returns a mock for trace.Span -func NewOpentelemetrySpanMock(t minimock.Tester) *OpentelemetrySpanMock { - m := &OpentelemetrySpanMock{t: t} - if controller, ok := t.(minimock.MockController); ok { - controller.RegisterMocker(m) - } - - m.AddEventMock = mOpentelemetrySpanMockAddEvent{mock: m} - m.AddEventMock.callArgs = []*OpentelemetrySpanMockAddEventParams{} - - m.EndMock = mOpentelemetrySpanMockEnd{mock: m} - m.EndMock.callArgs = []*OpentelemetrySpanMockEndParams{} - - m.IsRecordingMock = mOpentelemetrySpanMockIsRecording{mock: m} - - m.RecordErrorMock = mOpentelemetrySpanMockRecordError{mock: m} - m.RecordErrorMock.callArgs = []*OpentelemetrySpanMockRecordErrorParams{} - - m.SetAttributesMock = mOpentelemetrySpanMockSetAttributes{mock: m} - m.SetAttributesMock.callArgs = []*OpentelemetrySpanMockSetAttributesParams{} - - m.SetNameMock = mOpentelemetrySpanMockSetName{mock: m} - m.SetNameMock.callArgs = []*OpentelemetrySpanMockSetNameParams{} - - m.SetStatusMock = mOpentelemetrySpanMockSetStatus{mock: m} - m.SetStatusMock.callArgs = []*OpentelemetrySpanMockSetStatusParams{} - - m.SpanContextMock = mOpentelemetrySpanMockSpanContext{mock: m} - - m.TracerProviderMock = mOpentelemetrySpanMockTracerProvider{mock: m} - - return m -} - -type mOpentelemetrySpanMockAddEvent struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockAddEventExpectation - expectations []*OpentelemetrySpanMockAddEventExpectation - - callArgs []*OpentelemetrySpanMockAddEventParams - mutex sync.RWMutex -} - -// OpentelemetrySpanMockAddEventExpectation specifies expectation struct of the Span.AddEvent -type OpentelemetrySpanMockAddEventExpectation struct { - mock *OpentelemetrySpanMock - params *OpentelemetrySpanMockAddEventParams - - Counter uint64 -} - -// OpentelemetrySpanMockAddEventParams contains parameters of the Span.AddEvent -type OpentelemetrySpanMockAddEventParams struct { - name string - options []mm_trace.EventOption -} - -// Expect sets up expected params for Span.AddEvent -func (mmAddEvent *mOpentelemetrySpanMockAddEvent) Expect(name string, options ...mm_trace.EventOption) *mOpentelemetrySpanMockAddEvent { - if mmAddEvent.mock.funcAddEvent != nil { - mmAddEvent.mock.t.Fatalf("OpentelemetrySpanMock.AddEvent mock is already set by Set") - } - - if mmAddEvent.defaultExpectation == nil { - mmAddEvent.defaultExpectation = &OpentelemetrySpanMockAddEventExpectation{} - } - - mmAddEvent.defaultExpectation.params = &OpentelemetrySpanMockAddEventParams{name, options} - for _, e := range mmAddEvent.expectations { - if minimock.Equal(e.params, mmAddEvent.defaultExpectation.params) { - mmAddEvent.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmAddEvent.defaultExpectation.params) - } - } - - return mmAddEvent -} - -// Inspect accepts an inspector function that has same arguments as the Span.AddEvent -func (mmAddEvent *mOpentelemetrySpanMockAddEvent) Inspect(f func(name string, options ...mm_trace.EventOption)) *mOpentelemetrySpanMockAddEvent { - if mmAddEvent.mock.inspectFuncAddEvent != nil { - mmAddEvent.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.AddEvent") - } - - mmAddEvent.mock.inspectFuncAddEvent = f - - return mmAddEvent -} - -// Return sets up results that will be returned by Span.AddEvent -func (mmAddEvent *mOpentelemetrySpanMockAddEvent) Return() *OpentelemetrySpanMock { - if mmAddEvent.mock.funcAddEvent != nil { - mmAddEvent.mock.t.Fatalf("OpentelemetrySpanMock.AddEvent mock is already set by Set") - } - - if mmAddEvent.defaultExpectation == nil { - mmAddEvent.defaultExpectation = &OpentelemetrySpanMockAddEventExpectation{mock: mmAddEvent.mock} - } - - return mmAddEvent.mock -} - -// Set uses given function f to mock the Span.AddEvent method -func (mmAddEvent *mOpentelemetrySpanMockAddEvent) Set(f func(name string, options ...mm_trace.EventOption)) *OpentelemetrySpanMock { - if mmAddEvent.defaultExpectation != nil { - mmAddEvent.mock.t.Fatalf("Default expectation is already set for the Span.AddEvent method") - } - - if len(mmAddEvent.expectations) > 0 { - mmAddEvent.mock.t.Fatalf("Some expectations are already set for the Span.AddEvent method") - } - - mmAddEvent.mock.funcAddEvent = f - return mmAddEvent.mock -} - -// AddEvent implements trace.Span -func (mmAddEvent *OpentelemetrySpanMock) AddEvent(name string, options ...mm_trace.EventOption) { - mm_atomic.AddUint64(&mmAddEvent.beforeAddEventCounter, 1) - defer mm_atomic.AddUint64(&mmAddEvent.afterAddEventCounter, 1) - - if mmAddEvent.inspectFuncAddEvent != nil { - mmAddEvent.inspectFuncAddEvent(name, options...) - } - - mm_params := &OpentelemetrySpanMockAddEventParams{name, options} - - // Record call args - mmAddEvent.AddEventMock.mutex.Lock() - mmAddEvent.AddEventMock.callArgs = append(mmAddEvent.AddEventMock.callArgs, mm_params) - mmAddEvent.AddEventMock.mutex.Unlock() - - for _, e := range mmAddEvent.AddEventMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return - } - } - - if mmAddEvent.AddEventMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmAddEvent.AddEventMock.defaultExpectation.Counter, 1) - mm_want := mmAddEvent.AddEventMock.defaultExpectation.params - mm_got := OpentelemetrySpanMockAddEventParams{name, options} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmAddEvent.t.Errorf("OpentelemetrySpanMock.AddEvent got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - return - - } - if mmAddEvent.funcAddEvent != nil { - mmAddEvent.funcAddEvent(name, options...) - return - } - mmAddEvent.t.Fatalf("Unexpected call to OpentelemetrySpanMock.AddEvent. %v %v", name, options) - -} - -// AddEventAfterCounter returns a count of finished OpentelemetrySpanMock.AddEvent invocations -func (mmAddEvent *OpentelemetrySpanMock) AddEventAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmAddEvent.afterAddEventCounter) -} - -// AddEventBeforeCounter returns a count of OpentelemetrySpanMock.AddEvent invocations -func (mmAddEvent *OpentelemetrySpanMock) AddEventBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmAddEvent.beforeAddEventCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetrySpanMock.AddEvent. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmAddEvent *mOpentelemetrySpanMockAddEvent) Calls() []*OpentelemetrySpanMockAddEventParams { - mmAddEvent.mutex.RLock() - - argCopy := make([]*OpentelemetrySpanMockAddEventParams, len(mmAddEvent.callArgs)) - copy(argCopy, mmAddEvent.callArgs) - - mmAddEvent.mutex.RUnlock() - - return argCopy -} - -// MinimockAddEventDone returns true if the count of the AddEvent invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockAddEventDone() bool { - for _, e := range m.AddEventMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.AddEventMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterAddEventCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcAddEvent != nil && mm_atomic.LoadUint64(&m.afterAddEventCounter) < 1 { - return false - } - return true -} - -// MinimockAddEventInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockAddEventInspect() { - for _, e := range m.AddEventMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetrySpanMock.AddEvent with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.AddEventMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterAddEventCounter) < 1 { - if m.AddEventMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetrySpanMock.AddEvent") - } else { - m.t.Errorf("Expected call to OpentelemetrySpanMock.AddEvent with params: %#v", *m.AddEventMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcAddEvent != nil && mm_atomic.LoadUint64(&m.afterAddEventCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.AddEvent") - } -} - -type mOpentelemetrySpanMockEnd struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockEndExpectation - expectations []*OpentelemetrySpanMockEndExpectation - - callArgs []*OpentelemetrySpanMockEndParams - mutex sync.RWMutex -} - -// OpentelemetrySpanMockEndExpectation specifies expectation struct of the Span.End -type OpentelemetrySpanMockEndExpectation struct { - mock *OpentelemetrySpanMock - params *OpentelemetrySpanMockEndParams - - Counter uint64 -} - -// OpentelemetrySpanMockEndParams contains parameters of the Span.End -type OpentelemetrySpanMockEndParams struct { - options []mm_trace.SpanEndOption -} - -// Expect sets up expected params for Span.End -func (mmEnd *mOpentelemetrySpanMockEnd) Expect(options ...mm_trace.SpanEndOption) *mOpentelemetrySpanMockEnd { - if mmEnd.mock.funcEnd != nil { - mmEnd.mock.t.Fatalf("OpentelemetrySpanMock.End mock is already set by Set") - } - - if mmEnd.defaultExpectation == nil { - mmEnd.defaultExpectation = &OpentelemetrySpanMockEndExpectation{} - } - - mmEnd.defaultExpectation.params = &OpentelemetrySpanMockEndParams{options} - for _, e := range mmEnd.expectations { - if minimock.Equal(e.params, mmEnd.defaultExpectation.params) { - mmEnd.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmEnd.defaultExpectation.params) - } - } - - return mmEnd -} - -// Inspect accepts an inspector function that has same arguments as the Span.End -func (mmEnd *mOpentelemetrySpanMockEnd) Inspect(f func(options ...mm_trace.SpanEndOption)) *mOpentelemetrySpanMockEnd { - if mmEnd.mock.inspectFuncEnd != nil { - mmEnd.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.End") - } - - mmEnd.mock.inspectFuncEnd = f - - return mmEnd -} - -// Return sets up results that will be returned by Span.End -func (mmEnd *mOpentelemetrySpanMockEnd) Return() *OpentelemetrySpanMock { - if mmEnd.mock.funcEnd != nil { - mmEnd.mock.t.Fatalf("OpentelemetrySpanMock.End mock is already set by Set") - } - - if mmEnd.defaultExpectation == nil { - mmEnd.defaultExpectation = &OpentelemetrySpanMockEndExpectation{mock: mmEnd.mock} - } - - return mmEnd.mock -} - -// Set uses given function f to mock the Span.End method -func (mmEnd *mOpentelemetrySpanMockEnd) Set(f func(options ...mm_trace.SpanEndOption)) *OpentelemetrySpanMock { - if mmEnd.defaultExpectation != nil { - mmEnd.mock.t.Fatalf("Default expectation is already set for the Span.End method") - } - - if len(mmEnd.expectations) > 0 { - mmEnd.mock.t.Fatalf("Some expectations are already set for the Span.End method") - } - - mmEnd.mock.funcEnd = f - return mmEnd.mock -} - -// End implements trace.Span -func (mmEnd *OpentelemetrySpanMock) End(options ...mm_trace.SpanEndOption) { - mm_atomic.AddUint64(&mmEnd.beforeEndCounter, 1) - defer mm_atomic.AddUint64(&mmEnd.afterEndCounter, 1) - - if mmEnd.inspectFuncEnd != nil { - mmEnd.inspectFuncEnd(options...) - } - - mm_params := &OpentelemetrySpanMockEndParams{options} - - // Record call args - mmEnd.EndMock.mutex.Lock() - mmEnd.EndMock.callArgs = append(mmEnd.EndMock.callArgs, mm_params) - mmEnd.EndMock.mutex.Unlock() - - for _, e := range mmEnd.EndMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return - } - } - - if mmEnd.EndMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmEnd.EndMock.defaultExpectation.Counter, 1) - mm_want := mmEnd.EndMock.defaultExpectation.params - mm_got := OpentelemetrySpanMockEndParams{options} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmEnd.t.Errorf("OpentelemetrySpanMock.End got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - return - - } - if mmEnd.funcEnd != nil { - mmEnd.funcEnd(options...) - return - } - mmEnd.t.Fatalf("Unexpected call to OpentelemetrySpanMock.End. %v", options) - -} - -// EndAfterCounter returns a count of finished OpentelemetrySpanMock.End invocations -func (mmEnd *OpentelemetrySpanMock) EndAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmEnd.afterEndCounter) -} - -// EndBeforeCounter returns a count of OpentelemetrySpanMock.End invocations -func (mmEnd *OpentelemetrySpanMock) EndBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmEnd.beforeEndCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetrySpanMock.End. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmEnd *mOpentelemetrySpanMockEnd) Calls() []*OpentelemetrySpanMockEndParams { - mmEnd.mutex.RLock() - - argCopy := make([]*OpentelemetrySpanMockEndParams, len(mmEnd.callArgs)) - copy(argCopy, mmEnd.callArgs) - - mmEnd.mutex.RUnlock() - - return argCopy -} - -// MinimockEndDone returns true if the count of the End invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockEndDone() bool { - for _, e := range m.EndMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.EndMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterEndCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcEnd != nil && mm_atomic.LoadUint64(&m.afterEndCounter) < 1 { - return false - } - return true -} - -// MinimockEndInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockEndInspect() { - for _, e := range m.EndMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetrySpanMock.End with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.EndMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterEndCounter) < 1 { - if m.EndMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetrySpanMock.End") - } else { - m.t.Errorf("Expected call to OpentelemetrySpanMock.End with params: %#v", *m.EndMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcEnd != nil && mm_atomic.LoadUint64(&m.afterEndCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.End") - } -} - -type mOpentelemetrySpanMockIsRecording struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockIsRecordingExpectation - expectations []*OpentelemetrySpanMockIsRecordingExpectation -} - -// OpentelemetrySpanMockIsRecordingExpectation specifies expectation struct of the Span.IsRecording -type OpentelemetrySpanMockIsRecordingExpectation struct { - mock *OpentelemetrySpanMock - - results *OpentelemetrySpanMockIsRecordingResults - Counter uint64 -} - -// OpentelemetrySpanMockIsRecordingResults contains results of the Span.IsRecording -type OpentelemetrySpanMockIsRecordingResults struct { - b1 bool -} - -// Expect sets up expected params for Span.IsRecording -func (mmIsRecording *mOpentelemetrySpanMockIsRecording) Expect() *mOpentelemetrySpanMockIsRecording { - if mmIsRecording.mock.funcIsRecording != nil { - mmIsRecording.mock.t.Fatalf("OpentelemetrySpanMock.IsRecording mock is already set by Set") - } - - if mmIsRecording.defaultExpectation == nil { - mmIsRecording.defaultExpectation = &OpentelemetrySpanMockIsRecordingExpectation{} - } - - return mmIsRecording -} - -// Inspect accepts an inspector function that has same arguments as the Span.IsRecording -func (mmIsRecording *mOpentelemetrySpanMockIsRecording) Inspect(f func()) *mOpentelemetrySpanMockIsRecording { - if mmIsRecording.mock.inspectFuncIsRecording != nil { - mmIsRecording.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.IsRecording") - } - - mmIsRecording.mock.inspectFuncIsRecording = f - - return mmIsRecording -} - -// Return sets up results that will be returned by Span.IsRecording -func (mmIsRecording *mOpentelemetrySpanMockIsRecording) Return(b1 bool) *OpentelemetrySpanMock { - if mmIsRecording.mock.funcIsRecording != nil { - mmIsRecording.mock.t.Fatalf("OpentelemetrySpanMock.IsRecording mock is already set by Set") - } - - if mmIsRecording.defaultExpectation == nil { - mmIsRecording.defaultExpectation = &OpentelemetrySpanMockIsRecordingExpectation{mock: mmIsRecording.mock} - } - mmIsRecording.defaultExpectation.results = &OpentelemetrySpanMockIsRecordingResults{b1} - return mmIsRecording.mock -} - -// Set uses given function f to mock the Span.IsRecording method -func (mmIsRecording *mOpentelemetrySpanMockIsRecording) Set(f func() (b1 bool)) *OpentelemetrySpanMock { - if mmIsRecording.defaultExpectation != nil { - mmIsRecording.mock.t.Fatalf("Default expectation is already set for the Span.IsRecording method") - } - - if len(mmIsRecording.expectations) > 0 { - mmIsRecording.mock.t.Fatalf("Some expectations are already set for the Span.IsRecording method") - } - - mmIsRecording.mock.funcIsRecording = f - return mmIsRecording.mock -} - -// IsRecording implements trace.Span -func (mmIsRecording *OpentelemetrySpanMock) IsRecording() (b1 bool) { - mm_atomic.AddUint64(&mmIsRecording.beforeIsRecordingCounter, 1) - defer mm_atomic.AddUint64(&mmIsRecording.afterIsRecordingCounter, 1) - - if mmIsRecording.inspectFuncIsRecording != nil { - mmIsRecording.inspectFuncIsRecording() - } - - if mmIsRecording.IsRecordingMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmIsRecording.IsRecordingMock.defaultExpectation.Counter, 1) - - mm_results := mmIsRecording.IsRecordingMock.defaultExpectation.results - if mm_results == nil { - mmIsRecording.t.Fatal("No results are set for the OpentelemetrySpanMock.IsRecording") - } - return (*mm_results).b1 - } - if mmIsRecording.funcIsRecording != nil { - return mmIsRecording.funcIsRecording() - } - mmIsRecording.t.Fatalf("Unexpected call to OpentelemetrySpanMock.IsRecording.") - return -} - -// IsRecordingAfterCounter returns a count of finished OpentelemetrySpanMock.IsRecording invocations -func (mmIsRecording *OpentelemetrySpanMock) IsRecordingAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmIsRecording.afterIsRecordingCounter) -} - -// IsRecordingBeforeCounter returns a count of OpentelemetrySpanMock.IsRecording invocations -func (mmIsRecording *OpentelemetrySpanMock) IsRecordingBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmIsRecording.beforeIsRecordingCounter) -} - -// MinimockIsRecordingDone returns true if the count of the IsRecording invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockIsRecordingDone() bool { - for _, e := range m.IsRecordingMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.IsRecordingMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterIsRecordingCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcIsRecording != nil && mm_atomic.LoadUint64(&m.afterIsRecordingCounter) < 1 { - return false - } - return true -} - -// MinimockIsRecordingInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockIsRecordingInspect() { - for _, e := range m.IsRecordingMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.IsRecording") - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.IsRecordingMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterIsRecordingCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.IsRecording") - } - // if func was set then invocations count should be greater than zero - if m.funcIsRecording != nil && mm_atomic.LoadUint64(&m.afterIsRecordingCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.IsRecording") - } -} - -type mOpentelemetrySpanMockRecordError struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockRecordErrorExpectation - expectations []*OpentelemetrySpanMockRecordErrorExpectation - - callArgs []*OpentelemetrySpanMockRecordErrorParams - mutex sync.RWMutex -} - -// OpentelemetrySpanMockRecordErrorExpectation specifies expectation struct of the Span.RecordError -type OpentelemetrySpanMockRecordErrorExpectation struct { - mock *OpentelemetrySpanMock - params *OpentelemetrySpanMockRecordErrorParams - - Counter uint64 -} - -// OpentelemetrySpanMockRecordErrorParams contains parameters of the Span.RecordError -type OpentelemetrySpanMockRecordErrorParams struct { - err error - options []mm_trace.EventOption -} - -// Expect sets up expected params for Span.RecordError -func (mmRecordError *mOpentelemetrySpanMockRecordError) Expect(err error, options ...mm_trace.EventOption) *mOpentelemetrySpanMockRecordError { - if mmRecordError.mock.funcRecordError != nil { - mmRecordError.mock.t.Fatalf("OpentelemetrySpanMock.RecordError mock is already set by Set") - } - - if mmRecordError.defaultExpectation == nil { - mmRecordError.defaultExpectation = &OpentelemetrySpanMockRecordErrorExpectation{} - } - - mmRecordError.defaultExpectation.params = &OpentelemetrySpanMockRecordErrorParams{err, options} - for _, e := range mmRecordError.expectations { - if minimock.Equal(e.params, mmRecordError.defaultExpectation.params) { - mmRecordError.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmRecordError.defaultExpectation.params) - } - } - - return mmRecordError -} - -// Inspect accepts an inspector function that has same arguments as the Span.RecordError -func (mmRecordError *mOpentelemetrySpanMockRecordError) Inspect(f func(err error, options ...mm_trace.EventOption)) *mOpentelemetrySpanMockRecordError { - if mmRecordError.mock.inspectFuncRecordError != nil { - mmRecordError.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.RecordError") - } - - mmRecordError.mock.inspectFuncRecordError = f - - return mmRecordError -} - -// Return sets up results that will be returned by Span.RecordError -func (mmRecordError *mOpentelemetrySpanMockRecordError) Return() *OpentelemetrySpanMock { - if mmRecordError.mock.funcRecordError != nil { - mmRecordError.mock.t.Fatalf("OpentelemetrySpanMock.RecordError mock is already set by Set") - } - - if mmRecordError.defaultExpectation == nil { - mmRecordError.defaultExpectation = &OpentelemetrySpanMockRecordErrorExpectation{mock: mmRecordError.mock} - } - - return mmRecordError.mock -} - -// Set uses given function f to mock the Span.RecordError method -func (mmRecordError *mOpentelemetrySpanMockRecordError) Set(f func(err error, options ...mm_trace.EventOption)) *OpentelemetrySpanMock { - if mmRecordError.defaultExpectation != nil { - mmRecordError.mock.t.Fatalf("Default expectation is already set for the Span.RecordError method") - } - - if len(mmRecordError.expectations) > 0 { - mmRecordError.mock.t.Fatalf("Some expectations are already set for the Span.RecordError method") - } - - mmRecordError.mock.funcRecordError = f - return mmRecordError.mock -} - -// RecordError implements trace.Span -func (mmRecordError *OpentelemetrySpanMock) RecordError(err error, options ...mm_trace.EventOption) { - mm_atomic.AddUint64(&mmRecordError.beforeRecordErrorCounter, 1) - defer mm_atomic.AddUint64(&mmRecordError.afterRecordErrorCounter, 1) - - if mmRecordError.inspectFuncRecordError != nil { - mmRecordError.inspectFuncRecordError(err, options...) - } - - mm_params := &OpentelemetrySpanMockRecordErrorParams{err, options} - - // Record call args - mmRecordError.RecordErrorMock.mutex.Lock() - mmRecordError.RecordErrorMock.callArgs = append(mmRecordError.RecordErrorMock.callArgs, mm_params) - mmRecordError.RecordErrorMock.mutex.Unlock() - - for _, e := range mmRecordError.RecordErrorMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return - } - } - - if mmRecordError.RecordErrorMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmRecordError.RecordErrorMock.defaultExpectation.Counter, 1) - mm_want := mmRecordError.RecordErrorMock.defaultExpectation.params - mm_got := OpentelemetrySpanMockRecordErrorParams{err, options} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmRecordError.t.Errorf("OpentelemetrySpanMock.RecordError got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - return - - } - if mmRecordError.funcRecordError != nil { - mmRecordError.funcRecordError(err, options...) - return - } - mmRecordError.t.Fatalf("Unexpected call to OpentelemetrySpanMock.RecordError. %v %v", err, options) - -} - -// RecordErrorAfterCounter returns a count of finished OpentelemetrySpanMock.RecordError invocations -func (mmRecordError *OpentelemetrySpanMock) RecordErrorAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmRecordError.afterRecordErrorCounter) -} - -// RecordErrorBeforeCounter returns a count of OpentelemetrySpanMock.RecordError invocations -func (mmRecordError *OpentelemetrySpanMock) RecordErrorBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmRecordError.beforeRecordErrorCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetrySpanMock.RecordError. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmRecordError *mOpentelemetrySpanMockRecordError) Calls() []*OpentelemetrySpanMockRecordErrorParams { - mmRecordError.mutex.RLock() - - argCopy := make([]*OpentelemetrySpanMockRecordErrorParams, len(mmRecordError.callArgs)) - copy(argCopy, mmRecordError.callArgs) - - mmRecordError.mutex.RUnlock() - - return argCopy -} - -// MinimockRecordErrorDone returns true if the count of the RecordError invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockRecordErrorDone() bool { - for _, e := range m.RecordErrorMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.RecordErrorMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterRecordErrorCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcRecordError != nil && mm_atomic.LoadUint64(&m.afterRecordErrorCounter) < 1 { - return false - } - return true -} - -// MinimockRecordErrorInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockRecordErrorInspect() { - for _, e := range m.RecordErrorMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetrySpanMock.RecordError with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.RecordErrorMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterRecordErrorCounter) < 1 { - if m.RecordErrorMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetrySpanMock.RecordError") - } else { - m.t.Errorf("Expected call to OpentelemetrySpanMock.RecordError with params: %#v", *m.RecordErrorMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcRecordError != nil && mm_atomic.LoadUint64(&m.afterRecordErrorCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.RecordError") - } -} - -type mOpentelemetrySpanMockSetAttributes struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockSetAttributesExpectation - expectations []*OpentelemetrySpanMockSetAttributesExpectation - - callArgs []*OpentelemetrySpanMockSetAttributesParams - mutex sync.RWMutex -} - -// OpentelemetrySpanMockSetAttributesExpectation specifies expectation struct of the Span.SetAttributes -type OpentelemetrySpanMockSetAttributesExpectation struct { - mock *OpentelemetrySpanMock - params *OpentelemetrySpanMockSetAttributesParams - - Counter uint64 -} - -// OpentelemetrySpanMockSetAttributesParams contains parameters of the Span.SetAttributes -type OpentelemetrySpanMockSetAttributesParams struct { - kv []attribute.KeyValue -} - -// Expect sets up expected params for Span.SetAttributes -func (mmSetAttributes *mOpentelemetrySpanMockSetAttributes) Expect(kv ...attribute.KeyValue) *mOpentelemetrySpanMockSetAttributes { - if mmSetAttributes.mock.funcSetAttributes != nil { - mmSetAttributes.mock.t.Fatalf("OpentelemetrySpanMock.SetAttributes mock is already set by Set") - } - - if mmSetAttributes.defaultExpectation == nil { - mmSetAttributes.defaultExpectation = &OpentelemetrySpanMockSetAttributesExpectation{} - } - - mmSetAttributes.defaultExpectation.params = &OpentelemetrySpanMockSetAttributesParams{kv} - for _, e := range mmSetAttributes.expectations { - if minimock.Equal(e.params, mmSetAttributes.defaultExpectation.params) { - mmSetAttributes.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmSetAttributes.defaultExpectation.params) - } - } - - return mmSetAttributes -} - -// Inspect accepts an inspector function that has same arguments as the Span.SetAttributes -func (mmSetAttributes *mOpentelemetrySpanMockSetAttributes) Inspect(f func(kv ...attribute.KeyValue)) *mOpentelemetrySpanMockSetAttributes { - if mmSetAttributes.mock.inspectFuncSetAttributes != nil { - mmSetAttributes.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.SetAttributes") - } - - mmSetAttributes.mock.inspectFuncSetAttributes = f - - return mmSetAttributes -} - -// Return sets up results that will be returned by Span.SetAttributes -func (mmSetAttributes *mOpentelemetrySpanMockSetAttributes) Return() *OpentelemetrySpanMock { - if mmSetAttributes.mock.funcSetAttributes != nil { - mmSetAttributes.mock.t.Fatalf("OpentelemetrySpanMock.SetAttributes mock is already set by Set") - } - - if mmSetAttributes.defaultExpectation == nil { - mmSetAttributes.defaultExpectation = &OpentelemetrySpanMockSetAttributesExpectation{mock: mmSetAttributes.mock} - } - - return mmSetAttributes.mock -} - -// Set uses given function f to mock the Span.SetAttributes method -func (mmSetAttributes *mOpentelemetrySpanMockSetAttributes) Set(f func(kv ...attribute.KeyValue)) *OpentelemetrySpanMock { - if mmSetAttributes.defaultExpectation != nil { - mmSetAttributes.mock.t.Fatalf("Default expectation is already set for the Span.SetAttributes method") - } - - if len(mmSetAttributes.expectations) > 0 { - mmSetAttributes.mock.t.Fatalf("Some expectations are already set for the Span.SetAttributes method") - } - - mmSetAttributes.mock.funcSetAttributes = f - return mmSetAttributes.mock -} - -// SetAttributes implements trace.Span -func (mmSetAttributes *OpentelemetrySpanMock) SetAttributes(kv ...attribute.KeyValue) { - mm_atomic.AddUint64(&mmSetAttributes.beforeSetAttributesCounter, 1) - defer mm_atomic.AddUint64(&mmSetAttributes.afterSetAttributesCounter, 1) - - if mmSetAttributes.inspectFuncSetAttributes != nil { - mmSetAttributes.inspectFuncSetAttributes(kv...) - } - - mm_params := &OpentelemetrySpanMockSetAttributesParams{kv} - - // Record call args - mmSetAttributes.SetAttributesMock.mutex.Lock() - mmSetAttributes.SetAttributesMock.callArgs = append(mmSetAttributes.SetAttributesMock.callArgs, mm_params) - mmSetAttributes.SetAttributesMock.mutex.Unlock() - - for _, e := range mmSetAttributes.SetAttributesMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return - } - } - - if mmSetAttributes.SetAttributesMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmSetAttributes.SetAttributesMock.defaultExpectation.Counter, 1) - mm_want := mmSetAttributes.SetAttributesMock.defaultExpectation.params - mm_got := OpentelemetrySpanMockSetAttributesParams{kv} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmSetAttributes.t.Errorf("OpentelemetrySpanMock.SetAttributes got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - return - - } - if mmSetAttributes.funcSetAttributes != nil { - mmSetAttributes.funcSetAttributes(kv...) - return - } - mmSetAttributes.t.Fatalf("Unexpected call to OpentelemetrySpanMock.SetAttributes. %v", kv) - -} - -// SetAttributesAfterCounter returns a count of finished OpentelemetrySpanMock.SetAttributes invocations -func (mmSetAttributes *OpentelemetrySpanMock) SetAttributesAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmSetAttributes.afterSetAttributesCounter) -} - -// SetAttributesBeforeCounter returns a count of OpentelemetrySpanMock.SetAttributes invocations -func (mmSetAttributes *OpentelemetrySpanMock) SetAttributesBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmSetAttributes.beforeSetAttributesCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetrySpanMock.SetAttributes. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmSetAttributes *mOpentelemetrySpanMockSetAttributes) Calls() []*OpentelemetrySpanMockSetAttributesParams { - mmSetAttributes.mutex.RLock() - - argCopy := make([]*OpentelemetrySpanMockSetAttributesParams, len(mmSetAttributes.callArgs)) - copy(argCopy, mmSetAttributes.callArgs) - - mmSetAttributes.mutex.RUnlock() - - return argCopy -} - -// MinimockSetAttributesDone returns true if the count of the SetAttributes invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockSetAttributesDone() bool { - for _, e := range m.SetAttributesMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SetAttributesMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSetAttributesCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcSetAttributes != nil && mm_atomic.LoadUint64(&m.afterSetAttributesCounter) < 1 { - return false - } - return true -} - -// MinimockSetAttributesInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockSetAttributesInspect() { - for _, e := range m.SetAttributesMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetrySpanMock.SetAttributes with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SetAttributesMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSetAttributesCounter) < 1 { - if m.SetAttributesMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetrySpanMock.SetAttributes") - } else { - m.t.Errorf("Expected call to OpentelemetrySpanMock.SetAttributes with params: %#v", *m.SetAttributesMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcSetAttributes != nil && mm_atomic.LoadUint64(&m.afterSetAttributesCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.SetAttributes") - } -} - -type mOpentelemetrySpanMockSetName struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockSetNameExpectation - expectations []*OpentelemetrySpanMockSetNameExpectation - - callArgs []*OpentelemetrySpanMockSetNameParams - mutex sync.RWMutex -} - -// OpentelemetrySpanMockSetNameExpectation specifies expectation struct of the Span.SetName -type OpentelemetrySpanMockSetNameExpectation struct { - mock *OpentelemetrySpanMock - params *OpentelemetrySpanMockSetNameParams - - Counter uint64 -} - -// OpentelemetrySpanMockSetNameParams contains parameters of the Span.SetName -type OpentelemetrySpanMockSetNameParams struct { - name string -} - -// Expect sets up expected params for Span.SetName -func (mmSetName *mOpentelemetrySpanMockSetName) Expect(name string) *mOpentelemetrySpanMockSetName { - if mmSetName.mock.funcSetName != nil { - mmSetName.mock.t.Fatalf("OpentelemetrySpanMock.SetName mock is already set by Set") - } - - if mmSetName.defaultExpectation == nil { - mmSetName.defaultExpectation = &OpentelemetrySpanMockSetNameExpectation{} - } - - mmSetName.defaultExpectation.params = &OpentelemetrySpanMockSetNameParams{name} - for _, e := range mmSetName.expectations { - if minimock.Equal(e.params, mmSetName.defaultExpectation.params) { - mmSetName.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmSetName.defaultExpectation.params) - } - } - - return mmSetName -} - -// Inspect accepts an inspector function that has same arguments as the Span.SetName -func (mmSetName *mOpentelemetrySpanMockSetName) Inspect(f func(name string)) *mOpentelemetrySpanMockSetName { - if mmSetName.mock.inspectFuncSetName != nil { - mmSetName.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.SetName") - } - - mmSetName.mock.inspectFuncSetName = f - - return mmSetName -} - -// Return sets up results that will be returned by Span.SetName -func (mmSetName *mOpentelemetrySpanMockSetName) Return() *OpentelemetrySpanMock { - if mmSetName.mock.funcSetName != nil { - mmSetName.mock.t.Fatalf("OpentelemetrySpanMock.SetName mock is already set by Set") - } - - if mmSetName.defaultExpectation == nil { - mmSetName.defaultExpectation = &OpentelemetrySpanMockSetNameExpectation{mock: mmSetName.mock} - } - - return mmSetName.mock -} - -// Set uses given function f to mock the Span.SetName method -func (mmSetName *mOpentelemetrySpanMockSetName) Set(f func(name string)) *OpentelemetrySpanMock { - if mmSetName.defaultExpectation != nil { - mmSetName.mock.t.Fatalf("Default expectation is already set for the Span.SetName method") - } - - if len(mmSetName.expectations) > 0 { - mmSetName.mock.t.Fatalf("Some expectations are already set for the Span.SetName method") - } - - mmSetName.mock.funcSetName = f - return mmSetName.mock -} - -// SetName implements trace.Span -func (mmSetName *OpentelemetrySpanMock) SetName(name string) { - mm_atomic.AddUint64(&mmSetName.beforeSetNameCounter, 1) - defer mm_atomic.AddUint64(&mmSetName.afterSetNameCounter, 1) - - if mmSetName.inspectFuncSetName != nil { - mmSetName.inspectFuncSetName(name) - } - - mm_params := &OpentelemetrySpanMockSetNameParams{name} - - // Record call args - mmSetName.SetNameMock.mutex.Lock() - mmSetName.SetNameMock.callArgs = append(mmSetName.SetNameMock.callArgs, mm_params) - mmSetName.SetNameMock.mutex.Unlock() - - for _, e := range mmSetName.SetNameMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return - } - } - - if mmSetName.SetNameMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmSetName.SetNameMock.defaultExpectation.Counter, 1) - mm_want := mmSetName.SetNameMock.defaultExpectation.params - mm_got := OpentelemetrySpanMockSetNameParams{name} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmSetName.t.Errorf("OpentelemetrySpanMock.SetName got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - return - - } - if mmSetName.funcSetName != nil { - mmSetName.funcSetName(name) - return - } - mmSetName.t.Fatalf("Unexpected call to OpentelemetrySpanMock.SetName. %v", name) - -} - -// SetNameAfterCounter returns a count of finished OpentelemetrySpanMock.SetName invocations -func (mmSetName *OpentelemetrySpanMock) SetNameAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmSetName.afterSetNameCounter) -} - -// SetNameBeforeCounter returns a count of OpentelemetrySpanMock.SetName invocations -func (mmSetName *OpentelemetrySpanMock) SetNameBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmSetName.beforeSetNameCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetrySpanMock.SetName. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmSetName *mOpentelemetrySpanMockSetName) Calls() []*OpentelemetrySpanMockSetNameParams { - mmSetName.mutex.RLock() - - argCopy := make([]*OpentelemetrySpanMockSetNameParams, len(mmSetName.callArgs)) - copy(argCopy, mmSetName.callArgs) - - mmSetName.mutex.RUnlock() - - return argCopy -} - -// MinimockSetNameDone returns true if the count of the SetName invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockSetNameDone() bool { - for _, e := range m.SetNameMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SetNameMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSetNameCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcSetName != nil && mm_atomic.LoadUint64(&m.afterSetNameCounter) < 1 { - return false - } - return true -} - -// MinimockSetNameInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockSetNameInspect() { - for _, e := range m.SetNameMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetrySpanMock.SetName with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SetNameMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSetNameCounter) < 1 { - if m.SetNameMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetrySpanMock.SetName") - } else { - m.t.Errorf("Expected call to OpentelemetrySpanMock.SetName with params: %#v", *m.SetNameMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcSetName != nil && mm_atomic.LoadUint64(&m.afterSetNameCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.SetName") - } -} - -type mOpentelemetrySpanMockSetStatus struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockSetStatusExpectation - expectations []*OpentelemetrySpanMockSetStatusExpectation - - callArgs []*OpentelemetrySpanMockSetStatusParams - mutex sync.RWMutex -} - -// OpentelemetrySpanMockSetStatusExpectation specifies expectation struct of the Span.SetStatus -type OpentelemetrySpanMockSetStatusExpectation struct { - mock *OpentelemetrySpanMock - params *OpentelemetrySpanMockSetStatusParams - - Counter uint64 -} - -// OpentelemetrySpanMockSetStatusParams contains parameters of the Span.SetStatus -type OpentelemetrySpanMockSetStatusParams struct { - code codes.Code - description string -} - -// Expect sets up expected params for Span.SetStatus -func (mmSetStatus *mOpentelemetrySpanMockSetStatus) Expect(code codes.Code, description string) *mOpentelemetrySpanMockSetStatus { - if mmSetStatus.mock.funcSetStatus != nil { - mmSetStatus.mock.t.Fatalf("OpentelemetrySpanMock.SetStatus mock is already set by Set") - } - - if mmSetStatus.defaultExpectation == nil { - mmSetStatus.defaultExpectation = &OpentelemetrySpanMockSetStatusExpectation{} - } - - mmSetStatus.defaultExpectation.params = &OpentelemetrySpanMockSetStatusParams{code, description} - for _, e := range mmSetStatus.expectations { - if minimock.Equal(e.params, mmSetStatus.defaultExpectation.params) { - mmSetStatus.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmSetStatus.defaultExpectation.params) - } - } - - return mmSetStatus -} - -// Inspect accepts an inspector function that has same arguments as the Span.SetStatus -func (mmSetStatus *mOpentelemetrySpanMockSetStatus) Inspect(f func(code codes.Code, description string)) *mOpentelemetrySpanMockSetStatus { - if mmSetStatus.mock.inspectFuncSetStatus != nil { - mmSetStatus.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.SetStatus") - } - - mmSetStatus.mock.inspectFuncSetStatus = f - - return mmSetStatus -} - -// Return sets up results that will be returned by Span.SetStatus -func (mmSetStatus *mOpentelemetrySpanMockSetStatus) Return() *OpentelemetrySpanMock { - if mmSetStatus.mock.funcSetStatus != nil { - mmSetStatus.mock.t.Fatalf("OpentelemetrySpanMock.SetStatus mock is already set by Set") - } - - if mmSetStatus.defaultExpectation == nil { - mmSetStatus.defaultExpectation = &OpentelemetrySpanMockSetStatusExpectation{mock: mmSetStatus.mock} - } - - return mmSetStatus.mock -} - -// Set uses given function f to mock the Span.SetStatus method -func (mmSetStatus *mOpentelemetrySpanMockSetStatus) Set(f func(code codes.Code, description string)) *OpentelemetrySpanMock { - if mmSetStatus.defaultExpectation != nil { - mmSetStatus.mock.t.Fatalf("Default expectation is already set for the Span.SetStatus method") - } - - if len(mmSetStatus.expectations) > 0 { - mmSetStatus.mock.t.Fatalf("Some expectations are already set for the Span.SetStatus method") - } - - mmSetStatus.mock.funcSetStatus = f - return mmSetStatus.mock -} - -// SetStatus implements trace.Span -func (mmSetStatus *OpentelemetrySpanMock) SetStatus(code codes.Code, description string) { - mm_atomic.AddUint64(&mmSetStatus.beforeSetStatusCounter, 1) - defer mm_atomic.AddUint64(&mmSetStatus.afterSetStatusCounter, 1) - - if mmSetStatus.inspectFuncSetStatus != nil { - mmSetStatus.inspectFuncSetStatus(code, description) - } - - mm_params := &OpentelemetrySpanMockSetStatusParams{code, description} - - // Record call args - mmSetStatus.SetStatusMock.mutex.Lock() - mmSetStatus.SetStatusMock.callArgs = append(mmSetStatus.SetStatusMock.callArgs, mm_params) - mmSetStatus.SetStatusMock.mutex.Unlock() - - for _, e := range mmSetStatus.SetStatusMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return - } - } - - if mmSetStatus.SetStatusMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmSetStatus.SetStatusMock.defaultExpectation.Counter, 1) - mm_want := mmSetStatus.SetStatusMock.defaultExpectation.params - mm_got := OpentelemetrySpanMockSetStatusParams{code, description} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmSetStatus.t.Errorf("OpentelemetrySpanMock.SetStatus got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - return - - } - if mmSetStatus.funcSetStatus != nil { - mmSetStatus.funcSetStatus(code, description) - return - } - mmSetStatus.t.Fatalf("Unexpected call to OpentelemetrySpanMock.SetStatus. %v %v", code, description) - -} - -// SetStatusAfterCounter returns a count of finished OpentelemetrySpanMock.SetStatus invocations -func (mmSetStatus *OpentelemetrySpanMock) SetStatusAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmSetStatus.afterSetStatusCounter) -} - -// SetStatusBeforeCounter returns a count of OpentelemetrySpanMock.SetStatus invocations -func (mmSetStatus *OpentelemetrySpanMock) SetStatusBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmSetStatus.beforeSetStatusCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetrySpanMock.SetStatus. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmSetStatus *mOpentelemetrySpanMockSetStatus) Calls() []*OpentelemetrySpanMockSetStatusParams { - mmSetStatus.mutex.RLock() - - argCopy := make([]*OpentelemetrySpanMockSetStatusParams, len(mmSetStatus.callArgs)) - copy(argCopy, mmSetStatus.callArgs) - - mmSetStatus.mutex.RUnlock() - - return argCopy -} - -// MinimockSetStatusDone returns true if the count of the SetStatus invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockSetStatusDone() bool { - for _, e := range m.SetStatusMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SetStatusMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSetStatusCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcSetStatus != nil && mm_atomic.LoadUint64(&m.afterSetStatusCounter) < 1 { - return false - } - return true -} - -// MinimockSetStatusInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockSetStatusInspect() { - for _, e := range m.SetStatusMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetrySpanMock.SetStatus with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SetStatusMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSetStatusCounter) < 1 { - if m.SetStatusMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetrySpanMock.SetStatus") - } else { - m.t.Errorf("Expected call to OpentelemetrySpanMock.SetStatus with params: %#v", *m.SetStatusMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcSetStatus != nil && mm_atomic.LoadUint64(&m.afterSetStatusCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.SetStatus") - } -} - -type mOpentelemetrySpanMockSpanContext struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockSpanContextExpectation - expectations []*OpentelemetrySpanMockSpanContextExpectation -} - -// OpentelemetrySpanMockSpanContextExpectation specifies expectation struct of the Span.SpanContext -type OpentelemetrySpanMockSpanContextExpectation struct { - mock *OpentelemetrySpanMock - - results *OpentelemetrySpanMockSpanContextResults - Counter uint64 -} - -// OpentelemetrySpanMockSpanContextResults contains results of the Span.SpanContext -type OpentelemetrySpanMockSpanContextResults struct { - s1 mm_trace.SpanContext -} - -// Expect sets up expected params for Span.SpanContext -func (mmSpanContext *mOpentelemetrySpanMockSpanContext) Expect() *mOpentelemetrySpanMockSpanContext { - if mmSpanContext.mock.funcSpanContext != nil { - mmSpanContext.mock.t.Fatalf("OpentelemetrySpanMock.SpanContext mock is already set by Set") - } - - if mmSpanContext.defaultExpectation == nil { - mmSpanContext.defaultExpectation = &OpentelemetrySpanMockSpanContextExpectation{} - } - - return mmSpanContext -} - -// Inspect accepts an inspector function that has same arguments as the Span.SpanContext -func (mmSpanContext *mOpentelemetrySpanMockSpanContext) Inspect(f func()) *mOpentelemetrySpanMockSpanContext { - if mmSpanContext.mock.inspectFuncSpanContext != nil { - mmSpanContext.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.SpanContext") - } - - mmSpanContext.mock.inspectFuncSpanContext = f - - return mmSpanContext -} - -// Return sets up results that will be returned by Span.SpanContext -func (mmSpanContext *mOpentelemetrySpanMockSpanContext) Return(s1 mm_trace.SpanContext) *OpentelemetrySpanMock { - if mmSpanContext.mock.funcSpanContext != nil { - mmSpanContext.mock.t.Fatalf("OpentelemetrySpanMock.SpanContext mock is already set by Set") - } - - if mmSpanContext.defaultExpectation == nil { - mmSpanContext.defaultExpectation = &OpentelemetrySpanMockSpanContextExpectation{mock: mmSpanContext.mock} - } - mmSpanContext.defaultExpectation.results = &OpentelemetrySpanMockSpanContextResults{s1} - return mmSpanContext.mock -} - -// Set uses given function f to mock the Span.SpanContext method -func (mmSpanContext *mOpentelemetrySpanMockSpanContext) Set(f func() (s1 mm_trace.SpanContext)) *OpentelemetrySpanMock { - if mmSpanContext.defaultExpectation != nil { - mmSpanContext.mock.t.Fatalf("Default expectation is already set for the Span.SpanContext method") - } - - if len(mmSpanContext.expectations) > 0 { - mmSpanContext.mock.t.Fatalf("Some expectations are already set for the Span.SpanContext method") - } - - mmSpanContext.mock.funcSpanContext = f - return mmSpanContext.mock -} - -// SpanContext implements trace.Span -func (mmSpanContext *OpentelemetrySpanMock) SpanContext() (s1 mm_trace.SpanContext) { - mm_atomic.AddUint64(&mmSpanContext.beforeSpanContextCounter, 1) - defer mm_atomic.AddUint64(&mmSpanContext.afterSpanContextCounter, 1) - - if mmSpanContext.inspectFuncSpanContext != nil { - mmSpanContext.inspectFuncSpanContext() - } - - if mmSpanContext.SpanContextMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmSpanContext.SpanContextMock.defaultExpectation.Counter, 1) - - mm_results := mmSpanContext.SpanContextMock.defaultExpectation.results - if mm_results == nil { - mmSpanContext.t.Fatal("No results are set for the OpentelemetrySpanMock.SpanContext") - } - return (*mm_results).s1 - } - if mmSpanContext.funcSpanContext != nil { - return mmSpanContext.funcSpanContext() - } - mmSpanContext.t.Fatalf("Unexpected call to OpentelemetrySpanMock.SpanContext.") - return -} - -// SpanContextAfterCounter returns a count of finished OpentelemetrySpanMock.SpanContext invocations -func (mmSpanContext *OpentelemetrySpanMock) SpanContextAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmSpanContext.afterSpanContextCounter) -} - -// SpanContextBeforeCounter returns a count of OpentelemetrySpanMock.SpanContext invocations -func (mmSpanContext *OpentelemetrySpanMock) SpanContextBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmSpanContext.beforeSpanContextCounter) -} - -// MinimockSpanContextDone returns true if the count of the SpanContext invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockSpanContextDone() bool { - for _, e := range m.SpanContextMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SpanContextMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSpanContextCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcSpanContext != nil && mm_atomic.LoadUint64(&m.afterSpanContextCounter) < 1 { - return false - } - return true -} - -// MinimockSpanContextInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockSpanContextInspect() { - for _, e := range m.SpanContextMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.SpanContext") - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.SpanContextMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterSpanContextCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.SpanContext") - } - // if func was set then invocations count should be greater than zero - if m.funcSpanContext != nil && mm_atomic.LoadUint64(&m.afterSpanContextCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.SpanContext") - } -} - -type mOpentelemetrySpanMockTracerProvider struct { - mock *OpentelemetrySpanMock - defaultExpectation *OpentelemetrySpanMockTracerProviderExpectation - expectations []*OpentelemetrySpanMockTracerProviderExpectation -} - -// OpentelemetrySpanMockTracerProviderExpectation specifies expectation struct of the Span.TracerProvider -type OpentelemetrySpanMockTracerProviderExpectation struct { - mock *OpentelemetrySpanMock - - results *OpentelemetrySpanMockTracerProviderResults - Counter uint64 -} - -// OpentelemetrySpanMockTracerProviderResults contains results of the Span.TracerProvider -type OpentelemetrySpanMockTracerProviderResults struct { - t1 mm_trace.TracerProvider -} - -// Expect sets up expected params for Span.TracerProvider -func (mmTracerProvider *mOpentelemetrySpanMockTracerProvider) Expect() *mOpentelemetrySpanMockTracerProvider { - if mmTracerProvider.mock.funcTracerProvider != nil { - mmTracerProvider.mock.t.Fatalf("OpentelemetrySpanMock.TracerProvider mock is already set by Set") - } - - if mmTracerProvider.defaultExpectation == nil { - mmTracerProvider.defaultExpectation = &OpentelemetrySpanMockTracerProviderExpectation{} - } - - return mmTracerProvider -} - -// Inspect accepts an inspector function that has same arguments as the Span.TracerProvider -func (mmTracerProvider *mOpentelemetrySpanMockTracerProvider) Inspect(f func()) *mOpentelemetrySpanMockTracerProvider { - if mmTracerProvider.mock.inspectFuncTracerProvider != nil { - mmTracerProvider.mock.t.Fatalf("Inspect function is already set for OpentelemetrySpanMock.TracerProvider") - } - - mmTracerProvider.mock.inspectFuncTracerProvider = f - - return mmTracerProvider -} - -// Return sets up results that will be returned by Span.TracerProvider -func (mmTracerProvider *mOpentelemetrySpanMockTracerProvider) Return(t1 mm_trace.TracerProvider) *OpentelemetrySpanMock { - if mmTracerProvider.mock.funcTracerProvider != nil { - mmTracerProvider.mock.t.Fatalf("OpentelemetrySpanMock.TracerProvider mock is already set by Set") - } - - if mmTracerProvider.defaultExpectation == nil { - mmTracerProvider.defaultExpectation = &OpentelemetrySpanMockTracerProviderExpectation{mock: mmTracerProvider.mock} - } - mmTracerProvider.defaultExpectation.results = &OpentelemetrySpanMockTracerProviderResults{t1} - return mmTracerProvider.mock -} - -// Set uses given function f to mock the Span.TracerProvider method -func (mmTracerProvider *mOpentelemetrySpanMockTracerProvider) Set(f func() (t1 mm_trace.TracerProvider)) *OpentelemetrySpanMock { - if mmTracerProvider.defaultExpectation != nil { - mmTracerProvider.mock.t.Fatalf("Default expectation is already set for the Span.TracerProvider method") - } - - if len(mmTracerProvider.expectations) > 0 { - mmTracerProvider.mock.t.Fatalf("Some expectations are already set for the Span.TracerProvider method") - } - - mmTracerProvider.mock.funcTracerProvider = f - return mmTracerProvider.mock -} - -// TracerProvider implements trace.Span -func (mmTracerProvider *OpentelemetrySpanMock) TracerProvider() (t1 mm_trace.TracerProvider) { - mm_atomic.AddUint64(&mmTracerProvider.beforeTracerProviderCounter, 1) - defer mm_atomic.AddUint64(&mmTracerProvider.afterTracerProviderCounter, 1) - - if mmTracerProvider.inspectFuncTracerProvider != nil { - mmTracerProvider.inspectFuncTracerProvider() - } - - if mmTracerProvider.TracerProviderMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmTracerProvider.TracerProviderMock.defaultExpectation.Counter, 1) - - mm_results := mmTracerProvider.TracerProviderMock.defaultExpectation.results - if mm_results == nil { - mmTracerProvider.t.Fatal("No results are set for the OpentelemetrySpanMock.TracerProvider") - } - return (*mm_results).t1 - } - if mmTracerProvider.funcTracerProvider != nil { - return mmTracerProvider.funcTracerProvider() - } - mmTracerProvider.t.Fatalf("Unexpected call to OpentelemetrySpanMock.TracerProvider.") - return -} - -// TracerProviderAfterCounter returns a count of finished OpentelemetrySpanMock.TracerProvider invocations -func (mmTracerProvider *OpentelemetrySpanMock) TracerProviderAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmTracerProvider.afterTracerProviderCounter) -} - -// TracerProviderBeforeCounter returns a count of OpentelemetrySpanMock.TracerProvider invocations -func (mmTracerProvider *OpentelemetrySpanMock) TracerProviderBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmTracerProvider.beforeTracerProviderCounter) -} - -// MinimockTracerProviderDone returns true if the count of the TracerProvider invocations corresponds -// the number of defined expectations -func (m *OpentelemetrySpanMock) MinimockTracerProviderDone() bool { - for _, e := range m.TracerProviderMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.TracerProviderMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterTracerProviderCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcTracerProvider != nil && mm_atomic.LoadUint64(&m.afterTracerProviderCounter) < 1 { - return false - } - return true -} - -// MinimockTracerProviderInspect logs each unmet expectation -func (m *OpentelemetrySpanMock) MinimockTracerProviderInspect() { - for _, e := range m.TracerProviderMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.TracerProvider") - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.TracerProviderMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterTracerProviderCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.TracerProvider") - } - // if func was set then invocations count should be greater than zero - if m.funcTracerProvider != nil && mm_atomic.LoadUint64(&m.afterTracerProviderCounter) < 1 { - m.t.Error("Expected call to OpentelemetrySpanMock.TracerProvider") - } -} - -// MinimockFinish checks that all mocked methods have been called the expected number of times -func (m *OpentelemetrySpanMock) MinimockFinish() { - if !m.minimockDone() { - m.MinimockAddEventInspect() - - m.MinimockEndInspect() - - m.MinimockIsRecordingInspect() - - m.MinimockRecordErrorInspect() - - m.MinimockSetAttributesInspect() - - m.MinimockSetNameInspect() - - m.MinimockSetStatusInspect() - - m.MinimockSpanContextInspect() - - m.MinimockTracerProviderInspect() - m.t.FailNow() - } -} - -// MinimockWait waits for all mocked methods to be called the expected number of times -func (m *OpentelemetrySpanMock) MinimockWait(timeout mm_time.Duration) { - timeoutCh := mm_time.After(timeout) - for { - if m.minimockDone() { - return - } - select { - case <-timeoutCh: - m.MinimockFinish() - return - case <-mm_time.After(10 * mm_time.Millisecond): - } - } -} - -func (m *OpentelemetrySpanMock) minimockDone() bool { - done := true - return done && - m.MinimockAddEventDone() && - m.MinimockEndDone() && - m.MinimockIsRecordingDone() && - m.MinimockRecordErrorDone() && - m.MinimockSetAttributesDone() && - m.MinimockSetNameDone() && - m.MinimockSetStatusDone() && - m.MinimockSpanContextDone() && - m.MinimockTracerProviderDone() -} diff --git a/templates_tests/tracer_opentelemery_mock_test.go b/templates_tests/tracer_opentelemery_mock_test.go deleted file mode 100644 index b9a750f..0000000 --- a/templates_tests/tracer_opentelemery_mock_test.go +++ /dev/null @@ -1,287 +0,0 @@ -package templatestests - -// Code generated by http://github.com/gojuno/minimock (dev). DO NOT EDIT. - -//go:generate minimock -i go.opentelemetry.io/otel/trace.Tracer -o ./tracer_opentelemery_mock_test.go -n OpentelemetryTracerMock - -import ( - "context" - "sync" - mm_atomic "sync/atomic" - mm_time "time" - - "github.com/gojuno/minimock/v3" - mm_trace "go.opentelemetry.io/otel/trace" -) - -// OpentelemetryTracerMock implements trace.Tracer -type OpentelemetryTracerMock struct { - t minimock.Tester - - funcStart func(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption) (c2 context.Context, s1 mm_trace.Span) - inspectFuncStart func(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption) - afterStartCounter uint64 - beforeStartCounter uint64 - StartMock mOpentelemetryTracerMockStart -} - -// NewOpentelemetryTracerMock returns a mock for trace.Tracer -func NewOpentelemetryTracerMock(t minimock.Tester) *OpentelemetryTracerMock { - m := &OpentelemetryTracerMock{t: t} - if controller, ok := t.(minimock.MockController); ok { - controller.RegisterMocker(m) - } - - m.StartMock = mOpentelemetryTracerMockStart{mock: m} - m.StartMock.callArgs = []*OpentelemetryTracerMockStartParams{} - - return m -} - -type mOpentelemetryTracerMockStart struct { - mock *OpentelemetryTracerMock - defaultExpectation *OpentelemetryTracerMockStartExpectation - expectations []*OpentelemetryTracerMockStartExpectation - - callArgs []*OpentelemetryTracerMockStartParams - mutex sync.RWMutex -} - -// OpentelemetryTracerMockStartExpectation specifies expectation struct of the Tracer.Start -type OpentelemetryTracerMockStartExpectation struct { - mock *OpentelemetryTracerMock - params *OpentelemetryTracerMockStartParams - results *OpentelemetryTracerMockStartResults - Counter uint64 -} - -// OpentelemetryTracerMockStartParams contains parameters of the Tracer.Start -type OpentelemetryTracerMockStartParams struct { - ctx context.Context - spanName string - opts []mm_trace.SpanStartOption -} - -// OpentelemetryTracerMockStartResults contains results of the Tracer.Start -type OpentelemetryTracerMockStartResults struct { - c2 context.Context - s1 mm_trace.Span -} - -// Expect sets up expected params for Tracer.Start -func (mmStart *mOpentelemetryTracerMockStart) Expect(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption) *mOpentelemetryTracerMockStart { - if mmStart.mock.funcStart != nil { - mmStart.mock.t.Fatalf("OpentelemetryTracerMock.Start mock is already set by Set") - } - - if mmStart.defaultExpectation == nil { - mmStart.defaultExpectation = &OpentelemetryTracerMockStartExpectation{} - } - - mmStart.defaultExpectation.params = &OpentelemetryTracerMockStartParams{ctx, spanName, opts} - for _, e := range mmStart.expectations { - if minimock.Equal(e.params, mmStart.defaultExpectation.params) { - mmStart.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmStart.defaultExpectation.params) - } - } - - return mmStart -} - -// Inspect accepts an inspector function that has same arguments as the Tracer.Start -func (mmStart *mOpentelemetryTracerMockStart) Inspect(f func(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption)) *mOpentelemetryTracerMockStart { - if mmStart.mock.inspectFuncStart != nil { - mmStart.mock.t.Fatalf("Inspect function is already set for OpentelemetryTracerMock.Start") - } - - mmStart.mock.inspectFuncStart = f - - return mmStart -} - -// Return sets up results that will be returned by Tracer.Start -func (mmStart *mOpentelemetryTracerMockStart) Return(c2 context.Context, s1 mm_trace.Span) *OpentelemetryTracerMock { - if mmStart.mock.funcStart != nil { - mmStart.mock.t.Fatalf("OpentelemetryTracerMock.Start mock is already set by Set") - } - - if mmStart.defaultExpectation == nil { - mmStart.defaultExpectation = &OpentelemetryTracerMockStartExpectation{mock: mmStart.mock} - } - mmStart.defaultExpectation.results = &OpentelemetryTracerMockStartResults{c2, s1} - return mmStart.mock -} - -// Set uses given function f to mock the Tracer.Start method -func (mmStart *mOpentelemetryTracerMockStart) Set(f func(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption) (c2 context.Context, s1 mm_trace.Span)) *OpentelemetryTracerMock { - if mmStart.defaultExpectation != nil { - mmStart.mock.t.Fatalf("Default expectation is already set for the Tracer.Start method") - } - - if len(mmStart.expectations) > 0 { - mmStart.mock.t.Fatalf("Some expectations are already set for the Tracer.Start method") - } - - mmStart.mock.funcStart = f - return mmStart.mock -} - -// When sets expectation for the Tracer.Start which will trigger the result defined by the following -// Then helper -func (mmStart *mOpentelemetryTracerMockStart) When(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption) *OpentelemetryTracerMockStartExpectation { - if mmStart.mock.funcStart != nil { - mmStart.mock.t.Fatalf("OpentelemetryTracerMock.Start mock is already set by Set") - } - - expectation := &OpentelemetryTracerMockStartExpectation{ - mock: mmStart.mock, - params: &OpentelemetryTracerMockStartParams{ctx, spanName, opts}, - } - mmStart.expectations = append(mmStart.expectations, expectation) - return expectation -} - -// Then sets up Tracer.Start return parameters for the expectation previously defined by the When method -func (e *OpentelemetryTracerMockStartExpectation) Then(c2 context.Context, s1 mm_trace.Span) *OpentelemetryTracerMock { - e.results = &OpentelemetryTracerMockStartResults{c2, s1} - return e.mock -} - -// Start implements trace.Tracer -func (mmStart *OpentelemetryTracerMock) Start(ctx context.Context, spanName string, opts ...mm_trace.SpanStartOption) (c2 context.Context, s1 mm_trace.Span) { - mm_atomic.AddUint64(&mmStart.beforeStartCounter, 1) - defer mm_atomic.AddUint64(&mmStart.afterStartCounter, 1) - - if mmStart.inspectFuncStart != nil { - mmStart.inspectFuncStart(ctx, spanName, opts...) - } - - mm_params := &OpentelemetryTracerMockStartParams{ctx, spanName, opts} - - // Record call args - mmStart.StartMock.mutex.Lock() - mmStart.StartMock.callArgs = append(mmStart.StartMock.callArgs, mm_params) - mmStart.StartMock.mutex.Unlock() - - for _, e := range mmStart.StartMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return e.results.c2, e.results.s1 - } - } - - if mmStart.StartMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmStart.StartMock.defaultExpectation.Counter, 1) - mm_want := mmStart.StartMock.defaultExpectation.params - mm_got := OpentelemetryTracerMockStartParams{ctx, spanName, opts} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmStart.t.Errorf("OpentelemetryTracerMock.Start got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - mm_results := mmStart.StartMock.defaultExpectation.results - if mm_results == nil { - mmStart.t.Fatal("No results are set for the OpentelemetryTracerMock.Start") - } - return (*mm_results).c2, (*mm_results).s1 - } - if mmStart.funcStart != nil { - return mmStart.funcStart(ctx, spanName, opts...) - } - mmStart.t.Fatalf("Unexpected call to OpentelemetryTracerMock.Start. %v %v %v", ctx, spanName, opts) - return -} - -// StartAfterCounter returns a count of finished OpentelemetryTracerMock.Start invocations -func (mmStart *OpentelemetryTracerMock) StartAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmStart.afterStartCounter) -} - -// StartBeforeCounter returns a count of OpentelemetryTracerMock.Start invocations -func (mmStart *OpentelemetryTracerMock) StartBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmStart.beforeStartCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetryTracerMock.Start. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmStart *mOpentelemetryTracerMockStart) Calls() []*OpentelemetryTracerMockStartParams { - mmStart.mutex.RLock() - - argCopy := make([]*OpentelemetryTracerMockStartParams, len(mmStart.callArgs)) - copy(argCopy, mmStart.callArgs) - - mmStart.mutex.RUnlock() - - return argCopy -} - -// MinimockStartDone returns true if the count of the Start invocations corresponds -// the number of defined expectations -func (m *OpentelemetryTracerMock) MinimockStartDone() bool { - for _, e := range m.StartMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.StartMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterStartCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcStart != nil && mm_atomic.LoadUint64(&m.afterStartCounter) < 1 { - return false - } - return true -} - -// MinimockStartInspect logs each unmet expectation -func (m *OpentelemetryTracerMock) MinimockStartInspect() { - for _, e := range m.StartMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetryTracerMock.Start with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.StartMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterStartCounter) < 1 { - if m.StartMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetryTracerMock.Start") - } else { - m.t.Errorf("Expected call to OpentelemetryTracerMock.Start with params: %#v", *m.StartMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcStart != nil && mm_atomic.LoadUint64(&m.afterStartCounter) < 1 { - m.t.Error("Expected call to OpentelemetryTracerMock.Start") - } -} - -// MinimockFinish checks that all mocked methods have been called the expected number of times -func (m *OpentelemetryTracerMock) MinimockFinish() { - if !m.minimockDone() { - m.MinimockStartInspect() - m.t.FailNow() - } -} - -// MinimockWait waits for all mocked methods to be called the expected number of times -func (m *OpentelemetryTracerMock) MinimockWait(timeout mm_time.Duration) { - timeoutCh := mm_time.After(timeout) - for { - if m.minimockDone() { - return - } - select { - case <-timeoutCh: - m.MinimockFinish() - return - case <-mm_time.After(10 * mm_time.Millisecond): - } - } -} - -func (m *OpentelemetryTracerMock) minimockDone() bool { - done := true - return done && - m.MinimockStartDone() -} diff --git a/templates_tests/tracer_provider_opentelemery_mock_test.go b/templates_tests/tracer_provider_opentelemery_mock_test.go deleted file mode 100644 index dc8f3f0..0000000 --- a/templates_tests/tracer_provider_opentelemery_mock_test.go +++ /dev/null @@ -1,284 +0,0 @@ -package templatestests - -// Code generated by http://github.com/gojuno/minimock (dev). DO NOT EDIT. - -//go:generate minimock -i go.opentelemetry.io/otel/trace.TracerProvider -o ./tracer_provider_opentelemery_mock_test.go -n OpentelemetryTracerProviderMock - -import ( - "sync" - mm_atomic "sync/atomic" - mm_time "time" - - "github.com/gojuno/minimock/v3" - mm_trace "go.opentelemetry.io/otel/trace" -) - -// OpentelemetryTracerProviderMock implements trace.TracerProvider -type OpentelemetryTracerProviderMock struct { - t minimock.Tester - - funcTracer func(instrumentationName string, opts ...mm_trace.TracerOption) (t1 mm_trace.Tracer) - inspectFuncTracer func(instrumentationName string, opts ...mm_trace.TracerOption) - afterTracerCounter uint64 - beforeTracerCounter uint64 - TracerMock mOpentelemetryTracerProviderMockTracer -} - -// NewOpentelemetryTracerProviderMock returns a mock for trace.TracerProvider -func NewOpentelemetryTracerProviderMock(t minimock.Tester) *OpentelemetryTracerProviderMock { - m := &OpentelemetryTracerProviderMock{t: t} - if controller, ok := t.(minimock.MockController); ok { - controller.RegisterMocker(m) - } - - m.TracerMock = mOpentelemetryTracerProviderMockTracer{mock: m} - m.TracerMock.callArgs = []*OpentelemetryTracerProviderMockTracerParams{} - - return m -} - -type mOpentelemetryTracerProviderMockTracer struct { - mock *OpentelemetryTracerProviderMock - defaultExpectation *OpentelemetryTracerProviderMockTracerExpectation - expectations []*OpentelemetryTracerProviderMockTracerExpectation - - callArgs []*OpentelemetryTracerProviderMockTracerParams - mutex sync.RWMutex -} - -// OpentelemetryTracerProviderMockTracerExpectation specifies expectation struct of the TracerProvider.Tracer -type OpentelemetryTracerProviderMockTracerExpectation struct { - mock *OpentelemetryTracerProviderMock - params *OpentelemetryTracerProviderMockTracerParams - results *OpentelemetryTracerProviderMockTracerResults - Counter uint64 -} - -// OpentelemetryTracerProviderMockTracerParams contains parameters of the TracerProvider.Tracer -type OpentelemetryTracerProviderMockTracerParams struct { - instrumentationName string - opts []mm_trace.TracerOption -} - -// OpentelemetryTracerProviderMockTracerResults contains results of the TracerProvider.Tracer -type OpentelemetryTracerProviderMockTracerResults struct { - t1 mm_trace.Tracer -} - -// Expect sets up expected params for TracerProvider.Tracer -func (mmTracer *mOpentelemetryTracerProviderMockTracer) Expect(instrumentationName string, opts ...mm_trace.TracerOption) *mOpentelemetryTracerProviderMockTracer { - if mmTracer.mock.funcTracer != nil { - mmTracer.mock.t.Fatalf("OpentelemetryTracerProviderMock.Tracer mock is already set by Set") - } - - if mmTracer.defaultExpectation == nil { - mmTracer.defaultExpectation = &OpentelemetryTracerProviderMockTracerExpectation{} - } - - mmTracer.defaultExpectation.params = &OpentelemetryTracerProviderMockTracerParams{instrumentationName, opts} - for _, e := range mmTracer.expectations { - if minimock.Equal(e.params, mmTracer.defaultExpectation.params) { - mmTracer.mock.t.Fatalf("Expectation set by When has same params: %#v", *mmTracer.defaultExpectation.params) - } - } - - return mmTracer -} - -// Inspect accepts an inspector function that has same arguments as the TracerProvider.Tracer -func (mmTracer *mOpentelemetryTracerProviderMockTracer) Inspect(f func(instrumentationName string, opts ...mm_trace.TracerOption)) *mOpentelemetryTracerProviderMockTracer { - if mmTracer.mock.inspectFuncTracer != nil { - mmTracer.mock.t.Fatalf("Inspect function is already set for OpentelemetryTracerProviderMock.Tracer") - } - - mmTracer.mock.inspectFuncTracer = f - - return mmTracer -} - -// Return sets up results that will be returned by TracerProvider.Tracer -func (mmTracer *mOpentelemetryTracerProviderMockTracer) Return(t1 mm_trace.Tracer) *OpentelemetryTracerProviderMock { - if mmTracer.mock.funcTracer != nil { - mmTracer.mock.t.Fatalf("OpentelemetryTracerProviderMock.Tracer mock is already set by Set") - } - - if mmTracer.defaultExpectation == nil { - mmTracer.defaultExpectation = &OpentelemetryTracerProviderMockTracerExpectation{mock: mmTracer.mock} - } - mmTracer.defaultExpectation.results = &OpentelemetryTracerProviderMockTracerResults{t1} - return mmTracer.mock -} - -// Set uses given function f to mock the TracerProvider.Tracer method -func (mmTracer *mOpentelemetryTracerProviderMockTracer) Set(f func(instrumentationName string, opts ...mm_trace.TracerOption) (t1 mm_trace.Tracer)) *OpentelemetryTracerProviderMock { - if mmTracer.defaultExpectation != nil { - mmTracer.mock.t.Fatalf("Default expectation is already set for the TracerProvider.Tracer method") - } - - if len(mmTracer.expectations) > 0 { - mmTracer.mock.t.Fatalf("Some expectations are already set for the TracerProvider.Tracer method") - } - - mmTracer.mock.funcTracer = f - return mmTracer.mock -} - -// When sets expectation for the TracerProvider.Tracer which will trigger the result defined by the following -// Then helper -func (mmTracer *mOpentelemetryTracerProviderMockTracer) When(instrumentationName string, opts ...mm_trace.TracerOption) *OpentelemetryTracerProviderMockTracerExpectation { - if mmTracer.mock.funcTracer != nil { - mmTracer.mock.t.Fatalf("OpentelemetryTracerProviderMock.Tracer mock is already set by Set") - } - - expectation := &OpentelemetryTracerProviderMockTracerExpectation{ - mock: mmTracer.mock, - params: &OpentelemetryTracerProviderMockTracerParams{instrumentationName, opts}, - } - mmTracer.expectations = append(mmTracer.expectations, expectation) - return expectation -} - -// Then sets up TracerProvider.Tracer return parameters for the expectation previously defined by the When method -func (e *OpentelemetryTracerProviderMockTracerExpectation) Then(t1 mm_trace.Tracer) *OpentelemetryTracerProviderMock { - e.results = &OpentelemetryTracerProviderMockTracerResults{t1} - return e.mock -} - -// Tracer implements trace.TracerProvider -func (mmTracer *OpentelemetryTracerProviderMock) Tracer(instrumentationName string, opts ...mm_trace.TracerOption) (t1 mm_trace.Tracer) { - mm_atomic.AddUint64(&mmTracer.beforeTracerCounter, 1) - defer mm_atomic.AddUint64(&mmTracer.afterTracerCounter, 1) - - if mmTracer.inspectFuncTracer != nil { - mmTracer.inspectFuncTracer(instrumentationName, opts...) - } - - mm_params := &OpentelemetryTracerProviderMockTracerParams{instrumentationName, opts} - - // Record call args - mmTracer.TracerMock.mutex.Lock() - mmTracer.TracerMock.callArgs = append(mmTracer.TracerMock.callArgs, mm_params) - mmTracer.TracerMock.mutex.Unlock() - - for _, e := range mmTracer.TracerMock.expectations { - if minimock.Equal(e.params, mm_params) { - mm_atomic.AddUint64(&e.Counter, 1) - return e.results.t1 - } - } - - if mmTracer.TracerMock.defaultExpectation != nil { - mm_atomic.AddUint64(&mmTracer.TracerMock.defaultExpectation.Counter, 1) - mm_want := mmTracer.TracerMock.defaultExpectation.params - mm_got := OpentelemetryTracerProviderMockTracerParams{instrumentationName, opts} - if mm_want != nil && !minimock.Equal(*mm_want, mm_got) { - mmTracer.t.Errorf("OpentelemetryTracerProviderMock.Tracer got unexpected parameters, want: %#v, got: %#v%s\n", *mm_want, mm_got, minimock.Diff(*mm_want, mm_got)) - } - - mm_results := mmTracer.TracerMock.defaultExpectation.results - if mm_results == nil { - mmTracer.t.Fatal("No results are set for the OpentelemetryTracerProviderMock.Tracer") - } - return (*mm_results).t1 - } - if mmTracer.funcTracer != nil { - return mmTracer.funcTracer(instrumentationName, opts...) - } - mmTracer.t.Fatalf("Unexpected call to OpentelemetryTracerProviderMock.Tracer. %v %v", instrumentationName, opts) - return -} - -// TracerAfterCounter returns a count of finished OpentelemetryTracerProviderMock.Tracer invocations -func (mmTracer *OpentelemetryTracerProviderMock) TracerAfterCounter() uint64 { - return mm_atomic.LoadUint64(&mmTracer.afterTracerCounter) -} - -// TracerBeforeCounter returns a count of OpentelemetryTracerProviderMock.Tracer invocations -func (mmTracer *OpentelemetryTracerProviderMock) TracerBeforeCounter() uint64 { - return mm_atomic.LoadUint64(&mmTracer.beforeTracerCounter) -} - -// Calls returns a list of arguments used in each call to OpentelemetryTracerProviderMock.Tracer. -// The list is in the same order as the calls were made (i.e. recent calls have a higher index) -func (mmTracer *mOpentelemetryTracerProviderMockTracer) Calls() []*OpentelemetryTracerProviderMockTracerParams { - mmTracer.mutex.RLock() - - argCopy := make([]*OpentelemetryTracerProviderMockTracerParams, len(mmTracer.callArgs)) - copy(argCopy, mmTracer.callArgs) - - mmTracer.mutex.RUnlock() - - return argCopy -} - -// MinimockTracerDone returns true if the count of the Tracer invocations corresponds -// the number of defined expectations -func (m *OpentelemetryTracerProviderMock) MinimockTracerDone() bool { - for _, e := range m.TracerMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - return false - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.TracerMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterTracerCounter) < 1 { - return false - } - // if func was set then invocations count should be greater than zero - if m.funcTracer != nil && mm_atomic.LoadUint64(&m.afterTracerCounter) < 1 { - return false - } - return true -} - -// MinimockTracerInspect logs each unmet expectation -func (m *OpentelemetryTracerProviderMock) MinimockTracerInspect() { - for _, e := range m.TracerMock.expectations { - if mm_atomic.LoadUint64(&e.Counter) < 1 { - m.t.Errorf("Expected call to OpentelemetryTracerProviderMock.Tracer with params: %#v", *e.params) - } - } - - // if default expectation was set then invocations count should be greater than zero - if m.TracerMock.defaultExpectation != nil && mm_atomic.LoadUint64(&m.afterTracerCounter) < 1 { - if m.TracerMock.defaultExpectation.params == nil { - m.t.Error("Expected call to OpentelemetryTracerProviderMock.Tracer") - } else { - m.t.Errorf("Expected call to OpentelemetryTracerProviderMock.Tracer with params: %#v", *m.TracerMock.defaultExpectation.params) - } - } - // if func was set then invocations count should be greater than zero - if m.funcTracer != nil && mm_atomic.LoadUint64(&m.afterTracerCounter) < 1 { - m.t.Error("Expected call to OpentelemetryTracerProviderMock.Tracer") - } -} - -// MinimockFinish checks that all mocked methods have been called the expected number of times -func (m *OpentelemetryTracerProviderMock) MinimockFinish() { - if !m.minimockDone() { - m.MinimockTracerInspect() - m.t.FailNow() - } -} - -// MinimockWait waits for all mocked methods to be called the expected number of times -func (m *OpentelemetryTracerProviderMock) MinimockWait(timeout mm_time.Duration) { - timeoutCh := mm_time.After(timeout) - for { - if m.minimockDone() { - return - } - select { - case <-timeoutCh: - m.MinimockFinish() - return - case <-mm_time.After(10 * mm_time.Millisecond): - } - } -} - -func (m *OpentelemetryTracerProviderMock) minimockDone() bool { - done := true - return done && - m.MinimockTracerDone() -}