16
16
from unittest .mock import Mock , patch
17
17
18
18
from opentelemetry import baggage
19
- from opentelemetry .baggage .propagation import BaggagePropagator
19
+ from opentelemetry .baggage .propagation import W3CBaggagePropagator
20
20
from opentelemetry .context import get_current
21
21
from opentelemetry .propagators .textmap import DictGetter
22
22
25
25
26
26
class TestBaggagePropagation (unittest .TestCase ):
27
27
def setUp (self ):
28
- self .propagator = BaggagePropagator ()
28
+ self .propagator = W3CBaggagePropagator ()
29
29
30
30
def _extract (self , header_value ):
31
31
"""Test helper"""
@@ -87,7 +87,7 @@ def test_invalid_header(self):
87
87
self .assertEqual (self ._extract (header ), expected )
88
88
89
89
def test_header_too_long (self ):
90
- long_value = "s" * (BaggagePropagator .MAX_HEADER_LENGTH + 1 )
90
+ long_value = "s" * (W3CBaggagePropagator .MAX_HEADER_LENGTH + 1 )
91
91
header = "key1={}" .format (long_value )
92
92
expected = {}
93
93
self .assertEqual (self ._extract (header ), expected )
@@ -96,15 +96,15 @@ def test_header_contains_too_many_entries(self):
96
96
header = "," .join (
97
97
[
98
98
"key{}=val" .format (k )
99
- for k in range (BaggagePropagator .MAX_PAIRS + 1 )
99
+ for k in range (W3CBaggagePropagator .MAX_PAIRS + 1 )
100
100
]
101
101
)
102
102
self .assertEqual (
103
- len (self ._extract (header )), BaggagePropagator .MAX_PAIRS
103
+ len (self ._extract (header )), W3CBaggagePropagator .MAX_PAIRS
104
104
)
105
105
106
106
def test_header_contains_pair_too_long (self ):
107
- long_value = "s" * (BaggagePropagator .MAX_PAIR_LENGTH + 1 )
107
+ long_value = "s" * (W3CBaggagePropagator .MAX_PAIR_LENGTH + 1 )
108
108
header = "key1=value1,key2={},key3=value3" .format (long_value )
109
109
expected = {"key1" : "value1" , "key3" : "value3" }
110
110
self .assertEqual (self ._extract (header ), expected )
0 commit comments