File tree Expand file tree Collapse file tree 2 files changed +11
-3
lines changed Expand file tree Collapse file tree 2 files changed +11
-3
lines changed Original file line number Diff line number Diff line change
1
+ from .traceid import TraceId
1
2
from .segment import Segment
2
3
from .subsegment import Subsegment
3
4
@@ -13,7 +14,7 @@ class DummySegment(Segment):
13
14
"""
14
15
def __init__ (self , name = 'dummy' ):
15
16
16
- super (DummySegment , self ).__init__ (name = name , traceid = 'dummy' )
17
+ super (DummySegment , self ).__init__ (name = name , traceid = TraceId (). to_id () )
17
18
self .sampled = False
18
19
19
20
def set_aws (self , aws_meta ):
Original file line number Diff line number Diff line change 3
3
4
4
5
5
def test_not_sampled ():
6
-
7
6
segment = DummySegment ()
8
7
subsegment = DummySubsegment (segment )
9
8
@@ -12,7 +11,6 @@ def test_not_sampled():
12
11
13
12
14
13
def test_no_ops ():
15
-
16
14
segment = DummySegment ()
17
15
segment .put_metadata ('key' , 'value' )
18
16
segment .put_annotation ('key' , 'value' )
@@ -63,3 +61,12 @@ def test_invalid_entity_name():
63
61
64
62
assert segment .name == 'DummySegment Test'
65
63
assert subsegment .name == 'DummySubsegment'
64
+
65
+
66
+ def test_dummy_segment_trace_id ():
67
+ segment = DummySegment ()
68
+
69
+ assert segment .trace_id != 'dummy'
70
+ assert '-' in segment .trace_id
71
+ # checking version of trace id
72
+ assert segment .trace_id [:1 ] == '1'
You can’t perform that action at this time.
0 commit comments