98 lines
2.1 KiB
Go
98 lines
2.1 KiB
Go
// Copyright The OpenTelemetry Authors
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Package client contains generic representations of clients connecting to
|
|
// different receivers
|
|
package client
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewContext(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
cl Info
|
|
}{
|
|
{
|
|
name: "valid client",
|
|
cl: Info{
|
|
Addr: &net.IPAddr{
|
|
IP: net.IPv4(1, 2, 3, 4),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "nil client",
|
|
cl: Info{},
|
|
},
|
|
}
|
|
for _, tt := range testCases {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ctx := NewContext(context.Background(), tt.cl)
|
|
assert.Equal(t, ctx.Value(ctxKey{}), tt.cl)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFromContext(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
input context.Context
|
|
expected Info
|
|
}{
|
|
{
|
|
name: "context with client",
|
|
input: context.WithValue(context.Background(), ctxKey{}, Info{
|
|
Addr: &net.IPAddr{
|
|
IP: net.IPv4(1, 2, 3, 4),
|
|
},
|
|
}),
|
|
expected: Info{
|
|
Addr: &net.IPAddr{
|
|
IP: net.IPv4(1, 2, 3, 4),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
name: "context without client",
|
|
input: context.Background(),
|
|
expected: Info{},
|
|
},
|
|
{
|
|
name: "context with something else in the key",
|
|
input: context.WithValue(context.Background(), ctxKey{}, "unexpected!"),
|
|
expected: Info{},
|
|
},
|
|
}
|
|
for _, tt := range testCases {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
assert.Equal(t, tt.expected, FromContext(tt.input))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestMetadata(t *testing.T) {
|
|
source := map[string][]string{"test-key": {"test-val"}, "TEST-KEY-2": {"test-val"}}
|
|
md := NewMetadata(source)
|
|
assert.Equal(t, []string{"test-val"}, md.Get("test-key"))
|
|
assert.Equal(t, []string{"test-val"}, md.Get("test-KEY")) // case insensitive lookup
|
|
assert.Equal(t, []string{"test-val"}, md.Get("test-key-2")) // case insensitive lookup
|
|
|
|
// test if copy. In regular use, source cannot change
|
|
val := md.Get("test-key")
|
|
source["test-key"][0] = "abc"
|
|
assert.Equal(t, []string{"test-val"}, val)
|
|
|
|
assert.Empty(t, md.Get("non-existent-key"))
|
|
}
|
|
|
|
func TestUninstantiatedMetadata(t *testing.T) {
|
|
i := Info{}
|
|
assert.Empty(t, i.Metadata.Get("test"))
|
|
}
|