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