mirror of https://github.com/status-im/consul.git
38 lines
991 B
Go
38 lines
991 B
Go
|
// Copyright (c) HashiCorp, Inc.
|
||
|
|
// SPDX-License-Identifier: BUSL-1.1
|
||
|
|
|
||
|
|
package prototest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"google.golang.org/protobuf/encoding/protojson"
|
||
|
|
"google.golang.org/protobuf/proto"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ProtoToJSON will encode a protobuf into a 2-space indented, deterministic,
|
||
|
|
// pretty form of JSON suitable for use in golden file output persistence. It
|
||
|
|
// does not aim to be performant.
|
||
|
|
func ProtoToJSON(t *testing.T, pb proto.Message) string {
|
||
|
|
t.Helper()
|
||
|
|
m := protojson.MarshalOptions{
|
||
|
|
Indent: " ",
|
||
|
|
}
|
||
|
|
gotJSON, err := m.Marshal(pb)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// protojson format is non-determinstic, so scrub it through the
|
||
|
|
// determinstic json.Marshal so 'git diff' only shows real changes
|
||
|
|
//
|
||
|
|
// https://github.com/golang/protobuf/issues/1269
|
||
|
|
var tmp map[string]any
|
||
|
|
require.NoError(t, json.Unmarshal(gotJSON, &tmp))
|
||
|
|
|
||
|
|
gotJSON, err = json.MarshalIndent(&tmp, "", " ")
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
return string(gotJSON)
|
||
|
|
}
|