forked from open-telemetry/opentelemetry-collector-contrib
-
Notifications
You must be signed in to change notification settings - Fork 0
/
integration_test.go
109 lines (97 loc) · 3.96 KB
/
integration_test.go
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0
//go:build integration
package snmpreceiver
import (
"context"
"path/filepath"
"testing"
"time"
"github.com/stretchr/testify/require"
"github.com/testcontainers/testcontainers-go"
"go.opentelemetry.io/collector/component"
"go.opentelemetry.io/collector/component/componenttest"
"go.opentelemetry.io/collector/consumer/consumertest"
"go.opentelemetry.io/collector/otelcol/otelcoltest"
"go.opentelemetry.io/collector/receiver/receivertest"
"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/golden"
"github.com/open-telemetry/opentelemetry-collector-contrib/pkg/pdatatest/pmetrictest"
"github.com/open-telemetry/opentelemetry-collector-contrib/receiver/snmpreceiver/internal/metadata"
)
func TestIntegration(t *testing.T) {
// remove nolint when https://github.com/open-telemetry/opentelemetry-collector-contrib/issues/24240 is resolved
// nolint:staticcheck
testCases := []struct {
desc string
configFilename string
expectedResultsFilename string
}{
{
desc: "Integration test with v2c configuration",
configFilename: "integration_test_v2c_config.yaml",
expectedResultsFilename: "v2c_config_expected_metrics.yaml",
},
{
desc: "Integration test with v3 configuration",
configFilename: "integration_test_v3_config.yaml",
expectedResultsFilename: "v3_config_expected_metrics.yaml",
},
}
container := getContainer(t, snmpAgentContainerRequest)
defer func() {
require.NoError(t, container.Terminate(context.Background()))
}()
_, err := container.Host(context.Background())
require.NoError(t, err)
factories, err := otelcoltest.NopFactories()
require.NoError(t, err)
for _, testCase := range testCases {
t.Run(testCase.desc, func(t *testing.T) {
t.Skip("Flaky test, see https://github.com/open-telemetry/opentelemetry-collector-contrib/issues/21086")
factory := NewFactory()
factories.Receivers[metadata.Type] = factory
configFile := filepath.Join("testdata", "integration", testCase.configFilename)
// https://github.com/open-telemetry/opentelemetry-collector-contrib/issues/33594
// nolint:staticcheck
cfg, err := otelcoltest.LoadConfigAndValidate(configFile, factories)
require.NoError(t, err)
snmpConfig := cfg.Receivers[component.NewID(metadata.Type)].(*Config)
consumer := new(consumertest.MetricsSink)
settings := receivertest.NewNopSettings()
rcvr, err := factory.CreateMetricsReceiver(context.Background(), settings, snmpConfig, consumer)
require.NoError(t, err, "failed creating metrics receiver")
require.NoError(t, rcvr.Start(context.Background(), componenttest.NewNopHost()))
require.Eventuallyf(t, func() bool {
return len(consumer.AllMetrics()) > 0
}, 2*time.Minute, 1*time.Second, "failed to receive more than 0 metrics")
require.NoError(t, rcvr.Shutdown(context.Background()))
actualMetrics := consumer.AllMetrics()[0]
expectedFile := filepath.Join("testdata", "integration", testCase.expectedResultsFilename)
expectedMetrics, err := golden.ReadMetrics(expectedFile)
require.NoError(t, err)
err = pmetrictest.CompareMetrics(expectedMetrics, actualMetrics, pmetrictest.IgnoreMetricsOrder(),
pmetrictest.IgnoreTimestamp(), pmetrictest.IgnoreStartTimestamp())
require.NoError(t, err)
})
}
}
var (
snmpAgentContainerRequest = testcontainers.ContainerRequest{
FromDockerfile: testcontainers.FromDockerfile{
Context: filepath.Join("testdata", "integration", "docker"),
Dockerfile: "snmp_agent.Dockerfile",
},
ExposedPorts: []string{"1024:1024/udp"},
}
)
func getContainer(t *testing.T, req testcontainers.ContainerRequest) testcontainers.Container {
require.NoError(t, req.Validate())
container, err := testcontainers.GenericContainer(
context.Background(),
testcontainers.GenericContainerRequest{
ContainerRequest: req,
Started: true,
})
require.NoError(t, err)
return container
}