File: signal_test.go

package info (click to toggle)
golang-opentelemetry-contrib 0.56.0-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 4,884 kB
  • sloc: makefile: 278; sh: 211; sed: 1
file content (52 lines) | stat: -rw-r--r-- 1,793 bytes parent folder | download
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
// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

package autoexport // import "go.opentelemetry.io/contrib/exporters/autoexport"

import (
	"context"
	"errors"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestOTLPExporterReturnedWhenNoEnvOrFallbackExporterConfigured(t *testing.T) {
	ts := newSignal[*testType]("TEST_TYPE_KEY")
	assert.NoError(t, ts.registry.store("otlp", factory("test-otlp-exporter")))
	exp, err := ts.create(context.Background())
	assert.NoError(t, err)
	assert.Equal(t, exp.string, "test-otlp-exporter")
}

func TestFallbackExporterReturnedWhenNoEnvExporterConfigured(t *testing.T) {
	ts := newSignal[*testType]("TEST_TYPE_KEY")
	exp, err := ts.create(context.Background(), withFallbackFactory(factory("test-fallback-exporter")))
	assert.NoError(t, err)
	assert.Equal(t, exp.string, "test-fallback-exporter")
}

func TestFallbackExporterFactoryErrorReturnedWhenNoEnvExporterConfiguredAndFallbackFactoryReturnsAnError(t *testing.T) {
	ts := newSignal[*testType]("TEST_TYPE_KEY")

	expectedErr := errors.New("error expected to return")
	errFactory := func(ctx context.Context) (*testType, error) {
		return nil, expectedErr
	}
	exp, err := ts.create(context.Background(), withFallbackFactory(errFactory))
	assert.ErrorIs(t, err, expectedErr)
	assert.Nil(t, exp)
}

func TestEnvExporterIsPreferredOverFallbackExporter(t *testing.T) {
	envVariable := "TEST_TYPE_KEY"
	ts := newSignal[*testType](envVariable)

	expName := "test-env-exporter-name"
	t.Setenv(envVariable, expName)
	assert.NoError(t, ts.registry.store(expName, factory("test-env-exporter")))

	exp, err := ts.create(context.Background(), withFallbackFactory(factory("test-fallback-exporter")))
	assert.NoError(t, err)
	assert.Equal(t, exp.string, "test-env-exporter")
}