Supplement Jsonprotocol module testcase
Client: rb
Patch: Zezeng Wang
This closes #2223
diff --git a/lib/rb/test/TestJsonProtocol.rb b/lib/rb/test/TestJsonProtocol.rb
new file mode 100644
index 0000000..562aad5
--- /dev/null
+++ b/lib/rb/test/TestJsonProtocol.rb
@@ -0,0 +1,208 @@
+#!/usr/bin/env ruby
+#
+# 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.
+#
+
+require 'thrift'
+require 'test/unit'
+
+class TestJsonProtocol < Test::Unit::TestCase
+ def test_defferent_data_types
+ begin
+ port = ARGV[0] || 7070
+ pid = Process.fork
+ if pid.nil?then
+ buf = String.new('1234567')
+ socket = Thrift::ServerSocket.new(port)
+ socket.listen()
+ client = socket.accept()
+ transport = Thrift::BufferedTransport.new(client)
+ protocol = Thrift::JsonProtocol.new(transport)
+
+ acc_json_string = protocol.read_json_string()
+ assert_equal('hello_world123!@#$%',acc_json_string)
+
+ acc_json_base64 = protocol.read_json_base64()
+ assert_equal('hello_world12233!@#$%',acc_json_base64)
+
+ protocol.read_json_array_start()
+ acc_json_integer = protocol.read_json_integer()
+ protocol.read_json_array_end()
+ assert_equal(2553369689,acc_json_integer)
+
+
+ protocol.read_json_array_start()
+ acc_json_double = protocol.read_json_double()
+ protocol.read_json_array_end()
+ assert_equal(3.1415926,acc_json_double)
+
+ acc_message = protocol.read_message_begin()
+ protocol.read_message_end()
+ assert_equal("[\"hello_world\", 155510, 102020]","#{acc_message}")
+
+ acc_field = protocol.read_field_begin()
+ protocol.read_field_end()
+ assert_equal([nil ,6 ,12],acc_field)
+
+ acc_map = protocol.read_map_begin()
+ protocol.read_map_end()
+ assert_equal([10, 8, 20],acc_map)
+
+ acc_list = protocol.read_list_begin()
+ protocol.read_list_end()
+ assert_equal([6, 10],acc_list)
+
+ acc_set = protocol.read_set_begin()
+ protocol.read_set_end()
+ assert_equal([8, 20],acc_set)
+
+ protocol.read_json_object_start()
+ acc_bool = protocol.read_bool()
+ protocol.read_json_object_end()
+ assert_equal(true,acc_bool)
+
+ protocol.read_json_object_start()
+ acc_byte = protocol.read_byte()
+ protocol.read_json_object_end()
+ assert_equal(123,acc_byte)
+
+ protocol.read_json_object_start()
+ acc_i16 = protocol.read_i16()
+ protocol.read_json_object_end()
+ assert_equal(4203,acc_i16)
+
+
+ protocol.read_json_object_start()
+ acc_i32 = protocol.read_i32()
+ protocol.read_json_object_end()
+ assert_equal(2000000032,acc_i32)
+
+ protocol.read_json_object_start()
+ acc_i64 = protocol.read_i64()
+ protocol.read_json_object_end()
+ assert_equal(1844674407370955161,acc_i64)
+
+ protocol.read_json_object_start()
+ acc_double = protocol.read_double()
+ protocol.read_json_object_end()
+ assert_equal(3.1415926,acc_double)
+
+ protocol.read_json_object_start()
+ acc_string = protocol.read_string()
+ protocol.read_json_object_end()
+ assert_equal("sello_world123456789!@#$%",acc_string)
+
+ protocol.read_json_object_start()
+ acc_binary = protocol.read_binary()
+ ret =acc_binary.bytes.to_a
+ protocol.read_json_object_end()
+ assert_equal((0...256).reverse_each.to_a,ret)
+
+ exit(0)
+ else
+ sleep(2)
+ socket = Thrift::Socket.new('localhost',port)
+ transport = Thrift::BufferedTransport.new(socket)
+ transport.open()
+ protocol = Thrift::JsonProtocol.new(transport)
+
+ protocol.write_json_string('hello_world123!@#$%')
+ transport.flush()
+
+ protocol.write_json_base64('hello_world12233!@#$%')
+ transport.flush()
+
+ protocol.write_json_array_start()
+ protocol.write_json_integer(2553369689)
+ protocol.write_json_array_end()
+ transport.flush()
+
+ protocol.write_json_array_start()
+ protocol.write_json_double(3.1415926)
+ protocol.write_json_array_end()
+ transport.flush()
+
+ protocol.write_message_begin('hello_world',155510,102020)
+ protocol.write_message_end()
+ transport.flush()
+
+ protocol.write_field_begin("hello_world",Thrift::Types::I16,12)
+ protocol.write_field_end()
+ transport.flush()
+
+ protocol.write_map_begin(Thrift::Types::I64,Thrift::Types::I32,20)
+ protocol.write_map_end()
+ transport.flush()
+
+ protocol.write_list_begin(Thrift::Types::I16,10)
+ protocol.write_list_end()
+ transport.flush()
+
+ protocol.write_set_begin(Thrift::Types::I32,20)
+ protocol.write_set_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ protocol.write_bool(true)
+ protocol.write_json_object_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ protocol.write_byte(123)
+ protocol.write_json_object_end()
+ transport.flush()
+
+
+ protocol.write_json_object_start()
+ protocol.write_i16(4203)
+ protocol.write_json_object_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ protocol.write_i32(2000000032)
+ protocol.write_json_object_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ protocol.write_i64(1844674407370955161)
+ protocol.write_json_object_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ protocol.write_double(3.1415926)
+ protocol.write_json_object_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ protocol.write_string("sello_world123456789!@#$%")
+ protocol.write_json_object_end()
+ transport.flush()
+
+ protocol.write_json_object_start()
+ val = (0...256).reverse_each.to_a
+ protocol.write_binary(val.pack('C*'))
+ protocol.write_json_object_end()
+ transport.flush()
+
+
+ transport.close()
+ Process.wait(pid)
+ end
+ end
+ end
+end