Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

feat: Add OTEL tracer to metricsstore reader component #4595

Merged
merged 13 commits into from Jul 29, 2023
Merged

feat: Add OTEL tracer to metricsstore reader component #4595

Changes from 1 commit
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
1faf91e
adds otel tracer to metrics reader component
afzal442 Jul 23, 2023
faaecd4
updates the tracer to use global tp
afzal442 Jul 24, 2023
ec7d45d
adds condition to jt closer
afzal442 Jul 25, 2023
ffcc85e
adds tp to main and creates tp in test
afzal442 Jul 25, 2023
4852cec
adds exporter.GetSpan() to test recorded spans
afzal442 Jul 25, 2023
a6b2884
fixes the span creation for query
afzal442 Jul 26, 2023
09b3fbf
fixes the shutdown issue
afzal442 Jul 26, 2023
a0dcb35
validates the no. of spans reported
afzal442 Jul 27, 2023
e279905
reorders the defer to execute the close
afzal442 Jul 27, 2023
4319e62
fixes spans reported err
afzal442 Jul 27, 2023
3295970
moves defer to its obj creation
afzal442 Jul 27, 2023
abeeeb6
reorders the assert in the sequence call
afzal442 Jul 28, 2023
7d4624b
sets the span status
afzal442 Jul 28, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
33 changes: 19 additions & 14 deletions plugin/metrics/prometheus/metricsstore/reader_test.go
View file
Open in desktop
Expand Up @@ -29,8 +29,8 @@ import (

"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
stdout "go.opentelemetry.io/otel/exporters/stdout/stdouttrace"
sdktrace "go.opentelemetry.io/otel/sdk/trace"
"go.opentelemetry.io/otel/sdk/trace/tracetest"
"go.opentelemetry.io/otel/trace"
"go.uber.org/zap"

Expand Down Expand Up @@ -64,36 +64,36 @@ var defaultConfig = config.Configuration{
LatencyUnit: "ms",
}

func createTracingProvider() trace.TracerProvider {
exporter, err := stdout.New(stdout.WithPrettyPrint())
if err != nil {
panic(err)
}
func tracerProvider(t *testing.T) (trace.TracerProvider, *tracetest.InMemoryExporter) {
exporter := tracetest.NewInMemoryExporter()
tp := sdktrace.NewTracerProvider(
sdktrace.WithSampler(sdktrace.AlwaysSample()),
sdktrace.WithBatcher(exporter),
)
return tp
defer tp.Shutdown(context.Background())
afzal442 marked this conversation as resolved.
Show resolved Hide resolved
return tp, exporter
}

func TestNewMetricsReaderValidAddress(t *testing.T) {
logger := zap.NewNop()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
reader, err := NewMetricsReader(config.Configuration{
ServerURL: "http://localhost:1234",
ConnectTimeout: defaultTimeout,
}, logger, tracer)
assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
require.NoError(t, err)
assert.NotNil(t, reader)
}

func TestNewMetricsReaderInvalidAddress(t *testing.T) {
logger := zap.NewNop()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
reader, err := NewMetricsReader(config.Configuration{
ServerURL: "\n",
ConnectTimeout: defaultTimeout,
}, logger, tracer)
assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
require.Error(t, err)
assert.Contains(t, err.Error(), "failed to initialize prometheus client")
assert.Nil(t, reader)
Expand All @@ -102,7 +102,7 @@ func TestNewMetricsReaderInvalidAddress(t *testing.T) {
func TestGetMinStepDuration(t *testing.T) {
params := metricsstore.MinStepDurationQueryParameters{}
logger := zap.NewNop()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
listener, err := net.Listen("tcp", "localhost:")
require.NoError(t, err)
assert.NotNil(t, listener)
Expand All @@ -111,6 +111,7 @@ func TestGetMinStepDuration(t *testing.T) {
ServerURL: "http://" + listener.Addr().String(),
ConnectTimeout: defaultTimeout,
}, logger, tracer)
afzal442 marked this conversation as resolved.
Show resolved Hide resolved
assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
afzal442 marked this conversation as resolved.
Show resolved Hide resolved
require.NoError(t, err)

minStep, err := reader.GetMinStepDuration(context.Background(), &params)
Expand Down Expand Up @@ -139,12 +140,13 @@ func TestMetricsServerError(t *testing.T) {
defer mockPrometheus.Close()

logger := zap.NewNop()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
address := mockPrometheus.Listener.Addr().String()
reader, err := NewMetricsReader(config.Configuration{
ServerURL: "http://" + address,
ConnectTimeout: defaultTimeout,
}, logger, tracer)
assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
require.NoError(t, err)

m, err := reader.GetCallRates(context.Background(), &params)
Expand Down Expand Up @@ -479,12 +481,13 @@ func TestInvalidLatencyUnit(t *testing.T) {
t.Errorf("Expected a panic due to invalid latency unit")
}
}()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
cfg := config.Configuration{
SupportSpanmetricsConnector: true,
NormalizeDuration: true,
LatencyUnit: "something invalid",
}
assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
_, _ = NewMetricsReader(cfg, zap.NewNop(), tracer)
}

Expand Down Expand Up @@ -591,7 +594,7 @@ func TestGetRoundTripperTokenError(t *testing.T) {

func TestInvalidCertFile(t *testing.T) {
logger := zap.NewNop()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
reader, err := NewMetricsReader(config.Configuration{
ServerURL: "https://localhost:1234",
ConnectTimeout: defaultTimeout,
Expand All @@ -600,6 +603,7 @@ func TestInvalidCertFile(t *testing.T) {
CAPath: "foo",
},
}, logger, tracer)
assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
require.Error(t, err)
assert.Nil(t, reader)
}
Expand Down Expand Up @@ -660,12 +664,13 @@ func prepareMetricsReaderAndServer(t *testing.T, config config.Configuration, wa
mockPrometheus := startMockPrometheusServer(t, wantPromQlQuery, wantWarnings)

logger := zap.NewNop()
tracer := createTracingProvider()
tracer, exp := tracerProvider(t)
address := mockPrometheus.Listener.Addr().String()

config.ServerURL = "http://" + address
config.ConnectTimeout = defaultTimeout

assert.NotEmpty(t, exp.GetSpans(), "No spans recorded during the test.")
reader, err := NewMetricsReader(config, logger, tracer)
require.NoError(t, err)

Expand Down