add ProtoJSON compatability tests

Author: herin049

exporter/opentelemetry-exporter-otlp-json-common/test-requirements.txt

@@ -3,6 +3,7 @@ importlib-metadata==6.11.0
 iniconfig==2.0.0
 packaging==24.0
 pluggy==1.6.0
+protobuf==6.31.1
 py-cpuinfo==9.0.0
 pytest==7.4.4
 tomli==2.0.1
@@ -14,4 +15,6 @@ zipp==3.19.2
 -e opentelemetry-semantic-conventions
 -e tests/opentelemetry-test-utils
 -e opentelemetry-proto-json
+-e opentelemetry-proto
 -e exporter/opentelemetry-exporter-otlp-json-common
+-e exporter/opentelemetry-exporter-otlp-proto-common

exporter/opentelemetry-exporter-otlp-json-common/tests/__init__.py

@@ -15,6 +15,50 @@
 import dataclasses
 import unittest
 
+from opentelemetry._logs import LogRecord, SeverityNumber
+from opentelemetry.sdk._logs import ReadableLogRecord
+from opentelemetry.sdk.metrics.export import (
+    AggregationTemporality,
+    Buckets,
+    ExponentialHistogram,
+    ExponentialHistogramDataPoint,
+    Gauge,
+    Histogram,
+    HistogramDataPoint,
+    Metric,
+    MetricsData,
+    NumberDataPoint,
+    ResourceMetrics,
+    ScopeMetrics,
+    Sum,
+)
+from opentelemetry.sdk.resources import Resource
+from opentelemetry.sdk.trace import SpanContext, _Span
+from opentelemetry.sdk.util.instrumentation import InstrumentationScope
+from opentelemetry.trace import (
+    NonRecordingSpan,
+    TraceFlags,
+    set_span_in_context,
+)
+
+# ---------------------------------------------------------------------------
+# Shared constants
+# ---------------------------------------------------------------------------
+
+TRACE_ID = 0x3E0C63257DE34C926F9EFCD03927272E
+SPAN_ID = 0x34BF92DEEFC58C92
+PARENT_SPAN_ID = 0x1111111111111111
+BASE_TIME = 683647322 * 10**9
+START_TIME = 1641946015139533244
+TIME = 1641946016139533244
+
+_UNSET = object()
+
+
+# ---------------------------------------------------------------------------
+# Assertion utilities
+# ---------------------------------------------------------------------------
+
 
 def _is_none_equivalent(val_a, val_b):
     """Check if two values should be treated as equal because one is None
@@ -61,3 +105,288 @@ def assert_proto_json_equal(
         test_case.assertEqual(
             obj_a, obj_b, f"Mismatch at {path}: {obj_a!r} != {obj_b!r}"
         )
+
+
+# ---------------------------------------------------------------------------
+# Span builders
+# ---------------------------------------------------------------------------
+
+
+def make_span_unended(
+    name="test-span",
+    trace_id=TRACE_ID,
+    span_id=SPAN_ID,
+    parent=None,
+    resource=None,
+    instrumentation_scope=None,
+    events=(),
+    links=(),
+    start_time=BASE_TIME,
+):
+    if resource is None:
+        resource = Resource({})
+    span = _Span(
+        name=name,
+        context=SpanContext(
+            trace_id,
+            span_id,
+            is_remote=False,
+            trace_flags=TraceFlags(TraceFlags.SAMPLED),
+        ),
+        parent=parent,
+        resource=resource,
+        instrumentation_scope=instrumentation_scope,
+        events=events,
+        links=links,
+    )
+    span.start(start_time=start_time)
+    return span
+
+
+def make_span(
+    name="test-span",
+    trace_id=TRACE_ID,
+    span_id=SPAN_ID,
+    parent=None,
+    resource=None,
+    instrumentation_scope=None,
+    events=(),
+    links=(),
+    start_time=BASE_TIME,
+    end_time=BASE_TIME + 50 * 10**6,
+):
+    span = make_span_unended(
+        name=name,
+        trace_id=trace_id,
+        span_id=span_id,
+        parent=parent,
+        resource=resource,
+        instrumentation_scope=instrumentation_scope,
+        events=events,
+        links=links,
+        start_time=start_time,
+    )
+    span.end(end_time=end_time)
+    return span
+
+
+# ---------------------------------------------------------------------------
+# Log builders
+# ---------------------------------------------------------------------------
+
+
+def make_log_context(trace_id=TRACE_ID, span_id=SPAN_ID):
+    return set_span_in_context(
+        NonRecordingSpan(
+            SpanContext(trace_id, span_id, False, TraceFlags(0x01))
+        )
+    )
+
+
+def make_log(
+    body="test log message",
+    severity_text="INFO",
+    severity_number=SeverityNumber.INFO,
+    attributes=None,
+    timestamp=TIME,
+    observed_timestamp=TIME + 1000,
+    resource=None,
+    instrumentation_scope=_UNSET,
+    event_name=None,
+    context=None,
+    limits=None,
+):
+    kwargs = dict(
+        timestamp=timestamp,
+        observed_timestamp=observed_timestamp,
+        severity_text=severity_text,
+        severity_number=severity_number,
+        body=body,
+        attributes=attributes or {},
+        event_name=event_name,
+    )
+    if context is not None:
+        kwargs["context"] = context
+
+    rw_kwargs = dict(
+        resource=resource or Resource({}),
+        instrumentation_scope=InstrumentationScope("test_scope", "1.0")
+        if instrumentation_scope is _UNSET
+        else instrumentation_scope,
+    )
+    if limits is not None:
+        rw_kwargs["limits"] = limits
+
+    return ReadableLogRecord(LogRecord(**kwargs), **rw_kwargs)
+
+
+# ---------------------------------------------------------------------------
+# Metric builders
+# ---------------------------------------------------------------------------
+
+
+def make_metrics_data(
+    metrics,
+    resource_attrs=None,
+    resource_schema_url=None,
+    scope_name="test_scope",
+    scope_version="1.0",
+    scope_schema_url=None,
+):
+    return MetricsData(
+        resource_metrics=[
+            ResourceMetrics(
+                resource=Resource(
+                    attributes=resource_attrs or {},
+                    schema_url=resource_schema_url,
+                ),
+                scope_metrics=[
+                    ScopeMetrics(
+                        scope=InstrumentationScope(
+                            name=scope_name,
+                            version=scope_version,
+                            schema_url=scope_schema_url,
+                        ),
+                        metrics=metrics,
+                        schema_url=scope_schema_url,
+                    )
+                ],
+                schema_url=resource_schema_url,
+            )
+        ]
+    )
+
+
+def make_sum(
+    name="test_sum",
+    value=33,
+    attributes=None,
+    temporality=AggregationTemporality.CUMULATIVE,
+    is_monotonic=True,
+    description="desc",
+    unit="s",
+):
+    return Metric(
+        name=name,
+        description=description,
+        unit=unit,
+        data=Sum(
+            data_points=[
+                NumberDataPoint(
+                    attributes=attributes or {"a": 1},
+                    start_time_unix_nano=START_TIME,
+                    time_unix_nano=TIME,
+                    value=value,
+                )
+            ],
+            aggregation_temporality=temporality,
+            is_monotonic=is_monotonic,
+        ),
+    )
+
+
+def make_gauge(
+    name="test_gauge",
+    value=9000,
+    attributes=None,
+    description="desc",
+    unit="1",
+):
+    return Metric(
+        name=name,
+        description=description,
+        unit=unit,
+        data=Gauge(
+            data_points=[
+                NumberDataPoint(
+                    attributes=attributes or {"a": 1},
+                    start_time_unix_nano=None,
+                    time_unix_nano=TIME,
+                    value=value,
+                )
+            ],
+        ),
+    )
+
+
+def make_histogram(
+    name="test_histogram",
+    attributes=None,
+    count=5,
+    sum_value=67,
+    bucket_counts=None,
+    explicit_bounds=None,
+    min_value=8,
+    max_value=18,
+    exemplars=None,
+    temporality=AggregationTemporality.DELTA,
+    description="desc",
+    unit="ms",
+):
+    return Metric(
+        name=name,
+        description=description,
+        unit=unit,
+        data=Histogram(
+            data_points=[
+                HistogramDataPoint(
+                    attributes=attributes or {"a": 1},
+                    start_time_unix_nano=START_TIME,
+                    time_unix_nano=TIME,
+                    count=count,
+                    sum=sum_value,
+                    bucket_counts=bucket_counts or [1, 4],
+                    explicit_bounds=explicit_bounds or [10.0, 20.0],
+                    min=min_value,
+                    max=max_value,
+                    exemplars=exemplars or [],
+                )
+            ],
+            aggregation_temporality=temporality,
+        ),
+    )
+
+
+def make_exponential_histogram(
+    name="test_exp_hist",
+    attributes=None,
+    count=10,
+    sum_value=100.5,
+    scale=1,
+    zero_count=2,
+    positive=None,
+    negative=None,
+    flags=0,
+    min_value=1.0,
+    max_value=50.0,
+    exemplars=None,
+    temporality=AggregationTemporality.CUMULATIVE,
+    description="desc",
+    unit="s",
+):
+    return Metric(
+        name=name,
+        description=description,
+        unit=unit,
+        data=ExponentialHistogram(
+            data_points=[
+                ExponentialHistogramDataPoint(
+                    attributes=attributes or {"a": 1},
+                    start_time_unix_nano=START_TIME,
+                    time_unix_nano=TIME,
+                    count=count,
+                    sum=sum_value,
+                    scale=scale,
+                    zero_count=zero_count,
+                    positive=positive
+                    or Buckets(offset=0, bucket_counts=[1, 2, 3]),
+                    negative=negative or Buckets(offset=1, bucket_counts=[1]),
+                    flags=flags,
+                    min=min_value,
+                    max=max_value,
+                    exemplars=exemplars or [],
+                )
+            ],
+            aggregation_temporality=temporality,
+        ),
+    )