152 lines
2.8 KiB
Go
152 lines
2.8 KiB
Go
package options_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"go.unistack.org/micro/v4/codec"
|
|
"go.unistack.org/micro/v4/metadata"
|
|
"go.unistack.org/micro/v4/options"
|
|
"go.unistack.org/micro/v4/util/reflect"
|
|
)
|
|
|
|
func TestAddress(t *testing.T) {
|
|
var err error
|
|
|
|
type s struct {
|
|
Address []string
|
|
}
|
|
|
|
src := &s{}
|
|
var opts []options.Option
|
|
opts = append(opts, options.Address("host:port"))
|
|
|
|
for _, o := range opts {
|
|
if err = o(src); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if src.Address[0] != "host:port" {
|
|
t.Fatal("failed to set Address")
|
|
}
|
|
}
|
|
|
|
func TestCodecs(t *testing.T) {
|
|
var err error
|
|
|
|
type s struct {
|
|
Codecs map[string]codec.Codec
|
|
}
|
|
|
|
src := &s{}
|
|
var opts []options.Option
|
|
c := codec.NewCodec()
|
|
opts = append(opts, options.Codecs("text/html", c))
|
|
|
|
for _, o := range opts {
|
|
if err = o(src); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
for k, v := range src.Codecs {
|
|
if k != "text/html" || v != c {
|
|
continue
|
|
}
|
|
return
|
|
}
|
|
|
|
t.Fatalf("failed to set Codecs")
|
|
}
|
|
|
|
func TestLabels(t *testing.T) {
|
|
type str1 struct {
|
|
Labels []string
|
|
}
|
|
type str2 struct {
|
|
Labels []interface{}
|
|
}
|
|
|
|
x1 := &str1{}
|
|
|
|
if err := options.Labels("one", "two")(x1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(x1.Labels) != 2 {
|
|
t.Fatal("failed to set labels")
|
|
}
|
|
x2 := &str2{}
|
|
if err := options.Labels("key", "val")(x2); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(x2.Labels) != 2 {
|
|
t.Fatal("failed to set labels")
|
|
}
|
|
if x2.Labels[0] != "key" {
|
|
t.Fatal("failed to set labels")
|
|
}
|
|
}
|
|
|
|
func TestMetadataAny(t *testing.T) {
|
|
type s struct {
|
|
Metadata metadata.Metadata
|
|
}
|
|
|
|
testCases := []struct {
|
|
Name string
|
|
Data any
|
|
Expected metadata.Metadata
|
|
}{
|
|
{
|
|
"strings_even",
|
|
[]any{"Strkey1", []string{"val1"}, "Strkey2", []string{"val2"}},
|
|
metadata.Pairs("Strkey1", "val1", "Strkey2", "val2"),
|
|
},
|
|
{
|
|
"strings_odd",
|
|
[]any{"key1", "val1", "key2"},
|
|
metadata.Pairs("Key1", "val1"),
|
|
},
|
|
{
|
|
Name: "map",
|
|
Data: map[string][]string{
|
|
"Mapkey1": {"val1"},
|
|
"Mapkey2": {"val2"},
|
|
},
|
|
Expected: metadata.Metadata{
|
|
"Mapkey1": []string{"val1"},
|
|
"Mapkey2": []string{"val2"},
|
|
},
|
|
},
|
|
{
|
|
"metadata.Metadata",
|
|
metadata.Pairs("key1", "val1", "key2", "val2"),
|
|
metadata.Pairs("Key1", "val1", "Key2", "val2"),
|
|
},
|
|
}
|
|
|
|
for _, tt := range testCases {
|
|
t.Run(tt.Name, func(t *testing.T) {
|
|
src := &s{}
|
|
var opts []options.Option
|
|
switch valData := tt.Data.(type) {
|
|
case []any:
|
|
fmt.Printf("%s any %#+v\n", tt.Name, valData)
|
|
opts = append(opts, options.Metadata(valData...))
|
|
case map[string]string, map[string][]string, metadata.Metadata:
|
|
opts = append(opts, options.Metadata(valData))
|
|
}
|
|
|
|
for _, o := range opts {
|
|
if err := o(src); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.Equal(tt.Expected, src.Metadata) {
|
|
t.Fatalf("expected: %v, actual: %v", tt.Expected, src.Metadata)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|