1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
package tracetest
import (
"context"
"sync"
"testing"
"github.com/stretchr/testify/assert"
sdktrace "go.opentelemetry.io/otel/sdk/trace"
)
type rwSpan struct {
sdktrace.ReadWriteSpan
}
func TestSpanRecorderOnStartAppends(t *testing.T) {
s0, s1 := new(rwSpan), new(rwSpan)
ctx := context.Background()
sr := new(SpanRecorder)
assert.Empty(t, sr.started)
sr.OnStart(ctx, s0)
assert.Len(t, sr.started, 1)
sr.OnStart(ctx, s1)
assert.Len(t, sr.started, 2)
// Ensure order correct.
started := sr.Started()
assert.Same(t, s0, started[0])
assert.Same(t, s1, started[1])
}
type roSpan struct {
sdktrace.ReadOnlySpan
}
func TestSpanRecorderOnEndAppends(t *testing.T) {
s0, s1 := new(roSpan), new(roSpan)
sr := new(SpanRecorder)
assert.Empty(t, sr.ended)
sr.OnEnd(s0)
assert.Len(t, sr.ended, 1)
sr.OnEnd(s1)
assert.Len(t, sr.ended, 2)
// Ensure order correct.
ended := sr.Ended()
assert.Same(t, s0, ended[0])
assert.Same(t, s1, ended[1])
}
func TestSpanRecorderShutdownNoError(t *testing.T) {
ctx := context.Background()
assert.NoError(t, new(SpanRecorder).Shutdown(ctx))
var c context.CancelFunc
ctx, c = context.WithCancel(ctx)
c()
assert.NoError(t, new(SpanRecorder).Shutdown(ctx))
}
func TestSpanRecorderForceFlushNoError(t *testing.T) {
ctx := context.Background()
assert.NoError(t, new(SpanRecorder).ForceFlush(ctx))
var c context.CancelFunc
ctx, c = context.WithCancel(ctx)
c()
assert.NoError(t, new(SpanRecorder).ForceFlush(ctx))
}
func runConcurrently(funcs ...func()) {
var wg sync.WaitGroup
for _, f := range funcs {
wg.Add(1)
go func(f func()) {
f()
wg.Done()
}(f)
}
wg.Wait()
}
func TestEndingConcurrentSafe(t *testing.T) {
sr := NewSpanRecorder()
runConcurrently(
func() { sr.OnEnd(new(roSpan)) },
func() { sr.OnEnd(new(roSpan)) },
func() { sr.Ended() },
)
assert.Len(t, sr.Ended(), 2)
}
func TestStartingConcurrentSafe(t *testing.T) {
sr := NewSpanRecorder()
ctx := context.Background()
runConcurrently(
func() { sr.OnStart(ctx, new(rwSpan)) },
func() { sr.OnStart(ctx, new(rwSpan)) },
func() { sr.Started() },
)
assert.Len(t, sr.Started(), 2)
}
|
