|
4 | 4 | package solarwindsapmsettingsextension
|
5 | 5 |
|
6 | 6 | import (
|
7 |
| - "os" |
8 | 7 | "path/filepath"
|
9 | 8 | "testing"
|
10 | 9 | "time"
|
@@ -275,21 +274,26 @@ func TestLoadConfig(t *testing.T) {
|
275 | 274 | }
|
276 | 275 | }
|
277 | 276 |
|
278 |
| -func TestResolveServiceNameBestEffort(t *testing.T) { |
| 277 | +func TestResolveServiceNameBestEffortNoEnv(t *testing.T) { |
279 | 278 | // Without any environment variables
|
280 | 279 | require.Empty(t, resolveServiceNameBestEffort())
|
281 |
| - // With OTEL_SERVICE_NAME only |
282 |
| - require.NoError(t, os.Setenv("OTEL_SERVICE_NAME", "otel_ser1")) |
| 280 | +} |
| 281 | + |
| 282 | +// With OTEL_SERVICE_NAME only |
| 283 | +func TestResolveServiceNameBestEffortOnlyOtelService(t *testing.T) { |
| 284 | + t.Setenv("OTEL_SERVICE_NAME", "otel_ser1") |
283 | 285 | require.Equal(t, "otel_ser1", resolveServiceNameBestEffort())
|
284 |
| - require.NoError(t, os.Unsetenv("OTEL_SERVICE_NAME")) |
285 |
| - // With AWS_LAMBDA_FUNCTION_NAME only |
286 |
| - require.NoError(t, os.Setenv("AWS_LAMBDA_FUNCTION_NAME", "lambda")) |
| 286 | +} |
| 287 | + |
| 288 | +// With AWS_LAMBDA_FUNCTION_NAME only |
| 289 | +func TestResolveServiceNameBestEffortOnlyAwsLambda(t *testing.T) { |
| 290 | + t.Setenv("AWS_LAMBDA_FUNCTION_NAME", "lambda") |
287 | 291 | require.Equal(t, "lambda", resolveServiceNameBestEffort())
|
288 |
| - require.NoError(t, os.Unsetenv("AWS_LAMBDA_FUNCTION_NAME")) |
289 |
| - // With both |
290 |
| - require.NoError(t, os.Setenv("OTEL_SERVICE_NAME", "otel_ser1")) |
291 |
| - require.NoError(t, os.Setenv("AWS_LAMBDA_FUNCTION_NAME", "lambda")) |
| 292 | +} |
| 293 | + |
| 294 | +// With both |
| 295 | +func TestResolveServiceNameBestEffortBoth(t *testing.T) { |
| 296 | + t.Setenv("OTEL_SERVICE_NAME", "otel_ser1") |
| 297 | + t.Setenv("AWS_LAMBDA_FUNCTION_NAME", "lambda") |
292 | 298 | require.Equal(t, "otel_ser1", resolveServiceNameBestEffort())
|
293 |
| - require.NoError(t, os.Unsetenv("AWS_LAMBDA_FUNCTION_NAME")) |
294 |
| - require.NoError(t, os.Unsetenv("OTEL_SERVICE_NAME")) |
295 | 299 | }
|
0 commit comments