Skip to content

Commit d5f7724

Browse files
fedetorres93carrieedwards
authored andcommitted
Add test helper functions
1 parent 83dd84e commit d5f7724

File tree

1 file changed

+64
-0
lines changed

1 file changed

+64
-0
lines changed

receiver/datadogreceiver/testutil.go

Lines changed: 64 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,64 @@
1+
// Copyright The OpenTelemetry Authors
2+
// SPDX-License-Identifier: Apache-2.0
3+
4+
package datadogreceiver // import "github.com/open-telemetry/opentelemetry-collector-contrib/receiver/datadogreceiver"
5+
6+
import (
7+
"testing"
8+
9+
"go.opentelemetry.io/collector/pdata/pcommon"
10+
11+
"github.com/stretchr/testify/require"
12+
"go.opentelemetry.io/collector/component"
13+
"go.opentelemetry.io/collector/pdata/pmetric"
14+
)
15+
16+
func createMetricsTranslator() *MetricsTranslator {
17+
mt := newMetricsTranslator()
18+
mt.buildInfo = component.BuildInfo{
19+
Command: "otelcol",
20+
Description: "OpenTelemetry Collector",
21+
Version: "latest",
22+
}
23+
return mt
24+
}
25+
26+
func requireResourceMetrics(t *testing.T, result pmetric.Metrics, expectedAttrs pcommon.Map, expectedLen int) {
27+
require.Equal(t, expectedLen, result.ResourceMetrics().Len())
28+
require.Equal(t, expectedAttrs, result.ResourceMetrics().At(0).Resource().Attributes())
29+
}
30+
31+
func requireScopeMetrics(t *testing.T, result pmetric.Metrics, expectedScopeMetricsLen, expectedMetricsLen int) {
32+
require.Equal(t, expectedScopeMetricsLen, result.ResourceMetrics().At(0).ScopeMetrics().Len())
33+
require.Equal(t, expectedMetricsLen, result.ResourceMetrics().At(0).ScopeMetrics().At(0).Metrics().Len())
34+
}
35+
36+
func requireScope(t *testing.T, result pmetric.Metrics, expectedAttrs pcommon.Map, expectedName, expectedVersion string) {
37+
require.Equal(t, expectedName, result.ResourceMetrics().At(0).ScopeMetrics().At(0).Scope().Name())
38+
require.Equal(t, expectedVersion, result.ResourceMetrics().At(0).ScopeMetrics().At(0).Scope().Version())
39+
require.Equal(t, expectedAttrs, result.ResourceMetrics().At(0).ScopeMetrics().At(0).Scope().Attributes())
40+
}
41+
42+
func requireMetricAndDataPointCounts(t *testing.T, result pmetric.Metrics, expectedMetricCount, expectedDpCount int) {
43+
require.Equal(t, expectedMetricCount, result.MetricCount())
44+
require.Equal(t, expectedDpCount, result.DataPointCount())
45+
}
46+
47+
func requireSum(t *testing.T, metric pmetric.Metric, expectedName string, expectedAggregationTemporality pmetric.AggregationTemporality, expectedDpsLen int) {
48+
require.Equal(t, expectedName, metric.Name())
49+
require.Equal(t, pmetric.MetricTypeSum, metric.Type())
50+
require.Equal(t, expectedAggregationTemporality, metric.Sum().AggregationTemporality())
51+
require.Equal(t, expectedDpsLen, metric.Sum().DataPoints().Len())
52+
}
53+
54+
func requireGauge(t *testing.T, metric pmetric.Metric, expectedName string, expectedDpsLen int) {
55+
require.Equal(t, expectedName, metric.Name())
56+
require.Equal(t, pmetric.MetricTypeGauge, metric.Type())
57+
require.Equal(t, expectedDpsLen, metric.Gauge().DataPoints().Len())
58+
}
59+
60+
func requireDp(t *testing.T, dp pmetric.NumberDataPoint, expectedAttrs pcommon.Map, expectedTime int64, expectedValue float64) {
61+
require.Equal(t, expectedTime, dp.Timestamp().AsTime().Unix())
62+
require.Equal(t, expectedValue, dp.DoubleValue())
63+
require.Equal(t, expectedAttrs, dp.Attributes())
64+
}

0 commit comments

Comments
 (0)