Add specs for Struct.
Include the generated code so the thrift compiler is not required to run the specs.
git-svn-id: https://svn.apache.org/repos/asf/incubator/thrift/trunk@668961 13f79535-47bb-0310-9956-ffa450edef68
diff --git a/lib/rb/spec/ThriftSpec.thrift b/lib/rb/spec/ThriftSpec.thrift
new file mode 100644
index 0000000..f22f17d
--- /dev/null
+++ b/lib/rb/spec/ThriftSpec.thrift
@@ -0,0 +1,14 @@
+namespace rb SpecNamespace
+
+struct Hello {
+ 1: string greeting = "hello world"
+}
+
+struct Foo {
+ 1: i32 simple = 53,
+ 2: string words = "words",
+ 3: Hello hello = {'greeting' : "hello, world!"},
+ 4: list<i32> ints = [1, 2, 2, 3],
+ 5: map<i32, map<string, double>> complex,
+ 6: set<i16> shorts = [5, 17, 239]
+}
diff --git a/lib/rb/spec/gen-rb/ThriftSpec_constants.rb b/lib/rb/spec/gen-rb/ThriftSpec_constants.rb
new file mode 100644
index 0000000..2fc84fa
--- /dev/null
+++ b/lib/rb/spec/gen-rb/ThriftSpec_constants.rb
@@ -0,0 +1,11 @@
+#
+# Autogenerated by Thrift
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift/protocol'
+require 'ThriftSpec_types'
+
+ module SpecNamespace
+end
diff --git a/lib/rb/spec/gen-rb/ThriftSpec_types.rb b/lib/rb/spec/gen-rb/ThriftSpec_types.rb
new file mode 100644
index 0000000..c4eab4d
--- /dev/null
+++ b/lib/rb/spec/gen-rb/ThriftSpec_types.rb
@@ -0,0 +1,41 @@
+#
+# Autogenerated by Thrift
+#
+# DO NOT EDIT UNLESS YOU ARE SURE THAT YOU KNOW WHAT YOU ARE DOING
+#
+
+require 'thrift/protocol'
+
+module SpecNamespace
+ class Hello
+ include Thrift::Struct
+ attr_accessor :greeting
+ FIELDS = {
+ 1 => {:type => Thrift::Types::STRING, :name => 'greeting', :default => 'hello world'}
+ }
+ end
+
+ class Foo
+ include Thrift::Struct
+ attr_accessor :simple, :words, :hello, :ints, :complex, :shorts
+ FIELDS = {
+ 1 => {:type => Thrift::Types::I32, :name => 'simple', :default => 53},
+ 2 => {:type => Thrift::Types::STRING, :name => 'words', :default => 'words'},
+ 3 => {:type => Thrift::Types::STRUCT, :name => 'hello', :default => Hello.new({
+ 'greeting' => 'hello, world!',
+ }), :class => Hello},
+ 4 => {:type => Thrift::Types::LIST, :name => 'ints', :default => [
+ 1,
+ 2,
+ 2,
+ 3,
+ ], :element => {:type => Thrift::Types::I32}},
+ 5 => {: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}}},
+ 6 => {:type => Thrift::Types::SET, :name => 'shorts', :default => Set.new([ 5,
+ 17,
+ 239,
+ ]), :element => {:type => Thrift::Types::I16}}
+ }
+ end
+
+ end
diff --git a/lib/rb/spec/struct_spec.rb b/lib/rb/spec/struct_spec.rb
new file mode 100644
index 0000000..5a4dc6a
--- /dev/null
+++ b/lib/rb/spec/struct_spec.rb
@@ -0,0 +1,153 @@
+require File.dirname(__FILE__) + '/spec_helper'
+require File.dirname(__FILE__) + "/gen-rb/ThriftSpec_types"
+
+class ThriftStructSpec < Spec::ExampleGroup
+ include Thrift
+ include SpecNamespace
+
+ class OldStruct
+ include Thrift::Struct
+ attr_accessor :set
+ FIELDS = {
+ 1 => {:type => Thrift::Types::SET, :name => 'val', :default => {:foo => true, :bar => true}}
+ }
+ end
+
+ it "should iterate over all fields properly" do
+ fields = {}
+ Foo.new.each_field { |fid,type,name,default| fields[fid] = [type,name,default] }
+ fields.should == {
+ 1 => [Types::I32, 'simple', 53],
+ 2 => [Types::STRING, 'words', "words"],
+ 3 => [Types::STRUCT, 'hello', Hello.new(:greeting => 'hello, world!')],
+ 4 => [Types::LIST, 'ints', [1, 2, 2, 3]],
+ 5 => [Types::MAP, 'complex', nil],
+ 6 => [Types::SET, 'shorts', Set.new([5, 17, 239])]
+ }
+ end
+
+ it "should initialize all fields to defaults" do
+ struct = Foo.new
+ struct.simple.should == 53
+ struct.words.should == "words"
+ struct.hello.should == Hello.new(:greeting => 'hello, world!')
+ struct.ints.should == [1, 2, 2, 3]
+ struct.complex.should be_nil
+ struct.shorts.should == Set.new([5, 17, 239])
+ end
+
+ it "should have proper == semantics" do
+ Foo.new.should_not == Hello.new
+ Foo.new.should == Foo.new
+ Foo.new(:simple => 52).should_not == Foo.new
+ end
+
+ it "should read itself off the wire" do
+ struct = Foo.new
+ prot = mock("Protocol")
+ prot.should_receive(:read_struct_begin).twice
+ prot.should_receive(:read_struct_end).twice
+ prot.should_receive(:read_field_begin).and_return(
+ ['complex', Types::MAP, 5], # Foo
+ ['words', Types::STRING, 2], # Foo
+ ['hello', Types::STRUCT, 3], # Foo
+ ['greeting', Types::STRING, 1], # Hello
+ [nil, Types::STOP, 0], # Hello
+ ['simple', Types::I32, 1], # Foo
+ ['ints', Types::LIST, 4], # Foo
+ ['shorts', Types::SET, 6], # Foo
+ [nil, Types::STOP, 0] # Hello
+ )
+ prot.should_receive(:read_field_end).exactly(7).times
+ prot.should_receive(:read_map_begin).and_return(
+ [Types::I32, Types::MAP, 2], # complex
+ [Types::STRING, Types::DOUBLE, 2], # complex/1/value
+ [Types::STRING, Types::DOUBLE, 1] # complex/2/value
+ )
+ prot.should_receive(:read_map_end).exactly(3).times
+ prot.should_receive(:read_list_begin).and_return([Types::I32, 4])
+ prot.should_receive(:read_list_end)
+ prot.should_receive(:read_set_begin).and_return([Types::I16, 2])
+ prot.should_receive(:read_set_end)
+ prot.should_receive(:read_type).with(Types::I32).and_return(
+ 1, 14, # complex keys
+ 42, # simple
+ 4, 23, 4, 29 # ints
+ )
+ prot.should_receive(:read_type).with(Types::STRING).and_return("pi", "e", "feigenbaum", "apple banana", "what's up?")
+ prot.should_receive(:read_type).with(Types::DOUBLE).and_return(Math::PI, Math::E, 4.669201609)
+ prot.should_receive(:read_type).with(Types::I16).and_return(2, 3)
+ prot.should_not_receive(:skip)
+ struct.read(prot)
+
+ struct.simple.should == 42
+ struct.complex.should == {1 => {"pi" => Math::PI, "e" => Math::E}, 14 => {"feigenbaum" => 4.669201609}}
+ struct.hello.should == Hello.new(:greeting => "what's up?")
+ struct.words.should == "apple banana"
+ struct.ints.should == [4, 23, 4, 29]
+ struct.shorts.should == Set.new([3, 2])
+ end
+
+ it "should skip unexpected fields in structs and use default values" do
+ struct = Foo.new
+ prot = mock("Protocol")
+ prot.should_receive(:read_struct_begin)
+ prot.should_receive(:read_struct_end)
+ prot.should_receive(:read_field_begin).and_return(
+ ['simple', Types::I32, 1],
+ ['complex', Types::STRUCT, 5],
+ ['thinz', Types::MAP, 7],
+ ['foobar', Types::I32, 3],
+ ['words', Types::STRING, 2],
+ [nil, Types::STOP, 0]
+ )
+ prot.should_receive(:read_field_end).exactly(5).times
+ prot.should_receive(:read_type).with(Types::I32).and_return(42)
+ prot.should_receive(:read_type).with(Types::STRING).and_return("foobar")
+ prot.should_receive(:skip).with(Types::STRUCT)
+ prot.should_receive(:skip).with(Types::MAP)
+ prot.should_receive(:skip).with(Types::I32)
+ struct.read(prot)
+
+ struct.simple.should == 42
+ struct.complex.should be_nil
+ struct.words.should == "foobar"
+ struct.hello.should == Hello.new(:greeting => 'hello, world!')
+ struct.ints.should == [1, 2, 2, 3]
+ struct.shorts.should == Set.new([5, 17, 239])
+ end
+
+ it "should write itself to the wire" do
+ prot = mock("Protocol")
+ prot.should_receive(:write_struct_begin).with("SpecNamespace::Foo")
+ prot.should_receive(:write_struct_end)
+ prot.should_receive(:write_field_begin).with('ints', Types::LIST, 4)
+ prot.should_receive(:write_field_begin).with('complex', Types::MAP, 5)
+ prot.should_receive(:write_field_begin).with('shorts', Types::SET, 6)
+ prot.should_receive(:write_field_stop)
+ prot.should_receive(:write_field_end).exactly(3).times
+ prot.should_receive(:write_field).with('simple', Types::I32, 1, 53)
+ prot.should_receive(:write_field).with('hello', Types::STRUCT, 3, Hello.new(:greeting => 'hello, world!'))
+ prot.should_receive(:write_map_begin).with(Types::I32, Types::MAP, 1)
+ prot.should_receive(:write_map_begin).with(Types::STRING, Types::DOUBLE, 1)
+ prot.should_receive(:write_type).with(Types::I32, 5) # complex/1/key
+ prot.should_receive(:write_type).with(Types::STRING, "foo") # complex/1/value/1/key
+ prot.should_receive(:write_type).with(Types::DOUBLE, 1.23) # complex/1/value/1/value
+ prot.should_receive(:write_map_end).twice
+ prot.should_receive(:write_list_begin).with(Types::I32, 4)
+ prot.should_receive(:write_type).with(Types::I32, 1)
+ prot.should_receive(:write_type).with(Types::I32, 2).twice
+ prot.should_receive(:write_type).with(Types::I32, 3)
+ prot.should_receive(:write_list_end)
+ prot.should_receive(:write_set_begin).with(Types::I16, 3)
+ prot.should_receive(:write_type).with(Types::I16, 5)
+ prot.should_receive(:write_type).with(Types::I16, 17)
+ prot.should_receive(:write_type).with(Types::I16, 239)
+ prot.should_receive(:write_set_end)
+
+ struct = Foo.new
+ struct.words = nil
+ struct.complex = {5 => {"foo" => 1.23}}
+ struct.write(prot)
+ end
+end