9
9
"testing"
10
10
11
11
"go.opentelemetry.io/otel/trace"
12
- "go.opentelemetry.io/otel/trace/noop"
13
12
14
13
"github.com/go-chi/chi/v5"
15
14
"github.com/stretchr/testify/require"
@@ -18,32 +17,22 @@ import (
18
17
"github.com/coder/coder/v2/testutil"
19
18
)
20
19
21
- type fakeTracerProvider struct {
22
- noop.TracerProvider
23
- startCalled int64
24
- }
25
-
26
20
type fakeTracer struct {
27
- noop.Tracer
28
- prov * fakeTracerProvider
21
+ startCalled int64
29
22
}
30
23
31
24
var (
32
- _ trace.TracerProvider = & fakeTracerProvider {}
25
+ _ trace.TracerProvider = & fakeTracer {}
33
26
_ trace.Tracer = & fakeTracer {}
34
27
)
35
28
36
- func (f * fakeTracer ) Start (ctx context.Context , str string , opts ... trace.SpanStartOption ) (context.Context , trace.Span ) {
37
- return f .prov .Start (ctx , str , opts ... )
38
- }
39
-
40
29
// Tracer implements trace.TracerProvider.
41
- func (f * fakeTracerProvider ) Tracer (_ string , _ ... trace.TracerOption ) trace.Tracer {
42
- return & fakeTracer { prov : f }
30
+ func (f * fakeTracer ) Tracer (_ string , _ ... trace.TracerOption ) trace.Tracer {
31
+ return f
43
32
}
44
33
45
34
// Start implements trace.Tracer.
46
- func (f * fakeTracerProvider ) Start (ctx context.Context , _ string , _ ... trace.SpanStartOption ) (context.Context , trace.Span ) {
35
+ func (f * fakeTracer ) Start (ctx context.Context , _ string , _ ... trace.SpanStartOption ) (context.Context , trace.Span ) {
47
36
atomic .AddInt64 (& f .startCalled , 1 )
48
37
return ctx , tracing .NoopSpan
49
38
}
@@ -87,7 +76,7 @@ func Test_Middleware(t *testing.T) {
87
76
t .Run (name , func (t * testing.T ) {
88
77
t .Parallel ()
89
78
90
- fake := & fakeTracerProvider {}
79
+ fake := & fakeTracer {}
91
80
92
81
rw := & tracing.StatusWriter {ResponseWriter : httptest .NewRecorder ()}
93
82
r := httptest .NewRequest ("GET" , c .path , nil )
0 commit comments