slayer-thrift 0.7.0

data/spec/gen-rb/thrift_spec_constants.rb

@@ -0,0 +1,10 @@
+#
+# Autogenerated by Thrift Compiler (0.7.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift_spec_types'
+
+  module SpecNamespace
+end

data/spec/gen-rb/thrift_spec_types.rb

@@ -0,0 +1,345 @@
+#
+# Autogenerated by Thrift Compiler (0.7.0)
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+
+module SpecNamespace
+    module SomeEnum
+      ONE = 0
+      TWO = 1
+      VALUE_MAP = {0 => "ONE", 1 => "TWO"}
+      VALID_VALUES = Set.new([ONE, TWO]).freeze
+    end
+
+    class Hello
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      GREETING = 1
+
+      FIELDS = {
+        GREETING => {:type => ::Thrift::Types::STRING, :name => 'greeting', :default => %q"hello world"}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class StructWithSomeEnum
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SOME_ENUM = 1
+
+      FIELDS = {
+        SOME_ENUM => {:type => ::Thrift::Types::I32, :name => 'some_enum', :enum_class => SpecNamespace::SomeEnum}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        unless @some_enum.nil? || SpecNamespace::SomeEnum::VALID_VALUES.include?(@some_enum)
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field some_enum!')
+        end
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class TestUnion < ::Thrift::Union
+      include ::Thrift::Struct_Union
+      class << self
+        def string_field(val)
+          TestUnion.new(:string_field, val)
+        end
+
+        def i32_field(val)
+          TestUnion.new(:i32_field, val)
+        end
+
+        def other_i32_field(val)
+          TestUnion.new(:other_i32_field, val)
+        end
+
+        def enum_field(val)
+          TestUnion.new(:enum_field, val)
+        end
+
+        def binary_field(val)
+          TestUnion.new(:binary_field, val)
+        end
+      end
+
+      STRING_FIELD = 1
+      I32_FIELD = 2
+      OTHER_I32_FIELD = 3
+      ENUM_FIELD = 4
+      BINARY_FIELD = 5
+
+      FIELDS = {
+        # A doc string
+        STRING_FIELD => {:type => ::Thrift::Types::STRING, :name => 'string_field'},
+        I32_FIELD => {:type => ::Thrift::Types::I32, :name => 'i32_field'},
+        OTHER_I32_FIELD => {:type => ::Thrift::Types::I32, :name => 'other_i32_field'},
+        ENUM_FIELD => {:type => ::Thrift::Types::I32, :name => 'enum_field', :enum_class => SpecNamespace::SomeEnum},
+        BINARY_FIELD => {:type => ::Thrift::Types::STRING, :name => 'binary_field', :binary => true}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise(StandardError, 'Union fields are not set.') if get_set_field.nil? || get_value.nil?
+        if get_set_field == :enum_field
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field enum_field!') unless SpecNamespace::SomeEnum::VALID_VALUES.include?(get_value)
+        end
+      end
+
+      ::Thrift::Union.generate_accessors self
+    end
+
+    class Foo
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      SIMPLE = 1
+      WORDS = 2
+      HELLO = 3
+      INTS = 4
+      COMPLEX = 5
+      SHORTS = 6
+      OPT_STRING = 7
+      MY_BOOL = 8
+
+      FIELDS = {
+        SIMPLE => {:type => ::Thrift::Types::I32, :name => 'simple', :default => 53},
+        WORDS => {:type => ::Thrift::Types::STRING, :name => 'words', :default => %q"words"},
+        HELLO => {:type => ::Thrift::Types::STRUCT, :name => 'hello', :default => SpecNamespace::Hello.new({
+          %q"greeting" => %q"hello, world!",
+        }), :class => SpecNamespace::Hello},
+        INTS => {:type => ::Thrift::Types::LIST, :name => 'ints', :default => [
+          1,
+          2,
+          2,
+          3,
+        ], :element => {:type => ::Thrift::Types::I32}},
+        COMPLEX => {:type => ::Thrift::Types::MAP, :name => 'complex', :key => {:type => ::Thrift::Types::I32}, :value => {:type => ::Thrift::Types::MAP, :key => {:type => ::Thrift::Types::STRING}, :value => {:type => ::Thrift::Types::DOUBLE}}},
+        SHORTS => {:type => ::Thrift::Types::SET, :name => 'shorts', :default => Set.new([
+          5,
+          17,
+          239,
+        ]), :element => {:type => ::Thrift::Types::I16}},
+        OPT_STRING => {:type => ::Thrift::Types::STRING, :name => 'opt_string', :optional => true},
+        MY_BOOL => {:type => ::Thrift::Types::BOOL, :name => 'my_bool'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class Foo2
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      MY_BINARY = 1
+
+      FIELDS = {
+        MY_BINARY => {:type => ::Thrift::Types::STRING, :name => 'my_binary', :binary => true}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class BoolStruct
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      YESNO = 1
+
+      FIELDS = {
+        YESNO => {:type => ::Thrift::Types::BOOL, :name => 'yesno', :default => true}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class SimpleList
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      BOOLS = 1
+      BYTES = 2
+      I16S = 3
+      I32S = 4
+      I64S = 5
+      DOUBLES = 6
+      STRINGS = 7
+      MAPS = 8
+      LISTS = 9
+      SETS = 10
+      HELLOS = 11
+
+      FIELDS = {
+        BOOLS => {:type => ::Thrift::Types::LIST, :name => 'bools', :element => {:type => ::Thrift::Types::BOOL}},
+        BYTES => {:type => ::Thrift::Types::LIST, :name => 'bytes', :element => {:type => ::Thrift::Types::BYTE}},
+        I16S => {:type => ::Thrift::Types::LIST, :name => 'i16s', :element => {:type => ::Thrift::Types::I16}},
+        I32S => {:type => ::Thrift::Types::LIST, :name => 'i32s', :element => {:type => ::Thrift::Types::I32}},
+        I64S => {:type => ::Thrift::Types::LIST, :name => 'i64s', :element => {:type => ::Thrift::Types::I64}},
+        DOUBLES => {:type => ::Thrift::Types::LIST, :name => 'doubles', :element => {:type => ::Thrift::Types::DOUBLE}},
+        STRINGS => {:type => ::Thrift::Types::LIST, :name => 'strings', :element => {:type => ::Thrift::Types::STRING}},
+        MAPS => {:type => ::Thrift::Types::LIST, :name => 'maps', :element => {:type => ::Thrift::Types::MAP, :key => {:type => ::Thrift::Types::I16}, :value => {:type => ::Thrift::Types::I16}}},
+        LISTS => {:type => ::Thrift::Types::LIST, :name => 'lists', :element => {:type => ::Thrift::Types::LIST, :element => {:type => ::Thrift::Types::I16}}},
+        SETS => {:type => ::Thrift::Types::LIST, :name => 'sets', :element => {:type => ::Thrift::Types::SET, :element => {:type => ::Thrift::Types::I16}}},
+        HELLOS => {:type => ::Thrift::Types::LIST, :name => 'hellos', :element => {:type => ::Thrift::Types::STRUCT, :class => SpecNamespace::Hello}}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class Xception < ::Thrift::Exception
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      MESSAGE = 1
+      CODE = 2
+
+      FIELDS = {
+        MESSAGE => {:type => ::Thrift::Types::STRING, :name => 'message'},
+        CODE => {:type => ::Thrift::Types::I32, :name => 'code', :default => 1}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class My_union < ::Thrift::Union
+      include ::Thrift::Struct_Union
+      class << self
+        def im_true(val)
+          My_union.new(:im_true, val)
+        end
+
+        def a_bite(val)
+          My_union.new(:a_bite, val)
+        end
+
+        def integer16(val)
+          My_union.new(:integer16, val)
+        end
+
+        def integer32(val)
+          My_union.new(:integer32, val)
+        end
+
+        def integer64(val)
+          My_union.new(:integer64, val)
+        end
+
+        def double_precision(val)
+          My_union.new(:double_precision, val)
+        end
+
+        def some_characters(val)
+          My_union.new(:some_characters, val)
+        end
+
+        def other_i32(val)
+          My_union.new(:other_i32, val)
+        end
+
+        def some_enum(val)
+          My_union.new(:some_enum, val)
+        end
+
+        def my_map(val)
+          My_union.new(:my_map, val)
+        end
+      end
+
+      IM_TRUE = 1
+      A_BITE = 2
+      INTEGER16 = 3
+      INTEGER32 = 4
+      INTEGER64 = 5
+      DOUBLE_PRECISION = 6
+      SOME_CHARACTERS = 7
+      OTHER_I32 = 8
+      SOME_ENUM = 9
+      MY_MAP = 10
+
+      FIELDS = {
+        IM_TRUE => {:type => ::Thrift::Types::BOOL, :name => 'im_true'},
+        A_BITE => {:type => ::Thrift::Types::BYTE, :name => 'a_bite'},
+        INTEGER16 => {:type => ::Thrift::Types::I16, :name => 'integer16'},
+        INTEGER32 => {:type => ::Thrift::Types::I32, :name => 'integer32'},
+        INTEGER64 => {:type => ::Thrift::Types::I64, :name => 'integer64'},
+        DOUBLE_PRECISION => {:type => ::Thrift::Types::DOUBLE, :name => 'double_precision'},
+        SOME_CHARACTERS => {:type => ::Thrift::Types::STRING, :name => 'some_characters'},
+        OTHER_I32 => {:type => ::Thrift::Types::I32, :name => 'other_i32'},
+        SOME_ENUM => {:type => ::Thrift::Types::I32, :name => 'some_enum', :enum_class => SpecNamespace::SomeEnum},
+        MY_MAP => {:type => ::Thrift::Types::MAP, :name => 'my_map', :key => {:type => ::Thrift::Types::I32, :enum_class => SpecNamespace::SomeEnum}, :value => {:type => ::Thrift::Types::LIST, :element => {:type => ::Thrift::Types::I32, :enum_class => SpecNamespace::SomeEnum}}}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+        raise(StandardError, 'Union fields are not set.') if get_set_field.nil? || get_value.nil?
+        if get_set_field == :some_enum
+          raise ::Thrift::ProtocolException.new(::Thrift::ProtocolException::UNKNOWN, 'Invalid value of field some_enum!') unless SpecNamespace::SomeEnum::VALID_VALUES.include?(get_value)
+        end
+      end
+
+      ::Thrift::Union.generate_accessors self
+    end
+
+    class Struct_with_union
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      FUN_UNION = 1
+      INTEGER32 = 2
+      SOME_CHARACTERS = 3
+
+      FIELDS = {
+        FUN_UNION => {:type => ::Thrift::Types::STRUCT, :name => 'fun_union', :class => SpecNamespace::My_union},
+        INTEGER32 => {:type => ::Thrift::Types::I32, :name => 'integer32'},
+        SOME_CHARACTERS => {:type => ::Thrift::Types::STRING, :name => 'some_characters'}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+    class StructWithEnumMap
+      include ::Thrift::Struct, ::Thrift::Struct_Union
+      MY_MAP = 1
+
+      FIELDS = {
+        MY_MAP => {:type => ::Thrift::Types::MAP, :name => 'my_map', :key => {:type => ::Thrift::Types::I32, :enum_class => SpecNamespace::SomeEnum}, :value => {:type => ::Thrift::Types::LIST, :element => {:type => ::Thrift::Types::I32, :enum_class => SpecNamespace::SomeEnum}}}
+      }
+
+      def struct_fields; FIELDS; end
+
+      def validate
+      end
+
+      ::Thrift::Struct.generate_accessors self
+    end
+
+  end

data/spec/http_client_spec.rb

@@ -0,0 +1,64 @@
+#
+# 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 File.dirname(__FILE__) + '/spec_helper'
+
+class ThriftHTTPClientTransportSpec < Spec::ExampleGroup
+  include Thrift
+
+  describe HTTPClientTransport do
+    before(:each) do
+      @client = HTTPClientTransport.new("http://my.domain.com/path/to/service?param=value")
+    end
+
+    it "should always be open" do
+      @client.should be_open
+      @client.close
+      @client.should be_open
+    end
+
+    it "should post via HTTP and return the results" do
+      @client.write "a test"
+      @client.write " frame"
+      Net::HTTP.should_receive(:new).with("my.domain.com", 80).and_return do
+        mock("Net::HTTP").tee do |http|
+          http.should_receive(:use_ssl=).with(false)
+          http.should_receive(:post).with("/path/to/service?param=value", "a test frame", {"Content-Type"=>"application/x-thrift"}).and_return([nil, "data"])
+        end
+      end
+      @client.flush
+      @client.read(10).should == "data"
+    end
+
+    it "should send custom headers if defined" do
+      @client.write "test"
+      custom_headers = {"Cookie" => "Foo"}
+      headers = {"Content-Type"=>"application/x-thrift"}.merge(custom_headers)
+
+      @client.add_headers(custom_headers)
+      Net::HTTP.should_receive(:new).with("my.domain.com", 80).and_return do
+        mock("Net::HTTP").tee do |http|
+          http.should_receive(:use_ssl=).with(false)
+          http.should_receive(:post).with("/path/to/service?param=value", "test", headers).and_return([nil, "data"])
+        end
+      end
+      @client.flush
+    end
+  end
+end

data/spec/mongrel_http_server_spec.rb

@@ -0,0 +1,117 @@
+#
+# 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 File.dirname(__FILE__) + '/spec_helper'
+require 'thrift/server/mongrel_http_server'
+
+class ThriftHTTPServerSpec < Spec::ExampleGroup
+  include Thrift
+
+  Handler = MongrelHTTPServer::Handler
+
+  describe MongrelHTTPServer do
+    it "should have appropriate defaults" do
+      mock_factory = mock("BinaryProtocolFactory")
+      mock_proc = mock("Processor")
+      BinaryProtocolFactory.should_receive(:new).and_return(mock_factory)
+      Mongrel::HttpServer.should_receive(:new).with("0.0.0.0", 80).and_return do
+        mock("Mongrel::HttpServer").tee do |mock|
+          handler = mock("Handler")
+          Handler.should_receive(:new).with(mock_proc, mock_factory).and_return(handler)
+          mock.should_receive(:register).with("/", handler)
+        end
+      end
+      MongrelHTTPServer.new(mock_proc)
+    end
+
+    it "should understand :ip, :port, :path, and :protocol_factory" do
+      mock_proc = mock("Processor")
+      mock_factory = mock("ProtocolFactory")
+      Mongrel::HttpServer.should_receive(:new).with("1.2.3.4", 1234).and_return do
+        mock("Mongrel::HttpServer").tee do |mock|
+          handler = mock("Handler")
+          Handler.should_receive(:new).with(mock_proc, mock_factory).and_return(handler)
+          mock.should_receive(:register).with("/foo", handler)
+        end
+      end
+      MongrelHTTPServer.new(mock_proc, :ip => "1.2.3.4", :port => 1234, :path => "foo",
+                                             :protocol_factory => mock_factory)
+    end
+
+    it "should serve using Mongrel::HttpServer" do
+      BinaryProtocolFactory.stub!(:new)
+      Mongrel::HttpServer.should_receive(:new).and_return do
+        mock("Mongrel::HttpServer").tee do |mock|
+          Handler.stub!(:new)
+          mock.stub!(:register)
+          mock.should_receive(:run).and_return do
+            mock("Mongrel::HttpServer.run").tee do |runner|
+              runner.should_receive(:join)
+            end
+          end
+        end
+      end
+      MongrelHTTPServer.new(nil).serve
+    end
+  end
+
+  describe MongrelHTTPServer::Handler do
+    before(:each) do
+      @processor = mock("Processor")
+      @factory = mock("ProtocolFactory")
+      @handler = Handler.new(@processor, @factory)
+    end
+
+    it "should return 404 for non-POST requests" do
+      request = mock("request", :params => {"REQUEST_METHOD" => "GET"})
+      response = mock("response")
+      response.should_receive(:start).with(404)
+      response.should_not_receive(:start).with(200)
+      @handler.process(request, response)
+    end
+
+    it "should serve using application/x-thrift" do
+      request = mock("request", :params => {"REQUEST_METHOD" => "POST"}, :body => nil)
+      response = mock("response")
+      head = mock("head")
+      head.should_receive(:[]=).with("Content-Type", "application/x-thrift")
+      IOStreamTransport.stub!(:new)
+      @factory.stub!(:get_protocol)
+      @processor.stub!(:process)
+      response.should_receive(:start).with(200).and_yield(head, nil)
+      @handler.process(request, response)
+    end
+
+    it "should use the IOStreamTransport" do
+      body = mock("body")
+      request = mock("request", :params => {"REQUEST_METHOD" => "POST"}, :body => body)
+      response = mock("response")
+      head = mock("head")
+      head.stub!(:[]=)
+      out = mock("out")
+      protocol = mock("protocol")
+      transport = mock("transport")
+      IOStreamTransport.should_receive(:new).with(body, out).and_return(transport)
+      @factory.should_receive(:get_protocol).with(transport).and_return(protocol)
+      @processor.should_receive(:process).with(protocol, protocol)
+      response.should_receive(:start).with(200).and_yield(head, out)
+      @handler.process(request, response)
+    end
+  end
+end