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
|
from aws_xray_sdk.core.models.dummy_entities import DummySegment, DummySubsegment
from aws_xray_sdk.core.models import http
def test_not_sampled():
segment = DummySegment()
subsegment = DummySubsegment(segment)
assert not segment.sampled
assert not subsegment.sampled
def test_no_ops():
segment = DummySegment()
segment.put_metadata('key', 'value')
segment.put_annotation('key', 'value')
segment.put_http_meta(http.URL, 'url')
segment.set_user('user')
assert not segment.metadata
assert not segment.annotations
assert not segment.http
assert not segment.user
subsegment = DummySubsegment(segment)
subsegment.put_metadata('key', 'value')
subsegment.put_annotation('key', 'value')
subsegment.put_http_meta(http.URL, 'url')
subsegment.set_user('user')
subsegment.set_aws({'key': 'value'})
subsegment.set_sql({'key': 'value'})
assert not subsegment.metadata
assert not subsegment.annotations
assert not subsegment.http
assert not subsegment.user
assert not subsegment.aws
assert not subsegment.sql
assert not segment.serialize()
assert not subsegment.serialize()
def test_structure_intact():
segment = DummySegment()
subsegment = DummySubsegment(segment)
subsegment2 = DummySubsegment(segment)
subsegment.add_subsegment(subsegment2)
segment.add_subsegment(subsegment)
assert segment.subsegments[0] is subsegment
assert subsegment.subsegments[0] is subsegment2
subsegment2.close()
subsegment.close()
segment.close()
assert segment.ready_to_send()
def test_invalid_entity_name():
segment = DummySegment('DummySegment() Test?')
subsegment = DummySubsegment(segment, 'Dummy*Sub!segment$')
assert segment.name == 'DummySegment Test'
assert subsegment.name == 'DummySubsegment'
|
