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
|
from aws_xray_sdk.core.models.trace_header import TraceHeader
TRACE_ID = '1-5759e988-bd862e3fe1be46a994272793'
PARENT_ID = '53995c3f42cd8ad8'
def test_no_sample():
header = TraceHeader(root=TRACE_ID, parent=PARENT_ID)
assert header.sampled is None
assert header.root == TRACE_ID
assert header.parent == PARENT_ID
assert header.to_header_str() == 'Root=%s;Parent=%s' % (TRACE_ID, PARENT_ID)
def test_no_parent():
header = TraceHeader(root=TRACE_ID, sampled=1)
assert header.parent is None
assert header.to_header_str() == 'Root=%s;Sampled=1' % TRACE_ID
def test_from_str():
# a full header string that has all fields present
header_str1 = 'Root=%s;Parent=%s;Sampled=1' % (TRACE_ID, PARENT_ID)
header1 = TraceHeader.from_header_str(header_str1)
assert header1.root == TRACE_ID
assert header1.parent == PARENT_ID
assert header1.sampled == 1
# missing parent id
header_str2 = 'Root=%s;Sampled=0' % TRACE_ID
header2 = TraceHeader.from_header_str(header_str2)
assert header2.root == TRACE_ID
assert header2.parent is None
assert header2.sampled == 0
# missing sampled
header_str3 = 'Root=%s;Parent=%s' % (TRACE_ID, PARENT_ID)
header3 = TraceHeader.from_header_str(header_str3)
assert header3.root == TRACE_ID
assert header3.parent == PARENT_ID
assert header3.sampled is None
def test_invalid_str():
header = TraceHeader.from_header_str('some invalid string')
assert header.root is None
assert header.parent is None
assert header.sampled is None
|
