otlp-test-data 0.9.0__py3-none-any.whl

otlp_test_data-0.9.0.dist-info/METADATA

@@ -0,0 +1,53 @@
+Metadata-Version: 2.2
+Name: otlp-test-data
+Version: 0.9.0
+Summary: Produces OTLP data using OTEL instrumentation
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Natural Language :: English
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Classifier: Topic :: Software Development :: Testing
+Classifier: Topic :: Software Development :: Testing :: Traffic Generation
+Classifier: Framework :: OpenTelemetry
+Classifier: Framework :: OpenTelemetry :: Exporters
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: freezegun~=1.5
+Requires-Dist: opentelemetry-api~=1.30
+Requires-Dist: opentelemetry-exporter-otlp~=1.30
+Requires-Dist: opentelemetry-exporter-otlp-proto-http~=1.30
+
+# otlp-test-data
+
+Produces OTLP data using OTEL instrumentation.
+
+### Features
+
+- fixed, configurable timestamps
+- aims to cover as much of OTEL API as possible
+- aims to cover all valid data types
+
+### Limitations
+
+- currently only tracing data is generated, PRs are welcome to add metrics and logs data
+- data is generated in process, remote (forwarded) spans are not supported
+
+### TODO
+
+- Events
+- Links
+- Baggage
+- Schemata, when https://github.com/open-telemetry/opentelemetry-python/pull/4359 lands

otlp_test_data-0.9.0.dist-info/RECORD

@@ -0,0 +1,5 @@
+otlp_test_data.py,sha256=nawhyR0ClypJ_1L1denG9uTcydqqAfqmuPU5xqwc_IM,2867
+otlp_test_data-0.9.0.dist-info/METADATA,sha256=aGGgCwBapIaHfnZHHq8B1gh8EQN_8bPdPRQRx6QryIk,1949
+otlp_test_data-0.9.0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+otlp_test_data-0.9.0.dist-info/top_level.txt,sha256=ft5rEmyMAUMgA-09EOjh--RP05FB8jlthlJ1FywFTso,15
+otlp_test_data-0.9.0.dist-info/RECORD,,

otlp_test_data-0.9.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.8.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

otlp_test_data-0.9.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+otlp_test_data

otlp_test_data.py

@@ -0,0 +1,89 @@
+from __future__ import annotations
+
+import base64
+import dataclasses
+import json
+import random
+from typing import Sequence, TYPE_CHECKING
+from typing_extensions import reveal_type as reveal_type  # temp
+
+import freezegun
+from google.protobuf.json_format import MessageToDict
+from opentelemetry.exporter.otlp.proto.common._internal import trace_encoder
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
+
+if TYPE_CHECKING:
+    from google.protobuf.message import Message
+    from opentelemetry.sdk.trace import ReadableSpan
+    from opentelemetry.proto.collector.trace.v1.trace_service_pb2 import (
+        ExportTraceServiceRequest,
+    )
+
+
+@dataclasses.dataclass
+class Config:
+    start_time: str = "2020-01-01 00:00:00Z"
+    random_seed: int = 42
+
+
+time = None
+
+
+def sample_proto(config: Config) -> bytes:
+    return _proto_to_bytes(_spans_to_proto_object(sample_spans(config)))
+
+
+def sample_json(config: Config) -> bytes:
+    return _proto_to_json(_spans_to_proto_object(sample_spans(config)))
+
+
+def sample_spans(config: Config) -> Sequence[ReadableSpan]:
+    """Creates and finishes two spans, then returns them as a list."""
+    global time
+    tracer_provider = TracerProvider()
+    exporter = InMemorySpanExporter()
+    tracer_provider.add_span_processor(SimpleSpanProcessor(exporter))
+    tracer = tracer_provider.get_tracer(__name__)
+
+    with freezegun.freeze_time(config.start_time) as time:
+        random.seed(config.random_seed)
+
+        # FIXME the workload section is expected to grow a lot
+        with tracer.start_as_current_span("span-one"):
+            time.tick()
+        with tracer.start_as_current_span("span-two"):
+            time.tick()
+
+    return exporter.get_finished_spans()
+
+
+def _spans_to_proto_object(spans: Sequence[ReadableSpan]) -> ExportTraceServiceRequest:
+    return trace_encoder.encode_spans(spans)
+
+
+def _proto_to_bytes(data: Message) -> bytes:
+    return data.SerializePartialToString()
+
+
+# FIXME: there are probably 3 different enumerated types in the API
+def _proto_to_json(data: Message) -> bytes:
+    dic = MessageToDict(data)
+
+    for rs in dic["resourceSpans"]:
+        for ss in rs["scopeSpans"]:
+            for sp in ss["spans"]:
+                for k in "parentSpanId spanId traceId".split():
+                    if k in sp:
+                        sp[k] = base64.b64decode(sp[k]).hex()
+                sp["kind"] = {
+                    "SPAN_KIND_UNSPECIFIED": 0,
+                    "SPAN_KIND_INTERNAL": 1,
+                    "SPAN_KIND_SERVER": 2,
+                    "SPAN_KIND_CLIENT": 3,
+                    "SPAN_KIND_PRODUCER": 4,
+                    "SPAN_KIND_CONSUMER": 5,
+                }[sp["kind"]]
+
+    return json.dumps(dic).encode("utf-8")