int main(int argc, char** argv) { cout.precision(19); int ERR_BASETYPES = 1; int ERR_STRUCTS = 2; int ERR_CONTAINERS = 4; int ERR_EXCEPTIONS = 8; int ERR_UNKNOWN = 64; string testDir = boost::filesystem::system_complete(argv[0]).parent_path().parent_path().parent_path().string(); string pemPath = testDir + "/keys/CA.pem"; #if _WIN32 transport::TWinsockSingleton::create(); #endif string host = "localhost"; int port = 9090; int numTests = 1; bool ssl = false; string transport_type = "buffered"; string protocol_type = "binary"; string domain_socket = ""; bool abstract_namespace = false; bool noinsane = false; int return_code = 0; boost::program_options::options_description desc("Allowed options"); desc.add_options()("help,h", "produce help message")("host", boost::program_options::value<string>(&host) ->default_value(host), "Host to connect")("port", boost::program_options::value<int>( &port)->default_value(port), "Port number to connect")( "domain-socket", boost::program_options::value<string>(&domain_socket)->default_value(domain_socket), "Domain Socket (e.g. /tmp/ThriftTest.thrift), instead of host and port")( "abstract-namespace", "Look for the domain socket in the Abstract Namespace (no connection with filesystem pathnames)")( "transport", boost::program_options::value<string>(&transport_type)->default_value(transport_type), "Transport: buffered, framed, http, evhttp")( "protocol", boost::program_options::value<string>(&protocol_type)->default_value(protocol_type), "Protocol: binary, header, compact, json")("ssl", "Encrypted Transport using SSL")( "testloops,n", boost::program_options::value<int>(&numTests)->default_value(numTests), "Number of Tests")("noinsane", "Do not run insanity test"); boost::program_options::variables_map vm; boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm); boost::program_options::notify(vm); if (vm.count("help")) { cout << desc << endl; return ERR_UNKNOWN; } try { if (!protocol_type.empty()) { if (protocol_type == "binary") { } else if (protocol_type == "compact") { } else if (protocol_type == "header") { } else if (protocol_type == "json") { } else { throw invalid_argument("Unknown protocol type " + protocol_type); } } if (!transport_type.empty()) { if (transport_type == "buffered") { } else if (transport_type == "framed") { } else if (transport_type == "http") { } else if (transport_type == "evhttp") { } else { throw invalid_argument("Unknown transport type " + transport_type); } } } catch (exception& e) { cerr << e.what() << endl; cout << desc << endl; return ERR_UNKNOWN; } if (vm.count("ssl")) { ssl = true; } if (vm.count("abstract-namespace")) { abstract_namespace = true; } if (vm.count("noinsane")) { noinsane = true; } boost::shared_ptr<TTransport> transport; boost::shared_ptr<TProtocol> protocol; boost::shared_ptr<TSocket> socket; boost::shared_ptr<TSSLSocketFactory> factory; if (ssl) { factory = boost::shared_ptr<TSSLSocketFactory>(new TSSLSocketFactory()); factory->ciphers("ALL:!ADH:!LOW:!EXP:!MD5:@STRENGTH"); factory->loadTrustedCertificates(pemPath.c_str()); factory->authenticate(true); socket = factory->createSocket(host, port); } else { if (domain_socket != "") { if (abstract_namespace) { std::string abstract_socket("\0", 1); abstract_socket += domain_socket; socket = boost::shared_ptr<TSocket>(new TSocket(abstract_socket)); } else { socket = boost::shared_ptr<TSocket>(new TSocket(domain_socket)); } port = 0; } else { socket = boost::shared_ptr<TSocket>(new TSocket(host, port)); } } if (transport_type.compare("http") == 0) { boost::shared_ptr<TTransport> httpSocket(new THttpClient(socket, host, "/service")); transport = httpSocket; } else if (transport_type.compare("framed") == 0) { boost::shared_ptr<TFramedTransport> framedSocket(new TFramedTransport(socket)); transport = framedSocket; } else { boost::shared_ptr<TBufferedTransport> bufferedSocket(new TBufferedTransport(socket)); transport = bufferedSocket; } if (protocol_type.compare("json") == 0) { boost::shared_ptr<TProtocol> jsonProtocol(new TJSONProtocol(transport)); protocol = jsonProtocol; } else if (protocol_type.compare("compact") == 0) { boost::shared_ptr<TProtocol> compactProtocol(new TCompactProtocol(transport)); protocol = compactProtocol; } else if (protocol_type == "header") { boost::shared_ptr<TProtocol> headerProtocol(new THeaderProtocol(transport)); protocol = headerProtocol; } else { boost::shared_ptr<TBinaryProtocol> binaryProtocol(new TBinaryProtocol(transport)); protocol = binaryProtocol; } // Connection info cout << "Connecting (" << transport_type << "/" << protocol_type << ") to: "; if (abstract_namespace) { cout << '@'; } cout << domain_socket; if (port != 0) { cout << host << ":" << port; } cout << endl; if (transport_type.compare("evhttp") == 0) { event_base* base = event_base_new(); cout << "Libevent Version: " << event_get_version() << endl; cout << "Libevent Method: " << event_base_get_method(base) << endl; #if LIBEVENT_VERSION_NUMBER >= 0x02000000 cout << "Libevent Features: 0x" << hex << event_base_get_features(base) << endl; #endif boost::shared_ptr<TProtocolFactory> protocolFactory(new TBinaryProtocolFactory()); boost::shared_ptr<TAsyncChannel> channel( new TEvhttpClientChannel(host.c_str(), "/", host.c_str(), port, base)); ThriftTestCobClient* client = new ThriftTestCobClient(channel, protocolFactory.get()); client->testVoid(tcxx::bind(testVoid_clientReturn, base, tcxx::placeholders::_1)); event_base_loop(base, 0); return 0; } ThriftTestClient testClient(protocol); uint64_t time_min = 0; uint64_t time_max = 0; uint64_t time_tot = 0; int test = 0; for (test = 0; test < numTests; ++test) { try { transport->open(); } catch (TTransportException& ex) { cout << "Connect failed: " << ex.what() << endl; return ERR_UNKNOWN; } /** * CONNECT TEST */ printf("Test #%d, connect %s:%d\n", test + 1, host.c_str(), port); uint64_t start = now(); /** * VOID TEST */ try { cout << "testVoid()" << flush; testClient.testVoid(); cout << " = void" << endl; } catch (TTransportException&) { // Stop here if transport got broken throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; } /** * STRING TEST */ cout << "testString(\"Test\")" << flush; string s; testClient.testString(s, "Test"); cout << " = " << s << endl; if (s != "Test") { cout << "*** FAILED ***" << endl; return_code |= ERR_BASETYPES; } try { string str( "}{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ú, 粵語"); cout << "testString(" << str << ") = " << flush; testClient.testString(s, str); cout << s << endl; if (s != str) { cout.imbue(locale("en_US.UTF8")); cout << "*** FAILED ***" << endl << "Expected string: " << str << " but got: " << s << endl << "CLEAR"; return_code |= ERR_BASETYPES; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; return return_code; } try { string str( "quote: \" backslash:" " forwardslash-escaped: \\/ " " 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: ]] \"]] \\\" }}}{ [[[ "); cout << "testString(" << str << ") = " << flush; testClient.testString(s, str); cout << s << endl; if (s != str) { cout.imbue(locale("en_US.UTF8")); cout << "*** FAILED ***" << endl << "Expected string: " << str << " but got: " << s << endl << "CLEAR"; ; return_code |= ERR_BASETYPES; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; return return_code; } /** * BOOL TEST */ cout << boolalpha; BASETYPE_IDENTITY_TEST(testBool, true); BASETYPE_IDENTITY_TEST(testBool, false); /** * BYTE TEST */ BASETYPE_IDENTITY_TEST(testByte, (int8_t)0); BASETYPE_IDENTITY_TEST(testByte, (int8_t)-1); BASETYPE_IDENTITY_TEST(testByte, (int8_t)42); BASETYPE_IDENTITY_TEST(testByte, (int8_t)-42); BASETYPE_IDENTITY_TEST(testByte, (int8_t)127); BASETYPE_IDENTITY_TEST(testByte, (int8_t)-128); /** * I32 TEST */ BASETYPE_IDENTITY_TEST(testI32, 0); BASETYPE_IDENTITY_TEST(testI32, -1); BASETYPE_IDENTITY_TEST(testI32, 190000013); BASETYPE_IDENTITY_TEST(testI32, -190000013); BASETYPE_IDENTITY_TEST(testI32, numeric_limits<int32_t>::max()); BASETYPE_IDENTITY_TEST(testI32, numeric_limits<int32_t>::min()); /** * I64 TEST */ BASETYPE_IDENTITY_TEST(testI64, (int64_t)0); BASETYPE_IDENTITY_TEST(testI64, (int64_t)-1); BASETYPE_IDENTITY_TEST(testI64, (int64_t)7000000000000000123LL); BASETYPE_IDENTITY_TEST(testI64, (int64_t)-7000000000000000123LL); BASETYPE_IDENTITY_TEST(testI64, (int64_t)pow(2LL, 32)); BASETYPE_IDENTITY_TEST(testI64, (int64_t)-pow(2LL, 32)); BASETYPE_IDENTITY_TEST(testI64, (int64_t)pow(2LL, 32) + 1); BASETYPE_IDENTITY_TEST(testI64, (int64_t)-pow(2LL, 32) - 1); BASETYPE_IDENTITY_TEST(testI64, numeric_limits<int64_t>::max()); BASETYPE_IDENTITY_TEST(testI64, numeric_limits<int64_t>::min()); /** * DOUBLE TEST */ // Comparing double values with plain equality because Thrift handles full precision of double BASETYPE_IDENTITY_TEST(testDouble, 0.0); BASETYPE_IDENTITY_TEST(testDouble, -1.0); BASETYPE_IDENTITY_TEST(testDouble, -5.2098523); BASETYPE_IDENTITY_TEST(testDouble, -0.000341012439638598279); BASETYPE_IDENTITY_TEST(testDouble, pow(2, 32)); BASETYPE_IDENTITY_TEST(testDouble, pow(2, 32) + 1); BASETYPE_IDENTITY_TEST(testDouble, pow(2, 53) - 1); BASETYPE_IDENTITY_TEST(testDouble, -pow(2, 32)); BASETYPE_IDENTITY_TEST(testDouble, -pow(2, 32) - 1); BASETYPE_IDENTITY_TEST(testDouble, -pow(2, 53) + 1); try { double expected = pow(10, 307); cout << "testDouble(" << expected << ") = " << flush; double actual = testClient.testDouble(expected); cout << "(" << actual << ")" << endl; if (expected - actual > pow(10, 292)) { cout << "*** FAILED ***" << endl << "Expected: " << expected << " but got: " << actual << endl; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; } try { double expected = pow(10, -292); cout << "testDouble(" << expected << ") = " << flush; double actual = testClient.testDouble(expected); cout << "(" << actual << ")" << endl; if (expected - actual > pow(10, -307)) { cout << "*** FAILED ***" << endl << "Expected: " << expected << " but got: " << actual << endl; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; } /** * BINARY TEST */ cout << "testBinary(empty)" << endl; try { string bin_result; testClient.testBinary(bin_result, string()); if (!bin_result.empty()) { cout << endl << "*** FAILED ***" << endl; cout << "invalid length: " << bin_result.size() << endl; return_code |= ERR_BASETYPES; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; } cout << "testBinary([-128..127]) = {" << flush; const signed char bin_data[256] = {-128, -127, -126, -125, -124, -123, -122, -121, -120, -119, -118, -117, -116, -115, -114, -113, -112, -111, -110, -109, -108, -107, -106, -105, -104, -103, -102, -101, -100, -99, -98, -97, -96, -95, -94, -93, -92, -91, -90, -89, -88, -87, -86, -85, -84, -83, -82, -81, -80, -79, -78, -77, -76, -75, -74, -73, -72, -71, -70, -69, -68, -67, -66, -65, -64, -63, -62, -61, -60, -59, -58, -57, -56, -55, -54, -53, -52, -51, -50, -49, -48, -47, -46, -45, -44, -43, -42, -41, -40, -39, -38, -37, -36, -35, -34, -33, -32, -31, -30, -29, -28, -27, -26, -25, -24, -23, -22, -21, -20, -19, -18, -17, -16, -15, -14, -13, -12, -11, -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104, 105, 106, 107, 108, 109, 110, 111, 112, 113, 114, 115, 116, 117, 118, 119, 120, 121, 122, 123, 124, 125, 126, 127}; try { string bin_result; testClient.testBinary(bin_result, string(reinterpret_cast<const char *>(bin_data), 256)); if (bin_result.size() != 256) { cout << endl << "*** FAILED ***" << endl; cout << "invalid length: " << bin_result.size() << endl; return_code |= ERR_BASETYPES; } else { bool first = true; bool failed = false; for (int i = 0; i < 256; ++i) { if (!first) cout << ","; else first = false; cout << static_cast<int>(bin_result[i]); if (!failed && bin_result[i] != i - 128) { failed = true; } } cout << "}" << endl; if (failed) { cout << "*** FAILED ***" << endl; return_code |= ERR_BASETYPES; } } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_BASETYPES; } /** * STRUCT TEST */ cout << "testStruct({\"Zero\", 1, -3, -5})" << flush; Xtruct out; out.string_thing = "Zero"; out.byte_thing = 1; out.i32_thing = -3; out.i64_thing = -5; Xtruct in; testClient.testStruct(in, out); printf(" = {\"%s\", %d, %d, %" PRId64 "}\n", in.string_thing.c_str(), (int)in.byte_thing, in.i32_thing, in.i64_thing); if (in != out) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } /** * NESTED STRUCT TEST */ cout << "testNest({1, {\"Zero\", 1, -3, -5}), 5}" << flush; Xtruct2 out2; out2.byte_thing = 1; out2.struct_thing = out; out2.i32_thing = 5; Xtruct2 in2; testClient.testNest(in2, out2); in = in2.struct_thing; printf(" = {%d, {\"%s\", %d, %d, %" PRId64 "}, %d}\n", in2.byte_thing, in.string_thing.c_str(), (int)in.byte_thing, in.i32_thing, in.i64_thing, in2.i32_thing); if (in2 != out2) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } /** * MAP TEST */ map<int32_t, int32_t> mapout; for (int32_t i = 0; i < 5; ++i) { mapout.insert(make_pair(i, i - 10)); } cout << "testMap({" << flush; map<int32_t, int32_t>::const_iterator m_iter; bool first = true; for (m_iter = mapout.begin(); m_iter != mapout.end(); ++m_iter) { if (first) { first = false; } else { cout << ","; } cout << m_iter->first << " => " << m_iter->second; } cout << "})"; map<int32_t, int32_t> mapin; testClient.testMap(mapin, mapout); cout << " = {"; first = true; for (m_iter = mapin.begin(); m_iter != mapin.end(); ++m_iter) { if (first) { first = false; } else { cout << ","; } cout << m_iter->first << " => " << m_iter->second; } cout << "}" << endl; if (mapin != mapout) { cout << "*** FAILED ***" << endl; return_code |= ERR_CONTAINERS; } /** * STRING MAP TEST */ cout << "testStringMap({a => 2, b => blah, some => thing}) = {" << flush; map<string, string> smapin; map<string, string> smapout; smapin["a"] = "2"; smapin["b"] = "blah"; smapin["some"] = "thing"; try { testClient.testStringMap(smapout, smapin); first = true; for (map<string, string>::const_iterator it = smapout.begin(); it != smapout.end(); ++it) { if (first) cout << ","; else first = false; cout << it->first << " => " << it->second; } cout << "}" << endl; if (smapin != smapout) { cout << "*** FAILED ***" << endl; return_code |= ERR_CONTAINERS; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_CONTAINERS; } /** * SET TEST */ set<int32_t> setout; for (int32_t i = -2; i < 3; ++i) { setout.insert(i); } cout << "testSet({" << flush; set<int32_t>::const_iterator s_iter; first = true; for (s_iter = setout.begin(); s_iter != setout.end(); ++s_iter) { if (first) { first = false; } else { cout << ","; } cout << *s_iter; } cout << "})"; set<int32_t> setin; testClient.testSet(setin, setout); cout << " = {"; first = true; for (s_iter = setin.begin(); s_iter != setin.end(); ++s_iter) { if (first) { first = false; } else { cout << ","; } cout << *s_iter; } cout << "}" << endl; if (setin != setout) { cout << "*** FAILED ***" << endl; return_code |= ERR_CONTAINERS; } /** * LIST TEST */ cout << "testList(empty)" << flush; try { vector<int32_t> listout; testClient.testList(listout, vector<int32_t>()); if (!listout.empty()) { cout << "*** FAILED ***" << endl; cout << "invalid length: " << listout.size() << endl; return_code |= ERR_CONTAINERS; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_CONTAINERS; } try { vector<int32_t> listout; for (int32_t i = -2; i < 3; ++i) { listout.push_back(i); } cout << "testList({" << flush; vector<int32_t>::const_iterator l_iter; first = true; for (l_iter = listout.begin(); l_iter != listout.end(); ++l_iter) { if (first) { first = false; } else { cout << ","; } cout << *l_iter; } cout << "})"; vector<int32_t> listin; testClient.testList(listin, listout); cout << " = {"; first = true; for (l_iter = listin.begin(); l_iter != listin.end(); ++l_iter) { if (first) { first = false; } else { cout << ","; } cout << *l_iter; } cout << "}" << endl; if (listin != listout) { cout << "*** FAILED ***" << endl; return_code |= ERR_CONTAINERS; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_CONTAINERS; } /** * ENUM TEST */ cout << "testEnum(ONE)" << flush; Numberz::type ret = testClient.testEnum(Numberz::ONE); cout << " = " << ret << endl; if (ret != Numberz::ONE) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } cout << "testEnum(TWO)" << flush; ret = testClient.testEnum(Numberz::TWO); cout << " = " << ret << endl; if (ret != Numberz::TWO) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } cout << "testEnum(THREE)" << flush; ret = testClient.testEnum(Numberz::THREE); cout << " = " << ret << endl; if (ret != Numberz::THREE) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } cout << "testEnum(FIVE)" << flush; ret = testClient.testEnum(Numberz::FIVE); cout << " = " << ret << endl; if (ret != Numberz::FIVE) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } cout << "testEnum(EIGHT)" << flush; ret = testClient.testEnum(Numberz::EIGHT); cout << " = " << ret << endl; if (ret != Numberz::EIGHT) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } /** * TYPEDEF TEST */ cout << "testTypedef(309858235082523)" << flush; UserId uid = testClient.testTypedef(309858235082523LL); cout << " = " << uid << endl; if (uid != 309858235082523LL) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } /** * NESTED MAP TEST */ cout << "testMapMap(1)" << flush; map<int32_t, map<int32_t, int32_t> > mm; testClient.testMapMap(mm, 1); cout << " = {"; map<int32_t, map<int32_t, int32_t> >::const_iterator mi; for (mi = mm.begin(); mi != mm.end(); ++mi) { printf("%d => {", mi->first); map<int32_t, int32_t>::const_iterator mi2; for (mi2 = mi->second.begin(); mi2 != mi->second.end(); ++mi2) { cout << mi2->first << " => " << mi2->second; } cout << "}, "; } cout << "}" << endl; if (mm.size() != 2 || mm[-4][-4] != -4 || mm[-4][-3] != -3 || mm[-4][-2] != -2 || mm[-4][-1] != -1 || mm[4][4] != 4 || mm[4][3] != 3 || mm[4][2] != 2 || mm[4][1] != 1) { cout << "*** FAILED ***" << endl; return_code |= ERR_CONTAINERS; } /** * INSANITY TEST */ if (!noinsane) { Insanity insane; insane.userMap.insert(make_pair(Numberz::FIVE, 5)); insane.userMap.insert(make_pair(Numberz::EIGHT, 8)); Xtruct truck; truck.string_thing = "Goodbye4"; truck.byte_thing = 4; truck.i32_thing = 4; truck.i64_thing = 4; Xtruct truck2; truck2.string_thing = "Hello2"; truck2.byte_thing = 2; truck2.i32_thing = 2; truck2.i64_thing = 2; insane.xtructs.push_back(truck); insane.xtructs.push_back(truck2); cout << "testInsanity()" << flush; map<UserId, map<Numberz::type, Insanity> > whoa; testClient.testInsanity(whoa, insane); cout << " = {"; map<UserId, map<Numberz::type, Insanity> >::const_iterator i_iter; for (i_iter = whoa.begin(); i_iter != whoa.end(); ++i_iter) { printf("%" PRId64 " => {", i_iter->first); map<Numberz::type, Insanity>::const_iterator i2_iter; for (i2_iter = i_iter->second.begin(); i2_iter != i_iter->second.end(); ++i2_iter) { printf("%d => {", i2_iter->first); map<Numberz::type, UserId> userMap = i2_iter->second.userMap; map<Numberz::type, UserId>::const_iterator um; cout << "{"; for (um = userMap.begin(); um != userMap.end(); ++um) { cout << um->first << " => " << um->second; } cout << "}, "; vector<Xtruct> xtructs = i2_iter->second.xtructs; vector<Xtruct>::const_iterator x; cout << "{"; for (x = xtructs.begin(); x != xtructs.end(); ++x) { printf("{\"%s\", %d, %d, %" PRId64 "}, ", x->string_thing.c_str(), (int)x->byte_thing, x->i32_thing, x->i64_thing); } cout << "}"; cout << "}, "; } cout << "}, "; } cout << "}" << endl; bool failed = false; map<UserId, map<Numberz::type, Insanity> >::const_iterator it1 = whoa.find(UserId(1)); if (whoa.size() != 2) { failed = true; } if (it1 == whoa.end()) { failed = true; } else { map<Numberz::type, Insanity>::const_iterator it12 = it1->second.find(Numberz::TWO); if (it12 == it1->second.end() || it12->second != insane) { failed = true; } map<Numberz::type, Insanity>::const_iterator it13 = it1->second.find(Numberz::THREE); if (it13 == it1->second.end() || it13->second != insane) { failed = true; } } map<UserId, map<Numberz::type, Insanity> >::const_iterator it2 = whoa.find(UserId(2)); if (it2 == whoa.end()) { failed = true; } else { map<Numberz::type, Insanity>::const_iterator it26 = it2->second.find(Numberz::SIX); if (it26 == it1->second.end() || it26->second != Insanity()) { failed = true; } } if (failed) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } } /** * MULTI TEST */ cout << "testMulti()" << endl; try { map<int16_t, string> mul_map; Xtruct mul_result; mul_map[1] = "blah"; mul_map[2] = "thing"; testClient.testMulti(mul_result, 42, 4242, 424242, mul_map, Numberz::EIGHT, UserId(24)); Xtruct xxs; xxs.string_thing = "Hello2"; xxs.byte_thing = 42; xxs.i32_thing = 4242; xxs.i64_thing = 424242; if (mul_result != xxs) { cout << "*** FAILED ***" << endl; return_code |= ERR_STRUCTS; } } catch (TTransportException&) { throw; } catch (exception& ex) { cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_STRUCTS; } /* test exception */ try { cout << "testClient.testException(\"Xception\") =>" << flush; testClient.testException("Xception"); cout << " void\n*** FAILED ***" << endl; return_code |= ERR_EXCEPTIONS; } catch (Xception& e) { printf(" {%u, \"%s\"}\n", e.errorCode, e.message.c_str()); } try { cout << "testClient.testException(\"TException\") =>" << flush; testClient.testException("TException"); cout << " void\n*** FAILED ***" << endl; return_code |= ERR_EXCEPTIONS; } catch (const TException&) { cout << " Caught TException" << endl; } try { cout << "testClient.testException(\"success\") =>" << flush; testClient.testException("success"); cout << " void" << endl; } catch (exception & ex) { \ cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_EXCEPTIONS; } /* test multi exception */ try { cout << "testClient.testMultiException(\"Xception\", \"test 1\") =>" << flush; Xtruct result; testClient.testMultiException(result, "Xception", "test 1"); cout << " result\n*** FAILED ***" << endl; return_code |= ERR_EXCEPTIONS; } catch (Xception& e) { printf(" {%u, \"%s\"}\n", e.errorCode, e.message.c_str()); } try { cout << "testClient.testMultiException(\"Xception2\", \"test 2\") =>" << flush; Xtruct result; testClient.testMultiException(result, "Xception2", "test 2"); cout << " result\n*** FAILED ***" << endl; return_code |= ERR_EXCEPTIONS; } catch (Xception2& e) { printf(" {%u, {\"%s\"}}\n", e.errorCode, e.struct_thing.string_thing.c_str()); } try { cout << "testClient.testMultiException(\"success\", \"test 3\") =>" << flush; Xtruct result; testClient.testMultiException(result, "success", "test 3"); printf(" {{\"%s\"}}\n", result.string_thing.c_str()); } catch (exception & ex) { \ cout << "*** FAILED ***" << endl << ex.what() << endl; return_code |= ERR_EXCEPTIONS; } /* test oneway void */ { cout << "testClient.testOneway(1) =>" << flush; uint64_t startOneway = now(); testClient.testOneway(1); uint64_t elapsed = now() - startOneway; if (elapsed > 200 * 1000) { // 0.2 seconds printf("*** FAILED *** - took %.2f ms\n", (double)elapsed / 1000.0); return_code |= ERR_BASETYPES; } else { printf(" success - took %.2f ms\n", (double)elapsed / 1000.0); } } /** * redo a simple test after the oneway to make sure we aren't "off by one" -- * if the server treated oneway void like normal void, this next test will * fail since it will get the void confirmation rather than the correct * result. In this circumstance, the client will throw the exception: * * TApplicationException: Wrong method namea */ /** * I32 TEST */ cout << "re-test testI32(-1)"; int i32 = testClient.testI32(-1); cout << " = " << i32 << endl; if (i32 != -1) return_code |= ERR_BASETYPES; uint64_t stop = now(); uint64_t tot = stop - start; cout << "Total time: " << stop - start << " us" << endl; time_tot += tot; if (time_min == 0 || tot < time_min) { time_min = tot; } if (tot > time_max) { time_max = tot; } transport->close(); } cout << endl << "All tests done." << endl; uint64_t time_avg = time_tot / numTests; cout << "Min time: " << time_min << " us" << endl; cout << "Max time: " << time_max << " us" << endl; cout << "Avg time: " << time_avg << " us" << endl; return return_code; }
int main(int argc, char** argv) { string file_path = boost::filesystem::system_complete(argv[0]).string(); string dir_path = file_path.substr(0, file_path.size() - EXECUTABLE_FILE_NAME_LENGTH); #if _WIN32 transport::TWinsockSingleton::create(); #endif string host = "localhost"; int port = 9090; int numTests = 1; bool ssl = false; string transport_type = "buffered"; string protocol_type = "binary"; string domain_socket = ""; bool noinsane = false; boost::program_options::options_description desc("Allowed options"); desc.add_options()("help,h", "produce help message")("host", boost::program_options::value<string>(&host) ->default_value(host), "Host to connect")("port", boost::program_options::value<int>( &port)->default_value(port), "Port number to connect")( "domain-socket", boost::program_options::value<string>(&domain_socket)->default_value(domain_socket), "Domain Socket (e.g. /tmp/ThriftTest.thrift), instead of host and port")( "transport", boost::program_options::value<string>(&transport_type)->default_value(transport_type), "Transport: buffered, framed, http, evhttp")( "protocol", boost::program_options::value<string>(&protocol_type)->default_value(protocol_type), "Protocol: binary, compact, json")("ssl", "Encrypted Transport using SSL")( "testloops,n", boost::program_options::value<int>(&numTests)->default_value(numTests), "Number of Tests")("noinsane", "Do not run insanity test"); boost::program_options::variables_map vm; boost::program_options::store(boost::program_options::parse_command_line(argc, argv, desc), vm); boost::program_options::notify(vm); if (vm.count("help")) { cout << desc << "\n"; return 1; } try { if (!protocol_type.empty()) { if (protocol_type == "binary") { } else if (protocol_type == "compact") { } else if (protocol_type == "json") { } else { throw invalid_argument("Unknown protocol type " + protocol_type); } } if (!transport_type.empty()) { if (transport_type == "buffered") { } else if (transport_type == "framed") { } else if (transport_type == "http") { } else if (transport_type == "evhttp") { } else { throw invalid_argument("Unknown transport type " + transport_type); } } } catch (std::exception& e) { cerr << e.what() << endl; cout << desc << "\n"; return 1; } if (vm.count("ssl")) { ssl = true; } if (vm.count("noinsane")) { noinsane = true; } boost::shared_ptr<TTransport> transport; boost::shared_ptr<TProtocol> protocol; boost::shared_ptr<TSocket> socket; boost::shared_ptr<TSSLSocketFactory> factory; if (ssl) { factory = boost::shared_ptr<TSSLSocketFactory>(new TSSLSocketFactory()); factory->ciphers("ALL:!ADH:!LOW:!EXP:!MD5:@STRENGTH"); factory->loadTrustedCertificates((dir_path + "../keys/CA.pem").c_str()); factory->authenticate(true); socket = factory->createSocket(host, port); } else { if (domain_socket != "") { socket = boost::shared_ptr<TSocket>(new TSocket(domain_socket)); port = 0; } else { socket = boost::shared_ptr<TSocket>(new TSocket(host, port)); } } if (transport_type.compare("http") == 0) { boost::shared_ptr<TTransport> httpSocket(new THttpClient(socket, host, "/service")); transport = httpSocket; } else if (transport_type.compare("framed") == 0) { boost::shared_ptr<TFramedTransport> framedSocket(new TFramedTransport(socket)); transport = framedSocket; } else { boost::shared_ptr<TBufferedTransport> bufferedSocket(new TBufferedTransport(socket)); transport = bufferedSocket; } if (protocol_type.compare("json") == 0) { boost::shared_ptr<TProtocol> jsonProtocol(new TJSONProtocol(transport)); protocol = jsonProtocol; } else if (protocol_type.compare("compact") == 0) { boost::shared_ptr<TProtocol> compactProtocol(new TCompactProtocol(transport)); protocol = compactProtocol; } else { boost::shared_ptr<TBinaryProtocol> binaryProtocol(new TBinaryProtocol(transport)); protocol = binaryProtocol; } // Connection info cout << "Connecting (" << transport_type << "/" << protocol_type << ") to: " << domain_socket; if (port != 0) { cout << host << ":" << port; } cout << endl; if (transport_type.compare("evhttp") == 0) { event_base* base = event_base_new(); cout << "Libevent Version: " << event_get_version() << endl; cout << "Libevent Method: " << event_base_get_method(base) << endl; #if LIBEVENT_VERSION_NUMBER >= 0x02000000 cout << "Libevent Features: 0x" << hex << event_base_get_features(base) << endl; #endif boost::shared_ptr<TProtocolFactory> protocolFactory(new TBinaryProtocolFactory()); boost::shared_ptr<TAsyncChannel> channel( new TEvhttpClientChannel(host.c_str(), "/", host.c_str(), port, base)); ThriftTestCobClient* client = new ThriftTestCobClient(channel, protocolFactory.get()); client->testVoid(tcxx::bind(testVoid_clientReturn, host.c_str(), port, base, protocolFactory.get(), tcxx::placeholders::_1)); event_base_loop(base, 0); return 0; } ThriftTestClient testClient(protocol); uint64_t time_min = 0; uint64_t time_max = 0; uint64_t time_tot = 0; int failCount = 0; int test = 0; for (test = 0; test < numTests; ++test) { try { transport->open(); } catch (TTransportException& ttx) { printf("Connect failed: %s\n", ttx.what()); return 1; } /** * CONNECT TEST */ printf("Test #%d, connect %s:%d\n", test + 1, host.c_str(), port); uint64_t start = now(); /** * VOID TEST */ try { printf("testVoid()"); testClient.testVoid(); printf(" = void\n"); } catch (TApplicationException& tax) { printf("%s\n", tax.what()); failCount++; } /** * STRING TEST */ printf("testString(\"Test\")"); string s; testClient.testString(s, "Test"); printf(" = \"%s\"\n", s.c_str()); if (s != "Test") failCount++; /** * BYTE TEST */ printf("testByte(1)"); uint8_t u8 = testClient.testByte(1); printf(" = %d\n", (int)u8); if (u8 != 1) failCount++; /** * I32 TEST */ printf("testI32(-1)"); int32_t i32 = testClient.testI32(-1); printf(" = %d\n", i32); if (i32 != -1) failCount++; /** * I64 TEST */ printf("testI64(-34359738368)"); int64_t i64 = testClient.testI64(-34359738368LL); printf(" = %" PRId64 "\n", i64); if (i64 != -34359738368LL) failCount++; /** * DOUBLE TEST */ printf("testDouble(-5.2098523)"); double dub = testClient.testDouble(-5.2098523); printf(" = %f\n", dub); if ((dub - (-5.2098523)) > 0.001) failCount++; /** * BINARY TEST */ // TODO: add testBinary() call /** * STRUCT TEST */ printf("testStruct({\"Zero\", 1, -3, -5})"); Xtruct out; out.string_thing = "Zero"; out.byte_thing = 1; out.i32_thing = -3; out.i64_thing = -5; Xtruct in; testClient.testStruct(in, out); printf(" = {\"%s\", %d, %d, %" PRId64 "}\n", in.string_thing.c_str(), (int)in.byte_thing, in.i32_thing, in.i64_thing); if (in != out) failCount++; /** * NESTED STRUCT TEST */ printf("testNest({1, {\"Zero\", 1, -3, -5}), 5}"); Xtruct2 out2; out2.byte_thing = 1; out2.struct_thing = out; out2.i32_thing = 5; Xtruct2 in2; testClient.testNest(in2, out2); in = in2.struct_thing; printf(" = {%d, {\"%s\", %d, %d, %" PRId64 "}, %d}\n", in2.byte_thing, in.string_thing.c_str(), (int)in.byte_thing, in.i32_thing, in.i64_thing, in2.i32_thing); if (in2 != out2) failCount++; /** * MAP TEST */ map<int32_t, int32_t> mapout; for (int32_t i = 0; i < 5; ++i) { mapout.insert(make_pair(i, i - 10)); } printf("testMap({"); map<int32_t, int32_t>::const_iterator m_iter; bool first = true; for (m_iter = mapout.begin(); m_iter != mapout.end(); ++m_iter) { if (first) { first = false; } else { printf(", "); } printf("%d => %d", m_iter->first, m_iter->second); } printf("})"); map<int32_t, int32_t> mapin; testClient.testMap(mapin, mapout); printf(" = {"); first = true; for (m_iter = mapin.begin(); m_iter != mapin.end(); ++m_iter) { if (first) { first = false; } else { printf(", "); } printf("%d => %d", m_iter->first, m_iter->second); } printf("}\n"); if (mapin != mapout) failCount++; /** * STRING MAP TEST * missing */ /** * SET TEST */ set<int32_t> setout; for (int32_t i = -2; i < 3; ++i) { setout.insert(i); } printf("testSet({"); set<int32_t>::const_iterator s_iter; first = true; for (s_iter = setout.begin(); s_iter != setout.end(); ++s_iter) { if (first) { first = false; } else { printf(", "); } printf("%d", *s_iter); } printf("})"); set<int32_t> setin; testClient.testSet(setin, setout); printf(" = {"); first = true; for (s_iter = setin.begin(); s_iter != setin.end(); ++s_iter) { if (first) { first = false; } else { printf(", "); } printf("%d", *s_iter); } printf("}\n"); if (setin != setout) failCount++; /** * LIST TEST */ vector<int32_t> listout; for (int32_t i = -2; i < 3; ++i) { listout.push_back(i); } printf("testList({"); vector<int32_t>::const_iterator l_iter; first = true; for (l_iter = listout.begin(); l_iter != listout.end(); ++l_iter) { if (first) { first = false; } else { printf(", "); } printf("%d", *l_iter); } printf("})"); vector<int32_t> listin; testClient.testList(listin, listout); printf(" = {"); first = true; for (l_iter = listin.begin(); l_iter != listin.end(); ++l_iter) { if (first) { first = false; } else { printf(", "); } printf("%d", *l_iter); } printf("}\n"); if (listin != listout) failCount++; /** * ENUM TEST */ printf("testEnum(ONE)"); Numberz::type ret = testClient.testEnum(Numberz::ONE); printf(" = %d\n", ret); if (ret != Numberz::ONE) failCount++; printf("testEnum(TWO)"); ret = testClient.testEnum(Numberz::TWO); printf(" = %d\n", ret); if (ret != Numberz::TWO) failCount++; printf("testEnum(THREE)"); ret = testClient.testEnum(Numberz::THREE); printf(" = %d\n", ret); if (ret != Numberz::THREE) failCount++; printf("testEnum(FIVE)"); ret = testClient.testEnum(Numberz::FIVE); printf(" = %d\n", ret); if (ret != Numberz::FIVE) failCount++; printf("testEnum(EIGHT)"); ret = testClient.testEnum(Numberz::EIGHT); printf(" = %d\n", ret); if (ret != Numberz::EIGHT) failCount++; /** * TYPEDEF TEST */ printf("testTypedef(309858235082523)"); UserId uid = testClient.testTypedef(309858235082523LL); printf(" = %" PRId64 "\n", uid); if (uid != 309858235082523LL) failCount++; /** * NESTED MAP TEST */ printf("testMapMap(1)"); map<int32_t, map<int32_t, int32_t> > mm; testClient.testMapMap(mm, 1); printf(" = {"); map<int32_t, map<int32_t, int32_t> >::const_iterator mi; for (mi = mm.begin(); mi != mm.end(); ++mi) { printf("%d => {", mi->first); map<int32_t, int32_t>::const_iterator mi2; for (mi2 = mi->second.begin(); mi2 != mi->second.end(); ++mi2) { printf("%d => %d, ", mi2->first, mi2->second); } printf("}, "); } printf("}\n"); /** * INSANITY TEST */ if (!noinsane) { Insanity insane; insane.userMap.insert(make_pair(Numberz::FIVE, 5000)); Xtruct truck; truck.string_thing = "Truck"; truck.byte_thing = 8; truck.i32_thing = 8; truck.i64_thing = 8; insane.xtructs.push_back(truck); printf("testInsanity()"); map<UserId, map<Numberz::type, Insanity> > whoa; testClient.testInsanity(whoa, insane); printf(" = {"); map<UserId, map<Numberz::type, Insanity> >::const_iterator i_iter; for (i_iter = whoa.begin(); i_iter != whoa.end(); ++i_iter) { printf("%" PRId64 " => {", i_iter->first); map<Numberz::type, Insanity>::const_iterator i2_iter; for (i2_iter = i_iter->second.begin(); i2_iter != i_iter->second.end(); ++i2_iter) { printf("%d => {", i2_iter->first); map<Numberz::type, UserId> userMap = i2_iter->second.userMap; map<Numberz::type, UserId>::const_iterator um; printf("{"); for (um = userMap.begin(); um != userMap.end(); ++um) { printf("%d => %" PRId64 ", ", um->first, um->second); } printf("}, "); vector<Xtruct> xtructs = i2_iter->second.xtructs; vector<Xtruct>::const_iterator x; printf("{"); for (x = xtructs.begin(); x != xtructs.end(); ++x) { printf("{\"%s\", %d, %d, %" PRId64 "}, ", x->string_thing.c_str(), (int)x->byte_thing, x->i32_thing, x->i64_thing); } printf("}"); printf("}, "); } printf("}, "); } printf("}\n"); } /* test exception */ try { printf("testClient.testException(\"Xception\") =>"); testClient.testException("Xception"); printf(" void\nFAILURE\n"); failCount++; } catch (Xception& e) { printf(" {%u, \"%s\"}\n", e.errorCode, e.message.c_str()); } try { printf("testClient.testException(\"TException\") =>"); testClient.testException("TException"); printf(" void\nFAILURE\n"); failCount++; } catch (const TException&) { printf(" Caught TException\n"); } try { printf("testClient.testException(\"success\") =>"); testClient.testException("success"); printf(" void\n"); } catch (...) { printf(" exception\nFAILURE\n"); failCount++; } /* test multi exception */ try { printf("testClient.testMultiException(\"Xception\", \"test 1\") =>"); Xtruct result; testClient.testMultiException(result, "Xception", "test 1"); printf(" result\nFAILURE\n"); failCount++; } catch (Xception& e) { printf(" {%u, \"%s\"}\n", e.errorCode, e.message.c_str()); } try { printf("testClient.testMultiException(\"Xception2\", \"test 2\") =>"); Xtruct result; testClient.testMultiException(result, "Xception2", "test 2"); printf(" result\nFAILURE\n"); failCount++; } catch (Xception2& e) { printf(" {%u, {\"%s\"}}\n", e.errorCode, e.struct_thing.string_thing.c_str()); } try { printf("testClient.testMultiException(\"success\", \"test 3\") =>"); Xtruct result; testClient.testMultiException(result, "success", "test 3"); printf(" {{\"%s\"}}\n", result.string_thing.c_str()); } catch (...) { printf(" exception\nFAILURE\n"); failCount++; } /* test oneway void */ { printf("testClient.testOneway(1) =>"); uint64_t startOneway = now(); testClient.testOneway(1); uint64_t elapsed = now() - startOneway; if (elapsed > 200 * 1000) { // 0.2 seconds printf(" FAILURE - took %.2f ms\n", (double)elapsed / 1000.0); failCount++; } else { printf(" success - took %.2f ms\n", (double)elapsed / 1000.0); } } /** * redo a simple test after the oneway to make sure we aren't "off by one" -- * if the server treated oneway void like normal void, this next test will * fail since it will get the void confirmation rather than the correct * result. In this circumstance, the client will throw the exception: * * TApplicationException: Wrong method namea */ /** * I32 TEST */ printf("re-test testI32(-1)"); i32 = testClient.testI32(-1); printf(" = %d\n", i32); if (i32 != -1) failCount++; uint64_t stop = now(); uint64_t tot = stop - start; printf("Total time: %" PRIu64 " us\n", stop - start); time_tot += tot; if (time_min == 0 || tot < time_min) { time_min = tot; } if (tot > time_max) { time_max = tot; } transport->close(); } printf("\nAll tests done.\n"); uint64_t time_avg = time_tot / numTests; printf("Min time: %" PRIu64 " us\n", time_min); printf("Max time: %" PRIu64 " us\n", time_max); printf("Avg time: %" PRIu64 " us\n", time_avg); return failCount; }