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
|
import os
from aws_xray_sdk.core import lambda_launcher
from aws_xray_sdk.core.models.subsegment import Subsegment
TRACE_ID = '1-5759e988-bd862e3fe1be46a994272793'
PARENT_ID = '53995c3f42cd8ad8'
HEADER_VAR = "Root=%s;Parent=%s;Sampled=1" % (TRACE_ID, PARENT_ID)
os.environ[lambda_launcher.LAMBDA_TRACE_HEADER_KEY] = HEADER_VAR
context = lambda_launcher.LambdaContext()
def test_facade_segment_generation():
segment = context.get_trace_entity()
assert segment.id == PARENT_ID
assert segment.trace_id == TRACE_ID
assert segment.sampled
def test_put_subsegment():
segment = context.get_trace_entity()
subsegment = Subsegment('name', 'local', segment)
context.put_subsegment(subsegment)
assert context.get_trace_entity().id == subsegment.id
subsegment2 = Subsegment('name', 'local', segment)
context.put_subsegment(subsegment2)
assert context.get_trace_entity().id == subsegment2.id
assert subsegment.subsegments[0] is subsegment2
assert subsegment2.parent_id == subsegment.id
assert subsegment.parent_id == segment.id
assert subsegment2.parent_segment is segment
context.end_subsegment()
assert context.get_trace_entity().id == subsegment.id
context.end_subsegment()
assert context.get_trace_entity().id == segment.id
|
