62
62
StatusCode ,
63
63
get_tracer ,
64
64
set_tracer_provider ,
65
- span ,
66
65
)
67
66
68
67
@@ -2070,14 +2069,13 @@ class TestRandomIdGenerator(unittest.TestCase):
2070
2069
2071
2070
@patch (
2072
2071
"random.getrandbits" ,
2073
- side_effect = [span .INVALID_SPAN_ID , 0x00000000DEADBEF0 ],
2072
+ side_effect = [trace_api .INVALID_SPAN_ID , 0x00000000DEADBEF0 ],
2074
2073
)
2075
2074
def test_generate_span_id_avoids_invalid (self , mock_getrandbits ):
2076
2075
generator = RandomIdGenerator ()
2077
2076
span_id = generator .generate_span_id ()
2078
2077
2079
- self .assertNotEqual (span_id , span .INVALID_SPAN_ID )
2080
- self .assertGreater (span_id , span .INVALID_SPAN_ID )
2078
+ self .assertGreater (span_id , trace_api .INVALID_SPAN_ID )
2081
2079
self .assertLessEqual (span_id , self ._SPAN_ID_MAX_VALUE )
2082
2080
self .assertEqual (
2083
2081
mock_getrandbits .call_count , 2
@@ -2086,16 +2084,15 @@ def test_generate_span_id_avoids_invalid(self, mock_getrandbits):
2086
2084
@patch (
2087
2085
"random.getrandbits" ,
2088
2086
side_effect = [
2089
- span .INVALID_TRACE_ID ,
2087
+ trace_api .INVALID_TRACE_ID ,
2090
2088
0x000000000000000000000000DEADBEEF ,
2091
2089
],
2092
2090
)
2093
2091
def test_generate_trace_id_avoids_invalid (self , mock_getrandbits ):
2094
2092
generator = RandomIdGenerator ()
2095
2093
trace_id = generator .generate_trace_id ()
2096
2094
2097
- self .assertNotEqual (trace_id , span .INVALID_TRACE_ID )
2098
- self .assertGreater (trace_id , span .INVALID_TRACE_ID )
2095
+ self .assertGreater (trace_id , trace_api .INVALID_TRACE_ID )
2099
2096
self .assertLessEqual (trace_id , self ._TRACE_ID_MAX_VALUE )
2100
2097
self .assertEqual (
2101
2098
mock_getrandbits .call_count , 2
0 commit comments