1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
#include <gtest/gtest.h>
#include <prometheus/metric_family.h>
#include <stddef.h>
#include <chrono>
#include <thread>
#include <vector>
#include "opentelemetry/exporters/prometheus/collector.h"
#include "opentelemetry/metrics/meter_provider.h"
#include "opentelemetry/sdk/metrics/export/metric_producer.h"
#include "opentelemetry/sdk/metrics/instruments.h"
#include "opentelemetry/sdk/metrics/metric_reader.h"
#include "prometheus_test_helper.h"
using opentelemetry::exporter::metrics::PrometheusCollector;
using opentelemetry::sdk::metrics::MetricProducer;
using opentelemetry::sdk::metrics::ResourceMetrics;
namespace metric_api = opentelemetry::metrics;
namespace metric_sdk = opentelemetry::sdk::metrics;
namespace metric_exporter = opentelemetry::exporter::metrics;
class MockMetricProducer : public MetricProducer
{
TestDataPoints test_data_points_;
public:
MockMetricProducer(std::chrono::microseconds sleep_ms = std::chrono::microseconds::zero())
: sleep_ms_{sleep_ms}
{}
MetricProducer::Result Produce() noexcept override
{
std::this_thread::sleep_for(sleep_ms_);
data_sent_size_++;
ResourceMetrics data = test_data_points_.CreateSumPointData();
return {data, MetricProducer::Status::kSuccess};
}
size_t GetDataCount() { return data_sent_size_; }
private:
std::chrono::microseconds sleep_ms_;
size_t data_sent_size_{0};
};
class MockMetricReader : public opentelemetry::sdk::metrics::MetricReader
{
public:
opentelemetry::sdk::metrics::AggregationTemporality GetAggregationTemporality(
opentelemetry::sdk::metrics::InstrumentType /* instrument_type */) const noexcept override
{
// Prometheus exporter only support Cumulative
return opentelemetry::sdk::metrics::AggregationTemporality::kCumulative;
}
private:
bool OnForceFlush(std::chrono::microseconds /* timeout */) noexcept override { return true; }
bool OnShutDown(std::chrono::microseconds /* timeout */) noexcept override { return true; }
void OnInitialized() noexcept override {}
};
// ==================== Test for addMetricsData() function ======================
/**
* AddMetricData() should be able to successfully add a collection
* of SumPointData. It checks that the cumulative
* sum of updates to the aggregator of a record before and after AddMetricData()
* is called are equal.
*/
TEST(PrometheusCollector, BasicTests)
{
MockMetricReader reader;
MockMetricProducer producer;
reader.SetMetricProducer(&producer);
PrometheusCollector collector(&reader, true, false);
auto data = collector.Collect();
// Collection size should be the same as the size
// of the records collection produced by MetricProducer.
ASSERT_EQ(data.size(), 2);
}
|
