@@ -17,6 +17,7 @@ package testtrace_test
17
17
import (
18
18
"context"
19
19
"errors"
20
+ "fmt"
20
21
"sync"
21
22
"testing"
22
23
"time"
@@ -164,7 +165,8 @@ func TestSpan(t *testing.T) {
164
165
},
165
166
}}
166
167
e .Expect (subject .Events ()).ToEqual (expectedEvents )
167
- e .Expect (subject .Status ()).ToEqual (codes .OK )
168
+ e .Expect (subject .StatusCode ()).ToEqual (codes .OK )
169
+ e .Expect (subject .StatusMessage ()).ToEqual ("" )
168
170
}
169
171
})
170
172
@@ -182,8 +184,8 @@ func TestSpan(t *testing.T) {
182
184
errMsg := "test error message"
183
185
testErr := ottest .NewTestError (errMsg )
184
186
testTime := time .Now ()
185
- expStatus := codes .Unknown
186
- subject .RecordError (ctx , testErr , trace .WithErrorTime (testTime ), trace .WithErrorStatus (expStatus ))
187
+ expStatusCode := codes .Unknown
188
+ subject .RecordError (ctx , testErr , trace .WithErrorTime (testTime ), trace .WithErrorStatus (expStatusCode ))
187
189
188
190
expectedEvents := []testtrace.Event {{
189
191
Timestamp : testTime ,
@@ -194,7 +196,7 @@ func TestSpan(t *testing.T) {
194
196
},
195
197
}}
196
198
e .Expect (subject .Events ()).ToEqual (expectedEvents )
197
- e .Expect (subject .Status ()).ToEqual (expStatus )
199
+ e .Expect (subject .StatusCode ()).ToEqual (expStatusCode )
198
200
})
199
201
200
202
t .Run ("cannot be set after the span has ended" , func (t * testing.T ) {
@@ -527,11 +529,11 @@ func TestSpan(t *testing.T) {
527
529
subject , ok := span .(* testtrace.Span )
528
530
e .Expect (ok ).ToBeTrue ()
529
531
530
- e .Expect (subject .Status ()).ToEqual (codes .OK )
532
+ e .Expect (subject .StatusCode ()).ToEqual (codes .OK )
531
533
532
534
subject .End ()
533
535
534
- e .Expect (subject .Status ()).ToEqual (codes .OK )
536
+ e .Expect (subject .StatusCode ()).ToEqual (codes .OK )
535
537
})
536
538
537
539
statuses := []codes.Code {
@@ -566,10 +568,11 @@ func TestSpan(t *testing.T) {
566
568
subject , ok := span .(* testtrace.Span )
567
569
e .Expect (ok ).ToBeTrue ()
568
570
569
- subject .SetStatus (codes .OK )
570
- subject .SetStatus (status )
571
+ subject .SetStatus (codes .OK , "OK" )
572
+ subject .SetStatus (status , "Yo!" )
571
573
572
- e .Expect (subject .Status ()).ToEqual (status )
574
+ e .Expect (subject .StatusCode ()).ToEqual (status )
575
+ e .Expect (subject .StatusMessage ()).ToEqual ("Yo!" )
573
576
})
574
577
575
578
t .Run ("cannot be changed after the span has been ended" , func (t * testing.T ) {
@@ -585,11 +588,12 @@ func TestSpan(t *testing.T) {
585
588
586
589
originalStatus := codes .OK
587
590
588
- subject .SetStatus (originalStatus )
591
+ subject .SetStatus (originalStatus , "OK" )
589
592
subject .End ()
590
- subject .SetStatus (status )
593
+ subject .SetStatus (status , fmt . Sprint ( status ) )
591
594
592
- e .Expect (subject .Status ()).ToEqual (originalStatus )
595
+ e .Expect (subject .StatusCode ()).ToEqual (originalStatus )
596
+ e .Expect (subject .StatusMessage ()).ToEqual ("OK" )
593
597
})
594
598
}
595
599
})
0 commit comments