thrift-client 0.0.7

data/test/client_test.rb

@@ -0,0 +1,172 @@
+$:.unshift File.expand_path("../../lib", __FILE__)
+$:.unshift File.expand_path("../../test", __FILE__)
+
+require "test/unit"
+require "thrift_client"
+require "foobar_service"
+require 'foobar_types'
+require 'thrift'
+require 'yaml'
+
+class ClientTest < Test::Unit::TestCase
+  include Test::Unit::Assertions
+  def setup
+    super
+    root = File.expand_path("../..", __FILE__)
+    @config = YAML.load_file(File.join(root, 'test/thrift_config.yaml'))
+    puts "setup"
+  end
+ 
+  def test_connection_close
+    puts "test_connection_close start"
+    client = ThriftClient.new(@config['thrift_test'])
+
+    assert_nothing_raised do client.ping
+    end
+
+    client.destroy
+
+    assert_nothing_raised do client.ping
+    end
+    bar = client.getBar
+    assert_not_nil(bar)
+
+    client.destroy
+    
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = 2
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert(client.putFoo(foo))
+    
+    client.destroy
+    
+    assert_nothing_raised do client.ping
+    end
+
+    puts "test_connection_close stop"
+  end
+ 
+  def test_application_error
+    client = ThriftClient.new(@config['thrift_test'])
+
+    client.ping
+    assert_raise do client.getError end
+
+    bar = client.getBar
+    assert_not_nil(bar)
+    
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = 2
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert(client.putFoo(foo))
+  end
+
+  def test_read_timeout
+    puts "test_read_timeout start"
+    client = ThriftClient.new(@config['thrift_test'])
+    assert_raise do client.getTimeout end
+
+    assert_nothing_raised do client.ping end
+
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = 2
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert(client.putFoo(foo))
+
+    assert_nothing_raised do client.ping end
+    assert_nothing_raised do bar = client.getBar end
+    bar = client.getBar
+    assert_not_nil(bar)
+    puts "test_read_timeout stop"
+  end
+
+  def test_server_restart
+    puts "test_server_restart start"
+    client = ThriftClient.new(@config['thrift_test'])
+    client.ping
+
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = 2
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert(client.putFoo(foo))
+    client.ping
+    puts "restart server!!!!!!!!!!"
+    sleep(5)
+    assert_raise do bar = client.getBar end
+    bar = client.getBar
+    assert_not_nil(bar)
+    puts "test_server_restart stop"
+  end
+
+  def test_server_graceful_restart
+    puts "test_server_graceful_restart start"
+
+    client = ThriftClient.new(@config['thrift_test'])
+    client.ping
+
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = 2
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert(client.putFoo(foo))
+    client.ping
+    puts "restart server!!!!!!!!!!"
+    sleep(5)
+    assert_raise do bar = client.getBar end
+    bar = client.getBar
+    assert_not_nil(bar)
+
+    puts "test_server_graceful_restart stop"
+  end
+
+  def test_client_parameter_error
+    puts "test_client_parameter_error start"
+    client = ThriftClient.new(@config['thrift_test'])
+    
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = "2"
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert_raise do client.putFoo(foo) end 
+
+    bar = client.getBar
+    assert_not_nil(bar)
+    puts "test_client_parameter_error stop"
+  end
+
+  def test_add_callback
+    puts "test_add_callback start"
+    client = ThriftClient.new(@config['thrift_test'])
+    
+    client.add_callback(:before_method) { |method_name|
+        puts "call method #{method_name}"
+    }
+
+    foo = Foo::Foo.new
+    foo.first = 1
+    foo.second = "2"
+    foo.third = 3.0
+    foo.fourth = "4"
+    assert_raise do client.putFoo(foo) end 
+
+    bar = client.getBar
+    assert_not_nil(bar)
+
+    puts "test_add_callback stop"
+  end
+
+  def teardown
+    super
+    puts "teardown"
+  end
+end

data/test/foobar_constants.rb

@@ -0,0 +1,11 @@
+#
+# Autogenerated by Thrift Compiler (0.9.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'foobar_types'
+
+module Foo
+end

data/test/foobar_service.rb

@@ -0,0 +1,282 @@
+#
+# Autogenerated by Thrift Compiler (0.9.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'foobar_types'
+
+module Foo
+  module FoobarService
+    class Client
+      include ::Thrift::Client
+
+      def ping()
+        send_ping()
+      end
+
+      def send_ping()
+        send_message('ping', Ping_args)
+      end
+      def getBar()
+        send_getBar()
+        return recv_getBar()
+      end
+
+      def send_getBar()
+        send_message('getBar', GetBar_args)
+      end
+
+      def recv_getBar()
+        result = receive_message(GetBar_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'getBar failed: unknown result')
+      end
+
+      def putFoo(foo)
+        send_putFoo(foo)
+        return recv_putFoo()
+      end
+
+      def send_putFoo(foo)
+        send_message('putFoo', PutFoo_args, :foo => foo)
+      end
+
+      def recv_putFoo()
+        result = receive_message(PutFoo_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'putFoo failed: unknown result')
+      end
+
+      def getTimeout()
+        send_getTimeout()
+        return recv_getTimeout()
+      end
+
+      def send_getTimeout()
+        send_message('getTimeout', GetTimeout_args)
+      end
+
+      def recv_getTimeout()
+        result = receive_message(GetTimeout_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'getTimeout failed: unknown result')
+      end
+
+      def getError()
+        send_getError()
+        return recv_getError()
+      end
+
+      def send_getError()
+        send_message('getError', GetError_args)
+      end
+
+      def recv_getError()
+        result = receive_message(GetError_result)
+        return result.success unless result.success.nil?
+        raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'getError failed: unknown result')
+      end
+
+    end
+
+    class Processor
+      include ::Thrift::Processor
+
+      def process_ping(seqid, iprot, oprot)
+        args = read_args(iprot, Ping_args)
+        @handler.ping()
+        return
+      end
+
+      def process_getBar(seqid, iprot, oprot)
+        args = read_args(iprot, GetBar_args)
+        result = GetBar_result.new()
+        result.success = @handler.getBar()
+        write_result(result, oprot, 'getBar', seqid)
+      end
+
+      def process_putFoo(seqid, iprot, oprot)
+        args = read_args(iprot, PutFoo_args)
+        result = PutFoo_result.new()
+        result.success = @handler.putFoo(args.foo)
+        write_result(result, oprot, 'putFoo', seqid)
+      end
+
+      def process_getTimeout(seqid, iprot, oprot)
+        args = read_args(iprot, GetTimeout_args)
+        result = GetTimeout_result.new()
+        result.success = @handler.getTimeout()
+        write_result(result, oprot, 'getTimeout', seqid)
+      end
+
+      def process_getError(seqid, iprot, oprot)
+        args = read_args(iprot, GetError_args)
+        result = GetError_result.new()
+        result.success = @handler.getError()
+        write_result(result, oprot, 'getError', seqid)
+      end
+
+    end
+
+    # HELPER FUNCTIONS AND STRUCTURES
+
+    class Ping_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+
+      FIELDS = {
+
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class Ping_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+
+      FIELDS = {
+
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetBar_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+
+      FIELDS = {
+
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetBar_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => ::Foo::Bar}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class PutFoo_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      FOO = 1
+
+      FIELDS = {
+        FOO => {:type => ::Thrift::Types::STRUCT, :name => 'foo', :class => ::Foo::Foo}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class PutFoo_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetTimeout_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+
+      FIELDS = {
+
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetTimeout_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetError_args
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+
+      FIELDS = {
+
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class GetError_result
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SUCCESS = 0
+
+      FIELDS = {
+        SUCCESS => {:type => ::Thrift::Types::BOOL, :name => 'success'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end
+
+end