mirror of
https://github.com/hanzoai/telemetry.git
synced 2026-10-30 20:01:49 +00:00
Joshua MacDonaldandGitHub
593537a94c
#### Description API change in extensionmiddleware, an unstable package to allow middleware access to the start context. HTTP only. gRPC is treated in #14536. This is technically a breaking change in configmiddleware and for all implementations of this extension. We believe no one is using this feature outside of this repository, see memorylimiter is fixed here. #### Link to tracking issue Part of https://github.com/open-telemetry/opentelemetry-collector/issues/14523 #### Documentation N/A
94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package extensionmiddleware
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc"
|
|
)
|
|
|
|
func TestGetHTTPRoundTripperFunc(t *testing.T) {
|
|
// Create a base round tripper for testing
|
|
baseRT := http.DefaultTransport
|
|
testctx := context.Background()
|
|
|
|
t.Run("nil function", func(t *testing.T) {
|
|
var nilFunc GetHTTPRoundTripperFunc
|
|
rtfunc, err := nilFunc.GetHTTPRoundTripper(testctx)
|
|
require.NoError(t, err)
|
|
|
|
rt, err := rtfunc(testctx, baseRT)
|
|
require.NoError(t, err)
|
|
require.Equal(t, baseRT, rt)
|
|
})
|
|
|
|
t.Run("identity function", func(t *testing.T) {
|
|
identityFunc := GetHTTPRoundTripperFunc(nil)
|
|
rtfunc, err := identityFunc.GetHTTPRoundTripper(testctx)
|
|
require.NoError(t, err)
|
|
rt, err := rtfunc(testctx, baseRT)
|
|
require.NoError(t, err)
|
|
require.Equal(t, baseRT, rt)
|
|
})
|
|
|
|
t.Run("error function", func(t *testing.T) {
|
|
expectedErr := errors.New("round tripper error")
|
|
errorFunc := GetHTTPRoundTripperFunc(func(_ context.Context) (WrapHTTPRoundTripperFunc, error) {
|
|
return nil, expectedErr
|
|
})
|
|
rtfunc, err := errorFunc.GetHTTPRoundTripper(testctx)
|
|
require.Error(t, err)
|
|
require.Equal(t, expectedErr, err)
|
|
require.Nil(t, rtfunc)
|
|
})
|
|
}
|
|
|
|
func TestGetGRPCClientOptionsFunc(t *testing.T) {
|
|
type testCtx struct{}
|
|
var (
|
|
key = testCtx{}
|
|
value = "testval"
|
|
)
|
|
testctx := context.WithValue(context.Background(), key, value)
|
|
|
|
t.Run("nil function", func(t *testing.T) {
|
|
var nilFunc GetGRPCClientOptionsFunc
|
|
options, err := nilFunc.GetGRPCClientOptions(testctx)
|
|
require.NoError(t, err)
|
|
require.Nil(t, options)
|
|
})
|
|
|
|
t.Run("options function", func(t *testing.T) {
|
|
dialOpt1 := grpc.WithAuthority("test-authority")
|
|
dialOpt2 := grpc.WithDisableRetry()
|
|
|
|
optionsFunc := GetGRPCClientOptionsFunc(func(ctx context.Context) ([]grpc.DialOption, error) {
|
|
require.Equal(t, ctx.Value(key), value)
|
|
return []grpc.DialOption{dialOpt1, dialOpt2}, nil
|
|
})
|
|
|
|
options, err := optionsFunc.GetGRPCClientOptions(testctx)
|
|
require.NoError(t, err)
|
|
require.Len(t, options, 2)
|
|
})
|
|
|
|
t.Run("error function", func(t *testing.T) {
|
|
expectedErr := errors.New("grpc options error")
|
|
errorFunc := GetGRPCClientOptionsFunc(func(ctx context.Context) ([]grpc.DialOption, error) {
|
|
require.Equal(t, ctx.Value(key), value)
|
|
return nil, expectedErr
|
|
})
|
|
|
|
options, err := errorFunc.GetGRPCClientOptions(testctx)
|
|
require.Error(t, err)
|
|
require.Equal(t, expectedErr, err)
|
|
require.Nil(t, options)
|
|
})
|
|
}
|