go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-03-17 21:07:54 +02:00
Code Issues Releases Activity

add: metadata test (#1324)

Browse Source
This commit is contained in:
月墨夕 2021-08-12 19:34:58 +08:00 committed by GitHub
parent a128566cef
commit 55cf83062f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 240 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

240
metadata/metadata_test.go Normal file
View File

@ -0,0 +1,240 @@
package metadata
import (
"context"
"reflect"
"testing"
)
func TestNew(t *testing.T) {
type args struct {
mds []map[string]string
}
tests := []struct {
name string
args args
want Metadata
}{
{
name: "hello",
args: args{[]map[string]string{{"hello": "kratos"}, {"hello2": "go-kratos"}}},
want: Metadata{"hello": "kratos", "hello2": "go-kratos"},
},
{
name: "hi",
args: args{[]map[string]string{{"hi": "kratos"}, {"hi2": "go-kratos"}}},
want: Metadata{"hi": "kratos", "hi2": "go-kratos"},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := New(tt.args.mds...); !reflect.DeepEqual(got, tt.want) {
t.Errorf("New() = %v, want %v", got, tt.want)
}
})
}
}
func TestMetadata_Get(t *testing.T) {
type args struct {
key string
}
tests := []struct {
name string
m Metadata
args args
want string
}{
{
name: "kratos",
m: Metadata{"kratos": "value", "env": "dev"},
args: args{key: "kratos"},
want: "value",
},
{
name: "env",
m: Metadata{"kratos": "value", "env": "dev"},
args: args{key: "env"},
want: "dev",
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if got := tt.m.Get(tt.args.key); got != tt.want {
t.Errorf("Get() = %v, want %v", got, tt.want)
}
})
}
}
func TestMetadata_Set(t *testing.T) {
type args struct {
key string
value string
}
tests := []struct {
name string
m Metadata
args args
want Metadata
}{
{
name: "kratos",
m: Metadata{},
args: args{key: "hello", value: "kratos"},
want: Metadata{"hello": "kratos"},
},
{
name: "env",
m: Metadata{"hello": "kratos"},
args: args{key: "env", value: "pro"},
want: Metadata{"hello": "kratos", "env": "pro"},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
tt.m.Set(tt.args.key, tt.args.value)
if !reflect.DeepEqual(tt.m, tt.want) {
t.Errorf("Set() = %v, want %v", tt.m, tt.want)
}
})
}
}
func TestClientContext(t *testing.T) {
type args struct {
ctx context.Context
md Metadata
}
tests := []struct {
name string
args args
}{
{
name: "kratos",
args: args{context.Background(), Metadata{"hello": "kratos", "kratos": "https://go-kratos.dev"}},
},
{
name: "hello",
args: args{context.Background(), Metadata{"hello": "kratos", "hello2": "https://go-kratos.dev"}},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
ctx := NewClientContext(tt.args.ctx, tt.args.md)
m, ok := FromClientContext(ctx)
if !ok {
t.Errorf("FromClientContext() = %v, want %v", ok, true)
}
if !reflect.DeepEqual(m, tt.args.md) {
t.Errorf("meta = %v, want %v", m, tt.args.md)
}
})
}
}
func TestServerContext(t *testing.T) {
type args struct {
ctx context.Context
md Metadata
}
tests := []struct {
name string
args args
}{
{
name: "kratos",
args: args{context.Background(), Metadata{"hello": "kratos", "kratos": "https://go-kratos.dev"}},
},
{
name: "hello",
args: args{context.Background(), Metadata{"hello": "kratos", "hello2": "https://go-kratos.dev"}},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
ctx := NewServerContext(tt.args.ctx, tt.args.md)
m, ok := FromServerContext(ctx)
if !ok {
t.Errorf("FromServerContext() = %v, want %v", ok, true)
}
if !reflect.DeepEqual(m, tt.args.md) {
t.Errorf("meta = %v, want %v", m, tt.args.md)
}
})
}
}
func TestAppendToClientContext(t *testing.T) {
type args struct {
md Metadata
kv []string
}
tests := []struct {
name string
args args
want Metadata
}{
{
name: "kratos",
args: args{Metadata{}, []string{"hello", "kratos", "env", "dev"}},
want: Metadata{"hello": "kratos", "env": "dev"},
},
{
name: "hello",
args: args{Metadata{"hi": "https://go-kratos.dev/"}, []string{"hello", "kratos", "env", "dev"}},
want: Metadata{"hello": "kratos", "env": "dev", "hi": "https://go-kratos.dev/"},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
ctx := NewClientContext(context.Background(), tt.args.md)
ctx = AppendToClientContext(ctx, tt.args.kv...)
md, ok := FromClientContext(ctx)
if !ok {
t.Errorf("FromServerContext() = %v, want %v", ok, true)
}
if !reflect.DeepEqual(md, tt.want) {
t.Errorf("metadata = %v, want %v", md, tt.want)
}
})
}
}
func TestMergeToClientContext(t *testing.T) {
type args struct {
md Metadata
appendMd Metadata
}
tests := []struct {
name string
args args
want Metadata
}{
{
name: "kratos",
args: args{Metadata{}, Metadata{"hello": "kratos", "env": "dev"}},
want: Metadata{"hello": "kratos", "env": "dev"},
},
{
name: "hello",
args: args{Metadata{"hi": "https://go-kratos.dev/"}, Metadata{"hello": "kratos", "env": "dev"}},
want: Metadata{"hello": "kratos", "env": "dev", "hi": "https://go-kratos.dev/"},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
ctx := NewClientContext(context.Background(), tt.args.md)
ctx = MergeToClientContext(ctx, tt.args.appendMd)
md, ok := FromClientContext(ctx)
if !ok {
t.Errorf("FromServerContext() = %v, want %v", ok, true)
}
if !reflect.DeepEqual(md, tt.want) {
t.Errorf("metadata = %v, want %v", md, tt.want)
}
})
}
}