Skip to content

Commit

Permalink
test: fix histograms monitoring test (#5509)
Browse files Browse the repository at this point in the history
  • Loading branch information
alaeddine-13 authored Dec 12, 2022
1 parent 7ab2d9c commit fd37a09
Showing 1 changed file with 5 additions and 4 deletions.
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
from typing import Dict, List, Tuple

import pytest
from typing import Tuple, List, Dict
from opentelemetry.metrics import Meter
from opentelemetry.sdk.metrics import MeterProvider
from opentelemetry.sdk.metrics.export import (
HistogramDataPoint,
InMemoryMetricReader,
Metric,
HistogramDataPoint,
)
from opentelemetry.metrics import Meter

from jina.serve.networking import _NetworkingHistograms

Expand Down Expand Up @@ -59,7 +60,7 @@ def test_recording_methods(metrics_setup: Tuple[InMemoryMetricReader, Meter]):
metric_reader.get_metrics_data().resource_metrics[0].scope_metrics[0].metrics
)
data_points_sums: Dict[str, HistogramDataPoint] = {
hist.name: next(hist.data.data_points).sum for hist in histogram_metrics
hist.name: next(iter(hist.data.data_points)).sum for hist in histogram_metrics
}
assert data_points_sums == {
'request_time': 10,
Expand Down

0 comments on commit fd37a09

Please sign in to comment.