mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-19 00:04:12 +00:00
91bd477f4f
* checkpoint changes * gaz * add error check * gaz * fix tests * fix tests * terence's review * terence's review Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
57 lines
2.2 KiB
Go
57 lines
2.2 KiB
Go
package grpcutils
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
"google.golang.org/grpc/metadata"
|
|
)
|
|
|
|
func TestAppendHeaders(t *testing.T) {
|
|
t.Run("One header", func(t *testing.T) {
|
|
ctx := AppendHeaders(context.Background(), []string{"first=value1"})
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.Equal(t, true, ok, "Failed to read context metadata")
|
|
require.Equal(t, 1, md.Len(), "MetadataV0 contains wrong number of values")
|
|
assert.Equal(t, "value1", md.Get("first")[0])
|
|
})
|
|
|
|
t.Run("Multiple headers", func(t *testing.T) {
|
|
ctx := AppendHeaders(context.Background(), []string{"first=value1", "second=value2"})
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.Equal(t, true, ok, "Failed to read context metadata")
|
|
require.Equal(t, 2, md.Len(), "MetadataV0 contains wrong number of values")
|
|
assert.Equal(t, "value1", md.Get("first")[0])
|
|
assert.Equal(t, "value2", md.Get("second")[0])
|
|
})
|
|
|
|
t.Run("One empty header", func(t *testing.T) {
|
|
ctx := AppendHeaders(context.Background(), []string{"first=value1", ""})
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.Equal(t, true, ok, "Failed to read context metadata")
|
|
require.Equal(t, 1, md.Len(), "MetadataV0 contains wrong number of values")
|
|
assert.Equal(t, "value1", md.Get("first")[0])
|
|
})
|
|
|
|
t.Run("Incorrect header", func(t *testing.T) {
|
|
logHook := logTest.NewGlobal()
|
|
ctx := AppendHeaders(context.Background(), []string{"first=value1", "second"})
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.Equal(t, true, ok, "Failed to read context metadata")
|
|
require.Equal(t, 1, md.Len(), "MetadataV0 contains wrong number of values")
|
|
assert.Equal(t, "value1", md.Get("first")[0])
|
|
assert.LogsContain(t, logHook, "Skipping second")
|
|
})
|
|
|
|
t.Run("Header value with equal sign", func(t *testing.T) {
|
|
ctx := AppendHeaders(context.Background(), []string{"first=value=1"})
|
|
md, ok := metadata.FromOutgoingContext(ctx)
|
|
require.Equal(t, true, ok, "Failed to read context metadata")
|
|
require.Equal(t, 1, md.Len(), "MetadataV0 contains wrong number of values")
|
|
assert.Equal(t, "value=1", md.Get("first")[0])
|
|
})
|
|
}
|