129 lines
2.6 KiB
Go
129 lines
2.6 KiB
Go
package pbcommon
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/protobuf/types/known/structpb"
|
|
)
|
|
|
|
func TestEnvoyExtensionsToStructs(t *testing.T) {
|
|
input := []*EnvoyExtension{
|
|
{
|
|
Name: "ext1",
|
|
Required: true,
|
|
Arguments: &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"field1": {Kind: &structpb.Value_StringValue{StringValue: "value1"}},
|
|
"field2": {Kind: &structpb.Value_NumberValue{NumberValue: 3.14}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "ext2",
|
|
Required: false,
|
|
Arguments: &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"field3": {Kind: &structpb.Value_StringValue{StringValue: "value2"}},
|
|
"field4": {Kind: &structpb.Value_NumberValue{NumberValue: 2.718}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
expected := []structs.EnvoyExtension{
|
|
{
|
|
Name: "ext1",
|
|
Required: true,
|
|
Arguments: map[string]interface{}{
|
|
"field1": "value1",
|
|
"field2": 3.14,
|
|
},
|
|
},
|
|
{
|
|
Name: "ext2",
|
|
Required: false,
|
|
Arguments: map[string]interface{}{
|
|
"field3": "value2",
|
|
"field4": 2.718,
|
|
},
|
|
},
|
|
}
|
|
|
|
result := EnvoyExtensionsToStructs(input)
|
|
assert.Equal(t, expected, result)
|
|
}
|
|
|
|
func TestEnvoyExtensionsFromStructs(t *testing.T) {
|
|
input := []structs.EnvoyExtension{
|
|
{
|
|
Name: "ext1",
|
|
Required: true,
|
|
Arguments: map[string]interface{}{
|
|
"field1": "value1",
|
|
"field2": 3.14,
|
|
},
|
|
},
|
|
{
|
|
Name: "ext2",
|
|
Required: false,
|
|
Arguments: map[string]interface{}{
|
|
"field3": "value2",
|
|
"field4": 2.718,
|
|
},
|
|
},
|
|
}
|
|
expected := []*EnvoyExtension{
|
|
{
|
|
Name: "ext1",
|
|
Required: true,
|
|
Arguments: &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"field1": {Kind: &structpb.Value_StringValue{StringValue: "value1"}},
|
|
"field2": {Kind: &structpb.Value_NumberValue{NumberValue: 3.14}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Name: "ext2",
|
|
Required: false,
|
|
Arguments: &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"field3": {Kind: &structpb.Value_StringValue{StringValue: "value2"}},
|
|
"field4": {Kind: &structpb.Value_NumberValue{NumberValue: 2.718}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
result := EnvoyExtensionsFromStructs(input)
|
|
assert.Equal(t, expected, result)
|
|
}
|
|
|
|
func TestSliceToPBListValue(t *testing.T) {
|
|
s := []interface{}{1, 2, 3}
|
|
expected, _ := structpb.NewList(s)
|
|
tests := []struct {
|
|
input []interface{}
|
|
expected *structpb.ListValue
|
|
}{
|
|
{
|
|
[]interface{}{1, 2, 3},
|
|
expected,
|
|
},
|
|
{
|
|
[]interface{}{},
|
|
nil,
|
|
},
|
|
{
|
|
nil,
|
|
nil,
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
result := SliceToPBListValue(tc.input)
|
|
assert.Equal(t, tc.expected, result)
|
|
}
|
|
}
|