6ab5c9334d
* Changing default timeout for start maintenance. * Upgrading dependencies to gorealis v2 and thrift 0.12.0 * Refactored to update to gorealis v2.
381 lines
12 KiB
Ruby
Executable file
381 lines
12 KiB
Ruby
Executable file
#!/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
|
||
|