| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077 | #!/usr/bin/rubyrequire 'google/protobuf'require 'test/unit'# ------------- generated code --------------module BasicTest  pool = Google::Protobuf::DescriptorPool.new  pool.build do    add_message "Foo" do      optional :bar, :message, 1, "Bar"      repeated :baz, :message, 2, "Baz"    end    add_message "Bar" do      optional :msg, :string, 1    end    add_message "Baz" do      optional :msg, :string, 1    end    add_message "TestMessage" do      optional :optional_int32,  :int32,        1      optional :optional_int64,  :int64,        2      optional :optional_uint32, :uint32,       3      optional :optional_uint64, :uint64,       4      optional :optional_bool,   :bool,         5      optional :optional_float,  :float,        6      optional :optional_double, :double,       7      optional :optional_string, :string,       8      optional :optional_bytes,  :bytes,        9      optional :optional_msg,    :message,      10, "TestMessage2"      optional :optional_enum,   :enum,         11, "TestEnum"      repeated :repeated_int32,  :int32,        12      repeated :repeated_int64,  :int64,        13      repeated :repeated_uint32, :uint32,       14      repeated :repeated_uint64, :uint64,       15      repeated :repeated_bool,   :bool,         16      repeated :repeated_float,  :float,        17      repeated :repeated_double, :double,       18      repeated :repeated_string, :string,       19      repeated :repeated_bytes,  :bytes,        20      repeated :repeated_msg,    :message,      21, "TestMessage2"      repeated :repeated_enum,   :enum,         22, "TestEnum"    end    add_message "TestMessage2" do      optional :foo, :int32, 1    end    add_message "Recursive1" do      optional :foo, :message, 1, "Recursive2"    end    add_message "Recursive2" do      optional :foo, :message, 1, "Recursive1"    end    add_enum "TestEnum" do      value :Default, 0      value :A, 1      value :B, 2      value :C, 3    end    add_message "BadFieldNames" do      optional :dup, :int32, 1      optional :class, :int32, 2      optional :"a.b", :int32, 3    end    add_message "MapMessage" do      map :map_string_int32, :string, :int32, 1      map :map_string_msg, :string, :message, 2, "TestMessage2"    end    add_message "MapMessageWireEquiv" do      repeated :map_string_int32, :message, 1, "MapMessageWireEquiv_entry1"      repeated :map_string_msg, :message, 2, "MapMessageWireEquiv_entry2"    end    add_message "MapMessageWireEquiv_entry1" do      optional :key, :string, 1      optional :value, :int32, 2    end    add_message "MapMessageWireEquiv_entry2" do      optional :key, :string, 1      optional :value, :message, 2, "TestMessage2"    end    add_message "OneofMessage" do      oneof :my_oneof do        optional :a, :string, 1        optional :b, :int32, 2        optional :c, :message, 3, "TestMessage2"        optional :d, :enum, 4, "TestEnum"      end    end  end  Foo = pool.lookup("Foo").msgclass  Bar = pool.lookup("Bar").msgclass  Baz = pool.lookup("Baz").msgclass  TestMessage = pool.lookup("TestMessage").msgclass  TestMessage2 = pool.lookup("TestMessage2").msgclass  Recursive1 = pool.lookup("Recursive1").msgclass  Recursive2 = pool.lookup("Recursive2").msgclass  TestEnum = pool.lookup("TestEnum").enummodule  BadFieldNames = pool.lookup("BadFieldNames").msgclass  MapMessage = pool.lookup("MapMessage").msgclass  MapMessageWireEquiv = pool.lookup("MapMessageWireEquiv").msgclass  MapMessageWireEquiv_entry1 =    pool.lookup("MapMessageWireEquiv_entry1").msgclass  MapMessageWireEquiv_entry2 =    pool.lookup("MapMessageWireEquiv_entry2").msgclass  OneofMessage = pool.lookup("OneofMessage").msgclass# ------------ test cases ---------------  class MessageContainerTest < Test::Unit::TestCase    def test_defaults      m = TestMessage.new      assert m.optional_int32 == 0      assert m.optional_int64 == 0      assert m.optional_uint32 == 0      assert m.optional_uint64 == 0      assert m.optional_bool == false      assert m.optional_float == 0.0      assert m.optional_double == 0.0      assert m.optional_string == ""      assert m.optional_bytes == ""      assert m.optional_msg == nil      assert m.optional_enum == :Default    end    def test_setters      m = TestMessage.new      m.optional_int32 = -42      assert m.optional_int32 == -42      m.optional_int64 = -0x1_0000_0000      assert m.optional_int64 == -0x1_0000_0000      m.optional_uint32 = 0x9000_0000      assert m.optional_uint32 == 0x9000_0000      m.optional_uint64 = 0x9000_0000_0000_0000      assert m.optional_uint64 == 0x9000_0000_0000_0000      m.optional_bool = true      assert m.optional_bool == true      m.optional_float = 0.5      assert m.optional_float == 0.5      m.optional_double = 0.5      m.optional_string = "hello"      assert m.optional_string == "hello"      m.optional_bytes = "world".encode!('ASCII-8BIT')      assert m.optional_bytes == "world"      m.optional_msg = TestMessage2.new(:foo => 42)      assert m.optional_msg == TestMessage2.new(:foo => 42)      m.optional_msg = nil      assert m.optional_msg == nil    end    def test_ctor_args      m = TestMessage.new(:optional_int32 => -42,                          :optional_msg => TestMessage2.new,                          :optional_enum => :C,                          :repeated_string => ["hello", "there", "world"])      assert m.optional_int32 == -42      assert m.optional_msg.class == TestMessage2      assert m.repeated_string.length == 3      assert m.optional_enum == :C      assert m.repeated_string[0] == "hello"      assert m.repeated_string[1] == "there"      assert m.repeated_string[2] == "world"    end    def test_inspect      m = TestMessage.new(:optional_int32 => -42,                          :optional_enum => :A,                          :optional_msg => TestMessage2.new,                          :repeated_string => ["hello", "there", "world"])      expected = '<BasicTest::TestMessage: optional_int32: -42, optional_int64: 0, optional_uint32: 0, optional_uint64: 0, optional_bool: false, optional_float: 0.0, optional_double: 0.0, optional_string: "", optional_bytes: "", optional_msg: <BasicTest::TestMessage2: foo: 0>, optional_enum: :A, repeated_int32: [], repeated_int64: [], repeated_uint32: [], repeated_uint64: [], repeated_bool: [], repeated_float: [], repeated_double: [], repeated_string: ["hello", "there", "world"], repeated_bytes: [], repeated_msg: [], repeated_enum: []>'      assert m.inspect == expected    end    def test_hash      m1 = TestMessage.new(:optional_int32 => 42)      m2 = TestMessage.new(:optional_int32 => 102)      assert m1.hash != 0      assert m2.hash != 0      # relying on the randomness here -- if hash function changes and we are      # unlucky enough to get a collision, then change the values above.      assert m1.hash != m2.hash    end    def test_type_errors      m = TestMessage.new      assert_raise TypeError do        m.optional_int32 = "hello"      end      assert_raise TypeError do        m.optional_string = 42      end      assert_raise TypeError do        m.optional_string = nil      end      assert_raise TypeError do        m.optional_bool = 42      end      assert_raise TypeError do        m.optional_msg = TestMessage.new  # expects TestMessage2      end      assert_raise TypeError do        m.repeated_int32 = []  # needs RepeatedField      end      assert_raise TypeError do        m.repeated_int32.push "hello"      end      assert_raise TypeError do        m.repeated_msg.push TestMessage.new      end    end    def test_string_encoding      m = TestMessage.new      # Assigning a normal (ASCII or UTF8) string to a bytes field, or      # ASCII-8BIT to a string field, raises an error.      assert_raise TypeError do        m.optional_bytes = "Test string ASCII".encode!('ASCII')      end      assert_raise TypeError do        m.optional_bytes = "Test string UTF-8 \u0100".encode!('UTF-8')      end      assert_raise TypeError do        m.optional_string = ["FFFF"].pack('H*')      end      # "Ordinary" use case.      m.optional_bytes = ["FFFF"].pack('H*')      m.optional_string = "\u0100"      # strings are mutable so we can do this, but serialize should catch it.      m.optional_string = "asdf".encode!('UTF-8')      m.optional_string.encode!('ASCII-8BIT')      assert_raise TypeError do        data = TestMessage.encode(m)      end    end    def test_rptfield_int32      l = Google::Protobuf::RepeatedField.new(:int32)      assert l.count == 0      l = Google::Protobuf::RepeatedField.new(:int32, [1, 2, 3])      assert l.count == 3      assert_equal [1, 2, 3], l      assert_equal l, [1, 2, 3]      l.push 4      assert l == [1, 2, 3, 4]      dst_list = []      l.each { |val| dst_list.push val }      assert dst_list == [1, 2, 3, 4]      assert l.to_a == [1, 2, 3, 4]      assert l[0] == 1      assert l[3] == 4      l[0] = 5      assert l == [5, 2, 3, 4]      l2 = l.dup      assert l == l2      assert l.object_id != l2.object_id      l2.push 6      assert l.count == 4      assert l2.count == 5      assert l.inspect == '[5, 2, 3, 4]'      l.insert(7, 8, 9)      assert l == [5, 2, 3, 4, 7, 8, 9]      assert l.pop == 9      assert l == [5, 2, 3, 4, 7, 8]      assert_raise TypeError do        m = TestMessage.new        l.push m      end      m = TestMessage.new      m.repeated_int32 = l      assert m.repeated_int32 == [5, 2, 3, 4, 7, 8]      assert m.repeated_int32.object_id == l.object_id      l.push 42      assert m.repeated_int32.pop == 42      l3 = l + l.dup      assert l3.count == l.count * 2      l.count.times do |i|        assert l3[i] == l[i]        assert l3[l.count + i] == l[i]      end      l.clear      assert l.count == 0      l += [1, 2, 3, 4]      l.replace([5, 6, 7, 8])      assert l == [5, 6, 7, 8]      l4 = Google::Protobuf::RepeatedField.new(:int32)      l4[5] = 42      assert l4 == [0, 0, 0, 0, 0, 42]      l4 << 100      assert l4 == [0, 0, 0, 0, 0, 42, 100]      l4 << 101 << 102      assert l4 == [0, 0, 0, 0, 0, 42, 100, 101, 102]    end    def test_parent_rptfield      #make sure we set the RepeatedField and can add to it      m = TestMessage.new      assert m.repeated_string == []      m.repeated_string << 'ok'      m.repeated_string.push('ok2')      assert m.repeated_string == ['ok', 'ok2']      m.repeated_string += ['ok3']      assert m.repeated_string == ['ok', 'ok2', 'ok3']    end    def test_rptfield_msg      l = Google::Protobuf::RepeatedField.new(:message, TestMessage)      l.push TestMessage.new      assert l.count == 1      assert_raise TypeError do        l.push TestMessage2.new      end      assert_raise TypeError do        l.push 42      end      l2 = l.dup      assert l2[0] == l[0]      assert l2[0].object_id == l[0].object_id      l2 = Google::Protobuf.deep_copy(l)      assert l2[0] == l[0]      assert l2[0].object_id != l[0].object_id      l3 = l + l2      assert l3.count == 2      assert l3[0] == l[0]      assert l3[1] == l2[0]      l3[0].optional_int32 = 1000      assert l[0].optional_int32 == 1000      new_msg = TestMessage.new(:optional_int32 => 200)      l4 = l + [new_msg]      assert l4.count == 2      new_msg.optional_int32 = 1000      assert l4[1].optional_int32 == 1000    end    def test_rptfield_enum      l = Google::Protobuf::RepeatedField.new(:enum, TestEnum)      l.push :A      l.push :B      l.push :C      assert l.count == 3      assert_raise RangeError do        l.push :D      end      assert l[0] == :A      l.push 4      assert l[3] == 4    end    def test_rptfield_initialize      assert_raise ArgumentError do        l = Google::Protobuf::RepeatedField.new      end      assert_raise ArgumentError do        l = Google::Protobuf::RepeatedField.new(:message)      end      assert_raise ArgumentError do        l = Google::Protobuf::RepeatedField.new([1, 2, 3])      end      assert_raise ArgumentError do        l = Google::Protobuf::RepeatedField.new(:message, [TestMessage2.new])      end    end    def test_rptfield_array_ducktyping      l = Google::Protobuf::RepeatedField.new(:int32)      length_methods = %w(count length size)      length_methods.each do |lm|        assert l.send(lm)  == 0      end      # out of bounds returns a nil      assert l[0] == nil      assert l[1] == nil      assert l[-1] == nil      l.push 4      length_methods.each do |lm|        assert l.send(lm) == 1      end      assert l[0] == 4      assert l[1] == nil      assert l[-1] == 4      assert l[-2] == nil      l.push 2      length_methods.each do |lm|        assert l.send(lm) == 2      end      assert l[0] == 4      assert l[1] == 2      assert l[2] == nil      assert l[-1] == 2      assert l[-2] == 4      assert l[-3] == nil      #adding out of scope will backfill with empty objects    end    def test_map_basic      # allowed key types:      # :int32, :int64, :uint32, :uint64, :bool, :string, :bytes.      m = Google::Protobuf::Map.new(:string, :int32)      m["asdf"] = 1      assert m["asdf"] == 1      m["jkl;"] = 42      assert m == { "jkl;" => 42, "asdf" => 1 }      assert m.has_key?("asdf")      assert !m.has_key?("qwerty")      assert m.length == 2      m2 = m.dup      assert m == m2      assert m.hash != 0      assert m.hash == m2.hash      collected = {}      m.each { |k,v| collected[v] = k }      assert collected == { 42 => "jkl;", 1 => "asdf" }      assert m.delete("asdf") == 1      assert !m.has_key?("asdf")      assert m["asdf"] == nil      assert !m.has_key?("asdf")      # We only assert on inspect value when there is one map entry because the      # order in which elements appear is unspecified (depends on the internal      # hash function). We don't want a brittle test.      assert m.inspect == "{\"jkl;\"=>42}"      assert m.keys == ["jkl;"]      assert m.values == [42]      m.clear      assert m.length == 0      assert m == {}      assert_raise TypeError do        m[1] = 1      end      assert_raise RangeError do        m["asdf"] = 0x1_0000_0000      end    end    def test_map_ctor      m = Google::Protobuf::Map.new(:string, :int32,                                    {"a" => 1, "b" => 2, "c" => 3})      assert m == {"a" => 1, "c" => 3, "b" => 2}    end    def test_map_keytypes      m = Google::Protobuf::Map.new(:int32, :int32)      m[1] = 42      m[-1] = 42      assert_raise RangeError do        m[0x8000_0000] = 1      end      assert_raise TypeError do        m["asdf"] = 1      end      m = Google::Protobuf::Map.new(:int64, :int32)      m[0x1000_0000_0000_0000] = 1      assert_raise RangeError do        m[0x1_0000_0000_0000_0000] = 1      end      assert_raise TypeError do        m["asdf"] = 1      end      m = Google::Protobuf::Map.new(:uint32, :int32)      m[0x8000_0000] = 1      assert_raise RangeError do        m[0x1_0000_0000] = 1      end      assert_raise RangeError do        m[-1] = 1      end      m = Google::Protobuf::Map.new(:uint64, :int32)      m[0x8000_0000_0000_0000] = 1      assert_raise RangeError do        m[0x1_0000_0000_0000_0000] = 1      end      assert_raise RangeError do        m[-1] = 1      end      m = Google::Protobuf::Map.new(:bool, :int32)      m[true] = 1      m[false] = 2      assert_raise TypeError do        m[1] = 1      end      assert_raise TypeError do        m["asdf"] = 1      end      m = Google::Protobuf::Map.new(:string, :int32)      m["asdf"] = 1      assert_raise TypeError do        m[1] = 1      end      assert_raise TypeError do        bytestring = ["FFFF"].pack("H*")        m[bytestring] = 1      end      m = Google::Protobuf::Map.new(:bytes, :int32)      bytestring = ["FFFF"].pack("H*")      m[bytestring] = 1      assert_raise TypeError do        m["asdf"] = 1      end      assert_raise TypeError do        m[1] = 1      end    end    def test_map_msg_enum_valuetypes      m = Google::Protobuf::Map.new(:string, :message, TestMessage)      m["asdf"] = TestMessage.new      assert_raise TypeError do        m["jkl;"] = TestMessage2.new      end      m = Google::Protobuf::Map.new(        :string, :message, TestMessage,        { "a" => TestMessage.new(:optional_int32 => 42),          "b" => TestMessage.new(:optional_int32 => 84) })      assert m.length == 2      assert m.values.map{|msg| msg.optional_int32}.sort == [42, 84]      m = Google::Protobuf::Map.new(:string, :enum, TestEnum,                                    { "x" => :A, "y" => :B, "z" => :C })      assert m.length == 3      assert m["z"] == :C      m["z"] = 2      assert m["z"] == :B      m["z"] = 4      assert m["z"] == 4      assert_raise RangeError do        m["z"] = :Z      end      assert_raise TypeError do        m["z"] = "z"      end    end    def test_map_dup_deep_copy      m = Google::Protobuf::Map.new(        :string, :message, TestMessage,        { "a" => TestMessage.new(:optional_int32 => 42),          "b" => TestMessage.new(:optional_int32 => 84) })      m2 = m.dup      assert m == m2      assert m.object_id != m2.object_id      assert m["a"].object_id == m2["a"].object_id      assert m["b"].object_id == m2["b"].object_id      m2 = Google::Protobuf.deep_copy(m)      assert m == m2      assert m.object_id != m2.object_id      assert m["a"].object_id != m2["a"].object_id      assert m["b"].object_id != m2["b"].object_id    end    def test_map_field      m = MapMessage.new      assert m.map_string_int32 == {}      assert m.map_string_msg == {}      m = MapMessage.new(        :map_string_int32 => {"a" => 1, "b" => 2},        :map_string_msg => {"a" => TestMessage2.new(:foo => 1),                            "b" => TestMessage2.new(:foo => 2)})      assert m.map_string_int32.keys.sort == ["a", "b"]      assert m.map_string_int32["a"] == 1      assert m.map_string_msg["b"].foo == 2      m.map_string_int32["c"] = 3      assert m.map_string_int32["c"] == 3      m.map_string_msg["c"] = TestMessage2.new(:foo => 3)      assert m.map_string_msg["c"] == TestMessage2.new(:foo => 3)      m.map_string_msg.delete("b")      m.map_string_msg.delete("c")      assert m.map_string_msg == { "a" => TestMessage2.new(:foo => 1) }      assert_raise TypeError do        m.map_string_msg["e"] = TestMessage.new # wrong value type      end      # ensure nothing was added by the above      assert m.map_string_msg == { "a" => TestMessage2.new(:foo => 1) }      m.map_string_int32 = Google::Protobuf::Map.new(:string, :int32)      assert_raise TypeError do        m.map_string_int32 = Google::Protobuf::Map.new(:string, :int64)      end      assert_raise TypeError do        m.map_string_int32 = {}      end      assert_raise TypeError do        m = MapMessage.new(:map_string_int32 => { 1 => "I am not a number" })      end    end    def test_map_encode_decode      m = MapMessage.new(        :map_string_int32 => {"a" => 1, "b" => 2},        :map_string_msg => {"a" => TestMessage2.new(:foo => 1),                            "b" => TestMessage2.new(:foo => 2)})      m2 = MapMessage.decode(MapMessage.encode(m))      assert m == m2      m3 = MapMessageWireEquiv.decode(MapMessage.encode(m))      assert m3.map_string_int32.length == 2      kv = {}      m3.map_string_int32.map { |msg| kv[msg.key] = msg.value }      assert kv == {"a" => 1, "b" => 2}      kv = {}      m3.map_string_msg.map { |msg| kv[msg.key] = msg.value }      assert kv == {"a" => TestMessage2.new(:foo => 1),                    "b" => TestMessage2.new(:foo => 2)}    end    def test_oneof_descriptors      d = OneofMessage.descriptor      o = d.lookup_oneof("my_oneof")      assert o != nil      assert o.class == Google::Protobuf::OneofDescriptor      assert o.name == "my_oneof"      oneof_count = 0      d.each_oneof{ |oneof|        oneof_count += 1        assert oneof == o      }      assert oneof_count == 1      assert o.count == 4      field_names = o.map{|f| f.name}.sort      assert field_names == ["a", "b", "c", "d"]    end    def test_oneof      d = OneofMessage.new      assert d.a == nil      assert d.b == nil      assert d.c == nil      assert d.d == nil      assert d.my_oneof == nil      d.a = "hi"      assert d.a == "hi"      assert d.b == nil      assert d.c == nil      assert d.d == nil      assert d.my_oneof == :a      d.b = 42      assert d.a == nil      assert d.b == 42      assert d.c == nil      assert d.d == nil      assert d.my_oneof == :b      d.c = TestMessage2.new(:foo => 100)      assert d.a == nil      assert d.b == nil      assert d.c.foo == 100      assert d.d == nil      assert d.my_oneof == :c      d.d = :C      assert d.a == nil      assert d.b == nil      assert d.c == nil      assert d.d == :C      assert d.my_oneof == :d      d2 = OneofMessage.decode(OneofMessage.encode(d))      assert d2 == d      encoded_field_a = OneofMessage.encode(OneofMessage.new(:a => "string"))      encoded_field_b = OneofMessage.encode(OneofMessage.new(:b => 1000))      encoded_field_c = OneofMessage.encode(        OneofMessage.new(:c => TestMessage2.new(:foo => 1)))      encoded_field_d = OneofMessage.encode(OneofMessage.new(:d => :B))      d3 = OneofMessage.decode(        encoded_field_c + encoded_field_a + encoded_field_d)      assert d3.a == nil      assert d3.b == nil      assert d3.c == nil      assert d3.d == :B      d4 = OneofMessage.decode(        encoded_field_c + encoded_field_a + encoded_field_d +        encoded_field_c)      assert d4.a == nil      assert d4.b == nil      assert d4.c.foo == 1      assert d4.d == nil      d5 = OneofMessage.new(:a => "hello")      assert d5.a != nil      d5.a = nil      assert d5.a == nil      assert OneofMessage.encode(d5) == ''      assert d5.my_oneof == nil    end    def test_enum_field      m = TestMessage.new      assert m.optional_enum == :Default      m.optional_enum = :A      assert m.optional_enum == :A      assert_raise RangeError do        m.optional_enum = :ASDF      end      m.optional_enum = 1      assert m.optional_enum == :A      m.optional_enum = 100      assert m.optional_enum == 100    end    def test_dup      m = TestMessage.new      m.optional_string = "hello"      m.optional_int32 = 42      tm1 = TestMessage2.new(:foo => 100)      tm2 = TestMessage2.new(:foo => 200)      m.repeated_msg.push tm1      assert m.repeated_msg[-1] == tm1      m.repeated_msg.push tm2      assert m.repeated_msg[-1] == tm2      m2 = m.dup      assert m == m2      m.optional_int32 += 1      assert m != m2      assert m.repeated_msg[0] == m2.repeated_msg[0]      assert m.repeated_msg[0].object_id == m2.repeated_msg[0].object_id    end    def test_deep_copy      m = TestMessage.new(:optional_int32 => 42,                          :repeated_msg => [TestMessage2.new(:foo => 100)])      m2 = Google::Protobuf.deep_copy(m)      assert m == m2      assert m.repeated_msg == m2.repeated_msg      assert m.repeated_msg.object_id != m2.repeated_msg.object_id      assert m.repeated_msg[0].object_id != m2.repeated_msg[0].object_id    end    def test_eq      m = TestMessage.new(:optional_int32 => 42,                          :repeated_int32 => [1, 2, 3])      m2 = TestMessage.new(:optional_int32 => 43,                           :repeated_int32 => [1, 2, 3])      assert m != m2    end    def test_enum_lookup      assert TestEnum::A == 1      assert TestEnum::B == 2      assert TestEnum::C == 3      assert TestEnum::lookup(1) == :A      assert TestEnum::lookup(2) == :B      assert TestEnum::lookup(3) == :C      assert TestEnum::resolve(:A) == 1      assert TestEnum::resolve(:B) == 2      assert TestEnum::resolve(:C) == 3    end    def test_parse_serialize      m = TestMessage.new(:optional_int32 => 42,                          :optional_string => "hello world",                          :optional_enum => :B,                          :repeated_string => ["a", "b", "c"],                          :repeated_int32 => [42, 43, 44],                          :repeated_enum => [:A, :B, :C, 100],                          :repeated_msg => [TestMessage2.new(:foo => 1),                                            TestMessage2.new(:foo => 2)])      data = TestMessage.encode m      m2 = TestMessage.decode data      assert m == m2      data = Google::Protobuf.encode m      m2 = Google::Protobuf.decode(TestMessage, data)      assert m == m2    end    def test_def_errors      s = Google::Protobuf::DescriptorPool.new      assert_raise TypeError do        s.build do          # enum with no default (integer value 0)          add_enum "MyEnum" do            value :A, 1          end        end      end      assert_raise TypeError do        s.build do          # message with required field (unsupported in proto3)          add_message "MyMessage" do            required :foo, :int32, 1          end        end      end    end    def test_corecursive      # just be sure that we can instantiate types with corecursive field-type      # references.      m = Recursive1.new(:foo => Recursive2.new(:foo => Recursive1.new))      assert Recursive1.descriptor.lookup("foo").subtype ==        Recursive2.descriptor      assert Recursive2.descriptor.lookup("foo").subtype ==        Recursive1.descriptor      serialized = Recursive1.encode(m)      m2 = Recursive1.decode(serialized)      assert m == m2    end    def test_serialize_cycle      m = Recursive1.new(:foo => Recursive2.new)      m.foo.foo = m      assert_raise RuntimeError do        serialized = Recursive1.encode(m)      end    end    def test_bad_field_names      m = BadFieldNames.new(:dup => 1, :class => 2)      m2 = m.dup      assert m == m2      assert m['dup'] == 1      assert m['class'] == 2      m['dup'] = 3      assert m['dup'] == 3      m['a.b'] = 4      assert m['a.b'] == 4    end    def test_int_ranges      m = TestMessage.new      m.optional_int32 = 0      m.optional_int32 = -0x8000_0000      m.optional_int32 = +0x7fff_ffff      m.optional_int32 = 1.0      m.optional_int32 = -1.0      m.optional_int32 = 2e9      assert_raise RangeError do        m.optional_int32 = -0x8000_0001      end      assert_raise RangeError do        m.optional_int32 = +0x8000_0000      end      assert_raise RangeError do        m.optional_int32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum      end      assert_raise RangeError do        m.optional_int32 = 1e12      end      assert_raise RangeError do        m.optional_int32 = 1.5      end      m.optional_uint32 = 0      m.optional_uint32 = +0xffff_ffff      m.optional_uint32 = 1.0      m.optional_uint32 = 4e9      assert_raise RangeError do        m.optional_uint32 = -1      end      assert_raise RangeError do        m.optional_uint32 = -1.5      end      assert_raise RangeError do        m.optional_uint32 = -1.5e12      end      assert_raise RangeError do        m.optional_uint32 = -0x1000_0000_0000_0000      end      assert_raise RangeError do        m.optional_uint32 = +0x1_0000_0000      end      assert_raise RangeError do        m.optional_uint32 = +0x1000_0000_0000_0000_0000_0000 # force Bignum      end      assert_raise RangeError do        m.optional_uint32 = 1e12      end      assert_raise RangeError do        m.optional_uint32 = 1.5      end      m.optional_int64 = 0      m.optional_int64 = -0x8000_0000_0000_0000      m.optional_int64 = +0x7fff_ffff_ffff_ffff      m.optional_int64 = 1.0      m.optional_int64 = -1.0      m.optional_int64 = 8e18      m.optional_int64 = -8e18      assert_raise RangeError do        m.optional_int64 = -0x8000_0000_0000_0001      end      assert_raise RangeError do        m.optional_int64 = +0x8000_0000_0000_0000      end      assert_raise RangeError do        m.optional_int64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum      end      assert_raise RangeError do        m.optional_int64 = 1e50      end      assert_raise RangeError do        m.optional_int64 = 1.5      end      m.optional_uint64 = 0      m.optional_uint64 = +0xffff_ffff_ffff_ffff      m.optional_uint64 = 1.0      m.optional_uint64 = 16e18      assert_raise RangeError do        m.optional_uint64 = -1      end      assert_raise RangeError do        m.optional_uint64 = -1.5      end      assert_raise RangeError do        m.optional_uint64 = -1.5e12      end      assert_raise RangeError do        m.optional_uint64 = -0x1_0000_0000_0000_0000      end      assert_raise RangeError do        m.optional_uint64 = +0x1_0000_0000_0000_0000      end      assert_raise RangeError do        m.optional_uint64 = +0x1000_0000_0000_0000_0000_0000 # force Bignum      end      assert_raise RangeError do        m.optional_uint64 = 1e50      end      assert_raise RangeError do        m.optional_uint64 = 1.5      end    end    def test_stress_test      m = TestMessage.new      m.optional_int32 = 42      m.optional_int64 = 0x100000000      m.optional_string = "hello world"      10.times do m.repeated_msg.push TestMessage2.new(:foo => 42) end      10.times do m.repeated_string.push "hello world" end      data = TestMessage.encode(m)      l = 0      10_000.times do        m = TestMessage.decode(data)        data_new = TestMessage.encode(m)        assert data_new == data        data = data_new      end    end    def test_reflection      m = TestMessage.new(:optional_int32 => 1234)      msgdef = m.class.descriptor      assert msgdef.class == Google::Protobuf::Descriptor      assert msgdef.any? {|field| field.name == "optional_int32"}      optional_int32 = msgdef.lookup "optional_int32"      assert optional_int32.class == Google::Protobuf::FieldDescriptor      assert optional_int32 != nil      assert optional_int32.name == "optional_int32"      assert optional_int32.type == :int32      optional_int32.set(m, 5678)      assert m.optional_int32 == 5678      m.optional_int32 = 1000      assert optional_int32.get(m) == 1000      optional_msg = msgdef.lookup "optional_msg"      assert optional_msg.subtype == TestMessage2.descriptor      optional_msg.set(m, optional_msg.subtype.msgclass.new)      assert msgdef.msgclass == TestMessage      optional_enum = msgdef.lookup "optional_enum"      assert optional_enum.subtype == TestEnum.descriptor      assert optional_enum.subtype.class == Google::Protobuf::EnumDescriptor      optional_enum.subtype.each do |k, v|        # set with integer, check resolution to symbolic name        optional_enum.set(m, v)        assert optional_enum.get(m) == k      end    end    def test_json      skip("Unimplemented") if RUBY_PLATFORM == "java"      m = TestMessage.new(:optional_int32 => 1234,                          :optional_int64 => -0x1_0000_0000,                          :optional_uint32 => 0x8000_0000,                          :optional_uint64 => 0xffff_ffff_ffff_ffff,                          :optional_bool => true,                          :optional_float => 1.0,                          :optional_double => -1e100,                          :optional_string => "Test string",                          :optional_bytes => ["FFFFFFFF"].pack('H*'),                          :optional_msg => TestMessage2.new(:foo => 42),                          :repeated_int32 => [1, 2, 3, 4],                          :repeated_string => ["a", "b", "c"],                          :repeated_bool => [true, false, true, false],                          :repeated_msg => [TestMessage2.new(:foo => 1),                                            TestMessage2.new(:foo => 2)])      json_text = TestMessage.encode_json(m)      m2 = TestMessage.decode_json(json_text)      assert m == m2      # Crash case from GitHub issue 283.      bar = Bar.new(msg: "bar")      baz1 = Baz.new(msg: "baz")      baz2 = Baz.new(msg: "quux")      Foo.encode_json(Foo.new)      Foo.encode_json(Foo.new(bar: bar))      Foo.encode_json(Foo.new(bar: bar, baz: [baz1, baz2]))    end    def test_json_maps      skip("Unimplemented") if RUBY_PLATFORM == "java"      m = MapMessage.new(:map_string_int32 => {"a" => 1})      expected = '{"map_string_int32":{"a":1},"map_string_msg":{}}'      assert MapMessage.encode_json(m) == expected      m2 = MapMessage.decode_json(MapMessage.encode_json(m))      assert m == m2    end  endend
 |