179 lines
3.3 KiB
Go
179 lines
3.3 KiB
Go
package metadata
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestMerge(t *testing.T) {
|
|
omd := Metadata{
|
|
"key1": "val1",
|
|
}
|
|
mmd := Metadata{
|
|
"key2": "val2",
|
|
}
|
|
|
|
nmd := Merge(omd, mmd, true)
|
|
if len(nmd) != 2 {
|
|
t.Fatalf("merge failed: %v", nmd)
|
|
}
|
|
}
|
|
|
|
func TestIterator(t *testing.T) {
|
|
md := Metadata{
|
|
"1Last": "last",
|
|
"2First": "first",
|
|
"3Second": "second",
|
|
}
|
|
|
|
iter := md.Iterator()
|
|
var k, v string
|
|
|
|
for iter.Next(&k, &v) {
|
|
fmt.Printf("k: %s, v: %s\n", k, v)
|
|
}
|
|
}
|
|
|
|
func TestMedataCanonicalKey(t *testing.T) {
|
|
ctx := Set(context.TODO(), "x-request-id", "12345")
|
|
v, ok := Get(ctx, "x-request-id")
|
|
if !ok {
|
|
t.Fatalf("failed to get x-request-id")
|
|
} else if v != "12345" {
|
|
t.Fatalf("invalid metadata value: %s != %s", "12345", v)
|
|
}
|
|
|
|
v, ok = Get(ctx, "X-Request-Id")
|
|
if !ok {
|
|
t.Fatalf("failed to get x-request-id")
|
|
} else if v != "12345" {
|
|
t.Fatalf("invalid metadata value: %s != %s", "12345", v)
|
|
}
|
|
v, ok = Get(ctx, "X-Request-ID")
|
|
if !ok {
|
|
t.Fatalf("failed to get x-request-id")
|
|
} else if v != "12345" {
|
|
t.Fatalf("invalid metadata value: %s != %s", "12345", v)
|
|
}
|
|
|
|
}
|
|
|
|
func TestMetadataSet(t *testing.T) {
|
|
ctx := Set(context.TODO(), "Key", "val")
|
|
|
|
val, ok := Get(ctx, "Key")
|
|
if !ok {
|
|
t.Fatal("key Key not found")
|
|
}
|
|
if val != "val" {
|
|
t.Errorf("key Key with value val != %v", val)
|
|
}
|
|
}
|
|
|
|
func TestMetadataDelete(t *testing.T) {
|
|
md := Metadata{
|
|
"Foo": "bar",
|
|
"Baz": "empty",
|
|
}
|
|
|
|
ctx := NewContext(context.TODO(), md)
|
|
ctx = Del(ctx, "Baz")
|
|
|
|
emd, ok := FromContext(ctx)
|
|
if !ok {
|
|
t.Fatal("key Key not found")
|
|
}
|
|
|
|
_, ok = emd["Baz"]
|
|
if ok {
|
|
t.Fatal("key Baz not deleted")
|
|
}
|
|
|
|
}
|
|
|
|
func TestNilContext(t *testing.T) {
|
|
var ctx context.Context
|
|
|
|
_, ok := FromContext(ctx)
|
|
if ok {
|
|
t.Fatal("nil context")
|
|
}
|
|
}
|
|
|
|
func TestMetadataCopy(t *testing.T) {
|
|
md := Metadata{
|
|
"Foo": "bar",
|
|
"Bar": "baz",
|
|
}
|
|
|
|
cp := Copy(md)
|
|
|
|
for k, v := range md {
|
|
if cv := cp[k]; cv != v {
|
|
t.Fatalf("Got %s:%s for %s:%s", k, cv, k, v)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMetadataContext(t *testing.T) {
|
|
md := Metadata{
|
|
"Foo": "bar",
|
|
}
|
|
|
|
ctx := NewContext(context.TODO(), md)
|
|
|
|
emd, ok := FromContext(ctx)
|
|
if !ok {
|
|
t.Errorf("Unexpected error retrieving metadata, got %t", ok)
|
|
}
|
|
|
|
if emd["Foo"] != md["Foo"] {
|
|
t.Errorf("Expected key: %s val: %s, got key: %s val: %s", "Foo", md["Foo"], "Foo", emd["Foo"])
|
|
}
|
|
|
|
if i := len(emd); i != 1 {
|
|
t.Errorf("Expected metadata length 1 got %d", i)
|
|
}
|
|
}
|
|
|
|
func TestMergeContext(t *testing.T) {
|
|
type args struct {
|
|
existing Metadata
|
|
append Metadata
|
|
overwrite bool
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want Metadata
|
|
}{
|
|
{
|
|
name: "matching key, overwrite false",
|
|
args: args{
|
|
existing: Metadata{"Foo": "bar", "Sumo": "demo"},
|
|
append: Metadata{"Sumo": "demo2"},
|
|
overwrite: false,
|
|
},
|
|
want: Metadata{"Foo": "bar", "Sumo": "demo"},
|
|
},
|
|
{
|
|
name: "matching key, overwrite true",
|
|
args: args{
|
|
existing: Metadata{"Foo": "bar", "Sumo": "demo"},
|
|
append: Metadata{"Sumo": "demo2"},
|
|
overwrite: true,
|
|
},
|
|
want: Metadata{"Foo": "bar", "Sumo": "demo2"},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if got, _ := FromContext(MergeContext(NewContext(context.TODO(), tt.args.existing), tt.args.append, tt.args.overwrite)); !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("MergeContext() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|