diff options
Diffstat (limited to 'src/jaegertracing/thrift/test/rb/integration/TestClient.rb')
| -rwxr-xr-x | src/jaegertracing/thrift/test/rb/integration/TestClient.rb | 381 |
1 files changed, 381 insertions, 0 deletions
diff --git a/src/jaegertracing/thrift/test/rb/integration/TestClient.rb b/src/jaegertracing/thrift/test/rb/integration/TestClient.rb new file mode 100755 index 000000000..639aca99e --- /dev/null +++ b/src/jaegertracing/thrift/test/rb/integration/TestClient.rb @@ -0,0 +1,381 @@ +#!/usr/bin/env ruby +# encoding: utf-8 + +# +# 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. +# + +$:.push File.dirname(__FILE__) + '/..' + +require 'test_helper' +require 'thrift' +require 'thrift_test' + +$domain_socket = nil +$host = "localhost" +$port = 9090 +$protocolType = "binary" +$ssl = false +$transport = "buffered" + +ARGV.each do|a| + if a == "--help" + puts "Allowed options:" + puts "\t -h [ --help ] \t produce help message" + puts "\t--domain-socket arg (=) \t Unix domain socket path" + puts "\t--host arg (=localhost) \t Host to connect \t not valid with domain-socket" + puts "\t--port arg (=9090) \t Port number to listen \t not valid with domain-socket" + puts "\t--protocol arg (=binary) \t protocol: accel, binary, compact, json" + puts "\t--ssl \t use ssl \t not valid with domain-socket" + puts "\t--transport arg (=buffered) transport: buffered, framed, http" + exit + elsif a.start_with?("--domain-socket") + $domain_socket = a.split("=")[1] + elsif a.start_with?("--host") + $host = a.split("=")[1] + elsif a.start_with?("--protocol") + $protocolType = a.split("=")[1] + elsif a == "--ssl" + $ssl = true + elsif a.start_with?("--transport") + $transport = a.split("=")[1] + elsif a.start_with?("--port") + $port = a.split("=")[1].to_i + end +end + +class SimpleClientTest < Test::Unit::TestCase + def setup + unless @socket + if $domain_socket.to_s.strip.empty? + if $ssl + # the working directory for ruby crosstest is test/rb/gen-rb + keysDir = File.join(File.dirname(File.dirname(Dir.pwd)), "keys") + ctx = OpenSSL::SSL::SSLContext.new + ctx.ca_file = File.join(keysDir, "CA.pem") + ctx.cert = OpenSSL::X509::Certificate.new(File.open(File.join(keysDir, "client.crt"))) + ctx.cert_store = OpenSSL::X509::Store.new + ctx.cert_store.add_file(File.join(keysDir, 'server.pem')) + ctx.key = OpenSSL::PKey::RSA.new(File.open(File.join(keysDir, "client.key"))) + ctx.options = OpenSSL::SSL::OP_NO_SSLv2 | OpenSSL::SSL::OP_NO_SSLv3 + ctx.ssl_version = :SSLv23 + ctx.verify_mode = OpenSSL::SSL::VERIFY_PEER + @socket = Thrift::SSLSocket.new($host, $port, nil, ctx) + else + @socket = Thrift::Socket.new($host, $port) + end + else + @socket = Thrift::UNIXSocket.new($domain_socket) + end + + if $transport == "buffered" + transportFactory = Thrift::BufferedTransport.new(@socket) + elsif $transport == "framed" + transportFactory = Thrift::FramedTransport.new(@socket) + else + raise 'Unknown transport type' + end + + if $protocolType == "binary" + @protocol = Thrift::BinaryProtocol.new(transportFactory) + elsif $protocolType == "compact" + @protocol = Thrift::CompactProtocol.new(transportFactory) + elsif $protocolType == "json" + @protocol = Thrift::JsonProtocol.new(transportFactory) + elsif $protocolType == "accel" + @protocol = Thrift::BinaryProtocolAccelerated.new(transportFactory) + else + raise 'Unknown protocol type' + end + @client = Thrift::Test::ThriftTest::Client.new(@protocol) + @socket.open + end + end + + def teardown + @socket.close + end + + def test_void + p 'test_void' + @client.testVoid() + end + + def test_string + p 'test_string' + test_string = + '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: ]] \"]] \\" }}}{ [[[ ' + test_string = "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ú, 粵語" + + result_string = @client.testString(test_string) + assert_equal(test_string, result_string.force_encoding(Encoding::UTF_8)) + end + + def test_bool + p 'test_bool' + assert_equal(@client.testBool(true), true) + assert_equal(@client.testBool(false), false) + end + + def test_byte + p 'test_byte' + val = 120 + assert_equal(@client.testByte(val), val) + assert_equal(@client.testByte(-val), -val) + end + + def test_i32 + p 'test_i32' + val = 2000000032 + assert_equal(@client.testI32(val), val) + assert_equal(@client.testI32(-val), -val) + end + + def test_i64 + p 'test_i64' + val = 9000000000000000064 + assert_equal(@client.testI64(val), val) + assert_equal(@client.testI64(-val), -val) + end + + def test_double + p 'test_double' + val = 3.14159265358979323846 + assert_equal(@client.testDouble(val), val) + assert_equal(@client.testDouble(-val), -val) + assert_kind_of(Float, @client.testDouble(val)) + end + + def test_binary + p 'test_binary' + val = (0...256).reverse_each.to_a + ret = @client.testBinary(val.pack('C*')) + assert_equal(val, ret.bytes.to_a) + end + + def test_map + p 'test_map' + val = {1 => 1, 2 => 2, 3 => 3} + assert_equal(@client.testMap(val), val) + assert_kind_of(Hash, @client.testMap(val)) + end + + def test_string_map + p 'test_string_map' + val = {'a' => '2', 'b' => 'blah', 'some' => 'thing'} + ret = @client.testStringMap(val) + assert_equal(val, ret) + assert_kind_of(Hash, ret) + end + + def test_list + p 'test_list' + val = [1,2,3,4,5] + assert_equal(@client.testList(val), val) + assert_kind_of(Array, @client.testList(val)) + end + + def test_enum + p 'test_enum' + val = Thrift::Test::Numberz::SIX + ret = @client.testEnum(val) + + assert_equal(ret, 6) + assert_kind_of(Fixnum, ret) + end + + def test_typedef + p 'test_typedef' + #UserId testTypedef(1: UserId thing), + assert_equal(@client.testTypedef(309858235082523), 309858235082523) + assert_kind_of(Fixnum, @client.testTypedef(309858235082523)) + true + end + + def test_set + p 'test_set' + val = Set.new([1,2,3]) + assert_equal(@client.testSet(val), val) + assert_kind_of(Set, @client.testSet(val)) + end + + def get_struct + Thrift::Test::Xtruct.new({'string_thing' => 'hi!', 'i32_thing' => 4 }) + end + + def test_struct + p 'test_struct' + ret = @client.testStruct(get_struct) + + # TODO: not sure what unspecified "default" requiredness values should be + assert(ret.byte_thing == nil || ret.byte_thing == 0) + assert(ret.i64_thing == nil || ret.i64_thing == 0) + + assert_equal(ret.string_thing, 'hi!') + assert_equal(ret.i32_thing, 4) + assert_kind_of(Thrift::Test::Xtruct, ret) + end + + def test_nest + p 'test_nest' + struct2 = Thrift::Test::Xtruct2.new({'struct_thing' => get_struct, 'i32_thing' => 10}) + + ret = @client.testNest(struct2) + + # TODO: not sure what unspecified "default" requiredness values should be + assert(ret.struct_thing.byte_thing == nil || ret.struct_thing.byte_thing == 0) + assert(ret.struct_thing.i64_thing == nil || ret.struct_thing.i64_thing == 0) + + assert_equal(ret.struct_thing.string_thing, 'hi!') + assert_equal(ret.struct_thing.i32_thing, 4) + assert_equal(ret.i32_thing, 10) + + assert_kind_of(Thrift::Test::Xtruct, ret.struct_thing) + assert_kind_of(Thrift::Test::Xtruct2, ret) + end + + def test_insanity + p 'test_insanity' + insane = Thrift::Test::Insanity.new({ + 'userMap' => { + Thrift::Test::Numberz::FIVE => 5, + Thrift::Test::Numberz::EIGHT => 8, + }, + 'xtructs' => [ + Thrift::Test::Xtruct.new({ + 'string_thing' => 'Goodbye4', + 'byte_thing' => 4, + 'i32_thing' => 4, + 'i64_thing' => 4, + }), + Thrift::Test::Xtruct.new({ + 'string_thing' => 'Hello2', + 'byte_thing' => 2, + 'i32_thing' => 2, + 'i64_thing' => 2, + }) + ] + }) + + ret = @client.testInsanity(insane) + + assert_equal(insane, ret[1][2]) + assert_equal(insane, ret[1][3]) + + assert(ret[2][6].userMap == nil || ret[2][6].userMap.length == 0) + assert(ret[2][6].xtructs == nil || ret[2][6].xtructs.length == 0) + end + + def test_map_map + p 'test_map_map' + ret = @client.testMapMap(4) + assert_kind_of(Hash, ret) + expected = { + -4 => { + -4 => -4, + -3 => -3, + -2 => -2, + -1 => -1, + }, + 4 => { + 4 => 4, + 3 => 3, + 2 => 2, + 1 => 1, + } + } + assert_equal(expected, ret) + end + + def test_multi + p 'test_multi' + ret = @client.testMulti(42, 4242, 424242, {1 => 'blah', 2 => 'thing'}, Thrift::Test::Numberz::EIGHT, 24) + expected = Thrift::Test::Xtruct.new({ + :string_thing => 'Hello2', + :byte_thing => 42, + :i32_thing => 4242, + :i64_thing => 424242 + }) + assert_equal(expected, ret) + end + + def test_exception + p 'test_exception' + assert_raise Thrift::Test::Xception do + @client.testException('Xception') + end + begin + @client.testException('TException') + rescue => e + assert e.class.ancestors.include?(Thrift::Exception) + end + assert_nothing_raised do + @client.testException('test') + end + end + + def test_multi_exception + p 'test_multi_exception' + assert_raise Thrift::Test::Xception do + @client.testMultiException("Xception", "test 1") + end + assert_raise Thrift::Test::Xception2 do + @client.testMultiException("Xception2", "test 2") + end + assert_equal( @client.testMultiException("Success", "test 3").string_thing, "test 3") + end + + def test_oneway + p 'test_oneway' + time1 = Time.now.to_f + @client.testOneway(1) + time2 = Time.now.to_f + assert_operator (time2-time1), :<, 0.1 + end + +end + |