blob: c9911e28b4e2eab9703eaa001428f7ad64c610da [file] [log] [blame]
/*
* 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.
*/
module thrift_test_client;
import std.conv;
import std.datetime;
import std.exception : enforce;
import std.getopt;
import std.stdio;
import std.string;
import std.traits;
import thrift.base;
import thrift.codegen.client;
import thrift.protocol.base;
import thrift.protocol.binary;
import thrift.protocol.compact;
import thrift.protocol.json;
import thrift.transport.base;
import thrift.transport.buffered;
import thrift.transport.framed;
import thrift.transport.http;
import thrift.transport.zlib;
import thrift.transport.socket;
import thrift.transport.ssl;
import thrift.util.hashset;
import thrift_test_common;
import thrift.test.ThriftTest;
import thrift.test.ThriftTest_types;
enum TransportType {
buffered,
framed,
http,
zlib,
raw
}
TProtocol createProtocol(T)(T trans, ProtocolType type) {
final switch (type) {
case ProtocolType.binary:
return tBinaryProtocol(trans);
case ProtocolType.compact:
return tCompactProtocol(trans);
case ProtocolType.json:
return tJsonProtocol(trans);
}
}
void main(string[] args) {
string host = "localhost";
ushort port = 9090;
uint numTests = 1;
bool ssl;
ProtocolType protocolType;
TransportType transportType;
bool zlib;
bool trace;
getopt(args,
"numTests|n", &numTests,
"protocol", &protocolType,
"ssl", &ssl,
"transport", &transportType,
"zlib", &zlib,
"trace", &trace,
"port", &port,
"host", (string _, string value) {
auto parts = split(value, ":");
if (parts.length > 1) {
// IPv6 addresses can contain colons, so take the last part for the
// port.
host = join(parts[0 .. $ - 1], ":");
port = to!ushort(parts[$ - 1]);
} else {
host = value;
}
}
);
port = to!ushort(port);
TSocket socket;
if (ssl) {
auto sslContext = new TSSLContext();
sslContext.ciphers = "ALL:!ADH:!LOW:!EXP:!MD5:@STRENGTH";
sslContext.authenticate = true;
sslContext.loadTrustedCertificates("../../../test/keys/CA.pem");
socket = new TSSLSocket(sslContext, host, port);
} else {
socket = new TSocket(host, port);
}
TTransport transport;
final switch (transportType) {
case TransportType.buffered:
transport = new TBufferedTransport(socket);
break;
case TransportType.framed:
transport = new TFramedTransport(socket);
break;
case TransportType.http:
transport = new TClientHttpTransport(socket, host, "/service");
break;
case TransportType.zlib:
transport = new TZlibTransport(socket);
break;
case TransportType.raw:
transport = socket;
break;
}
if (zlib && transportType != TransportType.zlib) {
transport = new TZlibTransport(socket);
}
TProtocol protocol = createProtocol(transport, protocolType);
auto client = tClient!ThriftTest(protocol);
ulong time_min;
ulong time_max;
ulong time_tot;
StopWatch sw;
foreach(test; 0 .. numTests) {
sw.start();
protocol.transport.open();
if (trace) writefln("Test #%s, connect %s:%s", test + 1, host, port);
if (trace) write("testVoid()");
client.testVoid();
if (trace) writeln(" = void");
if (trace) write("testString(\"Test\")");
string s = client.testString("Test");
if (trace) writefln(" = \"%s\"", s);
enforce(s == "Test");
if (trace) write("testByte(1)");
byte u8 = client.testByte(1);
if (trace) writefln(" = %s", u8);
enforce(u8 == 1);
if (trace) write("testI32(-1)");
int i32 = client.testI32(-1);
if (trace) writefln(" = %s", i32);
enforce(i32 == -1);
if (trace) write("testI64(-34359738368)");
long i64 = client.testI64(-34359738368L);
if (trace) writefln(" = %s", i64);
enforce(i64 == -34359738368L);
if (trace) write("testDouble(-5.2098523)");
double dub = client.testDouble(-5.2098523);
if (trace) writefln(" = %s", dub);
enforce(dub == -5.2098523);
// TODO: add testBinary() call
Xtruct out1;
out1.string_thing = "Zero";
out1.byte_thing = 1;
out1.i32_thing = -3;
out1.i64_thing = -5;
if (trace) writef("testStruct(%s)", out1);
auto in1 = client.testStruct(out1);
if (trace) writefln(" = %s", in1);
enforce(in1 == out1);
if (trace) write("testNest({1, {\"Zero\", 1, -3, -5}), 5}");
Xtruct2 out2;
out2.byte_thing = 1;
out2.struct_thing = out1;
out2.i32_thing = 5;
auto in2 = client.testNest(out2);
in1 = in2.struct_thing;
if (trace) writefln(" = {%s, {\"%s\", %s, %s, %s}, %s}", in2.byte_thing,
in1.string_thing, in1.byte_thing, in1.i32_thing, in1.i64_thing,
in2.i32_thing);
enforce(in2 == out2);
int[int] mapout;
for (int i = 0; i < 5; ++i) {
mapout[i] = i - 10;
}
if (trace) writef("testMap({%s})", mapout);
auto mapin = client.testMap(mapout);
if (trace) writefln(" = {%s}", mapin);
enforce(mapin == mapout);
auto setout = new HashSet!int;
for (int i = -2; i < 3; ++i) {
setout ~= i;
}
if (trace) writef("testSet(%s)", setout);
auto setin = client.testSet(setout);
if (trace) writefln(" = %s", setin);
enforce(setin == setout);
int[] listout;
for (int i = -2; i < 3; ++i) {
listout ~= i;
}
if (trace) writef("testList(%s)", listout);
auto listin = client.testList(listout);
if (trace) writefln(" = %s", listin);
enforce(listin == listout);
{
if (trace) write("testEnum(ONE)");
auto ret = client.testEnum(Numberz.ONE);
if (trace) writefln(" = %s", ret);
enforce(ret == Numberz.ONE);
if (trace) write("testEnum(TWO)");
ret = client.testEnum(Numberz.TWO);
if (trace) writefln(" = %s", ret);
enforce(ret == Numberz.TWO);
if (trace) write("testEnum(THREE)");
ret = client.testEnum(Numberz.THREE);
if (trace) writefln(" = %s", ret);
enforce(ret == Numberz.THREE);
if (trace) write("testEnum(FIVE)");
ret = client.testEnum(Numberz.FIVE);
if (trace) writefln(" = %s", ret);
enforce(ret == Numberz.FIVE);
if (trace) write("testEnum(EIGHT)");
ret = client.testEnum(Numberz.EIGHT);
if (trace) writefln(" = %s", ret);
enforce(ret == Numberz.EIGHT);
}
if (trace) write("testTypedef(309858235082523)");
UserId uid = client.testTypedef(309858235082523L);
if (trace) writefln(" = %s", uid);
enforce(uid == 309858235082523L);
if (trace) write("testMapMap(1)");
auto mm = client.testMapMap(1);
if (trace) writefln(" = {%s}", mm);
// Simply doing == doesn't seem to work for nested AAs.
foreach (key, value; mm) {
enforce(testMapMapReturn[key] == value);
}
foreach (key, value; testMapMapReturn) {
enforce(mm[key] == value);
}
Insanity insane;
insane.userMap[Numberz.FIVE] = 5000;
Xtruct truck;
truck.string_thing = "Truck";
truck.byte_thing = 8;
truck.i32_thing = 8;
truck.i64_thing = 8;
insane.xtructs ~= truck;
if (trace) write("testInsanity()");
auto whoa = client.testInsanity(insane);
if (trace) writefln(" = %s", whoa);
// Commented for now, this is cumbersome to write without opEqual getting
// called on AA comparison.
// enforce(whoa == testInsanityReturn);
{
try {
if (trace) write("client.testException(\"Xception\") =>");
client.testException("Xception");
if (trace) writeln(" void\nFAILURE");
throw new Exception("testException failed.");
} catch (Xception e) {
if (trace) writefln(" {%s, \"%s\"}", e.errorCode, e.message);
}
try {
if (trace) write("client.testException(\"TException\") =>");
client.testException("Xception");
if (trace) writeln(" void\nFAILURE");
throw new Exception("testException failed.");
} catch (TException e) {
if (trace) writefln(" {%s}", e.msg);
}
try {
if (trace) write("client.testException(\"success\") =>");
client.testException("success");
if (trace) writeln(" void");
} catch (Exception e) {
if (trace) writeln(" exception\nFAILURE");
throw new Exception("testException failed.");
}
}
{
try {
if (trace) write("client.testMultiException(\"Xception\", \"test 1\") =>");
auto result = client.testMultiException("Xception", "test 1");
if (trace) writeln(" result\nFAILURE");
throw new Exception("testMultiException failed.");
} catch (Xception e) {
if (trace) writefln(" {%s, \"%s\"}", e.errorCode, e.message);
}
try {
if (trace) write("client.testMultiException(\"Xception2\", \"test 2\") =>");
auto result = client.testMultiException("Xception2", "test 2");
if (trace) writeln(" result\nFAILURE");
throw new Exception("testMultiException failed.");
} catch (Xception2 e) {
if (trace) writefln(" {%s, {\"%s\"}}",
e.errorCode, e.struct_thing.string_thing);
}
try {
if (trace) writef("client.testMultiException(\"success\", \"test 3\") =>");
auto result = client.testMultiException("success", "test 3");
if (trace) writefln(" {{\"%s\"}}", result.string_thing);
} catch (Exception e) {
if (trace) writeln(" exception\nFAILURE");
throw new Exception("testMultiException failed.");
}
}
// Do not run oneway test when doing multiple iterations, as it blocks the
// server for three seconds.
if (numTests == 1) {
if (trace) writef("client.testOneway(3) =>");
auto onewayWatch = StopWatch(AutoStart.yes);
client.testOneway(3);
onewayWatch.stop();
if (onewayWatch.peek().msecs > 200) {
if (trace) {
writefln(" FAILURE - took %s ms", onewayWatch.peek().usecs / 1000.0);
}
throw new Exception("testOneway failed.");
} else {
if (trace) {
writefln(" success - took %s ms", onewayWatch.peek().usecs / 1000.0);
}
}
// Redo a simple test after the oneway to make sure we aren't "off by
// one", which would be the case if the server treated oneway methods
// like normal ones.
if (trace) write("re-test testI32(-1)");
i32 = client.testI32(-1);
if (trace) writefln(" = %s", i32);
}
// Time metering.
sw.stop();
immutable tot = sw.peek().usecs;
if (trace) writefln("Total time: %s us\n", tot);
time_tot += tot;
if (time_min == 0 || tot < time_min) {
time_min = tot;
}
if (tot > time_max) {
time_max = tot;
}
protocol.transport.close();
sw.reset();
}
writeln("All tests done.");
if (numTests > 1) {
auto time_avg = time_tot / numTests;
writefln("Min time: %s us", time_min);
writefln("Max time: %s us", time_max);
writefln("Avg time: %s us", time_avg);
}
}