|  | /* | 
|  | * Licensed to the Apache Software Foundation (ASF) under one | 
|  | * or more contributor license agreements. See the NOTICE file | 
|  | * distributed with this work for additional information | 
|  | * regarding copyright ownership. The ASF licenses this file | 
|  | * to you under the Apache License, Version 2.0 (the | 
|  | * "License"); you may not use this file except in compliance | 
|  | * with the License. You may obtain a copy of the License at | 
|  | * | 
|  | *   http://www.apache.org/licenses/LICENSE-2.0 | 
|  | * | 
|  | * Unless required by applicable law or agreed to in writing, | 
|  | * software distributed under the License is distributed on an | 
|  | * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | 
|  | * KIND, either express or implied. See the License for the | 
|  | * specific language governing permissions and limitations | 
|  | * under the License. | 
|  | */ | 
|  |  | 
|  | package main | 
|  |  | 
|  | import ( | 
|  | "bytes" | 
|  | "context" | 
|  | "flag" | 
|  | "fmt" | 
|  | t "log" | 
|  | "reflect" | 
|  |  | 
|  | "github.com/apache/thrift/lib/go/thrift" | 
|  | "github.com/apache/thrift/test/go/src/common" | 
|  | "github.com/apache/thrift/test/go/src/gen/thrifttest" | 
|  | ) | 
|  |  | 
|  | var host = flag.String("host", "localhost", "Host to connect") | 
|  | var port = flag.Int64("port", 9090, "Port number to connect") | 
|  | var domain_socket = flag.String("domain-socket", "", "Domain Socket (e.g. /tmp/thrifttest.thrift), instead of host and port") | 
|  | var transport = flag.String("transport", "buffered", "Transport: buffered, framed, http, zlib") | 
|  | var _ = flag.Bool("zlib", false, "For compatibility. Ignored.") | 
|  | var protocol = flag.String("protocol", "binary", "Protocol: binary, compact, json") | 
|  | var ssl = flag.Bool("ssl", false, "Encrypted Transport using SSL") | 
|  | var testloops = flag.Int("testloops", 1, "Number of Tests") | 
|  |  | 
|  | func main() { | 
|  | flag.Parse() | 
|  | addr := *domain_socket | 
|  | if addr == "" { | 
|  | addr = fmt.Sprintf("%s:%d", *host, *port) | 
|  | } | 
|  | client, _, err := common.StartClient(addr, *transport, *protocol, *ssl) | 
|  | if err != nil { | 
|  | t.Fatalf("Unable to start client: ", err) | 
|  | } | 
|  | for range *testloops { | 
|  | callEverything(client) | 
|  | } | 
|  | } | 
|  |  | 
|  | var rmapmap = map[int32]map[int32]int32{ | 
|  | -4: {-4: -4, -3: -3, -2: -2, -1: -1}, | 
|  | 4:  {4: 4, 3: 3, 2: 2, 1: 1}, | 
|  | } | 
|  |  | 
|  | var xxs = &thrifttest.Xtruct{ | 
|  | StringThing: "Hello2", | 
|  | ByteThing:   42, | 
|  | I32Thing:    4242, | 
|  | I64Thing:    424242, | 
|  | } | 
|  |  | 
|  | var xcept = &thrifttest.Xception{ErrorCode: 1001, Message: "Xception"} | 
|  | var defaultCtx = context.Background() | 
|  |  | 
|  | func callEverything(client *thrifttest.ThriftTestClient) { | 
|  | var err error | 
|  | if err = client.TestVoid(defaultCtx); err != nil { | 
|  | t.Fatalf("Unexpected error in TestVoid() call: ", err) | 
|  | } | 
|  |  | 
|  | thing, err := client.TestString(defaultCtx, "thing") | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestString() call: ", err) | 
|  | } | 
|  | if thing != "thing" { | 
|  | t.Fatalf("Unexpected TestString() result, expected 'thing' got '%s' ", thing) | 
|  | } | 
|  |  | 
|  | bl, err := client.TestBool(defaultCtx, true) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestBool() call: ", err) | 
|  | } | 
|  | if !bl { | 
|  | t.Fatalf("Unexpected TestBool() result expected true, got %f ", bl) | 
|  | } | 
|  | bl, err = client.TestBool(defaultCtx, false) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestBool() call: ", err) | 
|  | } | 
|  | if bl { | 
|  | t.Fatalf("Unexpected TestBool() result expected false, got %f ", bl) | 
|  | } | 
|  |  | 
|  | b, err := client.TestByte(defaultCtx, 42) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestByte() call: ", err) | 
|  | } | 
|  | if b != 42 { | 
|  | t.Fatalf("Unexpected TestByte() result expected 42, got %d ", b) | 
|  | } | 
|  |  | 
|  | i32, err := client.TestI32(defaultCtx, 4242) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestI32() call: ", err) | 
|  | } | 
|  | if i32 != 4242 { | 
|  | t.Fatalf("Unexpected TestI32() result expected 4242, got %d ", i32) | 
|  | } | 
|  |  | 
|  | i64, err := client.TestI64(defaultCtx, 424242) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestI64() call: ", err) | 
|  | } | 
|  | if i64 != 424242 { | 
|  | t.Fatalf("Unexpected TestI64() result expected 424242, got %d ", i64) | 
|  | } | 
|  |  | 
|  | d, err := client.TestDouble(defaultCtx, 42.42) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestDouble() call: ", err) | 
|  | } | 
|  | if d != 42.42 { | 
|  | t.Fatalf("Unexpected TestDouble() result expected 42.42, got %f ", d) | 
|  | } | 
|  |  | 
|  | binout := make([]byte, 256) | 
|  | for i := range binout { | 
|  | binout[i] = byte(i) | 
|  | } | 
|  | bin, err := client.TestBinary(defaultCtx, binout) | 
|  | if err != nil { | 
|  | t.Fatalf("TestBinary failed with %v", err) | 
|  | } | 
|  | if !bytes.Equal(binout, bin) { | 
|  | t.Fatalf("Unexpected TestBinary() result expected % 02x, got % 02x ", binout, bin) | 
|  | } | 
|  |  | 
|  | uout := thrift.Tuuid{ | 
|  | 0x00, 0x11, 0x22, 0x33, | 
|  | 0x44, 0x55, | 
|  | 0x66, 0x77, | 
|  | 0x88, 0x99, | 
|  | 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff, | 
|  | } | 
|  | u, err := client.TestUuid(defaultCtx, uout) | 
|  | if err != nil { | 
|  | t.Fatalf("TestUuid failed with %v", err) | 
|  | } | 
|  | if u != uout { | 
|  | t.Fatalf("Unexpected TestUuid() result expected %v, got %v", uout, u) | 
|  | } | 
|  |  | 
|  | xs := thrifttest.NewXtruct() | 
|  | xs.StringThing = "thing" | 
|  | xs.ByteThing = 42 | 
|  | xs.I32Thing = 4242 | 
|  | xs.I64Thing = 424242 | 
|  | xsret, err := client.TestStruct(defaultCtx, xs) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestStruct() call: ", err) | 
|  | } | 
|  | if *xs != *xsret { | 
|  | t.Fatalf("Unexpected TestStruct() result expected %#v, got %#v ", xs, xsret) | 
|  | } | 
|  |  | 
|  | x2 := thrifttest.NewXtruct2() | 
|  | x2.StructThing = xs | 
|  | x2ret, err := client.TestNest(defaultCtx, x2) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestNest() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(x2, x2ret) { | 
|  | t.Fatalf("Unexpected TestNest() result expected %#v, got %#v ", x2, x2ret) | 
|  | } | 
|  |  | 
|  | m := map[int32]int32{1: 2, 3: 4, 5: 42} | 
|  | mret, err := client.TestMap(defaultCtx, m) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestMap() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(m, mret) { | 
|  | t.Fatalf("Unexpected TestMap() result expected %#v, got %#v ", m, mret) | 
|  | } | 
|  |  | 
|  | sm := map[string]string{"a": "2", "b": "blah", "some": "thing"} | 
|  | smret, err := client.TestStringMap(defaultCtx, sm) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestStringMap() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(sm, smret) { | 
|  | t.Fatalf("Unexpected TestStringMap() result expected %#v, got %#v ", sm, smret) | 
|  | } | 
|  |  | 
|  | s := []int32{1, 2, 42} | 
|  | sret, err := client.TestSet(defaultCtx, s) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestSet() call: ", err) | 
|  | } | 
|  | // Sets can be in any order, but Go slices are ordered, so reflect.DeepEqual won't work. | 
|  | stemp := map[int32]struct{}{} | 
|  | for _, val := range s { | 
|  | stemp[val] = struct{}{} | 
|  | } | 
|  | for _, val := range sret { | 
|  | if _, ok := stemp[val]; !ok { | 
|  | t.Fatalf("Unexpected TestSet() result expected %#v, got %#v ", s, sret) | 
|  | } | 
|  | } | 
|  |  | 
|  | l := []int32{1, 2, 42} | 
|  | lret, err := client.TestList(defaultCtx, l) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestList() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(l, lret) { | 
|  | t.Fatalf("Unexpected TestList() result expected %#v, got %#v ", l, lret) | 
|  | } | 
|  |  | 
|  | eret, err := client.TestEnum(defaultCtx, thrifttest.Numberz_TWO) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestEnum() call: ", err) | 
|  | } | 
|  | if eret != thrifttest.Numberz_TWO { | 
|  | t.Fatalf("Unexpected TestEnum() result expected %#v, got %#v ", thrifttest.Numberz_TWO, eret) | 
|  | } | 
|  |  | 
|  | tret, err := client.TestTypedef(defaultCtx, thrifttest.UserId(42)) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestTypedef() call: ", err) | 
|  | } | 
|  | if tret != thrifttest.UserId(42) { | 
|  | t.Fatalf("Unexpected TestTypedef() result expected %#v, got %#v ", thrifttest.UserId(42), tret) | 
|  | } | 
|  |  | 
|  | mapmap, err := client.TestMapMap(defaultCtx, 42) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestMapMap() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(mapmap, rmapmap) { | 
|  | t.Fatalf("Unexpected TestMapMap() result expected %#v, got %#v ", rmapmap, mapmap) | 
|  | } | 
|  |  | 
|  | crazy := thrifttest.NewInsanity() | 
|  | crazy.UserMap = map[thrifttest.Numberz]thrifttest.UserId{ | 
|  | thrifttest.Numberz_FIVE:  5, | 
|  | thrifttest.Numberz_EIGHT: 8, | 
|  | } | 
|  | truck1 := thrifttest.NewXtruct() | 
|  | truck1.StringThing = "Goodbye4" | 
|  | truck1.ByteThing = 4 | 
|  | truck1.I32Thing = 4 | 
|  | truck1.I64Thing = 4 | 
|  | truck2 := thrifttest.NewXtruct() | 
|  | truck2.StringThing = "Hello2" | 
|  | truck2.ByteThing = 2 | 
|  | truck2.I32Thing = 2 | 
|  | truck2.I64Thing = 2 | 
|  | crazy.Xtructs = []*thrifttest.Xtruct{ | 
|  | truck1, | 
|  | truck2, | 
|  | } | 
|  | insanity, err := client.TestInsanity(defaultCtx, crazy) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestInsanity() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(crazy, insanity[1][2]) { | 
|  | t.Fatalf("Unexpected TestInsanity() first result expected %#v, got %#v ", | 
|  | crazy, | 
|  | insanity[1][2]) | 
|  | } | 
|  | if !reflect.DeepEqual(crazy, insanity[1][3]) { | 
|  | t.Fatalf("Unexpected TestInsanity() second result expected %#v, got %#v ", | 
|  | crazy, | 
|  | insanity[1][3]) | 
|  | } | 
|  | if len(insanity[2][6].UserMap) > 0 || len(insanity[2][6].Xtructs) > 0 { | 
|  | t.Fatalf("Unexpected TestInsanity() non-empty result got %#v ", | 
|  | insanity[2][6]) | 
|  | } | 
|  |  | 
|  | xxsret, err := client.TestMulti(defaultCtx, 42, 4242, 424242, map[int16]string{1: "blah", 2: "thing"}, thrifttest.Numberz_EIGHT, thrifttest.UserId(24)) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestMulti() call: ", err) | 
|  | } | 
|  | if !reflect.DeepEqual(xxs, xxsret) { | 
|  | t.Fatalf("Unexpected TestMulti() result expected %#v, got %#v ", xxs, xxsret) | 
|  | } | 
|  |  | 
|  | err = client.TestException(defaultCtx, "Xception") | 
|  | if err == nil { | 
|  | t.Fatalf("Expecting exception in TestException() call") | 
|  | } | 
|  | if !reflect.DeepEqual(err, xcept) { | 
|  | t.Fatalf("Unexpected TestException() result expected %#v, got %#v ", xcept, err) | 
|  | } | 
|  |  | 
|  | err = client.TestException(defaultCtx, "TException") | 
|  | _, ok := err.(thrift.TApplicationException) | 
|  | if err == nil || !ok { | 
|  | t.Fatalf("Unexpected TestException() result expected ApplicationError, got %#v ", err) | 
|  | } | 
|  |  | 
|  | ign, err := client.TestMultiException(defaultCtx, "Xception", "ignoreme") | 
|  | if ign != nil || err == nil { | 
|  | t.Fatalf("Expecting exception in TestMultiException() call") | 
|  | } | 
|  | if !reflect.DeepEqual(err, &thrifttest.Xception{ErrorCode: 1001, Message: "This is an Xception"}) { | 
|  | t.Fatalf("Unexpected TestMultiException() %#v ", err) | 
|  | } | 
|  |  | 
|  | ign, err = client.TestMultiException(defaultCtx, "Xception2", "ignoreme") | 
|  | if ign != nil || err == nil { | 
|  | t.Fatalf("Expecting exception in TestMultiException() call") | 
|  | } | 
|  | expecting := &thrifttest.Xception2{ErrorCode: 2002, StructThing: &thrifttest.Xtruct{StringThing: "This is an Xception2"}} | 
|  |  | 
|  | if !reflect.DeepEqual(err, expecting) { | 
|  | t.Fatalf("Unexpected TestMultiException() %#v ", err) | 
|  | } | 
|  |  | 
|  | err = client.TestOneway(defaultCtx, 2) | 
|  | if err != nil { | 
|  | t.Fatalf("Unexpected error in TestOneway() call: ", err) | 
|  | } | 
|  |  | 
|  | //Make sure the connection still alive | 
|  | if err = client.TestVoid(defaultCtx); err != nil { | 
|  | t.Fatalf("Unexpected error in TestVoid() call: ", err) | 
|  | } | 
|  | } |