fix(tests): resolve linting issues in telemetry tests

This commit is contained in:
Emilio Garcia 2025-10-29 11:26:11 -04:00
parent a8a8aa56c0
commit 583df48479
3 changed files with 27 additions and 19 deletions

View file

@ -6,7 +6,7 @@
"""Shared helpers for telemetry test collectors."""
from collections.abc import Iterable
from collections.abc import Iterable, Mapping
from dataclasses import dataclass
from typing import Any
@ -14,7 +14,7 @@ from typing import Any
@dataclass
class SpanStub:
name: str
attributes: dict[str, Any]
attributes: Mapping[str, Any] | None = None
resource_attributes: dict[str, Any] | None = None
events: list[dict[str, Any]] | None = None
trace_id: str | None = None

View file

@ -55,9 +55,12 @@ class InMemoryTelemetryCollector(BaseTelemetryCollector):
def _snapshot_metrics(self) -> Any | None:
data = self._metric_reader.get_metrics_data()
if data and data.resource_metrics:
resource_metric = data.resource_metrics[0]
if resource_metric.scope_metrics:
return resource_metric.scope_metrics[0].metrics
all_metrics = []
for resource_metric in data.resource_metrics:
if resource_metric.scope_metrics:
for scope_metric in resource_metric.scope_metrics:
all_metrics.extend(scope_metric.metrics)
return all_metrics if all_metrics else None
return None
def _clear_impl(self) -> None: