|
4 | 4 | package loadbalancingexporter |
5 | 5 |
|
6 | 6 | import ( |
| 7 | + "context" |
7 | 8 | "path/filepath" |
8 | 9 | "testing" |
9 | 10 |
|
10 | 11 | "github.com/stretchr/testify/assert" |
11 | 12 | "github.com/stretchr/testify/require" |
12 | 13 | "go.opentelemetry.io/collector/component" |
| 14 | + "go.opentelemetry.io/collector/component/componenttest" |
13 | 15 | "go.opentelemetry.io/collector/config/configoptional" |
14 | 16 | "go.opentelemetry.io/collector/config/configretry" |
15 | 17 | "go.opentelemetry.io/collector/exporter/exporterhelper" |
16 | 18 | "go.opentelemetry.io/collector/exporter/exportertest" |
17 | 19 | "go.opentelemetry.io/collector/exporter/otlpexporter" |
18 | 20 | "go.opentelemetry.io/collector/otelcol/otelcoltest" |
| 21 | + metricnoop "go.opentelemetry.io/otel/metric/noop" |
| 22 | + tracenoop "go.opentelemetry.io/otel/trace/noop" |
19 | 23 | "go.uber.org/zap" |
20 | 24 | "go.uber.org/zap/zaptest/observer" |
21 | 25 |
|
@@ -98,24 +102,75 @@ func TestBuildExporterConfig(t *testing.T) { |
98 | 102 | } |
99 | 103 |
|
100 | 104 | func TestBuildExporterSettings(t *testing.T) { |
101 | | - // prepare |
102 | | - creationParams := exportertest.NewNopSettings(metadata.Type) |
103 | | - testEndpoint := "the-endpoint" |
104 | | - observedZapCore, observedLogs := observer.New(zap.InfoLevel) |
105 | | - creationParams.Logger = zap.New(observedZapCore) |
106 | | - |
107 | | - // test |
108 | | - exporterParams := buildExporterSettings(creationParams, testEndpoint) |
109 | | - exporterParams.Logger.Info("test") |
110 | | - |
111 | | - assert.Equal(t, creationParams.ID, exporterParams.ID) |
| 105 | + otlpType := otlpexporter.NewFactory().Type() |
| 106 | + |
| 107 | + t.Run("without exporter name", func(t *testing.T) { |
| 108 | + ctx := context.Background() //nolint:usetesting // Context must outlive test for cleanup |
| 109 | + creationParams := exportertest.NewNopSettings(metadata.Type) |
| 110 | + creationParams.ID = component.NewID(metadata.Type) |
| 111 | + telemetry := componenttest.NewTelemetry() |
| 112 | + t.Cleanup(func() { |
| 113 | + require.NoError(t, telemetry.Shutdown(ctx)) |
| 114 | + }) |
| 115 | + creationParams.TelemetrySettings = telemetry.NewTelemetrySettings() |
| 116 | + originalTelemetry := creationParams.TelemetrySettings |
| 117 | + testEndpoint := "the-endpoint" |
| 118 | + observedZapCore, observedLogs := observer.New(zap.InfoLevel) |
| 119 | + creationParams.Logger = zap.New(observedZapCore) |
| 120 | + originalLogger := creationParams.Logger |
| 121 | + |
| 122 | + exporterParams := buildExporterSettings(otlpType, creationParams, testEndpoint) |
| 123 | + exporterParams.Logger.Info("test") |
| 124 | + |
| 125 | + assert.Equal(t, component.NewID(otlpType), exporterParams.ID) |
| 126 | + assert.IsType(t, metricnoop.NewMeterProvider(), exporterParams.MeterProvider) |
| 127 | + assert.IsType(t, tracenoop.NewTracerProvider(), exporterParams.TracerProvider) |
| 128 | + |
| 129 | + assert.Same(t, originalTelemetry.MeterProvider, creationParams.MeterProvider) |
| 130 | + assert.Same(t, originalTelemetry.TracerProvider, creationParams.TracerProvider) |
| 131 | + assert.Same(t, originalLogger, creationParams.Logger) |
| 132 | + |
| 133 | + allLogs := observedLogs.All() |
| 134 | + require.Equal(t, 1, observedLogs.Len()) |
| 135 | + assert.Contains(t, |
| 136 | + allLogs[0].Context, |
| 137 | + zap.String(zapEndpointKey, testEndpoint), |
| 138 | + ) |
| 139 | + }) |
112 | 140 |
|
113 | | - allLogs := observedLogs.All() |
114 | | - require.Equal(t, 1, observedLogs.Len()) |
115 | | - assert.Contains(t, |
116 | | - allLogs[0].Context, |
117 | | - zap.String(zapEndpointKey, testEndpoint), |
118 | | - ) |
| 141 | + t.Run("with exporter name", func(t *testing.T) { |
| 142 | + ctx := context.Background() //nolint:usetesting // Context must outlive test for cleanup |
| 143 | + creationParams := exportertest.NewNopSettings(metadata.Type) |
| 144 | + creationParams.ID = component.NewIDWithName(metadata.Type, "custom") |
| 145 | + telemetry := componenttest.NewTelemetry() |
| 146 | + t.Cleanup(func() { |
| 147 | + require.NoError(t, telemetry.Shutdown(ctx)) |
| 148 | + }) |
| 149 | + creationParams.TelemetrySettings = telemetry.NewTelemetrySettings() |
| 150 | + originalTelemetry := creationParams.TelemetrySettings |
| 151 | + testEndpoint := "the-endpoint" |
| 152 | + observedZapCore, observedLogs := observer.New(zap.InfoLevel) |
| 153 | + creationParams.Logger = zap.New(observedZapCore) |
| 154 | + originalLogger := creationParams.Logger |
| 155 | + |
| 156 | + exporterParams := buildExporterSettings(otlpType, creationParams, testEndpoint) |
| 157 | + exporterParams.Logger.Info("test") |
| 158 | + |
| 159 | + assert.Equal(t, component.NewIDWithName(otlpType, "custom"), exporterParams.ID) |
| 160 | + assert.IsType(t, metricnoop.NewMeterProvider(), exporterParams.MeterProvider) |
| 161 | + assert.IsType(t, tracenoop.NewTracerProvider(), exporterParams.TracerProvider) |
| 162 | + |
| 163 | + assert.Same(t, originalTelemetry.MeterProvider, creationParams.MeterProvider) |
| 164 | + assert.Same(t, originalTelemetry.TracerProvider, creationParams.TracerProvider) |
| 165 | + assert.Same(t, originalLogger, creationParams.Logger) |
| 166 | + |
| 167 | + allLogs := observedLogs.All() |
| 168 | + require.Equal(t, 1, observedLogs.Len()) |
| 169 | + assert.Contains(t, |
| 170 | + allLogs[0].Context, |
| 171 | + zap.String(zapEndpointKey, testEndpoint), |
| 172 | + ) |
| 173 | + }) |
119 | 174 | } |
120 | 175 |
|
121 | 176 | func TestWrappedExporterHasEndpointAttribute(t *testing.T) { |
|
0 commit comments