func (p *ThriftTestDriver) Start() {
client := p.client
t := p.t
if client.TestVoid() != nil {
t.Fatal("TestVoid failed")
}
if r, err := client.TestString("Test"); r != "Test" || err != nil {
t.Fatal("TestString with simple text failed")
}
if r, err := client.TestString(""); r != "" || err != nil {
t.Fatal("TestString with empty text failed")
}
stringTest := "Afrikaans, Alemannisch, Aragonés, العربية, مصرى, " +
"Asturianu, Aymar aru, Azərbaycan, Башҡорт, Boarisch, Žemaitėška, " +
"Беларуская, Беларуская (тарашкевіца), Български, Bamanankan, " +
"বাংলা, Brezhoneg, Bosanski, Català, Mìng-dĕ̤ng-ngṳ̄, Нохчийн, " +
"Cebuano, ᏣᎳᎩ, Česky, Словѣ́ньскъ / ⰔⰎⰑⰂⰡⰐⰠⰔⰍⰟ, Чӑвашла, Cymraeg, " +
"Dansk, Zazaki, ދިވެހިބަސް, Ελληνικά, Emiliàn e rumagnòl, English, " +
"Esperanto, Español, Eesti, Euskara, فارسی, Suomi, Võro, Føroyskt, " +
"Français, Arpetan, Furlan, Frysk, Gaeilge, 贛語, Gàidhlig, Galego, " +
"Avañe'ẽ, ગુજરાતી, Gaelg, עברית, हिन्दी, Fiji Hindi, Hrvatski, " +
"Kreyòl ayisyen, Magyar, Հայերեն, Interlingua, Bahasa Indonesia, " +
"Ilokano, Ido, Íslenska, Italiano, 日本語, Lojban, Basa Jawa, " +
"ქართული, Kongo, Kalaallisut, ಕನ್ನಡ, 한국어, Къарачай-Малкъар, " +
"Ripoarisch, Kurdî, Коми, Kernewek, Кыргызча, Latina, Ladino, " +
"Lëtzebuergesch, Limburgs, Lingála, ລາວ, Lietuvių, Latviešu, Basa " +
"Banyumasan, Malagasy, Македонски, മലയാളം, मराठी, مازِرونی, Bahasa " +
"Melayu, Nnapulitano, Nedersaksisch, नेपाल भाषा, Nederlands, " +
"Norsk (nynorsk), Norsk (bokmål), Nouormand, Diné bizaad, " +
"Occitan, Иронау, Papiamentu, Deitsch, Polski, پنجابی, پښتو, " +
"Norfuk / Pitkern, Português, Runa Simi, Rumantsch, Romani, Română, " +
"Русский, Саха тыла, Sardu, Sicilianu, Scots, Sámegiella, Simple " +
"English, Slovenčina, Slovenščina, Српски / Srpski, Seeltersk, " +
"Svenska, Kiswahili, தமிழ், తెలుగు, Тоҷикӣ, ไทย, Türkmençe, Tagalog, " +
"Türkçe, Татарча/Tatarça, Українська, اردو, Tiếng Việt, Volapük, " +
"Walon, Winaray, 吴语, isiXhosa, ייִדיש, Yorùbá, Zeêuws, 中文, " +
"Bân-lâm-gú, 粵語"
if r, err := client.TestString(stringTest); r != stringTest || err != nil {
t.Fatal("TestString with all languages failed")
}
specialCharacters := "quote: \" backslash:" +
" backspace: \b formfeed: \f newline: \n return: \r tab: " +
" now-all-of-them-together: '\\\b\n\r\t'" +
" now-a-bunch-of-junk: !@#$%&()(&%$#{}{}<><><" +
" char-to-test-json-parsing: ]] \"]] \\\" }}}{ [[[ "
if r, err := client.TestString(specialCharacters); r != specialCharacters || err != nil {
t.Fatal("TestString with specialCharacters failed")
}
if r, err := client.TestByte(1); r != 1 || err != nil {
t.Fatal("TestByte(1) failed")
}
if r, err := client.TestByte(0); r != 0 || err != nil {
t.Fatal("TestByte(0) failed")
}
if r, err := client.TestByte(-1); r != -1 || err != nil {
t.Fatal("TestByte(-1) failed")
}
if r, err := client.TestByte(-127); r != -127 || err != nil {
t.Fatal("TestByte(-127) failed")
}
if r, err := client.TestI32(-1); r != -1 || err != nil {
t.Fatal("TestI32(-1) failed")
}
if r, err := client.TestI32(1); r != 1 || err != nil {
t.Fatal("TestI32(1) failed")
}
if r, err := client.TestI64(-5); r != -5 || err != nil {
t.Fatal("TestI64(-5) failed")
}
if r, err := client.TestI64(5); r != 5 || err != nil {
t.Fatal("TestI64(5) failed")
}
if r, err := client.TestI64(-34359738368); r != -34359738368 || err != nil {
t.Fatal("TestI64(-34359738368) failed")
}
if r, err := client.TestDouble(-5.2098523); r != -5.2098523 || err != nil {
t.Fatal("TestDouble(-5.2098523) failed")
}
if r, err := client.TestDouble(-7.012052175215044); r != -7.012052175215044 || err != nil {
t.Fatal("TestDouble(-7.012052175215044) failed")
}
// TODO: add testBinary() call
out := thrifttest.NewXtruct()
out.StringThing = "Zero"
out.ByteThing = 1
out.I32Thing = -3
out.I64Thing = 1000000
if r, err := client.TestStruct(out); !reflect.DeepEqual(r, out) || err != nil {
t.Fatal("TestStruct failed")
}
out2 := thrifttest.NewXtruct2()
out2.ByteThing = 1
out2.StructThing = out
out2.I32Thing = 5
if r, err := client.TestNest(out2); !reflect.DeepEqual(r, out2) || err != nil {
t.Fatal("TestNest failed")
}
mapout := make(map[int32]int32)
for i := int32(0); i < 5; i++ {
mapout[i] = i - 10
}
if r, err := client.TestMap(mapout); !reflect.DeepEqual(r, mapout) || err != nil {
t.Fatal("TestMap failed")
}
mapTestInput := map[string]string{
"a": "123", "a b": "with spaces ", "same": "same", "0": "numeric key",
"longValue": stringTest, stringTest: "long key",
}
if r, err := client.TestStringMap(mapTestInput); !reflect.DeepEqual(r, mapTestInput) || err != nil {
t.Fatal("TestStringMap failed")
}
setTestInput := map[int32]struct{}{1: {}, 2: {}, 3: {}}
if r, err := client.TestSet(setTestInput); !reflect.DeepEqual(r, setTestInput) || err != nil {
t.Fatal("TestSet failed")
}
listTest := []int32{1, 2, 3}
if r, err := client.TestList(listTest); !reflect.DeepEqual(r, listTest) || err != nil {
t.Fatal("TestList failed")
}
if r, err := client.TestEnum(thrifttest.Numberz_ONE); r != thrifttest.Numberz_ONE || err != nil {
t.Fatal("TestEnum failed")
}
if r, err := client.TestTypedef(69); r != 69 || err != nil {
t.Fatal("TestTypedef failed")
}
mapMapTest := map[int32]map[int32]int32{
4: {1: 1, 2: 2, 3: 3, 4: 4},
-4: {-4: -4, -3: -3, -2: -2, -1: -1},
}
if r, err := client.TestMapMap(1); !reflect.DeepEqual(r, mapMapTest) || err != nil {
t.Fatal("TestMapMap failed")
}
crazyX1 := thrifttest.NewXtruct()
crazyX1.StringThing = "Goodbye4"
crazyX1.ByteThing = 4
crazyX1.I32Thing = 4
crazyX1.I64Thing = 4
crazyX2 := thrifttest.NewXtruct()
crazyX2.StringThing = "Hello2"
crazyX2.ByteThing = 2
crazyX2.I32Thing = 2
crazyX2.I64Thing = 2
crazy := thrifttest.NewInsanity()
crazy.UserMap = map[thrifttest.Numberz]thrifttest.UserId{5: 5, 8: 8}
crazy.Xtructs = []*thrifttest.Xtruct{crazyX1, crazyX2}
crazyEmpty := thrifttest.NewInsanity()
crazyEmpty.UserMap = map[thrifttest.Numberz]thrifttest.UserId{}
crazyEmpty.Xtructs = []*thrifttest.Xtruct{}
insanity := map[thrifttest.UserId]map[thrifttest.Numberz]*thrifttest.Insanity{
1: {thrifttest.Numberz_TWO: crazy, thrifttest.Numberz_THREE: crazy},
2: {thrifttest.Numberz_SIX: crazyEmpty},
}
if r, err := client.TestInsanity(crazy); !reflect.DeepEqual(r, insanity) || err != nil {
t.Fatal("TestInsanity failed")
}
if err := client.TestException("TException"); err == nil {
t.Fatal("TestException TException failed")
}
if err, ok := client.TestException("Xception").(*thrifttest.Xception); ok == false || err == nil {
t.Fatal("TestException Xception failed")
} else if err.ErrorCode != 1001 || err.Message != "Xception" {
t.Fatal("TestException Xception failed")
}
if err := client.TestException("no Exception"); err != nil {
t.Fatal("TestException no Exception failed")
}
if err := client.TestOneway(0); err != nil {
t.Fatal("TestOneway failed")
}
}
func main() {
flag.Usage = Usage
var host string
var port int
var protocol string
var urlString string
var framed bool
var useHttp bool
var parsedUrl url.URL
var trans thrift.TTransport
_ = strconv.Atoi
_ = math.Abs
flag.Usage = Usage
flag.StringVar(&host, "h", "localhost", "Specify host and port")
flag.IntVar(&port, "p", 9090, "Specify port")
flag.StringVar(&protocol, "P", "binary", "Specify the protocol (binary, compact, simplejson, json)")
flag.StringVar(&urlString, "u", "", "Specify the url")
flag.BoolVar(&framed, "framed", false, "Use framed transport")
flag.BoolVar(&useHttp, "http", false, "Use http")
flag.Parse()
if len(urlString) > 0 {
parsedUrl, err := url.Parse(urlString)
if err != nil {
fmt.Fprintln(os.Stderr, "Error parsing URL: ", err)
flag.Usage()
}
host = parsedUrl.Host
useHttp = len(parsedUrl.Scheme) <= 0 || parsedUrl.Scheme == "http"
} else if useHttp {
_, err := url.Parse(fmt.Sprint("http://", host, ":", port))
if err != nil {
fmt.Fprintln(os.Stderr, "Error parsing URL: ", err)
flag.Usage()
}
}
cmd := flag.Arg(0)
var err error
if useHttp {
trans, err = thrift.NewTHttpClient(parsedUrl.String())
} else {
portStr := fmt.Sprint(port)
if strings.Contains(host, ":") {
host, portStr, err = net.SplitHostPort(host)
if err != nil {
fmt.Fprintln(os.Stderr, "error with host:", err)
os.Exit(1)
}
}
trans, err = thrift.NewTSocket(net.JoinHostPort(host, portStr))
if err != nil {
fmt.Fprintln(os.Stderr, "error resolving address:", err)
os.Exit(1)
}
if framed {
trans = thrift.NewTFramedTransport(trans)
}
}
if err != nil {
fmt.Fprintln(os.Stderr, "Error creating transport", err)
os.Exit(1)
}
defer trans.Close()
var protocolFactory thrift.TProtocolFactory
switch protocol {
case "compact":
protocolFactory = thrift.NewTCompactProtocolFactory()
break
case "simplejson":
protocolFactory = thrift.NewTSimpleJSONProtocolFactory()
break
case "json":
protocolFactory = thrift.NewTJSONProtocolFactory()
break
case "binary", "":
protocolFactory = thrift.NewTBinaryProtocolFactoryDefault()
break
default:
fmt.Fprintln(os.Stderr, "Invalid protocol specified: ", protocol)
Usage()
os.Exit(1)
}
client := thrifttest.NewThriftTestClientFactory(trans, protocolFactory)
if err := trans.Open(); err != nil {
fmt.Fprintln(os.Stderr, "Error opening socket to ", host, ":", port, " ", err)
os.Exit(1)
}
switch cmd {
case "testVoid":
if flag.NArg()-1 != 0 {
fmt.Fprintln(os.Stderr, "TestVoid requires 0 args")
flag.Usage()
}
fmt.Print(client.TestVoid())
fmt.Print("\n")
break
case "testString":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestString requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.TestString(value0))
fmt.Print("\n")
break
case "testBool":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestBool requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1) == "true"
value0 := argvalue0
fmt.Print(client.TestBool(value0))
fmt.Print("\n")
break
case "testByte":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestByte requires 1 args")
flag.Usage()
}
tmp0, err100 := (strconv.Atoi(flag.Arg(1)))
if err100 != nil {
Usage()
return
}
argvalue0 := byte(tmp0)
value0 := argvalue0
fmt.Print(client.TestByte(value0))
fmt.Print("\n")
break
case "testI32":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestI32 requires 1 args")
flag.Usage()
}
tmp0, err101 := (strconv.Atoi(flag.Arg(1)))
if err101 != nil {
Usage()
return
}
argvalue0 := int32(tmp0)
value0 := argvalue0
fmt.Print(client.TestI32(value0))
fmt.Print("\n")
break
case "testI64":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestI64 requires 1 args")
flag.Usage()
}
argvalue0, err102 := (strconv.ParseInt(flag.Arg(1), 10, 64))
if err102 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.TestI64(value0))
fmt.Print("\n")
break
case "testDouble":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestDouble requires 1 args")
flag.Usage()
}
argvalue0, err103 := (strconv.ParseFloat(flag.Arg(1), 64))
if err103 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.TestDouble(value0))
fmt.Print("\n")
break
case "testBinary":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestBinary requires 1 args")
flag.Usage()
}
argvalue0 := []byte(flag.Arg(1))
value0 := argvalue0
fmt.Print(client.TestBinary(value0))
fmt.Print("\n")
break
case "testStruct":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestStruct requires 1 args")
flag.Usage()
}
arg105 := flag.Arg(1)
mbTrans106 := thrift.NewTMemoryBufferLen(len(arg105))
defer mbTrans106.Close()
_, err107 := mbTrans106.WriteString(arg105)
if err107 != nil {
Usage()
return
}
factory108 := thrift.NewTSimpleJSONProtocolFactory()
jsProt109 := factory108.GetProtocol(mbTrans106)
argvalue0 := thrifttest.NewXtruct()
err110 := argvalue0.Read(jsProt109)
if err110 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.TestStruct(value0))
fmt.Print("\n")
break
case "testNest":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestNest requires 1 args")
flag.Usage()
}
arg111 := flag.Arg(1)
mbTrans112 := thrift.NewTMemoryBufferLen(len(arg111))
defer mbTrans112.Close()
_, err113 := mbTrans112.WriteString(arg111)
if err113 != nil {
Usage()
return
}
factory114 := thrift.NewTSimpleJSONProtocolFactory()
jsProt115 := factory114.GetProtocol(mbTrans112)
argvalue0 := thrifttest.NewXtruct2()
err116 := argvalue0.Read(jsProt115)
if err116 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.TestNest(value0))
fmt.Print("\n")
break
case "testMap":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestMap requires 1 args")
flag.Usage()
}
arg117 := flag.Arg(1)
mbTrans118 := thrift.NewTMemoryBufferLen(len(arg117))
defer mbTrans118.Close()
_, err119 := mbTrans118.WriteString(arg117)
if err119 != nil {
Usage()
return
}
factory120 := thrift.NewTSimpleJSONProtocolFactory()
jsProt121 := factory120.GetProtocol(mbTrans118)
containerStruct0 := thrifttest.NewThriftTestTestMapArgs()
err122 := containerStruct0.ReadField1(jsProt121)
if err122 != nil {
Usage()
return
}
argvalue0 := containerStruct0.Thing
value0 := argvalue0
fmt.Print(client.TestMap(value0))
fmt.Print("\n")
break
case "testStringMap":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestStringMap requires 1 args")
flag.Usage()
}
arg123 := flag.Arg(1)
mbTrans124 := thrift.NewTMemoryBufferLen(len(arg123))
defer mbTrans124.Close()
_, err125 := mbTrans124.WriteString(arg123)
if err125 != nil {
Usage()
return
}
factory126 := thrift.NewTSimpleJSONProtocolFactory()
jsProt127 := factory126.GetProtocol(mbTrans124)
containerStruct0 := thrifttest.NewThriftTestTestStringMapArgs()
err128 := containerStruct0.ReadField1(jsProt127)
if err128 != nil {
Usage()
return
}
argvalue0 := containerStruct0.Thing
value0 := argvalue0
fmt.Print(client.TestStringMap(value0))
fmt.Print("\n")
break
case "testSet":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestSet requires 1 args")
flag.Usage()
}
arg129 := flag.Arg(1)
mbTrans130 := thrift.NewTMemoryBufferLen(len(arg129))
defer mbTrans130.Close()
_, err131 := mbTrans130.WriteString(arg129)
if err131 != nil {
Usage()
return
}
factory132 := thrift.NewTSimpleJSONProtocolFactory()
jsProt133 := factory132.GetProtocol(mbTrans130)
containerStruct0 := thrifttest.NewThriftTestTestSetArgs()
err134 := containerStruct0.ReadField1(jsProt133)
if err134 != nil {
Usage()
return
}
argvalue0 := containerStruct0.Thing
value0 := argvalue0
fmt.Print(client.TestSet(value0))
fmt.Print("\n")
break
case "testList":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestList requires 1 args")
flag.Usage()
}
arg135 := flag.Arg(1)
mbTrans136 := thrift.NewTMemoryBufferLen(len(arg135))
defer mbTrans136.Close()
_, err137 := mbTrans136.WriteString(arg135)
if err137 != nil {
Usage()
return
}
factory138 := thrift.NewTSimpleJSONProtocolFactory()
jsProt139 := factory138.GetProtocol(mbTrans136)
containerStruct0 := thrifttest.NewThriftTestTestListArgs()
err140 := containerStruct0.ReadField1(jsProt139)
if err140 != nil {
Usage()
return
}
argvalue0 := containerStruct0.Thing
value0 := argvalue0
fmt.Print(client.TestList(value0))
fmt.Print("\n")
break
case "testEnum":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestEnum requires 1 args")
flag.Usage()
}
tmp0, err := (strconv.Atoi(flag.Arg(1)))
if err != nil {
Usage()
return
}
argvalue0 := thrifttest.Numberz(tmp0)
value0 := argvalue0
fmt.Print(client.TestEnum(value0))
fmt.Print("\n")
break
case "testTypedef":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestTypedef requires 1 args")
flag.Usage()
}
argvalue0, err141 := (strconv.ParseInt(flag.Arg(1), 10, 64))
if err141 != nil {
Usage()
return
}
value0 := thrifttest.UserId(argvalue0)
fmt.Print(client.TestTypedef(value0))
fmt.Print("\n")
break
case "testMapMap":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestMapMap requires 1 args")
flag.Usage()
}
tmp0, err142 := (strconv.Atoi(flag.Arg(1)))
if err142 != nil {
Usage()
return
}
argvalue0 := int32(tmp0)
value0 := argvalue0
fmt.Print(client.TestMapMap(value0))
fmt.Print("\n")
break
case "testInsanity":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestInsanity requires 1 args")
flag.Usage()
}
arg143 := flag.Arg(1)
mbTrans144 := thrift.NewTMemoryBufferLen(len(arg143))
defer mbTrans144.Close()
_, err145 := mbTrans144.WriteString(arg143)
if err145 != nil {
Usage()
return
}
factory146 := thrift.NewTSimpleJSONProtocolFactory()
jsProt147 := factory146.GetProtocol(mbTrans144)
argvalue0 := thrifttest.NewInsanity()
err148 := argvalue0.Read(jsProt147)
if err148 != nil {
Usage()
return
}
value0 := argvalue0
fmt.Print(client.TestInsanity(value0))
fmt.Print("\n")
break
case "testMulti":
if flag.NArg()-1 != 6 {
fmt.Fprintln(os.Stderr, "TestMulti requires 6 args")
flag.Usage()
}
tmp0, err149 := (strconv.Atoi(flag.Arg(1)))
if err149 != nil {
Usage()
return
}
argvalue0 := byte(tmp0)
value0 := argvalue0
tmp1, err150 := (strconv.Atoi(flag.Arg(2)))
if err150 != nil {
Usage()
return
}
argvalue1 := int32(tmp1)
value1 := argvalue1
argvalue2, err151 := (strconv.ParseInt(flag.Arg(3), 10, 64))
if err151 != nil {
Usage()
return
}
value2 := argvalue2
arg152 := flag.Arg(4)
mbTrans153 := thrift.NewTMemoryBufferLen(len(arg152))
defer mbTrans153.Close()
_, err154 := mbTrans153.WriteString(arg152)
if err154 != nil {
Usage()
return
}
factory155 := thrift.NewTSimpleJSONProtocolFactory()
jsProt156 := factory155.GetProtocol(mbTrans153)
containerStruct3 := thrifttest.NewThriftTestTestMultiArgs()
err157 := containerStruct3.ReadField4(jsProt156)
if err157 != nil {
Usage()
return
}
argvalue3 := containerStruct3.Arg3
value3 := argvalue3
tmp4, err := (strconv.Atoi(flag.Arg(5)))
if err != nil {
Usage()
return
}
argvalue4 := thrifttest.Numberz(tmp4)
value4 := argvalue4
argvalue5, err158 := (strconv.ParseInt(flag.Arg(6), 10, 64))
if err158 != nil {
Usage()
return
}
value5 := thrifttest.UserId(argvalue5)
fmt.Print(client.TestMulti(value0, value1, value2, value3, value4, value5))
fmt.Print("\n")
break
case "testException":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestException requires 1 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
fmt.Print(client.TestException(value0))
fmt.Print("\n")
break
case "testMultiException":
if flag.NArg()-1 != 2 {
fmt.Fprintln(os.Stderr, "TestMultiException requires 2 args")
flag.Usage()
}
argvalue0 := flag.Arg(1)
value0 := argvalue0
argvalue1 := flag.Arg(2)
value1 := argvalue1
fmt.Print(client.TestMultiException(value0, value1))
fmt.Print("\n")
break
case "testOneway":
if flag.NArg()-1 != 1 {
fmt.Fprintln(os.Stderr, "TestOneway requires 1 args")
flag.Usage()
}
tmp0, err162 := (strconv.Atoi(flag.Arg(1)))
if err162 != nil {
Usage()
return
}
argvalue0 := int32(tmp0)
value0 := argvalue0
fmt.Print(client.TestOneway(value0))
fmt.Print("\n")
break
case "":
Usage()
break
default:
fmt.Fprintln(os.Stderr, "Invalid function ", cmd)
}
}