stark 0.5.0

data/test/legacy_profile/user_storage.rb

@@ -0,0 +1,451 @@
+#
+# Autogenerated by Thrift Compiler (0.8.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift'
+require 'profile_types'
+
+module UserStorage
+  class Client
+    include ::Thrift::Client
+
+    def store(xuser)
+      send_store(xuser)
+      recv_store()
+    end
+
+    def send_store(xuser)
+      send_message('store', Store_args, :xuser => xuser)
+    end
+
+    def recv_store()
+      result = receive_message(Store_result)
+      return
+    end
+
+    def retrieve(xuid)
+      send_retrieve(xuid)
+      return recv_retrieve()
+    end
+
+    def send_retrieve(xuid)
+      send_message('retrieve', Retrieve_args, :xuid => xuid)
+    end
+
+    def recv_retrieve()
+      result = receive_message(Retrieve_result)
+      return result.success unless result.success.nil?
+      raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'retrieve failed: unknown result')
+    end
+
+    def set_map(m)
+      send_set_map(m)
+      recv_set_map()
+    end
+
+    def send_set_map(m)
+      send_message('set_map', Set_map_args, :m => m)
+    end
+
+    def recv_set_map()
+      result = receive_message(Set_map_result)
+      return
+    end
+
+    def last_map()
+      send_last_map()
+      return recv_last_map()
+    end
+
+    def send_last_map()
+      send_message('last_map', Last_map_args)
+    end
+
+    def recv_last_map()
+      result = receive_message(Last_map_result)
+      return result.success unless result.success.nil?
+      raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'last_map failed: unknown result')
+    end
+
+    def set_list(l)
+      send_set_list(l)
+      recv_set_list()
+    end
+
+    def send_set_list(l)
+      send_message('set_list', Set_list_args, :l => l)
+    end
+
+    def recv_set_list()
+      result = receive_message(Set_list_result)
+      return
+    end
+
+    def last_list()
+      send_last_list()
+      return recv_last_list()
+    end
+
+    def send_last_list()
+      send_message('last_list', Last_list_args)
+    end
+
+    def recv_last_list()
+      result = receive_message(Last_list_result)
+      return result.success unless result.success.nil?
+      raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'last_list failed: unknown result')
+    end
+
+    def set_status(s)
+      send_set_status(s)
+      recv_set_status()
+    end
+
+    def send_set_status(s)
+      send_message('set_status', Set_status_args, :s => s)
+    end
+
+    def recv_set_status()
+      result = receive_message(Set_status_result)
+      return
+    end
+
+    def last_status()
+      send_last_status()
+      return recv_last_status()
+    end
+
+    def send_last_status()
+      send_message('last_status', Last_status_args)
+    end
+
+    def recv_last_status()
+      result = receive_message(Last_status_result)
+      return result.success unless result.success.nil?
+      raise ::Thrift::ApplicationException.new(::Thrift::ApplicationException::MISSING_RESULT, 'last_status failed: unknown result')
+    end
+
+  end
+
+  class Processor
+    include ::Thrift::Processor
+
+    def process_store(seqid, iprot, oprot)
+      args = read_args(iprot, Store_args)
+      result = Store_result.new()
+      @handler.store(args.xuser)
+      write_result(result, oprot, 'store', seqid)
+    end
+
+    def process_retrieve(seqid, iprot, oprot)
+      args = read_args(iprot, Retrieve_args)
+      result = Retrieve_result.new()
+      result.success = @handler.retrieve(args.xuid)
+      write_result(result, oprot, 'retrieve', seqid)
+    end
+
+    def process_set_map(seqid, iprot, oprot)
+      args = read_args(iprot, Set_map_args)
+      result = Set_map_result.new()
+      @handler.set_map(args.m)
+      write_result(result, oprot, 'set_map', seqid)
+    end
+
+    def process_last_map(seqid, iprot, oprot)
+      args = read_args(iprot, Last_map_args)
+      result = Last_map_result.new()
+      result.success = @handler.last_map()
+      write_result(result, oprot, 'last_map', seqid)
+    end
+
+    def process_set_list(seqid, iprot, oprot)
+      args = read_args(iprot, Set_list_args)
+      result = Set_list_result.new()
+      @handler.set_list(args.l)
+      write_result(result, oprot, 'set_list', seqid)
+    end
+
+    def process_last_list(seqid, iprot, oprot)
+      args = read_args(iprot, Last_list_args)
+      result = Last_list_result.new()
+      result.success = @handler.last_list()
+      write_result(result, oprot, 'last_list', seqid)
+    end
+
+    def process_set_status(seqid, iprot, oprot)
+      args = read_args(iprot, Set_status_args)
+      result = Set_status_result.new()
+      @handler.set_status(args.s)
+      write_result(result, oprot, 'set_status', seqid)
+    end
+
+    def process_last_status(seqid, iprot, oprot)
+      args = read_args(iprot, Last_status_args)
+      result = Last_status_result.new()
+      result.success = @handler.last_status()
+      write_result(result, oprot, 'last_status', seqid)
+    end
+
+  end
+
+  # HELPER FUNCTIONS AND STRUCTURES
+
+  class Store_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    XUSER = 1
+
+    FIELDS = {
+      XUSER => {:type => ::Thrift::Types::STRUCT, :name => 'xuser', :class => UserProfile}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Store_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Retrieve_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    XUID = 1
+
+    FIELDS = {
+      XUID => {:type => ::Thrift::Types::I32, :name => 'xuid'}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Retrieve_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    SUCCESS = 0
+
+    FIELDS = {
+      SUCCESS => {:type => ::Thrift::Types::STRUCT, :name => 'success', :class => UserProfile}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Set_map_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    M = 1
+
+    FIELDS = {
+      M => {:type => ::Thrift::Types::MAP, :name => 'm', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Set_map_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Last_map_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Last_map_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    SUCCESS = 0
+
+    FIELDS = {
+      SUCCESS => {:type => ::Thrift::Types::MAP, :name => 'success', :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::STRING}}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Set_list_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    L = 1
+
+    FIELDS = {
+      L => {:type => ::Thrift::Types::LIST, :name => 'l', :element => {:type => ::Thrift::Types::STRING}}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Set_list_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Last_list_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Last_list_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    SUCCESS = 0
+
+    FIELDS = {
+      SUCCESS => {:type => ::Thrift::Types::LIST, :name => 'success', :element => {:type => ::Thrift::Types::STRING}}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Set_status_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    S = 1
+
+    FIELDS = {
+      S => {:type => ::Thrift::Types::I32, :name => 's', :enum_class => Status}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+      unless @s.nil? || Status::VALID_VALUES.include?(@s)
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field s!')
+      end
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Set_status_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Last_status_args
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+
+    FIELDS = {
+
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+  class Last_status_result
+    include ::Thrift::Struct, ::Thrift::Struct_Union
+    SUCCESS = 0
+
+    FIELDS = {
+      SUCCESS => {:type => ::Thrift::Types::I32, :name => 'success', :enum_class => Status}
+    }
+
+    def struct_fields; FIELDS; end
+
+    def validate
+      unless @success.nil? || Status::VALID_VALUES.include?(@success)
+        raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field success!')
+      end
+    end
+
+    ::Thrift::Struct.generate_accessors self
+  end
+
+end
+

data/test/profile.thrift

@@ -0,0 +1,24 @@
+struct UserProfile {
+  1: i32 uid,
+  2: string name,
+  3: string blurb
+}
+
+enum Status {
+  ON
+  OFF
+  DEAD
+  ALIVE
+}
+
+service UserStorage {
+  void store(1: UserProfile xuser),
+  UserProfile retrieve(1: i32 xuid),
+  void set_map(1: map<string, string> m),
+  map<string, string> last_map(),
+  void set_list(1: list<string> l),
+  list<string> last_list(),
+  void set_status(1: Status s),
+  Status last_status()
+}
+

data/test/test_client.rb

@@ -0,0 +1,184 @@
+require 'test/unit'
+
+require 'stark'
+
+require 'rubygems'
+require 'thrift'
+
+$: << "test/legacy_profile"
+
+require 'user_storage'
+
+class TestClient < Test::Unit::TestCase
+  def setup
+    @client_t, @server_t = Stark.pipe_transport
+    @client_p = Thrift::BinaryProtocol.new @client_t
+    @server_p = Thrift::BinaryProtocol.new @server_t
+
+    @n = Module.new
+    Stark.materialize "test/profile.thrift", @n
+
+    @client = @n::UserStorage::Client.new @client_p, @client_p
+    @handler = Handler.new
+    @server = UserStorage::Processor.new @handler
+  end
+
+  def teardown
+    @client_t.close
+    @server_t.close
+  end
+
+  class Handler
+    def initialize
+      @users = {}
+      @last_map = nil
+      @last_list = nil
+      @last_status = nil
+    end
+
+    attr_accessor :last_map, :last_list, :last_status
+
+    def store(obj)
+      @users[obj.uid] = obj
+    end
+
+    def retrieve(id)
+      @users[id]
+    end
+
+    def set_map(m)
+      @last_map = m
+    end
+
+    def set_list(l)
+      @last_list = l
+    end
+
+    def set_status(s)
+      @last_status = s
+    end
+  end
+
+  def test_store_and_retrieve
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    xuser = @n::UserProfile.new 'uid' => 0, 'name' => 'root', 'blurb' => 'god'
+
+    @client.store xuser
+
+    st.join
+
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    obj = @client.retrieve 0
+
+    st.join
+
+    assert_equal 0, obj.uid
+    assert_equal "root", obj.name
+    assert_equal "god", obj.blurb
+  end
+
+  def test_set_map
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    m = { "blah" => "foo", "a" => "b" }
+
+    @client.set_map m
+
+    st.join
+
+    assert_equal "foo", @handler.last_map["blah"]
+    assert_equal "b", @handler.last_map["a"]
+  end
+
+  def test_last_map
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    @handler.last_map = { "blah" => "foo", "a" => "b" }
+
+    m = @client.last_map
+
+    st.join
+
+    assert_equal "foo", m["blah"]
+    assert_equal "b", m["a"]
+  end
+
+  def test_set_list
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    m = [ "blah", "foo", "a", "b" ]
+
+    @client.set_list m
+
+    st.join
+
+    assert_equal m, @handler.last_list
+  end
+
+  def test_last_list
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    l = [ "blah", "foo", "a", "b" ]
+    @handler.last_list = l
+
+    begin
+      assert_equal l, @client.last_list
+    rescue Interrupt => e
+      puts e.backtrace
+    end
+
+    st.join
+  end
+
+  def test_last_list_is_nil
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    begin
+      assert_equal nil, @client.last_list
+    rescue Interrupt => e
+      puts e.backtrace
+    end
+
+    st.join
+  end
+
+  def test_enum
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    @client.set_status :ON
+
+    st.join
+
+    assert_equal 0, @handler.last_status
+  end
+
+  def test_enum_recv
+    st = Thread.new do
+      @server.process @server_p, @server_p
+    end
+
+    @handler.last_status = 0
+
+    assert_equal :ON, @client.last_status
+
+    st.join
+  end
+end