2018-01-09 18:57:14 +00:00
// 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 (
"bytes"
"encoding/json"
"io"
"math"
"reflect"
"strings"
"testing"
"github.com/golang/protobuf/proto"
pb "github.com/golang/protobuf/jsonpb/jsonpb_test_proto"
proto3pb "github.com/golang/protobuf/proto/proto3_proto"
"github.com/golang/protobuf/ptypes"
anypb "github.com/golang/protobuf/ptypes/any"
durpb "github.com/golang/protobuf/ptypes/duration"
stpb "github.com/golang/protobuf/ptypes/struct"
tspb "github.com/golang/protobuf/ptypes/timestamp"
wpb "github.com/golang/protobuf/ptypes/wrappers"
)
var (
marshaler = Marshaler { }
marshalerAllOptions = Marshaler {
Indent : " " ,
}
simpleObject = & pb . Simple {
2018-11-26 18:23:56 +00:00
OInt32 : proto . Int32 ( - 32 ) ,
OInt32Str : proto . Int32 ( - 32 ) ,
OInt64 : proto . Int64 ( - 6400000000 ) ,
OInt64Str : proto . Int64 ( - 6400000000 ) ,
OUint32 : proto . Uint32 ( 32 ) ,
OUint32Str : proto . Uint32 ( 32 ) ,
OUint64 : proto . Uint64 ( 6400000000 ) ,
OUint64Str : proto . Uint64 ( 6400000000 ) ,
OSint32 : proto . Int32 ( - 13 ) ,
OSint32Str : proto . Int32 ( - 13 ) ,
OSint64 : proto . Int64 ( - 2600000000 ) ,
OSint64Str : proto . Int64 ( - 2600000000 ) ,
OFloat : proto . Float32 ( 3.14 ) ,
OFloatStr : proto . Float32 ( 3.14 ) ,
ODouble : proto . Float64 ( 6.02214179e23 ) ,
ODoubleStr : proto . Float64 ( 6.02214179e23 ) ,
OBool : proto . Bool ( true ) ,
OString : proto . String ( "hello \"there\"" ) ,
OBytes : [ ] byte ( "beep boop" ) ,
2018-01-09 18:57:14 +00:00
}
2018-11-26 18:23:56 +00:00
simpleObjectInputJSON = ` { ` +
` "oBool":true, ` +
` "oInt32":-32, ` +
` "oInt32Str":"-32", ` +
` "oInt64":-6400000000, ` +
` "oInt64Str":"-6400000000", ` +
` "oUint32":32, ` +
` "oUint32Str":"32", ` +
` "oUint64":6400000000, ` +
` "oUint64Str":"6400000000", ` +
` "oSint32":-13, ` +
` "oSint32Str":"-13", ` +
` "oSint64":-2600000000, ` +
` "oSint64Str":"-2600000000", ` +
` "oFloat":3.14, ` +
` "oFloatStr":"3.14", ` +
` "oDouble":6.02214179e+23, ` +
` "oDoubleStr":"6.02214179e+23", ` +
` "oString":"hello \"there\"", ` +
` "oBytes":"YmVlcCBib29w" ` +
` } `
simpleObjectOutputJSON = ` { ` +
2018-01-09 18:57:14 +00:00
` "oBool":true, ` +
` "oInt32":-32, ` +
2018-11-26 18:23:56 +00:00
` "oInt32Str":-32, ` +
2018-01-09 18:57:14 +00:00
` "oInt64":"-6400000000", ` +
2018-11-26 18:23:56 +00:00
` "oInt64Str":"-6400000000", ` +
2018-01-09 18:57:14 +00:00
` "oUint32":32, ` +
2018-11-26 18:23:56 +00:00
` "oUint32Str":32, ` +
2018-01-09 18:57:14 +00:00
` "oUint64":"6400000000", ` +
2018-11-26 18:23:56 +00:00
` "oUint64Str":"6400000000", ` +
2018-01-09 18:57:14 +00:00
` "oSint32":-13, ` +
2018-11-26 18:23:56 +00:00
` "oSint32Str":-13, ` +
2018-01-09 18:57:14 +00:00
` "oSint64":"-2600000000", ` +
2018-11-26 18:23:56 +00:00
` "oSint64Str":"-2600000000", ` +
2018-01-09 18:57:14 +00:00
` "oFloat":3.14, ` +
2018-11-26 18:23:56 +00:00
` "oFloatStr":3.14, ` +
2018-01-09 18:57:14 +00:00
` "oDouble":6.02214179e+23, ` +
2018-11-26 18:23:56 +00:00
` "oDoubleStr":6.02214179e+23, ` +
2018-01-09 18:57:14 +00:00
` "oString":"hello \"there\"", ` +
` "oBytes":"YmVlcCBib29w" ` +
` } `
2018-11-26 18:23:56 +00:00
simpleObjectInputPrettyJSON = ` {
"oBool" : true ,
"oInt32" : - 32 ,
"oInt32Str" : "-32" ,
"oInt64" : - 6400000000 ,
"oInt64Str" : "-6400000000" ,
"oUint32" : 32 ,
"oUint32Str" : "32" ,
"oUint64" : 6400000000 ,
"oUint64Str" : "6400000000" ,
"oSint32" : - 13 ,
"oSint32Str" : "-13" ,
"oSint64" : - 2600000000 ,
"oSint64Str" : "-2600000000" ,
"oFloat" : 3.14 ,
"oFloatStr" : "3.14" ,
"oDouble" : 6.02214179e+23 ,
"oDoubleStr" : "6.02214179e+23" ,
"oString" : "hello \"there\"" ,
"oBytes" : "YmVlcCBib29w"
} `
simpleObjectOutputPrettyJSON = ` {
2018-01-09 18:57:14 +00:00
"oBool" : true ,
"oInt32" : - 32 ,
2018-11-26 18:23:56 +00:00
"oInt32Str" : - 32 ,
2018-01-09 18:57:14 +00:00
"oInt64" : "-6400000000" ,
2018-11-26 18:23:56 +00:00
"oInt64Str" : "-6400000000" ,
2018-01-09 18:57:14 +00:00
"oUint32" : 32 ,
2018-11-26 18:23:56 +00:00
"oUint32Str" : 32 ,
2018-01-09 18:57:14 +00:00
"oUint64" : "6400000000" ,
2018-11-26 18:23:56 +00:00
"oUint64Str" : "6400000000" ,
2018-01-09 18:57:14 +00:00
"oSint32" : - 13 ,
2018-11-26 18:23:56 +00:00
"oSint32Str" : - 13 ,
2018-01-09 18:57:14 +00:00
"oSint64" : "-2600000000" ,
2018-11-26 18:23:56 +00:00
"oSint64Str" : "-2600000000" ,
2018-01-09 18:57:14 +00:00
"oFloat" : 3.14 ,
2018-11-26 18:23:56 +00:00
"oFloatStr" : 3.14 ,
2018-01-09 18:57:14 +00:00
"oDouble" : 6.02214179e+23 ,
2018-11-26 18:23:56 +00:00
"oDoubleStr" : 6.02214179e+23 ,
2018-01-09 18:57:14 +00:00
"oString" : "hello \"there\"" ,
"oBytes" : "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 * 1e20 , 6.62606957e-34 } ,
RString : [ ] string { "happy" , "days" } ,
RBytes : [ ] [ ] byte { [ ] byte ( "skittles" ) , [ ] byte ( "m&m's" ) } ,
}
repeatsObjectJSON = ` { ` +
` "rBool":[true,false,true], ` +
` "rInt32":[-3,-4,-5], ` +
` "rInt64":["-123456789","-987654321"], ` +
` "rUint32":[1,2,3], ` +
` "rUint64":["6789012345","3456789012"], ` +
` "rSint32":[-1,-2,-3], ` +
` "rSint64":["-6789012345","-3456789012"], ` +
` "rFloat":[3.14,6.28], ` +
` "rDouble":[2.99792458e+28,6.62606957e-34], ` +
` "rString":["happy","days"], ` +
` "rBytes":["c2tpdHRsZXM=","bSZtJ3M="] ` +
` } `
repeatsObjectPrettyJSON = ` {
"rBool" : [
true ,
false ,
true
] ,
"rInt32" : [
- 3 ,
- 4 ,
- 5
] ,
"rInt64" : [
"-123456789" ,
"-987654321"
] ,
"rUint32" : [
1 ,
2 ,
3
] ,
"rUint64" : [
"6789012345" ,
"3456789012"
] ,
"rSint32" : [
- 1 ,
- 2 ,
- 3
] ,
"rSint64" : [
"-6789012345" ,
"-3456789012"
] ,
"rFloat" : [
3.14 ,
6.28
] ,
"rDouble" : [
2.99792458e+28 ,
6.62606957e-34
] ,
"rString" : [
"happy" ,
"days"
] ,
"rBytes" : [
"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":"GREEN", ` +
` "rColor":["RED","GREEN","BLUE"], ` +
` "simple": { "oInt32":-32}, ` +
` "rSimple":[ { "oInt32":-32}, { "oInt64":"25"}], ` +
` "repeats": { "rString":["roses","red"]}, ` +
` "rRepeats":[ { "rString":["roses","red"]}, { "rString":["violets","blue"]}] ` +
` } `
complexObjectPrettyJSON = ` {
"color" : "GREEN" ,
"rColor" : [
"RED" ,
"GREEN" ,
"BLUE"
] ,
"simple" : {
"oInt32" : - 32
} ,
"rSimple" : [
{
"oInt32" : - 32
} ,
{
"oInt64" : "25"
}
] ,
"repeats" : {
"rString" : [
"roses" ,
"red"
]
} ,
"rRepeats" : [
{
"rString" : [
"roses" ,
"red"
]
} ,
{
"rString" : [
"violets" ,
"blue"
]
}
]
} `
colorPrettyJSON = ` {
"color" : 2
} `
colorListPrettyJSON = ` {
"color" : 1000 ,
"rColor" : [
"RED"
]
} `
nummyPrettyJSON = ` {
"nummy" : {
"1" : 2 ,
"3" : 4
}
} `
objjyPrettyJSON = ` {
"objjy" : {
"1" : {
"dub" : 1
}
}
} `
realNumber = & pb . Real { Value : proto . Float64 ( 3.14159265359 ) }
realNumberName = "Pi"
complexNumber = & pb . Complex { Imaginary : proto . Float64 ( 0.5772156649 ) }
realNumberJSON = ` { ` +
` "value":3.14159265359, ` +
` "[jsonpb.Complex.real_extension]": { "imaginary":0.5772156649}, ` +
` "[jsonpb.name]":"Pi" ` +
` } `
anySimple = & pb . KnownTypes {
An : & anypb . Any {
TypeUrl : "something.example.com/jsonpb.Simple" ,
Value : [ ] byte {
// &pb.Simple{OBool:true}
1 << 3 , 1 ,
} ,
} ,
}
anySimpleJSON = ` { "an": { "@type":"something.example.com/jsonpb.Simple","oBool":true}} `
anySimplePrettyJSON = ` {
"an" : {
"@type" : "something.example.com/jsonpb.Simple" ,
"oBool" : true
}
} `
anyWellKnown = & pb . KnownTypes {
An : & anypb . Any {
TypeUrl : "type.googleapis.com/google.protobuf.Duration" ,
Value : [ ] byte {
// &durpb.Duration{Seconds: 1, Nanos: 212000000 }
1 << 3 , 1 , // seconds
2 << 3 , 0x80 , 0xba , 0x8b , 0x65 , // nanos
} ,
} ,
}
anyWellKnownJSON = ` { "an": { "@type":"type.googleapis.com/google.protobuf.Duration","value":"1.212s"}} `
anyWellKnownPrettyJSON = ` {
"an" : {
"@type" : "type.googleapis.com/google.protobuf.Duration" ,
"value" : "1.212s"
}
} `
nonFinites = & pb . NonFinites {
FNan : proto . Float32 ( float32 ( math . NaN ( ) ) ) ,
FPinf : proto . Float32 ( float32 ( math . Inf ( 1 ) ) ) ,
FNinf : proto . Float32 ( float32 ( math . Inf ( - 1 ) ) ) ,
DNan : proto . Float64 ( float64 ( math . NaN ( ) ) ) ,
DPinf : proto . Float64 ( float64 ( math . Inf ( 1 ) ) ) ,
DNinf : proto . Float64 ( float64 ( math . Inf ( - 1 ) ) ) ,
}
nonFinitesJSON = ` { ` +
` "fNan":"NaN", ` +
` "fPinf":"Infinity", ` +
` "fNinf":"-Infinity", ` +
` "dNan":"NaN", ` +
` "dPinf":"Infinity", ` +
` "dNinf":"-Infinity" ` +
` } `
)
func init ( ) {
if err := proto . SetExtension ( realNumber , pb . E_Name , & realNumberName ) ; err != nil {
panic ( err )
}
if err := proto . SetExtension ( realNumber , pb . E_Complex_RealExtension , complexNumber ) ; err != nil {
panic ( err )
}
}
var marshalingTests = [ ] struct {
desc string
marshaler Marshaler
pb proto . Message
json string
} {
2018-11-26 18:23:56 +00:00
{ "simple flat object" , marshaler , simpleObject , simpleObjectOutputJSON } ,
{ "simple pretty object" , marshalerAllOptions , simpleObject , simpleObjectOutputPrettyJSON } ,
2018-01-09 18:57:14 +00:00
{ "non-finite floats fields object" , marshaler , nonFinites , nonFinitesJSON } ,
{ "repeated fields flat object" , marshaler , repeatsObject , repeatsObjectJSON } ,
{ "repeated fields pretty object" , marshalerAllOptions , repeatsObject , repeatsObjectPrettyJSON } ,
{ "nested message/enum flat object" , marshaler , complexObject , complexObjectJSON } ,
{ "nested message/enum pretty object" , marshalerAllOptions , complexObject , complexObjectPrettyJSON } ,
{ "enum-string flat object" , Marshaler { } ,
& pb . Widget { Color : pb . Widget_BLUE . Enum ( ) } , ` { "color":"BLUE"} ` } ,
{ "enum-value pretty object" , Marshaler { EnumsAsInts : true , Indent : " " } ,
& pb . Widget { Color : pb . Widget_BLUE . Enum ( ) } , colorPrettyJSON } ,
{ "unknown enum value object" , marshalerAllOptions ,
& pb . Widget { Color : pb . Widget_Color ( 1000 ) . Enum ( ) , RColor : [ ] pb . Widget_Color { pb . Widget_RED } } , colorListPrettyJSON } ,
{ "repeated proto3 enum" , Marshaler { } ,
& proto3pb . Message { RFunny : [ ] proto3pb . Message_Humour {
proto3pb . Message_PUNS ,
proto3pb . Message_SLAPSTICK ,
} } ,
` { "rFunny":["PUNS","SLAPSTICK"]} ` } ,
{ "repeated proto3 enum as int" , Marshaler { EnumsAsInts : true } ,
& proto3pb . Message { RFunny : [ ] proto3pb . Message_Humour {
proto3pb . Message_PUNS ,
proto3pb . Message_SLAPSTICK ,
} } ,
` { "rFunny":[1,2]} ` } ,
{ "empty value" , marshaler , & pb . Simple3 { } , ` { } ` } ,
{ "empty value emitted" , Marshaler { EmitDefaults : true } , & pb . Simple3 { } , ` { "dub":0} ` } ,
{ "empty repeated emitted" , Marshaler { EmitDefaults : true } , & pb . SimpleSlice3 { } , ` { "slices":[]} ` } ,
{ "empty map emitted" , Marshaler { EmitDefaults : true } , & pb . SimpleMap3 { } , ` { "stringy": { }} ` } ,
{ "nested struct null" , Marshaler { EmitDefaults : true } , & pb . SimpleNull3 { } , ` { "simple":null} ` } ,
{ "map<int64, int32>" , marshaler , & pb . Mappy { Nummy : map [ int64 ] int32 { 1 : 2 , 3 : 4 } } , ` { "nummy": { "1":2,"3":4}} ` } ,
{ "map<int64, int32>" , marshalerAllOptions , & pb . Mappy { Nummy : map [ int64 ] int32 { 1 : 2 , 3 : 4 } } , nummyPrettyJSON } ,
{ "map<string, string>" , marshaler ,
& pb . Mappy { Strry : map [ string ] string { ` "one" ` : "two" , "three" : "four" } } ,
` { "strry": { "\"one\"":"two","three":"four"}} ` } ,
{ "map<int32, Object>" , marshaler ,
& pb . Mappy { Objjy : map [ int32 ] * pb . Simple3 { 1 : { Dub : 1 } } } , ` { "objjy": { "1": { "dub":1}}} ` } ,
{ "map<int32, Object>" , marshalerAllOptions ,
& pb . Mappy { Objjy : map [ int32 ] * pb . Simple3 { 1 : { Dub : 1 } } } , objjyPrettyJSON } ,
{ "map<int64, string>" , marshaler , & pb . Mappy { Buggy : map [ int64 ] string { 1234 : "yup" } } ,
` { "buggy": { "1234":"yup"}} ` } ,
{ "map<bool, bool>" , marshaler , & pb . Mappy { Booly : map [ bool ] bool { false : true } } , ` { "booly": { "false":true}} ` } ,
2018-11-26 18:23:56 +00:00
{ "map<string, enum>" , marshaler , & pb . Mappy { Enumy : map [ string ] pb . Numeral { "XIV" : pb . Numeral_ROMAN } } , ` { "enumy": { "XIV":"ROMAN"}} ` } ,
2018-01-09 18:57:14 +00:00
{ "map<string, enum as int>" , Marshaler { EnumsAsInts : true } , & pb . Mappy { Enumy : map [ string ] pb . Numeral { "XIV" : pb . Numeral_ROMAN } } , ` { "enumy": { "XIV":2}} ` } ,
{ "map<int32, bool>" , marshaler , & pb . Mappy { S32Booly : map [ int32 ] bool { 1 : true , 3 : false , 10 : true , 12 : false } } , ` { "s32booly": { "1":true,"3":false,"10":true,"12":false}} ` } ,
{ "map<int64, bool>" , marshaler , & pb . Mappy { S64Booly : map [ int64 ] bool { 1 : true , 3 : false , 10 : true , 12 : false } } , ` { "s64booly": { "1":true,"3":false,"10":true,"12":false}} ` } ,
{ "map<uint32, bool>" , marshaler , & pb . Mappy { U32Booly : map [ uint32 ] bool { 1 : true , 3 : false , 10 : true , 12 : false } } , ` { "u32booly": { "1":true,"3":false,"10":true,"12":false}} ` } ,
{ "map<uint64, bool>" , marshaler , & pb . Mappy { U64Booly : map [ uint64 ] bool { 1 : true , 3 : false , 10 : true , 12 : false } } , ` { "u64booly": { "1":true,"3":false,"10":true,"12":false}} ` } ,
{ "proto2 map<int64, string>" , marshaler , & pb . Maps { MInt64Str : map [ int64 ] string { 213 : "cat" } } ,
` { "mInt64Str": { "213":"cat"}} ` } ,
{ "proto2 map<bool, Object>" , marshaler ,
& pb . Maps { MBoolSimple : map [ bool ] * pb . Simple { true : { OInt32 : proto . Int32 ( 1 ) } } } ,
` { "mBoolSimple": { "true": { "oInt32":1}}} ` } ,
{ "oneof, not set" , marshaler , & pb . MsgWithOneof { } , ` { } ` } ,
{ "oneof, set" , marshaler , & pb . MsgWithOneof { Union : & pb . MsgWithOneof_Title { "Grand Poobah" } } , ` { "title":"Grand Poobah"} ` } ,
{ "force orig_name" , Marshaler { OrigName : true } , & pb . Simple { OInt32 : proto . Int32 ( 4 ) } ,
` { "o_int32":4} ` } ,
{ "proto2 extension" , marshaler , realNumber , realNumberJSON } ,
{ "Any with message" , marshaler , anySimple , anySimpleJSON } ,
{ "Any with message and indent" , marshalerAllOptions , anySimple , anySimplePrettyJSON } ,
{ "Any with WKT" , marshaler , anyWellKnown , anyWellKnownJSON } ,
{ "Any with WKT and indent" , marshalerAllOptions , anyWellKnown , anyWellKnownPrettyJSON } ,
2018-07-18 14:47:22 +00:00
{ "Duration" , marshaler , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 3 } } , ` { "dur":"3s"} ` } ,
{ "Duration" , marshaler , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 3 , Nanos : 1e6 } } , ` { "dur":"3.001s"} ` } ,
{ "Duration beyond float64 precision" , marshaler , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 100000000 , Nanos : 1 } } , ` { "dur":"100000000.000000001s"} ` } ,
{ "negative Duration" , marshaler , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : - 123 , Nanos : - 456 } } , ` { "dur":"-123.000000456s"} ` } ,
2018-01-09 18:57:14 +00:00
{ "Struct" , marshaler , & pb . KnownTypes { St : & stpb . Struct {
Fields : map [ string ] * stpb . Value {
"one" : { Kind : & stpb . Value_StringValue { "loneliest number" } } ,
"two" : { Kind : & stpb . Value_NullValue { stpb . NullValue_NULL_VALUE } } ,
} ,
} } , ` { "st": { "one":"loneliest number","two":null}} ` } ,
{ "empty ListValue" , marshaler , & pb . KnownTypes { Lv : & stpb . ListValue { } } , ` { "lv":[]} ` } ,
{ "basic ListValue" , marshaler , & pb . KnownTypes { Lv : & stpb . ListValue { Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_StringValue { "x" } } ,
{ Kind : & stpb . Value_NullValue { } } ,
{ Kind : & stpb . Value_NumberValue { 3 } } ,
{ Kind : & stpb . Value_BoolValue { true } } ,
} } } , ` { "lv":["x",null,3,true]} ` } ,
{ "Timestamp" , marshaler , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 14e8 , Nanos : 21e6 } } , ` { "ts":"2014-05-13T16:53:20.021Z"} ` } ,
2018-07-18 14:47:22 +00:00
{ "Timestamp" , marshaler , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 14e8 , Nanos : 0 } } , ` { "ts":"2014-05-13T16:53:20Z"} ` } ,
2018-01-09 18:57:14 +00:00
{ "number Value" , marshaler , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_NumberValue { 1 } } } , ` { "val":1} ` } ,
{ "null Value" , marshaler , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_NullValue { stpb . NullValue_NULL_VALUE } } } , ` { "val":null} ` } ,
{ "string number value" , marshaler , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_StringValue { "9223372036854775807" } } } , ` { "val":"9223372036854775807"} ` } ,
{ "list of lists Value" , marshaler , & pb . KnownTypes { Val : & stpb . Value {
Kind : & stpb . Value_ListValue { & stpb . ListValue {
Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_StringValue { "x" } } ,
{ Kind : & stpb . Value_ListValue { & stpb . ListValue {
Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_ListValue { & stpb . ListValue {
Values : [ ] * stpb . Value { { Kind : & stpb . Value_StringValue { "y" } } } ,
} } } ,
{ Kind : & stpb . Value_StringValue { "z" } } ,
} ,
} } } ,
} ,
} } ,
} } , ` { "val":["x",[["y"],"z"]]} ` } ,
{ "DoubleValue" , marshaler , & pb . KnownTypes { Dbl : & wpb . DoubleValue { Value : 1.2 } } , ` { "dbl":1.2} ` } ,
{ "FloatValue" , marshaler , & pb . KnownTypes { Flt : & wpb . FloatValue { Value : 1.2 } } , ` { "flt":1.2} ` } ,
{ "Int64Value" , marshaler , & pb . KnownTypes { I64 : & wpb . Int64Value { Value : - 3 } } , ` { "i64":"-3"} ` } ,
{ "UInt64Value" , marshaler , & pb . KnownTypes { U64 : & wpb . UInt64Value { Value : 3 } } , ` { "u64":"3"} ` } ,
{ "Int32Value" , marshaler , & pb . KnownTypes { I32 : & wpb . Int32Value { Value : - 4 } } , ` { "i32":-4} ` } ,
{ "UInt32Value" , marshaler , & pb . KnownTypes { U32 : & wpb . UInt32Value { Value : 4 } } , ` { "u32":4} ` } ,
{ "BoolValue" , marshaler , & pb . KnownTypes { Bool : & wpb . BoolValue { Value : true } } , ` { "bool":true} ` } ,
{ "StringValue" , marshaler , & pb . KnownTypes { Str : & wpb . StringValue { Value : "plush" } } , ` { "str":"plush"} ` } ,
{ "BytesValue" , marshaler , & pb . KnownTypes { Bytes : & wpb . BytesValue { Value : [ ] byte ( "wow" ) } } , ` { "bytes":"d293"} ` } ,
2018-07-18 14:47:22 +00:00
{ "required" , marshaler , & pb . MsgWithRequired { Str : proto . String ( "hello" ) } , ` { "str":"hello"} ` } ,
{ "required bytes" , marshaler , & pb . MsgWithRequiredBytes { Byts : [ ] byte { } } , ` { "byts":""} ` } ,
2018-01-09 18:57:14 +00:00
}
func TestMarshaling ( t * testing . T ) {
for _ , tt := range marshalingTests {
json , err := tt . marshaler . MarshalToString ( tt . pb )
if err != nil {
t . Errorf ( "%s: marshaling error: %v" , tt . desc , err )
} else if tt . json != json {
t . Errorf ( "%s: got [%v] want [%v]" , tt . desc , json , tt . json )
}
}
}
2018-07-18 14:47:22 +00:00
func TestMarshalingNil ( t * testing . T ) {
var msg * pb . Simple
m := & Marshaler { }
if _ , err := m . MarshalToString ( msg ) ; err == nil {
t . Errorf ( "mashaling nil returned no error" )
}
}
func TestMarshalIllegalTime ( t * testing . T ) {
tests := [ ] struct {
pb proto . Message
fail bool
} {
{ & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 1 , Nanos : 0 } } , false } ,
{ & pb . KnownTypes { Dur : & durpb . Duration { Seconds : - 1 , Nanos : 0 } } , false } ,
{ & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 1 , Nanos : - 1 } } , true } ,
{ & pb . KnownTypes { Dur : & durpb . Duration { Seconds : - 1 , Nanos : 1 } } , true } ,
{ & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 1 , Nanos : 1000000000 } } , true } ,
{ & pb . KnownTypes { Dur : & durpb . Duration { Seconds : - 1 , Nanos : - 1000000000 } } , true } ,
{ & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 1 , Nanos : 1 } } , false } ,
{ & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 1 , Nanos : - 1 } } , true } ,
{ & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 1 , Nanos : 1000000000 } } , true } ,
}
for _ , tt := range tests {
_ , err := marshaler . MarshalToString ( tt . pb )
if err == nil && tt . fail {
t . Errorf ( "marshaler.MarshalToString(%v) = _, <nil>; want _, <non-nil>" , tt . pb )
}
if err != nil && ! tt . fail {
t . Errorf ( "marshaler.MarshalToString(%v) = _, %v; want _, <nil>" , tt . pb , err )
}
}
}
2018-01-09 18:57:14 +00:00
func TestMarshalJSONPBMarshaler ( t * testing . T ) {
rawJson := ` { "foo": "bar", "baz": [0, 1, 2, 3] } `
2018-11-26 18:23:56 +00:00
msg := dynamicMessage { RawJson : rawJson }
2018-01-09 18:57:14 +00:00
str , err := new ( Marshaler ) . MarshalToString ( & msg )
if err != nil {
t . Errorf ( "an unexpected error occurred when marshalling JSONPBMarshaler: %v" , err )
}
if str != rawJson {
t . Errorf ( "marshalling JSON produced incorrect output: got %s, wanted %s" , str , rawJson )
}
}
func TestMarshalAnyJSONPBMarshaler ( t * testing . T ) {
2018-11-26 18:23:56 +00:00
msg := dynamicMessage { RawJson : ` { "foo": "bar", "baz": [0, 1, 2, 3] } ` }
2018-01-09 18:57:14 +00:00
a , err := ptypes . MarshalAny ( & msg )
if err != nil {
t . Errorf ( "an unexpected error occurred when marshalling to Any: %v" , err )
}
str , err := new ( Marshaler ) . MarshalToString ( a )
if err != nil {
t . Errorf ( "an unexpected error occurred when marshalling Any to JSON: %v" , err )
}
// after custom marshaling, it's round-tripped through JSON decoding/encoding already,
// so the keys are sorted, whitespace is compacted, and "@type" key has been added
expected := ` { "@type":"type.googleapis.com/ ` + dynamicMessageName + ` ","baz":[0,1,2,3],"foo":"bar"} `
if str != expected {
t . Errorf ( "marshalling JSON produced incorrect output: got %s, wanted %s" , str , expected )
}
}
2018-07-18 14:47:22 +00:00
func TestMarshalWithCustomValidation ( t * testing . T ) {
2018-11-26 18:23:56 +00:00
msg := dynamicMessage { RawJson : ` { "foo": "bar", "baz": [0, 1, 2, 3] } ` , Dummy : & dynamicMessage { } }
2018-07-18 14:47:22 +00:00
js , err := new ( Marshaler ) . MarshalToString ( & msg )
if err != nil {
t . Errorf ( "an unexpected error occurred when marshalling to json: %v" , err )
}
err = Unmarshal ( strings . NewReader ( js ) , & msg )
if err != nil {
t . Errorf ( "an unexpected error occurred when unmarshalling from json: %v" , err )
}
}
// Test marshaling message containing unset required fields should produce error.
func TestMarshalUnsetRequiredFields ( t * testing . T ) {
msgExt := & pb . Real { }
proto . SetExtension ( msgExt , pb . E_Extm , & pb . MsgWithRequired { } )
tests := [ ] struct {
desc string
marshaler * Marshaler
pb proto . Message
} {
{
desc : "direct required field" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithRequired { } ,
} ,
{
desc : "direct required field + emit defaults" ,
marshaler : & Marshaler { EmitDefaults : true } ,
pb : & pb . MsgWithRequired { } ,
} ,
{
desc : "indirect required field" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithIndirectRequired { Subm : & pb . MsgWithRequired { } } ,
} ,
{
desc : "indirect required field + emit defaults" ,
marshaler : & Marshaler { EmitDefaults : true } ,
pb : & pb . MsgWithIndirectRequired { Subm : & pb . MsgWithRequired { } } ,
} ,
{
desc : "direct required wkt field" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithRequiredWKT { } ,
} ,
{
desc : "direct required wkt field + emit defaults" ,
marshaler : & Marshaler { EmitDefaults : true } ,
pb : & pb . MsgWithRequiredWKT { } ,
} ,
{
desc : "direct required bytes field" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithRequiredBytes { } ,
} ,
{
desc : "required in map value" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithIndirectRequired {
MapField : map [ string ] * pb . MsgWithRequired {
"key" : { } ,
} ,
} ,
} ,
{
desc : "required in repeated item" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithIndirectRequired {
SliceField : [ ] * pb . MsgWithRequired {
{ Str : proto . String ( "hello" ) } ,
{ } ,
} ,
} ,
} ,
{
desc : "required inside oneof" ,
marshaler : & Marshaler { } ,
pb : & pb . MsgWithOneof {
Union : & pb . MsgWithOneof_MsgWithRequired { & pb . MsgWithRequired { } } ,
} ,
} ,
{
desc : "required inside extension" ,
marshaler : & Marshaler { } ,
pb : msgExt ,
} ,
}
for _ , tc := range tests {
if _ , err := tc . marshaler . MarshalToString ( tc . pb ) ; err == nil {
t . Errorf ( "%s: expecting error in marshaling with unset required fields %+v" , tc . desc , tc . pb )
}
}
}
2018-01-09 18:57:14 +00:00
var unmarshalingTests = [ ] struct {
desc string
unmarshaler Unmarshaler
json string
pb proto . Message
} {
2018-11-26 18:23:56 +00:00
{ "simple flat object" , Unmarshaler { } , simpleObjectInputJSON , simpleObject } ,
{ "simple pretty object" , Unmarshaler { } , simpleObjectInputPrettyJSON , simpleObject } ,
2018-01-09 18:57:14 +00:00
{ "repeated fields flat object" , Unmarshaler { } , repeatsObjectJSON , repeatsObject } ,
{ "repeated fields pretty object" , Unmarshaler { } , repeatsObjectPrettyJSON , repeatsObject } ,
{ "nested message/enum flat object" , Unmarshaler { } , complexObjectJSON , complexObject } ,
{ "nested message/enum pretty object" , Unmarshaler { } , complexObjectPrettyJSON , complexObject } ,
{ "enum-string object" , Unmarshaler { } , ` { "color":"BLUE"} ` , & pb . Widget { Color : pb . Widget_BLUE . Enum ( ) } } ,
{ "enum-value object" , Unmarshaler { } , "{\n \"color\": 2\n}" , & pb . Widget { Color : pb . Widget_BLUE . Enum ( ) } } ,
{ "unknown field with allowed option" , Unmarshaler { AllowUnknownFields : true } , ` { "unknown": "foo"} ` , new ( pb . Simple ) } ,
{ "proto3 enum string" , Unmarshaler { } , ` { "hilarity":"PUNS"} ` , & proto3pb . Message { Hilarity : proto3pb . Message_PUNS } } ,
{ "proto3 enum value" , Unmarshaler { } , ` { "hilarity":1} ` , & proto3pb . Message { Hilarity : proto3pb . Message_PUNS } } ,
{ "unknown enum value object" ,
Unmarshaler { } ,
"{\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 } } } ,
{ "repeated proto3 enum" , Unmarshaler { } , ` { "rFunny":["PUNS","SLAPSTICK"]} ` ,
& proto3pb . Message { RFunny : [ ] proto3pb . Message_Humour {
proto3pb . Message_PUNS ,
proto3pb . Message_SLAPSTICK ,
} } } ,
{ "repeated proto3 enum as int" , Unmarshaler { } , ` { "rFunny":[1,2]} ` ,
& proto3pb . Message { RFunny : [ ] proto3pb . Message_Humour {
proto3pb . Message_PUNS ,
proto3pb . Message_SLAPSTICK ,
} } } ,
{ "repeated proto3 enum as mix of strings and ints" , Unmarshaler { } , ` { "rFunny":["PUNS",2]} ` ,
& proto3pb . Message { RFunny : [ ] proto3pb . Message_Humour {
proto3pb . Message_PUNS ,
proto3pb . Message_SLAPSTICK ,
} } } ,
{ "unquoted int64 object" , Unmarshaler { } , ` { "oInt64":-314} ` , & pb . Simple { OInt64 : proto . Int64 ( - 314 ) } } ,
{ "unquoted uint64 object" , Unmarshaler { } , ` { "oUint64":123} ` , & pb . Simple { OUint64 : proto . Uint64 ( 123 ) } } ,
{ "NaN" , Unmarshaler { } , ` { "oDouble":"NaN"} ` , & pb . Simple { ODouble : proto . Float64 ( math . NaN ( ) ) } } ,
{ "Inf" , Unmarshaler { } , ` { "oFloat":"Infinity"} ` , & pb . Simple { OFloat : proto . Float32 ( float32 ( math . Inf ( 1 ) ) ) } } ,
{ "-Inf" , Unmarshaler { } , ` { "oDouble":"-Infinity"} ` , & pb . Simple { ODouble : proto . Float64 ( math . Inf ( - 1 ) ) } } ,
{ "map<int64, int32>" , Unmarshaler { } , ` { "nummy": { "1":2,"3":4}} ` , & pb . Mappy { Nummy : map [ int64 ] int32 { 1 : 2 , 3 : 4 } } } ,
{ "map<string, string>" , Unmarshaler { } , ` { "strry": { "\"one\"":"two","three":"four"}} ` , & pb . Mappy { Strry : map [ string ] string { ` "one" ` : "two" , "three" : "four" } } } ,
{ "map<int32, Object>" , Unmarshaler { } , ` { "objjy": { "1": { "dub":1}}} ` , & pb . Mappy { Objjy : map [ int32 ] * pb . Simple3 { 1 : { Dub : 1 } } } } ,
{ "proto2 extension" , Unmarshaler { } , realNumberJSON , realNumber } ,
{ "Any with message" , Unmarshaler { } , anySimpleJSON , anySimple } ,
{ "Any with message and indent" , Unmarshaler { } , anySimplePrettyJSON , anySimple } ,
{ "Any with WKT" , Unmarshaler { } , anyWellKnownJSON , anyWellKnown } ,
{ "Any with WKT and indent" , Unmarshaler { } , anyWellKnownPrettyJSON , anyWellKnown } ,
2018-11-26 18:23:56 +00:00
{ "map<string, enum>" , Unmarshaler { } , ` { "enumy": { "XIV":"ROMAN"}} ` , & pb . Mappy { Enumy : map [ string ] pb . Numeral { "XIV" : pb . Numeral_ROMAN } } } ,
2018-01-09 18:57:14 +00:00
{ "map<string, enum as int>" , Unmarshaler { } , ` { "enumy": { "XIV":2}} ` , & pb . Mappy { Enumy : map [ string ] pb . Numeral { "XIV" : pb . Numeral_ROMAN } } } ,
{ "oneof" , Unmarshaler { } , ` { "salary":31000} ` , & pb . MsgWithOneof { Union : & pb . MsgWithOneof_Salary { 31000 } } } ,
{ "oneof spec name" , Unmarshaler { } , ` { "Country":"Australia"} ` , & pb . MsgWithOneof { Union : & pb . MsgWithOneof_Country { "Australia" } } } ,
{ "oneof orig_name" , Unmarshaler { } , ` { "Country":"Australia"} ` , & pb . MsgWithOneof { Union : & pb . MsgWithOneof_Country { "Australia" } } } ,
{ "oneof spec name2" , Unmarshaler { } , ` { "homeAddress":"Australia"} ` , & pb . MsgWithOneof { Union : & pb . MsgWithOneof_HomeAddress { "Australia" } } } ,
{ "oneof orig_name2" , Unmarshaler { } , ` { "home_address":"Australia"} ` , & pb . MsgWithOneof { Union : & pb . MsgWithOneof_HomeAddress { "Australia" } } } ,
{ "orig_name input" , Unmarshaler { } , ` { "o_bool":true} ` , & pb . Simple { OBool : proto . Bool ( true ) } } ,
{ "camelName input" , Unmarshaler { } , ` { "oBool":true} ` , & pb . Simple { OBool : proto . Bool ( true ) } } ,
{ "Duration" , Unmarshaler { } , ` { "dur":"3.000s"} ` , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 3 } } } ,
2018-07-18 14:47:22 +00:00
{ "Duration" , Unmarshaler { } , ` { "dur":"4s"} ` , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 4 } } } ,
2018-11-26 18:23:56 +00:00
{ "Duration with unicode" , Unmarshaler { } , ` { "dur": "3\u0073"} ` , & pb . KnownTypes { Dur : & durpb . Duration { Seconds : 3 } } } ,
2018-01-09 18:57:14 +00:00
{ "null Duration" , Unmarshaler { } , ` { "dur":null} ` , & pb . KnownTypes { Dur : nil } } ,
{ "Timestamp" , Unmarshaler { } , ` { "ts":"2014-05-13T16:53:20.021Z"} ` , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 14e8 , Nanos : 21e6 } } } ,
2018-07-18 14:47:22 +00:00
{ "Timestamp" , Unmarshaler { } , ` { "ts":"2014-05-13T16:53:20Z"} ` , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 14e8 , Nanos : 0 } } } ,
2018-11-26 18:23:56 +00:00
{ "Timestamp with unicode" , Unmarshaler { } , ` { "ts": "2014-05-13T16:53:20\u005a"} ` , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : 14e8 , Nanos : 0 } } } ,
2018-01-09 18:57:14 +00:00
{ "PreEpochTimestamp" , Unmarshaler { } , ` { "ts":"1969-12-31T23:59:58.999999995Z"} ` , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : - 2 , Nanos : 999999995 } } } ,
{ "ZeroTimeTimestamp" , Unmarshaler { } , ` { "ts":"0001-01-01T00:00:00Z"} ` , & pb . KnownTypes { Ts : & tspb . Timestamp { Seconds : - 62135596800 , Nanos : 0 } } } ,
{ "null Timestamp" , Unmarshaler { } , ` { "ts":null} ` , & pb . KnownTypes { Ts : nil } } ,
{ "null Struct" , Unmarshaler { } , ` { "st": null} ` , & pb . KnownTypes { St : nil } } ,
{ "empty Struct" , Unmarshaler { } , ` { "st": { }} ` , & pb . KnownTypes { St : & stpb . Struct { } } } ,
{ "basic Struct" , Unmarshaler { } , ` { "st": { "a": "x", "b": null, "c": 3, "d": true}} ` , & pb . KnownTypes { St : & stpb . Struct { Fields : map [ string ] * stpb . Value {
"a" : { Kind : & stpb . Value_StringValue { "x" } } ,
"b" : { Kind : & stpb . Value_NullValue { } } ,
"c" : { Kind : & stpb . Value_NumberValue { 3 } } ,
"d" : { Kind : & stpb . Value_BoolValue { true } } ,
} } } } ,
{ "nested Struct" , Unmarshaler { } , ` { "st": { "a": { "b": 1, "c": [ { "d": true}, "f"]}}} ` , & pb . KnownTypes { St : & stpb . Struct { Fields : map [ string ] * stpb . Value {
"a" : { Kind : & stpb . Value_StructValue { & stpb . Struct { Fields : map [ string ] * stpb . Value {
"b" : { Kind : & stpb . Value_NumberValue { 1 } } ,
"c" : { Kind : & stpb . Value_ListValue { & stpb . ListValue { Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_StructValue { & stpb . Struct { Fields : map [ string ] * stpb . Value { "d" : { Kind : & stpb . Value_BoolValue { true } } } } } } ,
{ Kind : & stpb . Value_StringValue { "f" } } ,
} } } } ,
} } } } ,
} } } } ,
{ "null ListValue" , Unmarshaler { } , ` { "lv": null} ` , & pb . KnownTypes { Lv : nil } } ,
{ "empty ListValue" , Unmarshaler { } , ` { "lv": []} ` , & pb . KnownTypes { Lv : & stpb . ListValue { } } } ,
{ "basic ListValue" , Unmarshaler { } , ` { "lv": ["x", null, 3, true]} ` , & pb . KnownTypes { Lv : & stpb . ListValue { Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_StringValue { "x" } } ,
{ Kind : & stpb . Value_NullValue { } } ,
{ Kind : & stpb . Value_NumberValue { 3 } } ,
{ Kind : & stpb . Value_BoolValue { true } } ,
} } } } ,
{ "number Value" , Unmarshaler { } , ` { "val":1} ` , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_NumberValue { 1 } } } } ,
{ "null Value" , Unmarshaler { } , ` { "val":null} ` , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_NullValue { stpb . NullValue_NULL_VALUE } } } } ,
{ "bool Value" , Unmarshaler { } , ` { "val":true} ` , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_BoolValue { true } } } } ,
{ "string Value" , Unmarshaler { } , ` { "val":"x"} ` , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_StringValue { "x" } } } } ,
{ "string number value" , Unmarshaler { } , ` { "val":"9223372036854775807"} ` , & pb . KnownTypes { Val : & stpb . Value { Kind : & stpb . Value_StringValue { "9223372036854775807" } } } } ,
{ "list of lists Value" , Unmarshaler { } , ` { "val":["x", [["y"], "z"]]} ` , & pb . KnownTypes { Val : & stpb . Value {
Kind : & stpb . Value_ListValue { & stpb . ListValue {
Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_StringValue { "x" } } ,
{ Kind : & stpb . Value_ListValue { & stpb . ListValue {
Values : [ ] * stpb . Value {
{ Kind : & stpb . Value_ListValue { & stpb . ListValue {
Values : [ ] * stpb . Value { { Kind : & stpb . Value_StringValue { "y" } } } ,
} } } ,
{ Kind : & stpb . Value_StringValue { "z" } } ,
} ,
} } } ,
} ,
} } } } } ,
{ "DoubleValue" , Unmarshaler { } , ` { "dbl":1.2} ` , & pb . KnownTypes { Dbl : & wpb . DoubleValue { Value : 1.2 } } } ,
{ "FloatValue" , Unmarshaler { } , ` { "flt":1.2} ` , & pb . KnownTypes { Flt : & wpb . FloatValue { Value : 1.2 } } } ,
{ "Int64Value" , Unmarshaler { } , ` { "i64":"-3"} ` , & pb . KnownTypes { I64 : & wpb . Int64Value { Value : - 3 } } } ,
{ "UInt64Value" , Unmarshaler { } , ` { "u64":"3"} ` , & pb . KnownTypes { U64 : & wpb . UInt64Value { Value : 3 } } } ,
{ "Int32Value" , Unmarshaler { } , ` { "i32":-4} ` , & pb . KnownTypes { I32 : & wpb . Int32Value { Value : - 4 } } } ,
{ "UInt32Value" , Unmarshaler { } , ` { "u32":4} ` , & pb . KnownTypes { U32 : & wpb . UInt32Value { Value : 4 } } } ,
{ "BoolValue" , Unmarshaler { } , ` { "bool":true} ` , & pb . KnownTypes { Bool : & wpb . BoolValue { Value : true } } } ,
{ "StringValue" , Unmarshaler { } , ` { "str":"plush"} ` , & pb . KnownTypes { Str : & wpb . StringValue { Value : "plush" } } } ,
2018-11-26 18:23:56 +00:00
{ "StringValue containing escaped character" , Unmarshaler { } , ` { "str":"a\/b"} ` , & pb . KnownTypes { Str : & wpb . StringValue { Value : "a/b" } } } ,
{ "StructValue containing StringValue's" , Unmarshaler { } , ` { "escaped": "a\/b", "unicode": "\u00004E16\u0000754C"} ` ,
& stpb . Struct {
Fields : map [ string ] * stpb . Value {
"escaped" : { Kind : & stpb . Value_StringValue { "a/b" } } ,
"unicode" : { Kind : & stpb . Value_StringValue { "\u00004E16\u0000754C" } } ,
} ,
} } ,
2018-01-09 18:57:14 +00:00
{ "BytesValue" , Unmarshaler { } , ` { "bytes":"d293"} ` , & pb . KnownTypes { Bytes : & wpb . BytesValue { Value : [ ] byte ( "wow" ) } } } ,
// Ensure that `null` as a value ends up with a nil pointer instead of a [type]Value struct.
{ "null DoubleValue" , Unmarshaler { } , ` { "dbl":null} ` , & pb . KnownTypes { Dbl : nil } } ,
{ "null FloatValue" , Unmarshaler { } , ` { "flt":null} ` , & pb . KnownTypes { Flt : nil } } ,
{ "null Int64Value" , Unmarshaler { } , ` { "i64":null} ` , & pb . KnownTypes { I64 : nil } } ,
{ "null UInt64Value" , Unmarshaler { } , ` { "u64":null} ` , & pb . KnownTypes { U64 : nil } } ,
{ "null Int32Value" , Unmarshaler { } , ` { "i32":null} ` , & pb . KnownTypes { I32 : nil } } ,
{ "null UInt32Value" , Unmarshaler { } , ` { "u32":null} ` , & pb . KnownTypes { U32 : nil } } ,
{ "null BoolValue" , Unmarshaler { } , ` { "bool":null} ` , & pb . KnownTypes { Bool : nil } } ,
{ "null StringValue" , Unmarshaler { } , ` { "str":null} ` , & pb . KnownTypes { Str : nil } } ,
{ "null BytesValue" , Unmarshaler { } , ` { "bytes":null} ` , & pb . KnownTypes { Bytes : nil } } ,
2018-07-18 14:47:22 +00:00
{ "required" , Unmarshaler { } , ` { "str":"hello"} ` , & pb . MsgWithRequired { Str : proto . String ( "hello" ) } } ,
{ "required bytes" , Unmarshaler { } , ` { "byts": []} ` , & pb . MsgWithRequiredBytes { Byts : [ ] byte { } } } ,
2018-01-09 18:57:14 +00:00
}
func TestUnmarshaling ( t * testing . T ) {
for _ , tt := range unmarshalingTests {
// Make a new instance of the type of our expected object.
p := reflect . New ( reflect . TypeOf ( tt . pb ) . Elem ( ) ) . Interface ( ) . ( proto . Message )
err := tt . unmarshaler . Unmarshal ( strings . NewReader ( tt . json ) , p )
if err != nil {
2018-11-26 18:23:56 +00:00
t . Errorf ( "unmarshalling %s: %v" , tt . desc , err )
2018-01-09 18:57:14 +00:00
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 )
}
}
}
func TestUnmarshalNullArray ( t * testing . T ) {
var repeats pb . Repeats
if err := UnmarshalString ( ` { "rBool":null} ` , & repeats ) ; err != nil {
t . Fatal ( err )
}
if ! reflect . DeepEqual ( repeats , pb . Repeats { } ) {
t . Errorf ( "got non-nil fields in [%#v]" , repeats )
}
}
func TestUnmarshalNullObject ( t * testing . T ) {
var maps pb . Maps
if err := UnmarshalString ( ` { "mInt64Str":null} ` , & maps ) ; err != nil {
t . Fatal ( err )
}
if ! reflect . DeepEqual ( maps , pb . Maps { } ) {
t . Errorf ( "got non-nil fields in [%#v]" , maps )
}
}
func TestUnmarshalNext ( t * testing . T ) {
// We only need to check against a few, not all of them.
tests := unmarshalingTests [ : 5 ]
// Create a buffer with many concatenated JSON objects.
var b bytes . Buffer
for _ , tt := range tests {
b . WriteString ( tt . json )
}
dec := json . NewDecoder ( & b )
for _ , tt := range tests {
// Make a new instance of the type of our expected object.
p := reflect . New ( reflect . TypeOf ( tt . pb ) . Elem ( ) ) . Interface ( ) . ( proto . Message )
err := tt . unmarshaler . UnmarshalNext ( dec , p )
if err != nil {
t . Errorf ( "%s: %v" , tt . desc , 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 )
}
}
p := & pb . Simple { }
err := new ( Unmarshaler ) . UnmarshalNext ( dec , p )
if err != io . EOF {
t . Errorf ( "eof: got %v, expected io.EOF" , err )
}
}
var unmarshalingShouldError = [ ] struct {
desc string
in string
pb proto . Message
} {
{ "a value" , "666" , new ( pb . Simple ) } ,
{ "gibberish" , "{adskja123;l23=-=" , new ( pb . Simple ) } ,
{ "unknown field" , ` { "unknown": "foo"} ` , new ( pb . Simple ) } ,
{ "unknown enum name" , ` { "hilarity":"DAVE"} ` , new ( proto3pb . Message ) } ,
2018-11-26 18:23:56 +00:00
{ "Duration containing invalid character" , ` { "dur": "3\U0073"} ` , & pb . KnownTypes { } } ,
{ "Timestamp containing invalid character" , ` { "ts": "2014-05-13T16:53:20\U005a"} ` , & pb . KnownTypes { } } ,
{ "StringValue containing invalid character" , ` { "str": "\U00004E16\U0000754C"} ` , & pb . KnownTypes { } } ,
{ "StructValue containing invalid character" , ` { "str": "\U00004E16\U0000754C"} ` , & stpb . Struct { } } ,
{ "repeated proto3 enum with non array input" , ` { "rFunny":"PUNS"} ` , & proto3pb . Message { RFunny : [ ] proto3pb . Message_Humour { } } } ,
2018-01-09 18:57:14 +00:00
}
func TestUnmarshalingBadInput ( t * testing . T ) {
for _ , tt := range unmarshalingShouldError {
err := UnmarshalString ( tt . in , tt . pb )
if err == nil {
t . Errorf ( "an error was expected when parsing %q instead of an object" , tt . desc )
}
}
}
type funcResolver func ( turl string ) ( proto . Message , error )
func ( fn funcResolver ) Resolve ( turl string ) ( proto . Message , error ) {
return fn ( turl )
}
func TestAnyWithCustomResolver ( t * testing . T ) {
var resolvedTypeUrls [ ] string
resolver := funcResolver ( func ( turl string ) ( proto . Message , error ) {
resolvedTypeUrls = append ( resolvedTypeUrls , turl )
return new ( pb . Simple ) , nil
} )
msg := & pb . Simple {
OBytes : [ ] byte { 1 , 2 , 3 , 4 } ,
OBool : proto . Bool ( true ) ,
OString : proto . String ( "foobar" ) ,
OInt64 : proto . Int64 ( 1020304 ) ,
}
msgBytes , err := proto . Marshal ( msg )
if err != nil {
t . Errorf ( "an unexpected error occurred when marshaling message: %v" , err )
}
// make an Any with a type URL that won't resolve w/out custom resolver
any := & anypb . Any {
TypeUrl : "https://foobar.com/some.random.MessageKind" ,
Value : msgBytes ,
}
m := Marshaler { AnyResolver : resolver }
js , err := m . MarshalToString ( any )
if err != nil {
t . Errorf ( "an unexpected error occurred when marshaling any to JSON: %v" , err )
}
if len ( resolvedTypeUrls ) != 1 {
t . Errorf ( "custom resolver was not invoked during marshaling" )
} else if resolvedTypeUrls [ 0 ] != "https://foobar.com/some.random.MessageKind" {
t . Errorf ( "custom resolver was invoked with wrong URL: got %q, wanted %q" , resolvedTypeUrls [ 0 ] , "https://foobar.com/some.random.MessageKind" )
}
wanted := ` { "@type":"https://foobar.com/some.random.MessageKind","oBool":true,"oInt64":"1020304","oString":"foobar","oBytes":"AQIDBA=="} `
if js != wanted {
t . Errorf ( "marshalling JSON produced incorrect output: got %s, wanted %s" , js , wanted )
}
u := Unmarshaler { AnyResolver : resolver }
roundTrip := & anypb . Any { }
err = u . Unmarshal ( bytes . NewReader ( [ ] byte ( js ) ) , roundTrip )
if err != nil {
t . Errorf ( "an unexpected error occurred when unmarshaling any from JSON: %v" , err )
}
if len ( resolvedTypeUrls ) != 2 {
t . Errorf ( "custom resolver was not invoked during marshaling" )
} else if resolvedTypeUrls [ 1 ] != "https://foobar.com/some.random.MessageKind" {
t . Errorf ( "custom resolver was invoked with wrong URL: got %q, wanted %q" , resolvedTypeUrls [ 1 ] , "https://foobar.com/some.random.MessageKind" )
}
if ! proto . Equal ( any , roundTrip ) {
t . Errorf ( "message contents not set correctly after unmarshalling JSON: got %s, wanted %s" , roundTrip , any )
}
}
func TestUnmarshalJSONPBUnmarshaler ( t * testing . T ) {
rawJson := ` { "foo": "bar", "baz": [0, 1, 2, 3] } `
var msg dynamicMessage
if err := Unmarshal ( strings . NewReader ( rawJson ) , & msg ) ; err != nil {
t . Errorf ( "an unexpected error occurred when parsing into JSONPBUnmarshaler: %v" , err )
}
2018-11-26 18:23:56 +00:00
if msg . RawJson != rawJson {
t . Errorf ( "message contents not set correctly after unmarshalling JSON: got %s, wanted %s" , msg . RawJson , rawJson )
2018-01-09 18:57:14 +00:00
}
}
func TestUnmarshalNullWithJSONPBUnmarshaler ( t * testing . T ) {
rawJson := ` { "stringField":null} `
var ptrFieldMsg ptrFieldMessage
if err := Unmarshal ( strings . NewReader ( rawJson ) , & ptrFieldMsg ) ; err != nil {
t . Errorf ( "unmarshal error: %v" , err )
}
want := ptrFieldMessage { StringField : & stringField { IsSet : true , StringValue : "null" } }
if ! proto . Equal ( & ptrFieldMsg , & want ) {
t . Errorf ( "unmarshal result StringField: got %v, want %v" , ptrFieldMsg , want )
}
}
func TestUnmarshalAnyJSONPBUnmarshaler ( t * testing . T ) {
rawJson := ` { "@type": "blah.com/ ` + dynamicMessageName + ` ", "foo": "bar", "baz": [0, 1, 2, 3] } `
var got anypb . Any
if err := Unmarshal ( strings . NewReader ( rawJson ) , & got ) ; err != nil {
t . Errorf ( "an unexpected error occurred when parsing into JSONPBUnmarshaler: %v" , err )
}
2018-11-26 18:23:56 +00:00
dm := & dynamicMessage { RawJson : ` { "baz":[0,1,2,3],"foo":"bar"} ` }
2018-01-09 18:57:14 +00:00
var want anypb . Any
if b , err := proto . Marshal ( dm ) ; err != nil {
t . Errorf ( "an unexpected error occurred when marshaling message: %v" , err )
} else {
want . TypeUrl = "blah.com/" + dynamicMessageName
want . Value = b
}
if ! proto . Equal ( & got , & want ) {
2018-07-18 14:47:22 +00:00
t . Errorf ( "message contents not set correctly after unmarshalling JSON: got %v, wanted %v" , got , want )
2018-01-09 18:57:14 +00:00
}
}
const (
dynamicMessageName = "google.protobuf.jsonpb.testing.dynamicMessage"
)
func init ( ) {
// we register the custom type below so that we can use it in Any types
proto . RegisterType ( ( * dynamicMessage ) ( nil ) , dynamicMessageName )
}
type ptrFieldMessage struct {
StringField * stringField ` protobuf:"bytes,1,opt,name=stringField" `
}
func ( m * ptrFieldMessage ) Reset ( ) {
}
func ( m * ptrFieldMessage ) String ( ) string {
return m . StringField . StringValue
}
func ( m * ptrFieldMessage ) ProtoMessage ( ) {
}
type stringField struct {
IsSet bool ` protobuf:"varint,1,opt,name=isSet" `
StringValue string ` protobuf:"bytes,2,opt,name=stringValue" `
}
func ( s * stringField ) Reset ( ) {
}
func ( s * stringField ) String ( ) string {
return s . StringValue
}
func ( s * stringField ) ProtoMessage ( ) {
}
func ( s * stringField ) UnmarshalJSONPB ( jum * Unmarshaler , js [ ] byte ) error {
s . IsSet = true
s . StringValue = string ( js )
return nil
}
// dynamicMessage implements protobuf.Message but is not a normal generated message type.
// It provides implementations of JSONPBMarshaler and JSONPBUnmarshaler for JSON support.
type dynamicMessage struct {
2018-11-26 18:23:56 +00:00
RawJson string ` protobuf:"bytes,1,opt,name=rawJson" `
2018-07-18 14:47:22 +00:00
// an unexported nested message is present just to ensure that it
// won't result in a panic (see issue #509)
2018-11-26 18:23:56 +00:00
Dummy * dynamicMessage ` protobuf:"bytes,2,opt,name=dummy" `
2018-01-09 18:57:14 +00:00
}
func ( m * dynamicMessage ) Reset ( ) {
2018-11-26 18:23:56 +00:00
m . RawJson = "{}"
2018-01-09 18:57:14 +00:00
}
func ( m * dynamicMessage ) String ( ) string {
2018-11-26 18:23:56 +00:00
return m . RawJson
2018-01-09 18:57:14 +00:00
}
func ( m * dynamicMessage ) ProtoMessage ( ) {
}
func ( m * dynamicMessage ) MarshalJSONPB ( jm * Marshaler ) ( [ ] byte , error ) {
2018-11-26 18:23:56 +00:00
return [ ] byte ( m . RawJson ) , nil
2018-01-09 18:57:14 +00:00
}
func ( m * dynamicMessage ) UnmarshalJSONPB ( jum * Unmarshaler , js [ ] byte ) error {
2018-11-26 18:23:56 +00:00
m . RawJson = string ( js )
2018-01-09 18:57:14 +00:00
return nil
}
2018-07-18 14:47:22 +00:00
// Test unmarshaling message containing unset required fields should produce error.
func TestUnmarshalUnsetRequiredFields ( t * testing . T ) {
tests := [ ] struct {
desc string
pb proto . Message
json string
} {
{
desc : "direct required field missing" ,
pb : & pb . MsgWithRequired { } ,
json : ` { } ` ,
} ,
{
desc : "direct required field set to null" ,
pb : & pb . MsgWithRequired { } ,
json : ` { "str": null} ` ,
} ,
{
desc : "indirect required field missing" ,
pb : & pb . MsgWithIndirectRequired { } ,
json : ` { "subm": { }} ` ,
} ,
{
desc : "indirect required field set to null" ,
pb : & pb . MsgWithIndirectRequired { } ,
json : ` { "subm": { "str": null}} ` ,
} ,
{
desc : "direct required bytes field missing" ,
pb : & pb . MsgWithRequiredBytes { } ,
json : ` { } ` ,
} ,
{
desc : "direct required bytes field set to null" ,
pb : & pb . MsgWithRequiredBytes { } ,
json : ` { "byts": null} ` ,
} ,
{
desc : "direct required wkt field missing" ,
pb : & pb . MsgWithRequiredWKT { } ,
json : ` { } ` ,
} ,
{
desc : "direct required wkt field set to null" ,
pb : & pb . MsgWithRequiredWKT { } ,
json : ` { "str": null} ` ,
} ,
{
desc : "any containing message with required field set to null" ,
pb : & pb . KnownTypes { } ,
json : ` { "an": { "@type": "example.com/jsonpb.MsgWithRequired", "str": null}} ` ,
} ,
{
desc : "any containing message with missing required field" ,
pb : & pb . KnownTypes { } ,
json : ` { "an": { "@type": "example.com/jsonpb.MsgWithRequired"}} ` ,
} ,
{
desc : "missing required in map value" ,
pb : & pb . MsgWithIndirectRequired { } ,
json : ` { "map_field": { "a": { }, "b": { "str": "hi"}}} ` ,
} ,
{
desc : "required in map value set to null" ,
pb : & pb . MsgWithIndirectRequired { } ,
json : ` { "map_field": { "a": { "str": "hello"}, "b": { "str": null}}} ` ,
} ,
{
desc : "missing required in slice item" ,
pb : & pb . MsgWithIndirectRequired { } ,
json : ` { "slice_field": [ { }, { "str": "hi"}]} ` ,
} ,
{
desc : "required in slice item set to null" ,
pb : & pb . MsgWithIndirectRequired { } ,
json : ` { "slice_field": [ { "str": "hello"}, { "str": null}]} ` ,
} ,
{
desc : "required inside oneof missing" ,
pb : & pb . MsgWithOneof { } ,
json : ` { "msgWithRequired": { }} ` ,
} ,
{
desc : "required inside oneof set to null" ,
pb : & pb . MsgWithOneof { } ,
json : ` { "msgWithRequired": { "str": null}} ` ,
} ,
{
desc : "required field in extension missing" ,
pb : & pb . Real { } ,
json : ` { "[jsonpb.extm]": { }} ` ,
} ,
{
desc : "required field in extension set to null" ,
pb : & pb . Real { } ,
json : ` { "[jsonpb.extm]": { "str": null}} ` ,
} ,
}
for _ , tc := range tests {
if err := UnmarshalString ( tc . json , tc . pb ) ; err == nil {
t . Errorf ( "%s: expecting error in unmarshaling with unset required fields %s" , tc . desc , tc . json )
}
}
}