mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 10:56:28 +02:00
dedf4b1428
* xds: use plain functions, add unit tests for control plane routes * xds: add test for grpc routes * xds: add test for pomerium http routes * xds: add test for policy routes * xds: use plain functions * xds: test get all routeable domains * xds: add build downstream tls context test * more tests * test for client cert * more tests
34 lines
1 KiB
Go
34 lines
1 KiB
Go
// Package testutil contains helper functions for unit tests.
|
|
package testutil
|
|
|
|
import (
|
|
"encoding/json"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/protobuf/encoding/protojson"
|
|
)
|
|
|
|
// AssertProtoJSONEqual asserts that a protobuf message matches the given JSON. The protoMsg can also be a slice
|
|
// of protobuf messages.
|
|
func AssertProtoJSONEqual(t *testing.T, expected string, protoMsg interface{}, msgAndArgs ...interface{}) bool {
|
|
protoMsgVal := reflect.ValueOf(protoMsg)
|
|
if protoMsgVal.Kind() == reflect.Slice {
|
|
var protoMsgs []json.RawMessage
|
|
for i := 0; i < protoMsgVal.Len(); i++ {
|
|
protoMsgs = append(protoMsgs, toProtoJSON(protoMsgVal.Index(i).Interface()))
|
|
}
|
|
bs, _ := json.Marshal(protoMsgs)
|
|
return assert.JSONEq(t, expected, string(bs), msgAndArgs...)
|
|
}
|
|
|
|
return assert.JSONEq(t, expected, string(toProtoJSON(protoMsg)), msgAndArgs...)
|
|
}
|
|
|
|
func toProtoJSON(protoMsg interface{}) json.RawMessage {
|
|
v2 := proto.MessageV2(protoMsg)
|
|
bs, _ := protojson.Marshal(v2)
|
|
return bs
|
|
}
|