mirror of
https://github.com/open-telemetry/opentelemetry-python-contrib.git
synced 2025-08-02 19:47:17 +08:00
Rename IdsGenerator to IdGenerator (#350)
This commit is contained in:
@ -12,14 +12,14 @@
|
||||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
from opentelemetry.sdk.extension.aws.trace import AwsXRayIdsGenerator
|
||||
from opentelemetry.sdk.extension.aws.trace import AwsXRayIdGenerator
|
||||
|
||||
ids_generator = AwsXRayIdsGenerator()
|
||||
id_generator = AwsXRayIdGenerator()
|
||||
|
||||
|
||||
def test_generate_xray_trace_id(benchmark):
|
||||
benchmark(ids_generator.generate_trace_id)
|
||||
benchmark(id_generator.generate_trace_id)
|
||||
|
||||
|
||||
def test_generate_xray_span_id(benchmark):
|
||||
benchmark(ids_generator.generate_span_id)
|
||||
benchmark(id_generator.generate_span_id)
|
||||
|
@ -16,23 +16,23 @@ import datetime
|
||||
import time
|
||||
import unittest
|
||||
|
||||
from opentelemetry.sdk.extension.aws.trace import AwsXRayIdsGenerator
|
||||
from opentelemetry.sdk.extension.aws.trace import AwsXRayIdGenerator
|
||||
from opentelemetry.trace.span import INVALID_TRACE_ID
|
||||
|
||||
|
||||
class AwsXRayIdsGeneratorTest(unittest.TestCase):
|
||||
class AwsXRayIdGeneratorTest(unittest.TestCase):
|
||||
def test_ids_are_valid(self):
|
||||
ids_generator = AwsXRayIdsGenerator()
|
||||
id_generator = AwsXRayIdGenerator()
|
||||
for _ in range(1000):
|
||||
trace_id = ids_generator.generate_trace_id()
|
||||
trace_id = id_generator.generate_trace_id()
|
||||
self.assertTrue(trace_id != INVALID_TRACE_ID)
|
||||
span_id = ids_generator.generate_span_id()
|
||||
span_id = id_generator.generate_span_id()
|
||||
self.assertTrue(span_id != INVALID_TRACE_ID)
|
||||
|
||||
def test_id_timestamps_are_acceptable_for_xray(self):
|
||||
ids_generator = AwsXRayIdsGenerator()
|
||||
id_generator = AwsXRayIdGenerator()
|
||||
for _ in range(1000):
|
||||
trace_id = ids_generator.generate_trace_id()
|
||||
trace_id = id_generator.generate_trace_id()
|
||||
trace_id_time = trace_id >> 96
|
||||
current_time = int(time.time())
|
||||
self.assertLessEqual(trace_id_time, current_time)
|
||||
|
Reference in New Issue
Block a user