123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506 |
- // Copyright (c) 2012-2015 Ugorji Nwoke. All rights reserved.
- // Use of this source code is governed by a MIT license found in the LICENSE file.
- package codec
- // Test works by using a slice of interfaces.
- // It can test for encoding/decoding into/from a nil interface{}
- // or passing the object to encode/decode into.
- //
- // There are basically 2 main tests here.
- // First test internally encodes and decodes things and verifies that
- // the artifact was as expected.
- // Second test will use python msgpack to create a bunch of golden files,
- // read those files, and compare them to what it should be. It then
- // writes those files back out and compares the byte streams.
- //
- // Taken together, the tests are pretty extensive.
- //
- // The following manual tests must be done:
- // - TestCodecUnderlyingType
- import (
- "bytes"
- "encoding/gob"
- "flag"
- "fmt"
- "io/ioutil"
- "math"
- "math/rand"
- "net"
- "net/rpc"
- "os"
- "os/exec"
- "path/filepath"
- "reflect"
- "runtime"
- "strconv"
- "strings"
- "sync/atomic"
- "testing"
- "time"
- )
- func init() {
- testInitFlags()
- testPreInitFns = append(testPreInitFns, testInit)
- }
- // make this a mapbyslice
- type testMbsT []interface{}
- func (_ testMbsT) MapBySlice() {}
- type testVerifyArg int
- const (
- testVerifyMapTypeSame testVerifyArg = iota
- testVerifyMapTypeStrIntf
- testVerifyMapTypeIntfIntf
- // testVerifySliceIntf
- testVerifyForPython
- )
- const testSkipRPCTests = false
- var (
- testTableNumPrimitives int
- testTableIdxTime int
- testTableNumMaps int
- )
- var (
- skipVerifyVal interface{} = &(struct{}{})
- testMapStrIntfTyp = reflect.TypeOf(map[string]interface{}(nil))
- // For Go Time, do not use a descriptive timezone.
- // It's unnecessary, and makes it harder to do a reflect.DeepEqual.
- // The Offset already tells what the offset should be, if not on UTC and unknown zone name.
- timeLoc = time.FixedZone("", -8*60*60) // UTC-08:00 //time.UTC-8
- timeToCompare1 = time.Date(2012, 2, 2, 2, 2, 2, 2000, timeLoc).UTC()
- timeToCompare2 = time.Date(1900, 2, 2, 2, 2, 2, 2000, timeLoc).UTC()
- timeToCompare3 = time.Unix(0, 270).UTC() // use value that must be encoded as uint64 for nanoseconds (for cbor/msgpack comparison)
- //timeToCompare4 = time.Time{}.UTC() // does not work well with simple cbor time encoding (overflow)
- timeToCompare4 = time.Unix(-2013855848, 4223).UTC()
- table []interface{} // main items we encode
- tableVerify []interface{} // we verify encoded things against this after decode
- tableTestNilVerify []interface{} // for nil interface, use this to verify (rules are different)
- tablePythonVerify []interface{} // for verifying for python, since Python sometimes
- // will encode a float32 as float64, or large int as uint
- testRpcInt = new(TestRpcInt)
- )
- func testInitFlags() {
- // delete(testDecOpts.ExtFuncs, timeTyp)
- flag.BoolVar(&testVerbose, "tv", false, "Test Verbose")
- flag.BoolVar(&testInitDebug, "tg", false, "Test Init Debug")
- flag.BoolVar(&testUseIoEncDec, "ti", false, "Use IO Reader/Writer for Marshal/Unmarshal")
- flag.BoolVar(&testStructToArray, "ts", false, "Set StructToArray option")
- flag.BoolVar(&testWriteNoSymbols, "tn", false, "Set NoSymbols option")
- flag.BoolVar(&testCanonical, "tc", false, "Set Canonical option")
- flag.BoolVar(&testInternStr, "te", false, "Set InternStr option")
- flag.BoolVar(&testSkipIntf, "tf", false, "Skip Interfaces")
- flag.BoolVar(&testUseReset, "tr", false, "Use Reset")
- flag.IntVar(&testJsonIndent, "td", 0, "Use JSON Indent")
- flag.BoolVar(&testUseMust, "tm", true, "Use Must(En|De)code")
- flag.BoolVar(&testCheckCircRef, "tl", false, "Use Check Circular Ref")
- }
- func testByteBuf(in []byte) *bytes.Buffer {
- return bytes.NewBuffer(in)
- }
- type TestABC struct {
- A, B, C string
- }
- func (x *TestABC) MarshalBinary() ([]byte, error) {
- return []byte(fmt.Sprintf("%s %s %s", x.A, x.B, x.C)), nil
- }
- func (x *TestABC) MarshalText() ([]byte, error) {
- return []byte(fmt.Sprintf("%s %s %s", x.A, x.B, x.C)), nil
- }
- func (x *TestABC) MarshalJSON() ([]byte, error) {
- return []byte(fmt.Sprintf(`"%s %s %s"`, x.A, x.B, x.C)), nil
- }
- func (x *TestABC) UnmarshalBinary(data []byte) (err error) {
- ss := strings.Split(string(data), " ")
- x.A, x.B, x.C = ss[0], ss[1], ss[2]
- return
- }
- func (x *TestABC) UnmarshalText(data []byte) (err error) {
- return x.UnmarshalBinary(data)
- }
- func (x *TestABC) UnmarshalJSON(data []byte) (err error) {
- return x.UnmarshalBinary(data[1 : len(data)-1])
- }
- type TestABC2 struct {
- A, B, C string
- }
- func (x TestABC2) MarshalText() ([]byte, error) {
- return []byte(fmt.Sprintf("%s %s %s", x.A, x.B, x.C)), nil
- }
- func (x *TestABC2) UnmarshalText(data []byte) (err error) {
- ss := strings.Split(string(data), " ")
- x.A, x.B, x.C = ss[0], ss[1], ss[2]
- return
- // _, err = fmt.Sscanf(string(data), "%s %s %s", &x.A, &x.B, &x.C)
- }
- type TestRpcABC struct {
- A, B, C string
- }
- type TestRpcInt struct {
- i int
- }
- func (r *TestRpcInt) Update(n int, res *int) error { r.i = n; *res = r.i; return nil }
- func (r *TestRpcInt) Square(ignore int, res *int) error { *res = r.i * r.i; return nil }
- func (r *TestRpcInt) Mult(n int, res *int) error { *res = r.i * n; return nil }
- func (r *TestRpcInt) EchoStruct(arg TestRpcABC, res *string) error {
- *res = fmt.Sprintf("%#v", arg)
- return nil
- }
- func (r *TestRpcInt) Echo123(args []string, res *string) error {
- *res = fmt.Sprintf("%#v", args)
- return nil
- }
- type TestRawValue struct {
- R Raw
- I int
- }
- type testUnixNanoTimeExt struct {
- // keep timestamp here, so that do not incur interface-conversion costs
- ts int64
- }
- // func (x *testUnixNanoTimeExt) WriteExt(interface{}) []byte { panic("unsupported") }
- // func (x *testUnixNanoTimeExt) ReadExt(interface{}, []byte) { panic("unsupported") }
- func (x *testUnixNanoTimeExt) ConvertExt(v interface{}) interface{} {
- switch v2 := v.(type) {
- case time.Time:
- x.ts = v2.UTC().UnixNano()
- case *time.Time:
- x.ts = v2.UTC().UnixNano()
- default:
- panic(fmt.Sprintf("unsupported format for time conversion: expecting time.Time; got %T", v))
- }
- return &x.ts
- }
- func (x *testUnixNanoTimeExt) UpdateExt(dest interface{}, v interface{}) {
- // fmt.Printf("testUnixNanoTimeExt.UpdateExt: v: %v\n", v)
- tt := dest.(*time.Time)
- switch v2 := v.(type) {
- case int64:
- *tt = time.Unix(0, v2).UTC()
- case *int64:
- *tt = time.Unix(0, *v2).UTC()
- case uint64:
- *tt = time.Unix(0, int64(v2)).UTC()
- case *uint64:
- *tt = time.Unix(0, int64(*v2)).UTC()
- //case float64:
- //case string:
- default:
- panic(fmt.Sprintf("unsupported format for time conversion: expecting int64/uint64; got %T", v))
- }
- // fmt.Printf("testUnixNanoTimeExt.UpdateExt: v: %v, tt: %#v\n", v, tt)
- }
- func testVerifyVal(v interface{}, arg testVerifyArg) (v2 interface{}) {
- //for python msgpack,
- // - all positive integers are unsigned 64-bit ints
- // - all floats are float64
- switch iv := v.(type) {
- case int8:
- if iv >= 0 {
- v2 = uint64(iv)
- } else {
- v2 = int64(iv)
- }
- case int16:
- if iv >= 0 {
- v2 = uint64(iv)
- } else {
- v2 = int64(iv)
- }
- case int32:
- if iv >= 0 {
- v2 = uint64(iv)
- } else {
- v2 = int64(iv)
- }
- case int64:
- if iv >= 0 {
- v2 = uint64(iv)
- } else {
- v2 = int64(iv)
- }
- case uint8:
- v2 = uint64(iv)
- case uint16:
- v2 = uint64(iv)
- case uint32:
- v2 = uint64(iv)
- case uint64:
- v2 = uint64(iv)
- case float32:
- v2 = float64(iv)
- case float64:
- v2 = float64(iv)
- case []interface{}:
- m2 := make([]interface{}, len(iv))
- for j, vj := range iv {
- m2[j] = testVerifyVal(vj, arg)
- }
- v2 = m2
- case testMbsT:
- m2 := make([]interface{}, len(iv))
- for j, vj := range iv {
- m2[j] = testVerifyVal(vj, arg)
- }
- v2 = testMbsT(m2)
- case map[string]bool:
- switch arg {
- case testVerifyMapTypeSame:
- m2 := make(map[string]bool)
- for kj, kv := range iv {
- m2[kj] = kv
- }
- v2 = m2
- case testVerifyMapTypeStrIntf, testVerifyForPython:
- m2 := make(map[string]interface{})
- for kj, kv := range iv {
- m2[kj] = kv
- }
- v2 = m2
- case testVerifyMapTypeIntfIntf:
- m2 := make(map[interface{}]interface{})
- for kj, kv := range iv {
- m2[kj] = kv
- }
- v2 = m2
- }
- case map[string]interface{}:
- switch arg {
- case testVerifyMapTypeSame:
- m2 := make(map[string]interface{})
- for kj, kv := range iv {
- m2[kj] = testVerifyVal(kv, arg)
- }
- v2 = m2
- case testVerifyMapTypeStrIntf, testVerifyForPython:
- m2 := make(map[string]interface{})
- for kj, kv := range iv {
- m2[kj] = testVerifyVal(kv, arg)
- }
- v2 = m2
- case testVerifyMapTypeIntfIntf:
- m2 := make(map[interface{}]interface{})
- for kj, kv := range iv {
- m2[kj] = testVerifyVal(kv, arg)
- }
- v2 = m2
- }
- case map[interface{}]interface{}:
- m2 := make(map[interface{}]interface{})
- for kj, kv := range iv {
- m2[testVerifyVal(kj, arg)] = testVerifyVal(kv, arg)
- }
- v2 = m2
- case time.Time:
- switch arg {
- case testVerifyForPython:
- if iv2 := iv.UnixNano(); iv2 >= 0 {
- v2 = uint64(iv2)
- } else {
- v2 = int64(iv2)
- }
- default:
- v2 = v
- }
- default:
- v2 = v
- }
- return
- }
- func testInit() {
- gob.Register(new(TestStruc))
- if testInitDebug {
- ts0 := newTestStruc(2, false, !testSkipIntf, false)
- fmt.Printf("====> depth: %v, ts: %#v\n", 2, ts0)
- }
- for _, v := range testHandles {
- bh := v.getBasicHandle()
- bh.InternString = testInternStr
- bh.Canonical = testCanonical
- bh.CheckCircularRef = testCheckCircRef
- bh.StructToArray = testStructToArray
- // mostly doing this for binc
- if testWriteNoSymbols {
- bh.AsSymbols = AsSymbolNone
- } else {
- bh.AsSymbols = AsSymbolAll
- }
- }
- testJsonH.Indent = int8(testJsonIndent)
- testMsgpackH.RawToString = true
- // testMsgpackH.AddExt(byteSliceTyp, 0, testMsgpackH.BinaryEncodeExt, testMsgpackH.BinaryDecodeExt)
- // testMsgpackH.AddExt(timeTyp, 1, testMsgpackH.TimeEncodeExt, testMsgpackH.TimeDecodeExt)
- // add extensions for msgpack, simple for time.Time, so we can encode/decode same way.
- // use different flavors of XXXExt calls, including deprecated ones.
- // NOTE:
- // DO NOT set extensions for JsonH, so we can test json(M|Unm)arshal support.
- testSimpleH.AddExt(timeTyp, 1, timeExtEncFn, timeExtDecFn)
- testMsgpackH.SetBytesExt(timeTyp, 1, timeExt{})
- testCborH.SetInterfaceExt(timeTyp, 1, &testUnixNanoTimeExt{})
- // testJsonH.SetInterfaceExt(timeTyp, 1, &testUnixNanoTimeExt{})
- // primitives MUST be an even number, so it can be used as a mapBySlice also.
- primitives := []interface{}{
- int8(-8),
- int16(-1616),
- int32(-32323232),
- int64(-6464646464646464),
- uint8(192),
- uint16(1616),
- uint32(32323232),
- uint64(6464646464646464),
- byte(192),
- float32(-3232.0),
- float64(-6464646464.0),
- float32(3232.0),
- float64(6464.0),
- float64(6464646464.0),
- false,
- true,
- "null",
- nil,
- "someday",
- timeToCompare1,
- "",
- timeToCompare2,
- "bytestring",
- timeToCompare3,
- "none",
- timeToCompare4,
- }
- maps := []interface{}{
- map[string]bool{
- "true": true,
- "false": false,
- },
- map[string]interface{}{
- "true": "True",
- "false": false,
- "uint16(1616)": uint16(1616),
- },
- //add a complex combo map in here. (map has list which has map)
- //note that after the first thing, everything else should be generic.
- map[string]interface{}{
- "list": []interface{}{
- int16(1616),
- int32(32323232),
- true,
- float32(-3232.0),
- map[string]interface{}{
- "TRUE": true,
- "FALSE": false,
- },
- []interface{}{true, false},
- },
- "int32": int32(32323232),
- "bool": true,
- "LONG STRING": "123456789012345678901234567890123456789012345678901234567890",
- "SHORT STRING": "1234567890",
- },
- map[interface{}]interface{}{
- true: "true",
- uint8(138): false,
- "false": uint8(200),
- },
- }
- testTableNumPrimitives = len(primitives)
- testTableIdxTime = testTableNumPrimitives - 8
- testTableNumMaps = len(maps)
- table = []interface{}{}
- table = append(table, primitives...)
- table = append(table, primitives)
- table = append(table, testMbsT(primitives))
- table = append(table, maps...)
- table = append(table, newTestStruc(0, false, !testSkipIntf, false))
- tableVerify = make([]interface{}, len(table))
- tableTestNilVerify = make([]interface{}, len(table))
- tablePythonVerify = make([]interface{}, len(table))
- lp := testTableNumPrimitives + 4
- av := tableVerify
- for i, v := range table {
- if i == lp {
- av[i] = skipVerifyVal
- continue
- }
- //av[i] = testVerifyVal(v, testVerifyMapTypeSame)
- switch v.(type) {
- case []interface{}:
- av[i] = testVerifyVal(v, testVerifyMapTypeSame)
- case testMbsT:
- av[i] = testVerifyVal(v, testVerifyMapTypeSame)
- case map[string]interface{}:
- av[i] = testVerifyVal(v, testVerifyMapTypeSame)
- case map[interface{}]interface{}:
- av[i] = testVerifyVal(v, testVerifyMapTypeSame)
- default:
- av[i] = v
- }
- }
- av = tableTestNilVerify
- for i, v := range table {
- if i > lp {
- av[i] = skipVerifyVal
- continue
- }
- av[i] = testVerifyVal(v, testVerifyMapTypeStrIntf)
- }
- av = tablePythonVerify
- for i, v := range table {
- if i == testTableNumPrimitives+1 || i > lp { // testTableNumPrimitives+1 is the mapBySlice
- av[i] = skipVerifyVal
- continue
- }
- av[i] = testVerifyVal(v, testVerifyForPython)
- }
- // only do the python verify up to the maps, skipping the last 2 maps.
- tablePythonVerify = tablePythonVerify[:testTableNumPrimitives+2+testTableNumMaps-2]
- }
- func testUnmarshal(v interface{}, data []byte, h Handle) (err error) {
- return testCodecDecode(data, v, h)
- }
- func testMarshal(v interface{}, h Handle) (bs []byte, err error) {
- return testCodecEncode(v, nil, testByteBuf, h)
- }
- func testMarshalErr(v interface{}, h Handle, t *testing.T, name string) (bs []byte, err error) {
- if bs, err = testMarshal(v, h); err != nil {
- logT(t, "Error encoding %s: %v, Err: %v", name, v, err)
- t.FailNow()
- }
- return
- }
- func testUnmarshalErr(v interface{}, data []byte, h Handle, t *testing.T, name string) (err error) {
- if err = testUnmarshal(v, data, h); err != nil {
- logT(t, "Error Decoding into %s: %v, Err: %v", name, v, err)
- t.FailNow()
- }
- return
- }
- // doTestCodecTableOne allows us test for different variations based on arguments passed.
- func doTestCodecTableOne(t *testing.T, testNil bool, h Handle,
- vs []interface{}, vsVerify []interface{}) {
- //if testNil, then just test for when a pointer to a nil interface{} is passed. It should work.
- //Current setup allows us test (at least manually) the nil interface or typed interface.
- logT(t, "================ TestNil: %v ================\n", testNil)
- for i, v0 := range vs {
- logT(t, "..............................................")
- logT(t, " Testing: #%d:, %T, %#v\n", i, v0, v0)
- b0, err := testMarshalErr(v0, h, t, "v0")
- if err != nil {
- continue
- }
- if h.isBinary() {
- logT(t, " Encoded bytes: len: %v, %v\n", len(b0), b0)
- } else {
- logT(t, " Encoded string: len: %v, %v\n", len(string(b0)), string(b0))
- // println("########### encoded string: " + string(b0))
- }
- var v1 interface{}
- if testNil {
- err = testUnmarshal(&v1, b0, h)
- } else {
- if v0 != nil {
- v0rt := reflect.TypeOf(v0) // ptr
- rv1 := reflect.New(v0rt)
- err = testUnmarshal(rv1.Interface(), b0, h)
- v1 = rv1.Elem().Interface()
- // v1 = reflect.Indirect(reflect.ValueOf(v1)).Interface()
- }
- }
- logT(t, " v1 returned: %T, %#v", v1, v1)
- // if v1 != nil {
- // logT(t, " v1 returned: %T, %#v", v1, v1)
- // //we always indirect, because ptr to typed value may be passed (if not testNil)
- // v1 = reflect.Indirect(reflect.ValueOf(v1)).Interface()
- // }
- if err != nil {
- logT(t, "-------- Error: %v. Partial return: %v", err, v1)
- failT(t)
- continue
- }
- v0check := vsVerify[i]
- if v0check == skipVerifyVal {
- logT(t, " Nil Check skipped: Decoded: %T, %#v\n", v1, v1)
- continue
- }
- if err = deepEqual(v0check, v1); err == nil {
- logT(t, "++++++++ Before and After marshal matched\n")
- } else {
- // logT(t, "-------- Before and After marshal do not match: Error: %v"+
- // " ====> GOLDEN: (%T) %#v, DECODED: (%T) %#v\n", err, v0check, v0check, v1, v1)
- logT(t, "-------- Before and After marshal do not match: Error: %v", err)
- logT(t, " ....... GOLDEN: (%T) %#v", v0check, v0check)
- logT(t, " ....... DECODED: (%T) %#v", v1, v1)
- failT(t)
- }
- }
- }
- func testCodecTableOne(t *testing.T, h Handle) {
- testOnce.Do(testInitAll)
- // func TestMsgpackAllExperimental(t *testing.T) {
- // dopts := testDecOpts(nil, nil, false, true, true),
- numPrim, numMap, idxTime, idxMap := testTableNumPrimitives, testTableNumMaps, testTableIdxTime, testTableNumPrimitives+2
- //println("#################")
- switch v := h.(type) {
- case *MsgpackHandle:
- var oldWriteExt, oldRawToString bool
- oldWriteExt, v.WriteExt = v.WriteExt, true
- oldRawToString, v.RawToString = v.RawToString, true
- doTestCodecTableOne(t, false, h, table, tableVerify)
- v.WriteExt, v.RawToString = oldWriteExt, oldRawToString
- case *JsonHandle:
- //skip []interface{} containing time.Time, as it encodes as a number, but cannot decode back to time.Time.
- //As there is no real support for extension tags in json, this must be skipped.
- doTestCodecTableOne(t, false, h, table[:numPrim], tableVerify[:numPrim])
- doTestCodecTableOne(t, false, h, table[idxMap:], tableVerify[idxMap:])
- default:
- doTestCodecTableOne(t, false, h, table, tableVerify)
- }
- // func TestMsgpackAll(t *testing.T) {
- // //skip []interface{} containing time.Time
- // doTestCodecTableOne(t, false, h, table[:numPrim], tableVerify[:numPrim])
- // doTestCodecTableOne(t, false, h, table[numPrim+1:], tableVerify[numPrim+1:])
- // func TestMsgpackNilStringMap(t *testing.T) {
- var oldMapType reflect.Type
- v := h.getBasicHandle()
- oldMapType, v.MapType = v.MapType, testMapStrIntfTyp
- //skip time.Time, []interface{} containing time.Time, last map, and newStruc
- doTestCodecTableOne(t, true, h, table[:idxTime], tableTestNilVerify[:idxTime])
- doTestCodecTableOne(t, true, h, table[idxMap:idxMap+numMap-1], tableTestNilVerify[idxMap:idxMap+numMap-1])
- v.MapType = oldMapType
- // func TestMsgpackNilIntf(t *testing.T) {
- //do last map and newStruc
- idx2 := idxMap + numMap - 1
- doTestCodecTableOne(t, true, h, table[idx2:], tableTestNilVerify[idx2:])
- //TODO? What is this one?
- //doTestCodecTableOne(t, true, h, table[17:18], tableTestNilVerify[17:18])
- }
- func testCodecMiscOne(t *testing.T, h Handle) {
- testOnce.Do(testInitAll)
- b, err := testMarshalErr(32, h, t, "32")
- // Cannot do this nil one, because faster type assertion decoding will panic
- // var i *int32
- // if err = testUnmarshal(b, i, nil); err == nil {
- // logT(t, "------- Expecting error because we cannot unmarshal to int32 nil ptr")
- // t.FailNow()
- // }
- var i2 int32 = 0
- err = testUnmarshalErr(&i2, b, h, t, "int32-ptr")
- if i2 != int32(32) {
- logT(t, "------- didn't unmarshal to 32: Received: %d", i2)
- t.FailNow()
- }
- // func TestMsgpackDecodePtr(t *testing.T) {
- ts := newTestStruc(0, false, !testSkipIntf, false)
- b, err = testMarshalErr(ts, h, t, "pointer-to-struct")
- if len(b) < 40 {
- logT(t, "------- Size must be > 40. Size: %d", len(b))
- t.FailNow()
- }
- if h.isBinary() {
- logT(t, "------- b: %v", b)
- } else {
- logT(t, "------- b: %s", b)
- }
- ts2 := new(TestStruc)
- err = testUnmarshalErr(ts2, b, h, t, "pointer-to-struct")
- if ts2.I64 != math.MaxInt64*2/3 {
- logT(t, "------- Unmarshal wrong. Expect I64 = 64. Got: %v", ts2.I64)
- t.FailNow()
- }
- // func TestMsgpackIntfDecode(t *testing.T) {
- m := map[string]int{"A": 2, "B": 3}
- p := []interface{}{m}
- bs, err := testMarshalErr(p, h, t, "p")
- m2 := map[string]int{}
- p2 := []interface{}{m2}
- err = testUnmarshalErr(&p2, bs, h, t, "&p2")
- if m2["A"] != 2 || m2["B"] != 3 {
- logT(t, "m2 not as expected: expecting: %v, got: %v", m, m2)
- t.FailNow()
- }
- // log("m: %v, m2: %v, p: %v, p2: %v", m, m2, p, p2)
- checkEqualT(t, p, p2, "p=p2")
- checkEqualT(t, m, m2, "m=m2")
- if err = deepEqual(p, p2); err == nil {
- logT(t, "p and p2 match")
- } else {
- logT(t, "Not Equal: %v. p: %v, p2: %v", err, p, p2)
- t.FailNow()
- }
- if err = deepEqual(m, m2); err == nil {
- logT(t, "m and m2 match")
- } else {
- logT(t, "Not Equal: %v. m: %v, m2: %v", err, m, m2)
- t.FailNow()
- }
- // func TestMsgpackDecodeStructSubset(t *testing.T) {
- // test that we can decode a subset of the stream
- mm := map[string]interface{}{"A": 5, "B": 99, "C": 333}
- bs, err = testMarshalErr(mm, h, t, "mm")
- type ttt struct {
- A uint8
- C int32
- }
- var t2 ttt
- testUnmarshalErr(&t2, bs, h, t, "t2")
- t3 := ttt{5, 333}
- checkEqualT(t, t2, t3, "t2=t3")
- // println(">>>>>")
- // test simple arrays, non-addressable arrays, slices
- type tarr struct {
- A int64
- B [3]int64
- C []byte
- D [3]byte
- }
- var tarr0 = tarr{1, [3]int64{2, 3, 4}, []byte{4, 5, 6}, [3]byte{7, 8, 9}}
- // test both pointer and non-pointer (value)
- for _, tarr1 := range []interface{}{tarr0, &tarr0} {
- bs, err = testMarshalErr(tarr1, h, t, "tarr1")
- if err != nil {
- logT(t, "Error marshalling: %v", err)
- t.FailNow()
- }
- if _, ok := h.(*JsonHandle); ok {
- logT(t, "Marshal as: %s", bs)
- }
- var tarr2 tarr
- testUnmarshalErr(&tarr2, bs, h, t, "tarr2")
- checkEqualT(t, tarr0, tarr2, "tarr0=tarr2")
- // fmt.Printf(">>>> err: %v. tarr1: %v, tarr2: %v\n", err, tarr0, tarr2)
- }
- // test byte array, even if empty (msgpack only)
- if h == testMsgpackH {
- type ystruct struct {
- Anarray []byte
- }
- var ya = ystruct{}
- testUnmarshalErr(&ya, []byte{0x91, 0x90}, h, t, "ya")
- }
- }
- func testCodecEmbeddedPointer(t *testing.T, h Handle) {
- testOnce.Do(testInitAll)
- type Z int
- type A struct {
- AnInt int
- }
- type B struct {
- *Z
- *A
- MoreInt int
- }
- var z Z = 4
- x1 := &B{&z, &A{5}, 6}
- bs, err := testMarshalErr(x1, h, t, "x1")
- // fmt.Printf("buf: len(%v): %x\n", buf.Len(), buf.Bytes())
- var x2 = new(B)
- err = testUnmarshalErr(x2, bs, h, t, "x2")
- err = checkEqualT(t, x1, x2, "x1=x2")
- _ = err
- }
- func testCodecUnderlyingType(t *testing.T, h Handle) {
- testOnce.Do(testInitAll)
- // Manual Test.
- // Run by hand, with accompanying print statements in fast-path.go
- // to ensure that the fast functions are called.
- type T1 map[string]string
- v := T1{"1": "1s", "2": "2s"}
- var bs []byte
- var err error
- NewEncoderBytes(&bs, h).MustEncode(v)
- if err != nil {
- logT(t, "Error during encode: %v", err)
- failT(t)
- }
- var v2 T1
- NewDecoderBytes(bs, h).MustDecode(&v2)
- if err != nil {
- logT(t, "Error during decode: %v", err)
- failT(t)
- }
- }
- func testCodecChan(t *testing.T, h Handle) {
- // - send a slice []*int64 (sl1) into an chan (ch1) with cap > len(s1)
- // - encode ch1 as a stream array
- // - decode a chan (ch2), with cap > len(s1) from the stream array
- // - receive from ch2 into slice sl2
- // - compare sl1 and sl2
- // - do this for codecs: json, cbor (covers all types)
- sl1 := make([]*int64, 4)
- for i := range sl1 {
- var j int64 = int64(i)
- sl1[i] = &j
- }
- ch1 := make(chan *int64, 4)
- for _, j := range sl1 {
- ch1 <- j
- }
- var bs []byte
- NewEncoderBytes(&bs, h).MustEncode(ch1)
- // if !h.isBinary() {
- // fmt.Printf("before: len(ch1): %v, bs: %s\n", len(ch1), bs)
- // }
- // var ch2 chan *int64 // this will block if json, etc.
- ch2 := make(chan *int64, 8)
- NewDecoderBytes(bs, h).MustDecode(&ch2)
- // logT(t, "Len(ch2): %v", len(ch2))
- // fmt.Printf("after: len(ch2): %v, ch2: %v\n", len(ch2), ch2)
- close(ch2)
- var sl2 []*int64
- for j := range ch2 {
- sl2 = append(sl2, j)
- }
- if err := deepEqual(sl1, sl2); err != nil {
- logT(t, "Not Match: %v; len: %v, %v", err, len(sl1), len(sl2))
- failT(t)
- }
- }
- func testCodecRpcOne(t *testing.T, rr Rpc, h Handle, doRequest bool, exitSleepMs time.Duration,
- ) (port int) {
- testOnce.Do(testInitAll)
- if testSkipRPCTests {
- return
- }
- // rpc needs EOF, which is sent via a panic, and so must be recovered.
- if !recoverPanicToErr {
- logT(t, "EXPECTED. set recoverPanicToErr=true, since rpc needs EOF")
- t.FailNow()
- }
- srv := rpc.NewServer()
- srv.Register(testRpcInt)
- ln, err := net.Listen("tcp", "127.0.0.1:0")
- // log("listener: %v", ln.Addr())
- checkErrT(t, err)
- port = (ln.Addr().(*net.TCPAddr)).Port
- // var opts *DecoderOptions
- // opts := testDecOpts
- // opts.MapType = mapStrIntfTyp
- // opts.RawToString = false
- serverExitChan := make(chan bool, 1)
- var serverExitFlag uint64 = 0
- serverFn := func() {
- for {
- conn1, err1 := ln.Accept()
- // if err1 != nil {
- // //fmt.Printf("accept err1: %v\n", err1)
- // continue
- // }
- if atomic.LoadUint64(&serverExitFlag) == 1 {
- serverExitChan <- true
- conn1.Close()
- return // exit serverFn goroutine
- }
- if err1 == nil {
- var sc rpc.ServerCodec = rr.ServerCodec(conn1, h)
- srv.ServeCodec(sc)
- }
- }
- }
- clientFn := func(cc rpc.ClientCodec) {
- cl := rpc.NewClientWithCodec(cc)
- defer cl.Close()
- // defer func() { println("##### client closing"); cl.Close() }()
- var up, sq, mult int
- var rstr string
- // log("Calling client")
- checkErrT(t, cl.Call("TestRpcInt.Update", 5, &up))
- // log("Called TestRpcInt.Update")
- checkEqualT(t, testRpcInt.i, 5, "testRpcInt.i=5")
- checkEqualT(t, up, 5, "up=5")
- checkErrT(t, cl.Call("TestRpcInt.Square", 1, &sq))
- checkEqualT(t, sq, 25, "sq=25")
- checkErrT(t, cl.Call("TestRpcInt.Mult", 20, &mult))
- checkEqualT(t, mult, 100, "mult=100")
- checkErrT(t, cl.Call("TestRpcInt.EchoStruct", TestRpcABC{"Aa", "Bb", "Cc"}, &rstr))
- checkEqualT(t, rstr, fmt.Sprintf("%#v", TestRpcABC{"Aa", "Bb", "Cc"}), "rstr=")
- checkErrT(t, cl.Call("TestRpcInt.Echo123", []string{"A1", "B2", "C3"}, &rstr))
- checkEqualT(t, rstr, fmt.Sprintf("%#v", []string{"A1", "B2", "C3"}), "rstr=")
- }
- connFn := func() (bs net.Conn) {
- // log("calling f1")
- bs, err2 := net.Dial(ln.Addr().Network(), ln.Addr().String())
- //fmt.Printf("f1. bs: %v, err2: %v\n", bs, err2)
- checkErrT(t, err2)
- return
- }
- exitFn := func() {
- atomic.StoreUint64(&serverExitFlag, 1)
- bs := connFn()
- <-serverExitChan
- bs.Close()
- // serverExitChan <- true
- }
- go serverFn()
- runtime.Gosched()
- //time.Sleep(100 * time.Millisecond)
- if exitSleepMs == 0 {
- defer ln.Close()
- defer exitFn()
- }
- if doRequest {
- bs := connFn()
- cc := rr.ClientCodec(bs, h)
- clientFn(cc)
- }
- if exitSleepMs != 0 {
- go func() {
- defer ln.Close()
- time.Sleep(exitSleepMs)
- exitFn()
- }()
- }
- return
- }
- func doTestMapEncodeForCanonical(t *testing.T, name string, h Handle) {
- v1 := map[string]interface{}{
- "a": 1,
- "b": "hello",
- "c": map[string]interface{}{
- "c/a": 1,
- "c/b": "world",
- "c/c": []int{1, 2, 3, 4},
- "c/d": map[string]interface{}{
- "c/d/a": "fdisajfoidsajfopdjsaopfjdsapofda",
- "c/d/b": "fdsafjdposakfodpsakfopdsakfpodsakfpodksaopfkdsopafkdopsa",
- "c/d/c": "poir02 ir30qif4p03qir0pogjfpoaerfgjp ofke[padfk[ewapf kdp[afep[aw",
- "c/d/d": "fdsopafkd[sa f-32qor-=4qeof -afo-erfo r-eafo 4e- o r4-qwo ag",
- "c/d/e": "kfep[a sfkr0[paf[a foe-[wq ewpfao-q ro3-q ro-4qof4-qor 3-e orfkropzjbvoisdb",
- "c/d/f": "",
- },
- "c/e": map[int]string{
- 1: "1",
- 22: "22",
- 333: "333",
- 4444: "4444",
- 55555: "55555",
- },
- "c/f": map[string]int{
- "1": 1,
- "22": 22,
- "333": 333,
- "4444": 4444,
- "55555": 55555,
- },
- },
- }
- var v2 map[string]interface{}
- var b1, b2 []byte
- // encode v1 into b1, decode b1 into v2, encode v2 into b2, compare b1 and b2
- bh := h.getBasicHandle()
- if !bh.Canonical {
- bh.Canonical = true
- defer func() { bh.Canonical = false }()
- }
- e1 := NewEncoderBytes(&b1, h)
- e1.MustEncode(v1)
- d1 := NewDecoderBytes(b1, h)
- d1.MustDecode(&v2)
- e2 := NewEncoderBytes(&b2, h)
- e2.MustEncode(v2)
- if !bytes.Equal(b1, b2) {
- logT(t, "Unequal bytes: %v VS %v", b1, b2)
- t.FailNow()
- }
- }
- func doTestStdEncIntf(t *testing.T, name string, h Handle) {
- args := [][2]interface{}{
- {&TestABC{"A", "BB", "CCC"}, new(TestABC)},
- {&TestABC2{"AAA", "BB", "C"}, new(TestABC2)},
- }
- for _, a := range args {
- var b []byte
- e := NewEncoderBytes(&b, h)
- e.MustEncode(a[0])
- d := NewDecoderBytes(b, h)
- d.MustDecode(a[1])
- if err := deepEqual(a[0], a[1]); err == nil {
- logT(t, "++++ Objects match")
- } else {
- logT(t, "---- Objects do not match: y1: %v, err: %v", a[1], err)
- failT(t)
- }
- }
- }
- func doTestEncCircularRef(t *testing.T, name string, h Handle) {
- type T1 struct {
- S string
- B bool
- T interface{}
- }
- type T2 struct {
- S string
- T *T1
- }
- type T3 struct {
- S string
- T *T2
- }
- t1 := T1{"t1", true, nil}
- t2 := T2{"t2", &t1}
- t3 := T3{"t3", &t2}
- t1.T = &t3
- var bs []byte
- var err error
- bh := h.getBasicHandle()
- if !bh.CheckCircularRef {
- bh.CheckCircularRef = true
- defer func() { bh.CheckCircularRef = false }()
- }
- err = NewEncoderBytes(&bs, h).Encode(&t3)
- if err == nil {
- logT(t, "expecting error due to circular reference. found none")
- t.FailNow()
- }
- if x := err.Error(); strings.Contains(x, "circular") || strings.Contains(x, "cyclic") {
- logT(t, "error detected as expected: %v", x)
- } else {
- logT(t, "error detected was not as expected: %v", x)
- t.FailNow()
- }
- }
- // TestAnonCycleT{1,2,3} types are used to test anonymous cycles.
- // They are top-level, so that they can have circular references.
- type (
- TestAnonCycleT1 struct {
- S string
- TestAnonCycleT2
- }
- TestAnonCycleT2 struct {
- S2 string
- TestAnonCycleT3
- }
- TestAnonCycleT3 struct {
- *TestAnonCycleT1
- }
- )
- func doTestAnonCycle(t *testing.T, name string, h Handle) {
- var x TestAnonCycleT1
- x.S = "hello"
- x.TestAnonCycleT2.S2 = "hello.2"
- x.TestAnonCycleT2.TestAnonCycleT3.TestAnonCycleT1 = &x
- // just check that you can get typeInfo for T1
- rt := reflect.TypeOf((*TestAnonCycleT1)(nil)).Elem()
- rtid := reflect.ValueOf(rt).Pointer()
- pti := h.getBasicHandle().getTypeInfo(rtid, rt)
- logT(t, "pti: %v", pti)
- }
- func doTestJsonLargeInteger(t *testing.T, v interface{}, ias uint8) {
- logT(t, "Running doTestJsonLargeInteger: v: %#v, ias: %c", v, ias)
- oldIAS := testJsonH.IntegerAsString
- defer func() { testJsonH.IntegerAsString = oldIAS }()
- testJsonH.IntegerAsString = ias
- var vu uint
- var vi int
- var vb bool
- var b []byte
- e := NewEncoderBytes(&b, testJsonH)
- e.MustEncode(v)
- e.MustEncode(true)
- d := NewDecoderBytes(b, testJsonH)
- // below, we validate that the json string or number was encoded,
- // then decode, and validate that the correct value was decoded.
- fnStrChk := func() {
- // check that output started with ", and ended with "true
- if !(b[0] == '"' && string(b[len(b)-5:]) == `"true`) {
- logT(t, "Expecting a JSON string, got: %s", b)
- failT(t)
- }
- }
- switch ias {
- case 'L':
- switch v2 := v.(type) {
- case int:
- v2n := int64(v2) // done to work with 32-bit OS
- if v2n > 1<<53 || (v2n < 0 && -v2n > 1<<53) {
- fnStrChk()
- }
- case uint:
- v2n := uint64(v2) // done to work with 32-bit OS
- if v2n > 1<<53 {
- fnStrChk()
- }
- }
- case 'A':
- fnStrChk()
- default:
- // check that output doesn't contain " at all
- for _, i := range b {
- if i == '"' {
- logT(t, "Expecting a JSON Number without quotation: got: %s", b)
- failT(t)
- }
- }
- }
- switch v2 := v.(type) {
- case int:
- d.MustDecode(&vi)
- d.MustDecode(&vb)
- // check that vb = true, and vi == v2
- if !(vb && vi == v2) {
- logT(t, "Expecting equal values from %s: got golden: %v, decoded: %v", b, v2, vi)
- failT(t)
- }
- case uint:
- d.MustDecode(&vu)
- d.MustDecode(&vb)
- // check that vb = true, and vi == v2
- if !(vb && vu == v2) {
- logT(t, "Expecting equal values from %s: got golden: %v, decoded: %v", b, v2, vu)
- failT(t)
- }
- // fmt.Printf("%v: %s, decode: %d, bool: %v, equal_on_decode: %v\n", v, b, vu, vb, vu == v.(uint))
- }
- }
- func doTestRawValue(t *testing.T, name string, h Handle) {
- bh := h.getBasicHandle()
- if !bh.Raw {
- bh.Raw = true
- defer func() { bh.Raw = false }()
- }
- var i, i2 int
- var v, v2 TestRawValue
- var bs, bs2 []byte
- i = 1234 //1234567890
- v = TestRawValue{I: i}
- e := NewEncoderBytes(&bs, h)
- e.MustEncode(v.I)
- logT(t, ">>> raw: %v\n", bs)
- v.R = Raw(bs)
- e.ResetBytes(&bs2)
- e.MustEncode(v)
- logT(t, ">>> bs2: %v\n", bs2)
- d := NewDecoderBytes(bs2, h)
- d.MustDecode(&v2)
- d.ResetBytes(v2.R)
- logT(t, ">>> v2.R: %v\n", ([]byte)(v2.R))
- d.MustDecode(&i2)
- logT(t, ">>> Encoded %v, decoded %v\n", i, i2)
- // logT(t, "Encoded %v, decoded %v", i, i2)
- if i != i2 {
- logT(t, "Error: encoded %v, decoded %v", i, i2)
- t.FailNow()
- }
- }
- // Comprehensive testing that generates data encoded from python handle (cbor, msgpack),
- // and validates that our code can read and write it out accordingly.
- // We keep this unexported here, and put actual test in ext_dep_test.go.
- // This way, it can be excluded by excluding file completely.
- func doTestPythonGenStreams(t *testing.T, name string, h Handle) {
- logT(t, "TestPythonGenStreams-%v", name)
- tmpdir, err := ioutil.TempDir("", "golang-"+name+"-test")
- if err != nil {
- logT(t, "-------- Unable to create temp directory\n")
- t.FailNow()
- }
- defer os.RemoveAll(tmpdir)
- logT(t, "tmpdir: %v", tmpdir)
- cmd := exec.Command("python", "test.py", "testdata", tmpdir)
- //cmd.Stdin = strings.NewReader("some input")
- //cmd.Stdout = &out
- var cmdout []byte
- if cmdout, err = cmd.CombinedOutput(); err != nil {
- logT(t, "-------- Error running test.py testdata. Err: %v", err)
- logT(t, " %v", string(cmdout))
- t.FailNow()
- }
- bh := h.getBasicHandle()
- oldMapType := bh.MapType
- for i, v := range tablePythonVerify {
- // if v == uint64(0) && h == testMsgpackH {
- // v = int64(0)
- // }
- bh.MapType = oldMapType
- //load up the golden file based on number
- //decode it
- //compare to in-mem object
- //encode it again
- //compare to output stream
- logT(t, "..............................................")
- logT(t, " Testing: #%d: %T, %#v\n", i, v, v)
- var bss []byte
- bss, err = ioutil.ReadFile(filepath.Join(tmpdir, strconv.Itoa(i)+"."+name+".golden"))
- if err != nil {
- logT(t, "-------- Error reading golden file: %d. Err: %v", i, err)
- failT(t)
- continue
- }
- bh.MapType = testMapStrIntfTyp
- var v1 interface{}
- if err = testUnmarshal(&v1, bss, h); err != nil {
- logT(t, "-------- Error decoding stream: %d: Err: %v", i, err)
- failT(t)
- continue
- }
- if v == skipVerifyVal {
- continue
- }
- //no need to indirect, because we pass a nil ptr, so we already have the value
- //if v1 != nil { v1 = reflect.Indirect(reflect.ValueOf(v1)).Interface() }
- if err = deepEqual(v, v1); err == nil {
- logT(t, "++++++++ Objects match: %T, %v", v, v)
- } else {
- logT(t, "-------- Objects do not match: %v. Source: %T. Decoded: %T", err, v, v1)
- logT(t, "-------- GOLDEN: %#v", v)
- // logT(t, "-------- DECODED: %#v <====> %#v", v1, reflect.Indirect(reflect.ValueOf(v1)).Interface())
- logT(t, "-------- DECODED: %#v <====> %#v", v1, reflect.Indirect(reflect.ValueOf(v1)).Interface())
- failT(t)
- }
- bsb, err := testMarshal(v1, h)
- if err != nil {
- logT(t, "Error encoding to stream: %d: Err: %v", i, err)
- failT(t)
- continue
- }
- if err = deepEqual(bsb, bss); err == nil {
- logT(t, "++++++++ Bytes match")
- } else {
- logT(t, "???????? Bytes do not match. %v.", err)
- xs := "--------"
- if reflect.ValueOf(v).Kind() == reflect.Map {
- xs = " "
- logT(t, "%s It's a map. Ok that they don't match (dependent on ordering).", xs)
- } else {
- logT(t, "%s It's not a map. They should match.", xs)
- failT(t)
- }
- logT(t, "%s FROM_FILE: %4d] %v", xs, len(bss), bss)
- logT(t, "%s ENCODED: %4d] %v", xs, len(bsb), bsb)
- }
- }
- bh.MapType = oldMapType
- }
- // To test MsgpackSpecRpc, we test 3 scenarios:
- // - Go Client to Go RPC Service (contained within TestMsgpackRpcSpec)
- // - Go client to Python RPC Service (contained within doTestMsgpackRpcSpecGoClientToPythonSvc)
- // - Python Client to Go RPC Service (contained within doTestMsgpackRpcSpecPythonClientToGoSvc)
- //
- // This allows us test the different calling conventions
- // - Go Service requires only one argument
- // - Python Service allows multiple arguments
- func doTestMsgpackRpcSpecGoClientToPythonSvc(t *testing.T) {
- if testSkipRPCTests {
- return
- }
- // openPorts are between 6700 and 6800
- r := rand.New(rand.NewSource(time.Now().UnixNano()))
- openPort := strconv.FormatInt(6700+r.Int63n(99), 10)
- // openPort := "6792"
- cmd := exec.Command("python", "test.py", "rpc-server", openPort, "4")
- checkErrT(t, cmd.Start())
- bs, err2 := net.Dial("tcp", ":"+openPort)
- for i := 0; i < 10 && err2 != nil; i++ {
- time.Sleep(50 * time.Millisecond) // time for python rpc server to start
- bs, err2 = net.Dial("tcp", ":"+openPort)
- }
- checkErrT(t, err2)
- cc := MsgpackSpecRpc.ClientCodec(bs, testMsgpackH)
- cl := rpc.NewClientWithCodec(cc)
- defer cl.Close()
- var rstr string
- checkErrT(t, cl.Call("EchoStruct", TestRpcABC{"Aa", "Bb", "Cc"}, &rstr))
- //checkEqualT(t, rstr, "{'A': 'Aa', 'B': 'Bb', 'C': 'Cc'}")
- var mArgs MsgpackSpecRpcMultiArgs = []interface{}{"A1", "B2", "C3"}
- checkErrT(t, cl.Call("Echo123", mArgs, &rstr))
- checkEqualT(t, rstr, "1:A1 2:B2 3:C3", "rstr=")
- cmd.Process.Kill()
- }
- func doTestMsgpackRpcSpecPythonClientToGoSvc(t *testing.T) {
- if testSkipRPCTests {
- return
- }
- port := testCodecRpcOne(t, MsgpackSpecRpc, testMsgpackH, false, 1*time.Second)
- //time.Sleep(1000 * time.Millisecond)
- cmd := exec.Command("python", "test.py", "rpc-client-go-service", strconv.Itoa(port))
- var cmdout []byte
- var err error
- if cmdout, err = cmd.CombinedOutput(); err != nil {
- logT(t, "-------- Error running test.py rpc-client-go-service. Err: %v", err)
- logT(t, " %v", string(cmdout))
- t.FailNow()
- }
- checkEqualT(t, string(cmdout),
- fmt.Sprintf("%#v\n%#v\n", []string{"A1", "B2", "C3"}, TestRpcABC{"Aa", "Bb", "Cc"}), "cmdout=")
- }
- func TestBincCodecsTable(t *testing.T) {
- testCodecTableOne(t, testBincH)
- }
- func TestBincCodecsMisc(t *testing.T) {
- testCodecMiscOne(t, testBincH)
- }
- func TestBincCodecsEmbeddedPointer(t *testing.T) {
- testCodecEmbeddedPointer(t, testBincH)
- }
- func TestBincStdEncIntf(t *testing.T) {
- doTestStdEncIntf(t, "binc", testBincH)
- }
- func TestSimpleCodecsTable(t *testing.T) {
- testCodecTableOne(t, testSimpleH)
- }
- func TestSimpleCodecsMisc(t *testing.T) {
- testCodecMiscOne(t, testSimpleH)
- }
- func TestSimpleCodecsEmbeddedPointer(t *testing.T) {
- testCodecEmbeddedPointer(t, testSimpleH)
- }
- func TestSimpleStdEncIntf(t *testing.T) {
- doTestStdEncIntf(t, "simple", testSimpleH)
- }
- func TestMsgpackCodecsTable(t *testing.T) {
- testCodecTableOne(t, testMsgpackH)
- }
- func TestMsgpackCodecsMisc(t *testing.T) {
- testCodecMiscOne(t, testMsgpackH)
- }
- func TestMsgpackCodecsEmbeddedPointer(t *testing.T) {
- testCodecEmbeddedPointer(t, testMsgpackH)
- }
- func TestMsgpackStdEncIntf(t *testing.T) {
- doTestStdEncIntf(t, "msgpack", testMsgpackH)
- }
- func TestCborCodecsTable(t *testing.T) {
- testCodecTableOne(t, testCborH)
- }
- func TestCborCodecsMisc(t *testing.T) {
- testCodecMiscOne(t, testCborH)
- }
- func TestCborCodecsEmbeddedPointer(t *testing.T) {
- testCodecEmbeddedPointer(t, testCborH)
- }
- func TestCborMapEncodeForCanonical(t *testing.T) {
- doTestMapEncodeForCanonical(t, "cbor", testCborH)
- }
- func TestCborCodecChan(t *testing.T) {
- testCodecChan(t, testCborH)
- }
- func TestCborStdEncIntf(t *testing.T) {
- doTestStdEncIntf(t, "cbor", testCborH)
- }
- func TestJsonCodecsTable(t *testing.T) {
- testCodecTableOne(t, testJsonH)
- }
- func TestJsonCodecsMisc(t *testing.T) {
- testCodecMiscOne(t, testJsonH)
- }
- func TestJsonCodecsEmbeddedPointer(t *testing.T) {
- testCodecEmbeddedPointer(t, testJsonH)
- }
- func TestJsonCodecChan(t *testing.T) {
- testCodecChan(t, testJsonH)
- }
- func TestJsonStdEncIntf(t *testing.T) {
- doTestStdEncIntf(t, "json", testJsonH)
- }
- // ----- Raw ---------
- func TestJsonRaw(t *testing.T) {
- doTestRawValue(t, "json", testJsonH)
- }
- func TestBincRaw(t *testing.T) {
- doTestRawValue(t, "binc", testBincH)
- }
- func TestMsgpackRaw(t *testing.T) {
- doTestRawValue(t, "msgpack", testMsgpackH)
- }
- func TestSimpleRaw(t *testing.T) {
- doTestRawValue(t, "simple", testSimpleH)
- }
- func TestCborRaw(t *testing.T) {
- doTestRawValue(t, "cbor", testCborH)
- }
- // ----- ALL (framework based) -----
- func TestAllEncCircularRef(t *testing.T) {
- doTestEncCircularRef(t, "cbor", testCborH)
- }
- func TestAllAnonCycle(t *testing.T) {
- doTestAnonCycle(t, "cbor", testCborH)
- }
- // ----- RPC -----
- func TestBincRpcGo(t *testing.T) {
- testCodecRpcOne(t, GoRpc, testBincH, true, 0)
- }
- func TestSimpleRpcGo(t *testing.T) {
- testCodecRpcOne(t, GoRpc, testSimpleH, true, 0)
- }
- func TestMsgpackRpcGo(t *testing.T) {
- testCodecRpcOne(t, GoRpc, testMsgpackH, true, 0)
- }
- func TestCborRpcGo(t *testing.T) {
- testCodecRpcOne(t, GoRpc, testCborH, true, 0)
- }
- func TestJsonRpcGo(t *testing.T) {
- testCodecRpcOne(t, GoRpc, testJsonH, true, 0)
- }
- func TestMsgpackRpcSpec(t *testing.T) {
- testCodecRpcOne(t, MsgpackSpecRpc, testMsgpackH, true, 0)
- }
- func TestBincUnderlyingType(t *testing.T) {
- testCodecUnderlyingType(t, testBincH)
- }
- func TestJsonLargeInteger(t *testing.T) {
- for _, i := range []uint8{'L', 'A', 0} {
- for _, j := range []interface{}{
- int64(1 << 60),
- -int64(1 << 60),
- 0,
- 1 << 20,
- -(1 << 20),
- uint64(1 << 60),
- uint(0),
- uint(1 << 20),
- } {
- doTestJsonLargeInteger(t, j, i)
- }
- }
- }
- // TODO:
- // Add Tests for:
- // - decoding empty list/map in stream into a nil slice/map
- // - binary(M|Unm)arsher support for time.Time (e.g. cbor encoding)
- // - text(M|Unm)arshaler support for time.Time (e.g. json encoding)
- // - non fast-path scenarios e.g. map[string]uint16, []customStruct.
- // Expand cbor to include indefinite length stuff for this non-fast-path types.
- // This may not be necessary, since we have the manual tests (fastpathEnabled=false) to test/validate with.
- // - CodecSelfer
- // Ensure it is called when (en|de)coding interface{} or reflect.Value (2 different codepaths).
- // - interfaces: textMarshaler, binaryMarshaler, codecSelfer
- // - struct tags:
- // on anonymous fields, _struct (all fields), etc
- // - codecgen of struct containing channels.
- // - bad input with large array length prefix
- //
- // Cleanup tests:
- // - The are brittle in their handling of validation and skipping
|