|
|
@@ -0,0 +1,367 @@
|
|
|
+// Go support for Protocol Buffers - Google's data interchange format
|
|
|
+//
|
|
|
+// Copyright 2015 The Go Authors. All rights reserved.
|
|
|
+// https://github.com/golang/protobuf
|
|
|
+//
|
|
|
+// Redistribution and use in source and binary forms, with or without
|
|
|
+// modification, are permitted provided that the following conditions are
|
|
|
+// met:
|
|
|
+//
|
|
|
+// * Redistributions of source code must retain the above copyright
|
|
|
+// notice, this list of conditions and the following disclaimer.
|
|
|
+// * Redistributions in binary form must reproduce the above
|
|
|
+// copyright notice, this list of conditions and the following disclaimer
|
|
|
+// in the documentation and/or other materials provided with the
|
|
|
+// distribution.
|
|
|
+// * Neither the name of Google Inc. nor the names of its
|
|
|
+// contributors may be used to endorse or promote products derived from
|
|
|
+// this software without specific prior written permission.
|
|
|
+//
|
|
|
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
+// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
|
+// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
|
+// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
|
+// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
|
+// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
|
+// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
+// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
+// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
+// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
+// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
+
|
|
|
+package jsonpb
|
|
|
+
|
|
|
+import (
|
|
|
+ "testing"
|
|
|
+
|
|
|
+ pb "github.com/golang/protobuf/jsonpb/jsonpb_test_proto"
|
|
|
+ "github.com/golang/protobuf/proto"
|
|
|
+)
|
|
|
+
|
|
|
+var (
|
|
|
+ marshaller = Marshaller{}
|
|
|
+
|
|
|
+ marshallerAllOptions = Marshaller{
|
|
|
+ EnumsAsString: true,
|
|
|
+ Indent: " ",
|
|
|
+ }
|
|
|
+
|
|
|
+ simpleObject = &pb.Simple{
|
|
|
+ OInt32: proto.Int32(-32),
|
|
|
+ OInt64: proto.Int64(-6400000000),
|
|
|
+ OUint32: proto.Uint32(32),
|
|
|
+ OUint64: proto.Uint64(6400000000),
|
|
|
+ OSint32: proto.Int32(-13),
|
|
|
+ OSint64: proto.Int64(-2600000000),
|
|
|
+ OFloat: proto.Float32(3.14),
|
|
|
+ ODouble: proto.Float64(6.02214179e23),
|
|
|
+ OBool: proto.Bool(true),
|
|
|
+ OString: proto.String("hello \"there\""),
|
|
|
+ OBytes: []byte("beep boop"),
|
|
|
+ }
|
|
|
+
|
|
|
+ simpleObjectJSON = `{` +
|
|
|
+ `"o_bool":true,` +
|
|
|
+ `"o_int32":-32,` +
|
|
|
+ `"o_int64":"-6400000000",` +
|
|
|
+ `"o_uint32":32,` +
|
|
|
+ `"o_uint64":"6400000000",` +
|
|
|
+ `"o_sint32":-13,` +
|
|
|
+ `"o_sint64":"-2600000000",` +
|
|
|
+ `"o_float":3.14,` +
|
|
|
+ `"o_double":6.02214179e+23,` +
|
|
|
+ `"o_string":"hello \"there\"",` +
|
|
|
+ `"o_bytes":"YmVlcCBib29w"` +
|
|
|
+ `}`
|
|
|
+
|
|
|
+ simpleObjectPrettyJSON = `{
|
|
|
+ "o_bool": true,
|
|
|
+ "o_int32": -32,
|
|
|
+ "o_int64": "-6400000000",
|
|
|
+ "o_uint32": 32,
|
|
|
+ "o_uint64": "6400000000",
|
|
|
+ "o_sint32": -13,
|
|
|
+ "o_sint64": "-2600000000",
|
|
|
+ "o_float": 3.14,
|
|
|
+ "o_double": 6.02214179e+23,
|
|
|
+ "o_string": "hello \"there\"",
|
|
|
+ "o_bytes": "YmVlcCBib29w"
|
|
|
+}`
|
|
|
+
|
|
|
+ repeatsObject = &pb.Repeats{
|
|
|
+ RBool: []bool{true, false, true},
|
|
|
+ RInt32: []int32{-3, -4, -5},
|
|
|
+ RInt64: []int64{-123456789, -987654321},
|
|
|
+ RUint32: []uint32{1, 2, 3},
|
|
|
+ RUint64: []uint64{6789012345, 3456789012},
|
|
|
+ RSint32: []int32{-1, -2, -3},
|
|
|
+ RSint64: []int64{-6789012345, -3456789012},
|
|
|
+ RFloat: []float32{3.14, 6.28},
|
|
|
+ RDouble: []float64{299792458, 6.62606957e-34},
|
|
|
+ RString: []string{"happy", "days"},
|
|
|
+ RBytes: [][]byte{[]byte("skittles"), []byte("m&m's")},
|
|
|
+ }
|
|
|
+
|
|
|
+ repeatsObjectJSON = `{` +
|
|
|
+ `"r_bool":[true,false,true],` +
|
|
|
+ `"r_int32":[-3,-4,-5],` +
|
|
|
+ `"r_int64":["-123456789","-987654321"],` +
|
|
|
+ `"r_uint32":[1,2,3],` +
|
|
|
+ `"r_uint64":["6789012345","3456789012"],` +
|
|
|
+ `"r_sint32":[-1,-2,-3],` +
|
|
|
+ `"r_sint64":["-6789012345","-3456789012"],` +
|
|
|
+ `"r_float":[3.14,6.28],` +
|
|
|
+ `"r_double":[2.99792458e+08,6.62606957e-34],` +
|
|
|
+ `"r_string":["happy","days"],` +
|
|
|
+ `"r_bytes":["c2tpdHRsZXM=","bSZtJ3M="]` +
|
|
|
+ `}`
|
|
|
+
|
|
|
+ repeatsObjectPrettyJSON = `{
|
|
|
+ "r_bool": [
|
|
|
+ true,
|
|
|
+ false,
|
|
|
+ true
|
|
|
+ ],
|
|
|
+ "r_int32": [
|
|
|
+ -3,
|
|
|
+ -4,
|
|
|
+ -5
|
|
|
+ ],
|
|
|
+ "r_int64": [
|
|
|
+ "-123456789",
|
|
|
+ "-987654321"
|
|
|
+ ],
|
|
|
+ "r_uint32": [
|
|
|
+ 1,
|
|
|
+ 2,
|
|
|
+ 3
|
|
|
+ ],
|
|
|
+ "r_uint64": [
|
|
|
+ "6789012345",
|
|
|
+ "3456789012"
|
|
|
+ ],
|
|
|
+ "r_sint32": [
|
|
|
+ -1,
|
|
|
+ -2,
|
|
|
+ -3
|
|
|
+ ],
|
|
|
+ "r_sint64": [
|
|
|
+ "-6789012345",
|
|
|
+ "-3456789012"
|
|
|
+ ],
|
|
|
+ "r_float": [
|
|
|
+ 3.14,
|
|
|
+ 6.28
|
|
|
+ ],
|
|
|
+ "r_double": [
|
|
|
+ 2.99792458e+08,
|
|
|
+ 6.62606957e-34
|
|
|
+ ],
|
|
|
+ "r_string": [
|
|
|
+ "happy",
|
|
|
+ "days"
|
|
|
+ ],
|
|
|
+ "r_bytes": [
|
|
|
+ "c2tpdHRsZXM=",
|
|
|
+ "bSZtJ3M="
|
|
|
+ ]
|
|
|
+}`
|
|
|
+
|
|
|
+ innerSimple = &pb.Simple{OInt32: proto.Int32(-32)}
|
|
|
+ innerSimple2 = &pb.Simple{OInt64: proto.Int64(25)}
|
|
|
+ innerRepeats = &pb.Repeats{RString: []string{"roses", "red"}}
|
|
|
+ innerRepeats2 = &pb.Repeats{RString: []string{"violets", "blue"}}
|
|
|
+ complexObject = &pb.Widget{
|
|
|
+ Color: pb.Widget_GREEN.Enum(),
|
|
|
+ RColor: []pb.Widget_Color{pb.Widget_RED, pb.Widget_GREEN, pb.Widget_BLUE},
|
|
|
+ Simple: innerSimple,
|
|
|
+ RSimple: []*pb.Simple{innerSimple, innerSimple2},
|
|
|
+ Repeats: innerRepeats,
|
|
|
+ RRepeats: []*pb.Repeats{innerRepeats, innerRepeats2},
|
|
|
+ }
|
|
|
+
|
|
|
+ complexObjectJSON = `{"color":1,` +
|
|
|
+ `"r_color":[0,1,2],` +
|
|
|
+ `"simple":{"o_int32":-32},` +
|
|
|
+ `"r_simple":[{"o_int32":-32},{"o_int64":"25"}],` +
|
|
|
+ `"repeats":{"r_string":["roses","red"]},` +
|
|
|
+ `"r_repeats":[{"r_string":["roses","red"]},{"r_string":["violets","blue"]}]` +
|
|
|
+ `}`
|
|
|
+
|
|
|
+ complexObjectPrettyJSON = `{
|
|
|
+ "color": "GREEN",
|
|
|
+ "r_color": [
|
|
|
+ "RED",
|
|
|
+ "GREEN",
|
|
|
+ "BLUE"
|
|
|
+ ],
|
|
|
+ "simple": {
|
|
|
+ "o_int32": -32
|
|
|
+ },
|
|
|
+ "r_simple": [
|
|
|
+ {
|
|
|
+ "o_int32": -32
|
|
|
+ },
|
|
|
+ {
|
|
|
+ "o_int64": "25"
|
|
|
+ }
|
|
|
+ ],
|
|
|
+ "repeats": {
|
|
|
+ "r_string": [
|
|
|
+ "roses",
|
|
|
+ "red"
|
|
|
+ ]
|
|
|
+ },
|
|
|
+ "r_repeats": [
|
|
|
+ {
|
|
|
+ "r_string": [
|
|
|
+ "roses",
|
|
|
+ "red"
|
|
|
+ ]
|
|
|
+ },
|
|
|
+ {
|
|
|
+ "r_string": [
|
|
|
+ "violets",
|
|
|
+ "blue"
|
|
|
+ ]
|
|
|
+ }
|
|
|
+ ]
|
|
|
+}`
|
|
|
+
|
|
|
+ colorPrettyJSON = `{
|
|
|
+ "color": 2
|
|
|
+}`
|
|
|
+
|
|
|
+ colorListPrettyJSON = `{
|
|
|
+ "color": 1000,
|
|
|
+ "r_color": [
|
|
|
+ "RED"
|
|
|
+ ]
|
|
|
+}`
|
|
|
+
|
|
|
+ nummyPrettyJSON = `{
|
|
|
+ "nummy": {
|
|
|
+ "1": 2,
|
|
|
+ "3": 4
|
|
|
+ }
|
|
|
+}`
|
|
|
+
|
|
|
+ objjyPrettyJSON = `{
|
|
|
+ "objjy": {
|
|
|
+ "1": {
|
|
|
+ "dub": 1
|
|
|
+ }
|
|
|
+ }
|
|
|
+}`
|
|
|
+)
|
|
|
+
|
|
|
+var marshallingTests = []struct {
|
|
|
+ desc string
|
|
|
+ marshaller Marshaller
|
|
|
+ pb proto.Message
|
|
|
+ json string
|
|
|
+}{
|
|
|
+ {"simple flat object", marshaller, simpleObject, simpleObjectJSON},
|
|
|
+ {"simple pretty object", marshallerAllOptions, simpleObject, simpleObjectPrettyJSON},
|
|
|
+ {"repeated fields flat object", marshaller, repeatsObject, repeatsObjectJSON},
|
|
|
+ {"repeated fields pretty object", marshallerAllOptions, repeatsObject, repeatsObjectPrettyJSON},
|
|
|
+ {"nested message/enum flat object", marshaller, complexObject, complexObjectJSON},
|
|
|
+ {"nested message/enum pretty object", marshallerAllOptions, complexObject, complexObjectPrettyJSON},
|
|
|
+ {"enum-string flat object", Marshaller{EnumsAsString: true},
|
|
|
+ &pb.Widget{Color: pb.Widget_BLUE.Enum()}, `{"color":"BLUE"}`},
|
|
|
+ {"enum-value pretty object", Marshaller{Indent: " "},
|
|
|
+ &pb.Widget{Color: pb.Widget_BLUE.Enum()}, colorPrettyJSON},
|
|
|
+ {"unknown enum value object", marshallerAllOptions,
|
|
|
+ &pb.Widget{Color: pb.Widget_Color(1000).Enum(), RColor: []pb.Widget_Color{pb.Widget_RED}}, colorListPrettyJSON},
|
|
|
+ {"proto3 object with empty value", marshaller, &pb.Simple3{}, `{"dub":0}`},
|
|
|
+ {"map<int64, int32>", marshaller, &pb.Mappy{Nummy: map[int64]int32{1: 2, 3: 4}}, `{"nummy":{"1":2,"3":4}}`},
|
|
|
+ {"map<int64, int32>", marshallerAllOptions, &pb.Mappy{Nummy: map[int64]int32{1: 2, 3: 4}}, nummyPrettyJSON},
|
|
|
+ {"map<string, string>", marshaller,
|
|
|
+ &pb.Mappy{Strry: map[string]string{`"one"`: "two", "three": "four"}},
|
|
|
+ `{"strry":{"\"one\"":"two","three":"four"}}`},
|
|
|
+ {"map<int32, Object>", marshaller,
|
|
|
+ &pb.Mappy{Objjy: map[int32]*pb.Simple3{1: &pb.Simple3{Dub: 1}}}, `{"objjy":{"1":{"dub":1}}}`},
|
|
|
+ {"map<int32, Object>", marshallerAllOptions,
|
|
|
+ &pb.Mappy{Objjy: map[int32]*pb.Simple3{1: &pb.Simple3{Dub: 1}}}, objjyPrettyJSON},
|
|
|
+ {"map<int64, string>", marshaller, &pb.Mappy{Buggy: map[int64]string{1234: "yup"}},
|
|
|
+ `{"buggy":{"1234":"yup"}}`},
|
|
|
+ {"map<bool, bool>", marshaller, &pb.Mappy{Booly: map[bool]bool{false: true}}, `{"booly":{"false":true}}`},
|
|
|
+ {"proto2 map<int64, string>", marshaller, &pb.Maps{MInt64Str: map[int64]string{213: "cat"}},
|
|
|
+ `{"m_int64_str":{"213":"cat"}}`},
|
|
|
+ {"proto2 map<bool, Object>", marshaller,
|
|
|
+ &pb.Maps{MBoolSimple: map[bool]*pb.Simple{true: &pb.Simple{OInt32: proto.Int32(1)}}},
|
|
|
+ `{"m_bool_simple":{"true":{"o_int32":1}}}`},
|
|
|
+}
|
|
|
+
|
|
|
+func TestMarshalling(t *testing.T) {
|
|
|
+ for _, tt := range marshallingTests {
|
|
|
+ json, err := tt.marshaller.MarshalToString(tt.pb)
|
|
|
+ if err != nil {
|
|
|
+ t.Errorf("%s: marshalling error: %v", tt.desc, err)
|
|
|
+ } else if tt.json != json {
|
|
|
+ t.Errorf("%s: got [%v] want [%v]", tt.desc, json, tt.json)
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+var unmarshallingTests = []struct {
|
|
|
+ desc string
|
|
|
+ json string
|
|
|
+ pb proto.Message
|
|
|
+}{
|
|
|
+ {"simple flat object", simpleObjectJSON, simpleObject},
|
|
|
+ {"simple pretty object", simpleObjectPrettyJSON, simpleObject},
|
|
|
+ {"repeated fields flat object", repeatsObjectJSON, repeatsObject},
|
|
|
+ {"repeated fields pretty object", repeatsObjectPrettyJSON, repeatsObject},
|
|
|
+ {"nested message/enum flat object", complexObjectJSON, complexObject},
|
|
|
+ {"nested message/enum pretty object", complexObjectPrettyJSON, complexObject},
|
|
|
+ {"enum-string object", `{"color":"BLUE"}`, &pb.Widget{Color: pb.Widget_BLUE.Enum()}},
|
|
|
+ {"enum-value object", "{\n \"color\": 2\n}", &pb.Widget{Color: pb.Widget_BLUE.Enum()}},
|
|
|
+ {"unknown enum value object",
|
|
|
+ "{\n \"color\": 1000,\n \"r_color\": [\n \"RED\"\n ]\n}",
|
|
|
+ &pb.Widget{Color: pb.Widget_Color(1000).Enum(), RColor: []pb.Widget_Color{pb.Widget_RED}}},
|
|
|
+ {"unquoted int64 object", `{"o_int64":-314}`, &pb.Simple{OInt64: proto.Int64(-314)}},
|
|
|
+ {"unquoted uint64 object", `{"o_uint64":123}`, &pb.Simple{OUint64: proto.Uint64(123)}},
|
|
|
+ {"map<int64, int32>", `{"nummy":{"1":2,"3":4}}`, &pb.Mappy{Nummy: map[int64]int32{1: 2, 3: 4}}},
|
|
|
+ {"map<string, string>", `{"strry":{"\"one\"":"two","three":"four"}}`, &pb.Mappy{Strry: map[string]string{`"one"`: "two", "three": "four"}}},
|
|
|
+ {"map<int32, Object>", `{"objjy":{"1":{"dub":1}}}`, &pb.Mappy{Objjy: map[int32]*pb.Simple3{1: &pb.Simple3{Dub: 1}}}},
|
|
|
+}
|
|
|
+
|
|
|
+func TestUnmarshalling(t *testing.T) {
|
|
|
+ for _, tt := range unmarshallingTests {
|
|
|
+ // Make a new instance of the type of our expected object.
|
|
|
+ p := proto.Clone(tt.pb)
|
|
|
+ p.Reset()
|
|
|
+
|
|
|
+ err := UnmarshalString(tt.json, p)
|
|
|
+ if err != nil {
|
|
|
+ t.Error(err)
|
|
|
+ continue
|
|
|
+ }
|
|
|
+
|
|
|
+ // For easier diffs, compare text strings of the protos.
|
|
|
+ exp := proto.MarshalTextString(tt.pb)
|
|
|
+ act := proto.MarshalTextString(p)
|
|
|
+ if string(exp) != string(act) {
|
|
|
+ t.Errorf("%s: got [%s] want [%s]", tt.desc, act, exp)
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+var unmarshallingShouldError = []struct {
|
|
|
+ desc string
|
|
|
+ in string
|
|
|
+}{
|
|
|
+ {"a value", "666"},
|
|
|
+ {"gibberish", "{adskja123;l23=-="},
|
|
|
+}
|
|
|
+
|
|
|
+func TestUnmarshallingBadInput(t *testing.T) {
|
|
|
+ for _, tt := range unmarshallingShouldError {
|
|
|
+ obj := &pb.Simple{}
|
|
|
+ err := UnmarshalString(tt.in, obj)
|
|
|
+ if err == nil {
|
|
|
+ t.Errorf("an error was expected when parsing %q instead of an object", tt.desc)
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|