Function TestTracePushMiddlewareCalled

modules/distributor/distributor_test.go:2113–2156

(t *testing.T)

}

func TestTracePushMiddlewareCalled(t *testing.T) {
	limits := overrides.Config{}
	limitCfg := &flag.FlagSet{}
	limits.RegisterFlagsAndApplyDefaults(limitCfg)

	distributorCfg, overridesSvc, loggingLevel, middleware := setupDependencies(t, limits)

	// Track middleware calls
	var middlewareCalled bool
	var receivedTraces ptrace.Traces

	// Create a test middleware
	testMiddleware := func(_ context.Context, td ptrace.Traces) error {
		middlewareCalled = true
		receivedTraces = td
		return nil
	}

	distributorCfg.TracePushMiddlewares = []TracePushMiddleware{testMiddleware}

	d, err := New(
		distributorCfg,
		LocalPushTargets{},
		nil,
		overridesSvc,
		middleware,
		kitlog.NewLogfmtLogger(os.Stdout),
		loggingLevel,
		prometheus.NewRegistry(),
	)
	require.NoError(t, err)

	// Create test traces
	traces := batchesToTraces(t, []*v1.ResourceSpans{test.MakeBatch(10, nil)})

	// Call PushTraces
	_, err = d.PushTraces(ctx, traces)
	require.NoError(t, err)

	// Verify middleware was called
	assert.True(t, middlewareCalled, "TracePushMiddleware should have been called")
	assert.NotNil(t, receivedTraces, "Middleware should have received traces")
	assert.Equal(t, traces.SpanCount(), receivedTraces.SpanCount(), "Middleware should receive the same traces")
}

func TestTracePushMiddlewareFailsOpen(t *testing.T) {
	limits := overrides.Config{}