mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 18:36:30 +02:00
bbed421cd8
* remove source, remove deadcode, fix linting issues * use github action for lint * fix missing envoy
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package storage
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/protobuf/types/known/structpb"
|
|
|
|
"github.com/pomerium/pomerium/pkg/protoutil"
|
|
)
|
|
|
|
func TestGetRecordIndex(t *testing.T) {
|
|
type M = map[string]interface{}
|
|
t.Run("missing", func(t *testing.T) {
|
|
v, err := structpb.NewStruct(M{
|
|
"notindex": "value",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Nil(t, GetRecordIndex(v))
|
|
})
|
|
t.Run("struct", func(t *testing.T) {
|
|
v, err := structpb.NewStruct(M{
|
|
"$index": M{
|
|
"cidr": "192.168.0.0/16",
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"cidr": structpb.NewStringValue("192.168.0.0/16"),
|
|
},
|
|
}, GetRecordIndex(v))
|
|
})
|
|
t.Run("value", func(t *testing.T) {
|
|
v, err := structpb.NewValue(M{
|
|
"$index": M{
|
|
"cidr": "192.168.0.0/16",
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"cidr": structpb.NewStringValue("192.168.0.0/16"),
|
|
},
|
|
}, GetRecordIndex(v))
|
|
})
|
|
t.Run("any", func(t *testing.T) {
|
|
v, err := structpb.NewValue(M{
|
|
"$index": M{
|
|
"cidr": "192.168.0.0/16",
|
|
},
|
|
})
|
|
require.NoError(t, err)
|
|
data := protoutil.NewAny(v)
|
|
assert.Equal(t, &structpb.Struct{
|
|
Fields: map[string]*structpb.Value{
|
|
"cidr": structpb.NewStringValue("192.168.0.0/16"),
|
|
},
|
|
}, GetRecordIndex(data))
|
|
})
|
|
}
|